source: sasview/src/sas/qtgui/Perspectives/Fitting/FittingWidget.py @ 6b50296

ESS_GUIESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc
Last change on this file since 6b50296 was 6b50296, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago

Added freeze action for inner main model datasets. SASVIEW-1002

  • Property mode set to 100644
File size: 120.5 KB
Line 
1import json
2import os
3from collections import defaultdict
4
5import copy
6import logging
7import traceback
8from twisted.internet import threads
9import numpy as np
10import webbrowser
11
12from PyQt5 import QtCore
13from PyQt5 import QtGui
14from PyQt5 import QtWidgets
15
16from sasmodels import generate
17from sasmodels import modelinfo
18from sasmodels.sasview_model import load_standard_models
19from sasmodels.sasview_model import MultiplicationModel
20from sasmodels.weights import MODELS as POLYDISPERSITY_MODELS
21
22from sas.sascalc.fit.BumpsFitting import BumpsFit as Fit
23from sas.sascalc.fit.pagestate import PageState
24
25import sas.qtgui.Utilities.GuiUtils as GuiUtils
26import sas.qtgui.Utilities.LocalConfig as LocalConfig
27from sas.qtgui.Utilities.CategoryInstaller import CategoryInstaller
28from sas.qtgui.Plotting.PlotterData import Data1D
29from sas.qtgui.Plotting.PlotterData import Data2D
30
31from sas.qtgui.Perspectives.Fitting.UI.FittingWidgetUI import Ui_FittingWidgetUI
32from sas.qtgui.Perspectives.Fitting.FitThread import FitThread
33from sas.qtgui.Perspectives.Fitting.ConsoleUpdate import ConsoleUpdate
34
35from sas.qtgui.Perspectives.Fitting.ModelThread import Calc1D
36from sas.qtgui.Perspectives.Fitting.ModelThread import Calc2D
37from sas.qtgui.Perspectives.Fitting.FittingLogic import FittingLogic
38from sas.qtgui.Perspectives.Fitting import FittingUtilities
39from sas.qtgui.Perspectives.Fitting import ModelUtilities
40from sas.qtgui.Perspectives.Fitting.SmearingWidget import SmearingWidget
41from sas.qtgui.Perspectives.Fitting.OptionsWidget import OptionsWidget
42from sas.qtgui.Perspectives.Fitting.FitPage import FitPage
43from sas.qtgui.Perspectives.Fitting.ViewDelegate import ModelViewDelegate
44from sas.qtgui.Perspectives.Fitting.ViewDelegate import PolyViewDelegate
45from sas.qtgui.Perspectives.Fitting.ViewDelegate import MagnetismViewDelegate
46from sas.qtgui.Perspectives.Fitting.Constraint import Constraint
47from sas.qtgui.Perspectives.Fitting.MultiConstraint import MultiConstraint
48from sas.qtgui.Perspectives.Fitting.ReportPageLogic import ReportPageLogic
49
50
51
52TAB_MAGNETISM = 4
53TAB_POLY = 3
54CATEGORY_DEFAULT = "Choose category..."
55CATEGORY_STRUCTURE = "Structure Factor"
56CATEGORY_CUSTOM = "Plugin Models"
57STRUCTURE_DEFAULT = "None"
58
59DEFAULT_POLYDISP_FUNCTION = 'gaussian'
60
61
62logger = logging.getLogger(__name__)
63
64
65class ToolTippedItemModel(QtGui.QStandardItemModel):
66    """
67    Subclass from QStandardItemModel to allow displaying tooltips in
68    QTableView model.
69    """
70    def __init__(self, parent=None):
71        QtGui.QStandardItemModel.__init__(self, parent)
72
73    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
74        """
75        Displays tooltip for each column's header
76        :param section:
77        :param orientation:
78        :param role:
79        :return:
80        """
81        if role == QtCore.Qt.ToolTipRole:
82            if orientation == QtCore.Qt.Horizontal:
83                return str(self.header_tooltips[section])
84
85        return QtGui.QStandardItemModel.headerData(self, section, orientation, role)
86
87class FittingWidget(QtWidgets.QWidget, Ui_FittingWidgetUI):
88    """
89    Main widget for selecting form and structure factor models
90    """
91    constraintAddedSignal = QtCore.pyqtSignal(list)
92    newModelSignal = QtCore.pyqtSignal()
93    fittingFinishedSignal = QtCore.pyqtSignal(tuple)
94    batchFittingFinishedSignal = QtCore.pyqtSignal(tuple)
95    Calc1DFinishedSignal = QtCore.pyqtSignal(tuple)
96    Calc2DFinishedSignal = QtCore.pyqtSignal(tuple)
97
98    def __init__(self, parent=None, data=None, tab_id=1):
99
100        super(FittingWidget, self).__init__()
101
102        # Necessary globals
103        self.parent = parent
104
105        # Which tab is this widget displayed in?
106        self.tab_id = tab_id
107
108        # Globals
109        self.initializeGlobals()
110
111        # data index for the batch set
112        self.data_index = 0
113        # Main Data[12]D holders
114        # Logics.data contains a single Data1D/Data2D object
115        self._logic = [FittingLogic()]
116
117        # Main GUI setup up
118        self.setupUi(self)
119        self.setWindowTitle("Fitting")
120
121        # Set up tabs widgets
122        self.initializeWidgets()
123
124        # Set up models and views
125        self.initializeModels()
126
127        # Defaults for the structure factors
128        self.setDefaultStructureCombo()
129
130        # Make structure factor and model CBs disabled
131        self.disableModelCombo()
132        self.disableStructureCombo()
133
134        # Generate the category list for display
135        self.initializeCategoryCombo()
136
137        # Initial control state
138        self.initializeControls()
139
140        QtWidgets.QApplication.processEvents()
141
142        # Connect signals to controls
143        self.initializeSignals()
144
145        if data is not None:
146            self.data = data
147
148        # New font to display angstrom symbol
149        new_font = 'font-family: -apple-system, "Helvetica Neue", "Ubuntu";'
150        self.label_17.setStyleSheet(new_font)
151        self.label_19.setStyleSheet(new_font)
152
153    @property
154    def logic(self):
155        # make sure the logic contains at least one element
156        assert self._logic
157        # logic connected to the currently shown data
158        return self._logic[self.data_index]
159
160    @property
161    def data(self):
162        return self.logic.data
163
164    @data.setter
165    def data(self, value):
166        """ data setter """
167        # Value is either a list of indices for batch fitting or a simple index
168        # for standard fitting. Assure we have a list, regardless.
169        if isinstance(value, list):
170            self.is_batch_fitting = True
171        else:
172            value = [value]
173
174        assert isinstance(value[0], QtGui.QStandardItem)
175
176        # Keep reference to all datasets for batch
177        self.all_data = value
178
179        # Create logics with data items
180        # Logics.data contains only a single Data1D/Data2D object
181        if len(value) == 1:
182            # single data logic is already defined, update data on it
183            self._logic[0].data = GuiUtils.dataFromItem(value[0])
184        else:
185            # batch datasets
186            self._logic = []
187            for data_item in value:
188                logic = FittingLogic(data=GuiUtils.dataFromItem(data_item))
189                self._logic.append(logic)
190
191        # Overwrite data type descriptor
192        self.is2D = True if isinstance(self.logic.data, Data2D) else False
193
194        # Let others know we're full of data now
195        self.data_is_loaded = True
196
197        # Enable/disable UI components
198        self.setEnablementOnDataLoad()
199
200    def initializeGlobals(self):
201        """
202        Initialize global variables used in this class
203        """
204        # SasModel is loaded
205        self.model_is_loaded = False
206        # Data[12]D passed and set
207        self.data_is_loaded = False
208        # Batch/single fitting
209        self.is_batch_fitting = False
210        self.is_chain_fitting = False
211        # Is the fit job running?
212        self.fit_started = False
213        # The current fit thread
214        self.calc_fit = None
215        # Current SasModel in view
216        self.kernel_module = None
217        # Current SasModel view dimension
218        self.is2D = False
219        # Current SasModel is multishell
220        self.model_has_shells = False
221        # Utility variable to enable unselectable option in category combobox
222        self._previous_category_index = 0
223        # Utility variable for multishell display
224        self._last_model_row = 0
225        # Dictionary of {model name: model class} for the current category
226        self.models = {}
227        # Parameters to fit
228        self.main_params_to_fit = []
229        self.poly_params_to_fit = []
230        self.magnet_params_to_fit = []
231
232        # Fit options
233        self.q_range_min = 0.005
234        self.q_range_max = 0.1
235        self.npts = 25
236        self.log_points = False
237        self.weighting = 0
238        self.chi2 = None
239        # Does the control support UNDO/REDO
240        # temporarily off
241        self.undo_supported = False
242        self.page_stack = []
243        self.all_data = []
244        # custom plugin models
245        # {model.name:model}
246        self.custom_models = self.customModels()
247        # Polydisp widget table default index for function combobox
248        self.orig_poly_index = 3
249        # copy of current kernel model
250        self.kernel_module_copy = None
251
252        # Page id for fitting
253        # To keep with previous SasView values, use 200 as the start offset
254        self.page_id = 200 + self.tab_id
255
256        # Data for chosen model
257        self.model_data = None
258
259        # Which shell is being currently displayed?
260        self.current_shell_displayed = 0
261        # List of all shell-unique parameters
262        self.shell_names = []
263
264        # Error column presence in parameter display
265        self.has_error_column = False
266        self.has_poly_error_column = False
267        self.has_magnet_error_column = False
268
269        # If the widget generated theory item, save it
270        self.theory_item = None
271
272        # signal communicator
273        self.communicate = self.parent.communicate
274
275    def initializeWidgets(self):
276        """
277        Initialize widgets for tabs
278        """
279        # Options widget
280        layout = QtWidgets.QGridLayout()
281        self.options_widget = OptionsWidget(self, self.logic)
282        layout.addWidget(self.options_widget)
283        self.tabOptions.setLayout(layout)
284
285        # Smearing widget
286        layout = QtWidgets.QGridLayout()
287        self.smearing_widget = SmearingWidget(self)
288        layout.addWidget(self.smearing_widget)
289        self.tabResolution.setLayout(layout)
290
291        # Define bold font for use in various controls
292        self.boldFont = QtGui.QFont()
293        self.boldFont.setBold(True)
294
295        # Set data label
296        self.label.setFont(self.boldFont)
297        self.label.setText("No data loaded")
298        self.lblFilename.setText("")
299
300        # Magnetic angles explained in one picture
301        self.magneticAnglesWidget = QtWidgets.QWidget()
302        labl = QtWidgets.QLabel(self.magneticAnglesWidget)
303        pixmap = QtGui.QPixmap(GuiUtils.IMAGES_DIRECTORY_LOCATION + '/M_angles_pic.bmp')
304        labl.setPixmap(pixmap)
305        self.magneticAnglesWidget.setFixedSize(pixmap.width(), pixmap.height())
306
307    def initializeModels(self):
308        """
309        Set up models and views
310        """
311        # Set the main models
312        # We can't use a single model here, due to restrictions on flattening
313        # the model tree with subclassed QAbstractProxyModel...
314        self._model_model = ToolTippedItemModel()
315        self._poly_model = ToolTippedItemModel()
316        self._magnet_model = ToolTippedItemModel()
317
318        # Param model displayed in param list
319        self.lstParams.setModel(self._model_model)
320        self.readCategoryInfo()
321
322        self.model_parameters = None
323
324        # Delegates for custom editing and display
325        self.lstParams.setItemDelegate(ModelViewDelegate(self))
326
327        self.lstParams.setAlternatingRowColors(True)
328        stylesheet = """
329
330            QTreeView {
331                paint-alternating-row-colors-for-empty-area:0;
332            }
333
334            QTreeView::item {
335                border: 1px;
336                padding: 2px 1px;
337            }
338
339            QTreeView::item:hover {
340                background: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #e7effd, stop: 1 #cbdaf1);
341                border: 1px solid #bfcde4;
342            }
343
344            QTreeView::item:selected {
345                border: 1px solid #567dbc;
346            }
347
348            QTreeView::item:selected:active{
349                background: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #6ea1f1, stop: 1 #567dbc);
350            }
351
352            QTreeView::item:selected:!active {
353                background: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #6b9be8, stop: 1 #577fbf);
354            }
355           """
356        self.lstParams.setStyleSheet(stylesheet)
357        self.lstParams.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
358        self.lstParams.customContextMenuRequested.connect(self.showModelContextMenu)
359        self.lstParams.setAttribute(QtCore.Qt.WA_MacShowFocusRect, False)
360        # Poly model displayed in poly list
361        self.lstPoly.setModel(self._poly_model)
362        self.setPolyModel()
363        self.setTableProperties(self.lstPoly)
364        # Delegates for custom editing and display
365        self.lstPoly.setItemDelegate(PolyViewDelegate(self))
366        # Polydispersity function combo response
367        self.lstPoly.itemDelegate().combo_updated.connect(self.onPolyComboIndexChange)
368        self.lstPoly.itemDelegate().filename_updated.connect(self.onPolyFilenameChange)
369
370        # Magnetism model displayed in magnetism list
371        self.lstMagnetic.setModel(self._magnet_model)
372        self.setMagneticModel()
373        self.setTableProperties(self.lstMagnetic)
374        # Delegates for custom editing and display
375        self.lstMagnetic.setItemDelegate(MagnetismViewDelegate(self))
376
377    def initializeCategoryCombo(self):
378        """
379        Model category combo setup
380        """
381        category_list = sorted(self.master_category_dict.keys())
382        self.cbCategory.addItem(CATEGORY_DEFAULT)
383        self.cbCategory.addItems(category_list)
384        if CATEGORY_STRUCTURE not in category_list:
385            self.cbCategory.addItem(CATEGORY_STRUCTURE)
386        self.cbCategory.setCurrentIndex(0)
387
388    def setEnablementOnDataLoad(self):
389        """
390        Enable/disable various UI elements based on data loaded
391        """
392        # Tag along functionality
393        self.label.setText("Data loaded from: ")
394        if self.logic.data.filename:
395            self.lblFilename.setText(self.logic.data.filename)
396        else:
397            self.lblFilename.setText(self.logic.data.name)
398        self.updateQRange()
399        # Switch off Data2D control
400        self.chk2DView.setEnabled(False)
401        self.chk2DView.setVisible(False)
402        self.chkMagnetism.setEnabled(self.is2D)
403        self.tabFitting.setTabEnabled(TAB_MAGNETISM, self.chkMagnetism.isChecked())
404        # Combo box or label for file name"
405        if self.is_batch_fitting:
406            self.lblFilename.setVisible(False)
407            for dataitem in self.all_data:
408                filename = GuiUtils.dataFromItem(dataitem).filename
409                self.cbFileNames.addItem(filename)
410            self.cbFileNames.setVisible(True)
411            self.chkChainFit.setEnabled(True)
412            self.chkChainFit.setVisible(True)
413            # This panel is not designed to view individual fits, so disable plotting
414            self.cmdPlot.setVisible(False)
415        # Similarly on other tabs
416        self.options_widget.setEnablementOnDataLoad()
417        self.onSelectModel()
418        # Smearing tab
419        self.smearing_widget.updateData(self.data)
420
421    def acceptsData(self):
422        """ Tells the caller this widget can accept new dataset """
423        return not self.data_is_loaded
424
425    def disableModelCombo(self):
426        """ Disable the combobox """
427        self.cbModel.setEnabled(False)
428        self.lblModel.setEnabled(False)
429
430    def enableModelCombo(self):
431        """ Enable the combobox """
432        self.cbModel.setEnabled(True)
433        self.lblModel.setEnabled(True)
434
435    def disableStructureCombo(self):
436        """ Disable the combobox """
437        self.cbStructureFactor.setEnabled(False)
438        self.lblStructure.setEnabled(False)
439
440    def enableStructureCombo(self):
441        """ Enable the combobox """
442        self.cbStructureFactor.setEnabled(True)
443        self.lblStructure.setEnabled(True)
444
445    def togglePoly(self, isChecked):
446        """ Enable/disable the polydispersity tab """
447        self.tabFitting.setTabEnabled(TAB_POLY, isChecked)
448
449    def toggleMagnetism(self, isChecked):
450        """ Enable/disable the magnetism tab """
451        self.tabFitting.setTabEnabled(TAB_MAGNETISM, isChecked)
452
453    def toggleChainFit(self, isChecked):
454        """ Enable/disable chain fitting """
455        self.is_chain_fitting = isChecked
456
457    def toggle2D(self, isChecked):
458        """ Enable/disable the controls dependent on 1D/2D data instance """
459        self.chkMagnetism.setEnabled(isChecked)
460        self.is2D = isChecked
461        # Reload the current model
462        if self.kernel_module:
463            self.onSelectModel()
464
465    @classmethod
466    def customModels(cls):
467        """ Reads in file names in the custom plugin directory """
468        return ModelUtilities._find_models()
469
470    def initializeControls(self):
471        """
472        Set initial control enablement
473        """
474        self.cbFileNames.setVisible(False)
475        self.cmdFit.setEnabled(False)
476        self.cmdPlot.setEnabled(False)
477        self.options_widget.cmdComputePoints.setVisible(False) # probably redundant
478        self.chkPolydispersity.setEnabled(True)
479        self.chkPolydispersity.setCheckState(False)
480        self.chk2DView.setEnabled(True)
481        self.chk2DView.setCheckState(False)
482        self.chkMagnetism.setEnabled(False)
483        self.chkMagnetism.setCheckState(False)
484        self.chkChainFit.setEnabled(False)
485        self.chkChainFit.setVisible(False)
486        # Tabs
487        self.tabFitting.setTabEnabled(TAB_POLY, False)
488        self.tabFitting.setTabEnabled(TAB_MAGNETISM, False)
489        self.lblChi2Value.setText("---")
490        # Smearing tab
491        self.smearing_widget.updateData(self.data)
492        # Line edits in the option tab
493        self.updateQRange()
494
495    def initializeSignals(self):
496        """
497        Connect GUI element signals
498        """
499        # Comboboxes
500        self.cbStructureFactor.currentIndexChanged.connect(self.onSelectStructureFactor)
501        self.cbCategory.currentIndexChanged.connect(self.onSelectCategory)
502        self.cbModel.currentIndexChanged.connect(self.onSelectModel)
503        self.cbFileNames.currentIndexChanged.connect(self.onSelectBatchFilename)
504        # Checkboxes
505        self.chk2DView.toggled.connect(self.toggle2D)
506        self.chkPolydispersity.toggled.connect(self.togglePoly)
507        self.chkMagnetism.toggled.connect(self.toggleMagnetism)
508        self.chkChainFit.toggled.connect(self.toggleChainFit)
509        # Buttons
510        self.cmdFit.clicked.connect(self.onFit)
511        self.cmdPlot.clicked.connect(self.onPlot)
512        self.cmdHelp.clicked.connect(self.onHelp)
513        self.cmdMagneticDisplay.clicked.connect(self.onDisplayMagneticAngles)
514
515        # Respond to change in parameters from the UI
516        self._model_model.itemChanged.connect(self.onMainParamsChange)
517        #self.constraintAddedSignal.connect(self.modifyViewOnConstraint)
518        self._poly_model.itemChanged.connect(self.onPolyModelChange)
519        self._magnet_model.itemChanged.connect(self.onMagnetModelChange)
520        self.lstParams.selectionModel().selectionChanged.connect(self.onSelectionChanged)
521
522        # Local signals
523        self.batchFittingFinishedSignal.connect(self.batchFitComplete)
524        self.fittingFinishedSignal.connect(self.fitComplete)
525        self.Calc1DFinishedSignal.connect(self.complete1D)
526        self.Calc2DFinishedSignal.connect(self.complete2D)
527
528        # Signals from separate tabs asking for replot
529        self.options_widget.plot_signal.connect(self.onOptionsUpdate)
530        self.options_widget.plot_signal.connect(self.onOptionsUpdate)
531
532        # Signals from other widgets
533        self.communicate.customModelDirectoryChanged.connect(self.onCustomModelChange)
534        self.communicate.saveAnalysisSignal.connect(self.savePageState)
535        self.smearing_widget.smearingChangedSignal.connect(self.onSmearingOptionsUpdate)
536        self.communicate.copyFitParamsSignal.connect(self.onParameterCopy)
537        self.communicate.pasteFitParamsSignal.connect(self.onParameterPaste)
538
539        # Communicator signal
540        self.communicate.updateModelCategoriesSignal.connect(self.onCategoriesChanged)
541
542    def modelName(self):
543        """
544        Returns model name, by default M<tab#>, e.g. M1, M2
545        """
546        return "M%i" % self.tab_id
547
548    def nameForFittedData(self, name):
549        """
550        Generate name for the current fit
551        """
552        if self.is2D:
553            name += "2d"
554        name = "%s [%s]" % (self.modelName(), name)
555        return name
556
557    def showModelContextMenu(self, position):
558        """
559        Show context specific menu in the parameter table.
560        When clicked on parameter(s): fitting/constraints options
561        When clicked on white space: model description
562        """
563        rows = [s.row() for s in self.lstParams.selectionModel().selectedRows()]
564        menu = self.showModelDescription() if not rows else self.modelContextMenu(rows)
565        try:
566            menu.exec_(self.lstParams.viewport().mapToGlobal(position))
567        except AttributeError as ex:
568            logging.error("Error generating context menu: %s" % ex)
569        return
570
571    def modelContextMenu(self, rows):
572        """
573        Create context menu for the parameter selection
574        """
575        menu = QtWidgets.QMenu()
576        num_rows = len(rows)
577        if num_rows < 1:
578            return menu
579        # Select for fitting
580        param_string = "parameter " if num_rows == 1 else "parameters "
581        to_string = "to its current value" if num_rows == 1 else "to their current values"
582        has_constraints = any([self.rowHasConstraint(i) for i in rows])
583
584        self.actionSelect = QtWidgets.QAction(self)
585        self.actionSelect.setObjectName("actionSelect")
586        self.actionSelect.setText(QtCore.QCoreApplication.translate("self", "Select "+param_string+" for fitting"))
587        # Unselect from fitting
588        self.actionDeselect = QtWidgets.QAction(self)
589        self.actionDeselect.setObjectName("actionDeselect")
590        self.actionDeselect.setText(QtCore.QCoreApplication.translate("self", "De-select "+param_string+" from fitting"))
591
592        self.actionConstrain = QtWidgets.QAction(self)
593        self.actionConstrain.setObjectName("actionConstrain")
594        self.actionConstrain.setText(QtCore.QCoreApplication.translate("self", "Constrain "+param_string + to_string))
595
596        self.actionRemoveConstraint = QtWidgets.QAction(self)
597        self.actionRemoveConstraint.setObjectName("actionRemoveConstrain")
598        self.actionRemoveConstraint.setText(QtCore.QCoreApplication.translate("self", "Remove constraint"))
599
600        self.actionMultiConstrain = QtWidgets.QAction(self)
601        self.actionMultiConstrain.setObjectName("actionMultiConstrain")
602        self.actionMultiConstrain.setText(QtCore.QCoreApplication.translate("self", "Constrain selected parameters to their current values"))
603
604        self.actionMutualMultiConstrain = QtWidgets.QAction(self)
605        self.actionMutualMultiConstrain.setObjectName("actionMutualMultiConstrain")
606        self.actionMutualMultiConstrain.setText(QtCore.QCoreApplication.translate("self", "Mutual constrain of selected parameters..."))
607
608        menu.addAction(self.actionSelect)
609        menu.addAction(self.actionDeselect)
610        menu.addSeparator()
611
612        if has_constraints:
613            menu.addAction(self.actionRemoveConstraint)
614            #if num_rows == 1:
615            #    menu.addAction(self.actionEditConstraint)
616        else:
617            menu.addAction(self.actionConstrain)
618            if num_rows == 2:
619                menu.addAction(self.actionMutualMultiConstrain)
620
621        # Define the callbacks
622        self.actionConstrain.triggered.connect(self.addSimpleConstraint)
623        self.actionRemoveConstraint.triggered.connect(self.deleteConstraint)
624        self.actionMutualMultiConstrain.triggered.connect(self.showMultiConstraint)
625        self.actionSelect.triggered.connect(self.selectParameters)
626        self.actionDeselect.triggered.connect(self.deselectParameters)
627        return menu
628
629    def showMultiConstraint(self):
630        """
631        Show the constraint widget and receive the expression
632        """
633        selected_rows = self.lstParams.selectionModel().selectedRows()
634        # There have to be only two rows selected. The caller takes care of that
635        # but let's check the correctness.
636        assert len(selected_rows) == 2
637
638        params_list = [s.data() for s in selected_rows]
639        # Create and display the widget for param1 and param2
640        mc_widget = MultiConstraint(self, params=params_list)
641        if mc_widget.exec_() != QtWidgets.QDialog.Accepted:
642            return
643
644        constraint = Constraint()
645        c_text = mc_widget.txtConstraint.text()
646
647        # widget.params[0] is the parameter we're constraining
648        constraint.param = mc_widget.params[0]
649        # parameter should have the model name preamble
650        model_name = self.kernel_module.name
651        # param_used is the parameter we're using in constraining function
652        param_used = mc_widget.params[1]
653        # Replace param_used with model_name.param_used
654        updated_param_used = model_name + "." + param_used
655        new_func = c_text.replace(param_used, updated_param_used)
656        constraint.func = new_func
657        # Which row is the constrained parameter in?
658        row = self.getRowFromName(constraint.param)
659
660        # Create a new item and add the Constraint object as a child
661        self.addConstraintToRow(constraint=constraint, row=row)
662
663    def getRowFromName(self, name):
664        """
665        Given parameter name get the row number in self._model_model
666        """
667        for row in range(self._model_model.rowCount()):
668            row_name = self._model_model.item(row).text()
669            if row_name == name:
670                return row
671        return None
672
673    def getParamNames(self):
674        """
675        Return list of all parameters for the current model
676        """
677        return [self._model_model.item(row).text() for row in range(self._model_model.rowCount())]
678
679    def modifyViewOnRow(self, row, font=None, brush=None):
680        """
681        Chage how the given row of the main model is shown
682        """
683        fields_enabled = False
684        if font is None:
685            font = QtGui.QFont()
686            fields_enabled = True
687        if brush is None:
688            brush = QtGui.QBrush()
689            fields_enabled = True
690        self._model_model.blockSignals(True)
691        # Modify font and foreground of affected rows
692        for column in range(0, self._model_model.columnCount()):
693            self._model_model.item(row, column).setForeground(brush)
694            self._model_model.item(row, column).setFont(font)
695            self._model_model.item(row, column).setEditable(fields_enabled)
696        self._model_model.blockSignals(False)
697
698    def addConstraintToRow(self, constraint=None, row=0):
699        """
700        Adds the constraint object to requested row
701        """
702        # Create a new item and add the Constraint object as a child
703        assert isinstance(constraint, Constraint)
704        assert 0 <= row <= self._model_model.rowCount()
705
706        item = QtGui.QStandardItem()
707        item.setData(constraint)
708        self._model_model.item(row, 1).setChild(0, item)
709        # Set min/max to the value constrained
710        self.constraintAddedSignal.emit([row])
711        # Show visual hints for the constraint
712        font = QtGui.QFont()
713        font.setItalic(True)
714        brush = QtGui.QBrush(QtGui.QColor('blue'))
715        self.modifyViewOnRow(row, font=font, brush=brush)
716        self.communicate.statusBarUpdateSignal.emit('Constraint added')
717
718    def addSimpleConstraint(self):
719        """
720        Adds a constraint on a single parameter.
721        """
722        min_col = self.lstParams.itemDelegate().param_min
723        max_col = self.lstParams.itemDelegate().param_max
724        for row in self.selectedParameters():
725            param = self._model_model.item(row, 0).text()
726            value = self._model_model.item(row, 1).text()
727            min_t = self._model_model.item(row, min_col).text()
728            max_t = self._model_model.item(row, max_col).text()
729            # Create a Constraint object
730            constraint = Constraint(param=param, value=value, min=min_t, max=max_t)
731            # Create a new item and add the Constraint object as a child
732            item = QtGui.QStandardItem()
733            item.setData(constraint)
734            self._model_model.item(row, 1).setChild(0, item)
735            # Assumed correctness from the validator
736            value = float(value)
737            # BUMPS calculates log(max-min) without any checks, so let's assign minor range
738            min_v = value - (value/10000.0)
739            max_v = value + (value/10000.0)
740            # Set min/max to the value constrained
741            self._model_model.item(row, min_col).setText(str(min_v))
742            self._model_model.item(row, max_col).setText(str(max_v))
743            self.constraintAddedSignal.emit([row])
744            # Show visual hints for the constraint
745            font = QtGui.QFont()
746            font.setItalic(True)
747            brush = QtGui.QBrush(QtGui.QColor('blue'))
748            self.modifyViewOnRow(row, font=font, brush=brush)
749        self.communicate.statusBarUpdateSignal.emit('Constraint added')
750
751    def deleteConstraint(self):
752        """
753        Delete constraints from selected parameters.
754        """
755        params = [s.data() for s in self.lstParams.selectionModel().selectedRows()
756                   if self.isCheckable(s.row())]
757        for param in params:
758            self.deleteConstraintOnParameter(param=param)
759
760    def deleteConstraintOnParameter(self, param=None):
761        """
762        Delete the constraint on model parameter 'param'
763        """
764        min_col = self.lstParams.itemDelegate().param_min
765        max_col = self.lstParams.itemDelegate().param_max
766        for row in range(self._model_model.rowCount()):
767            if not self.rowHasConstraint(row):
768                continue
769            # Get the Constraint object from of the model item
770            item = self._model_model.item(row, 1)
771            constraint = self.getConstraintForRow(row)
772            if constraint is None:
773                continue
774            if not isinstance(constraint, Constraint):
775                continue
776            if param and constraint.param != param:
777                continue
778            # Now we got the right row. Delete the constraint and clean up
779            # Retrieve old values and put them on the model
780            if constraint.min is not None:
781                self._model_model.item(row, min_col).setText(constraint.min)
782            if constraint.max is not None:
783                self._model_model.item(row, max_col).setText(constraint.max)
784            # Remove constraint item
785            item.removeRow(0)
786            self.constraintAddedSignal.emit([row])
787            self.modifyViewOnRow(row)
788
789        self.communicate.statusBarUpdateSignal.emit('Constraint removed')
790
791    def getConstraintForRow(self, row):
792        """
793        For the given row, return its constraint, if any
794        """
795        try:
796            item = self._model_model.item(row, 1)
797            return item.child(0).data()
798        except AttributeError:
799            # return none when no constraints
800            return None
801
802    def rowHasConstraint(self, row):
803        """
804        Finds out if row of the main model has a constraint child
805        """
806        item = self._model_model.item(row, 1)
807        if item.hasChildren():
808            c = item.child(0).data()
809            if isinstance(c, Constraint):
810                return True
811        return False
812
813    def rowHasActiveConstraint(self, row):
814        """
815        Finds out if row of the main model has an active constraint child
816        """
817        item = self._model_model.item(row, 1)
818        if item.hasChildren():
819            c = item.child(0).data()
820            if isinstance(c, Constraint) and c.active:
821                return True
822        return False
823
824    def rowHasActiveComplexConstraint(self, row):
825        """
826        Finds out if row of the main model has an active, nontrivial constraint child
827        """
828        item = self._model_model.item(row, 1)
829        if item.hasChildren():
830            c = item.child(0).data()
831            if isinstance(c, Constraint) and c.func and c.active:
832                return True
833        return False
834
835    def selectParameters(self):
836        """
837        Selected parameter is chosen for fitting
838        """
839        status = QtCore.Qt.Checked
840        self.setParameterSelection(status)
841
842    def deselectParameters(self):
843        """
844        Selected parameters are removed for fitting
845        """
846        status = QtCore.Qt.Unchecked
847        self.setParameterSelection(status)
848
849    def selectedParameters(self):
850        """ Returns list of selected (highlighted) parameters """
851        return [s.row() for s in self.lstParams.selectionModel().selectedRows()
852                if self.isCheckable(s.row())]
853
854    def setParameterSelection(self, status=QtCore.Qt.Unchecked):
855        """
856        Selected parameters are chosen for fitting
857        """
858        # Convert to proper indices and set requested enablement
859        for row in self.selectedParameters():
860            self._model_model.item(row, 0).setCheckState(status)
861
862    def getConstraintsForModel(self):
863        """
864        Return a list of tuples. Each tuple contains constraints mapped as
865        ('constrained parameter', 'function to constrain')
866        e.g. [('sld','5*sld_solvent')]
867        """
868        param_number = self._model_model.rowCount()
869        params = [(self._model_model.item(s, 0).text(),
870                    self._model_model.item(s, 1).child(0).data().func)
871                    for s in range(param_number) if self.rowHasActiveConstraint(s)]
872        return params
873
874    def getComplexConstraintsForModel(self):
875        """
876        Return a list of tuples. Each tuple contains constraints mapped as
877        ('constrained parameter', 'function to constrain')
878        e.g. [('sld','5*M2.sld_solvent')].
879        Only for constraints with defined VALUE
880        """
881        param_number = self._model_model.rowCount()
882        params = [(self._model_model.item(s, 0).text(),
883                    self._model_model.item(s, 1).child(0).data().func)
884                    for s in range(param_number) if self.rowHasActiveComplexConstraint(s)]
885        return params
886
887    def getConstraintObjectsForModel(self):
888        """
889        Returns Constraint objects present on the whole model
890        """
891        param_number = self._model_model.rowCount()
892        constraints = [self._model_model.item(s, 1).child(0).data()
893                       for s in range(param_number) if self.rowHasConstraint(s)]
894
895        return constraints
896
897    def getConstraintsForFitting(self):
898        """
899        Return a list of constraints in format ready for use in fiting
900        """
901        # Get constraints
902        constraints = self.getComplexConstraintsForModel()
903        # See if there are any constraints across models
904        multi_constraints = [cons for cons in constraints if self.isConstraintMultimodel(cons[1])]
905
906        if multi_constraints:
907            # Let users choose what to do
908            msg = "The current fit contains constraints relying on other fit pages.\n"
909            msg += "Parameters with those constraints are:\n" +\
910                '\n'.join([cons[0] for cons in multi_constraints])
911            msg += "\n\nWould you like to remove these constraints or cancel fitting?"
912            msgbox = QtWidgets.QMessageBox(self)
913            msgbox.setIcon(QtWidgets.QMessageBox.Warning)
914            msgbox.setText(msg)
915            msgbox.setWindowTitle("Existing Constraints")
916            # custom buttons
917            button_remove = QtWidgets.QPushButton("Remove")
918            msgbox.addButton(button_remove, QtWidgets.QMessageBox.YesRole)
919            button_cancel = QtWidgets.QPushButton("Cancel")
920            msgbox.addButton(button_cancel, QtWidgets.QMessageBox.RejectRole)
921            retval = msgbox.exec_()
922            if retval == QtWidgets.QMessageBox.RejectRole:
923                # cancel fit
924                raise ValueError("Fitting cancelled")
925            else:
926                # remove constraint
927                for cons in multi_constraints:
928                    self.deleteConstraintOnParameter(param=cons[0])
929                # re-read the constraints
930                constraints = self.getComplexConstraintsForModel()
931
932        return constraints
933
934    def showModelDescription(self):
935        """
936        Creates a window with model description, when right clicked in the treeview
937        """
938        msg = 'Model description:\n'
939        if self.kernel_module is not None:
940            if str(self.kernel_module.description).rstrip().lstrip() == '':
941                msg += "Sorry, no information is available for this model."
942            else:
943                msg += self.kernel_module.description + '\n'
944        else:
945            msg += "You must select a model to get information on this"
946
947        menu = QtWidgets.QMenu()
948        label = QtWidgets.QLabel(msg)
949        action = QtWidgets.QWidgetAction(self)
950        action.setDefaultWidget(label)
951        menu.addAction(action)
952        return menu
953
954    def onSelectModel(self):
955        """
956        Respond to select Model from list event
957        """
958        model = self.cbModel.currentText()
959
960        # empty combobox forced to be read
961        if not model:
962            return
963        # Reset structure factor
964        self.cbStructureFactor.setCurrentIndex(0)
965
966        # Reset parameters to fit
967        self.resetParametersToFit()
968        self.has_error_column = False
969        self.has_poly_error_column = False
970
971        self.respondToModelStructure(model=model, structure_factor=None)
972
973    def onSelectBatchFilename(self, data_index):
974        """
975        Update the logic based on the selected file in batch fitting
976        """
977        self.data_index = data_index
978        self.updateQRange()
979
980    def onSelectStructureFactor(self):
981        """
982        Select Structure Factor from list
983        """
984        model = str(self.cbModel.currentText())
985        category = str(self.cbCategory.currentText())
986        structure = str(self.cbStructureFactor.currentText())
987        if category == CATEGORY_STRUCTURE:
988            model = None
989
990        # Reset parameters to fit
991        self.resetParametersToFit()
992        self.has_error_column = False
993        self.has_poly_error_column = False
994
995        self.respondToModelStructure(model=model, structure_factor=structure)
996
997    def resetParametersToFit(self):
998        """
999        Clears the list of parameters to be fitted
1000        """
1001        self.main_params_to_fit = []
1002        self.poly_params_to_fit = []
1003        self.magnet_params_to_fit = []
1004
1005    def onCustomModelChange(self):
1006        """
1007        Reload the custom model combobox
1008        """
1009        self.custom_models = self.customModels()
1010        self.readCustomCategoryInfo()
1011        # See if we need to update the combo in-place
1012        if self.cbCategory.currentText() != CATEGORY_CUSTOM: return
1013
1014        current_text = self.cbModel.currentText()
1015        self.cbModel.blockSignals(True)
1016        self.cbModel.clear()
1017        self.cbModel.blockSignals(False)
1018        self.enableModelCombo()
1019        self.disableStructureCombo()
1020        # Retrieve the list of models
1021        model_list = self.master_category_dict[CATEGORY_CUSTOM]
1022        # Populate the models combobox
1023        self.cbModel.addItems(sorted([model for (model, _) in model_list]))
1024        new_index = self.cbModel.findText(current_text)
1025        if new_index != -1:
1026            self.cbModel.setCurrentIndex(self.cbModel.findText(current_text))
1027
1028    def onSelectionChanged(self):
1029        """
1030        React to parameter selection
1031        """
1032        rows = self.lstParams.selectionModel().selectedRows()
1033        # Clean previous messages
1034        self.communicate.statusBarUpdateSignal.emit("")
1035        if len(rows) == 1:
1036            # Show constraint, if present
1037            row = rows[0].row()
1038            if self.rowHasConstraint(row):
1039                func = self.getConstraintForRow(row).func
1040                if func is not None:
1041                    self.communicate.statusBarUpdateSignal.emit("Active constrain: "+func)
1042
1043    def replaceConstraintName(self, old_name, new_name=""):
1044        """
1045        Replace names of models in defined constraints
1046        """
1047        param_number = self._model_model.rowCount()
1048        # loop over parameters
1049        for row in range(param_number):
1050            if self.rowHasConstraint(row):
1051                func = self._model_model.item(row, 1).child(0).data().func
1052                if old_name in func:
1053                    new_func = func.replace(old_name, new_name)
1054                    self._model_model.item(row, 1).child(0).data().func = new_func
1055
1056    def isConstraintMultimodel(self, constraint):
1057        """
1058        Check if the constraint function text contains current model name
1059        """
1060        current_model_name = self.kernel_module.name
1061        if current_model_name in constraint:
1062            return False
1063        else:
1064            return True
1065
1066    def updateData(self):
1067        """
1068        Helper function for recalculation of data used in plotting
1069        """
1070        # Update the chart
1071        if self.data_is_loaded:
1072            self.cmdPlot.setText("Show Plot")
1073            self.calculateQGridForModel()
1074        else:
1075            self.cmdPlot.setText("Calculate")
1076            # Create default datasets if no data passed
1077            self.createDefaultDataset()
1078
1079    def respondToModelStructure(self, model=None, structure_factor=None):
1080        # Set enablement on calculate/plot
1081        self.cmdPlot.setEnabled(True)
1082
1083        # kernel parameters -> model_model
1084        self.SASModelToQModel(model, structure_factor)
1085
1086        # Update plot
1087        self.updateData()
1088
1089        # Update state stack
1090        self.updateUndo()
1091
1092        # Let others know
1093        self.newModelSignal.emit()
1094
1095    def onSelectCategory(self):
1096        """
1097        Select Category from list
1098        """
1099        category = self.cbCategory.currentText()
1100        # Check if the user chose "Choose category entry"
1101        if category == CATEGORY_DEFAULT:
1102            # if the previous category was not the default, keep it.
1103            # Otherwise, just return
1104            if self._previous_category_index != 0:
1105                # We need to block signals, or else state changes on perceived unchanged conditions
1106                self.cbCategory.blockSignals(True)
1107                self.cbCategory.setCurrentIndex(self._previous_category_index)
1108                self.cbCategory.blockSignals(False)
1109            return
1110
1111        if category == CATEGORY_STRUCTURE:
1112            self.disableModelCombo()
1113            self.enableStructureCombo()
1114            self._model_model.clear()
1115            return
1116
1117        # Safely clear and enable the model combo
1118        self.cbModel.blockSignals(True)
1119        self.cbModel.clear()
1120        self.cbModel.blockSignals(False)
1121        self.enableModelCombo()
1122        self.disableStructureCombo()
1123
1124        self._previous_category_index = self.cbCategory.currentIndex()
1125        # Retrieve the list of models
1126        model_list = self.master_category_dict[category]
1127        # Populate the models combobox
1128        self.cbModel.addItems(sorted([model for (model, _) in model_list]))
1129
1130    def onPolyModelChange(self, item):
1131        """
1132        Callback method for updating the main model and sasmodel
1133        parameters with the GUI values in the polydispersity view
1134        """
1135        model_column = item.column()
1136        model_row = item.row()
1137        name_index = self._poly_model.index(model_row, 0)
1138        parameter_name = str(name_index.data()).lower() # "distribution of sld" etc.
1139        if "distribution of" in parameter_name:
1140            # just the last word
1141            parameter_name = parameter_name.rsplit()[-1]
1142
1143        delegate = self.lstPoly.itemDelegate()
1144
1145        # Extract changed value.
1146        if model_column == delegate.poly_parameter:
1147            # Is the parameter checked for fitting?
1148            value = item.checkState()
1149            parameter_name = parameter_name + '.width'
1150            if value == QtCore.Qt.Checked:
1151                self.poly_params_to_fit.append(parameter_name)
1152            else:
1153                if parameter_name in self.poly_params_to_fit:
1154                    self.poly_params_to_fit.remove(parameter_name)
1155            self.cmdFit.setEnabled(self.haveParamsToFit())
1156
1157        elif model_column in [delegate.poly_min, delegate.poly_max]:
1158            try:
1159                value = GuiUtils.toDouble(item.text())
1160            except TypeError:
1161                # Can't be converted properly, bring back the old value and exit
1162                return
1163
1164            current_details = self.kernel_module.details[parameter_name]
1165            if self.has_poly_error_column:
1166                # err column changes the indexing
1167                current_details[model_column-2] = value
1168            else:
1169                current_details[model_column-1] = value
1170
1171        elif model_column == delegate.poly_function:
1172            # name of the function - just pass
1173            pass
1174
1175        else:
1176            try:
1177                value = GuiUtils.toDouble(item.text())
1178            except TypeError:
1179                # Can't be converted properly, bring back the old value and exit
1180                return
1181
1182            # Update the sasmodel
1183            # PD[ratio] -> width, npts -> npts, nsigs -> nsigmas
1184            self.kernel_module.setParam(parameter_name + '.' + delegate.columnDict()[model_column], value)
1185
1186            # Update plot
1187            self.updateData()
1188
1189        # update in param model
1190        if model_column in [delegate.poly_pd, delegate.poly_error, delegate.poly_min, delegate.poly_max]:
1191            row = self.getRowFromName(parameter_name)
1192            param_item = self._model_model.item(row)
1193            param_item.child(0).child(0, model_column).setText(item.text())
1194
1195    def onMagnetModelChange(self, item):
1196        """
1197        Callback method for updating the sasmodel magnetic parameters with the GUI values
1198        """
1199        model_column = item.column()
1200        model_row = item.row()
1201        name_index = self._magnet_model.index(model_row, 0)
1202        parameter_name = str(self._magnet_model.data(name_index))
1203
1204        if model_column == 0:
1205            value = item.checkState()
1206            if value == QtCore.Qt.Checked:
1207                self.magnet_params_to_fit.append(parameter_name)
1208            else:
1209                if parameter_name in self.magnet_params_to_fit:
1210                    self.magnet_params_to_fit.remove(parameter_name)
1211            self.cmdFit.setEnabled(self.haveParamsToFit())
1212            # Update state stack
1213            self.updateUndo()
1214            return
1215
1216        # Extract changed value.
1217        try:
1218            value = GuiUtils.toDouble(item.text())
1219        except TypeError:
1220            # Unparsable field
1221            return
1222
1223        property_index = self._magnet_model.headerData(1, model_column)-1 # Value, min, max, etc.
1224
1225        # Update the parameter value - note: this supports +/-inf as well
1226        self.kernel_module.params[parameter_name] = value
1227
1228        # min/max to be changed in self.kernel_module.details[parameter_name] = ['Ang', 0.0, inf]
1229        self.kernel_module.details[parameter_name][property_index] = value
1230
1231        # Force the chart update when actual parameters changed
1232        if model_column == 1:
1233            self.recalculatePlotData()
1234
1235        # Update state stack
1236        self.updateUndo()
1237
1238    def onHelp(self):
1239        """
1240        Show the "Fitting" section of help
1241        """
1242        tree_location = "/user/qtgui/Perspectives/Fitting/"
1243
1244        # Actual file will depend on the current tab
1245        tab_id = self.tabFitting.currentIndex()
1246        helpfile = "fitting.html"
1247        if tab_id == 0:
1248            helpfile = "fitting_help.html"
1249        elif tab_id == 1:
1250            helpfile = "residuals_help.html"
1251        elif tab_id == 2:
1252            helpfile = "resolution.html"
1253        elif tab_id == 3:
1254            helpfile = "pd/polydispersity.html"
1255        elif tab_id == 4:
1256            helpfile = "magnetism/magnetism.html"
1257        help_location = tree_location + helpfile
1258
1259        self.showHelp(help_location)
1260
1261    def showHelp(self, url):
1262        """
1263        Calls parent's method for opening an HTML page
1264        """
1265        self.parent.showHelp(url)
1266
1267    def onDisplayMagneticAngles(self):
1268        """
1269        Display a simple image showing direction of magnetic angles
1270        """
1271        self.magneticAnglesWidget.show()
1272
1273    def onFit(self):
1274        """
1275        Perform fitting on the current data
1276        """
1277        if self.fit_started:
1278            self.stopFit()
1279            return
1280
1281        # initialize fitter constants
1282        fit_id = 0
1283        handler = None
1284        batch_inputs = {}
1285        batch_outputs = {}
1286        #---------------------------------
1287        if LocalConfig.USING_TWISTED:
1288            handler = None
1289            updater = None
1290        else:
1291            handler = ConsoleUpdate(parent=self.parent,
1292                                    manager=self,
1293                                    improvement_delta=0.1)
1294            updater = handler.update_fit
1295
1296        # Prepare the fitter object
1297        try:
1298            fitters, _ = self.prepareFitters()
1299        except ValueError as ex:
1300            # This should not happen! GUI explicitly forbids this situation
1301            self.communicate.statusBarUpdateSignal.emit(str(ex))
1302            return
1303
1304        # keep local copy of kernel parameters, as they will change during the update
1305        self.kernel_module_copy = copy.deepcopy(self.kernel_module)
1306
1307        # Create the fitting thread, based on the fitter
1308        completefn = self.batchFittingCompleted if self.is_batch_fitting else self.fittingCompleted
1309
1310        self.calc_fit = FitThread(handler=handler,
1311                            fn=fitters,
1312                            batch_inputs=batch_inputs,
1313                            batch_outputs=batch_outputs,
1314                            page_id=[[self.page_id]],
1315                            updatefn=updater,
1316                            completefn=completefn,
1317                            reset_flag=self.is_chain_fitting)
1318
1319        if LocalConfig.USING_TWISTED:
1320            # start the trhrhread with twisted
1321            calc_thread = threads.deferToThread(self.calc_fit.compute)
1322            calc_thread.addCallback(completefn)
1323            calc_thread.addErrback(self.fitFailed)
1324        else:
1325            # Use the old python threads + Queue
1326            self.calc_fit.queue()
1327            self.calc_fit.ready(2.5)
1328
1329        self.communicate.statusBarUpdateSignal.emit('Fitting started...')
1330        self.fit_started = True
1331        # Disable some elements
1332        self.setFittingStarted()
1333
1334    def stopFit(self):
1335        """
1336        Attempt to stop the fitting thread
1337        """
1338        if self.calc_fit is None or not self.calc_fit.isrunning():
1339            return
1340        self.calc_fit.stop()
1341        #self.fit_started=False
1342        #re-enable the Fit button
1343        self.setFittingStopped()
1344
1345        msg = "Fitting cancelled."
1346        self.communicate.statusBarUpdateSignal.emit(msg)
1347
1348    def updateFit(self):
1349        """
1350        """
1351        print("UPDATE FIT")
1352        pass
1353
1354    def fitFailed(self, reason):
1355        """
1356        """
1357        self.setFittingStopped()
1358        msg = "Fitting failed with: "+ str(reason)
1359        self.communicate.statusBarUpdateSignal.emit(msg)
1360
1361    def batchFittingCompleted(self, result):
1362        """
1363        Send the finish message from calculate threads to main thread
1364        """
1365        if result is None:
1366            result = tuple()
1367        self.batchFittingFinishedSignal.emit(result)
1368
1369    def batchFitComplete(self, result):
1370        """
1371        Receive and display batch fitting results
1372        """
1373        #re-enable the Fit button
1374        self.setFittingStopped()
1375
1376        if len(result) == 0:
1377            msg = "Fitting failed."
1378            self.communicate.statusBarUpdateSignal.emit(msg)
1379            return
1380
1381        # Show the grid panel
1382        self.communicate.sendDataToGridSignal.emit(result[0])
1383
1384        elapsed = result[1]
1385        msg = "Fitting completed successfully in: %s s.\n" % GuiUtils.formatNumber(elapsed)
1386        self.communicate.statusBarUpdateSignal.emit(msg)
1387
1388        # Run over the list of results and update the items
1389        for res_index, res_list in enumerate(result[0]):
1390            # results
1391            res = res_list[0]
1392            param_dict = self.paramDictFromResults(res)
1393
1394            # create local kernel_module
1395            kernel_module = FittingUtilities.updateKernelWithResults(self.kernel_module, param_dict)
1396            # pull out current data
1397            data = self._logic[res_index].data
1398
1399            # Switch indexes
1400            self.onSelectBatchFilename(res_index)
1401
1402            method = self.complete1D if isinstance(self.data, Data1D) else self.complete2D
1403            self.calculateQGridForModelExt(data=data, model=kernel_module, completefn=method, use_threads=False)
1404
1405        # Restore original kernel_module, so subsequent fits on the same model don't pick up the new params
1406        if self.kernel_module is not None:
1407            self.kernel_module = copy.deepcopy(self.kernel_module_copy)
1408
1409    def paramDictFromResults(self, results):
1410        """
1411        Given the fit results structure, pull out optimized parameters and return them as nicely
1412        formatted dict
1413        """
1414        if results.fitness is None or \
1415            not np.isfinite(results.fitness) or \
1416            np.any(results.pvec is None) or \
1417            not np.all(np.isfinite(results.pvec)):
1418            msg = "Fitting did not converge!"
1419            self.communicate.statusBarUpdateSignal.emit(msg)
1420            msg += results.mesg
1421            logging.error(msg)
1422            return
1423
1424        param_list = results.param_list # ['radius', 'radius.width']
1425        param_values = results.pvec     # array([ 0.36221662,  0.0146783 ])
1426        param_stderr = results.stderr   # array([ 1.71293015,  1.71294233])
1427        params_and_errors = list(zip(param_values, param_stderr))
1428        param_dict = dict(zip(param_list, params_and_errors))
1429
1430        return param_dict
1431
1432    def fittingCompleted(self, result):
1433        """
1434        Send the finish message from calculate threads to main thread
1435        """
1436        if result is None:
1437            result = tuple()
1438        self.fittingFinishedSignal.emit(result)
1439
1440    def fitComplete(self, result):
1441        """
1442        Receive and display fitting results
1443        "result" is a tuple of actual result list and the fit time in seconds
1444        """
1445        #re-enable the Fit button
1446        self.setFittingStopped()
1447
1448        if len(result) == 0:
1449            msg = "Fitting failed."
1450            self.communicate.statusBarUpdateSignal.emit(msg)
1451            return
1452
1453        res_list = result[0][0]
1454        res = res_list[0]
1455        self.chi2 = res.fitness
1456        param_dict = self.paramDictFromResults(res)
1457
1458        if param_dict is None:
1459            return
1460
1461        elapsed = result[1]
1462        if self.calc_fit._interrupting:
1463            msg = "Fitting cancelled by user after: %s s." % GuiUtils.formatNumber(elapsed)
1464            logging.warning("\n"+msg+"\n")
1465        else:
1466            msg = "Fitting completed successfully in: %s s." % GuiUtils.formatNumber(elapsed)
1467        self.communicate.statusBarUpdateSignal.emit(msg)
1468
1469        # Dictionary of fitted parameter: value, error
1470        # e.g. param_dic = {"sld":(1.703, 0.0034), "length":(33.455, -0.0983)}
1471        self.updateModelFromList(param_dict)
1472
1473        self.updatePolyModelFromList(param_dict)
1474
1475        self.updateMagnetModelFromList(param_dict)
1476
1477        # update charts
1478        self.onPlot()
1479
1480        # Read only value - we can get away by just printing it here
1481        chi2_repr = GuiUtils.formatNumber(self.chi2, high=True)
1482        self.lblChi2Value.setText(chi2_repr)
1483
1484    def prepareFitters(self, fitter=None, fit_id=0):
1485        """
1486        Prepare the Fitter object for use in fitting
1487        """
1488        # fitter = None -> single/batch fitting
1489        # fitter = Fit() -> simultaneous fitting
1490
1491        # Data going in
1492        data = self.logic.data
1493        model = self.kernel_module
1494        qmin = self.q_range_min
1495        qmax = self.q_range_max
1496
1497        params_to_fit = self.main_params_to_fit
1498        if self.chkPolydispersity.isChecked():
1499            params_to_fit += self.poly_params_to_fit
1500        if self.chkMagnetism.isChecked():
1501            params_to_fit += self.magnet_params_to_fit
1502        if not params_to_fit:
1503            raise ValueError('Fitting requires at least one parameter to optimize.')
1504
1505        # Potential smearing added
1506        # Remember that smearing_min/max can be None ->
1507        # deal with it until Python gets discriminated unions
1508        self.addWeightingToData(data)
1509
1510        # Get the constraints.
1511        constraints = self.getComplexConstraintsForModel()
1512        if fitter is None:
1513            # For single fits - check for inter-model constraints
1514            constraints = self.getConstraintsForFitting()
1515
1516        smearer = self.smearing_widget.smearer()
1517        handler = None
1518        batch_inputs = {}
1519        batch_outputs = {}
1520
1521        fitters = []
1522        for fit_index in self.all_data:
1523            fitter_single = Fit() if fitter is None else fitter
1524            data = GuiUtils.dataFromItem(fit_index)
1525            # Potential weights added directly to data
1526            self.addWeightingToData(data)
1527            try:
1528                fitter_single.set_model(model, fit_id, params_to_fit, data=data,
1529                             constraints=constraints)
1530            except ValueError as ex:
1531                raise ValueError("Setting model parameters failed with: %s" % ex)
1532
1533            qmin, qmax, _ = self.logic.computeRangeFromData(data)
1534            fitter_single.set_data(data=data, id=fit_id, smearer=smearer, qmin=qmin,
1535                            qmax=qmax)
1536            fitter_single.select_problem_for_fit(id=fit_id, value=1)
1537            if fitter is None:
1538                # Assign id to the new fitter only
1539                fitter_single.fitter_id = [self.page_id]
1540            fit_id += 1
1541            fitters.append(fitter_single)
1542
1543        return fitters, fit_id
1544
1545    def iterateOverModel(self, func):
1546        """
1547        Take func and throw it inside the model row loop
1548        """
1549        for row_i in range(self._model_model.rowCount()):
1550            func(row_i)
1551
1552    def updateModelFromList(self, param_dict):
1553        """
1554        Update the model with new parameters, create the errors column
1555        """
1556        assert isinstance(param_dict, dict)
1557        if not dict:
1558            return
1559
1560        def updateFittedValues(row):
1561            # Utility function for main model update
1562            # internal so can use closure for param_dict
1563            param_name = str(self._model_model.item(row, 0).text())
1564            if param_name not in list(param_dict.keys()):
1565                return
1566            # modify the param value
1567            param_repr = GuiUtils.formatNumber(param_dict[param_name][0], high=True)
1568            self._model_model.item(row, 1).setText(param_repr)
1569            if self.has_error_column:
1570                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1571                self._model_model.item(row, 2).setText(error_repr)
1572
1573        def updatePolyValues(row):
1574            # Utility function for updateof polydispersity part of the main model
1575            param_name = str(self._model_model.item(row, 0).text())+'.width'
1576            if param_name not in list(param_dict.keys()):
1577                return
1578            # modify the param value
1579            param_repr = GuiUtils.formatNumber(param_dict[param_name][0], high=True)
1580            self._model_model.item(row, 0).child(0).child(0,1).setText(param_repr)
1581            # modify the param error
1582            if self.has_error_column:
1583                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1584                self._model_model.item(row, 0).child(0).child(0,2).setText(error_repr)
1585
1586        def createErrorColumn(row):
1587            # Utility function for error column update
1588            item = QtGui.QStandardItem()
1589            def createItem(param_name):
1590                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1591                item.setText(error_repr)
1592            def curr_param():
1593                return str(self._model_model.item(row, 0).text())
1594
1595            [createItem(param_name) for param_name in list(param_dict.keys()) if curr_param() == param_name]
1596
1597            error_column.append(item)
1598
1599        def createPolyErrorColumn(row):
1600            # Utility function for error column update in the polydispersity sub-rows
1601            # NOTE: only creates empty items; updatePolyValues adds the error value
1602            item = self._model_model.item(row, 0)
1603            if not item.hasChildren():
1604                return
1605            poly_item = item.child(0)
1606            if not poly_item.hasChildren():
1607                return
1608            poly_item.insertColumn(2, [QtGui.QStandardItem("")])
1609
1610        # block signals temporarily, so we don't end up
1611        # updating charts with every single model change on the end of fitting
1612        self._model_model.blockSignals(True)
1613
1614        if not self.has_error_column:
1615            # create top-level error column
1616            error_column = []
1617            self.lstParams.itemDelegate().addErrorColumn()
1618            self.iterateOverModel(createErrorColumn)
1619
1620            # we need to enable signals for this, otherwise the final column mysteriously disappears (don't ask, I don't
1621            # know)
1622            self._model_model.blockSignals(False)
1623            self._model_model.insertColumn(2, error_column)
1624            self._model_model.blockSignals(True)
1625
1626            FittingUtilities.addErrorHeadersToModel(self._model_model)
1627
1628            # create error column in polydispersity sub-rows
1629            self.iterateOverModel(createPolyErrorColumn)
1630
1631            self.has_error_column = True
1632
1633        self.iterateOverModel(updateFittedValues)
1634        self.iterateOverModel(updatePolyValues)
1635
1636        self._model_model.blockSignals(False)
1637
1638        # Adjust the table cells width.
1639        # TODO: find a way to dynamically adjust column width while resized expanding
1640        self.lstParams.resizeColumnToContents(0)
1641        self.lstParams.resizeColumnToContents(4)
1642        self.lstParams.resizeColumnToContents(5)
1643        self.lstParams.setSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.Expanding)
1644
1645    def iterateOverPolyModel(self, func):
1646        """
1647        Take func and throw it inside the poly model row loop
1648        """
1649        for row_i in range(self._poly_model.rowCount()):
1650            func(row_i)
1651
1652    def updatePolyModelFromList(self, param_dict):
1653        """
1654        Update the polydispersity model with new parameters, create the errors column
1655        """
1656        assert isinstance(param_dict, dict)
1657        if not dict:
1658            return
1659
1660        def updateFittedValues(row_i):
1661            # Utility function for main model update
1662            # internal so can use closure for param_dict
1663            if row_i >= self._poly_model.rowCount():
1664                return
1665            param_name = str(self._poly_model.item(row_i, 0).text()).rsplit()[-1] + '.width'
1666            if param_name not in list(param_dict.keys()):
1667                return
1668            # modify the param value
1669            param_repr = GuiUtils.formatNumber(param_dict[param_name][0], high=True)
1670            self._poly_model.item(row_i, 1).setText(param_repr)
1671            if self.has_poly_error_column:
1672                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1673                self._poly_model.item(row_i, 2).setText(error_repr)
1674
1675
1676        def createErrorColumn(row_i):
1677            # Utility function for error column update
1678            if row_i >= self._poly_model.rowCount():
1679                return
1680            item = QtGui.QStandardItem()
1681
1682            def createItem(param_name):
1683                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1684                item.setText(error_repr)
1685
1686            def poly_param():
1687                return str(self._poly_model.item(row_i, 0).text()).rsplit()[-1] + '.width'
1688
1689            [createItem(param_name) for param_name in list(param_dict.keys()) if poly_param() == param_name]
1690
1691            error_column.append(item)
1692
1693        # block signals temporarily, so we don't end up
1694        # updating charts with every single model change on the end of fitting
1695        self._poly_model.blockSignals(True)
1696        self.iterateOverPolyModel(updateFittedValues)
1697        self._poly_model.blockSignals(False)
1698
1699        if self.has_poly_error_column:
1700            return
1701
1702        self.lstPoly.itemDelegate().addErrorColumn()
1703        error_column = []
1704        self.iterateOverPolyModel(createErrorColumn)
1705
1706        # switch off reponse to model change
1707        self._poly_model.blockSignals(True)
1708        self._poly_model.insertColumn(2, error_column)
1709        self._poly_model.blockSignals(False)
1710        FittingUtilities.addErrorPolyHeadersToModel(self._poly_model)
1711
1712        self.has_poly_error_column = True
1713
1714    def iterateOverMagnetModel(self, func):
1715        """
1716        Take func and throw it inside the magnet model row loop
1717        """
1718        for row_i in range(self._magnet_model.rowCount()):
1719            func(row_i)
1720
1721    def updateMagnetModelFromList(self, param_dict):
1722        """
1723        Update the magnetic model with new parameters, create the errors column
1724        """
1725        assert isinstance(param_dict, dict)
1726        if not dict:
1727            return
1728        if self._magnet_model.rowCount() == 0:
1729            return
1730
1731        def updateFittedValues(row):
1732            # Utility function for main model update
1733            # internal so can use closure for param_dict
1734            if self._magnet_model.item(row, 0) is None:
1735                return
1736            param_name = str(self._magnet_model.item(row, 0).text())
1737            if param_name not in list(param_dict.keys()):
1738                return
1739            # modify the param value
1740            param_repr = GuiUtils.formatNumber(param_dict[param_name][0], high=True)
1741            self._magnet_model.item(row, 1).setText(param_repr)
1742            if self.has_magnet_error_column:
1743                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1744                self._magnet_model.item(row, 2).setText(error_repr)
1745
1746        def createErrorColumn(row):
1747            # Utility function for error column update
1748            item = QtGui.QStandardItem()
1749            def createItem(param_name):
1750                error_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
1751                item.setText(error_repr)
1752            def curr_param():
1753                return str(self._magnet_model.item(row, 0).text())
1754
1755            [createItem(param_name) for param_name in list(param_dict.keys()) if curr_param() == param_name]
1756
1757            error_column.append(item)
1758
1759        # block signals temporarily, so we don't end up
1760        # updating charts with every single model change on the end of fitting
1761        self._magnet_model.blockSignals(True)
1762        self.iterateOverMagnetModel(updateFittedValues)
1763        self._magnet_model.blockSignals(False)
1764
1765        if self.has_magnet_error_column:
1766            return
1767
1768        self.lstMagnetic.itemDelegate().addErrorColumn()
1769        error_column = []
1770        self.iterateOverMagnetModel(createErrorColumn)
1771
1772        # switch off reponse to model change
1773        self._magnet_model.blockSignals(True)
1774        self._magnet_model.insertColumn(2, error_column)
1775        self._magnet_model.blockSignals(False)
1776        FittingUtilities.addErrorHeadersToModel(self._magnet_model)
1777
1778        self.has_magnet_error_column = True
1779
1780    def onPlot(self):
1781        """
1782        Plot the current set of data
1783        """
1784        # Regardless of previous state, this should now be `plot show` functionality only
1785        self.cmdPlot.setText("Show Plot")
1786        # Force data recalculation so existing charts are updated
1787        self.recalculatePlotData()
1788        self.showPlot()
1789
1790    def onSmearingOptionsUpdate(self):
1791        """
1792        React to changes in the smearing widget
1793        """
1794        self.calculateQGridForModel()
1795
1796    def recalculatePlotData(self):
1797        """
1798        Generate a new dataset for model
1799        """
1800        if not self.data_is_loaded:
1801            self.createDefaultDataset()
1802        self.calculateQGridForModel()
1803
1804    def showPlot(self):
1805        """
1806        Show the current plot in MPL
1807        """
1808        # Show the chart if ready
1809        data_to_show = self.data if self.data_is_loaded else self.model_data
1810        if data_to_show is not None:
1811            self.communicate.plotRequestedSignal.emit([data_to_show])
1812
1813    def onOptionsUpdate(self):
1814        """
1815        Update local option values and replot
1816        """
1817        self.q_range_min, self.q_range_max, self.npts, self.log_points, self.weighting = \
1818            self.options_widget.state()
1819        # set Q range labels on the main tab
1820        self.lblMinRangeDef.setText(str(self.q_range_min))
1821        self.lblMaxRangeDef.setText(str(self.q_range_max))
1822        self.recalculatePlotData()
1823
1824    def setDefaultStructureCombo(self):
1825        """
1826        Fill in the structure factors combo box with defaults
1827        """
1828        structure_factor_list = self.master_category_dict.pop(CATEGORY_STRUCTURE)
1829        factors = [factor[0] for factor in structure_factor_list]
1830        factors.insert(0, STRUCTURE_DEFAULT)
1831        self.cbStructureFactor.clear()
1832        self.cbStructureFactor.addItems(sorted(factors))
1833
1834    def createDefaultDataset(self):
1835        """
1836        Generate default Dataset 1D/2D for the given model
1837        """
1838        # Create default datasets if no data passed
1839        if self.is2D:
1840            qmax = self.q_range_max/np.sqrt(2)
1841            qstep = self.npts
1842            self.logic.createDefault2dData(qmax, qstep, self.tab_id)
1843            return
1844        elif self.log_points:
1845            qmin = -10.0 if self.q_range_min < 1.e-10 else np.log10(self.q_range_min)
1846            qmax = 10.0 if self.q_range_max > 1.e10 else np.log10(self.q_range_max)
1847            interval = np.logspace(start=qmin, stop=qmax, num=self.npts, endpoint=True, base=10.0)
1848        else:
1849            interval = np.linspace(start=self.q_range_min, stop=self.q_range_max,
1850                                   num=self.npts, endpoint=True)
1851        self.logic.createDefault1dData(interval, self.tab_id)
1852
1853    def readCategoryInfo(self):
1854        """
1855        Reads the categories in from file
1856        """
1857        self.master_category_dict = defaultdict(list)
1858        self.by_model_dict = defaultdict(list)
1859        self.model_enabled_dict = defaultdict(bool)
1860
1861        categorization_file = CategoryInstaller.get_user_file()
1862        if not os.path.isfile(categorization_file):
1863            categorization_file = CategoryInstaller.get_default_file()
1864        with open(categorization_file, 'rb') as cat_file:
1865            self.master_category_dict = json.load(cat_file)
1866            self.regenerateModelDict()
1867
1868        # Load the model dict
1869        models = load_standard_models()
1870        for model in models:
1871            self.models[model.name] = model
1872
1873        self.readCustomCategoryInfo()
1874
1875    def readCustomCategoryInfo(self):
1876        """
1877        Reads the custom model category
1878        """
1879        #Looking for plugins
1880        self.plugins = list(self.custom_models.values())
1881        plugin_list = []
1882        for name, plug in self.custom_models.items():
1883            self.models[name] = plug
1884            plugin_list.append([name, True])
1885        self.master_category_dict[CATEGORY_CUSTOM] = plugin_list
1886
1887    def regenerateModelDict(self):
1888        """
1889        Regenerates self.by_model_dict which has each model name as the
1890        key and the list of categories belonging to that model
1891        along with the enabled mapping
1892        """
1893        self.by_model_dict = defaultdict(list)
1894        for category in self.master_category_dict:
1895            for (model, enabled) in self.master_category_dict[category]:
1896                self.by_model_dict[model].append(category)
1897                self.model_enabled_dict[model] = enabled
1898
1899    def addBackgroundToModel(self, model):
1900        """
1901        Adds background parameter with default values to the model
1902        """
1903        assert isinstance(model, QtGui.QStandardItemModel)
1904        checked_list = ['background', '0.001', '-inf', 'inf', '1/cm']
1905        FittingUtilities.addCheckedListToModel(model, checked_list)
1906        last_row = model.rowCount()-1
1907        model.item(last_row, 0).setEditable(False)
1908        model.item(last_row, 4).setEditable(False)
1909
1910    def addScaleToModel(self, model):
1911        """
1912        Adds scale parameter with default values to the model
1913        """
1914        assert isinstance(model, QtGui.QStandardItemModel)
1915        checked_list = ['scale', '1.0', '0.0', 'inf', '']
1916        FittingUtilities.addCheckedListToModel(model, checked_list)
1917        last_row = model.rowCount()-1
1918        model.item(last_row, 0).setEditable(False)
1919        model.item(last_row, 4).setEditable(False)
1920
1921    def addWeightingToData(self, data):
1922        """
1923        Adds weighting contribution to fitting data
1924        """
1925        # Send original data for weighting
1926        weight = FittingUtilities.getWeight(data=data, is2d=self.is2D, flag=self.weighting)
1927        if self.is2D:
1928            data.err_data = weight
1929        else:
1930            data.dy = weight
1931        pass
1932
1933    def updateQRange(self):
1934        """
1935        Updates Q Range display
1936        """
1937        if self.data_is_loaded:
1938            self.q_range_min, self.q_range_max, self.npts = self.logic.computeDataRange()
1939        # set Q range labels on the main tab
1940        self.lblMinRangeDef.setText(str(self.q_range_min))
1941        self.lblMaxRangeDef.setText(str(self.q_range_max))
1942        # set Q range labels on the options tab
1943        self.options_widget.updateQRange(self.q_range_min, self.q_range_max, self.npts)
1944
1945    def SASModelToQModel(self, model_name, structure_factor=None):
1946        """
1947        Setting model parameters into table based on selected category
1948        """
1949        # Crete/overwrite model items
1950        self._model_model.clear()
1951
1952        # First, add parameters from the main model
1953        if model_name is not None:
1954            self.fromModelToQModel(model_name)
1955
1956        # Then, add structure factor derived parameters
1957        if structure_factor is not None and structure_factor != "None":
1958            if model_name is None:
1959                # Instantiate the current sasmodel for SF-only models
1960                self.kernel_module = self.models[structure_factor]()
1961            self.fromStructureFactorToQModel(structure_factor)
1962        else:
1963            # Allow the SF combobox visibility for the given sasmodel
1964            self.enableStructureFactorControl(structure_factor)
1965
1966        # Then, add multishells
1967        if model_name is not None:
1968            # Multishell models need additional treatment
1969            self.addExtraShells()
1970
1971        # Add polydispersity to the model
1972        self.setPolyModel()
1973        # Add magnetic parameters to the model
1974        self.setMagneticModel()
1975
1976        # Adjust the table cells width
1977        self.lstParams.resizeColumnToContents(0)
1978        self.lstParams.setSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.Expanding)
1979
1980        # Now we claim the model has been loaded
1981        self.model_is_loaded = True
1982        # Change the model name to a monicker
1983        self.kernel_module.name = self.modelName()
1984        # Update the smearing tab
1985        self.smearing_widget.updateKernelModel(kernel_model=self.kernel_module)
1986
1987        # (Re)-create headers
1988        FittingUtilities.addHeadersToModel(self._model_model)
1989        self.lstParams.header().setFont(self.boldFont)
1990
1991        # Update Q Ranges
1992        self.updateQRange()
1993
1994    def fromModelToQModel(self, model_name):
1995        """
1996        Setting model parameters into QStandardItemModel based on selected _model_
1997        """
1998        name = model_name
1999        if self.cbCategory.currentText() == CATEGORY_CUSTOM:
2000            # custom kernel load requires full path
2001            name = os.path.join(ModelUtilities.find_plugins_dir(), model_name+".py")
2002        try:
2003            kernel_module = generate.load_kernel_module(name)
2004        except ModuleNotFoundError:
2005            # maybe it's a recategorised custom model?
2006            name = os.path.join(ModelUtilities.find_plugins_dir(), model_name+".py")
2007            # If this rises, it's a valid problem.
2008            kernel_module = generate.load_kernel_module(name)
2009
2010        if hasattr(kernel_module, 'parameters'):
2011            # built-in and custom models
2012            self.model_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
2013
2014        elif hasattr(kernel_module, 'model_info'):
2015            # for sum/multiply models
2016            self.model_parameters = kernel_module.model_info.parameters
2017
2018        elif hasattr(kernel_module, 'Model') and hasattr(kernel_module.Model, "_model_info"):
2019            # this probably won't work if there's no model_info, but just in case
2020            self.model_parameters = kernel_module.Model._model_info.parameters
2021        else:
2022            # no parameters - default to blank table
2023            msg = "No parameters found in model '{}'.".format(model_name)
2024            logger.warning(msg)
2025            self.model_parameters = modelinfo.ParameterTable([])
2026
2027        # Instantiate the current sasmodel
2028        self.kernel_module = self.models[model_name]()
2029
2030        # Explicitly add scale and background with default values
2031        temp_undo_state = self.undo_supported
2032        self.undo_supported = False
2033        self.addScaleToModel(self._model_model)
2034        self.addBackgroundToModel(self._model_model)
2035        self.undo_supported = temp_undo_state
2036
2037        self.shell_names = self.shellNamesList()
2038
2039        # Update the QModel
2040        new_rows = FittingUtilities.addParametersToModel(self.model_parameters, self.kernel_module, self.is2D)
2041
2042        for row in new_rows:
2043            self._model_model.appendRow(row)
2044        # Update the counter used for multishell display
2045        self._last_model_row = self._model_model.rowCount()
2046
2047    def fromStructureFactorToQModel(self, structure_factor):
2048        """
2049        Setting model parameters into QStandardItemModel based on selected _structure factor_
2050        """
2051        structure_module = generate.load_kernel_module(structure_factor)
2052        structure_parameters = modelinfo.make_parameter_table(getattr(structure_module, 'parameters', []))
2053
2054        structure_kernel = self.models[structure_factor]()
2055        form_kernel = self.kernel_module
2056
2057        self.kernel_module = MultiplicationModel(form_kernel, structure_kernel)
2058
2059        new_rows = FittingUtilities.addSimpleParametersToModel(structure_parameters, self.is2D)
2060        for row in new_rows:
2061            self._model_model.appendRow(row)
2062            # disable fitting of parameters not listed in self.kernel_module (probably radius_effective)
2063            if row[0].text() not in self.kernel_module.params.keys():
2064                row_num = self._model_model.rowCount() - 1
2065                FittingUtilities.markParameterDisabled(self._model_model, row_num)
2066
2067        # Update the counter used for multishell display
2068        self._last_model_row = self._model_model.rowCount()
2069
2070    def haveParamsToFit(self):
2071        """
2072        Finds out if there are any parameters ready to be fitted
2073        """
2074        return (self.main_params_to_fit!=[]
2075                or self.poly_params_to_fit!=[]
2076                or self.magnet_params_to_fit != []) and \
2077                self.logic.data_is_loaded
2078
2079    def onMainParamsChange(self, item):
2080        """
2081        Callback method for updating the sasmodel parameters with the GUI values
2082        """
2083        model_column = item.column()
2084
2085        if model_column == 0:
2086            self.checkboxSelected(item)
2087            self.cmdFit.setEnabled(self.haveParamsToFit())
2088            # Update state stack
2089            self.updateUndo()
2090            return
2091
2092        model_row = item.row()
2093        name_index = self._model_model.index(model_row, 0)
2094
2095        # Extract changed value.
2096        try:
2097            value = GuiUtils.toDouble(item.text())
2098        except TypeError:
2099            # Unparsable field
2100            return
2101
2102        parameter_name = str(self._model_model.data(name_index)) # sld, background etc.
2103
2104        # Update the parameter value - note: this supports +/-inf as well
2105        self.kernel_module.params[parameter_name] = value
2106
2107        # Update the parameter value - note: this supports +/-inf as well
2108        param_column = self.lstParams.itemDelegate().param_value
2109        min_column = self.lstParams.itemDelegate().param_min
2110        max_column = self.lstParams.itemDelegate().param_max
2111        if model_column == param_column:
2112            self.kernel_module.setParam(parameter_name, value)
2113        elif model_column == min_column:
2114            # min/max to be changed in self.kernel_module.details[parameter_name] = ['Ang', 0.0, inf]
2115            self.kernel_module.details[parameter_name][1] = value
2116        elif model_column == max_column:
2117            self.kernel_module.details[parameter_name][2] = value
2118        else:
2119            # don't update the chart
2120            return
2121
2122        # TODO: magnetic params in self.kernel_module.details['M0:parameter_name'] = value
2123        # TODO: multishell params in self.kernel_module.details[??] = value
2124
2125        # Force the chart update when actual parameters changed
2126        if model_column == 1:
2127            self.recalculatePlotData()
2128
2129        # Update state stack
2130        self.updateUndo()
2131
2132    def isCheckable(self, row):
2133        return self._model_model.item(row, 0).isCheckable()
2134
2135    def checkboxSelected(self, item):
2136        # Assure we're dealing with checkboxes
2137        if not item.isCheckable():
2138            return
2139        status = item.checkState()
2140
2141        # If multiple rows selected - toggle all of them, filtering uncheckable
2142        # Switch off signaling from the model to avoid recursion
2143        self._model_model.blockSignals(True)
2144        # Convert to proper indices and set requested enablement
2145        self.setParameterSelection(status)
2146        self._model_model.blockSignals(False)
2147
2148        # update the list of parameters to fit
2149        self.main_params_to_fit = self.checkedListFromModel(self._model_model)
2150
2151    def checkedListFromModel(self, model):
2152        """
2153        Returns list of checked parameters for given model
2154        """
2155        def isChecked(row):
2156            return model.item(row, 0).checkState() == QtCore.Qt.Checked
2157
2158        return [str(model.item(row_index, 0).text())
2159                for row_index in range(model.rowCount())
2160                if isChecked(row_index)]
2161
2162    def createNewIndex(self, fitted_data):
2163        """
2164        Create a model or theory index with passed Data1D/Data2D
2165        """
2166        if self.data_is_loaded:
2167            if not fitted_data.name:
2168                name = self.nameForFittedData(self.data.filename)
2169                fitted_data.title = name
2170                fitted_data.name = name
2171                fitted_data.filename = name
2172                fitted_data.symbol = "Line"
2173            self.updateModelIndex(fitted_data)
2174        else:
2175            if not fitted_data.name:
2176                name = self.nameForFittedData(self.kernel_module.id)
2177            else:
2178                name = fitted_data.name
2179            fitted_data.title = name
2180            fitted_data.filename = name
2181            fitted_data.symbol = "Line"
2182            self.createTheoryIndex(fitted_data)
2183
2184    def updateModelIndex(self, fitted_data):
2185        """
2186        Update a QStandardModelIndex containing model data
2187        """
2188        name = self.nameFromData(fitted_data)
2189        # Make this a line if no other defined
2190        if hasattr(fitted_data, 'symbol') and fitted_data.symbol is None:
2191            fitted_data.symbol = 'Line'
2192        # Notify the GUI manager so it can update the main model in DataExplorer
2193        GuiUtils.updateModelItemWithPlot(self.all_data[self.data_index], fitted_data, name)
2194
2195    def createTheoryIndex(self, fitted_data):
2196        """
2197        Create a QStandardModelIndex containing model data
2198        """
2199        name = self.nameFromData(fitted_data)
2200        # Notify the GUI manager so it can create the theory model in DataExplorer
2201        self.theory_item = GuiUtils.createModelItemWithPlot(fitted_data, name=name)
2202        self.communicate.updateTheoryFromPerspectiveSignal.emit(self.theory_item)
2203
2204    def nameFromData(self, fitted_data):
2205        """
2206        Return name for the dataset. Terribly impure function.
2207        """
2208        if fitted_data.name is None:
2209            name = self.nameForFittedData(self.logic.data.filename)
2210            fitted_data.title = name
2211            fitted_data.name = name
2212            fitted_data.filename = name
2213        else:
2214            name = fitted_data.name
2215        return name
2216
2217    def methodCalculateForData(self):
2218        '''return the method for data calculation'''
2219        return Calc1D if isinstance(self.data, Data1D) else Calc2D
2220
2221    def methodCompleteForData(self):
2222        '''return the method for result parsin on calc complete '''
2223        return self.completed1D if isinstance(self.data, Data1D) else self.completed2D
2224
2225    def calculateQGridForModelExt(self, data=None, model=None, completefn=None, use_threads=True):
2226        """
2227        Wrapper for Calc1D/2D calls
2228        """
2229        if data is None:
2230            data = self.data
2231        if model is None:
2232            model = self.kernel_module
2233        if completefn is None:
2234            completefn = self.methodCompleteForData()
2235        smearer = self.smearing_widget.smearer()
2236        # Awful API to a backend method.
2237        calc_thread = self.methodCalculateForData()(data=data,
2238                                               model=model,
2239                                               page_id=0,
2240                                               qmin=self.q_range_min,
2241                                               qmax=self.q_range_max,
2242                                               smearer=smearer,
2243                                               state=None,
2244                                               weight=None,
2245                                               fid=None,
2246                                               toggle_mode_on=False,
2247                                               completefn=completefn,
2248                                               update_chisqr=True,
2249                                               exception_handler=self.calcException,
2250                                               source=None)
2251        if use_threads:
2252            if LocalConfig.USING_TWISTED:
2253                # start the thread with twisted
2254                thread = threads.deferToThread(calc_thread.compute)
2255                thread.addCallback(completefn)
2256                thread.addErrback(self.calculateDataFailed)
2257            else:
2258                # Use the old python threads + Queue
2259                calc_thread.queue()
2260                calc_thread.ready(2.5)
2261        else:
2262            results = calc_thread.compute()
2263            completefn(results)
2264
2265    def calculateQGridForModel(self):
2266        """
2267        Prepare the fitting data object, based on current ModelModel
2268        """
2269        if self.kernel_module is None:
2270            return
2271        self.calculateQGridForModelExt()
2272
2273    def calculateDataFailed(self, reason):
2274        """
2275        Thread returned error
2276        """
2277        print("Calculate Data failed with ", reason)
2278
2279    def completed1D(self, return_data):
2280        self.Calc1DFinishedSignal.emit(return_data)
2281
2282    def completed2D(self, return_data):
2283        self.Calc2DFinishedSignal.emit(return_data)
2284
2285    def complete1D(self, return_data):
2286        """
2287        Plot the current 1D data
2288        """
2289        fitted_data = self.logic.new1DPlot(return_data, self.tab_id)
2290        residuals = self.calculateResiduals(fitted_data)
2291        self.model_data = fitted_data
2292
2293        new_plots = [fitted_data, residuals]
2294
2295        # Create plots for intermediate product data
2296        pq_data, sq_data = self.logic.new1DProductPlots(return_data, self.tab_id)
2297        if pq_data is not None:
2298            pq_data.symbol = "Line"
2299            self.createNewIndex(pq_data)
2300            # self.communicate.plotUpdateSignal.emit([pq_data])
2301            new_plots.append(pq_data)
2302        if sq_data is not None:
2303            sq_data.symbol = "Line"
2304            self.createNewIndex(sq_data)
2305            # self.communicate.plotUpdateSignal.emit([sq_data])
2306            new_plots.append(sq_data)
2307
2308        if self.data_is_loaded:
2309            GuiUtils.deleteRedundantPlots(self.all_data[self.data_index], new_plots)
2310
2311        for plot in new_plots:
2312            if hasattr(plot, "id") and "esidual" in plot.id:
2313                # TODO: fix updates to residuals plot
2314                pass
2315            elif plot is not None:
2316                self.communicate.plotUpdateSignal.emit([plot])
2317
2318    def complete2D(self, return_data):
2319        """
2320        Plot the current 2D data
2321        """
2322        fitted_data = self.logic.new2DPlot(return_data)
2323        self.calculateResiduals(fitted_data)
2324        self.model_data = fitted_data
2325
2326    def calculateResiduals(self, fitted_data):
2327        """
2328        Calculate and print Chi2 and display chart of residuals. Returns residuals plot object.
2329        """
2330        # Create a new index for holding data
2331        fitted_data.symbol = "Line"
2332
2333        # Modify fitted_data with weighting
2334        self.addWeightingToData(fitted_data)
2335
2336        self.createNewIndex(fitted_data)
2337        # Calculate difference between return_data and logic.data
2338        self.chi2 = FittingUtilities.calculateChi2(fitted_data, self.logic.data)
2339        # Update the control
2340        chi2_repr = "---" if self.chi2 is None else GuiUtils.formatNumber(self.chi2, high=True)
2341        self.lblChi2Value.setText(chi2_repr)
2342
2343        # self.communicate.plotUpdateSignal.emit([fitted_data])
2344
2345        # Plot residuals if actual data
2346        if not self.data_is_loaded:
2347            return
2348
2349        residuals_plot = FittingUtilities.plotResiduals(self.data, fitted_data)
2350        residuals_plot.id = "Residual " + residuals_plot.id
2351        self.createNewIndex(residuals_plot)
2352        return residuals_plot
2353
2354    def onCategoriesChanged(self):
2355            """
2356            Reload the category/model comboboxes
2357            """
2358            # Store the current combo indices
2359            current_cat = self.cbCategory.currentText()
2360            current_model = self.cbModel.currentText()
2361
2362            # reread the category file and repopulate the combo
2363            self.cbCategory.blockSignals(True)
2364            self.cbCategory.clear()
2365            self.readCategoryInfo()
2366            self.initializeCategoryCombo()
2367
2368            # Scroll back to the original index in Categories
2369            new_index = self.cbCategory.findText(current_cat)
2370            if new_index != -1:
2371                self.cbCategory.setCurrentIndex(new_index)
2372            self.cbCategory.blockSignals(False)
2373            # ...and in the Models
2374            self.cbModel.blockSignals(True)
2375            new_index = self.cbModel.findText(current_model)
2376            if new_index != -1:
2377                self.cbModel.setCurrentIndex(new_index)
2378            self.cbModel.blockSignals(False)
2379
2380            return
2381
2382    def calcException(self, etype, value, tb):
2383        """
2384        Thread threw an exception.
2385        """
2386        # TODO: remimplement thread cancellation
2387        logging.error("".join(traceback.format_exception(etype, value, tb)))
2388
2389    def setTableProperties(self, table):
2390        """
2391        Setting table properties
2392        """
2393        # Table properties
2394        table.verticalHeader().setVisible(False)
2395        table.setAlternatingRowColors(True)
2396        table.setSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.Expanding)
2397        table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
2398        table.resizeColumnsToContents()
2399
2400        # Header
2401        header = table.horizontalHeader()
2402        header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
2403        header.ResizeMode(QtWidgets.QHeaderView.Interactive)
2404
2405        # Qt5: the following 2 lines crash - figure out why!
2406        # Resize column 0 and 7 to content
2407        #header.setSectionResizeMode(0, QtWidgets.QHeaderView.ResizeToContents)
2408        #header.setSectionResizeMode(7, QtWidgets.QHeaderView.ResizeToContents)
2409
2410    def setPolyModel(self):
2411        """
2412        Set polydispersity values
2413        """
2414        if not self.model_parameters:
2415            return
2416        self._poly_model.clear()
2417
2418        parameters = self.model_parameters.form_volume_parameters
2419        if self.is2D:
2420            parameters += self.model_parameters.orientation_parameters
2421
2422        [self.setPolyModelParameters(i, param) for i, param in \
2423            enumerate(parameters) if param.polydisperse]
2424
2425        FittingUtilities.addPolyHeadersToModel(self._poly_model)
2426
2427    def setPolyModelParameters(self, i, param):
2428        """
2429        Standard of multishell poly parameter driver
2430        """
2431        param_name = param.name
2432        # see it the parameter is multishell
2433        if '[' in param.name:
2434            # Skip empty shells
2435            if self.current_shell_displayed == 0:
2436                return
2437            else:
2438                # Create as many entries as current shells
2439                for ishell in range(1, self.current_shell_displayed+1):
2440                    # Remove [n] and add the shell numeral
2441                    name = param_name[0:param_name.index('[')] + str(ishell)
2442                    self.addNameToPolyModel(i, name)
2443        else:
2444            # Just create a simple param entry
2445            self.addNameToPolyModel(i, param_name)
2446
2447    def addNameToPolyModel(self, i, param_name):
2448        """
2449        Creates a checked row in the poly model with param_name
2450        """
2451        # Polydisp. values from the sasmodel
2452        width = self.kernel_module.getParam(param_name + '.width')
2453        npts = self.kernel_module.getParam(param_name + '.npts')
2454        nsigs = self.kernel_module.getParam(param_name + '.nsigmas')
2455        _, min, max = self.kernel_module.details[param_name]
2456
2457        # Construct a row with polydisp. related variable.
2458        # This will get added to the polydisp. model
2459        # Note: last argument needs extra space padding for decent display of the control
2460        checked_list = ["Distribution of " + param_name, str(width),
2461                        str(min), str(max),
2462                        str(npts), str(nsigs), "gaussian      ",'']
2463        FittingUtilities.addCheckedListToModel(self._poly_model, checked_list)
2464
2465        # All possible polydisp. functions as strings in combobox
2466        func = QtWidgets.QComboBox()
2467        func.addItems([str(name_disp) for name_disp in POLYDISPERSITY_MODELS.keys()])
2468        # Set the default index
2469        func.setCurrentIndex(func.findText(DEFAULT_POLYDISP_FUNCTION))
2470        ind = self._poly_model.index(i,self.lstPoly.itemDelegate().poly_function)
2471        self.lstPoly.setIndexWidget(ind, func)
2472        func.currentIndexChanged.connect(lambda: self.onPolyComboIndexChange(str(func.currentText()), i))
2473
2474    def onPolyFilenameChange(self, row_index):
2475        """
2476        Respond to filename_updated signal from the delegate
2477        """
2478        # For the given row, invoke the "array" combo handler
2479        array_caption = 'array'
2480
2481        # Get the combo box reference
2482        ind = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_function)
2483        widget = self.lstPoly.indexWidget(ind)
2484
2485        # Update the combo box so it displays "array"
2486        widget.blockSignals(True)
2487        widget.setCurrentIndex(self.lstPoly.itemDelegate().POLYDISPERSE_FUNCTIONS.index(array_caption))
2488        widget.blockSignals(False)
2489
2490        # Invoke the file reader
2491        self.onPolyComboIndexChange(array_caption, row_index)
2492
2493    def onPolyComboIndexChange(self, combo_string, row_index):
2494        """
2495        Modify polydisp. defaults on function choice
2496        """
2497        # Get npts/nsigs for current selection
2498        param = self.model_parameters.form_volume_parameters[row_index]
2499        file_index = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_function)
2500        combo_box = self.lstPoly.indexWidget(file_index)
2501
2502        def updateFunctionCaption(row):
2503            # Utility function for update of polydispersity function name in the main model
2504            param_name = str(self._model_model.item(row, 0).text())
2505            if param_name !=  param.name:
2506                return
2507            # Modify the param value
2508            if self.has_error_column:
2509                # err column changes the indexing
2510                self._model_model.item(row, 0).child(0).child(0,5).setText(combo_string)
2511            else:
2512                self._model_model.item(row, 0).child(0).child(0,4).setText(combo_string)
2513
2514        if combo_string == 'array':
2515            try:
2516                self.loadPolydispArray(row_index)
2517                # Update main model for display
2518                self.iterateOverModel(updateFunctionCaption)
2519                # disable the row
2520                lo = self.lstPoly.itemDelegate().poly_pd
2521                hi = self.lstPoly.itemDelegate().poly_function
2522                [self._poly_model.item(row_index, i).setEnabled(False) for i in range(lo, hi)]
2523                return
2524            except IOError:
2525                combo_box.setCurrentIndex(self.orig_poly_index)
2526                # Pass for cancel/bad read
2527                pass
2528
2529        # Enable the row in case it was disabled by Array
2530        self._poly_model.blockSignals(True)
2531        max_range = self.lstPoly.itemDelegate().poly_filename
2532        [self._poly_model.item(row_index, i).setEnabled(True) for i in range(7)]
2533        file_index = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_filename)
2534        self._poly_model.setData(file_index, "")
2535        self._poly_model.blockSignals(False)
2536
2537        npts_index = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_npts)
2538        nsigs_index = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_nsigs)
2539
2540        npts = POLYDISPERSITY_MODELS[str(combo_string)].default['npts']
2541        nsigs = POLYDISPERSITY_MODELS[str(combo_string)].default['nsigmas']
2542
2543        self._poly_model.setData(npts_index, npts)
2544        self._poly_model.setData(nsigs_index, nsigs)
2545
2546        self.iterateOverModel(updateFunctionCaption)
2547        self.orig_poly_index = combo_box.currentIndex()
2548
2549    def loadPolydispArray(self, row_index):
2550        """
2551        Show the load file dialog and loads requested data into state
2552        """
2553        datafile = QtWidgets.QFileDialog.getOpenFileName(
2554            self, "Choose a weight file", "", "All files (*.*)", None,
2555            QtWidgets.QFileDialog.DontUseNativeDialog)[0]
2556
2557        if not datafile:
2558            logging.info("No weight data chosen.")
2559            raise IOError
2560
2561        values = []
2562        weights = []
2563        def appendData(data_tuple):
2564            """
2565            Fish out floats from a tuple of strings
2566            """
2567            try:
2568                values.append(float(data_tuple[0]))
2569                weights.append(float(data_tuple[1]))
2570            except (ValueError, IndexError):
2571                # just pass through if line with bad data
2572                return
2573
2574        with open(datafile, 'r') as column_file:
2575            column_data = [line.rstrip().split() for line in column_file.readlines()]
2576            [appendData(line) for line in column_data]
2577
2578        # If everything went well - update the sasmodel values
2579        self.disp_model = POLYDISPERSITY_MODELS['array']()
2580        self.disp_model.set_weights(np.array(values), np.array(weights))
2581        # + update the cell with filename
2582        fname = os.path.basename(str(datafile))
2583        fname_index = self._poly_model.index(row_index, self.lstPoly.itemDelegate().poly_filename)
2584        self._poly_model.setData(fname_index, fname)
2585
2586    def setMagneticModel(self):
2587        """
2588        Set magnetism values on model
2589        """
2590        if not self.model_parameters:
2591            return
2592        self._magnet_model.clear()
2593        [self.addCheckedMagneticListToModel(param, self._magnet_model) for param in \
2594            self.model_parameters.call_parameters if param.type == 'magnetic']
2595        FittingUtilities.addHeadersToModel(self._magnet_model)
2596
2597    def shellNamesList(self):
2598        """
2599        Returns list of names of all multi-shell parameters
2600        E.g. for sld[n], radius[n], n=1..3 it will return
2601        [sld1, sld2, sld3, radius1, radius2, radius3]
2602        """
2603        multi_names = [p.name[:p.name.index('[')] for p in self.model_parameters.iq_parameters if '[' in p.name]
2604        top_index = self.kernel_module.multiplicity_info.number
2605        shell_names = []
2606        for i in range(1, top_index+1):
2607            for name in multi_names:
2608                shell_names.append(name+str(i))
2609        return shell_names
2610
2611    def addCheckedMagneticListToModel(self, param, model):
2612        """
2613        Wrapper for model update with a subset of magnetic parameters
2614        """
2615        if param.name[param.name.index(':')+1:] in self.shell_names:
2616            # check if two-digit shell number
2617            try:
2618                shell_index = int(param.name[-2:])
2619            except ValueError:
2620                shell_index = int(param.name[-1:])
2621
2622            if shell_index > self.current_shell_displayed:
2623                return
2624
2625        checked_list = [param.name,
2626                        str(param.default),
2627                        str(param.limits[0]),
2628                        str(param.limits[1]),
2629                        param.units]
2630
2631        FittingUtilities.addCheckedListToModel(model, checked_list)
2632
2633    def enableStructureFactorControl(self, structure_factor):
2634        """
2635        Add structure factors to the list of parameters
2636        """
2637        if self.kernel_module.is_form_factor or structure_factor == 'None':
2638            self.enableStructureCombo()
2639        else:
2640            self.disableStructureCombo()
2641
2642    def addExtraShells(self):
2643        """
2644        Add a combobox for multiple shell display
2645        """
2646        param_name, param_length = FittingUtilities.getMultiplicity(self.model_parameters)
2647
2648        if param_length == 0:
2649            return
2650
2651        # cell 1: variable name
2652        item1 = QtGui.QStandardItem(param_name)
2653
2654        func = QtWidgets.QComboBox()
2655        # Available range of shells displayed in the combobox
2656        func.addItems([str(i) for i in range(param_length+1)])
2657
2658        # Respond to index change
2659        func.currentIndexChanged.connect(self.modifyShellsInList)
2660
2661        # cell 2: combobox
2662        item2 = QtGui.QStandardItem()
2663        self._model_model.appendRow([item1, item2])
2664
2665        # Beautify the row:  span columns 2-4
2666        shell_row = self._model_model.rowCount()
2667        shell_index = self._model_model.index(shell_row-1, 1)
2668
2669        self.lstParams.setIndexWidget(shell_index, func)
2670        self._last_model_row = self._model_model.rowCount()
2671
2672        # Set the index to the state-kept value
2673        func.setCurrentIndex(self.current_shell_displayed
2674                             if self.current_shell_displayed < func.count() else 0)
2675
2676    def modifyShellsInList(self, index):
2677        """
2678        Add/remove additional multishell parameters
2679        """
2680        # Find row location of the combobox
2681        last_row = self._last_model_row
2682        remove_rows = self._model_model.rowCount() - last_row
2683
2684        if remove_rows > 1:
2685            self._model_model.removeRows(last_row, remove_rows)
2686
2687        FittingUtilities.addShellsToModel(self.model_parameters, self._model_model, index)
2688        self.current_shell_displayed = index
2689
2690        # Update relevant models
2691        self.setPolyModel()
2692        self.setMagneticModel()
2693
2694    def setFittingStarted(self):
2695        """
2696        Set buttion caption on fitting start
2697        """
2698        # Notify the user that fitting is being run
2699        # Allow for stopping the job
2700        self.cmdFit.setStyleSheet('QPushButton {color: red;}')
2701        self.cmdFit.setText('Stop fit')
2702
2703    def setFittingStopped(self):
2704        """
2705        Set button caption on fitting stop
2706        """
2707        # Notify the user that fitting is available
2708        self.cmdFit.setStyleSheet('QPushButton {color: black;}')
2709        self.cmdFit.setText("Fit")
2710        self.fit_started = False
2711
2712    def readFitPage(self, fp):
2713        """
2714        Read in state from a fitpage object and update GUI
2715        """
2716        assert isinstance(fp, FitPage)
2717        # Main tab info
2718        self.logic.data.filename = fp.filename
2719        self.data_is_loaded = fp.data_is_loaded
2720        self.chkPolydispersity.setCheckState(fp.is_polydisperse)
2721        self.chkMagnetism.setCheckState(fp.is_magnetic)
2722        self.chk2DView.setCheckState(fp.is2D)
2723
2724        # Update the comboboxes
2725        self.cbCategory.setCurrentIndex(self.cbCategory.findText(fp.current_category))
2726        self.cbModel.setCurrentIndex(self.cbModel.findText(fp.current_model))
2727        if fp.current_factor:
2728            self.cbStructureFactor.setCurrentIndex(self.cbStructureFactor.findText(fp.current_factor))
2729
2730        self.chi2 = fp.chi2
2731
2732        # Options tab
2733        self.q_range_min = fp.fit_options[fp.MIN_RANGE]
2734        self.q_range_max = fp.fit_options[fp.MAX_RANGE]
2735        self.npts = fp.fit_options[fp.NPTS]
2736        self.log_points = fp.fit_options[fp.LOG_POINTS]
2737        self.weighting = fp.fit_options[fp.WEIGHTING]
2738
2739        # Models
2740        self._model_model = fp.model_model
2741        self._poly_model = fp.poly_model
2742        self._magnet_model = fp.magnetism_model
2743
2744        # Resolution tab
2745        smearing = fp.smearing_options[fp.SMEARING_OPTION]
2746        accuracy = fp.smearing_options[fp.SMEARING_ACCURACY]
2747        smearing_min = fp.smearing_options[fp.SMEARING_MIN]
2748        smearing_max = fp.smearing_options[fp.SMEARING_MAX]
2749        self.smearing_widget.setState(smearing, accuracy, smearing_min, smearing_max)
2750
2751        # TODO: add polidyspersity and magnetism
2752
2753    def saveToFitPage(self, fp):
2754        """
2755        Write current state to the given fitpage
2756        """
2757        assert isinstance(fp, FitPage)
2758
2759        # Main tab info
2760        fp.filename = self.logic.data.filename
2761        fp.data_is_loaded = self.data_is_loaded
2762        fp.is_polydisperse = self.chkPolydispersity.isChecked()
2763        fp.is_magnetic = self.chkMagnetism.isChecked()
2764        fp.is2D = self.chk2DView.isChecked()
2765        fp.data = self.data
2766
2767        # Use current models - they contain all the required parameters
2768        fp.model_model = self._model_model
2769        fp.poly_model = self._poly_model
2770        fp.magnetism_model = self._magnet_model
2771
2772        if self.cbCategory.currentIndex() != 0:
2773            fp.current_category = str(self.cbCategory.currentText())
2774            fp.current_model = str(self.cbModel.currentText())
2775
2776        if self.cbStructureFactor.isEnabled() and self.cbStructureFactor.currentIndex() != 0:
2777            fp.current_factor = str(self.cbStructureFactor.currentText())
2778        else:
2779            fp.current_factor = ''
2780
2781        fp.chi2 = self.chi2
2782        fp.main_params_to_fit = self.main_params_to_fit
2783        fp.poly_params_to_fit = self.poly_params_to_fit
2784        fp.magnet_params_to_fit = self.magnet_params_to_fit
2785        fp.kernel_module = self.kernel_module
2786
2787        # Algorithm options
2788        # fp.algorithm = self.parent.fit_options.selected_id
2789
2790        # Options tab
2791        fp.fit_options[fp.MIN_RANGE] = self.q_range_min
2792        fp.fit_options[fp.MAX_RANGE] = self.q_range_max
2793        fp.fit_options[fp.NPTS] = self.npts
2794        #fp.fit_options[fp.NPTS_FIT] = self.npts_fit
2795        fp.fit_options[fp.LOG_POINTS] = self.log_points
2796        fp.fit_options[fp.WEIGHTING] = self.weighting
2797
2798        # Resolution tab
2799        smearing, accuracy, smearing_min, smearing_max = self.smearing_widget.state()
2800        fp.smearing_options[fp.SMEARING_OPTION] = smearing
2801        fp.smearing_options[fp.SMEARING_ACCURACY] = accuracy
2802        fp.smearing_options[fp.SMEARING_MIN] = smearing_min
2803        fp.smearing_options[fp.SMEARING_MAX] = smearing_max
2804
2805        # TODO: add polidyspersity and magnetism
2806
2807
2808    def updateUndo(self):
2809        """
2810        Create a new state page and add it to the stack
2811        """
2812        if self.undo_supported:
2813            self.pushFitPage(self.currentState())
2814
2815    def currentState(self):
2816        """
2817        Return fit page with current state
2818        """
2819        new_page = FitPage()
2820        self.saveToFitPage(new_page)
2821
2822        return new_page
2823
2824    def pushFitPage(self, new_page):
2825        """
2826        Add a new fit page object with current state
2827        """
2828        self.page_stack.append(new_page)
2829
2830    def popFitPage(self):
2831        """
2832        Remove top fit page from stack
2833        """
2834        if self.page_stack:
2835            self.page_stack.pop()
2836
2837    def getReport(self):
2838        """
2839        Create and return HTML report with parameters and charts
2840        """
2841        index = None
2842        if self.all_data:
2843            index = self.all_data[self.data_index]
2844        else:
2845            index = self.theory_item
2846        report_logic = ReportPageLogic(self,
2847                                       kernel_module=self.kernel_module,
2848                                       data=self.data,
2849                                       index=index,
2850                                       model=self._model_model)
2851
2852        return report_logic.reportList()
2853
2854    def savePageState(self):
2855        """
2856        Create and serialize local PageState
2857        """
2858        from sas.sascalc.fit.pagestate import Reader
2859        model = self.kernel_module
2860
2861        # Old style PageState object
2862        state = PageState(model=model, data=self.data)
2863
2864        # Add parameter data to the state
2865        self.getCurrentFitState(state)
2866
2867        # Create the filewriter, aptly named 'Reader'
2868        state_reader = Reader(self.loadPageStateCallback)
2869        filepath = self.saveAsAnalysisFile()
2870        if filepath is None or filepath == "":
2871            return
2872        state_reader.write(filename=filepath, fitstate=state)
2873        pass
2874
2875    def saveAsAnalysisFile(self):
2876        """
2877        Show the save as... dialog and return the chosen filepath
2878        """
2879        default_name = "FitPage"+str(self.tab_id)+".fitv"
2880
2881        wildcard = "fitv files (*.fitv)"
2882        kwargs = {
2883            'caption'   : 'Save As',
2884            'directory' : default_name,
2885            'filter'    : wildcard,
2886            'parent'    : None,
2887        }
2888        # Query user for filename.
2889        filename_tuple = QtWidgets.QFileDialog.getSaveFileName(**kwargs)
2890        filename = filename_tuple[0]
2891        return filename
2892
2893    def loadPageStateCallback(self,state=None, datainfo=None, format=None):
2894        """
2895        This is a callback method called from the CANSAS reader.
2896        We need the instance of this reader only for writing out a file,
2897        so there's nothing here.
2898        Until Load Analysis is implemented, that is.
2899        """
2900        pass
2901
2902    def loadPageState(self, pagestate=None):
2903        """
2904        Load the PageState object and update the current widget
2905        """
2906        pass
2907
2908    def getCurrentFitState(self, state=None):
2909        """
2910        Store current state for fit_page
2911        """
2912        # save model option
2913        #if self.model is not None:
2914        #    self.disp_list = self.getDispParamList()
2915        #    state.disp_list = copy.deepcopy(self.disp_list)
2916        #    #state.model = self.model.clone()
2917
2918        # Comboboxes
2919        state.categorycombobox = self.cbCategory.currentText()
2920        state.formfactorcombobox = self.cbModel.currentText()
2921        if self.cbStructureFactor.isEnabled():
2922            state.structurecombobox = self.cbStructureFactor.currentText()
2923        state.tcChi = self.chi2
2924
2925        state.enable2D = self.is2D
2926
2927        #state.weights = copy.deepcopy(self.weights)
2928        # save data
2929        state.data = copy.deepcopy(self.data)
2930
2931        # save plotting range
2932        state.qmin = self.q_range_min
2933        state.qmax = self.q_range_max
2934        state.npts = self.npts
2935
2936        #    self.state.enable_disp = self.enable_disp.GetValue()
2937        #    self.state.disable_disp = self.disable_disp.GetValue()
2938
2939        #    self.state.enable_smearer = \
2940        #                        copy.deepcopy(self.enable_smearer.GetValue())
2941        #    self.state.disable_smearer = \
2942        #                        copy.deepcopy(self.disable_smearer.GetValue())
2943
2944        #self.state.pinhole_smearer = \
2945        #                        copy.deepcopy(self.pinhole_smearer.GetValue())
2946        #self.state.slit_smearer = copy.deepcopy(self.slit_smearer.GetValue())
2947        #self.state.dI_noweight = copy.deepcopy(self.dI_noweight.GetValue())
2948        #self.state.dI_didata = copy.deepcopy(self.dI_didata.GetValue())
2949        #self.state.dI_sqrdata = copy.deepcopy(self.dI_sqrdata.GetValue())
2950        #self.state.dI_idata = copy.deepcopy(self.dI_idata.GetValue())
2951
2952        p = self.model_parameters
2953        # save checkbutton state and txtcrtl values
2954        state.parameters = FittingUtilities.getStandardParam(self._model_model)
2955        state.orientation_params_disp = FittingUtilities.getOrientationParam(self.kernel_module)
2956
2957        #self._copy_parameters_state(self.orientation_params_disp, self.state.orientation_params_disp)
2958        #self._copy_parameters_state(self.parameters, self.state.parameters)
2959        #self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
2960        #self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
2961
2962    def onParameterCopy(self, format=None):
2963        """
2964        Copy current parameters into the clipboard
2965        """
2966        # run a loop over all parameters and pull out
2967        # first - regular params
2968        param_list = []
2969        def gatherParams(row):
2970            """
2971            Create list of main parameters based on _model_model
2972            """
2973            param_name = str(self._model_model.item(row, 0).text())
2974            param_checked = str(self._model_model.item(row, 0).checkState() == QtCore.Qt.Checked)
2975            param_value = str(self._model_model.item(row, 1).text())
2976            param_error = None
2977            column_offset = 0
2978            if self.has_error_column:
2979                param_error = str(self._model_model.item(row, 2).text())
2980                column_offset = 1
2981            param_min = str(self._model_model.item(row, 2+column_offset).text())
2982            param_max = str(self._model_model.item(row, 3+column_offset).text())
2983            param_list.append([param_name, param_checked, param_value, param_error, param_min, param_max])
2984
2985        def gatherPolyParams(row):
2986            """
2987            Create list of polydisperse parameters based on _poly_model
2988            """
2989            param_name = str(self._poly_model.item(row, 0).text()).split()[-1]
2990            param_checked = str(self._poly_model.item(row, 0).checkState() == QtCore.Qt.Checked)
2991            param_value = str(self._poly_model.item(row, 1).text())
2992            param_error = None
2993            column_offset = 0
2994            if self.has_poly_error_column:
2995                param_error = str(self._poly_model.item(row, 2).text())
2996                column_offset = 1
2997            param_min   = str(self._poly_model.item(row, 2+column_offset).text())
2998            param_max   = str(self._poly_model.item(row, 3+column_offset).text())
2999            param_npts  = str(self._poly_model.item(row, 4+column_offset).text())
3000            param_nsigs = str(self._poly_model.item(row, 5+column_offset).text())
3001            param_fun   = str(self._poly_model.item(row, 6+column_offset).text()).rstrip()
3002            # width
3003            name = param_name+".width"
3004            param_list.append([name, param_checked, param_value, param_error,
3005                                param_npts, param_nsigs, param_min, param_max, param_fun])
3006
3007        def gatherMagnetParams(row):
3008            """
3009            Create list of magnetic parameters based on _magnet_model
3010            """
3011            param_name = str(self._magnet_model.item(row, 0).text())
3012            param_checked = str(self._magnet_model.item(row, 0).checkState() == QtCore.Qt.Checked)
3013            param_value = str(self._magnet_model.item(row, 1).text())
3014            param_error = None
3015            column_offset = 0
3016            if self.has_magnet_error_column:
3017                param_error = str(self._magnet_model.item(row, 2).text())
3018                column_offset = 1
3019            param_min = str(self._magnet_model.item(row, 2+column_offset).text())
3020            param_max = str(self._magnet_model.item(row, 3+column_offset).text())
3021            param_list.append([param_name, param_checked, param_value, param_error, param_min, param_max])
3022
3023        self.iterateOverModel(gatherParams)
3024        if self.chkPolydispersity.isChecked():
3025            self.iterateOverPolyModel(gatherPolyParams)
3026        if self.chkMagnetism.isChecked() and self.chkMagnetism.isEnabled():
3027            self.iterateOverMagnetModel(gatherMagnetParams)
3028
3029        if format=="":
3030            formatted_output = FittingUtilities.formatParameters(param_list)
3031        elif format == "Excel":
3032            formatted_output = FittingUtilities.formatParametersExcel(param_list)
3033        elif format == "Latex":
3034            formatted_output = FittingUtilities.formatParametersLatex(param_list)
3035        else:
3036            raise AttributeError("Bad format specifier.")
3037
3038        # Dump formatted_output to the clipboard
3039        cb = QtWidgets.QApplication.clipboard()
3040        cb.setText(formatted_output)
3041
3042    def onParameterPaste(self):
3043        """
3044        Use the clipboard to update fit state
3045        """
3046        # Check if the clipboard contains right stuff
3047        cb = QtWidgets.QApplication.clipboard()
3048        cb_text = cb.text()
3049
3050        context = {}
3051        # put the text into dictionary
3052        lines = cb_text.split(':')
3053        if lines[0] != 'sasview_parameter_values':
3054            return False
3055        for line in lines[1:-1]:
3056            if len(line) != 0:
3057                item = line.split(',')
3058                check = item[1]
3059                name = item[0]
3060                value = item[2]
3061                # Transfer the text to content[dictionary]
3062                context[name] = [check, value]
3063
3064                # limits
3065                limit_lo = item[3]
3066                context[name].append(limit_lo)
3067                limit_hi = item[4]
3068                context[name].append(limit_hi)
3069
3070                # Polydisp
3071                if len(item) > 5:
3072                    value = item[5]
3073                    context[name].append(value)
3074                    try:
3075                        value = item[6]
3076                        context[name].append(value)
3077                        value = item[7]
3078                        context[name].append(value)
3079                    except IndexError:
3080                        pass
3081
3082        self.updateFullModel(context)
3083        self.updateFullPolyModel(context)
3084
3085    def updateFullModel(self, param_dict):
3086        """
3087        Update the model with new parameters
3088        """
3089        assert isinstance(param_dict, dict)
3090        if not dict:
3091            return
3092
3093        def updateFittedValues(row):
3094            # Utility function for main model update
3095            # internal so can use closure for param_dict
3096            param_name = str(self._model_model.item(row, 0).text())
3097            if param_name not in list(param_dict.keys()):
3098                return
3099            # checkbox state
3100            param_checked = QtCore.Qt.Checked if param_dict[param_name][0] == "True" else QtCore.Qt.Unchecked
3101            self._model_model.item(row, 0).setCheckState(param_checked)
3102
3103            # modify the param value
3104            param_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
3105            self._model_model.item(row, 1).setText(param_repr)
3106
3107            # Potentially the error column
3108            ioffset = 0
3109            if len(param_dict[param_name])>4 and self.has_error_column:
3110                # error values are not editable - no need to update
3111                #error_repr = GuiUtils.formatNumber(param_dict[param_name][2], high=True)
3112                #self._model_model.item(row, 2).setText(error_repr)
3113                ioffset = 1
3114            # min/max
3115            param_repr = GuiUtils.formatNumber(param_dict[param_name][2+ioffset], high=True)
3116            self._model_model.item(row, 2+ioffset).setText(param_repr)
3117            param_repr = GuiUtils.formatNumber(param_dict[param_name][3+ioffset], high=True)
3118            self._model_model.item(row, 3+ioffset).setText(param_repr)
3119
3120        # block signals temporarily, so we don't end up
3121        # updating charts with every single model change on the end of fitting
3122        self._model_model.blockSignals(True)
3123        self.iterateOverModel(updateFittedValues)
3124        self._model_model.blockSignals(False)
3125
3126    def updateFullPolyModel(self, param_dict):
3127        """
3128        Update the polydispersity model with new parameters, create the errors column
3129        """
3130        assert isinstance(param_dict, dict)
3131        if not dict:
3132            return
3133
3134        def updateFittedValues(row):
3135            # Utility function for main model update
3136            # internal so can use closure for param_dict
3137            if row >= self._poly_model.rowCount():
3138                return
3139            param_name = str(self._poly_model.item(row, 0).text()).rsplit()[-1] + '.width'
3140            if param_name not in list(param_dict.keys()):
3141                return
3142            # checkbox state
3143            param_checked = QtCore.Qt.Checked if param_dict[param_name][0] == "True" else QtCore.Qt.Unchecked
3144            self._poly_model.item(row,0).setCheckState(param_checked)
3145
3146            # modify the param value
3147            param_repr = GuiUtils.formatNumber(param_dict[param_name][1], high=True)
3148            self._poly_model.item(row, 1).setText(param_repr)
3149
3150            # Potentially the error column
3151            ioffset = 0
3152            if len(param_dict[param_name])>4 and self.has_poly_error_column:
3153                ioffset = 1
3154            # min
3155            param_repr = GuiUtils.formatNumber(param_dict[param_name][2+ioffset], high=True)
3156            self._poly_model.item(row, 2+ioffset).setText(param_repr)
3157            # max
3158            param_repr = GuiUtils.formatNumber(param_dict[param_name][3+ioffset], high=True)
3159            self._poly_model.item(row, 3+ioffset).setText(param_repr)
3160            # Npts
3161            param_repr = GuiUtils.formatNumber(param_dict[param_name][4+ioffset], high=True)
3162            self._poly_model.item(row, 4+ioffset).setText(param_repr)
3163            # Nsigs
3164            param_repr = GuiUtils.formatNumber(param_dict[param_name][5+ioffset], high=True)
3165            self._poly_model.item(row, 5+ioffset).setText(param_repr)
3166
3167            param_repr = GuiUtils.formatNumber(param_dict[param_name][5+ioffset], high=True)
3168            self._poly_model.item(row, 5+ioffset).setText(param_repr)
3169
3170        # block signals temporarily, so we don't end up
3171        # updating charts with every single model change on the end of fitting
3172        self._poly_model.blockSignals(True)
3173        self.iterateOverPolyModel(updateFittedValues)
3174        self._poly_model.blockSignals(False)
3175
3176
Note: See TracBrowser for help on using the repository browser.