source: sasview/src/sas/sasgui/guiframe/report_image_handler.py @ d0ce666f

import wx
import logging

logger = logging.getLogger(__name__)


class ReportImageHandler:
    # Singleton class that manages all report plot images
    # To call the handler, call the static method set_figs

    class _ReportImageHandler:

        def __init__(self):
            wx.FileSystem.AddHandler(wx.MemoryFSHandler())
            self.img_holder = wx.MemoryFSHandler()
            self.refs = {}
            self.indices = []

        def set_figs(self, figs, bitmaps, perspective):
            """
            Save figures and images to memory and return refernces
            :param figs: A list of matplotlib Figures
            :param bitmaps: A list of bitmaps
            :param perspective: A String with the perspective name
            :return: A tuple of a list of Figures and a list of memory refs
            """
            imgs = []
            refs = []
            if figs is None or len(figs) == 0:
                figs = [None]
            for fig in figs:
                if fig is not None:
                    ind = figs.index(fig)
                    bitmap = bitmaps[ind]

                # name of the fig
                name = self.create_unique_name(perspective)
                # AddFile, image can be retrieved with 'memory:filename'
                ref = 'memory:' + name
                self.refs[ref] = fig
                self.img_holder.AddFile(name, bitmap, wx.BITMAP_TYPE_PNG)
                refs.append(ref)
                imgs.append(fig)
            return imgs, refs

        def create_unique_name(self, perspective, index=None):
            """
            Create a unique key for each item in memory
            :param perspective: The perspective name as a string
            :param index: The base index used for incrementing the name
            :return: A unique file name not currently in use
            """
            if not index:
                index = len(self.indices)
            if index in self.indices:
                name = self.create_unique_name(index + 1)
            else:
                self.indices.append(index)
                name = 'img_{}_{:03d}.png'.format(str(perspective), index)
            return name

    instance = None

    @staticmethod
    def check_for_empty_instance():
        if ReportImageHandler.instance is None:
            ReportImageHandler.instance = \
                ReportImageHandler._ReportImageHandler()

    @staticmethod
    def set_figs(figs, bitmaps, perspective):
        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)