source: sasview/src/sas/sasgui/perspectives/fitting/resultpanel.py @ a8d79fd

"""
FitPanel class contains fields allowing to fit  models and  data

:note: For Fit to be performed the user should check at least one parameter
    on fit Panel window.

"""
import wx
import wx.lib.newevent
from wx.aui import AuiNotebook as Notebook

import datetime

from bumps.gui.convergence_view import ConvergenceView
from bumps.gui.uncertainty_view import UncertaintyView, CorrelationView, TraceView
from bumps.dream.stats import var_stats, format_vars

from sas.sasgui.guiframe.panel_base import PanelBase
from sas.sasgui.guiframe.events import StatusEvent

(PlotResultEvent, EVT_PLOT_RESULT) = wx.lib.newevent.NewEvent()


class ResultPanel(Notebook, PanelBase):
    """
    FitPanel class contains fields allowing to fit  models and  data

    :note: For Fit to be performed the user should check at least one parameter
        on fit Panel window.

    """
    ## Internal name for the AUI manager
    window_name = "Result panel"
    ## Title to appear on top of the window
    window_caption = "Result Panel"
    CENTER_PANE = True

    def __init__(self, parent, manager=None, *args, **kwargs):
        """
        """
        style = ((wx.aui.AUI_NB_WINDOWLIST_BUTTON
                 | wx.aui.AUI_NB_DEFAULT_STYLE
                 | wx.CLIP_CHILDREN)
                 & ~wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB)
        Notebook.__init__(self, parent, wx.ID_ANY, style=style)
        PanelBase.__init__(self, parent)
        self.frame = parent
        self.Bind(EVT_PLOT_RESULT, self.on_plot_results)
        self.frame.Bind(wx.EVT_CLOSE, self.on_close)
        self._manager = None

    def on_close(self, event):
        if event.CanVeto():
            self.frame.Hide()
            event.Veto()
        else:
            event.Skip()

    def on_plot_results(self, event):
        self.frame.Show(True)
        result = event.result[0][0]
        filename = result.data.sas_data.filename
        current_time = datetime.datetime.now().strftime("%I:%M%p, %B %d, %Y")
        self.parent.SetTitle(self.window_name + " - " + filename + " - " + current_time)
        if hasattr(result, 'convergence'):
            best, pop = result.convergence[:, 0], result.convergence[:, 1:]
            self._get_view(ConvergenceView).update(best, pop)
        else:
            self._del_view(ConvergenceView)
        if hasattr(result, 'uncertainty_state'):
            stats = var_stats(result.uncertainty_state.draw())
            msg = format_vars(stats)
            self._get_view(CorrelationView).update(result.uncertainty_state)
            self._get_view(UncertaintyView).update((result.uncertainty_state, stats))
            self._get_view(TraceView).update(result.uncertainty_state)
            # TODO: stats should be stored in result rather than computed in bumps UncertaintyView
            wx.PostEvent(self.frame.parent,
                         StatusEvent(status=msg, info="info"))
        else:
            for view in (CorrelationView, UncertaintyView, TraceView):
                self._del_view(view)

    def get_frame(self):
        return self.frame

    def _get_view(self, view_class):
        for idx in range(self.PageCount):
            if self.GetPageText(idx) == view_class.title:
                return self.GetPage(idx)
        else:
            panel = view_class(self)
            self.AddPage(panel, panel.title)
            return panel

    def _del_view(self, view_class):
        for idx in range(self.PageCount):
            if self.GetPageText(idx) == view_class.title:
                self.DeletePage(idx)