source: sasview/src/sas/sasgui/perspectives/fitting/basepage.py @ 8898558b

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 8898558b was 8898558b, checked in by krzywon, 7 years ago

#795: Added a check to see if model names are from sasmodels in SasView? v4+ or from v3.x.y. Conversion tool is skipped if from v4 or later.

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