source: sasview/sansview/perspectives/fitting/fitting.py @ e0e2c44f

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 e0e2c44f was 2296316, checked in by Jae Cho <jhjcho@…>, 14 years ago

moving features from the branch

  • Property mode set to 100644
File size: 62.7 KB
RevLine 
[5612152]1
2
[5062bbf]3################################################################################
4#This software was developed by the University of Tennessee as part of the
5#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
6#project funded by the US National Science Foundation.
7#
8#See the license text in license.txt
9#
10#copyright 2009, University of Tennessee
11################################################################################
12
13
[a0da535]14import re
[5536035]15import sys
16import wx
17import logging
18import numpy
19import string
[c3b4dcb]20import time
[edcbd467]21from copy import deepcopy
[f83b94a]22import models
23import fitpage
24
[66ff250]25
[7f69d4d]26from DataLoader.loader import Loader
[c3b4dcb]27from sans.guiframe.dataFitting import Data2D
28from sans.guiframe.dataFitting import Data1D
[6bbeacd4]29from sans.guiframe.events import NewPlotEvent
30from sans.guiframe.events import StatusEvent 
[a0da535]31from sans.guiframe.events import EVT_SLICER_PANEL
32from sans.guiframe.events import EVT_REMOVE_DATA
33from sans.guiframe.events import EVT_SLICER_PARS_UPDATE
[6bbeacd4]34from sans.guiframe.gui_style import GUIFRAME_ID
35from sans.guiframe.plugin_base import PluginBase
[6ed82c7]36
[a0da535]37from .console import ConsoleUpdate
38from .fitproblem import FitProblem
39from .fitpanel import FitPanel
40from .fit_thread import FitThread
41from .pagestate import Reader
42from .fitpage import Chi2UpdateEvent
[ed2ea6a]43
[d250f7d]44DEFAULT_BEAM = 0.005
[6ced1cc]45DEFAULT_QMIN = 0.001
[9be7432]46DEFAULT_QMAX = 0.13
[7b758fd]47DEFAULT_NPTS = 50
[c26a64b]48MAX_NBR_DATA = 4
[77e23a2]49
[5062bbf]50
[6bbeacd4]51(PageInfoEvent, EVT_PAGE_INFO)   = wx.lib.newevent.NewEvent()
[5062bbf]52
[6bbeacd4]53   
[3658abed]54
55class Plugin(PluginBase):
[d89f09b]56    """
[5062bbf]57    Fitting plugin is used to perform fit
[d89f09b]58    """
[3658abed]59    def __init__(self, standalone=False):
60        PluginBase.__init__(self, name="Fitting", standalone=standalone)
[d89f09b]61       
[ed2ea6a]62        #Provide list of models existing in the application
[d89f09b]63        self.menu_mng = models.ModelManager()
[3658abed]64       
[ed2ea6a]65        #list of panel to send to guiframe
[f83b94a]66        self.mypanels = []
[ed2ea6a]67        # reference to the current running thread
[f83b94a]68        self.calc_2D = None
69        self.calc_1D = None
[66ff250]70        self.fit_thread_list = {}
[2296316]71        self.residuals = None
[9466f2d6]72        self.fit_panel = None
[d89f09b]73        # Start with a good default
74        self.elapsed = 0.022
[ed2ea6a]75        # the type of optimizer selected, park or scipy
[d89f09b]76        self.fitter  = None
[2296316]77        self.fit_panel = None
[c660493]78        #let fit ready
79        self.fitproblem_count = None
[ed2ea6a]80        #Flag to let the plug-in know that it is running stand alone
[f83b94a]81        self.standalone = True
[6f023e8]82        ## dictionary of page closed and id
[f83b94a]83        self.closed_page_dict = {}
[d89f09b]84        ## Fit engine
[e9e914f]85        self._fit_engine = 'scipy'
[2296316]86        ## Relative error desired in the sum of squares (float); scipy only
87        self.ftol=1.49012e-08
[ed2ea6a]88        #List of selected data
[f83b94a]89        self.selected_data_list = []
[c3435b7f]90        ## list of slicer panel created to display slicer parameters and results
[f83b94a]91        self.slicer_panels = []
[32d802f]92        # model 2D view
[f83b94a]93        self.model2D_id = None
[ed2ea6a]94        #keep reference of the simultaneous fit page
[f83b94a]95        self.sim_page = None
[ae4ade7]96        self.index_model = 0
[f83b94a]97        #Create a reader for fit page's state
[3bf4a032]98        self.state_reader = None 
[c8deee5]99        self._extensions = '.fitv'
[4da35bc]100        self.temp_state = []
[ef16f59]101        self.state_index = 0
[b63dc6e]102        self.sfile_ext = None
[6bbeacd4]103        # take care of saving  data, model and page associated with each other
104        self.page_finder = {}
[f83b94a]105        # Log startup
106        logging.info("Fitting plug-in started") 
[2a8fac1]107       
[cab076b]108    def populate_menu(self, owner):
[d89f09b]109        """
[5062bbf]110        Create a menu for the Fitting plug-in
111       
112        :param id: id to create a menu
113        :param owner: owner of menu
114       
115        :return: list of information to populate the main menu
116       
[d89f09b]117        """
118        #Menu for fitting
119        self.menu1 = wx.Menu()
[75fbd17]120       
[b5c537f]121        #Set park engine
[3b19ac9]122        id3 = wx.NewId()
[bb18ef1]123        scipy_help= "Scipy Engine: Perform Simple fit. More in Help window...."
[89ef796]124        self.menu1.AppendCheckItem(id3, "Simple Fit  [Scipy]",scipy_help) 
[bb18ef1]125        wx.EVT_MENU(owner, id3,  self._onset_engine_scipy)
[6f023e8]126       
[bb18ef1]127        id3 = wx.NewId()
128        park_help = "Park Engine: Perform Complex fit. More in Help window...."
[89ef796]129        self.menu1.AppendCheckItem(id3, "Complex Fit  [Park]",park_help) 
[bb18ef1]130        wx.EVT_MENU(owner, id3,  self._onset_engine_park)
[707436d]131       
132        self.menu1.FindItemByPosition(0).Check(True)
133        self.menu1.FindItemByPosition(1).Check(False)
134           
[6f023e8]135        self.menu1.AppendSeparator()
136        id1 = wx.NewId()
[6bbeacd4]137        simul_help = "Simultaneous Fit"
[2296316]138        self.menu1.Append(id1, '&Simultaneous Fit',simul_help)
[6f023e8]139        wx.EVT_MENU(owner, id1, self.on_add_sim_page)
[6bbeacd4]140       
141        id1 = wx.NewId()
142        simul_help = "Add new fit page"
143        self.menu1.Append(id1, '&Create New Page',simul_help)
144        wx.EVT_MENU(owner, id1, self.on_add_new_page)
145       
146        #self.fit_panel.set_model_list(self.menu_mng.get_model_list())
[0b12abb5]147   
[3b19ac9]148        #create  menubar items
[2296316]149        return [(self.menu1, "FitEngine")]
[2db1d66]150               
[51d47b5]151    def on_add_sim_page(self, event):
[ed2ea6a]152        """
[5062bbf]153        Create a page to access simultaneous fit option
[ed2ea6a]154        """
[a0da535]155        if self.sim_page != None:
[2140e68]156            msg= "Simultaneous Fit page already opened"
157            wx.PostEvent(self.parent, StatusEvent(status= msg))
158            return 
159       
[51d47b5]160        self.sim_page= self.fit_panel.add_sim_page()
[b28717b]161       
[d89f09b]162    def help(self, evt):
163        """
[5062bbf]164        Show a general help dialog.
[d89f09b]165        """
[484faf7]166        from help_panel import  HelpWindow
[2268d10]167        frame = HelpWindow(None, -1, 'HelpWindow')   
168        frame.Show(True)
169       
[6bbeacd4]170    def get_context_menu(self, plotpanel=None):
[d89f09b]171        """
[5062bbf]172        Get the context menu items available for P(r).them allow fitting option
173        for Data2D and Data1D only.
174       
175        :param graph: the Graph object to which we attach the context menu
176       
177        :return: a list of menu items with call-back function
178       
179        :note: if Data1D was generated from Theory1D 
180                the fitting option is not allowed
181               
[d89f09b]182        """
[6bbeacd4]183        graph = plotpanel.graph
[484faf7]184        fit_option = "Select data for fitting"
185        fit_hint =  "Dialog with fitting parameters "
[6bbeacd4]186       
187        if graph.selected_plottable not in plotpanel.plots:
188            return []
189        item = plotpanel.plots[graph.selected_plottable]
190        if item.__class__.__name__ is "Data2D": 
191            if hasattr(item,"is_data"):
192                if item.is_data:
193                    return [[fit_option, fit_hint, self._onSelect]]
194                else:
195                    return [] 
196            return [[fit_option, fit_hint, self._onSelect]]
197        else:
198           
199            # if is_data is true , this in an actual data loaded
200            #else it is a data created from a theory model
201            if hasattr(item,"is_data"):
202                if item.is_data:
203                    return [[fit_option, fit_hint,
204                              self._onSelect]]
205                else:
206                    return [] 
[d89f09b]207        return []   
208
209
210    def get_panels(self, parent):
211        """
[5062bbf]212        Create and return a list of panel objects
[d89f09b]213        """
214        self.parent = parent
[75fbd17]215        #self.parent.Bind(EVT_FITSTATE_UPDATE, self.on_set_state_helper)
[d89f09b]216        # Creation of the fit panel
[6bbeacd4]217        self.fit_panel = FitPanel(parent=self.parent, manager=self)
218        self.on_add_new_page(event=None)
[f93dfcb]219        #Set the manager for the main panel
[d89f09b]220        self.fit_panel.set_manager(self)
221        # List of windows used for the perspective
222        self.perspective = []
223        self.perspective.append(self.fit_panel.window_name)
[6bbeacd4]224       
[3b19ac9]225        #index number to create random model name
226        self.index_model = 0
[264df67]227        self.index_theory= 0
[32673ac]228        self.parent.Bind(EVT_SLICER_PANEL, self._on_slicer_event)
[6bbeacd4]229       
[848a2ef]230        self.parent.Bind(EVT_REMOVE_DATA, self._closed_fitpage)
231        self.parent.Bind(EVT_SLICER_PARS_UPDATE, self._onEVT_SLICER_PANEL)
[df4c3ad]232        self.parent._mgr.Bind(wx.aui.EVT_AUI_PANE_CLOSE,self._onclearslicer)   
[3bf4a032]233        #Create reader when fitting panel are created
234        self.state_reader = Reader(self.set_state)   
235        #append that reader to list of available reader
236        loader = Loader()
237        loader.associate_file_reader(".fitv", self.state_reader)
[b35d3d1]238        loader.associate_file_reader(".svs", self.state_reader)
[75fbd17]239        from sans.perspectives.calculator.sld_panel import SldPanel
[f93dfcb]240        #Send the fitting panel to guiframe
[0912feab]241        self.mypanels.append(self.fit_panel) 
[75fbd17]242        self.mypanels.append(SldPanel(parent=self.parent, base=self.parent))
[568e1a5]243        return self.mypanels
[232c3db]244   
[90a7bbd]245    def clear_panel(self):
246        """
247        """
[81f00d7]248        self.fit_panel.clear_panel()
[90a7bbd]249       
[1610976]250    def set_default_perspective(self):
251        """
[5062bbf]252        Call back method that True to notify the parent that the current plug-in
253        can be set as default  perspective.
254        when returning False, the plug-in is not candidate for an automatic
255        default perspective setting
[1610976]256        """
257        return True
[a0da535]258   
[17553ae]259    def delete_data(self, data):
[8ee56a9]260        """
261        delete  the given data from panel
262        """
263        self.fit_panel.delete_data(data)
264       
[e88ebfd]265    def set_data(self, data_list=None):
[a0da535]266        """
267        receive a list of data to fit
268        """
[b2d9826]269        if data_list is None:
270            data_list = []
[c26a64b]271        selected_data_list = []
272        if len(data_list) > MAX_NBR_DATA :
273            from fitting_widgets import DataDialog
274            dlg = DataDialog(data_list=data_list, nb_data=MAX_NBR_DATA)
275            if dlg.ShowModal() == wx.ID_OK:
276                selected_data_list = dlg.get_data()
277        else:
278            selected_data_list = data_list
[e88ebfd]279        try:
280            for data in selected_data_list:
281                self.add_fit_page(data=data)
282                wx.PostEvent(self.parent, NewPlotEvent(plot=data, 
283                                                       title=str(data.title)))
284        except:
[66ff250]285            raise
286            #msg = "Fitting Set_data: " + str(sys.exc_value)
287            #wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
[e88ebfd]288           
289    def set_theory(self,  theory_list=None):
290        """
291        """
292        #set the model state for a given theory_state:
293        for item in theory_list:
294            try:
295                _, theory_state = item
296                self.fit_panel.set_model_state(theory_state)
297            except:
298                msg = "Fitting: cannot deal with the theory received"
299                logging.error("set_theory " + msg + "\n" + str(sys.exc_value))
300                wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
[a0da535]301           
[b63dc6e]302    def set_state(self, state=None, datainfo=None, format=None):
[f83b94a]303        """
[5062bbf]304        Call-back method for the fit page state reader.
[8897d66]305        This method is called when a .fitv/.svs file is loaded.
[5062bbf]306       
[8897d66]307        : param state: PageState object
308        : param datainfo: data
[f83b94a]309        """
[90a7bbd]310        #state = self.state_reader.get_state()
[b63dc6e]311        if state != None:
312            # store fitting state in temp_state
313            self.temp_state.append(state) 
314        else:
315            self.temp_state = []
[ef16f59]316        # index to start with for a new set_state
317        self.state_index = 0
[b63dc6e]318        # state file format
319        self.sfile_ext = format
[75fbd17]320       
321        self.on_set_state_helper(event=None)
[ef16f59]322
[75fbd17]323    def  on_set_state_helper(self,event=None):
[4da35bc]324        """
[8897d66]325        Set_state_helper. This actually sets state after plotting data from state file.
[ef16f59]326       
[9b18735]327        : event: FitStateUpdateEvent called by dataloader.plot_data from guiframe
[4da35bc]328        """
[7a07864]329        if len(self.temp_state) == 0:
330            if self.state_index==0 and len(self.mypanels) <= 0 and self.sfile_ext =='.svs':
[9b18735]331                self.fit_panel.add_default_pages()
[b63dc6e]332                self.temp_state = []
333                self.state_index = 0
[4da35bc]334            return
[3eb2811]335       
[ef16f59]336        try:
337            # Load fitting state
338            state = self.temp_state[self.state_index]
[3eb2811]339            #panel state should have model selection to set_state
340            if state.formfactorcombobox != None:
341                #set state
[75fbd17]342                data = self.parent.create_gui_data(state.data)
[6bbeacd4]343               
[75fbd17]344                data.group_id = state.data.group_id
[f95301b]345                self.parent.add_data(data_list={data.id:data})
[75fbd17]346                wx.PostEvent(self.parent, NewPlotEvent(plot=data,
347                                        title=data.title))
[f95301b]348                #need to be fix later make sure we are sendind guiframe.data
349                #to panel
350                state.data = data
[3eb2811]351                page = self.fit_panel.set_state(state)   
352            else:
353                #just set data because set_state won't work
[75fbd17]354                data = self.parent.create_gui_data(state.data)
355                data.group_id = state.data.group_id
[f95301b]356                self.parent.add_data(data_list={data.id:data})
357                wx.PostEvent(self.parent, NewPlotEvent(plot=data,
358                                        title=data.title))
[6bbeacd4]359                self.add_fit_page(data)
360                caption = panel.window_name
361                self.store_data(page=panel.id, data=data, caption=caption)
[b63dc6e]362                self.mypanels.append(panel) 
[3eb2811]363               
[9b18735]364            # get ready for the next set_state
[ef16f59]365            self.state_index += 1
[9b18735]366
[ef16f59]367            #reset state variables to default when all set_state is finished.
368            if len(self.temp_state) == self.state_index:
[9b18735]369               
[ef16f59]370                self.temp_state = []
[b63dc6e]371                #self.state_index = 0
[ef16f59]372                # Make sure the user sees the fitting panel after loading
[75fbd17]373                #self.parent.set_perspective(self.perspective)
374                self.on_perspective(event=None)
[ef16f59]375        except:
[b63dc6e]376            self.state_index==0
[8897d66]377            self.temp_state = []
[ef16f59]378            raise
[4da35bc]379                 
[f83b94a]380    def save_fit_state(self, filepath, fitstate): 
381        """
[5062bbf]382        save fit page state into file
[f83b94a]383        """
384        self.state_reader.write(filename=filepath, fitstate=fitstate)
385       
[66ff250]386    def set_fit_range(self, uid, qmin, qmax):
[ca7a626]387        """
[5062bbf]388        Set the fitting range of a given page
[ca7a626]389        """
[66ff250]390        self.page_finder[uid].set_range(qmin=qmin, qmax=qmax)
[2a8fac1]391                   
[66ff250]392    def schedule_for_fit(self,value=0, uid=None,fitproblem =None): 
[948add7]393        """
[5062bbf]394        Set the fit problem field to 0 or 1 to schedule that problem to fit.
395        Schedule the specified fitproblem or get the fit problem related to
396        the current page and set value.
397       
398        :param value: integer 0 or 1
399        :param fitproblem: fitproblem to schedule or not to fit
400       
[948add7]401        """   
402        if fitproblem !=None:
403            fitproblem.schedule_tofit(value)
404        else:
[66ff250]405            self.page_finder[uid].schedule_tofit(value)
[ca7a626]406         
[d89f09b]407    def get_page_finder(self):
[5062bbf]408        """
409        return self.page_finder used also by simfitpage.py
410        """ 
[d89f09b]411        return self.page_finder
412   
[3b19ac9]413    def set_page_finder(self,modelname,names,values):
[d89f09b]414        """
[5062bbf]415        Used by simfitpage.py to reset a parameter given the string constrainst.
416         
417        :param modelname: the name ot the model for with the parameter has to reset
418        :param value: can be a string in this case.
419        :param names: the paramter name
420         
421        :note: expecting park used for fit.
422         
[d89f09b]423        """ 
[66ff250]424        sim_page_id = self.sim_page.uid
425        for uid, value in self.page_finder.iteritems():
426            if uid != sim_page_id:
[6bbeacd4]427                list = value.get_model()
[f93dfcb]428                model = list[0]
[6bbeacd4]429                if model.name == modelname:
430                    value.set_model_param(names, values)
[d89f09b]431                    break
[2db1d66]432         
[d89f09b]433    def split_string(self,item): 
434        """
[5062bbf]435        receive a word containing dot and split it. used to split parameterset
436        name into model name and parameter name example: ::
437       
[3b19ac9]438            paramaterset (item) = M1.A
[5062bbf]439            Will return model_name = M1 , parameter name = A
440           
[d89f09b]441        """
442        if string.find(item,".")!=-1:
443            param_names= re.split("\.",item)
[6d91073]444            model_name=param_names[0]           
445            ##Assume max len is 3; eg., M0.radius.width
446            if len(param_names) == 3:
447                param_name=param_names[1]+"."+param_names[2]
448            else:
449                param_name=param_names[1]                   
[d89f09b]450            return model_name,param_name
[2296316]451   
452    def set_ftol(self, ftol=None):
453        """
454        Set ftol: Relative error desired in the sum of chi squares. 
455        """
456        # check if it is flaot
457        try:
458            f_tol = float(ftol)
459        except:
460            # default
461            f_tol = 1.49012e-08
462           
463        self.ftol = f_tol
464             
[66ff250]465    def stop_fit(self, uid):
[ed2ea6a]466        """
[5062bbf]467        Stop the fit engine
[ed2ea6a]468        """
[66ff250]469        if uid in self.fit_thread_list.keys():
470            calc_fit = self.fit_thread_list[uid]
471            if calc_fit is not  None and calc_fit.isrunning():
472                calc_fit.stop()
473                msg = "Fit stop!"
474                wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
475        #set the fit button label of page when fit stop is trigger from
476        #simultaneous fit pane
477        if  self.sim_page is not None and uid == self.sim_page.uid:
478            for uid, value in self.page_finder.iteritems():
479                if value.get_scheduled() == 1:
480                    if uid in self.fit_panel.opened_pages.keys():
481                        panel = self.fit_panel.opened_pages[uid]
482                        panel. _on_fit_complete()
[6bbeacd4]483 
[66ff250]484    def set_smearer(self, uid, smearer, qmin=None, qmax=None, draw=True):
[d89f09b]485        """
[5062bbf]486        Get a smear object and store it to a fit problem
487       
488        :param smearer: smear object to allow smearing data
489       
[ca7a626]490        """   
[66ff250]491        if uid not in self.page_finder.keys():
492            msg = "Cannot find ID: %s in page_finder" % str(uid)
[6bbeacd4]493            raise ValueError, msg
[66ff250]494        self.page_finder[uid].set_smearer(smearer)
[6bbeacd4]495        if draw:
496            ## draw model 1D with smeared data
[66ff250]497            data =  self.page_finder[uid].get_fit_data()
498            model = self.page_finder[uid].get_model()
[6bbeacd4]499            if model is None:
500                return
501            ## if user has already selected a model to plot
502            ## redraw the model with data smeared
[66ff250]503            smear = self.page_finder[uid].get_smearer()
504            self.draw_model(model=model, data=data, page_id=uid, smearer=smear,
[67e258c]505                qmin=qmin, qmax=qmax)
[ca7a626]506
[66ff250]507    def draw_model(self, model, page_id, data=None, smearer=None,
[67e258c]508                   enable1D=True, enable2D=False,
[5ef55d2]509                   state=None,
[fa65e99]510                   toggle_mode_on=False,
[2296316]511                   qmin=DEFAULT_QMIN, qmax=DEFAULT_QMAX, 
512                   qstep=DEFAULT_NPTS,
513                   update_chisqr=True):
[ca7a626]514        """
[5062bbf]515        Draw model.
516       
517        :param model: the model to draw
518        :param name: the name of the model to draw
519        :param data: the data on which the model is based to be drawn
520        :param description: model's description
521        :param enable1D: if true enable drawing model 1D
522        :param enable2D: if true enable drawing model 2D
523        :param qmin:  Range's minimum value to draw model
524        :param qmax:  Range's maximum value to draw model
525        :param qstep: number of step to divide the x and y-axis
[2296316]526        :param update_chisqr: update chisqr [bool]
[ed2ea6a]527             
[d89f09b]528        """
[8b6f489]529        if data.__class__.__name__ == "Data1D" or not enable2D:   
[f72333f]530            ## draw model 1D with no loaded data
[6bbeacd4]531           
[67e258c]532            self._draw_model1D(model=model, 
533                               data=data,
[66ff250]534                               page_id=page_id,
[67e258c]535                               enable1D=enable1D, 
536                               smearer=smearer,
537                               qmin=qmin,
538                               qmax=qmax, 
[fa65e99]539                               toggle_mode_on=toggle_mode_on,
[5ef55d2]540                               state=state,
[2296316]541                               qstep=qstep,
542                               update_chisqr=update_chisqr)
[f72333f]543        else:     
544            ## draw model 2D with no initial data
545             self._draw_model2D(model=model,
[66ff250]546                                page_id=page_id,
[67e258c]547                                data=data,
548                                enable2D=enable2D,
549                                smearer=smearer,
550                                qmin=qmin,
551                                qmax=qmax,
[5ef55d2]552                                state=state,
[fa65e99]553                                toggle_mode_on=toggle_mode_on,
[2296316]554                                qstep=qstep,
555                                update_chisqr=update_chisqr)
[2db1d66]556           
[ca7a626]557    def onFit(self):
558        """
[5062bbf]559        perform fit
[ca7a626]560        """
[bb18ef1]561        ##  count the number of fitproblem schedule to fit
[67e258c]562        fitproblem_count = 0
[66ff250]563        for value in self.page_finder.values():
[67e258c]564            if value.get_scheduled() == 1:
[bb18ef1]565                fitproblem_count += 1
[2140e68]566               
[bb18ef1]567        ## if simultaneous fit change automatically the engine to park
[67e258c]568        if fitproblem_count > 1:
[bb18ef1]569            self._on_change_engine(engine='park')
[c9a4377]570           
[c660493]571        self.fitproblem_count = fitproblem_count 
572         
[bb18ef1]573        from sans.fit.Fitting import Fit
[66ff250]574        fitter = Fit(self._fit_engine)
[bb18ef1]575       
[67e258c]576        if self._fit_engine == "park":
577            engineType = "Simultaneous Fit"
[ca7a626]578        else:
[67e258c]579            engineType = "Single Fit"
[ca7a626]580           
581        fproblemId = 0
[67e258c]582        self.current_pg = None
[66ff250]583        list_page_id = []
584        for page_id, value in self.page_finder.iteritems():
[d89f09b]585            try:
[67e258c]586                if value.get_scheduled() == 1:
[f93dfcb]587                    #Get list of parameters name to fit
[d89f09b]588                    pars = []
589                    templist = []
[66ff250]590                   
591                    page = self.fit_panel.get_page_by_id(page_id)
[3b19ac9]592                    templist = page.get_param_list()
[d89f09b]593                    for element in templist:
[513115c]594                        name = str(element[1])
[ca7a626]595                        pars.append(name)
596                    #Set Engine  (model , data) related to the page on
[67e258c]597                    self._fit_helper(value=value, pars=pars,
[66ff250]598                                     fitter=fitter,
599                                      fitproblem_id=fproblemId,
600                                      title=engineType) 
601                    list_page_id.append(page_id)
[ca7a626]602                    fproblemId += 1 
[66ff250]603                    current_page_id = page_id
[d89f09b]604            except:
[6bbeacd4]605                raise
606                #msg= "%s error: %s" % (engineType, sys.exc_value)
607                #wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
608                #                                      type="stop"))
609                #return
[e54d2c32]610        ## If a thread is already started, stop it
611        #if self.calc_fit!= None and self.calc_fit.isrunning():
612        #    self.calc_fit.stop()
613         #Handler used for park engine displayed message
[66ff250]614        handler = ConsoleUpdate(parent=self.parent,
615                                manager=self,
616                                improvement_delta=0.1)
617       
[e54d2c32]618        ## perform single fit
619        if fitproblem_count == 1:
[58e0c83]620            calc_fit = FitThread(handler = handler,
[66ff250]621                                    fn=fitter,
[2296316]622                                    pars=pars,
623                                    page_id=list_page_id,
624                                    completefn=self._single_fit_completed,
625                                    ftol=self.ftol)
[e54d2c32]626        else:
[66ff250]627            current_page_id = self.sim_page.uid
[e54d2c32]628            ## Perform more than 1 fit at the time
[58e0c83]629            calc_fit = FitThread(handler=handler,
[66ff250]630                                    fn=fitter,
631                                    page_id=list_page_id,
[2296316]632                                    updatefn=handler.update_fit,
633                                    ftol=self.ftol)
[66ff250]634        self.fit_thread_list[current_page_id] = calc_fit
[e54d2c32]635        calc_fit.queue()
636        self.ready_fit(calc_fit=calc_fit)
637     
[66ff250]638           
[662da312]639    def ready_fit(self, calc_fit):
[c660493]640        """
641        Ready for another fit
642        """
643        if self.fitproblem_count != None and self.fitproblem_count > 1:
[662da312]644            calc_fit.ready(2.5)
[c660493]645           
646        else:
647            time.sleep(0.4)
[2f189dc]648           
[66ff250]649    def remove_plot(self, uid, theory=False):
[2f189dc]650        """
[5062bbf]651        remove model plot when a fit page is closed
[2f189dc]652        """
[66ff250]653        fitproblem = self.page_finder[uid]
[2f189dc]654        data = fitproblem.get_fit_data()
655        model = fitproblem.get_model()
[66ff250]656        plot_id = None
[2f189dc]657        if model is not None:
[66ff250]658            plot_id = data.id + name
[2f189dc]659        if theory:
[66ff250]660            plot_id = data.id
[e88ebfd]661        group_id = data.group_id
[66ff250]662        wx.PostEvent(self.parent, NewPlotEvent(id=plot_id,
[6bbeacd4]663                                                   group_id=group_id,
[3fafe15]664                                                   action='remove'))
[edcbd467]665           
[66ff250]666    def store_data(self, uid, data=None, caption=None):
[31b0c47]667        """
[5062bbf]668        Helper to save page reference into the plug-in
669       
670        :param page: page to store
671       
[31b0c47]672        """
673        #create a fitproblem storing all link to data,model,page creation
[66ff250]674        if not uid in self.page_finder.keys():
675            self.page_finder[uid] = FitProblem()
676        self.page_finder[uid].set_fit_data(data)
677        self.page_finder[uid].set_fit_tab_caption(caption)
[31b0c47]678       
[6bbeacd4]679    def on_add_new_page(self, event=None):
[edcbd467]680        """
[6bbeacd4]681        ask fit panel to create a new empty page
[edcbd467]682        """
683        try:
[6bbeacd4]684            page = self.fit_panel.add_empty_page()
685            page_caption = page.window_name
[6450d8c]686            # add data associated to the page created
687            if page != None: 
[66ff250]688                self.store_data(uid=page.uid, caption=page_caption,
[6bbeacd4]689                                data=page.get_data())
[f304194]690                wx.PostEvent(self.parent, StatusEvent(status="Page Created",
[31b0c47]691                                               info="info"))
[edcbd467]692            else:
[f304194]693                msg = "Page was already Created"
[a0da535]694                wx.PostEvent(self.parent, StatusEvent(status=msg,
695                                                       info="warning"))
[edcbd467]696        except:
[6bbeacd4]697            raise
698            #msg = "Creating Fit page: %s"%sys.exc_value
699            #wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
700       
701       
702    def add_fit_page(self, data):
703        """
704        given a data, ask to the fitting panel to create a new fitting page,
705        get this page and store it into the page_finder of this plug-in
706        """
707        page = self.fit_panel.set_data(data)
708        page_caption = page.window_name
709        #append Data1D to the panel containing its theory
710        #if theory already plotted
[66ff250]711        if page.uid in self.page_finder:
712            theory_data = self.page_finder[page.uid].get_theory_data()
[6bbeacd4]713            if issubclass(data.__class__, Data2D):
[e88ebfd]714                data.group_id = wx.NewId()
[a5701e6]715                if theory_data is not None:
[66ff250]716                    group_id = str(page.uid) + " Model1D"
[a5701e6]717                    wx.PostEvent(self.parent, 
718                             NewPlotEvent(group_id=group_id,
719                                               action="delete"))
[ae4ade7]720                    self.parent.update_data(prev_data=theory_data, new_data=data)     
[6bbeacd4]721            else:
722                if theory_data is not None:
[66ff250]723                    group_id = str(page.uid) + " Model2D"
[e88ebfd]724                    data.group_id = theory_data.group_id
[a5701e6]725                    wx.PostEvent(self.parent, 
726                             NewPlotEvent(group_id=group_id,
727                                               action="delete"))
[e88ebfd]728                    self.parent.update_data(prev_data=theory_data, new_data=data)   
729             
[66ff250]730        self.store_data(uid=page.uid, data=data, caption=page.window_name)
[6bbeacd4]731        if self.sim_page is not None:
732            self.sim_page.draw_page()
733           
[848a2ef]734    def _onEVT_SLICER_PANEL(self, event):
735        """
[5062bbf]736        receive and event telling to update a panel with a name starting with
737        event.panel_name. this method update slicer panel for a given interactor.
738       
739        :param event: contains type of slicer , paramaters for updating the panel
[848a2ef]740            and panel_name to find the slicer 's panel concerned.
741        """
742        for item in self.parent.panels:
[df4c3ad]743            if self.parent.panels[item].window_caption.startswith(event.panel_name):
[848a2ef]744                self.parent.panels[item].set_slicer(event.type, event.params)
[df4c3ad]745               
746        self.parent._mgr.Update()
[b324aa9]747   
[848a2ef]748    def _closed_fitpage(self, event):   
749        """
[5062bbf]750        request fitpanel to close a given page when its unique data is removed
751        from the plot. close fitpage only when the a loaded data is removed
[848a2ef]752        """   
[784e2fa]753        if event is None or event.data is None:
754            return
755        if hasattr(event.data,"is_data"):
[a0da535]756            if not event.data.is_data or \
[66ff250]757                event.data.__class__.__name__ == "Data1D":
[784e2fa]758                self.fit_panel.close_page_with_data(event.data) 
[848a2ef]759       
[66ff250]760    def _add_page_onmenu(self, name, fitproblem=None):
[6f023e8]761        """
[5062bbf]762        Add name of a closed page of fitpanel in a menu
[6f023e8]763        """
[dcf29d7]764        list = self.menu1.GetMenuItems()
765        for item in list:
766            if name == item.GetItemLabel():
[cfc0913]767                self.closed_page_dict[name][1] = fitproblem
[dcf29d7]768               
769        if not name in self.closed_page_dict.keys():   
770            # Post paramters
[77e23a2]771            event_id = wx.NewId()
772            self.menu1.Append(event_id, name, "Show %s fit panel" % name)
[cfc0913]773            self.closed_page_dict[name]= [event_id, fitproblem]
[66ff250]774            wx.EVT_MENU(self.parent, event_id, self._open_closed_page)
[ca7a626]775       
[6f023e8]776    def _open_closed_page(self, event):   
777        """
[5062bbf]778        reopen a closed page
[6f023e8]779        """
[cfc0913]780        for name, value in self.closed_page_dict.iteritems():
[dcf29d7]781            if event.GetId() in value:
[66ff250]782                uid,fitproblem = value
[b787e68c]783                if name !="Model":
[cfc0913]784                    data= fitproblem.get_fit_data()
[e88ebfd]785                    page = self.fit_panel.add_fit_page(data=data, reset=True)
[0f5fe6b]786                    if fitproblem != None:
[66ff250]787                        self.page_finder[uid] = fitproblem
[7c845cb]788                        if self.sim_page != None:
789                            self.sim_page.draw_page()
790                           
[0f5fe6b]791                else:
[b787e68c]792                    model = fitproblem
793                    self.fit_panel.add_model_page(model=model, topmenu=True,
794                                                  reset= True)
[0f5fe6b]795                    break
[5062bbf]796   
[66ff250]797    def _reset_schedule_problem(self, value=0):
[6f023e8]798        """
[5062bbf]799        unschedule or schedule all fitproblem to be fit
[6f023e8]800        """
[883e5f5]801        for page_id in self.page_finder.keys():
[66ff250]802            self.page_finder[page_id].schedule_tofit(value)
[6f023e8]803           
[66ff250]804    def _fit_helper(self, pars, value, fitproblem_id,fitter, title="Single Fit " ):
[2140e68]805        """
[5062bbf]806        helper for fitting
[2140e68]807        """
[ca7a626]808        metadata = value.get_fit_data()
[24cab5d]809        model = value.get_model()
810        smearer = value.get_smearer()
[67e258c]811        qmin, qmax = value.get_range()
[66ff250]812        self.fit_id = fitproblem_id
[24cab5d]813        #Create list of parameters for fitting used
[67e258c]814        templist = []
[464fce54]815       
[24cab5d]816        try:
[464fce54]817            #Extra list of parameters and their constraints
[67e258c]818            listOfConstraint = []
[464fce54]819           
[24cab5d]820            param = value.get_model_param()
[67e258c]821            if len(param) > 0:
[24cab5d]822                for item in param:
823                    ## check if constraint
[67e258c]824                    if item[0] != None and item[1] != None:
[464fce54]825                        listOfConstraint.append((item[0],item[1]))
826                   
[24cab5d]827            #Do the single fit
[66ff250]828            fitter.set_model(model, self.fit_id,
[67e258c]829                                   pars, constraints=listOfConstraint)
830           
[66ff250]831            fitter.set_data(data=metadata, id=self.fit_id,
[67e258c]832                                 smearer=smearer, qmin=qmin, qmax=qmax)
[784e2fa]833           
[66ff250]834            fitter.select_problem_for_fit(id=self.fit_id,
[67e258c]835                                               value=value.get_scheduled())
[60d6c23]836            value.clear_model_param()
[24cab5d]837        except:
[6bbeacd4]838            raise
839            #msg = title + " error: %s" % sys.exc_value
840            #wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[67e258c]841         
[c77d859]842    def _onSelect(self,event):
843        """
[5062bbf]844        when Select data to fit a new page is created .Its reference is
845        added to self.page_finder
[d89f09b]846        """
[c77d859]847        self.panel = event.GetEventObject()
[67e258c]848        Plugin.on_perspective(self, event=event)
[6ed82c7]849        for plottable in self.panel.graph.plottables:
[2fff4db]850            if plottable.__class__.__name__ in ["Data1D", "Theory1D"]:
[edcbd467]851                if plottable.name == self.panel.graph.selected_plottable:
[6bbeacd4]852                    data = plottable
[edcbd467]853                    self.add_fit_page(data=data)
854                    return
[c77d859]855            else:
[6bbeacd4]856                data = plottable
[edcbd467]857                self.add_fit_page(data=data)
[1c1436d]858           
[66ff250]859    def update_fit(self, result=None, msg=""):
860        """
861        """
862        print "update_fit result", result
863       
864    def _single_fit_completed(self, result, pars, page_id, elapsed=None):
[c77d859]865        """
[5062bbf]866        Display fit result on one page of the notebook.
867       
868        :param result: result of fit
869        :param pars: list of names of parameters fitted
870        :param current_pg: the page where information will be displayed
871        :param qmin: the minimum value of x to replot the model
872        :param qmax: the maximum value of x to replot model
[c77d859]873         
[7975f2b]874        """     
[c77d859]875        try:
[66ff250]876            if result == None:
[2296316]877                msg= "Single Fitting did not converge!!!"
878                wx.PostEvent(self.parent, 
879                             StatusEvent(status=msg, 
880                                         info="warning",
881                                         type="stop"))
[ad6dd4c]882                return
[67e258c]883            if not numpy.isfinite(result.fitness) or \
884                    numpy.any(result.pvec == None) or \
885                    not numpy.all(numpy.isfinite(result.pvec)):
886                msg = "Single Fitting did not converge!!!"
887                wx.PostEvent(self.parent, 
[2296316]888                             StatusEvent(status=msg, 
889                                         info="warning",
890                                         type="stop"))
[d902cf0]891                return
[66ff250]892            for uid in page_id:
893                value = self.page_finder[uid]   
894                model = value.get_model()
895                page_id = uid
896                   
897                param_name = []
898                for name in pars:
899                    param_name.append(name)
900                   
901                cpage = self.fit_panel.get_page_by_id(uid)
902                cpage.onsetValues(result.fitness, 
903                                  param_name, result.pvec,result.stderr)
904                cpage._on_fit_complete()
[2296316]905
906        except ValueError:
907            msg = "Single Fitting did not converge!!!"
908            wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
909                                                  type="stop"))
910            return   
[c77d859]911        except:
[66ff250]912            raise
913            #msg = "Single Fit completed but Following"
914            #msg += " error occurred:%s" % sys.exc_value
915            #wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
916            #                                      type="stop"))
[25e3fc9]917            return
[c77d859]918       
[66ff250]919    def _simul_fit_completed(self, result, page_id,pars=None, elapsed=None):
[c77d859]920        """
[5062bbf]921        Parameter estimation completed,
922        display the results to the user
923       
924        :param alpha: estimated best alpha
925        :param elapsed: computation time
926       
[c77d859]927        """
[66ff250]928        if page_id is None:
929            page_id = []
[ca7a626]930        ## fit more than 1 model at the same time
931        try:
[c69b6d5]932            msg = "" 
[66ff250]933            if result == None:
[2296316]934                msg= "Complex Fitting did not converge!!!"
[66ff250]935                wx.PostEvent(self.parent, StatusEvent(status=msg,
936                                                      type="stop"))
[ad6dd4c]937                return
[66ff250]938            if not numpy.isfinite(result.fitness) or \
939                numpy.any(result.pvec == None) or not \
940                numpy.all(numpy.isfinite(result.pvec)):
[6bbeacd4]941                msg= "Simultaneous Fitting did not converge!!!"
[ad6dd4c]942                wx.PostEvent(self.parent, StatusEvent(status=msg,type="stop"))
943                return
[c69b6d5]944             
[66ff250]945            for uid in page_id:   
946                value = self.page_finder[uid]
947                model = value.get_model()
948                data =  value.get_fit_data()
949                small_param_name = []
950                small_out = []
951                small_cov = []
952                #Separate result in to data corresponding to each page
953                for p in result.parameters:
954                    model_name, param_name = self.split_string(p.name) 
955                    if model.name == model_name:
956                        p_name= model.name+"."+param_name
957                        if p.name == p_name:     
958                            if p.value != None and numpy.isfinite(p.value):
959                                small_out.append(p.value)
960                                small_param_name.append(param_name)
961                                small_cov.append(p.stderr)
962                # Display result on each page
963                cpage = self.fit_panel.get_page_by_id(uid)
964                cpage.onsetValues(result.fitness,
965                                  small_param_name,
966                                  small_out,small_cov)
967                cpage._on_fit_complete()
[2296316]968               
969        except Exception:
970            msg = "Complex Fitting did not converge!!!"
971            wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
972                                                  type="stop"))
973            return
974
[ca7a626]975        except:
[66ff250]976            msg = "Simultaneous Fit completed"
977            msg += " but Following error occurred:%s" % sys.exc_value
978            wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
979   
[c77d859]980    def _on_show_panel(self, event):
[5062bbf]981        """
982        """
983        pass
984       
[c77d859]985    def _onset_engine_park(self,event):
986        """
[5062bbf]987        set engine to park
[c77d859]988        """
989        self._on_change_engine('park')
990       
991    def _onset_engine_scipy(self,event):
992        """
[5062bbf]993        set engine to scipy
[c77d859]994        """
995        self._on_change_engine('scipy')
996       
997    def _on_slicer_event(self, event):
998        """
[5062bbf]999        Receive a panel as event and send it to guiframe
1000       
1001        :param event: event containing a panel
1002       
[c77d859]1003        """
[5062bbf]1004        if event.panel is not None:
[c77d859]1005            new_panel = event.panel
[c3435b7f]1006            self.slicer_panels.append(event.panel)
[c77d859]1007            # Set group ID if available
1008            event_id = self.parent.popup_panel(new_panel)
[3fef0a8]1009            #self.menu3.Append(event_id, new_panel.window_caption,
1010            #                 "Show %s plot panel" % new_panel.window_caption)
[67e258c]1011            # Set id to allow us to reference the panel later
[df4c3ad]1012         
[c77d859]1013            new_panel.uid = event_id
1014            self.mypanels.append(new_panel) 
[5062bbf]1015       
[848a2ef]1016    def _onclearslicer(self, event):
1017        """
[5062bbf]1018        Clear the boxslicer when close the panel associate with this slicer
[848a2ef]1019        """
[df4c3ad]1020        name =event.GetPane().caption
[c3435b7f]1021   
1022        for panel in self.slicer_panels:
1023            if panel.window_caption==name:
1024               
[df4c3ad]1025                for item in self.parent.panels:
[c3435b7f]1026                    if hasattr(self.parent.panels[item],"uid"):
1027                        if self.parent.panels[item].uid ==panel.base.uid:
[df4c3ad]1028                            self.parent.panels[item].onClearSlicer(event)
1029                            self.parent._mgr.Update()
1030                            break 
[c3435b7f]1031                break
[5062bbf]1032   
[707436d]1033    def _return_engine_type(self):
1034        """
[5062bbf]1035        return the current type of engine
[707436d]1036        """
1037        return self._fit_engine
1038     
1039     
[c77d859]1040    def _on_change_engine(self, engine='park'):
1041        """
[5062bbf]1042        Allow to select the type of engine to perform fit
1043       
1044        :param engine: the key work of the engine
1045       
[c77d859]1046        """
[77e23a2]1047        ## saving fit engine name
[c77d859]1048        self._fit_engine = engine
[707436d]1049        ## change menu item state
1050        if engine=="park":
1051            self.menu1.FindItemByPosition(0).Check(False)
1052            self.menu1.FindItemByPosition(1).Check(True)
1053        else:
1054            self.menu1.FindItemByPosition(0).Check(True)
1055            self.menu1.FindItemByPosition(1).Check(False)
[77e23a2]1056        ## post a message to status bar
[a0da535]1057        msg = "Engine set to: %s" % self._fit_engine
1058        wx.PostEvent(self.parent, 
1059                     StatusEvent(status=msg))
[66ff250]1060        ## send the current engine type to fitpanel
1061        self.fit_panel._on_engine_change(name=self._fit_engine)
[77e23a2]1062       
[c77d859]1063    def _on_model_panel(self, evt):
1064        """
[5062bbf]1065        react to model selection on any combo box or model menu.plot the model 
1066       
1067        :param evt: wx.combobox event
1068       
[c77d859]1069        """
1070        model = evt.model
[66ff250]1071        uid = evt.uid
[6bbeacd4]1072        qmin = evt.qmin
1073        qmax = evt.qmax
1074        smearer = evt.smearer
[ae4ade7]1075       
[9237df4]1076        if model == None:
[c77d859]1077            return
[6bbeacd4]1078       
[66ff250]1079        if self.page_finder[uid].get_model() is None:
[6bbeacd4]1080            model.name = "M" + str(self.index_model)
1081            self.index_model += 1 
1082        else:
[66ff250]1083            model.name = self.page_finder[uid].get_model().name
[6bbeacd4]1084        # save the name containing the data name with the appropriate model
[66ff250]1085        self.page_finder[uid].set_model(model)
1086        self.page_finder[uid].set_range(qmin=qmin, qmax=qmax)
[6bbeacd4]1087        if self.sim_page is not None:
1088            self.sim_page.draw_page()
[5062bbf]1089       
[c77d859]1090    def _update1D(self,x, output):
[bb18ef1]1091        """
[5062bbf]1092        Update the output of plotting model 1D
[bb18ef1]1093        """
[58e0c83]1094        msg = "Plot updating ... "
1095        wx.PostEvent(self.parent, StatusEvent(status=msg,type="update"))
1096        self.ready_fit()
1097       
[bb18ef1]1098   
[66ff250]1099    def _fill_default_model2D(self, theory, page_id, qmax,qstep, qmin=None):
[ed2ea6a]1100        """
[5062bbf]1101        fill Data2D with default value
1102       
1103        :param theory: Data2D to fill
1104       
[ed2ea6a]1105        """
[d15c0202]1106        from DataLoader.data_info import Detector, Source
[bb18ef1]1107       
[d15c0202]1108        detector = Detector()
[e575db9]1109        theory.detector.append(detector)         
[d15c0202]1110        theory.source= Source()
[f93dfcb]1111       
[e575db9]1112        ## Default values   
1113        theory.detector[0].distance= 8000   # mm       
1114        theory.source.wavelength= 6         # A     
1115        theory.detector[0].pixel_size.x= 5  # mm
1116        theory.detector[0].pixel_size.y= 5  # mm
1117       
[ac2cc940]1118        theory.detector[0].beam_center.x= qmax
1119        theory.detector[0].beam_center.y= qmax
[5062bbf]1120   
[f93dfcb]1121        ## create x_bins and y_bins of the model 2D
[13e120a]1122        pixel_width_x = theory.detector[0].pixel_size.x
1123        pixel_width_y = theory.detector[0].pixel_size.y
1124        center_x      = theory.detector[0].beam_center.x/pixel_width_x
1125        center_y      = theory.detector[0].beam_center.y/pixel_width_y
[e575db9]1126
[a0da535]1127        # theory default: assume the beam
1128        #center is located at the center of sqr detector
[e575db9]1129        xmax = qmax
1130        xmin = -qmax
1131        ymax = qmax
1132        ymin = -qmax
1133       
1134        x=  numpy.linspace(start= -1*qmax,
[a0da535]1135                               stop=qmax,
1136                               num=qstep,
1137                               endpoint=True) 
1138        y = numpy.linspace(start=-1*qmax,
[e575db9]1139                               stop= qmax,
1140                               num= qstep,
[a0da535]1141                               endpoint=True)
[e575db9]1142         
1143        ## use data info instead
1144        new_x = numpy.tile(x, (len(y),1))
1145        new_y = numpy.tile(y, (len(x),1))
1146        new_y = new_y.swapaxes(0,1)
[d15c0202]1147       
[e575db9]1148        # all data reuire now in 1d array
1149        qx_data = new_x.flatten()
1150        qy_data = new_y.flatten()
[13e120a]1151       
[e575db9]1152        q_data = numpy.sqrt(qx_data*qx_data+qy_data*qy_data)
1153        # set all True (standing for unmasked) as default
1154        mask    = numpy.ones(len(qx_data), dtype = bool)
1155       
[a0da535]1156        # calculate the range of qx and qy: this way,
1157        # it is a little more independent
[e575db9]1158        x_size = xmax- xmin
1159        y_size = ymax -ymin
1160       
1161        # store x and y bin centers in q space
1162        x_bins  = x
1163        y_bins  = y
1164        # bin size: x- & y-directions
1165        xstep = x_size/len(x_bins-1)
1166        ystep = y_size/len(y_bins-1)
1167       
1168        #theory.data = numpy.zeros(len(mask))
[75a8723]1169        theory.err_data = numpy.ones(len(mask))
[e575db9]1170        theory.qx_data = qx_data
1171        theory.qy_data = qy_data 
1172        theory.q_data = q_data
1173        theory.mask = mask           
1174        theory.x_bins = x_bins 
1175        theory.y_bins = y_bins   
1176       
1177        # max and min taking account of the bin sizes
[a0da535]1178        theory.xmin = xmin
1179        theory.xmax = xmax
1180        theory.ymin = ymin
1181        theory.ymax = ymax
[66ff250]1182        theory.group_id = str(page_id) + " Model2D"
1183        theory.id = str(page_id) + " Model2D"
[6bbeacd4]1184 
[66ff250]1185    def _complete1D(self, x,y, page_id, elapsed,index,model,
[2296316]1186                    toggle_mode_on=False,state=None, 
1187                    data=None, update_chisqr=True):
[c77d859]1188        """
[5062bbf]1189        Complete plotting 1D data
[c77d859]1190        """ 
1191        try:
[6bbeacd4]1192            new_plot = Data1D(x=x, y=y)
[dce84c0]1193            new_plot.is_data = False
[6bbeacd4]1194            new_plot.symbol = GUIFRAME_ID.CURVE_SYMBOL_NUM
[a0da535]1195            if data != None:
[6bbeacd4]1196                _xaxis, _xunit = data.get_xaxis() 
1197                _yaxis, _yunit = data.get_yaxis() 
1198                new_plot.title = data.name
1199                #if the theory is already plotted use the same group id
1200                #to replot
[66ff250]1201                if page_id in self.page_finder:
1202                    theory_data = self.page_finder[page_id].get_theory_data()
[6bbeacd4]1203                    if theory_data is not None:
[e88ebfd]1204                       data.group_id = theory_data.group_id
[6bbeacd4]1205                #data is plotted before the theory, then take its group_id
1206                #assign to the new theory
[e88ebfd]1207                new_plot.group_id = data.group_id
[a5701e6]1208               
[6bbeacd4]1209            else:
1210                _xaxis, _xunit = "\\rm{Q}", 'A^{-1}'
1211                _yaxis, _yunit = "\\rm{Intensity} ", "cm^{-1}"
1212                new_plot.title = "Analytical model 1D "
1213                #find a group id to plot theory without data
[66ff250]1214                new_plot.group_id =  str(page_id) + " Model1D" 
1215            new_plot.id =  str(page_id) + " Model1D" 
[3fafe15]1216           
[6bbeacd4]1217            #find if this theory was already plotted and replace that plot given
1218            #the same id
[66ff250]1219           
1220            theory_data = self.page_finder[page_id].get_theory_data()
1221            if theory_data is not None:
1222                new_plot.id = theory_data.id
[3fafe15]1223             
[6bbeacd4]1224            new_plot.name = model.name + " ["+ str(model.__class__.__name__)+ "]"
1225            new_plot.xaxis(_xaxis, _xunit)
1226            new_plot.yaxis(_yaxis, _yunit)
[3fafe15]1227            if toggle_mode_on:
[66ff250]1228                new_plot.id =  str(page_id) + " Model" 
[3fafe15]1229                wx.PostEvent(self.parent, 
[66ff250]1230                             NewPlotEvent(group_id=str(page_id) + " Model2D",
[3fafe15]1231                                               action="Hide"))
[e88ebfd]1232           
[66ff250]1233            self.page_finder[page_id].set_theory_data(new_plot)
[a0da535]1234            if data is None:
[72323d1]1235                data_id = None
[c77d859]1236            else:
[5ef55d2]1237                data_id = data.id
[e88ebfd]1238            self.parent.update_theory(data_id=data_id, 
[05319e2]1239                                       theory=new_plot,
[e88ebfd]1240                                       state=state)     
[66ff250]1241            current_pg = self.fit_panel.get_page_by_id(page_id)
[ae4ade7]1242            title = new_plot.title
1243            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot,
1244                                            title= str(title)))
[2296316]1245            if update_chisqr:
1246                wx.PostEvent(current_pg,
1247                             Chi2UpdateEvent(output=self._cal_chisqr(data=data,
[66ff250]1248                                                        page_id=page_id,
[6bbeacd4]1249                                                        index=index)))
[2296316]1250            else:
1251                self._plot_residuals(page_id, data, index)
1252
[847091f]1253            msg = "Plot 1D  complete !"
[a0da535]1254            wx.PostEvent( self.parent, StatusEvent(status=msg, type="stop" ))
[2296316]1255            #self.current_pg.state.theory_data = deepcopy(self.theory_data)
[c77d859]1256        except:
[6bbeacd4]1257            raise
1258            #msg = " Error occurred when drawing %s Model 1D: " % new_plot.name
1259            #msg += " %s"  % sys.exc_value
1260            #wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[5062bbf]1261   
[c77d859]1262    def _update2D(self, output,time=None):
1263        """
[5062bbf]1264        Update the output of plotting model
[c77d859]1265        """
[58e0c83]1266        wx.PostEvent(self.parent, StatusEvent(status="Plot \
[a0da535]1267        #updating ... ", type="update"))
[58e0c83]1268        self.ready_fit()
1269 
[66ff250]1270    def _complete2D(self, image, data, model, page_id,  elapsed, index, qmin,
[2296316]1271                     qmax, toggle_mode_on=False,state=None,qstep=DEFAULT_NPTS, 
1272                     update_chisqr=True):
[c77d859]1273        """
[5062bbf]1274        Complete get the result of modelthread and create model 2D
1275        that can be plot.
[c77d859]1276        """
1277        err_image = numpy.zeros(numpy.shape(image))
[a8088d7]1278       
[fa65e99]1279        new_plot= Data2D(image=image, err_image=err_image)
1280        new_plot.name = model.name
[c77d859]1281       
[a0da535]1282        if data is None:
[fa65e99]1283            self._fill_default_model2D(theory=new_plot, 
[a0da535]1284                                       qmax=qmax, 
[66ff250]1285                                       page_id=page_id,
[a0da535]1286                                       qstep=qstep,
[5062bbf]1287                                        qmin= qmin)
[6bbeacd4]1288           
[c77d859]1289        else:
[66ff250]1290            new_plot.id = str(page_id) + " Model2D"
1291            new_plot.group_id = str(page_id) + " Model2D"
[fa65e99]1292            new_plot.x_bins = data.x_bins
1293            new_plot.y_bins = data.y_bins
1294            new_plot.detector = data.detector
1295            new_plot.source = data.source
1296            new_plot.is_data = False 
1297            new_plot.qx_data = data.qx_data
1298            new_plot.qy_data = data.qy_data
1299            new_plot.q_data = data.q_data
[a0da535]1300            #numpy.zeros(len(data.err_data))#data.err_data
[fa65e99]1301            new_plot.err_data = err_image
1302            new_plot.mask = data.mask
[c77d859]1303            ## plot boundaries
[fa65e99]1304            new_plot.ymin = data.ymin
1305            new_plot.ymax = data.ymax
1306            new_plot.xmin = data.xmin
1307            new_plot.xmax = data.xmax
[dce84c0]1308        new_plot.is_data = False
[fa65e99]1309        new_plot.name = model.name + " ["+ str(model.__class__.__name__)+ "]"
1310        new_plot.title = "Analytical model 2D "
1311        theory_data = deepcopy(new_plot)
[5ef55d2]1312        theory_data.name = "Unknown"
[3fafe15]1313        if toggle_mode_on:
[66ff250]1314            new_plot.id = str(page_id) + " Model"     
[3fafe15]1315            wx.PostEvent(self.parent, 
[66ff250]1316                             NewPlotEvent(group_id=str(page_id) + " Model1D",
[3fafe15]1317                                               action="Hide"))
[6bbeacd4]1318       
[66ff250]1319        self.page_finder[page_id].set_theory_data(new_plot)
[5ef55d2]1320        if data is None:
[72323d1]1321            data_id = None
[5ef55d2]1322        else:
1323            data_id = data.id
[e88ebfd]1324        self.parent.update_theory(data_id=data_id, 
[05319e2]1325                                       theory=new_plot,
[e88ebfd]1326                                       state=state) 
[66ff250]1327        current_pg = self.fit_panel.get_page_by_id(page_id)
[ae4ade7]1328        title = new_plot.title
[fa65e99]1329        wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot,
[ae4ade7]1330                                               title=title))
[f72333f]1331        # Chisqr in fitpage
[2296316]1332        if update_chisqr:
1333            wx.PostEvent(current_pg,
1334                         Chi2UpdateEvent(output=\
1335                                    self._cal_chisqr(data=data,
[66ff250]1336                                                     page_id=page_id,
1337                                                     index=index)))
[2296316]1338        else:
1339            self._plot_residuals(page_id, data, index)
[847091f]1340        msg = "Plot 2D complete !"
[a0da535]1341        wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[6bbeacd4]1342   
[66ff250]1343    def _draw_model2D(self, model, page_id, data=None, smearer=None,
[a0da535]1344                      description=None, enable2D=False,
[5ef55d2]1345                      state=None,
[fa65e99]1346                      toggle_mode_on=False,
[a0da535]1347                      qmin=DEFAULT_QMIN, qmax=DEFAULT_QMAX,
[2296316]1348                      qstep=DEFAULT_NPTS,
1349                      update_chisqr=True):
[f93dfcb]1350        """
[5062bbf]1351        draw model in 2D
1352       
1353        :param model: instance of the model to draw
1354        :param description: the description of the model
1355        :param enable2D: when True allows to draw model 2D
1356        :param qmin: the minimum value to  draw model 2D
1357        :param qmax: the maximum value to draw model 2D
1358        :param qstep: the number of division of Qx and Qy of the model to draw
[f93dfcb]1359           
1360        """
[a0da535]1361        x=  numpy.linspace(start=-1*qmax,
1362                               stop=qmax,
1363                               num=qstep,
1364                               endpoint=True) 
[d15c0202]1365        y = numpy.linspace(start= -1*qmax,
[a0da535]1366                               stop=qmax,
1367                               num=qstep,
1368                               endpoint=True)
[6bbeacd4]1369        if model is None:
[66ff250]1370            msg = "Panel with ID: %s does not contained model" % str(page_id)
[6bbeacd4]1371            raise ValueError, msg
[c77d859]1372        ## use data info instead
[a0da535]1373        if data is not None:
[c77d859]1374            ## check if data2D to plot
1375            if hasattr(data, "x_bins"):
1376                enable2D = True
[a0da535]1377                x = data.x_bins
1378                y = data.y_bins
[f72333f]1379               
[c77d859]1380        if not enable2D:
[a0da535]1381            return None, None
[c77d859]1382        try:
1383            from model_thread import Calc2D
1384            ## If a thread is already started, stop it
[a0da535]1385            if (self.calc_2D is not None) and self.calc_2D.isrunning():
[c77d859]1386                self.calc_2D.stop()
[f72333f]1387
[a0da535]1388            self.calc_2D = Calc2D(x=x,
1389                                    y=y,
1390                                    model=model, 
1391                                    data=data,
[66ff250]1392                                    page_id=page_id,
[a0da535]1393                                    smearer=smearer,
1394                                    qmin=qmin,
1395                                    qmax=qmax,
1396                                    qstep=qstep,
[fa65e99]1397                                    toggle_mode_on=toggle_mode_on,
[5ef55d2]1398                                    state=state,
[2296316]1399                                    completefn=self._complete2D,
1400                                    #updatefn= self._update2D,
1401                                    update_chisqr=update_chisqr)
1402
[c77d859]1403            self.calc_2D.queue()
[f72333f]1404
[c77d859]1405        except:
[6bbeacd4]1406            raise
1407            #msg = " Error occurred when drawing %s Model 2D: " % model.name
1408            #msg += " %s" % sys.exc_value
1409            #wx.PostEvent(self.parent, StatusEvent(status=msg))
[f72333f]1410
[66ff250]1411    def _draw_model1D(self, model, page_id, data=None, smearer=None,
[67e258c]1412                qmin=DEFAULT_QMIN, qmax=DEFAULT_QMAX, 
[5ef55d2]1413                state=None,
[fa65e99]1414                toggle_mode_on=False,
[2296316]1415                qstep=DEFAULT_NPTS, update_chisqr=True, 
1416                enable1D=True):
[c77d859]1417        """
[5062bbf]1418        Draw model 1D from loaded data1D
1419       
1420        :param data: loaded data
1421        :param model: the model to plot
1422       
[c77d859]1423        """
[a0da535]1424        x=  numpy.linspace(start=qmin,
1425                           stop=qmax,
1426                           num=qstep,
[c77d859]1427                           endpoint=True
1428                           )
[a0da535]1429        if data is not None:
[c77d859]1430            ## check for data2D
1431            if hasattr(data,"x_bins"):
1432                return
1433            x = data.x
[2296316]1434            if qmin == None :
1435                qmin == DEFAULT_QMIN
1436
1437            if qmax == None:
1438                qmax == DEFAULT_QMAX
[c77d859]1439        if not enable1D:
[f72333f]1440            return 
[c77d859]1441        try:
1442            from model_thread import Calc1D
1443            ## If a thread is already started, stop it
[a0da535]1444            if (self.calc_1D is not None) and self.calc_1D.isrunning():
[c77d859]1445                self.calc_1D.stop()
[a0da535]1446            self.calc_1D = Calc1D(x=x,
1447                                  data=data,
[6bbeacd4]1448                                  model=model,
[66ff250]1449                                  page_id=page_id, 
[a0da535]1450                                  qmin=qmin,
1451                                  qmax=qmax,
1452                                  smearer=smearer,
[5ef55d2]1453                                  state=state,
[fa65e99]1454                                  toggle_mode_on=toggle_mode_on,
[2296316]1455                                  completefn=self._complete1D,
1456                                  #updatefn = self._update1D,
1457                                  update_chisqr=update_chisqr)
1458
[c77d859]1459            self.calc_1D.queue()
[f72333f]1460
[c77d859]1461        except:
[a0da535]1462            msg = " Error occurred when drawing %s Model 1D: " % model.name
1463            msg += " %s" % sys.exc_value
1464            wx.PostEvent(self.parent, StatusEvent(status=msg))
[f72333f]1465
[66ff250]1466    def _cal_chisqr(self, page_id, data=None, index=None): 
[f72333f]1467        """
[5062bbf]1468        Get handy Chisqr using the output from draw1D and 2D,
1469        instead of calling expansive CalcChisqr in guithread
[f72333f]1470        """
1471        # default chisqr
1472        chisqr = None
[f151ddf]1473       
[f72333f]1474        # return None if data == None
1475        if data == None: return chisqr
[f151ddf]1476       
[f72333f]1477        # Get data: data I, theory I, and data dI in order
[2296316]1478        if data.__class__.__name__ == "Data2D":
[a0da535]1479            if index == None: 
1480                index = numpy.ones(len(data.data),ntype=bool)
1481            # get rid of zero error points
[2296316]1482            index = index & (data.err_data != 0 ) 
1483            index = index & (numpy.isfinite(data.data)) 
[f72333f]1484            fn = data.data[index] 
[66ff250]1485            theory_data = self.page_finder[page_id].get_theory_data()
[6bbeacd4]1486            gn = theory_data.data[index]
[f72333f]1487            en = data.err_data[index]
1488        else:
1489            # 1 d theory from model_thread is only in the range of index
[2296316]1490            if index == None: 
[a0da535]1491                index = numpy.ones(len(data.y), ntype=bool)
1492            if data.dy == None or data.dy == []:
[4d5fa8c]1493                dy = numpy.ones(len(data.y))
[f151ddf]1494            else:
[a0da535]1495                ## Set consitently w/AbstractFitengine:
1496                # But this should be corrected later.
[2296316]1497                dy = deepcopy(data.dy)
[4d5fa8c]1498                dy[dy==0] = 1 
[f72333f]1499            fn = data.y[index] 
[66ff250]1500            theory_data = self.page_finder[page_id].get_theory_data()
[6bbeacd4]1501            gn = theory_data.y
[4d5fa8c]1502            en = dy[index]
[f72333f]1503        # residual
[2296316]1504        res = (fn - gn) / en
1505        residuals = res[numpy.isfinite(res)]
[f72333f]1506        # get chisqr only w/finite
[2296316]1507        chisqr = numpy.average(residuals * residuals)
1508       
1509        self._plot_residuals(page_id, data, index)
[f72333f]1510        return chisqr
1511   
[2296316]1512
1513       
1514    def _plot_residuals(self, page_id, data=None, index=None): 
1515        """
1516        Plot the residuals
1517       
1518        :param data: data
1519        :param index: index array (bool)
1520        : Note: this is different from the residuals in cal_chisqr()
1521        """
1522        if data == None: 
1523            return 
1524       
1525        # Get data: data I, theory I, and data dI in order
1526        if data.__class__.__name__ == "Data2D":
1527            # build residuals
1528            #print data
1529            residuals = Data2D()
1530            #residuals.copy_from_datainfo(data)
1531            # Not for trunk the line below, instead use the line above
1532            data.clone_without_data(len(data.data), residuals)
1533            residuals.data = None
1534            fn = data.data#[index]
1535            theory_data = self.page_finder[page_id].get_theory_data()
1536            gn = theory_data.data#[index]
1537            en = data.err_data#[index]
1538            residuals.data = (fn - gn) / en
1539            residuals.qx_data = data.qx_data#[index]
1540            residuals.qy_data = data.qy_data #[index]
1541            residuals.q_data = data.q_data#[index]
1542            residuals.err_data = numpy.ones(len(residuals.data))#[index]
1543            residuals.xmin = min(residuals.qx_data)
1544            residuals.xmax = max(residuals.qx_data)
1545            residuals.ymin = min(residuals.qy_data)
1546            residuals.ymax = max(residuals.qy_data)
1547            residuals.q_data = data.q_data#[index]
1548            residuals.mask = data.mask
1549            residuals.scale = 'linear'
1550            #print "print data",residuals
1551            # check the lengths
1552            if len(residuals.data) != len(residuals.q_data):
1553                return
1554
1555        else:
1556            # 1 d theory from model_thread is only in the range of index
1557            if data.dy == None or data.dy == []:
1558                dy = numpy.ones(len(data.y))
1559            else:
1560                ## Set consitently w/AbstractFitengine:
1561                ## But this should be corrected later.
1562                dy = deepcopy(data.dy)
1563                dy[dy==0] = 1 
1564            fn = data.y[index] 
1565            theory_data = self.page_finder[page_id].get_theory_data()
1566            gn = theory_data.y
1567            en = dy[index]
1568            # build residuals
1569            residuals = Data1D()
1570            residuals.y = (fn - gn) / en
1571            residuals.x = data.x[index]
1572            residuals.dy = numpy.ones(len(residuals.y))
1573            residuals.dx = None
1574            residuals.dxl = None
1575            residuals.dxw = None
1576            residuals.ytransform = 'y'
1577            # For latter scale changes
1578            residuals.xaxis('\\rm{Q} ', 'A^{-1}')
1579            residuals.yaxis('\\rm{Residuals} ', 'normalized')
1580           
1581        new_plot = residuals
1582        if data.id == None:
1583            data.id = data.name
1584        name  = data.id
1585        new_plot.name = "Residuals for " + str(data.name)
1586        ## allow to highlight data when plotted
1587        new_plot.interactive = True
1588        ## when 2 data have the same id override the 1 st plotted
1589        new_plot.id = new_plot.name#name + " residuals"
1590        ##group_id specify on which panel to plot this data
1591        new_plot.group_id = new_plot.id
1592        #new_plot.is_data = True
1593        ##post data to plot
1594        title = new_plot.name
1595       
1596        # plot data
1597        wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title=title))   
1598   
[5062bbf]1599#def profile(fn, *args, **kw):
1600#    import cProfile, pstats, os
1601#    global call_result
1602#    def call():
1603#        global call_result
1604#        call_result = fn(*args, **kw)
1605#    cProfile.runctx('call()', dict(call=call), {}, 'profile.out')
1606#    stats = pstats.Stats('profile.out')
1607#    #stats.sort_stats('time')
1608#    stats.sort_stats('calls')
1609#    stats.print_stats()
1610#    os.unlink('profile.out')
1611#    return call_result
[d89f09b]1612if __name__ == "__main__":
1613    i = Plugin()
1614   
1615   
1616   
1617   
Note: See TracBrowser for help on using the repository browser.