Changeset d0ce666f in sasview for src/sas/sasgui/guiframe

Timestamp:

Sep 6, 2018 12:11:25 PM (6 years ago)

Author:

krzywon

Branches:

master, magnetic_scatt, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, unittest-saveload

Children:

44e8f48

Parents:

fa412df

Message:

Remove image references when report is closed and point to correct file locations when saving reports to PDF.

Location:

src/sas/sasgui/guiframe

Files:

• report_dialog.py (modified) (4 diffs)
• report_image_handler.py (modified) (3 diffs)

src/sas/sasgui/guiframe/report_dialog.py

@@ -7 +7 @@
 import sys
 import wx.html as html
+from sas.sasgui.guiframe.report_image_handler import ReportImageHandler
 
 logger = logging.getLogger(__name__)
@@ -70 +71 @@
         button_close = wx.Button(self, wx.ID_OK, "Close")
         button_close.SetToolTipString("Close this report window.")
+        button_close.Bind(wx.EVT_BUTTON, self.onClose,
+                          id=button_close.GetId())
         hbox.Add(button_close)
         button_close.SetFocus()
@@ -117 +120 @@
 
 
-    def OnClose(self, event=None):
+    def onClose(self, event=None):
         """
         Close the Dialog
@@ -123 +126 @@
         """
         for fig in self.fig_urls:
-            self.imgRAM.RemoveFile(fig)
+            ReportImageHandler.remove_figure(fig)
 
-        self.Close()
+        self.Destroy()
 
     def HTML2PDF(self, data, filename):

src/sas/sasgui/guiframe/report_image_handler.py

@@ -1 +1 @@
 import wx
+import logging
+
+logger = logging.getLogger(__name__)
 
 
@@ -9 +12 @@
 
         def __init__(self):
+            wx.FileSystem.AddHandler(wx.MemoryFSHandler())
             self.img_holder = wx.MemoryFSHandler()
-            wx.FileSystem.AddHandler(wx.MemoryFSHandler())
             self.refs = {}
             self.indices = []
@@ -60 +63 @@
 
     @staticmethod
+    def check_for_empty_instance():
+        if ReportImageHandler.instance is None:
+            ReportImageHandler.instance = \
+                ReportImageHandler._ReportImageHandler()
+
+    @staticmethod
     def set_figs(figs, bitmaps, perspective):
-        if not ReportImageHandler.instance:
-            ReportImageHandler.instance = ReportImageHandler._ReportImageHandler()
+        ReportImageHandler.check_for_empty_instance()
         return ReportImageHandler.instance.set_figs(figs, bitmaps, perspective)
+
+    @staticmethod
+    def remove_figure(fig_url):
+        try:
+            ReportImageHandler.check_for_empty_instance()
+            ReportImageHandler.instance.refs.pop(fig_url)
+        except Exception as e:
+            logger.warn(e)