updated to use np.array(fig.canvas.buffer_rgba() and changed reshape …

…to 4 channel
usnistgov · Dec 22, 2024 · fdaa935 · fdaa935
1 parent 0341c10
commit fdaa935
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/NRSS/visualizer.py b/src/NRSS/visualizer.py
@@ -592,8 +592,8 @@ def morphology_visualizer(
 
             if outputmat and (i in outputmat):
                 fig.canvas.draw()
-                rgb_return = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-                rgb_return = rgb_return.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+                rgb_return = np.array(fig.canvas.buffer_rgba(), dtype=np.uint8)
+                rgb_return = rgb_return.reshape(fig.canvas.get_width_height()[::-1] + (4,))
                 if outputplot and ("vfrac" in outputplot):
                     if outputaxes:
                         fig_xl = ax1.get_tightbbox().intervalx[0].astype(int)