source: sasview/src/sas/perspectives/fitting/basepage.py @ 8d302cd

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.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 8d302cd was 7cd87c2, checked in by Doucet, Mathieu <doucetm@…>, 10 years ago

pylint fixes and remove except followed by pass

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