def __init__(self, parent, model=None, view=None):
if view is None:
self.view = FittingPlotView(parent)
else:
self.view = view
if model is None:
self.model = FittingPlotModel()
else:
self.model = model
self.workspace_added_observer = GenericObserverWithArgPassing(self.add_workspace_to_plot)
self.workspace_removed_observer = GenericObserverWithArgPassing(self.remove_workspace_from_plot)
self.all_workspaces_removed_observer = GenericObserver(self.clear_plot)
class FittingPlotPresenter(object):
def __init__(self, parent, model=None, view=None):
if view is None:
self.view = FittingPlotView(parent)
else:
self.view = view
if model is None:
self.model = FittingPlotModel()
else:
self.model = model
self.workspace_added_observer = GenericObserverWithArgPassing(self.add_workspace_to_plot)
self.workspace_removed_observer = GenericObserverWithArgPassing(self.remove_workspace_from_plot)
self.all_workspaces_removed_observer = GenericObserver(self.clear_plot)
def add_workspace_to_plot(self, ws):
axes = self.view.get_axes()
for ax in axes:
self.model.add_workspace_to_plot(ws, ax, PLOT_KWARGS)
self.view.update_figure()
def remove_workspace_from_plot(self, ws):
for ax in self.view.get_axes():
self.model.remove_workspace_from_plot(ws, ax)
self.view.update_figure()
def clear_plot(self):
for ax in self.view.get_axes():
self.model.remove_all_workspaces_from_plot(ax)
self.view.clear_figure()
class FittingPlotPresenter(object):
def __init__(self, parent, model=None, view=None):
if view is None:
self.view = FittingPlotView(parent)
else:
self.view = view
if model is None:
self.model = FittingPlotModel()
else:
self.model = model
self.workspace_added_observer = GenericObserverWithArgPassing(
self.add_workspace_to_plot)
self.workspace_removed_observer = GenericObserverWithArgPassing(
self.remove_workspace_from_plot)
self.all_workspaces_removed_observer = GenericObserver(self.clear_plot)
self.fit_all_started_observer = GenericObserverWithArgPassing(
self.do_fit_all)
self.fit_all_done_notifier = GenericObservable()
def add_workspace_to_plot(self, ws):
axes = self.view.get_axes()
for ax in axes:
self.model.add_workspace_to_plot(ws, ax, PLOT_KWARGS)
self.view.update_figure()
def remove_workspace_from_plot(self, ws):
for ax in self.view.get_axes():
self.model.remove_workspace_from_plot(ws, ax)
self.view.remove_ws_from_fitbrowser(ws)
self.view.update_figure()
def clear_plot(self):
for ax in self.view.get_axes():
self.model.remove_all_workspaces_from_plot(ax)
self.view.clear_figure()
self.view.update_fitbrowser()
def do_fit_all(self, ws_list, do_sequential=True):
fitprop_list = []
prev_fitprop = self.view.read_fitprop_from_browser()
for ws in ws_list:
logger.notice(f'Starting to fit workspace {ws}')
fitprop = deepcopy(prev_fitprop)
# update I/O workspace name
fitprop['properties']['Output'] = ws
fitprop['properties']['InputWorkspace'] = ws
# do fit
fit_output = Fit(**fitprop['properties'])
# update results
fitprop['status'] = fit_output.OutputStatus
funcstr = str(fit_output.Function.fun)
fitprop['properties']['Function'] = funcstr
if "success" in fitprop['status'].lower() and do_sequential:
# update function in prev fitprop to use for next workspace
prev_fitprop['properties']['Function'] = funcstr
# update last fit in fit browser and save setup
self.view.update_browser(fit_output.OutputStatus, funcstr, ws)
# append a deep copy to output list (will be initial parameters if not successful)
fitprop_list.append(fitprop)
logger.notice('Sequential fitting finished.')
self.fit_all_done_notifier.notify_subscribers(fitprop_list)