source: sasview/src/sas/sasgui/perspectives/fitting/basepage.py @ 880e845

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 880e845 was 880e845, checked in by wojciech, 8 years ago

Adding short time delay upon model loading to prevent double compilation. It is releated to _handle error reported in #778

  • Property mode set to 100644
File size: 142.2 KB
Line 
1"""
2Base Page for fitting
3"""
4import sys
5import os
6import wx
7import numpy
8import time
9import copy
10import math
11import json
12import logging
13import traceback
14
15from collections import defaultdict
16from wx.lib.scrolledpanel import ScrolledPanel
17
18from sasmodels.weights import MODELS as POLYDISPERSITY_MODELS
19from sasmodels.core import HAVE_OPENCL
20from sas.sasgui.guiframe.panel_base import PanelBase
21from sas.sasgui.guiframe.utils import format_number, check_float, IdList, \
22    check_int
23from sas.sasgui.guiframe.events import PanelOnFocusEvent
24from sas.sasgui.guiframe.events import StatusEvent
25from sas.sasgui.guiframe.events import AppendBookmarkEvent
26from sas.sasgui.guiframe.dataFitting import Data2D
27from sas.sasgui.guiframe.dataFitting import Data1D
28from sas.sasgui.guiframe.dataFitting import check_data_validity
29from sas.sasgui.guiframe.gui_style import GUIFRAME_ID
30from sas.sascalc.dataloader.data_info import Detector
31from sas.sascalc.dataloader.data_info import Source
32from sas.sasgui.perspectives.fitting.pagestate import PageState
33from sas.sasgui.guiframe.CategoryInstaller import CategoryInstaller
34from sas.sasgui.guiframe.documentation_window import DocumentationWindow
35
36
37(PageInfoEvent, EVT_PAGE_INFO) = wx.lib.newevent.NewEvent()
38(PreviousStateEvent, EVT_PREVIOUS_STATE) = wx.lib.newevent.NewEvent()
39(NextStateEvent, EVT_NEXT_STATE) = wx.lib.newevent.NewEvent()
40
41_BOX_WIDTH = 76
42_QMIN_DEFAULT = 0.0005
43_QMAX_DEFAULT = 0.5
44_NPTS_DEFAULT = 50
45# Control panel width
46if sys.platform.count("win32") > 0:
47    PANEL_WIDTH = 450
48    FONT_VARIANT = 0
49    ON_MAC = False
50else:
51    PANEL_WIDTH = 500
52    FONT_VARIANT = 1
53    ON_MAC = True
54
55class BasicPage(ScrolledPanel, PanelBase):
56    """
57    This class provide general structure of  fitpanel page
58    """
59    # Internal name for the AUI manager
60    window_name = "Fit Page"
61    # Title to appear on top of the window
62    window_caption = "Fit Page "
63
64    # These two buttons have specific IDs since they seem to be created more
65    # frequently than they need to.  In particular, set_dispers_sizer() is
66    # called by _on_select_model
67    ID_BOOKMARK = wx.NewId()
68    ID_DISPERSER_HELP = wx.NewId()
69    _id_pool = IdList()
70
71    def __init__(self, parent, color='blue', **kwargs):
72        """
73        """
74        ScrolledPanel.__init__(self, parent, **kwargs)
75        PanelBase.__init__(self, parent)
76        self.SetupScrolling()
77        # Set window's font size
78        self.SetWindowVariant(variant=FONT_VARIANT)
79        self.SetBackgroundColour(color)
80
81        self._ids = iter(self._id_pool)
82        # parent of the page
83        self.parent = parent
84        # manager is the fitting plugin
85        # owner of the page (fitting plugin)
86        self.event_owner = None
87        # current model
88        self.model = None
89        self.m_name = None
90        self.index_model = None
91        self.panel = None
92        # data
93        self.data = None
94        # list of available data
95        self.data_list = []
96        self.mask = None
97        self.uid = wx.NewId()
98        self.graph_id = None
99        # Q range for data set
100        self.qmin_data_set = numpy.inf
101        self.qmax_data_set = None
102        self.npts_data_set = 0
103        # Q range
104        self.qmin = None
105        self.qmax = None
106        self.qmax_x = _QMAX_DEFAULT
107        self.qmin_x = _QMIN_DEFAULT
108        self.npts_x = _NPTS_DEFAULT
109        # total number of point: float
110        self.npts = None
111        self.num_points = None
112        # smear default
113        self.current_smearer = None
114        # 2D smear accuracy default
115        self.smear2d_accuracy = 'Low'
116        # slit smear:
117        self.dxl = None
118        self.dxw = None
119        # pinhole smear
120        self.dx_min = None
121        self.dx_max = None
122        # smear attrbs
123        self.enable_smearer = None
124        self.disable_smearer = None
125        self.pinhole_smearer = None
126        self.slit_smearer = None
127        # weight attrbs
128        self.dI_noweight = None
129        self.dI_didata = None
130        self.dI_sqrdata = None
131        self.dI_idata = None
132        # other attrbs
133        self.dq_l = None
134        self.dq_r = None
135        self.tcChi = None
136        self.disp_box = None
137        self.model_disp = None
138        self.Npts_fit = None
139        self.Npts_total = None
140        self.theory_qmin = None
141        self.theory_qmax = None
142        self.theory_qmin_x = None
143        self.theory_qmax_x = None
144        self.btEditMask = None
145        self.btFit = None
146        self.sld_axes = None
147        self.multi_factor = None
148
149        self.disp_cb_dict = {}
150
151        # self.state = PageState(parent=parent)
152        # dictionary containing list of models
153        self.model_list_box = {}
154
155        # Data member to store the dispersion object created
156        self._disp_obj_dict = {}
157        # selected parameters to apply dispersion
158        self.disp_cb_dict = {}
159        # smearer object
160        self.enable2D = False
161        self._has_magnetic = False
162        self.magnetic_on = False
163        self.is_mac = ON_MAC
164        self.formfactorbox = None
165        self.structurebox = None
166        self.categorybox = None
167        # list of model parameters. each item must have same length
168        # each item related to a given parameters
169        # [cb state, name, value, "+/-", error of fit, min, max , units]
170        self.parameters = []
171        # non-fittable parameter whose value is astring
172        self.str_parameters = []
173        # list of parameters to fit , must be like self.parameters
174        self.param_toFit = []
175        # list of looking like parameters but with non fittable parameters info
176        self.fixed_param = []
177        # list of looking like parameters but with  fittable parameters info
178        self.fittable_param = []
179        # list of dispersion parameters
180        self.disp_list = []
181        self.disp_name = ""
182
183        # list of orientation parameters
184        self.orientation_params = []
185        self.orientation_params_disp = []
186#       Self.model should ALWAYS be None here.  It was set to none above in
187#       this long init setting.  no obvious function call in between setting
188#       and this - commenting out on 4/8/2014 by PDB.  Remove once clear
189#       it is pointless.
190#        if self.model is not None:
191#            self.disp_list = self.model.getDispParamList()
192        self.temp_multi_functional = False
193        # enable model 2D draw
194        self.enable2D = False
195        # check that the fit range is correct to plot the model again
196        self.fitrange = True
197        # Create memento to save the current state
198        self.state = PageState(parent=self.parent,
199                               model=self.model, data=self.data)
200        # flag to determine if state has change
201        self.state_change = False
202        # save customized array
203        self.values = {}   # type: Dict[str, List[float, ...]]
204        self.weights = {}   # type: Dict[str, List[float, ...]]
205        # retrieve saved state
206        self.number_saved_state = 0
207        # dictionary of saved state
208        self.saved_states = {}
209        # Create context menu for page
210        self.popUpMenu = wx.Menu()
211
212        wx_id = self._ids.next()
213        self._keep = wx.MenuItem(self.popUpMenu, wx_id, "Add bookmark",
214                                 " Keep the panel status to recall it later")
215        self.popUpMenu.AppendItem(self._keep)
216        self._keep.Enable(False)
217        self._set_bookmark_flag(False)
218        self._set_save_flag(False)
219        wx.EVT_MENU(self, wx_id, self.on_bookmark)
220        self.popUpMenu.AppendSeparator()
221
222        # Default locations
223        self._default_save_location = os.getcwd()
224        # save initial state on context menu
225        # self.onSave(event=None)
226        self.Bind(wx.EVT_CONTEXT_MENU, self.onContextMenu)
227
228        # bind key event
229        self.Bind(wx.EVT_LEFT_DOWN, self.on_left_down)
230
231        # create the basic structure of the panel with empty sizer
232        self.define_page_structure()
233        # drawing Initial dispersion parameters sizer
234        self.set_dispers_sizer()
235
236        # layout
237        self.set_layout()
238
239    def set_index_model(self, index):
240        """
241        Index related to this page
242        """
243        self.index_model = index
244
245    def create_default_data(self):
246        """
247        Given the user selection, creates a 1D or 2D data
248        Only when the page is on theory mode.
249        """
250        if not hasattr(self, "model_view"):
251            return
252        toggle_mode_on = self.model_view.IsEnabled()
253        if toggle_mode_on:
254            if self.enable2D and not check_data_validity(self.data):
255                self._create_default_2d_data()
256            else:
257                if self.pointsbox.GetValue():
258                    self._create_log_1d_data()
259                else:
260                    self._create_default_1d_data()
261
262            if self.model is not None:
263                if not self.data.is_data:
264                    self._manager.page_finder[self.uid].set_fit_data(
265                        data=[self.data])
266            self.on_smear_helper(update=True)
267            self.state.enable_smearer = self.enable_smearer.GetValue()
268            self.state.disable_smearer = self.disable_smearer.GetValue()
269            self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
270            self.state.slit_smearer = self.slit_smearer.GetValue()
271
272    def _create_default_1d_data(self):
273        """
274        Create default data for fitting perspective
275        Only when the page is on theory mode.
276        :warning: This data is never plotted.
277
278        """
279        x = numpy.linspace(start=self.qmin_x, stop=self.qmax_x,
280                           num=self.npts_x, endpoint=True)
281        self.data = Data1D(x=x)
282        self.data.xaxis('\\rm{Q}', "A^{-1}")
283        self.data.yaxis('\\rm{Intensity}', "cm^{-1}")
284        self.data.is_data = False
285        self.data.id = str(self.uid) + " data"
286        self.data.group_id = str(self.uid) + " Model1D"
287
288    def _create_log_1d_data(self):
289        """
290        Create log-spaced data for fitting perspective
291        Only when the page is on theory mode.
292        :warning: This data is never plotted.
293
294        """
295        if self.qmin_x >= 1.e-10:
296            qmin = numpy.log10(self.qmin_x)
297        else:
298            qmin = -10.
299
300        if self.qmax_x <= 1.e10:
301            qmax = numpy.log10(self.qmax_x)
302        else:
303            qmax = 10.
304
305        x = numpy.logspace(start=qmin, stop=qmax,
306                           num=self.npts_x, endpoint=True, base=10.0)
307        self.data = Data1D(x=x)
308        self.data.xaxis('\\rm{Q}', "A^{-1}")
309        self.data.yaxis('\\rm{Intensity}', "cm^{-1}")
310        self.data.is_data = False
311        self.data.id = str(self.uid) + " data"
312        self.data.group_id = str(self.uid) + " Model1D"
313
314    def _create_default_2d_data(self):
315        """
316        Create 2D data by default
317        Only when the page is on theory mode.
318        :warning: This data is never plotted.
319        """
320        self.data = Data2D()
321        qmax = self.qmax_x / math.sqrt(2)
322        self.data.xaxis('\\rm{Q_{x}}', 'A^{-1}')
323        self.data.yaxis('\\rm{Q_{y}}', 'A^{-1}')
324        self.data.is_data = False
325        self.data.id = str(self.uid) + " data"
326        self.data.group_id = str(self.uid) + " Model2D"
327        # Default values
328        self.data.detector.append(Detector())
329        index = len(self.data.detector) - 1
330        self.data.detector[index].distance = 8000   # mm
331        self.data.source.wavelength = 6             # A
332        self.data.detector[index].pixel_size.x = 5  # mm
333        self.data.detector[index].pixel_size.y = 5  # mm
334        self.data.detector[index].beam_center.x = qmax
335        self.data.detector[index].beam_center.y = qmax
336        xmax = qmax
337        xmin = -qmax
338        ymax = qmax
339        ymin = -qmax
340        qstep = self.npts_x
341
342        x = numpy.linspace(start=xmin, stop=xmax, num=qstep, endpoint=True)
343        y = numpy.linspace(start=ymin, stop=ymax, num=qstep, endpoint=True)
344        # use data info instead
345        new_x = numpy.tile(x, (len(y), 1))
346        new_y = numpy.tile(y, (len(x), 1))
347        new_y = new_y.swapaxes(0, 1)
348        # all data reuire now in 1d array
349        qx_data = new_x.flatten()
350        qy_data = new_y.flatten()
351        q_data = numpy.sqrt(qx_data * qx_data + qy_data * qy_data)
352        # set all True (standing for unmasked) as default
353        mask = numpy.ones(len(qx_data), dtype=bool)
354        # store x and y bin centers in q space
355        x_bins = x
356        y_bins = y
357
358        self.data.source = Source()
359        self.data.data = numpy.ones(len(mask))
360        self.data.err_data = numpy.ones(len(mask))
361        self.data.qx_data = qx_data
362        self.data.qy_data = qy_data
363        self.data.q_data = q_data
364        self.data.mask = mask
365        self.data.x_bins = x_bins
366        self.data.y_bins = y_bins
367        # max and min taking account of the bin sizes
368        self.data.xmin = xmin
369        self.data.xmax = xmax
370        self.data.ymin = ymin
371        self.data.ymax = ymax
372
373    def on_set_focus(self, event):
374        """
375        On Set Focus, update guimanger and menu
376        """
377        if self._manager is not None:
378            wx.PostEvent(self._manager.parent, PanelOnFocusEvent(panel=self))
379            self.on_tap_focus()
380
381    def on_tap_focus(self):
382        """
383        Update menu1 on cliking the page tap
384        """
385        if self._manager.menu1 is not None:
386            chain_menu = self._manager.menu1.FindItemById(
387                                                   self._manager.id_reset_flag)
388            chain_menu.Enable(self.batch_on)
389            sim_menu = self._manager.menu1.FindItemById(self._manager.id_simfit)
390            flag = self.data.is_data\
391                            and (self.model is not None)
392            sim_menu.Enable(not self.batch_on and flag)
393            batch_menu = \
394                    self._manager.menu1.FindItemById(self._manager.id_batchfit)
395            batch_menu.Enable(self.batch_on and flag)
396
397    def onContextMenu(self, event):
398        """
399        Retrieve the state selected state
400        """
401        pos = event.GetPosition()
402        pos = self.ScreenToClient(pos)
403        self.PopupMenu(self.popUpMenu, pos)
404
405    def onUndo(self, event):
406        """
407        Cancel the previous action
408        """
409        event = PreviousStateEvent(page=self)
410        wx.PostEvent(self.parent, event)
411
412    def onRedo(self, event):
413        """
414        Restore the previous action cancelled
415        """
416        event = NextStateEvent(page=self)
417        wx.PostEvent(self.parent, event)
418
419    def define_page_structure(self):
420        """
421        Create empty sizer for a panel
422        """
423        self.vbox = wx.BoxSizer(wx.VERTICAL)
424        self.sizer0 = wx.BoxSizer(wx.VERTICAL)
425        self.sizer1 = wx.BoxSizer(wx.VERTICAL)
426        self.sizer2 = wx.BoxSizer(wx.VERTICAL)
427        self.sizer3 = wx.BoxSizer(wx.VERTICAL)
428        self.sizer4 = wx.BoxSizer(wx.VERTICAL)
429        self.sizer5 = wx.BoxSizer(wx.VERTICAL)
430        self.sizer6 = wx.BoxSizer(wx.VERTICAL)
431
432        self.sizer0.SetMinSize((PANEL_WIDTH, -1))
433        self.sizer1.SetMinSize((PANEL_WIDTH, -1))
434        self.sizer2.SetMinSize((PANEL_WIDTH, -1))
435        self.sizer3.SetMinSize((PANEL_WIDTH, -1))
436        self.sizer4.SetMinSize((PANEL_WIDTH, -1))
437        self.sizer5.SetMinSize((PANEL_WIDTH, -1))
438        self.sizer6.SetMinSize((PANEL_WIDTH, -1))
439
440        self.vbox.Add(self.sizer0)
441        self.vbox.Add(self.sizer1)
442        self.vbox.Add(self.sizer2)
443        self.vbox.Add(self.sizer3)
444        self.vbox.Add(self.sizer4)
445        self.vbox.Add(self.sizer5)
446        self.vbox.Add(self.sizer6)
447
448    def set_layout(self):
449        """
450        layout
451        """
452        self.vbox.Layout()
453        self.vbox.Fit(self)
454        self.SetSizer(self.vbox)
455        self.Centre()
456
457    def set_owner(self, owner):
458        """
459        set owner of fitpage
460
461        :param owner: the class responsible of plotting
462
463        """
464        self.event_owner = owner
465        self.state.event_owner = owner
466
467    def get_state(self):
468        """
469        return the current page state
470        """
471        return self.state
472
473    def get_data(self):
474        """
475        return the current data
476        """
477        return self.data
478
479    def get_data_list(self):
480        """
481        return the current data
482        """
483        return self.data_list
484
485    def set_manager(self, manager):
486        """
487        set panel manager
488
489        :param manager: instance of plugin fitting
490
491        """
492        self._manager = manager
493        self.state.manager = manager
494
495    def populate_box(self, model_dict):
496        """
497        Store list of model
498
499        :param model_dict: dictionary containing list of models
500
501        """
502        self.model_list_box = model_dict
503        self.state.model_list_box = self.model_list_box
504        self.initialize_combox()
505
506    def set_model_dictionary(self, model_dict):
507        """
508        Store a dictionary linking model name -> model object
509
510        :param model_dict: dictionary containing list of models
511        """
512        self.model_dict = model_dict
513
514    def initialize_combox(self):
515        """
516        put default value in the combo box
517        """
518        if self.model_list_box is not None and len(self.model_list_box) > 0:
519            self._populate_box(self.structurebox,
520                               self.model_list_box["Structure Factors"])
521            self.structurebox.Insert("None", 0, None)
522            self.structurebox.SetSelection(0)
523            self.structurebox.Hide()
524            self.text2.Hide()
525            self.structurebox.Disable()
526            self.text2.Disable()
527
528    def set_dispers_sizer(self):
529        """
530        fill sizer containing dispersity info
531        """
532        # print "==== entering set_dispers_sizer ==="
533        self.sizer4.Clear(True)
534        name = "Polydispersity and Orientational Distribution"
535        box_description = wx.StaticBox(self, wx.ID_ANY, name)
536        box_description.SetForegroundColour(wx.BLUE)
537        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
538        # ----------------------------------------------------
539        self.disable_disp = wx.RadioButton(self, wx.ID_ANY, 'Off', (10, 10),
540                                           style=wx.RB_GROUP)
541        self.enable_disp = wx.RadioButton(self, wx.ID_ANY, 'On', (10, 30))
542        # best size for MAC and PC
543        if ON_MAC:
544            size_q = (30, 20)
545        else:
546            size_q = (20, 15)
547        self.disp_help_bt = wx.Button(self, self.ID_DISPERSER_HELP, '?',
548                                      style=wx.BU_EXACTFIT,
549                                      size=size_q)
550        self.disp_help_bt.Bind(wx.EVT_BUTTON, self.on_pd_help_clicked,
551                               id=self.disp_help_bt.GetId())
552        self.disp_help_bt.SetToolTipString("Help for polydispersion.")
553
554        self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param,
555                  id=self.disable_disp.GetId())
556        self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param,
557                  id=self.enable_disp.GetId())
558        # MAC needs SetValue
559        self.disable_disp.SetValue(True)
560        sizer_dispersion = wx.BoxSizer(wx.HORIZONTAL)
561        sizer_dispersion.Add((20, 20))
562        name = ""  # Polydispersity and \nOrientational Distribution "
563        sizer_dispersion.Add(wx.StaticText(self, wx.ID_ANY, name))
564        sizer_dispersion.Add(self.enable_disp)
565        sizer_dispersion.Add((20, 20))
566        sizer_dispersion.Add(self.disable_disp)
567        sizer_dispersion.Add((25, 20))
568        sizer_dispersion.Add(self.disp_help_bt)
569
570        # fill a sizer for dispersion
571        boxsizer1.Add(sizer_dispersion, 0,
572                      wx.TOP|wx.BOTTOM|wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE,
573                      border=5)
574        self.sizer4_4 = wx.GridBagSizer(6, 5)
575
576        boxsizer1.Add(self.sizer4_4)
577        # -----------------------------------------------------
578        self.sizer4.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
579        self.sizer4_4.Layout()
580        self.sizer4.Layout()
581        self.Layout()
582
583        self.Refresh()
584        # saving the state of enable dispersity button
585        self.state.enable_disp = self.enable_disp.GetValue()
586        self.state.disable_disp = self.disable_disp.GetValue()
587        self.SetupScrolling()
588
589    def onResetModel(self, event):
590        """
591        Reset model state
592        """
593        menu = event.GetEventObject()
594        # post help message for the selected model
595        msg = menu.GetHelpString(event.GetId())
596        msg += " reloaded"
597        wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
598        self.Show(False)
599        name = menu.GetLabel(event.GetId())
600        self._on_select_model_helper()
601        if self.model is not None:
602            self.m_name = self.model.name
603        if name in self.saved_states.keys():
604            previous_state = self.saved_states[name]
605            # reset state of checkbox,textcrtl  and  regular parameters value
606
607            self.reset_page(previous_state)
608        self.state.m_name = self.m_name
609        self.Show(True)
610
611    def on_preview(self, event):
612        """
613        Report the current fit results
614        """
615        # Get plot image from plotpanel
616        images, canvases = self.get_images()
617        # get the report dialog
618        self.state.report(images, canvases)
619
620    def on_save(self, event):
621        """
622        Save the current state into file
623        """
624        self.save_current_state()
625        new_state = self.state.clone()
626        # Ask the user the location of the file to write to.
627        path = None
628        if self.parent is not None:
629            self._default_save_location = \
630                        self._manager.parent._default_save_location
631        dlg = wx.FileDialog(self, "Choose a file", self._default_save_location,
632                            self.window_caption, "*.fitv", wx.SAVE)
633
634        if dlg.ShowModal() == wx.ID_OK:
635            path = dlg.GetPath()
636            self._default_save_location = os.path.dirname(path)
637            self._manager.parent._default_save_location = \
638                self._default_save_location
639        else:
640            return None
641        # MAC always needs the extension for saving
642        extens = ".fitv"
643        # Make sure the ext included in the file name
644        fName = os.path.splitext(path)[0] + extens
645        # the manager write the state into file
646        self._manager.save_fit_state(filepath=fName, fitstate=new_state)
647        return new_state
648
649    def on_copy(self, event):
650        """
651        Copy Parameter values to the clipboad
652        """
653        if event is not None:
654            event.Skip()
655        # It seems MAC needs wxCallAfter
656        if event.GetId() == GUIFRAME_ID.COPYEX_ID:
657            print "copy excel"
658            wx.CallAfter(self.get_copy_excel)
659        elif event.GetId() == GUIFRAME_ID.COPYLAT_ID:
660            print "copy latex"
661            wx.CallAfter(self.get_copy_latex)
662        else:
663            wx.CallAfter(self.get_copy)
664
665    def on_paste(self, event):
666        """
667        Paste Parameter values to the panel if possible
668        """
669        # if event is not None:
670        #    event.Skip()
671        # It seems MAC needs wxCallAfter for the setvalues
672        # for multiple textctrl items, otherwise it tends to crash once a while
673        wx.CallAfter(self.get_paste)
674        # messages depending on the flag
675        # self._copy_info(True)
676
677    def _copy_info(self, flag):
678        """
679        Send event dpemding on flag
680
681        : Param flag: flag that distinguish event
682        """
683        # messages depending on the flag
684        if flag is None:
685            msg = " Parameter values are copied to the clipboard..."
686            infor = 'warning'
687        elif flag:
688            msg = " Parameter values are pasted from the clipboard..."
689            infor = "warning"
690        else:
691            msg = "Error occurred: "
692            msg += "No valid parameter values to paste from the clipboard..."
693            infor = "warning"
694        # inform msg to wx
695        wx.PostEvent(self._manager.parent,
696                     StatusEvent(status=msg, info=infor))
697
698    def _get_time_stamp(self):
699        """
700        return time and date stings
701        """
702        # date and time
703        year, month, day, hour, minute, second, _, _, _ = time.localtime()
704        current_time = str(hour) + ":" + str(minute) + ":" + str(second)
705        current_date = str(month) + "/" + str(day) + "/" + str(year)
706        return current_time, current_date
707
708    def on_bookmark(self, event):
709        """
710        save history of the data and model
711        """
712        if self.model is None:
713            msg = "Can not bookmark; Please select Data and Model first..."
714            wx.MessageBox(msg, 'Info')
715            return
716        self.save_current_state()
717        new_state = self.state.clone()
718        # Add model state on context menu
719        self.number_saved_state += 1
720        current_time, current_date = self._get_time_stamp()
721        # name= self.model.name+"[%g]"%self.number_saved_state
722        name = "Fitting: %g]" % self.number_saved_state
723        name += self.model.__class__.__name__
724        name += "bookmarked at %s on %s" % (current_time, current_date)
725        self.saved_states[name] = new_state
726
727        # Add item in the context menu
728        msg = "Model saved at %s on %s" % (current_time, current_date)
729        # post help message for the selected model
730        msg += " Saved! right click on this page to retrieve this model"
731        wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
732
733        self.popUpMenu.Append(self.ID_BOOKMARK, name, str(msg))
734        wx.EVT_MENU(self, self.ID_BOOKMARK, self.onResetModel)
735        wx.PostEvent(self._manager.parent,
736                     AppendBookmarkEvent(title=name,
737                                         hint=str(msg),
738                                         handler=self._back_to_bookmark))
739
740    def _back_to_bookmark(self, event):
741        """
742        Back to bookmark
743        """
744        self._manager.on_perspective(event)
745        self.onResetModel(event)
746        self._draw_model()
747
748    def onSetFocus(self, evt):
749        """
750        highlight the current textcrtl and hide the error text control shown
751        after fitting
752        """
753        return
754
755    def read_file(self, path):
756        """
757        Read two columns file
758
759        :param path: the path to the file to read
760
761        """
762        try:
763            if path is None:
764                status = " Selected Distribution was not loaded: %s" % path
765                wx.PostEvent(self._manager.parent,
766                             StatusEvent(status=status))
767                return None, None
768            input_f = open(path, 'r')
769            buff = input_f.read()
770            lines = buff.split('\n')
771            input_f.close()
772            angles = []
773            weights = []
774            for line in lines:
775                toks = line.split()
776                try:
777                    angle = float(toks[0])
778                    weight = float(toks[1])
779                    angles.append(angle)
780                    weights.append(weight)
781                except Exception:
782                    # Skip non-data lines
783                    logging.error(traceback.format_exc())
784            return numpy.array(angles), numpy.array(weights)
785        except:
786            raise
787
788    def createMemento(self):
789        """
790        return the current state of the page
791        """
792        return self.state.clone()
793
794    def save_current_state(self):
795        """
796        Store current state
797        """
798        # save model option
799        if self.model is not None:
800            self.disp_list = self.model.getDispParamList()
801            self.state.disp_list = copy.deepcopy(self.disp_list)
802            self.state.model = self.model.clone()
803
804            # model combobox: complex code because of mac's silent error
805            if self.structurebox is not None:
806                if self.structurebox.IsShown():
807                    self.state.structurecombobox = 'None'
808                    s_select = self.structurebox.GetSelection()
809                    if s_select > 0:
810                        self.state.structurecombobox = \
811                            self.structurebox.GetString(s_select)
812            if self.formfactorbox is not None:
813                f_select = self.formfactorbox.GetSelection()
814                if f_select > 0:
815                    self.state.formfactorcombobox = \
816                        self.formfactorbox.GetString(f_select)
817        if self.categorybox is not None:
818            cb_select = self.categorybox.GetSelection()
819            if cb_select > 0:
820                self.state.categorycombobox = \
821                    self.categorybox.GetString(cb_select)
822
823        self.state.enable2D = copy.deepcopy(self.enable2D)
824        self.state.values = copy.deepcopy(self.values)
825        self.state.weights = copy.deepcopy(self.weights)
826        # save data
827        self.state.data = copy.deepcopy(self.data)
828        self.state.qmax_x = self.qmax_x
829        self.state.qmin_x = self.qmin_x
830        self.state.dI_noweight = copy.deepcopy(self.dI_noweight.GetValue())
831        self.state.dI_didata = copy.deepcopy(self.dI_didata.GetValue())
832        self.state.dI_sqrdata = copy.deepcopy(self.dI_sqrdata.GetValue())
833        self.state.dI_idata = copy.deepcopy(self.dI_idata.GetValue())
834        self.state.dq_l = self.dq_l
835        self.state.dq_r = self.dq_r
836        if hasattr(self, "enable_disp"):
837            self.state.enable_disp = self.enable_disp.GetValue()
838            self.state.disable_disp = self.disable_disp.GetValue()
839
840        self.state.smearer = copy.deepcopy(self.current_smearer)
841        if hasattr(self, "enable_smearer"):
842            self.state.enable_smearer = \
843                                copy.deepcopy(self.enable_smearer.GetValue())
844            self.state.disable_smearer = \
845                                copy.deepcopy(self.disable_smearer.GetValue())
846
847        self.state.pinhole_smearer = \
848                                copy.deepcopy(self.pinhole_smearer.GetValue())
849        self.state.dx_max = copy.deepcopy(self.dx_max)
850        self.state.dx_min = copy.deepcopy(self.dx_min)
851        self.state.dxl = copy.deepcopy(self.dxl)
852        self.state.dxw = copy.deepcopy(self.dxw)
853        self.state.slit_smearer = copy.deepcopy(self.slit_smearer.GetValue())
854
855        if len(self._disp_obj_dict) > 0:
856            for k, v in self._disp_obj_dict.iteritems():
857                self.state._disp_obj_dict[k] = v.type
858
859            self.state.values = copy.deepcopy(self.values)
860            self.state.weights = copy.deepcopy(self.weights)
861        # save plotting range
862        self._save_plotting_range()
863
864        self.state.orientation_params = []
865        self.state.orientation_params_disp = []
866        self.state.parameters = []
867        self.state.fittable_param = []
868        self.state.fixed_param = []
869        self.state.str_parameters = []
870
871        # save checkbutton state and txtcrtl values
872        self._copy_parameters_state(self.str_parameters,
873                                    self.state.str_parameters)
874        self._copy_parameters_state(self.orientation_params,
875                                     self.state.orientation_params)
876        self._copy_parameters_state(self.orientation_params_disp,
877                                    self.state.orientation_params_disp)
878
879        self._copy_parameters_state(self.parameters, self.state.parameters)
880        self._copy_parameters_state(self.fittable_param,
881                                    self.state.fittable_param)
882        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
883        # save chisqr
884        self.state.tcChi = self.tcChi.GetValue()
885
886    def save_current_state_fit(self):
887        """
888        Store current state for fit_page
889        """
890        # save model option
891        if self.model is not None:
892            self.disp_list = self.model.getDispParamList()
893            self.state.disp_list = copy.deepcopy(self.disp_list)
894            self.state.model = self.model.clone()
895
896        self.state.enable2D = copy.deepcopy(self.enable2D)
897        self.state.values = copy.deepcopy(self.values)
898        self.state.weights = copy.deepcopy(self.weights)
899        # save data
900        self.state.data = copy.deepcopy(self.data)
901
902        if hasattr(self, "enable_disp"):
903            self.state.enable_disp = self.enable_disp.GetValue()
904            self.state.disable_disp = self.disable_disp.GetValue()
905
906        self.state.smearer = copy.deepcopy(self.current_smearer)
907        if hasattr(self, "enable_smearer"):
908            self.state.enable_smearer = \
909                                copy.deepcopy(self.enable_smearer.GetValue())
910            self.state.disable_smearer = \
911                                copy.deepcopy(self.disable_smearer.GetValue())
912
913        self.state.pinhole_smearer = \
914                                copy.deepcopy(self.pinhole_smearer.GetValue())
915        self.state.slit_smearer = copy.deepcopy(self.slit_smearer.GetValue())
916        self.state.dI_noweight = copy.deepcopy(self.dI_noweight.GetValue())
917        self.state.dI_didata = copy.deepcopy(self.dI_didata.GetValue())
918        self.state.dI_sqrdata = copy.deepcopy(self.dI_sqrdata.GetValue())
919        self.state.dI_idata = copy.deepcopy(self.dI_idata.GetValue())
920        if hasattr(self, "disp_box") and self.disp_box is not None:
921            self.state.disp_box = self.disp_box.GetCurrentSelection()
922
923            if len(self.disp_cb_dict) > 0:
924                for k, v in self.disp_cb_dict.iteritems():
925                    if v is None:
926                        self.state.disp_cb_dict[k] = v
927                    else:
928                        try:
929                            self.state.disp_cb_dict[k] = v.GetValue()
930                        except:
931                            self.state.disp_cb_dict[k] = None
932            if len(self._disp_obj_dict) > 0:
933                for k, v in self._disp_obj_dict.iteritems():
934                    self.state._disp_obj_dict[k] = v.type
935
936            self.state.values = copy.deepcopy(self.values)
937            self.state.weights = copy.deepcopy(self.weights)
938
939        # save plotting range
940        self._save_plotting_range()
941
942        # save checkbutton state and txtcrtl values
943        self._copy_parameters_state(self.orientation_params,
944                                    self.state.orientation_params)
945        self._copy_parameters_state(self.orientation_params_disp,
946                                    self.state.orientation_params_disp)
947        self._copy_parameters_state(self.parameters, self.state.parameters)
948        self._copy_parameters_state(self.fittable_param,
949                                    self.state.fittable_param)
950        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
951
952    def check_invalid_panel(self):
953        """
954        check if the user can already perform some action with this panel
955        """
956        if self.data is None:
957            self.disable_smearer.SetValue(True)
958            self.disable_disp.SetValue(True)
959            msg = "Please load Data and select Model to start..."
960            wx.MessageBox(msg, 'Info')
961            return True
962
963    def set_model_state(self, state):
964        """
965        reset page given a model state
966        """
967        self.disp_cb_dict = state.disp_cb_dict
968        self.disp_list = state.disp_list
969
970        # fill model combobox
971        self._show_combox_helper()
972        # select the current model
973        try:
974            # to support older version
975            category_pos = int(state.categorycombobox)
976        except:
977            category_pos = 0
978            for ind_cat in range(self.categorybox.GetCount()):
979                if self.categorycombobox.GetString(ind_cat) == \
980                                        state.categorycombobox:
981                    category_pos = int(ind_cat)
982                    break
983
984        self.categorybox.Select(category_pos)
985        try:
986            # to support older version
987            formfactor_pos = int(state.formfactorcombobox)
988        except:
989            formfactor_pos = 0
990            for ind_form in range(self.formfactorbox.GetCount()):
991                if self.formfactorbox.GetString(ind_form) == \
992                                        state.formfactorcombobox:
993                    formfactor_pos = int(ind_form)
994                    break
995
996        self.formfactorbox.Select(formfactor_pos)
997
998        try:
999            # to support older version
1000            structfactor_pos = int(state.structurecombobox)
1001        except:
1002            structfactor_pos = 0
1003            for ind_struct in range(self.structurebox.GetCount()):
1004                if self.structurebox.GetString(ind_struct) == \
1005                                        state.structurecombobox:
1006                    structfactor_pos = int(ind_struct)
1007                    break
1008
1009        self.structurebox.SetSelection(structfactor_pos)
1010
1011        if state.multi_factor is not None:
1012            self.multifactorbox.SetSelection(state.multi_factor)
1013
1014        # reset state of checkbox,textcrtl  and  regular parameters value
1015        self._reset_parameters_state(self.orientation_params_disp,
1016                                     state.orientation_params_disp)
1017        self._reset_parameters_state(self.orientation_params,
1018                                     state.orientation_params)
1019        self._reset_parameters_state(self.str_parameters,
1020                                     state.str_parameters)
1021        self._reset_parameters_state(self.parameters, state.parameters)
1022        # display dispersion info layer
1023        self.enable_disp.SetValue(state.enable_disp)
1024        self.disable_disp.SetValue(state.disable_disp)
1025
1026        if hasattr(self, "disp_box") and self.disp_box is not None:
1027            self.disp_box.SetSelection(state.disp_box)
1028            n = self.disp_box.GetCurrentSelection()
1029            dispersity = self.disp_box.GetClientData(n)
1030            name = dispersity.__name__
1031            self._set_dipers_Param(event=None)
1032
1033            if name == "ArrayDispersion":
1034
1035                for item in self.disp_cb_dict.keys():
1036
1037                    if hasattr(self.disp_cb_dict[item], "SetValue"):
1038                        self.disp_cb_dict[item].SetValue(
1039                                                    state.disp_cb_dict[item])
1040                        # Create the dispersion objects
1041                        disp_model = POLYDISPERSITY_MODELS['array']()
1042                        if hasattr(state, "values") and \
1043                                 self.disp_cb_dict[item].GetValue() is True:
1044                            if len(state.values) > 0:
1045                                self.values = state.values
1046                                self.weights = state.weights
1047                                disp_model.set_weights(self.values,
1048                                                       state.weights)
1049                            else:
1050                                self._reset_dispersity()
1051
1052                        self._disp_obj_dict[item] = disp_model
1053                        # Set the new model as the dispersion object
1054                        # for the selected parameter
1055                        self.model.set_dispersion(item, disp_model)
1056
1057                        self.model._persistency_dict[item] = \
1058                                                [state.values, state.weights]
1059
1060            else:
1061                keys = self.model.getParamList()
1062                for item in keys:
1063                    if item in self.disp_list and \
1064                            item not in self.model.details:
1065                        self.model.details[item] = ["", None, None]
1066                self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1067                self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1068        # smearing info  restore
1069        if hasattr(self, "enable_smearer"):
1070            # set smearing value whether or not the data
1071            # contain the smearing info
1072            self.enable_smearer.SetValue(state.enable_smearer)
1073            self.disable_smearer.SetValue(state.disable_smearer)
1074            self.onSmear(event=None)
1075        self.pinhole_smearer.SetValue(state.pinhole_smearer)
1076        self.slit_smearer.SetValue(state.slit_smearer)
1077
1078        self.dI_noweight.SetValue(state.dI_noweight)
1079        self.dI_didata.SetValue(state.dI_didata)
1080        self.dI_sqrdata.SetValue(state.dI_sqrdata)
1081        self.dI_idata.SetValue(state.dI_idata)
1082
1083        # we have two more options for smearing
1084        if self.pinhole_smearer.GetValue():
1085            self.onPinholeSmear(event=None)
1086        elif self.slit_smearer.GetValue():
1087            self.onSlitSmear(event=None)
1088
1089        # reset state of checkbox,textcrtl  and dispersity parameters value
1090        self._reset_parameters_state(self.fittable_param, state.fittable_param)
1091        self._reset_parameters_state(self.fixed_param, state.fixed_param)
1092
1093        # draw the model with previous parameters value
1094        self._onparamEnter_helper()
1095        self.select_param(event=None)
1096        # Save state_fit
1097        self.save_current_state_fit()
1098        self._lay_out()
1099        self.Refresh()
1100
1101    def get_cat_combo_box_pos(self, state):
1102        """
1103        Iterate through the categories to find the structurefactor
1104        :return: combo_box_position
1105        """
1106        for key, value in self.master_category_dict.iteritems():
1107            for list_item in value:
1108                if state.formfactorcombobox in list_item:
1109                    return self.categorybox.Items.index(key)
1110        return 0
1111
1112    def reset_page_helper(self, state):
1113        """
1114        Use page_state and change the state of existing page
1115
1116        :precondition: the page is already drawn or created
1117
1118        :postcondition: the state of the underlying data change as well as the
1119            state of the graphic interface
1120        """
1121        if state is None:
1122            return
1123        # set data, etc. from the state
1124        # reset page between theory and fitting from bookmarking
1125        data = state.data
1126
1127        if data is None:
1128            data_min = state.qmin
1129            data_max = state.qmax
1130            self.qmin_x = data_min
1131            self.qmax_x = data_max
1132            self.qmin.SetValue(str(data_min))
1133            self.qmax.SetValue(str(data_max))
1134
1135            self.state.data = data
1136            self.state.qmin = self.qmin_x
1137            self.state.qmax = self.qmax_x
1138        else:
1139            self.set_data(data)
1140
1141        self.enable2D = state.enable2D
1142        try:
1143            self.magnetic_on = state.magnetic_on
1144        except:
1145            # Backward compatibility (for older state files)
1146            self.magnetic_on = False
1147
1148        self.disp_cb_dict = state.disp_cb_dict
1149        self.disp_list = state.disp_list
1150
1151        # fill model combobox
1152        self._show_combox_helper()
1153        # select the current model
1154        try:
1155            # to support older version
1156            category_pos = int(state.categorycombobox)
1157        except:
1158            state.formfactorcombobox = state.formfactorcombobox.lower()
1159            state.formfactorcombobox = \
1160                state.formfactorcombobox.replace('model', '')
1161            state.formfactorcombobox = unicode(state.formfactorcombobox)
1162            state.categorycombobox = unicode(state.categorycombobox)
1163            if state.categorycombobox in self.categorybox.Items:
1164                category_pos = self.categorybox.Items.index(
1165                    state.categorycombobox)
1166            else:
1167                # Look in master list for model name (model.lower)
1168                category_pos = self.get_cat_combo_box_pos(state)
1169
1170        self.categorybox.Select(category_pos)
1171        self._show_combox(None)
1172        try:
1173            # to support older version
1174            formfactor_pos = int(state.formfactorcombobox)
1175        except:
1176            formfactor_pos = 0
1177            for ind_form in range(self.formfactorbox.GetCount()):
1178                if self.formfactorbox.GetString(ind_form) == \
1179                                                    (state.formfactorcombobox):
1180                    formfactor_pos = int(ind_form)
1181                    break
1182
1183        self.formfactorbox.Select(formfactor_pos)
1184
1185        structfactor_pos = 0
1186        try:
1187            # to support older version
1188            structfactor_pos = int(state.structurecombobox)
1189        except:
1190            if state.structurecombobox is not None:
1191                state.structurecombobox = unicode(state.structurecombobox)
1192                for ind_struct in range(self.structurebox.GetCount()):
1193                    if self.structurebox.GetString(ind_struct) == \
1194                                                    (state.structurecombobox):
1195                        structfactor_pos = int(ind_struct)
1196                        break
1197
1198        self.structurebox.SetSelection(structfactor_pos)
1199
1200        if state.multi_factor is not None:
1201            self.multifactorbox.SetSelection(state.multi_factor)
1202
1203        # draw the panel according to the new model parameter
1204        self._on_select_model(event=None)
1205
1206        # take care of 2D button
1207        if data is None and self.model_view.IsEnabled():
1208            if self.enable2D:
1209                self.model_view.SetLabel("2D Mode")
1210            else:
1211                self.model_view.SetLabel("1D Mode")
1212
1213        # reset state of checkbox,textcrtl  and  regular parameters value
1214        self._reset_parameters_state(self.orientation_params_disp,
1215                                     state.orientation_params_disp)
1216        self._reset_parameters_state(self.orientation_params,
1217                                     state.orientation_params)
1218        self._reset_parameters_state(self.str_parameters,
1219                                     state.str_parameters)
1220        self._reset_parameters_state(self.parameters, state.parameters)
1221        # display dispersion info layer
1222        self.enable_disp.SetValue(state.enable_disp)
1223        self.disable_disp.SetValue(state.disable_disp)
1224        # If the polydispersion is ON
1225        if state.enable_disp:
1226            # reset dispersion according the state
1227            self._set_dipers_Param(event=None)
1228            self._reset_page_disp_helper(state)
1229        # plotting range restore
1230        self._reset_plotting_range(state)
1231        # smearing info  restore
1232        if hasattr(self, "enable_smearer"):
1233            # set smearing value whether or not the data
1234            # contain the smearing info
1235            self.enable_smearer.SetValue(state.enable_smearer)
1236            self.disable_smearer.SetValue(state.disable_smearer)
1237            self.onSmear(event=None)
1238        self.pinhole_smearer.SetValue(state.pinhole_smearer)
1239        self.slit_smearer.SetValue(state.slit_smearer)
1240        try:
1241            self.dI_noweight.SetValue(state.dI_noweight)
1242            self.dI_didata.SetValue(state.dI_didata)
1243            self.dI_sqrdata.SetValue(state.dI_sqrdata)
1244            self.dI_idata.SetValue(state.dI_idata)
1245        except:
1246            # to support older state file formats
1247            self.dI_noweight.SetValue(False)
1248            self.dI_didata.SetValue(True)
1249            self.dI_sqrdata.SetValue(False)
1250            self.dI_idata.SetValue(False)
1251
1252        # we have two more options for smearing
1253        if self.pinhole_smearer.GetValue():
1254            self.dx_min = state.dx_min
1255            self.dx_max = state.dx_max
1256            if self.dx_min is not None:
1257                self.smear_pinhole_min.SetValue(str(self.dx_min))
1258            if self.dx_max is not None:
1259                self.smear_pinhole_max.SetValue(str(self.dx_max))
1260            self.onPinholeSmear(event=None)
1261        elif self.slit_smearer.GetValue():
1262            self.dxl = state.dxl
1263            self.dxw = state.dxw
1264            if self.dxl is not None:
1265                self.smear_slit_height.SetValue(str(self.dxl))
1266            if self.dxw is not None:
1267                self.smear_slit_width.SetValue(str(self.dxw))
1268            else:
1269                self.smear_slit_width.SetValue('')
1270            self.onSlitSmear(event=None)
1271
1272        # reset state of checkbox,textcrtl  and dispersity parameters value
1273        self._reset_parameters_state(self.fittable_param, state.fittable_param)
1274        self._reset_parameters_state(self.fixed_param, state.fixed_param)
1275
1276        # draw the model with previous parameters value
1277        self._onparamEnter_helper()
1278        # reset the value of chisqr when not consistent with the value computed
1279        self.tcChi.SetValue(str(self.state.tcChi))
1280        # reset context menu items
1281        self._reset_context_menu()
1282
1283        # set the value of the current state to the state given as parameter
1284        self.state = state.clone()
1285        self.state.m_name = self.m_name
1286
1287    def _reset_page_disp_helper(self, state):
1288        """
1289        Help to rest page for dispersions
1290        """
1291        keys = self.model.getParamList()
1292        for item in keys:
1293            if item in self.disp_list and \
1294                            item not in self.model.details:
1295                self.model.details[item] = ["", None, None]
1296        # for k,v in self.state.disp_cb_dict.iteritems():
1297        self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1298        self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1299        self.values = copy.deepcopy(state.values)
1300        self.weights = copy.deepcopy(state.weights)
1301
1302        for key, disp_type in state._disp_obj_dict.iteritems():
1303            # disp_model = disp
1304            disp_model = POLYDISPERSITY_MODELS[disp_type]()
1305            self._disp_obj_dict[key] = disp_model
1306            param_name = key.split('.')[0]
1307            # Try to set dispersion only when available
1308            # for eg., pass the orient. angles for 1D Cal
1309            try:
1310                self.model.set_dispersion(param_name, disp_model)
1311                self.model._persistency_dict[key] = \
1312                    [state.values, state.weights]
1313            except Exception:
1314                logging.error(traceback.format_exc())
1315            selection = self._find_polyfunc_selection(disp_model)
1316            for list in self.fittable_param:
1317                if list[1] == key and list[7] is not None:
1318                    list[7].SetSelection(selection)
1319                    # For the array disp_model, set the values and weights
1320                    if selection == 1:
1321                        disp_model.set_weights(self.values[key],
1322                                               self.weights[key])
1323                        try:
1324                            # Diables all fittable params for array
1325                            list[0].SetValue(False)
1326                            list[0].Disable()
1327                            list[2].Disable()
1328                            list[5].Disable()
1329                            list[6].Disable()
1330                        except Exception:
1331                            logging.error(traceback.format_exc())
1332            # For array, disable all fixed params
1333            if selection == 1:
1334                for item in self.fixed_param:
1335                    if item[1].split(".")[0] == key.split(".")[0]:
1336                        # try it and pass it for the orientation for 1D
1337                        try:
1338                            item[2].Disable()
1339                        except Exception:
1340                            logging.error(traceback.format_exc())
1341
1342    def _selectDlg(self):
1343        """
1344        open a dialog file to selected the customized dispersity
1345        """
1346        if self.parent is not None:
1347            self._default_save_location = \
1348                        self._manager.parent.get_save_location()
1349        dlg = wx.FileDialog(self, "Choose a weight file",
1350                            self._default_save_location, "",
1351                            "*.*", wx.OPEN)
1352        path = None
1353        if dlg.ShowModal() == wx.ID_OK:
1354            path = dlg.GetPath()
1355        dlg.Destroy()
1356        return path
1357
1358    def _reset_context_menu(self):
1359        """
1360        reset the context menu
1361        """
1362        ids = iter(self._id_pool)  # Reusing ids for context menu
1363        for name, _ in self.state.saved_states.iteritems():
1364            self.number_saved_state += 1
1365            # Add item in the context menu
1366            wx_id = ids.next()
1367            msg = 'Save model and state %g' % self.number_saved_state
1368            self.popUpMenu.Append(wx_id, name, msg)
1369            wx.EVT_MENU(self, wx_id, self.onResetModel)
1370
1371    def _reset_plotting_range(self, state):
1372        """
1373        Reset the plotting range to a given state
1374        """
1375        self.qmin.SetValue(str(state.qmin))
1376        self.qmax.SetValue(str(state.qmax))
1377
1378    def _save_typeOfmodel(self):
1379        """
1380        save radiobutton containing the type model that can be selected
1381        """
1382        # self.state.shape_rbutton = self.shape_rbutton.GetValue()
1383        # self.state.shape_indep_rbutton = self.shape_indep_rbutton.GetValue()
1384        # self.state.struct_rbutton = self.struct_rbutton.GetValue()
1385        # self.state.plugin_rbutton = self.plugin_rbutton.GetValue()
1386        self.state.structurecombobox = self.structurebox.GetLabel()
1387        self.state.formfactorcombobox = self.formfactorbox.GetLabel()
1388        self.state.categorycombobox = self.categorybox.GetLabel()
1389
1390        # post state to fit panel
1391        event = PageInfoEvent(page=self)
1392        wx.PostEvent(self.parent, event)
1393
1394    def _save_plotting_range(self):
1395        """
1396        save the state of plotting range
1397        """
1398        self.state.qmin = self.qmin_x
1399        self.state.qmax = self.qmax_x
1400        self.state.npts = self.npts_x
1401
1402    def _onparamEnter_helper(self, is_modified=False):
1403        """
1404        check if values entered by the user are changed and valid to replot
1405        model
1406        """
1407        # Flag to register when a parameter has changed.
1408        # is_modified = False
1409        self.fitrange = True
1410        is_2Ddata = False
1411        # self._undo.Enable(True)
1412        # check if 2d data
1413        if self.data.__class__.__name__ == "Data2D":
1414            is_2Ddata = True
1415        if self.model is not None:
1416            # Either we get a is_modified = True passed in because
1417            # _update_paramv_on_fit() has been called already or
1418            # we need to check here ourselves.
1419            if not is_modified:
1420                is_modified = (self._check_value_enter(self.fittable_param)
1421                               or self._check_value_enter(self.fixed_param)
1422                               or self._check_value_enter(self.parameters))
1423
1424            # Here we should check whether the boundaries have been modified.
1425            # If qmin and qmax have been modified, update qmin and qmax and
1426            # set the is_modified flag to True
1427            if self._validate_qrange(self.qmin, self.qmax):
1428                tempmin = float(self.qmin.GetValue())
1429                if tempmin != self.qmin_x:
1430                    self.qmin_x = tempmin
1431                    is_modified = True
1432                tempmax = float(self.qmax.GetValue())
1433                if tempmax != self.qmax_x:
1434                    self.qmax_x = tempmax
1435                    is_modified = True
1436
1437                if is_2Ddata:
1438                    # set mask
1439                    is_modified = self._validate_Npts()
1440
1441            else:
1442                self.fitrange = False
1443
1444            # if any value is modify draw model with new value
1445            if not self.fitrange:
1446                # self.btFit.Disable()
1447                if is_2Ddata:
1448                    self.btEditMask.Disable()
1449            else:
1450                if is_2Ddata and self.data.is_data and not self.batch_on:
1451                    self.btEditMask.Enable(True)
1452            if is_modified and self.fitrange:
1453                # Theory case: need to get npts value to draw
1454                self.npts_x = float(self.Npts_total.GetValue())
1455                self.create_default_data()
1456                self.state_change = True
1457                self._draw_model()
1458                #Time delay introduced to prevent double compilation on systems
1459                #without OPENCL. It seems to be threading related issue that
1460                #triggers compilation twice at the same time
1461                if HAVE_OPENCL == False:
1462                    time.sleep(0.1)
1463                self.Refresh()
1464
1465        # logging.info("is_modified flag set to %g",is_modified)
1466        return is_modified
1467
1468    def _update_paramv_on_fit(self):
1469        """
1470        make sure that update param values just before the fitting
1471        """
1472        # flag for qmin qmax check values
1473        flag = True
1474        self.fitrange = True
1475        is_modified = False
1476
1477        # wx.PostEvent(self._manager.parent, StatusEvent(status=" \
1478        # updating ... ",type="update"))
1479
1480        # So make sure that update param values on_Fit.
1481        # self._undo.Enable(True)
1482        if self.model is not None:
1483            if self.Npts_total.GetValue() != self.Npts_fit.GetValue():
1484                if not self.data.is_data:
1485                    self._manager.page_finder[self.uid].set_fit_data(
1486                        data=[self.data])
1487            # Check the values
1488            is_modified = (self._check_value_enter(self.fittable_param)
1489                           or self._check_value_enter(self.fixed_param)
1490                           or self._check_value_enter(self.parameters))
1491
1492            # If qmin and qmax have been modified, update qmin and qmax and
1493            # Here we should check whether the boundaries have been modified.
1494            # If qmin and qmax have been modified, update qmin and qmax and
1495            # set the is_modified flag to True
1496            self.fitrange = self._validate_qrange(self.qmin, self.qmax)
1497            if self.fitrange:
1498                tempmin = float(self.qmin.GetValue())
1499                if tempmin != self.qmin_x:
1500                    self.qmin_x = tempmin
1501                tempmax = float(self.qmax.GetValue())
1502                if tempmax != self.qmax_x:
1503                    self.qmax_x = tempmax
1504                if tempmax == tempmin:
1505                    flag = False
1506                temp_smearer = None
1507                if not self.disable_smearer.GetValue():
1508                    temp_smearer = self.current_smearer
1509                    if self.slit_smearer.GetValue():
1510                        flag = self.update_slit_smear()
1511                    elif self.pinhole_smearer.GetValue():
1512                        flag = self.update_pinhole_smear()
1513                    else:
1514                        enable_smearer = not self.disable_smearer.GetValue()
1515                        self._manager.set_smearer(smearer=temp_smearer,
1516                                                  uid=self.uid,
1517                                                  fid=self.data.id,
1518                                                  qmin=float(self.qmin_x),
1519                                                  qmax=float(self.qmax_x),
1520                                                  enable_smearer=enable_smearer,
1521                                                  draw=False)
1522                elif not self._is_2D():
1523                    enable_smearer = not self.disable_smearer.GetValue()
1524                    self._manager.set_smearer(smearer=temp_smearer,
1525                                              qmin=float(self.qmin_x),
1526                                              uid=self.uid,
1527                                              fid=self.data.id,
1528                                              qmax=float(self.qmax_x),
1529                                              enable_smearer=enable_smearer,
1530                                              draw=False)
1531                    if self.data is not None:
1532                        index_data = ((self.qmin_x <= self.data.x) &
1533                                      (self.data.x <= self.qmax_x))
1534                        val = str(len(self.data.x[index_data is True]))
1535                        self.Npts_fit.SetValue(val)
1536                    else:
1537                        # No data in the panel
1538                        try:
1539                            self.npts_x = float(self.Npts_total.GetValue())
1540                        except:
1541                            flag = False
1542                            return flag
1543                    flag = True
1544                if self._is_2D():
1545                    # only 2D case set mask
1546                    flag = self._validate_Npts()
1547                    if not flag:
1548                        return flag
1549            else:
1550                flag = False
1551        else:
1552            flag = False
1553
1554        # For invalid q range, disable the mask editor and fit button, vs.
1555        if not self.fitrange:
1556            if self._is_2D():
1557                self.btEditMask.Disable()
1558        else:
1559            if self._is_2D() and self.data.is_data and not self.batch_on:
1560                self.btEditMask.Enable(True)
1561
1562        if not flag:
1563            msg = "Cannot Plot or Fit :Must select a "
1564            msg += " model or Fitting range is not valid!!!  "
1565            wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1566
1567        try:
1568            self.save_current_state()
1569        except Exception:
1570            logging.error(traceback.format_exc())
1571
1572        return flag, is_modified
1573
1574    def _reset_parameters_state(self, listtorestore, statelist):
1575        """
1576        Reset the parameters at the given state
1577        """
1578        if len(statelist) == 0 or len(listtorestore) == 0:
1579            return
1580        if len(statelist) != len(listtorestore):
1581            return
1582
1583        for j in range(len(listtorestore)):
1584            item_page = listtorestore[j]
1585            item_page_info = statelist[j]
1586            # change the state of the check box for simple parameters
1587            if item_page[0] is not None:
1588                item_page[0].SetValue(item_page_info[0])
1589            if item_page[2] is not None:
1590                item_page[2].SetValue(item_page_info[2])
1591                if item_page[2].__class__.__name__ == "ComboBox":
1592                    if item_page_info[2] in self.model.fun_list:
1593                        fun_val = self.model.fun_list[item_page_info[2]]
1594                        self.model.setParam(item_page_info[1], fun_val)
1595            if item_page[3] is not None:
1596                # show or hide text +/-
1597                if item_page_info[2]:
1598                    item_page[3].Show(True)
1599                else:
1600                    item_page[3].Hide()
1601            if item_page[4] is not None:
1602                # show of hide the text crtl for fitting error
1603                if item_page_info[4][0]:
1604                    item_page[4].Show(True)
1605                    item_page[4].SetValue(item_page_info[4][1])
1606                else:
1607                    item_page[3].Hide()
1608            if item_page[5] is not None:
1609                # show of hide the text crtl for fitting error
1610                item_page[5].Show(item_page_info[5][0])
1611                item_page[5].SetValue(item_page_info[5][1])
1612
1613            if item_page[6] is not None:
1614                # show of hide the text crtl for fitting error
1615                item_page[6].Show(item_page_info[6][0])
1616                item_page[6].SetValue(item_page_info[6][1])
1617
1618    def _reset_strparam_state(self, listtorestore, statelist):
1619        """
1620        Reset the string parameters at the given state
1621        """
1622        if len(statelist) == 0:
1623            return
1624
1625        listtorestore = copy.deepcopy(statelist)
1626
1627        for j in range(len(listtorestore)):
1628            item_page = listtorestore[j]
1629            item_page_info = statelist[j]
1630            # change the state of the check box for simple parameters
1631
1632            if item_page[0] is not None:
1633                item_page[0].SetValue(format_number(item_page_info[0], True))
1634
1635            if item_page[2] is not None:
1636                param_name = item_page_info[1]
1637                value = item_page_info[2]
1638                selection = value
1639                if value in self.model.fun_list:
1640                    selection = self.model.fun_list[value]
1641                item_page[2].SetValue(selection)
1642                self.model.setParam(param_name, selection)
1643
1644    def _copy_parameters_state(self, listtocopy, statelist):
1645        """
1646        copy the state of button
1647
1648        :param listtocopy: the list of check button to copy
1649        :param statelist: list of state object to store the current state
1650
1651        """
1652        if len(listtocopy) == 0:
1653            return
1654
1655        for item in listtocopy:
1656
1657            checkbox_state = None
1658            if item[0] is not None:
1659                checkbox_state = item[0].GetValue()
1660            parameter_name = item[1]
1661            parameter_value = None
1662            if item[2] is not None:
1663                parameter_value = item[2].GetValue()
1664            static_text = None
1665            if item[3] is not None:
1666                static_text = item[3].IsShown()
1667            error_value = None
1668            error_state = None
1669            if item[4] is not None:
1670                error_value = item[4].GetValue()
1671                error_state = item[4].IsShown()
1672
1673            min_value = None
1674            min_state = None
1675            if item[5] is not None:
1676                min_value = item[5].GetValue()
1677                min_state = item[5].IsShown()
1678
1679            max_value = None
1680            max_state = None
1681            if item[6] is not None:
1682                max_value = item[6].GetValue()
1683                max_state = item[6].IsShown()
1684            unit = None
1685            if item[7] is not None:
1686                unit = item[7].GetLabel()
1687
1688            statelist.append([checkbox_state, parameter_name, parameter_value,
1689                              static_text, [error_state, error_value],
1690                              [min_state, min_value],
1691                              [max_state, max_value], unit])
1692
1693    def _draw_model(self, update_chisqr=True, source='model'):
1694        """
1695        Method to draw or refresh a plotted model.
1696        The method will use the data member from the model page
1697        to build a call to the fitting perspective manager.
1698
1699        :param chisqr: update chisqr value [bool]
1700        """
1701        wx.CallAfter(self._draw_model_after, update_chisqr, source)
1702
1703    def _draw_model_after(self, update_chisqr=True, source='model'):
1704        """
1705        Method to draw or refresh a plotted model.
1706        The method will use the data member from the model page
1707        to build a call to the fitting perspective manager.
1708
1709        :param chisqr: update chisqr value [bool]
1710        """
1711        # if self.check_invalid_panel():
1712        #    return
1713        if self.model is not None:
1714            temp_smear = None
1715            if hasattr(self, "enable_smearer"):
1716                if not self.disable_smearer.GetValue():
1717                    temp_smear = self.current_smearer
1718            # compute weight for the current data
1719            from sas.sasgui.perspectives.fitting.utils import get_weight
1720            flag = self.get_weight_flag()
1721            weight = get_weight(data=self.data, is2d=self._is_2D(), flag=flag)
1722            toggle_mode_on = self.model_view.IsEnabled()
1723            is_2d = self._is_2D()
1724            self._manager.draw_model(self.model,
1725                                     data=self.data,
1726                                     smearer=temp_smear,
1727                                     qmin=float(self.qmin_x),
1728                                     qmax=float(self.qmax_x),
1729                                     page_id=self.uid,
1730                                     toggle_mode_on=toggle_mode_on,
1731                                     state=self.state,
1732                                     enable2D=is_2d,
1733                                     update_chisqr=update_chisqr,
1734                                     source='model',
1735                                     weight=weight)
1736
1737    def _on_show_sld(self, event=None):
1738        """
1739        Plot SLD profile
1740        """
1741        # get profile data
1742        x, y = self.model.getProfile()
1743
1744        from sas.sasgui.plottools import Data1D as pf_data1d
1745        # from sas.sasgui.perspectives.theory.profile_dialog import SLDPanel
1746        from sas.sasgui.guiframe.local_perspectives.plotting.profile_dialog \
1747            import SLDPanel
1748        sld_data = pf_data1d(x, y)
1749        sld_data.name = 'SLD'
1750        sld_data.axes = self.sld_axes
1751        self.panel = SLDPanel(self, data=sld_data, axes=self.sld_axes,
1752                              id=wx.ID_ANY)
1753        self.panel.ShowModal()
1754
1755    def _set_multfactor_combobox(self, multiplicity=10):
1756        """
1757        Set comboBox for muitfactor of CoreMultiShellModel
1758        :param multiplicit: no. of multi-functionality
1759        """
1760        # build content of the combobox
1761        for idx in range(0, multiplicity):
1762            self.multifactorbox.Append(str(idx), int(idx))
1763        self._hide_multfactor_combobox()
1764
1765    def _show_multfactor_combobox(self):
1766        """
1767        Show the comboBox of muitfactor of CoreMultiShellModel
1768        """
1769        if not self.mutifactor_text.IsShown():
1770            self.mutifactor_text.Show(True)
1771            self.mutifactor_text1.Show(True)
1772        if not self.multifactorbox.IsShown():
1773            self.multifactorbox.Show(True)
1774
1775    def _hide_multfactor_combobox(self):
1776        """
1777        Hide the comboBox of muitfactor of CoreMultiShellModel
1778        """
1779        if self.mutifactor_text.IsShown():
1780            self.mutifactor_text.Hide()
1781            self.mutifactor_text1.Hide()
1782        if self.multifactorbox.IsShown():
1783            self.multifactorbox.Hide()
1784
1785    def formfactor_combo_init(self):
1786        """
1787        First time calls _show_combox_helper
1788        """
1789        self._show_combox(None)
1790
1791    def _show_combox_helper(self):
1792        """
1793        Fill panel's combo box according to the type of model selected
1794        """
1795        custom_model = 'Customized Models'
1796        mod_cat = self.categorybox.GetStringSelection()
1797        self.structurebox.SetSelection(0)
1798        self.structurebox.Disable()
1799        self.formfactorbox.Clear()
1800        if mod_cat is None:
1801            return
1802        m_list = []
1803        try:
1804            if mod_cat == custom_model:
1805                for model in self.model_list_box[mod_cat]:
1806                    m_list.append(self.model_dict[model.name])
1807            else:
1808                cat_dic = self.master_category_dict[mod_cat]
1809                for (model, enabled) in cat_dic:
1810                    if enabled:
1811                        m_list.append(self.model_dict[model])
1812        except Exception:
1813            msg = traceback.format_exc()
1814            wx.PostEvent(self._manager.parent,
1815                         StatusEvent(status=msg, info="error"))
1816        self._populate_box(self.formfactorbox, m_list)
1817
1818    def _on_modify_cat(self, event=None):
1819        """
1820        Called when category manager is opened
1821        """
1822        self._manager.parent.on_category_panel(event)
1823
1824    def _show_combox(self, event=None):
1825        """
1826        Show combox box associate with type of model selected
1827        """
1828        self.Show(False)
1829        self._show_combox_helper()
1830        self._on_select_model(event=None)
1831        self.Show(True)
1832        self._save_typeOfmodel()
1833        self.sizer4_4.Layout()
1834        self.sizer4.Layout()
1835        self.Layout()
1836        self.Refresh()
1837
1838    def _populate_box(self, combobox, list):
1839        """
1840        fill combox box with dict item
1841
1842        :param list: contains item to fill the combox
1843            item must model class
1844        """
1845        mlist = []
1846        for models in list:
1847            if models.name != "NoStructure":
1848                mlist.append((models.name, models))
1849
1850        # Sort the models
1851        mlist_sorted = sorted(mlist)
1852        for item in mlist_sorted:
1853            combobox.Append(item[0], item[1])
1854        return 0
1855
1856    def _onQrangeEnter(self, event):
1857        """
1858        Check validity of value enter in the Q range field
1859
1860        """
1861        tcrtl = event.GetEventObject()
1862        # Clear msg if previously shown.
1863        msg = ""
1864        wx.PostEvent(self.parent, StatusEvent(status=msg))
1865        # Flag to register when a parameter has changed.
1866        if tcrtl.GetValue().lstrip().rstrip() != "":
1867            try:
1868                float(tcrtl.GetValue())
1869                tcrtl.SetBackgroundColour(wx.WHITE)
1870                # If qmin and qmax have been modified, update qmin and qmax
1871                if self._validate_qrange(self.qmin, self.qmax):
1872                    tempmin = float(self.qmin.GetValue())
1873                    if tempmin != self.qmin_x:
1874                        self.qmin_x = tempmin
1875                    tempmax = float(self.qmax.GetValue())
1876                    if tempmax != self.qmax_x:
1877                        self.qmax_x = tempmax
1878                else:
1879                    tcrtl.SetBackgroundColour("pink")
1880                    msg = "Model Error: wrong value entered: %s" % \
1881                          sys.exc_info()[1]
1882                    wx.PostEvent(self.parent, StatusEvent(status=msg))
1883                    return
1884            except:
1885                tcrtl.SetBackgroundColour("pink")
1886                msg = "Model Error: wrong value entered: %s" % sys.exc_info()[1]
1887                wx.PostEvent(self.parent, StatusEvent(status=msg))
1888                return
1889            # Check if # of points for theory model are valid(>0).
1890            if self.npts is not None:
1891                if check_float(self.npts):
1892                    temp_npts = float(self.npts.GetValue())
1893                    if temp_npts != self.num_points:
1894                        self.num_points = temp_npts
1895                else:
1896                    msg = "Cannot plot: No points in Q range!!!  "
1897                    wx.PostEvent(self.parent, StatusEvent(status=msg))
1898        else:
1899            tcrtl.SetBackgroundColour("pink")
1900            msg = "Model Error: wrong value entered!!!"
1901            wx.PostEvent(self.parent, StatusEvent(status=msg))
1902        self.save_current_state()
1903        event = PageInfoEvent(page=self)
1904        wx.PostEvent(self.parent, event)
1905        self.state_change = False
1906        # Draw the model for a different range
1907        if not self.data.is_data:
1908            self.create_default_data()
1909        self._draw_model()
1910
1911    def _theory_qrange_enter(self, event):
1912        """
1913        Check validity of value enter in the Q range field
1914        """
1915
1916        tcrtl = event.GetEventObject()
1917        # Clear msg if previously shown.
1918        msg = ""
1919        wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1920        # Flag to register when a parameter has changed.
1921        is_modified = False
1922        if tcrtl.GetValue().lstrip().rstrip() != "":
1923            try:
1924                value = float(tcrtl.GetValue())
1925                tcrtl.SetBackgroundColour(wx.WHITE)
1926
1927                # If qmin and qmax have been modified, update qmin and qmax
1928                if self._validate_qrange(self.theory_qmin, self.theory_qmax):
1929                    tempmin = float(self.theory_qmin.GetValue())
1930                    if tempmin != self.theory_qmin_x:
1931                        self.theory_qmin_x = tempmin
1932                    tempmax = float(self.theory_qmax.GetValue())
1933                    if tempmax != self.qmax_x:
1934                        self.theory_qmax_x = tempmax
1935                else:
1936                    tcrtl.SetBackgroundColour("pink")
1937                    msg = "Model Error: wrong value entered: %s" % \
1938                          sys.exc_info()[1]
1939                    wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1940                    return
1941            except:
1942                tcrtl.SetBackgroundColour("pink")
1943                msg = "Model Error: wrong value entered: %s" % sys.exc_info()[1]
1944                wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1945                return
1946            # Check if # of points for theory model are valid(>0).
1947            if self.Npts_total.IsEditable():
1948                if check_float(self.Npts_total):
1949                    temp_npts = float(self.Npts_total.GetValue())
1950                    if temp_npts != self.num_points:
1951                        self.num_points = temp_npts
1952                        is_modified = True
1953                else:
1954                    msg = "Cannot Plot: No points in Q range!!!  "
1955                    wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1956        else:
1957            tcrtl.SetBackgroundColour("pink")
1958            msg = "Model Error: wrong value entered!!!"
1959            wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1960        self.save_current_state()
1961        event = PageInfoEvent(page=self)
1962        wx.PostEvent(self.parent, event)
1963        self.state_change = False
1964        # Draw the model for a different range
1965        self.create_default_data()
1966        self._draw_model()
1967
1968    def _on_select_model_helper(self):
1969        """
1970        call back for model selection
1971        """
1972        # reset dictionary containing reference to dispersion
1973        self._disp_obj_dict = {}
1974        self.disp_cb_dict = {}
1975        self.temp_multi_functional = False
1976        f_id = self.formfactorbox.GetCurrentSelection()
1977        # For MAC
1978        form_factor = None
1979        if f_id >= 0:
1980            form_factor = self.formfactorbox.GetClientData(f_id)
1981
1982        if form_factor is None or \
1983            not hasattr(form_factor, 'is_form_factor') or \
1984                not form_factor.is_form_factor:
1985            self.structurebox.Hide()
1986            self.text2.Hide()
1987            self.structurebox.Disable()
1988            self.structurebox.SetSelection(0)
1989            self.text2.Disable()
1990        else:
1991            self.structurebox.Show()
1992            self.text2.Show()
1993            self.structurebox.Enable()
1994            self.text2.Enable()
1995
1996        if form_factor is not None:
1997            # set multifactor for Mutifunctional models
1998            if form_factor.is_multiplicity_model:
1999                m_id = self.multifactorbox.GetCurrentSelection()
2000                multiplicity = form_factor.multiplicity_info[0]
2001                self.multifactorbox.Clear()
2002                self._set_multfactor_combobox(multiplicity)
2003                self._show_multfactor_combobox()
2004                # ToDo: this info should be called directly from the model
2005                text = form_factor.multiplicity_info[1]  # 'No. of Shells: '
2006
2007                self.mutifactor_text.SetLabel(text)
2008                if m_id > multiplicity - 1:
2009                    # default value
2010                    m_id = 1
2011
2012                self.multi_factor = self.multifactorbox.GetClientData(m_id)
2013                if self.multi_factor is None:
2014                    self.multi_factor = 0
2015                self.multifactorbox.SetSelection(m_id)
2016                # Check len of the text1 and max_multiplicity
2017                text = ''
2018                if form_factor.multiplicity_info[0] == \
2019                        len(form_factor.multiplicity_info[2]):
2020                    text = form_factor.multiplicity_info[2][self.multi_factor]
2021                self.mutifactor_text1.SetLabel(text)
2022                # Check if model has  get sld profile.
2023                if len(form_factor.multiplicity_info[3]) > 0:
2024                    self.sld_axes = form_factor.multiplicity_info[3]
2025                    self.show_sld_button.Show(True)
2026                else:
2027                    self.sld_axes = ""
2028            else:
2029                self._hide_multfactor_combobox()
2030                self.show_sld_button.Hide()
2031                self.multi_factor = None
2032        else:
2033            self._hide_multfactor_combobox()
2034            self.show_sld_button.Hide()
2035            self.multi_factor = None
2036
2037        s_id = self.structurebox.GetCurrentSelection()
2038        struct_factor = self.structurebox.GetClientData(s_id)
2039
2040        if struct_factor is not None:
2041            from sasmodels.sasview_model import MultiplicationModel
2042            self.model = MultiplicationModel(form_factor(self.multi_factor),
2043                                             struct_factor())
2044            # multifunctional form factor
2045            if len(form_factor.non_fittable) > 0:
2046                self.temp_multi_functional = True
2047        elif form_factor is not None:
2048            if self.multi_factor is not None:
2049                self.model = form_factor(self.multi_factor)
2050            else:
2051                # old style plugin models do not accept a multiplicity argument
2052                self.model = form_factor()
2053        else:
2054            self.model = None
2055            return
2056
2057        # check if model has magnetic parameters
2058        if len(self.model.magnetic_params) > 0:
2059            self._has_magnetic = True
2060        else:
2061            self._has_magnetic = False
2062        # post state to fit panel
2063        self.state.parameters = []
2064        self.state.model = self.model
2065        self.state.qmin = self.qmin_x
2066        self.state.multi_factor = self.multi_factor
2067        self.disp_list = self.model.getDispParamList()
2068        self.state.disp_list = self.disp_list
2069        self.on_set_focus(None)
2070        self.Layout()
2071
2072    def _validate_qrange(self, qmin_ctrl, qmax_ctrl):
2073        """
2074        Verify that the Q range controls have valid values
2075        and that Qmin < Qmax.
2076
2077        :param qmin_ctrl: text control for Qmin
2078        :param qmax_ctrl: text control for Qmax
2079
2080        :return: True is the Q range is value, False otherwise
2081
2082        """
2083        qmin_validity = check_float(qmin_ctrl)
2084        qmax_validity = check_float(qmax_ctrl)
2085        if not (qmin_validity and qmax_validity):
2086            return False
2087        else:
2088            qmin = float(qmin_ctrl.GetValue())
2089            qmax = float(qmax_ctrl.GetValue())
2090            if qmin < qmax:
2091                # Make sure to set both colours white.
2092                qmin_ctrl.SetBackgroundColour(wx.WHITE)
2093                qmin_ctrl.Refresh()
2094                qmax_ctrl.SetBackgroundColour(wx.WHITE)
2095                qmax_ctrl.Refresh()
2096            else:
2097                qmin_ctrl.SetBackgroundColour("pink")
2098                qmin_ctrl.Refresh()
2099                qmax_ctrl.SetBackgroundColour("pink")
2100                qmax_ctrl.Refresh()
2101                msg = "Invalid Q range: Q min must be smaller than Q max"
2102                wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
2103                return False
2104        return True
2105
2106    def _validate_Npts(self):
2107        """
2108        Validate the number of points for fitting is more than 10 points.
2109        If valid, setvalues Npts_fit otherwise post msg.
2110        """
2111        # default flag
2112        flag = True
2113        # Theory
2114        if self.data is None and self.enable2D:
2115            return flag
2116        for data in self.data_list:
2117            # q value from qx and qy
2118            radius = numpy.sqrt(data.qx_data * data.qx_data +
2119                                data.qy_data * data.qy_data)
2120            # get unmasked index
2121            index_data = (float(self.qmin.GetValue()) <= radius) & \
2122                         (radius <= float(self.qmax.GetValue()))
2123            index_data = (index_data) & (data.mask)
2124            index_data = (index_data) & (numpy.isfinite(data.data))
2125
2126            if len(index_data[index_data]) < 10:
2127                # change the color pink.
2128                self.qmin.SetBackgroundColour("pink")
2129                self.qmin.Refresh()
2130                self.qmax.SetBackgroundColour("pink")
2131                self.qmax.Refresh()
2132                msg = "Data Error: "
2133                msg += "Too few points in %s." % data.name
2134                wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
2135                self.fitrange = False
2136                flag = False
2137            else:
2138                self.Npts_fit.SetValue(str(len(index_data[index_data is True])))
2139                self.fitrange = True
2140
2141        return flag
2142
2143    def _validate_Npts_1D(self):
2144        """
2145        Validate the number of points for fitting is more than 5 points.
2146        If valid, setvalues Npts_fit otherwise post msg.
2147        """
2148        # default flag
2149        flag = True
2150        # Theory
2151        if self.data is None:
2152            return flag
2153        for data in self.data_list:
2154            # q value from qx and qy
2155            radius = data.x
2156            # get unmasked index
2157            index_data = (float(self.qmin.GetValue()) <= radius) & \
2158                         (radius <= float(self.qmax.GetValue()))
2159            index_data = (index_data) & (numpy.isfinite(data.y))
2160
2161            if len(index_data[index_data]) < 5:
2162                # change the color pink.
2163                self.qmin.SetBackgroundColour("pink")
2164                self.qmin.Refresh()
2165                self.qmax.SetBackgroundColour("pink")
2166                self.qmax.Refresh()
2167                msg = "Data Error: "
2168                msg += "Too few points in %s." % data.name
2169                wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
2170                self.fitrange = False
2171                flag = False
2172            else:
2173                self.Npts_fit.SetValue(str(len(index_data[index_data is True])))
2174                self.fitrange = True
2175
2176        return flag
2177
2178    def _check_value_enter(self, list):
2179        """
2180        :param list: model parameter and panel info
2181        :Note: each item of the list should be as follow:
2182            item=[check button state, parameter's name,
2183                paramater's value, string="+/-",
2184                parameter's error of fit,
2185                parameter's minimum value,
2186                parameter's maximum value ,
2187                parameter's units]
2188
2189        Returns True if the model parameters have changed.
2190        """
2191        is_modified = False
2192        for item in list:
2193            # skip angle parameters for 1D
2194            if not self.enable2D and item in self.orientation_params:
2195                continue
2196
2197            value_ctrl = item[2]
2198            if not value_ctrl.IsEnabled():
2199                # ArrayDispersion disables PD, Min, Max, Npts, Nsigs
2200                continue
2201
2202            name = item[1]
2203            value_str = value_ctrl.GetValue().strip()
2204            if name.endswith(".npts"):
2205                validity = check_int(value_ctrl)
2206                if not validity:
2207                    continue
2208                value = int(value_str)
2209
2210            elif name.endswith(".nsigmas"):
2211                validity = check_float(value_ctrl)
2212                if not validity:
2213                    continue
2214                value = float(value_str)
2215
2216            else:  # value or polydispersity
2217
2218                # Check that min, max and value are floats
2219                min_ctrl, max_ctrl = item[5], item[6]
2220                min_str = min_ctrl.GetValue().strip()
2221                max_str = max_ctrl.GetValue().strip()
2222                validity = check_float(value_ctrl)
2223                if min_str != "":
2224                    validity = validity and check_float(min_ctrl)
2225                if max_str != "":
2226                    validity = validity and check_float(max_ctrl)
2227                if not validity:
2228                    continue
2229
2230                # Check that min is less than max
2231                low = -numpy.inf if min_str == "" else float(min_str)
2232                high = numpy.inf if max_str == "" else float(max_str)
2233                if high < low:
2234                    min_ctrl.SetBackgroundColour("pink")
2235                    min_ctrl.Refresh()
2236                    max_ctrl.SetBackgroundColour("pink")
2237                    max_ctrl.Refresh()
2238                    # msg = "Invalid fit range for %s: min must be smaller
2239                    # than max"%name
2240                    # wx.PostEvent(self._manager.parent,
2241                    # StatusEvent(status=msg))
2242                    continue
2243
2244                # Force value between min and max
2245                value = float(value_str)
2246                if value < low:
2247                    value = low
2248                    value_ctrl.SetValue(format_number(value))
2249                elif value > high:
2250                    value = high
2251                    value_ctrl.SetValue(format_number(value))
2252
2253                if name not in self.model.details.keys():
2254                    self.model.details[name] = ["", None, None]
2255                old_low, old_high = self.model.details[name][1:3]
2256                if old_low != low or old_high != high:
2257                    # The configuration has changed but it won't change the
2258                    # computed curve so no need to set is_modified to True
2259                    # is_modified = True
2260                    self.model.details[name][1:3] = low, high
2261
2262            # Update value in model if it has changed
2263            if value != self.model.getParam(name):
2264                self.model.setParam(name, value)
2265                is_modified = True
2266
2267        return is_modified
2268
2269    def _set_dipers_Param(self, event):
2270        """
2271        respond to self.enable_disp and self.disable_disp radio box.
2272        The dispersity object is reset inside the model into Gaussian.
2273        When the user select yes , this method display a combo box for
2274        more selection when the user selects No,the combo box disappears.
2275        Redraw the model with the default dispersity (Gaussian)
2276        """
2277        # On selction if no model exists.
2278        if self.model is None:
2279            self.disable_disp.SetValue(True)
2280            msg = "Please select a Model first..."
2281            wx.MessageBox(msg, 'Info')
2282            wx.PostEvent(self._manager.parent,
2283                         StatusEvent(status="Polydispersion: %s" % msg))
2284            return
2285
2286        self._reset_dispersity()
2287
2288        if self.model is None:
2289            self.model_disp.Hide()
2290            self.sizer4_4.Clear(True)
2291            return
2292
2293        if self.enable_disp.GetValue():
2294            # layout for model containing no dispersity parameters
2295
2296            self.disp_list = self.model.getDispParamList()
2297
2298            if len(self.disp_list) == 0 and len(self.disp_cb_dict) == 0:
2299                self._layout_sizer_noDipers()
2300            else:
2301                # set gaussian sizer
2302                self._on_select_Disp(event=None)
2303        else:
2304            self.sizer4_4.Clear(True)
2305
2306        # post state to fit panel
2307        self.save_current_state()
2308        if event is not None:
2309            event = PageInfoEvent(page=self)
2310            wx.PostEvent(self.parent, event)
2311        # draw the model with the current dispersity
2312
2313        # Wojtek P, Oct 8, 2016: Calling draw_model seems to be unessecary.
2314        # By comenting it we save an extra Iq calculation
2315        # self._draw_model()
2316
2317        # Need to use FitInside again here to replace the next four lines.
2318        # Otherwised polydispersity off does not resize the scrollwindow.
2319        # PDB Nov 28, 2015
2320        self.FitInside()
2321#        self.sizer4_4.Layout()
2322#        self.sizer5.Layout()
2323#        self.Layout()
2324#        self.Refresh()
2325
2326    def _layout_sizer_noDipers(self):
2327        """
2328        Draw a sizer with no dispersity info
2329        """
2330        ix = 0
2331        iy = 1
2332        self.fittable_param = []
2333        self.fixed_param = []
2334        self.orientation_params_disp = []
2335
2336        self.sizer4_4.Clear(True)
2337        text = "No polydispersity available for this model"
2338        model_disp = wx.StaticText(self, wx.ID_ANY, text)
2339        self.sizer4_4.Add(model_disp, (iy, ix), (1, 1),
2340                          wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 10)
2341        self.sizer4_4.Layout()
2342        self.sizer4.Layout()
2343
2344    def _reset_dispersity(self):
2345        """
2346        put gaussian dispersity into current model
2347        """
2348        if len(self.param_toFit) > 0:
2349            for item in self.fittable_param:
2350                if item in self.param_toFit:
2351                    self.param_toFit.remove(item)
2352
2353            for item in self.orientation_params_disp:
2354                if item in self.param_toFit:
2355                    self.param_toFit.remove(item)
2356
2357        self.fittable_param = []
2358        self.fixed_param = []
2359        self.orientation_params_disp = []
2360        self.values = {}
2361        self.weights = {}
2362
2363        # from sas.models.dispersion_models import GaussianDispersion
2364        from sasmodels.weights import GaussianDispersion
2365        if len(self.disp_cb_dict) == 0:
2366            self.save_current_state()
2367            self.sizer4_4.Clear(True)
2368            self.Layout()
2369            return
2370        if (len(self.disp_cb_dict) > 0):
2371            for p in self.disp_cb_dict:
2372                # The parameter was un-selected.
2373                # Go back to Gaussian model (with 0 pts)
2374                disp_model = GaussianDispersion()
2375
2376                self._disp_obj_dict[p] = disp_model
2377                # Set the new model as the dispersion object
2378                # for the selected parameter
2379                try:
2380                    self.model.set_dispersion(p, disp_model)
2381                except Exception:
2382                    logging.error(traceback.format_exc())
2383
2384        # save state into
2385        self.save_current_state()
2386        self.Layout()
2387        self.Refresh()
2388
2389    def _on_select_Disp(self, event):
2390        """
2391        allow selecting different dispersion
2392        self.disp_list should change type later .now only gaussian
2393        """
2394        self._set_sizer_dispersion()
2395
2396        # Redraw the model
2397        self._draw_model()
2398        # self._undo.Enable(True)
2399        event = PageInfoEvent(page=self)
2400        wx.PostEvent(self.parent, event)
2401
2402        self.sizer4_4.Layout()
2403        self.sizer4.Layout()
2404        self.SetupScrolling()
2405
2406    def _on_disp_func(self, event=None):
2407        """
2408        Select a distribution function for the polydispersion
2409
2410        :Param event: ComboBox event
2411        """
2412        # get ready for new event
2413        if event is not None:
2414            event.Skip()
2415        # Get event object
2416        disp_box = event.GetEventObject()
2417
2418        # Try to select a Distr. function
2419        try:
2420            disp_box.SetBackgroundColour("white")
2421            selection = disp_box.GetCurrentSelection()
2422            param_name = disp_box.Name.split('.')[0]
2423            disp_name = disp_box.GetValue()
2424            dispersity = disp_box.GetClientData(selection)
2425
2426            # disp_model =  GaussianDispersion()
2427            disp_model = dispersity()
2428            # Get param names to reset the values of the param
2429            name1 = param_name + ".width"
2430            name2 = param_name + ".npts"
2431            name3 = param_name + ".nsigmas"
2432            # Check Disp. function whether or not it is 'array'
2433            if disp_name.lower() == "array":
2434                value2 = ""
2435                value3 = ""
2436                value1 = self._set_array_disp(name=name1, disp=disp_model)
2437            else:
2438                self._del_array_values(name1)
2439                # self._reset_array_disp(param_name)
2440                self._disp_obj_dict[name1] = disp_model
2441                self.model.set_dispersion(param_name, disp_model)
2442                self.state._disp_obj_dict[name1] = disp_model.type
2443
2444                value1 = str(format_number(self.model.getParam(name1), True))
2445                value2 = str(format_number(self.model.getParam(name2)))
2446                value3 = str(format_number(self.model.getParam(name3)))
2447            # Reset fittable polydispersin parameter value
2448            for item in self.fittable_param:
2449                if item[1] == name1:
2450                    item[2].SetValue(value1)
2451                    item[5].SetValue("")
2452                    item[6].SetValue("")
2453                    # Disable for array
2454                    if disp_name.lower() == "array":
2455                        item[0].SetValue(False)
2456                        item[0].Disable()
2457                        item[2].Disable()
2458                        item[3].Show(False)
2459                        item[4].Show(False)
2460                        item[5].Disable()
2461                        item[6].Disable()
2462                    else:
2463                        item[0].Enable()
2464                        item[2].Enable()
2465                        item[3].Show(True)
2466                        item[4].Show(True)
2467                        item[5].Enable()
2468                        item[6].Enable()
2469                    break
2470            # Reset fixed polydispersion params
2471            for item in self.fixed_param:
2472                if item[1] == name2:
2473                    item[2].SetValue(value2)
2474                    # Disable Npts for array
2475                    if disp_name.lower() == "array":
2476                        item[2].Disable()
2477                    else:
2478                        item[2].Enable()
2479                if item[1] == name3:
2480                    item[2].SetValue(value3)
2481                    # Disable Nsigs for array
2482                    if disp_name.lower() == "array":
2483                        item[2].Disable()
2484                    else:
2485                        item[2].Enable()
2486
2487            # Make sure the check box updated
2488            self.get_all_checked_params()
2489
2490            # update params
2491            self._update_paramv_on_fit()
2492            # draw
2493            self._draw_model()
2494            self.Refresh()
2495        except Exception:
2496            logging.error(traceback.format_exc())
2497            # Error msg
2498            msg = "Error occurred:"
2499            msg += " Could not select the distribution function..."
2500            msg += " Please select another distribution function."
2501            disp_box.SetBackgroundColour("pink")
2502            # Focus on Fit button so that users can see the pinky box
2503            self.btFit.SetFocus()
2504            wx.PostEvent(self._manager.parent,
2505                         StatusEvent(status=msg, info="error"))
2506
2507    def _set_array_disp(self, name=None, disp=None):
2508        """
2509        Set array dispersion
2510
2511        :param name: name of the parameter for the dispersion to be set
2512        :param disp: the polydisperion object
2513        """
2514        # The user wants this parameter to be averaged.
2515        # Pop up the file selection dialog.
2516        path = self._selectDlg()
2517        # Array data
2518        values = []
2519        weights = []
2520        # If nothing was selected, just return
2521        if path is None:
2522            self.disp_cb_dict[name].SetValue(False)
2523            # self.noDisper_rbox.SetValue(True)
2524            return
2525        self._default_save_location = os.path.dirname(path)
2526        if self._manager is not None:
2527            self._manager.parent._default_save_location = \
2528                             self._default_save_location
2529
2530        basename = os.path.basename(path)
2531        values, weights = self.read_file(path)
2532
2533        # If any of the two arrays is empty, notify the user that we won't
2534        # proceed
2535        if len(self.param_toFit) > 0:
2536            if name in self.param_toFit:
2537                self.param_toFit.remove(name)
2538
2539        # Tell the user that we are about to apply the distribution
2540        msg = "Applying loaded %s distribution: %s" % (name, path)
2541        wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
2542        self._set_array_disp_model(name=name, disp=disp,
2543                                   values=values, weights=weights)
2544        return basename
2545
2546    def _set_array_disp_model(self, name=None, disp=None,
2547                              values=[], weights=[]):
2548        """
2549        Set array dispersion model
2550
2551        :param name: name of the parameter for the dispersion to be set
2552        :param disp: the polydisperion object
2553        """
2554        disp.set_weights(values, weights)
2555        self._disp_obj_dict[name] = disp
2556        self.model.set_dispersion(name.split('.')[0], disp)
2557        self.state._disp_obj_dict[name] = disp.type
2558        self.values[name] = values
2559        self.weights[name] = weights
2560        # Store the object to make it persist outside the
2561        # scope of this method
2562        # TODO: refactor model to clean this up?
2563        self.state.values = {}
2564        self.state.weights = {}
2565        self.state.values = copy.deepcopy(self.values)
2566        self.state.weights = copy.deepcopy(self.weights)
2567
2568        # Set the new model as the dispersion object for the
2569        # selected parameter
2570        # self.model.set_dispersion(p, disp_model)
2571        # Store a reference to the weights in the model object
2572        # so that
2573        # it's not lost when we use the model within another thread.
2574        self.state.model = self.model.clone()
2575        self.model._persistency_dict[name.split('.')[0]] = \
2576            [values, weights]
2577        self.state.model._persistency_dict[name.split('.')[0]] = \
2578            [values, weights]
2579
2580    def _del_array_values(self, name=None):
2581        """
2582        Reset array dispersion
2583
2584        :param name: name of the parameter for the dispersion to be set
2585        """
2586        # Try to delete values and weight of the names array dic if exists
2587        try:
2588            if name in self.values:
2589                del self.values[name]
2590                del self.weights[name]
2591                # delete all other dic
2592                del self.state.values[name]
2593                del self.state.weights[name]
2594                del self.model._persistency_dict[name.split('.')[0]]
2595                del self.state.model._persistency_dict[name.split('.')[0]]
2596        except Exception:
2597            logging.error(traceback.format_exc())
2598
2599    def _lay_out(self):
2600        """
2601        returns self.Layout
2602
2603        :Note: Mac seems to like this better when self.
2604            Layout is called after fitting.
2605        """
2606        self._sleep4sec()
2607        self.Layout()
2608        return
2609
2610    def _sleep4sec(self):
2611        """
2612            sleep for 1 sec only applied on Mac
2613            Note: This 1sec helps for Mac not to crash on self.
2614            Layout after self._draw_model
2615        """
2616        if ON_MAC is True:
2617            time.sleep(1)
2618
2619    def _find_polyfunc_selection(self, disp_func=None):
2620        """
2621        FInd Comboox selection from disp_func
2622
2623        :param disp_function: dispersion distr. function
2624        """
2625        # Find the selection
2626        if disp_func is not None:
2627            try:
2628                return POLYDISPERSITY_MODELS.values().index(disp_func.__class__)
2629            except ValueError:
2630                pass  # Fall through to default class
2631        return POLYDISPERSITY_MODELS.keys().index('gaussian')
2632
2633    def on_reset_clicked(self, event):
2634        """
2635        On 'Reset' button  for Q range clicked
2636        """
2637        flag = True
2638        # For 3 different cases: Data2D, Data1D, and theory
2639        if self.model is None:
2640            msg = "Please select a model first..."
2641            wx.MessageBox(msg, 'Info')
2642            flag = False
2643            return
2644
2645        elif self.data.__class__.__name__ == "Data2D":
2646            data_min = 0
2647            x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
2648            y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
2649            self.qmin_x = data_min
2650            self.qmax_x = math.sqrt(x * x + y * y)
2651            # self.data.mask = numpy.ones(len(self.data.data),dtype=bool)
2652            # check smearing
2653            if not self.disable_smearer.GetValue():
2654                # set smearing value whether or
2655                # not the data contain the smearing info
2656                if self.pinhole_smearer.GetValue():
2657                    flag = self.update_pinhole_smear()
2658                else:
2659                    flag = True
2660
2661        elif self.data is None:
2662            self.qmin_x = _QMIN_DEFAULT
2663            self.qmax_x = _QMAX_DEFAULT
2664            self.num_points = _NPTS_DEFAULT
2665            self.state.npts = self.num_points
2666
2667        elif self.data.__class__.__name__ != "Data2D":
2668            self.qmin_x = min(self.data.x)
2669            self.qmax_x = max(self.data.x)
2670            # check smearing
2671            if not self.disable_smearer.GetValue():
2672                # set smearing value whether or
2673                # not the data contain the smearing info
2674                if self.slit_smearer.GetValue():
2675                    flag = self.update_slit_smear()
2676                elif self.pinhole_smearer.GetValue():
2677                    flag = self.update_pinhole_smear()
2678                else:
2679                    flag = True
2680        else:
2681            flag = False
2682
2683        if flag is False:
2684            msg = "Cannot Plot :Must enter a number!!!  "
2685            wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
2686        else:
2687            # set relative text ctrs.
2688            self.qmin.SetValue(str(self.qmin_x))
2689            self.qmax.SetValue(str(self.qmax_x))
2690            self.show_npts2fit()
2691            # At this point, some button and variables satatus (disabled?)
2692            # should be checked such as color that should be reset to
2693            # white in case that it was pink.
2694            self._onparamEnter_helper()
2695
2696        self.save_current_state()
2697        self.state.qmin = self.qmin_x
2698        self.state.qmax = self.qmax_x
2699
2700        # reset the q range values
2701        self._reset_plotting_range(self.state)
2702        self._draw_model()
2703
2704    def select_log(self, event):
2705        """
2706        Log checked to generate log spaced points for theory model
2707        """
2708
2709    def get_images(self):
2710        """
2711        Get the images of the plots corresponding this panel for report
2712
2713        : return graphs: list of figures
2714        : Need Move to guiframe
2715        """
2716        # set list of graphs
2717        graphs = []
2718        canvases = []
2719        res_item = None
2720        # call gui_manager
2721        gui_manager = self._manager.parent
2722        # loops through the panels [dic]
2723        for _, item2 in gui_manager.plot_panels.iteritems():
2724            data_title = self.data.group_id
2725            # try to get all plots belonging to this control panel
2726            try:
2727                g_id = item2.group_id
2728                if g_id == data_title or \
2729                        str(g_id).count("res" + str(self.graph_id)) or \
2730                        str(g_id).count(str(self.uid)) > 0:
2731                    if str(g_id).count("res" + str(self.graph_id)) > 0:
2732                        res_item = [item2.figure, item2.canvas]
2733                    else:
2734                        # append to the list
2735                        graphs.append(item2.figure)
2736                        canvases.append(item2.canvas)
2737            except Exception:
2738                # Not for control panels
2739                logging.error(traceback.format_exc())
2740        # Make sure the resduals plot goes to the last
2741        if res_item is not None:
2742            graphs.append(res_item[0])
2743            canvases.append(res_item[1])
2744        # return the list of graphs
2745        return graphs, canvases
2746
2747    def on_function_help_clicked(self, event):
2748        """
2749        Function called when 'Help' button is pressed next to model
2750        of interest.  This calls DocumentationWindow from
2751        documentation_window.py. It will load the top level of the model
2752        help documenation sphinx generated html if no model is presented.
2753        If a model IS present then if documention for that model exists
2754        it will load to that  point otherwise again it will go to the top.
2755        For Wx2.8 and below is used (i.e. non-released through installer)
2756        a browser is loaded and the top of the model documentation only is
2757        accessible because webbrowser module does not pass anything after
2758        the # to the browser.
2759
2760        :param event: on Help Button pressed event
2761        """
2762
2763        if self.model is not None:
2764            name = self.formfactorbox.GetValue()
2765            _TreeLocation = 'user/models/' + name.lower()+'.html'
2766            _doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation,
2767                                              "", name + " Help")
2768        else:
2769            _TreeLocation = 'user/index.html'
2770            _doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation,
2771                                              "", "General Model Help")
2772
2773    def on_model_help_clicked(self, event):
2774        """
2775        Function called when 'Description' button is pressed next to model
2776        of interest.  This calls the Description embedded in the model. This
2777        should work with either Wx2.8 and lower or higher. If no model is
2778        selected it will give the message that a model must be chosen first
2779        in the box that would normally contain the description.  If a badly
2780        behaved model is encountered which has no description then it will
2781        give the message that none is available.
2782
2783        :param event: on Description Button pressed event
2784        """
2785
2786        if self.model is None:
2787            name = 'index.html'
2788        else:
2789            name = self.formfactorbox.GetValue()
2790
2791        msg = 'Model description:\n'
2792        info = "Info"
2793        if self.model is not None:
2794            # frame.Destroy()
2795            if str(self.model.description).rstrip().lstrip() == '':
2796                msg += "Sorry, no information is available for this model."
2797            else:
2798                msg += self.model.description + '\n'
2799            wx.MessageBox(msg, info)
2800        else:
2801            msg += "You must select a model to get information on this"
2802            wx.MessageBox(msg, info)
2803
2804    def _on_mag_angle_help(self, event):
2805        """
2806        Bring up Magnetic Angle definition bmp image whenever the ? button
2807        is clicked. Calls DocumentationWindow with the path of the location
2808        within the documentation tree (after /doc/ ....". When using old
2809        versions of Wx (i.e. before 2.9 and therefore not part of release
2810        versions distributed via installer) it brings up an image viewer
2811        box which allows the user to click through the rest of the images in
2812        the directory.  Not ideal but probably better than alternative which
2813        would bring up the entire discussion of how magnetic models work?
2814        Specially since it is not likely to be accessed.  The normal release
2815        versions bring up the normal image box.
2816
2817        :param evt: Triggers on clicking ? in Magnetic Angles? box
2818        """
2819
2820        _TreeLocation = "_images/M_angles_pic.bmp"
2821        _doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation, "",
2822                                          "Magnetic Angle Defintions")
2823
2824    def _on_mag_help(self, event):
2825        """
2826        Bring up Magnetic Angle definition bmp image whenever the ? button
2827        is clicked. Calls DocumentationWindow with the path of the location
2828        within the documentation tree (after /doc/ ....". When using old
2829        versions of Wx (i.e. before 2.9 and therefore not part of release
2830        versions distributed via installer) it brings up an image viewer
2831        box which allows the user to click through the rest of the images in
2832        the directory.  Not ideal but probably better than alternative which
2833        would bring up the entire discussion of how magnetic models work?
2834        Specially since it is not likely to be accessed.  The normal release
2835        versions bring up the normal image box.
2836
2837        :param evt: Triggers on clicking ? in Magnetic Angles? box
2838        """
2839
2840        _TreeLocation = "user/magnetism.html"
2841        _doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation, "",
2842                                          "Polarized Beam/Magnetc Help")
2843
2844    def _on_mag_on(self, event):
2845        """
2846        Magnetic Parameters ON/OFF
2847        """
2848        button = event.GetEventObject()
2849
2850        if button.GetLabel().count('ON') > 0:
2851            self.magnetic_on = True
2852            button.SetLabel("Magnetic OFF")
2853            m_value = 1.0e-06
2854            for key in self.model.magnetic_params:
2855                if key.count('M0') > 0:
2856                    self.model.setParam(key, m_value)
2857                    m_value += 0.5e-06
2858        else:
2859            self.magnetic_on = False
2860            button.SetLabel("Magnetic ON")
2861            for key in self.model.magnetic_params:
2862                if key.count('M0') > 0:
2863                    # reset mag value to zero fo safety
2864                    self.model.setParam(key, 0.0)
2865
2866        self.Show(False)
2867        self.set_model_param_sizer(self.model)
2868        # self._set_sizer_dispersion()
2869        self.state.magnetic_on = self.magnetic_on
2870        self.SetupScrolling()
2871        self.Show(True)
2872
2873    def on_pd_help_clicked(self, event):
2874        """
2875        Bring up Polydispersity Documentation whenever the ? button is clicked.
2876        Calls DocumentationWindow with the path of the location within the
2877        documentation tree (after /doc/ ....".  Note that when using old
2878        versions of Wx (before 2.9) and thus not the release version of
2879        istallers, the help comes up at the top level of the file as
2880        webbrowser does not pass anything past the # to the browser when it is
2881        running "file:///...."
2882
2883        :param event: Triggers on clicking ? in polydispersity box
2884        """
2885
2886        _TreeLocation = "user/sasgui/perspectives/fitting/pd_help.html"
2887        _PageAnchor = ""
2888        _doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation,
2889                                          _PageAnchor, "Polydispersity Help")
2890
2891    def on_left_down(self, event):
2892        """
2893        Get key stroke event
2894        """
2895        # Figuring out key combo: Cmd for copy, Alt for paste
2896        if event.CmdDown() and event.ShiftDown():
2897            self.get_paste()
2898        elif event.CmdDown():
2899            self.get_copy()
2900        else:
2901            event.Skip()
2902            return
2903        # make event free
2904        event.Skip()
2905
2906    def get_copy(self):
2907        """
2908        Get copy params to clipboard
2909        """
2910        content = self.get_copy_params()
2911        flag = self.set_clipboard(content)
2912        self._copy_info(flag)
2913        return flag
2914
2915    def get_copy_params(self):
2916        """
2917        Get the string copies of the param names and values in the tap
2918        """
2919        content = 'sasview_parameter_values:'
2920        # Do it if params exist
2921        if self.parameters:
2922
2923            # go through the parameters
2924            strings = self._get_copy_helper(self.parameters,
2925                                           self.orientation_params)
2926            content += strings
2927
2928            # go through the fittables
2929            strings = self._get_copy_helper(self.fittable_param,
2930                                           self.orientation_params_disp)
2931            content += strings
2932
2933            # go through the fixed params
2934            strings = self._get_copy_helper(self.fixed_param,
2935                                           self.orientation_params_disp)
2936            content += strings
2937
2938            # go through the str params
2939            strings = self._get_copy_helper(self.str_parameters,
2940                                           self.orientation_params)
2941            content += strings
2942            return content
2943        else:
2944            return False
2945
2946    def get_copy_excel(self):
2947        """
2948        Get copy params to clipboard
2949        """
2950        content = self.get_copy_params_excel()
2951        flag = self.set_clipboard(content)
2952        self._copy_info(flag)
2953        return flag
2954
2955    def get_copy_params_excel(self):
2956        """
2957        Get the string copies of the param names and values in the tap
2958        """
2959        content = ''
2960
2961        crlf = chr(13) + chr(10)
2962        tab = chr(9)
2963
2964        # Do it if params exist
2965        if self.parameters:
2966
2967            for param in self.parameters:
2968                content += param[1]  # parameter name
2969                content += tab
2970                content += param[1] + "_err"
2971                content += tab
2972
2973            content += crlf
2974
2975            # row of values and errors...
2976            for param in self.parameters:
2977                content += param[2].GetValue()  # value
2978                content += tab
2979                content += param[4].GetValue()  # error
2980                content += tab
2981
2982            return content
2983        else:
2984            return False
2985
2986    def get_copy_latex(self):
2987        """
2988        Get copy params to clipboard
2989        """
2990        content = self.get_copy_params_latex()
2991        flag = self.set_clipboard(content)
2992        self._copy_info(flag)
2993        return flag
2994
2995    def get_copy_params_latex(self):
2996        """
2997        Get the string copies of the param names and values in the tap
2998        """
2999        content = '\\begin{table}'
3000        content += '\\begin{tabular}[h]'
3001
3002        crlf = chr(13) + chr(10)
3003        tab = chr(9)
3004
3005        # Do it if params exist
3006        if self.parameters:
3007
3008            content += '{|'
3009            for param in self.parameters:
3010                content += 'l|l|'
3011            content += '}\hline'
3012            content += crlf
3013
3014            for index, param in enumerate(self.parameters):
3015                content += param[1].replace('_', '\_')  # parameter name
3016                content += ' & '
3017                content += param[1].replace('_', '\_') + "\_err"
3018                if index < len(self.parameters) - 1:
3019                    content += ' & '
3020            content += '\\\\ \\hline'
3021            content += crlf
3022
3023            # row of values and errors...
3024            for index, param in enumerate(self.parameters):
3025                content += param[2].GetValue()  # parameter value
3026                content += ' & '
3027                content += param[4].GetValue()  # parameter error
3028                if index < len(self.parameters) - 1:
3029                    content += ' & '
3030            content += '\\\\ \\hline'
3031            content += crlf
3032
3033            content += '\\end{tabular}'
3034            content += '\\end{table}'
3035            return content
3036        else:
3037            return False
3038
3039    def set_clipboard(self, content=None):
3040        """
3041        Put the string to the clipboard
3042        """
3043        if not content:
3044            return False
3045        if wx.TheClipboard.Open():
3046            wx.TheClipboard.SetData(wx.TextDataObject(str(content)))
3047            wx.TheClipboard.Close()
3048            return True
3049        return None
3050
3051    def _get_copy_helper(self, param, orient_param):
3052        """
3053        Helping get value and name of the params
3054
3055        : param param:  parameters
3056        : param orient_param = oritational params
3057        : return content: strings [list] [name,value:....]
3058        """
3059        content = ''
3060        bound_hi = ''
3061        bound_lo = ''
3062        # go through the str params
3063        for item in param:
3064            # copy only the params shown
3065            if not item[2].IsShown():
3066                continue
3067            disfunc = ''
3068            try:
3069                if item[7].__class__.__name__ == 'ComboBox':
3070                    disfunc = str(item[7].GetValue())
3071            except Exception:
3072                logging.error(traceback.format_exc())
3073
3074            # 2D
3075            if self.data.__class__.__name__ == "Data2D":
3076                try:
3077                    check = item[0].GetValue()
3078                except Exception:
3079                    check = None
3080                name = item[1]
3081                value = item[2].GetValue()
3082            # 1D
3083            else:
3084                # for 1D all parameters except orientation
3085                if not item[1] in orient_param:
3086                    try:
3087                        check = item[0].GetValue()
3088                    except:
3089                        check = None
3090                    name = item[1]
3091                    value = item[2].GetValue()
3092
3093            # Bounds
3094            try:
3095                bound_lo = item[5].GetValue()
3096                bound_hi = item[6].GetValue()
3097            except Exception:
3098                # harmless - need to just pass
3099                pass
3100
3101            # add to the content
3102            if disfunc != '':
3103
3104                disfunc = ',' + disfunc
3105            # Need to support array func for copy/paste
3106            try:
3107                if disfunc.count('array') > 0:
3108                    disfunc += ','
3109                    for val in self.values[name]:
3110                        disfunc += ' ' + str(val)
3111                    disfunc += ','
3112                    for weight in self.weights[name]:
3113                        disfunc += ' ' + str(weight)
3114            except Exception:
3115                logging.error(traceback.format_exc())
3116            content += name + ',' + str(check) + ',' + value + disfunc + ',' + \
3117                       bound_lo + ',' + bound_hi + ':'
3118
3119        return content
3120
3121    def get_clipboard(self):
3122        """
3123        Get strings in the clipboard
3124        """
3125        text = ""
3126        # Get text from the clip board
3127        if wx.TheClipboard.Open():
3128            if wx.TheClipboard.IsSupported(wx.DataFormat(wx.DF_TEXT)):
3129                data = wx.TextDataObject()
3130                # get wx dataobject
3131                success = wx.TheClipboard.GetData(data)
3132                # get text
3133                if success:
3134                    text = data.GetText()
3135                else:
3136                    text = ''
3137            # close clipboard
3138            wx.TheClipboard.Close()
3139        return text
3140
3141    def get_paste(self):
3142        """
3143        Paste params from the clipboard
3144        """
3145        text = self.get_clipboard()
3146        flag = self.get_paste_params(text)
3147        self._copy_info(flag)
3148        return flag
3149
3150    def get_paste_params(self, text=''):
3151        """
3152        Get the string copies of the param names and values in the tap
3153        """
3154        context = {}
3155        # put the text into dictionary
3156        lines = text.split(':')
3157        if lines[0] != 'sasview_parameter_values':
3158            self._copy_info(False)
3159            return False
3160        for line in lines[1:-1]:
3161            if len(line) != 0:
3162                item = line.split(',')
3163                check = item[1]
3164                name = item[0]
3165                value = item[2]
3166                # Transfer the text to content[dictionary]
3167                context[name] = [check, value]
3168
3169                # limits
3170                limit_lo = item[3]
3171                context[name].append(limit_lo)
3172                limit_hi = item[4]
3173                context[name].append(limit_hi)
3174
3175            # ToDo: PlugIn this poly disp function for pasting
3176            try:
3177                poly_func = item[5]
3178                context[name].append(poly_func)
3179                try:
3180                    # take the vals and weights for  array
3181                    array_values = item[6].split(' ')
3182                    array_weights = item[7].split(' ')
3183                    val = [float(a_val) for a_val in array_values[1:]]
3184                    weit = [float(a_weit) for a_weit in array_weights[1:]]
3185
3186                    context[name].append(val)
3187                    context[name].append(weit)
3188                except:
3189                    raise
3190            except:
3191                poly_func = ''
3192                context[name].append(poly_func)
3193
3194        # Do it if params exist
3195        if self.parameters:
3196            # go through the parameters
3197            self._get_paste_helper(self.parameters,
3198                                   self.orientation_params, context)
3199
3200            # go through the fittables
3201            self._get_paste_helper(self.fittable_param,
3202                                   self.orientation_params_disp,
3203                                   context)
3204
3205            # go through the fixed params
3206            self._get_paste_helper(self.fixed_param,
3207                                   self.orientation_params_disp, context)
3208
3209            # go through the str params
3210            self._get_paste_helper(self.str_parameters,
3211                                   self.orientation_params, context)
3212
3213            return True
3214        return None
3215
3216    def _get_paste_helper(self, param, orient_param, content):
3217        """
3218        Helping set values of the params
3219
3220        : param param:  parameters
3221        : param orient_param: oritational params
3222        : param content: dictionary [ name, value: name1.value1,...]
3223        """
3224        # go through the str params
3225        for item in param:
3226            # 2D
3227            if self.data.__class__.__name__ == "Data2D":
3228                name = item[1]
3229                if name in content.keys():
3230                    values = content[name]
3231                    check = values[0]
3232                    pd = values[1]
3233
3234                    if name.count('.') > 0:
3235                        # If this is parameter.width, then pd may be a floating
3236                        # point value or it may be an array distribution.
3237                        # Nothing to do for parameter.npts or parameter.nsigmas.
3238                        try:
3239                            float(pd)
3240                            if name.endswith('.npts'):
3241                                pd = int(pd)
3242                        except Exception:
3243                            # continue
3244                            if not pd and pd != '':
3245                                continue
3246                    item[2].SetValue(str(pd))
3247                    if item in self.fixed_param and pd == '':
3248                        # Only array func has pd == '' case.
3249                        item[2].Enable(False)
3250                    else:
3251                        item[2].Enable(True)
3252                    if item[2].__class__.__name__ == "ComboBox":
3253                        if content[name][1] in self.model.fun_list:
3254                            fun_val = self.model.fun_list[content[name][1]]
3255                            self.model.setParam(name, fun_val)
3256                    try:
3257                        item[5].SetValue(str(values[-3]))
3258                        item[6].SetValue(str(values[-2]))
3259                    except Exception:
3260                        # passing as harmless non-update
3261                        pass
3262
3263                    value = content[name][1:]
3264                    self._paste_poly_help(item, value)
3265                    if check == 'True':
3266                        is_true = True
3267                    elif check == 'False':
3268                        is_true = False
3269                    else:
3270                        is_true = None
3271                    if is_true is not None:
3272                        item[0].SetValue(is_true)
3273            # 1D
3274            else:
3275                # for 1D all parameters except orientation
3276                if not item[1] in orient_param:
3277                    name = item[1]
3278                    if name in content.keys():
3279                        check = content[name][0]
3280                        # Avoid changing combox content
3281                        value = content[name][1:]
3282                        pd = value[0]
3283                        if name.count('.') > 0:
3284                            # If this is parameter.width, then pd may be a
3285                            # floating point value or it may be an array
3286                            # distribution. Nothing to do for parameter.npts or
3287                            # parameter.nsigmas.
3288                            try:
3289                                pd = float(pd)
3290                                if name.endswith('.npts'):
3291                                    pd = int(pd)
3292                            except:
3293                                # continue
3294                                if not pd and pd != '':
3295                                    continue
3296                        item[2].SetValue(str(pd))
3297                        if item in self.fixed_param and pd == '':
3298                            # Only array func has pd == '' case.
3299                            item[2].Enable(False)
3300                        else:
3301                            item[2].Enable(True)
3302                        if item[2].__class__.__name__ == "ComboBox":
3303                            if value[0] in self.model.fun_list:
3304                                fun_val = self.model.fun_list[value[0]]
3305                                self.model.setParam(name, fun_val)
3306                                # save state
3307                        try:
3308                            item[5].SetValue(str(value[-3]))
3309                            item[6].SetValue(str(value[-2]))
3310                        except Exception:
3311                            # passing as harmless non-update
3312                            pass
3313
3314                        self._paste_poly_help(item, value)
3315                        if check == 'True':
3316                            is_true = True
3317                        elif check == 'False':
3318                            is_true = False
3319                        else:
3320                            is_true = None
3321                        if is_true is not None:
3322                            item[0].SetValue(is_true)
3323
3324        self.select_param(event=None)
3325        self.Refresh()
3326
3327    def _paste_poly_help(self, item, value):
3328        """
3329        Helps get paste for poly function
3330
3331        *item* is the parameter name
3332
3333        *value* depends on which parameter is being processed, and whether it
3334        has array polydispersity.
3335
3336        For parameters without array polydispersity:
3337
3338            parameter => ['FLOAT', '']
3339            parameter.width => ['FLOAT', 'DISTRIBUTION', '']
3340            parameter.npts => ['FLOAT', '']
3341            parameter.nsigmas => ['FLOAT', '']
3342
3343        For parameters with array polydispersity:
3344
3345            parameter => ['FLOAT', '']
3346            parameter.width => ['FILENAME', 'array', [x1, ...], [w1, ...]]
3347            parameter.npts => ['FLOAT', '']
3348            parameter.nsigmas => ['FLOAT', '']
3349        """
3350        # Do nothing if not setting polydispersity
3351        if len(value[3]) == 0:
3352            return
3353
3354        try:
3355            name = item[7].Name
3356            param_name = name.split('.')[0]
3357            item[7].SetValue(value[1])
3358            selection = item[7].GetCurrentSelection()
3359            dispersity = item[7].GetClientData(selection)
3360            disp_model = dispersity()
3361
3362            if value[1] == 'array':
3363                pd_vals = numpy.array(value[2])
3364                pd_weights = numpy.array(value[3])
3365                if len(pd_vals) == 0 or len(pd_vals) != len(pd_weights):
3366                    msg = ("bad array distribution parameters for %s"
3367                           % param_name)
3368                    raise ValueError(msg)
3369                self._set_disp_cb(True, item=item)
3370                self._set_array_disp_model(name=name,
3371                                           disp=disp_model,
3372                                           values=pd_vals,
3373                                           weights=pd_weights)
3374            else:
3375                self._set_disp_cb(False, item=item)
3376                self._disp_obj_dict[name] = disp_model
3377                self.model.set_dispersion(param_name, disp_model)
3378                self.state._disp_obj_dict[name] = disp_model.type
3379                # TODO: It's not an array, why update values and weights?
3380                self.model._persistency_dict[param_name] = \
3381                    [self.values, self.weights]
3382                self.state.values = self.values
3383                self.state.weights = self.weights
3384
3385        except Exception:
3386            logging.error(traceback.format_exc())
3387            print "Error in BasePage._paste_poly_help: %s" % \
3388                  sys.exc_info()[1]
3389
3390    def _set_disp_cb(self, isarray, item):
3391        """
3392        Set cb for array disp
3393        """
3394        if isarray:
3395            item[0].SetValue(False)
3396            item[0].Enable(False)
3397            item[2].Enable(False)
3398            item[3].Show(False)
3399            item[4].Show(False)
3400            item[5].SetValue('')
3401            item[5].Enable(False)
3402            item[6].SetValue('')
3403            item[6].Enable(False)
3404        else:
3405            item[0].Enable()
3406            item[2].Enable()
3407            item[3].Show(True)
3408            item[4].Show(True)
3409            item[5].Enable()
3410            item[6].Enable()
3411
3412    def update_pinhole_smear(self):
3413        """
3414            Method to be called by sub-classes
3415            Moveit; This method doesn't belong here
3416        """
3417        print "BasicPage.update_pinhole_smear was called: skipping"
3418        return
3419
3420    def _read_category_info(self):
3421        """
3422        Reads the categories in from file
3423        """
3424        # # ILL mod starts here - July 2012 kieranrcampbell@gmail.com
3425        self.master_category_dict = defaultdict(list)
3426        self.by_model_dict = defaultdict(list)
3427        self.model_enabled_dict = defaultdict(bool)
3428        categorization_file = CategoryInstaller.get_user_file()
3429        with open(categorization_file, 'rb') as f:
3430            self.master_category_dict = json.load(f)
3431        self._regenerate_model_dict()
3432
3433    def _regenerate_model_dict(self):
3434        """
3435        regenerates self.by_model_dict which has each model name as the
3436        key and the list of categories belonging to that model
3437        along with the enabled mapping
3438        """
3439        self.by_model_dict = defaultdict(list)
3440        for category in self.master_category_dict:
3441            for (model, enabled) in self.master_category_dict[category]:
3442                self.by_model_dict[model].append(category)
3443                self.model_enabled_dict[model] = enabled
3444
3445    def _populate_listbox(self):
3446        """
3447        fills out the category list box
3448        """
3449        uncat_str = 'Customized Models'
3450        self._read_category_info()
3451
3452        self.categorybox.Clear()
3453        cat_list = sorted(self.master_category_dict.keys())
3454        if uncat_str not in cat_list:
3455            cat_list.append(uncat_str)
3456
3457        for category in cat_list:
3458            if category != '':
3459                self.categorybox.Append(category)
3460
3461        if self.categorybox.GetSelection() == wx.NOT_FOUND:
3462            self.categorybox.SetSelection(0)
3463        else:
3464            self.categorybox.SetSelection(
3465                self.categorybox.GetSelection())
3466        # self._on_change_cat(None)
3467
3468    def _on_change_cat(self, event):
3469        """
3470        Callback for category change action
3471        """
3472        self.model_name = None
3473        category = self.categorybox.GetStringSelection()
3474        if category is None:
3475            return
3476        self.model_box.Clear()
3477
3478        if category == 'Customized Models':
3479            for model in self.model_list_box[category]:
3480                str_m = str(model).split(".")[0]
3481                self.model_box.Append(str_m)
3482
3483        else:
3484            for (model, enabled) in sorted(self.master_category_dict[category],
3485                                           key=lambda name: name[0]):
3486                if(enabled):
3487                    self.model_box.Append(model)
3488
3489    def _fill_model_sizer(self, sizer):
3490        """
3491        fill sizer containing model info
3492        """
3493        # This should only be called once per fit tab
3494        # print "==== Entering _fill_model_sizer"
3495        # Add model function Details button in fitpanel.
3496        # The following 3 lines are for Mac. Let JHC know before modifying...
3497        title = "Model"
3498        self.formfactorbox = None
3499        self.multifactorbox = None
3500        self.mbox_description = wx.StaticBox(self, wx.ID_ANY, str(title))
3501        boxsizer1 = wx.StaticBoxSizer(self.mbox_description, wx.VERTICAL)
3502        sizer_cat = wx.BoxSizer(wx.HORIZONTAL)
3503        self.mbox_description.SetForegroundColour(wx.RED)
3504        wx_id = self._ids.next()
3505        self.model_func = wx.Button(self, wx_id, 'Help', size=(80, 23))
3506        self.model_func.Bind(wx.EVT_BUTTON, self.on_function_help_clicked,
3507                             id=wx_id)
3508        self.model_func.SetToolTipString("Full Model Function Help")
3509        wx_id = self._ids.next()
3510        self.model_help = wx.Button(self, wx_id, 'Description', size=(80, 23))
3511        self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,
3512                             id=wx_id)
3513        self.model_help.SetToolTipString("Short Model Function Description")
3514        wx_id = self._ids.next()
3515        self.model_view = wx.Button(self, wx_id, "Show 2D", size=(80, 23))
3516        self.model_view.Bind(wx.EVT_BUTTON, self._onModel2D, id=wx_id)
3517        hint = "toggle view of model from 1D to 2D  or 2D to 1D"
3518        self.model_view.SetToolTipString(hint)
3519
3520        cat_set_box = wx.StaticBox(self, wx.ID_ANY, 'Category')
3521        sizer_cat_box = wx.StaticBoxSizer(cat_set_box, wx.HORIZONTAL)
3522        sizer_cat_box.SetMinSize((200, 50))
3523        self.categorybox = wx.ComboBox(self, wx.ID_ANY,
3524                                       style=wx.CB_READONLY)
3525        self.categorybox.SetToolTip(wx.ToolTip("Select a Category/Type"))
3526        self._populate_listbox()
3527        wx.EVT_COMBOBOX(self.categorybox, wx.ID_ANY, self._show_combox)
3528        # self.shape_rbutton = wx.RadioButton(self, wx.ID_ANY, 'Shapes',
3529        #                                     style=wx.RB_GROUP)
3530        # self.shape_indep_rbutton = wx.RadioButton(self, wx.ID_ANY,
3531        #                                          "Shape-Independent")
3532        # self.struct_rbutton = wx.RadioButton(self, wx.ID_ANY,
3533        #                                     "Structure Factor ")
3534        # self.plugin_rbutton = wx.RadioButton(self, wx.ID_ANY,
3535        #                                     "Uncategorized")
3536
3537        # self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
3538        #                   id=self.shape_rbutton.GetId())
3539        # self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
3540        #                    id=self.shape_indep_rbutton.GetId())
3541        # self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
3542        #                    id=self.struct_rbutton.GetId())
3543        # self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
3544        #                    id=self.plugin_rbutton.GetId())
3545        # MAC needs SetValue
3546
3547        show_cat_button = wx.Button(self, wx.ID_ANY, "Modify")
3548        cat_tip = "Modify model categories \n"
3549        cat_tip += "(also accessible from the menu bar)."
3550        show_cat_button.SetToolTip(wx.ToolTip(cat_tip))
3551        show_cat_button.Bind(wx.EVT_BUTTON, self._on_modify_cat)
3552        sizer_cat_box.Add(self.categorybox, 1, wx.RIGHT, 3)
3553        sizer_cat_box.Add((10, 10))
3554        sizer_cat_box.Add(show_cat_button)
3555        # self.shape_rbutton.SetValue(True)
3556
3557        sizer_radiobutton = wx.GridSizer(2, 2, 5, 5)
3558        # sizer_radiobutton.Add(self.shape_rbutton)
3559        # sizer_radiobutton.Add(self.shape_indep_rbutton)
3560        sizer_radiobutton.Add((5, 5))
3561        sizer_radiobutton.Add(self.model_view, 1, wx.RIGHT, 5)
3562        # sizer_radiobutton.Add(self.plugin_rbutton)
3563        # sizer_radiobutton.Add(self.struct_rbutton)
3564        # sizer_radiobutton.Add((5,5))
3565        sizer_radiobutton.Add(self.model_help, 1, wx.RIGHT | wx.LEFT, 5)
3566        # sizer_radiobutton.Add((5,5))
3567        sizer_radiobutton.Add(self.model_func, 1, wx.RIGHT, 5)
3568        sizer_cat.Add(sizer_cat_box, 1, wx.LEFT, 2.5)
3569        sizer_cat.Add(sizer_radiobutton)
3570        sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
3571        mutifactor_selection = wx.BoxSizer(wx.HORIZONTAL)
3572
3573        self.text1 = wx.StaticText(self, wx.ID_ANY, "")
3574        self.text2 = wx.StaticText(self, wx.ID_ANY, "P(Q)*S(Q)")
3575        self.mutifactor_text = wx.StaticText(self, wx.ID_ANY, "No. of Shells: ")
3576        self.mutifactor_text1 = wx.StaticText(self, wx.ID_ANY, "")
3577        self.show_sld_button = wx.Button(self, wx.ID_ANY, "Show SLD Profile")
3578        self.show_sld_button.Bind(wx.EVT_BUTTON, self._on_show_sld)
3579
3580        self.formfactorbox = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
3581        self.formfactorbox.SetToolTip(wx.ToolTip("Select a Model"))
3582        if self.model is not None:
3583            self.formfactorbox.SetValue(self.model.name)
3584        self.structurebox = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
3585        self.multifactorbox = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
3586        self.initialize_combox()
3587        wx.EVT_COMBOBOX(self.formfactorbox, wx.ID_ANY, self._on_select_model)
3588
3589        wx.EVT_COMBOBOX(self.structurebox, wx.ID_ANY, self._on_select_model)
3590        wx.EVT_COMBOBOX(self.multifactorbox, wx.ID_ANY, self._on_select_model)
3591        # check model type to show sizer
3592        if self.model is not None:
3593            print "_set_model_sizer_selection: disabled."
3594            # self._set_model_sizer_selection(self.model)
3595
3596        sizer_selection.Add(self.text1)
3597        sizer_selection.Add((10, 5))
3598        sizer_selection.Add(self.formfactorbox)
3599        sizer_selection.Add((5, 5))
3600        sizer_selection.Add(self.text2)
3601        sizer_selection.Add((5, 5))
3602        sizer_selection.Add(self.structurebox)
3603
3604        mutifactor_selection.Add((13, 5))
3605        mutifactor_selection.Add(self.mutifactor_text)
3606        mutifactor_selection.Add(self.multifactorbox)
3607        mutifactor_selection.Add((5, 5))
3608        mutifactor_selection.Add(self.mutifactor_text1)
3609        mutifactor_selection.Add((10, 5))
3610        mutifactor_selection.Add(self.show_sld_button)
3611
3612        boxsizer1.Add(sizer_cat)
3613        boxsizer1.Add((10, 10))
3614        boxsizer1.Add(sizer_selection)
3615        boxsizer1.Add((10, 10))
3616        boxsizer1.Add(mutifactor_selection)
3617
3618        self._set_multfactor_combobox()
3619        self.multifactorbox.SetSelection(1)
3620        self.show_sld_button.Hide()
3621        sizer.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
3622        sizer.Layout()
3623
3624    def on_smear_helper(self, update=False):
3625        """
3626        Help for onSmear if implemented
3627
3628        :param update: force or not to update
3629        """
3630    def reset_page(self, state, first=False):
3631        """
3632        reset the state  if implemented
3633        """
3634    def onSmear(self, event):
3635        """
3636        Create a smear object if implemented
3637        """
3638    def onPinholeSmear(self, event):
3639        """
3640        Create a custom pinhole smear object if implemented
3641        """
3642    def onSlitSmear(self, event):
3643        """
3644        Create a custom slit smear object if implemented
3645        """
3646    def update_slit_smear(self):
3647        """
3648        called by kill_focus on pinhole TextCntrl
3649        to update the changes if implemented
3650        """
3651    def select_param(self, event):
3652        """
3653        Select TextCtrl  checked if implemented
3654        """
3655    def set_data(self, data=None):
3656        """
3657        Sets data if implemented
3658        """
3659    def _is_2D(self):
3660        """
3661        Check if data_name is Data2D if implemented
3662        """
3663    def _on_select_model(self, event=None):
3664        """
3665        call back for model selection if implemented
3666        """
3667    def get_weight_flag(self):
3668        """
3669        Get flag corresponding to a given weighting dI data if implemented
3670        """
3671    def _set_sizer_dispersion(self):
3672        """
3673        draw sizer for dispersity if implemented
3674        """
3675    def get_all_checked_params(self):
3676        """
3677        Found all parameters current check and add them to list of parameters
3678        to fit if implemented
3679        """
3680    def show_npts2fit(self):
3681        """
3682        setValue Npts for fitting if implemented
3683        """
3684    def _onModel2D(self, event):
3685        """
3686        toggle view of model from 1D to 2D  or 2D from 1D if implemented
3687        """
3688
3689
3690class ModelTextCtrl(wx.TextCtrl):
3691    """
3692    Text control for model and fit parameters.
3693    Binds the appropriate events for user interactions.
3694    Default callback methods can be overwritten on initialization
3695
3696    :param kill_focus_callback: callback method for EVT_KILL_FOCUS event
3697    :param set_focus_callback:  callback method for EVT_SET_FOCUS event
3698    :param mouse_up_callback:   callback method for EVT_LEFT_UP event
3699    :param text_enter_callback: callback method for EVT_TEXT_ENTER event
3700
3701    """
3702    # Set to True when the mouse is clicked while whole string is selected
3703    full_selection = False
3704    # Call back for EVT_SET_FOCUS events
3705    _on_set_focus_callback = None
3706
3707    def __init__(self, parent, id=-1,
3708                 value=wx.EmptyString,
3709                 pos=wx.DefaultPosition,
3710                 size=wx.DefaultSize,
3711                 style=0,
3712                 validator=wx.DefaultValidator,
3713                 name=wx.TextCtrlNameStr,
3714                 kill_focus_callback=None,
3715                 set_focus_callback=None,
3716                 mouse_up_callback=None,
3717                 text_enter_callback=None):
3718
3719        wx.TextCtrl.__init__(self, parent, id, value, pos,
3720                             size, style, validator, name)
3721
3722        # Bind appropriate events
3723        self._on_set_focus_callback = parent.onSetFocus \
3724            if set_focus_callback is None else set_focus_callback
3725        self.Bind(wx.EVT_SET_FOCUS, self._on_set_focus)
3726        self.Bind(wx.EVT_KILL_FOCUS, self._silent_kill_focus
3727        if kill_focus_callback is None else kill_focus_callback)
3728        self.Bind(wx.EVT_TEXT_ENTER, parent._onparamEnter
3729        if text_enter_callback is None else text_enter_callback)
3730        if not ON_MAC:
3731            self.Bind(wx.EVT_LEFT_UP, self._highlight_text
3732            if mouse_up_callback is None else mouse_up_callback)
3733
3734    def _on_set_focus(self, event):
3735        """
3736        Catch when the text control is set in focus to highlight the whole
3737        text if necessary
3738
3739        :param event: mouse event
3740
3741        """
3742        event.Skip()
3743        self.full_selection = True
3744        return self._on_set_focus_callback(event)
3745
3746    def _highlight_text(self, event):
3747        """
3748        Highlight text of a TextCtrl only of no text has be selected
3749
3750        :param event: mouse event
3751
3752        """
3753        # Make sure the mouse event is available to other listeners
3754        event.Skip()
3755        control = event.GetEventObject()
3756        if self.full_selection:
3757            self.full_selection = False
3758            # Check that we have a TextCtrl
3759            if issubclass(control.__class__, wx.TextCtrl):
3760                # Check whether text has been selected,
3761                # if not, select the whole string
3762                (start, end) = control.GetSelection()
3763                if start == end:
3764                    control.SetSelection(-1, -1)
3765
3766    def _silent_kill_focus(self, event):
3767        """
3768        Save the state of the page
3769        """
3770
3771        event.Skip()
3772        # pass
Note: See TracBrowser for help on using the repository browser.