Changeset 5330be2 in sasview for src/sas/qtgui/Perspectives/Fitting

Timestamp:

Jul 31, 2018 6:06:15 AM (6 years ago)

Author:

Piotr Rozyczko <rozyczko@…>

Branches:

ESS_GUI, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc

Children:

4453043

Parents:

933f9e9

git-author:

Piotr Rozyczko <rozyczko@…> (07/31/18 06:04:46)

git-committer:

Piotr Rozyczko <rozyczko@…> (07/31/18 06:06:15)

Message:

Fix 2D chart failures with reporting. SASVIEW-950

File:

• src/sas/qtgui/Perspectives/Fitting/ReportPageLogic.py (modified) (4 diffs)

src/sas/qtgui/Perspectives/Fitting/ReportPageLogic.py

@@ -4 +4 @@
 import re
 import sys
+import tempfile
+
 import logging
 from io import BytesIO
@@ -9 +11 @@
 
 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
-
-from sas.qtgui.Plotting.Plotter import Plotter
-from sas.qtgui.Plotting.Plotter2D import Plotter2D
-from sas.qtgui.Plotting.PlotterData import Data1D
-from sas.qtgui.Plotting.PlotterData import Data2D
 
 import sas.qtgui.Utilities.GuiUtils as GuiUtils
@@ -102 +99 @@
             canvas = FigureCanvas(fig)
             png_output = BytesIO()
-            canvas.print_png(png_output)
-            data = png_output.getvalue()
-            data64 = base64.b64encode(data)
+
+            # Create a "safe" location - system tmp
+            tmp_file = tempfile.TemporaryFile(suffix=".png")
+            try:
+                fig.savefig(tmp_file.name, dpi=75)
+                fig.savefig(png_output, dpi=75)
+            except PermissionError:
+                # sometimes one gets "permission denied" for temp files
+                # mainly on Windows 7 *gasp*. Let's try local directory
+                tmp_file = open("_tmp.png", "w+")
+                try:
+                    fig.savefig(tmp_file.name, dpi=75)
+                    fig.savefig(png_output, dpi=75)
+                except Exception as ex:
+                    logging.error("Creating of the report failed: %s"%str(ex))
+                    return
+
+            data_to_print = png_output.getvalue() == open(tmp_file.name, 'rb').read()
+            tmp_file.close()
+            data64 = base64.b64encode(png_output.getvalue())
             data_to_print = urllib.parse.quote(data64)
-            feet=FEET_2
+
+            feet = FEET_2
             if sys.platform == "darwin":  # Mac
                 feet = FEET_3
@@ -153 +168 @@
         if not modelname or self._index is None:
             return None
-        plots = GuiUtils.plotsFromModel(modelname, self._index)
-        # Call show on requested plots
-        # All same-type charts in one plot
-        for plot_set in plots:
-            if isinstance(plot_set, Data1D):
-                if 'residuals' in plot_set.title.lower():
-                    res_plot = Plotter(self, quickplot=True)
-                    res_plot.plot(plot_set)
-                    graphs.append(res_plot.figure)
-                    continue
-                if not 'new_plot' in locals():
-                    new_plot = Plotter(self, quickplot=True)
-                new_plot.plot(plot_set)
-            elif isinstance(plot_set, Data2D):
-                plot2D = Plotter2D(self, quickplot=True)
-                plot2D.item = self._index
-                plot2D.plot(plot_set)
-                graphs.append(plot2D.figure)
-
-            else:
-                msg = "Incorrect data type passed to Plotting"
-                raise AttributeError(msg)
-
-        if 'new_plot' in locals() and isinstance(new_plot.data, Data1D):
-            graphs.append(new_plot.figure)
+        plot_ids = [plot.id for plot in GuiUtils.plotsFromModel(modelname, self._index)]
+
+        # Active plots
+        import sas.qtgui.Plotting.PlotHelper as PlotHelper
+        shown_plot_names = PlotHelper.currentPlots()
+
+        # current_plots = list of graph names of currently shown plots
+        # which are related to this dataset
+        current_plots = [name for name in shown_plot_names if PlotHelper.plotById(name).data.id in plot_ids]
+
+        for name in current_plots:
+            # get the plotter object first
+            plotter = PlotHelper.plotById(name)
+            graphs.append(plotter.figure)
 
         return graphs