Changeset 2140e68 in sasview for sansview/perspectives/fitting


Ignore:
Timestamp:
Mar 25, 2009 3:05:23 PM (16 years ago)
Author:
Gervaise Alina <gervyh@…>
Branches:
master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload
Children:
ac11e40
Parents:
c7ac15e
Message:

drawing sim_page

Location:
sansview/perspectives/fitting
Files:
1 added
5 edited

Legend:

Unmodified
Added
Removed

  • sansview/perspectives/fitting/fitpage.py

    rc77d859 r2140e68  
    354354            ## error occured on chisqr computation 
    355355            pass 
    356          
    357          
     356    def get_range(self): 
     357        """ 
     358            return the fitting range 
     359        """ 
     360        return self.qmin_x , self.qmax_x 
    358361         
    359362    def get_param_list(self): 
    … …  
    432435        if self.enable_smearer.GetValue(): 
    433436            from DataLoader.qsmearing import smear_selection 
    434             smear =smear_selection( self.data ) 
     437            smear = smear_selection( self.data ) 
    435438            if hasattr(self.data,"dxl"): 
    436439                msg= ": Resolution smearing parameters" 
    … …  
    443446                wx.PostEvent(self.manager.parent, StatusEvent(status=\ 
    444447                            "Data contains smearing information %s"%msg)) 
    445         self.manager.set_smearer(smear, qmin= self.qmin_x, qmax= self.qmax_x)    
     448            self.manager.set_smearer(smear, qmin= self.qmin_x, qmax= self.qmax_x)    
    446449               
    447450         

  • sansview/perspectives/fitting/fitpanel.py

    rc77d859 r2140e68  
    6363        page_number = self.GetSelection() 
    6464        if self.sim_page != selected_page and selected_page!=self.about_page: 
    65             # remove the check box link to the model name of this page (selected_page) 
    66             if self.sim_page !=None : 
    67                 self.sim_page.remove_model(selected_page) 
    6865            #remove that page from page_finder of fitting module 
    6966            page_finder=self.manager.get_page_finder()  
    … …  
    7269                    del page_finder[page] 
    7370                    break 
     71            # remove the check box link to the model name of this page (selected_page) 
     72            if self.sim_page !=None : 
     73                self.sim_page.draw_page(page_finder) 
     74             
    7475            #Delete the page from notebook 
    7576            if selected_page.name in self.fit_page_name: 

  • sansview/perspectives/fitting/fitproblem.py

    rbb18ef1 r2140e68  
    1 from sans.fit.AbstractFitEngine import Model 
    21 
    32class FitProblem: 
    … …  
    1110         
    1211        """ 
    13             @ self.data :is the data selected to perform the fit 
    14             @ self.theory_name: the name of the theory created with self.model 
    15             @ self.model_list:  is a list containing a model as first element  
    16             and its name assign example [lineModel, M0] 
     12           contains information about data and model to fit 
    1713        """ 
    1814        ## data used for fitting 
    1915        self.fit_data=None 
    20         ## list containing couple of model and its name 
    21         self.model_list=[] 
     16        ## the current model 
     17        self.model = None 
    2218        ## if 1 this fit problem will be selected to fit , if 0  
    2319        ## it will not be selected for fit 
    … …  
    3026        ## axis unit info and so on see plottables definition 
    3127        self.plotted_data=None 
     28        ## fitting range 
     29        self.qmin = None 
     30        self.qmax = None 
    3231         
    3332         
    … …  
    5251        return self.name_per_page 
    5352     
    54     def set_model(self,model,name): 
     53    def set_model(self,model): 
    5554        """  
    5655             associates each model with its new created name 
    … …  
    5857             @param name: name created for model 
    5958        """ 
     59        self.model= model 
    6060         
    61         if len(self.model_list)>0 and name==None: 
    62             self.model_list=[model,self.model_list[1]] 
    63         else: 
    64             self.model_list=[model,name] 
    65  
     61         
     62    def get_model(self): 
     63        """ @return: saved model """ 
     64        return self.model 
     65   
    6666   
    6767    def add_plotted_data(self,data): 
    … …  
    7272        self.plotted_data = data 
    7373         
    74          
     74 
     75    def get_plotted_data(self): 
     76        """ @return:  list of data dList""" 
     77        return self.plotted_data 
     78     
     79     
    7580    def add_fit_data(self,data): 
    7681        """  
    … …  
    8085        self.fit_data = data 
    8186             
    82     def get_model(self): 
    83         """ @return: saved model """ 
    84         return self.model_list 
    85       
    86     def get_plotted_data(self): 
    87         """ @return:  list of data dList""" 
    88         return self.plotted_data 
    89      
    90      
     87    
    9188    def get_fit_data(self): 
    9289        return self.fit_data 
    9390     
    94      
    95     def get_theory(self): 
    96         """ @return the name of theory for plotting purpose""" 
    97         return self.theory_name 
    98      
    99      
    100     def set_theory(self,name): 
    101         """ 
    102             Set theory name 
    103             @param name: name of the theory 
    104         """ 
    105         self.theory_name = name 
    106  
    10791         
    108     def set_model_param(self,name,value): 
     92    def set_model_param(self,name,value=None): 
    10993        """  
    11094            Store the name and value of a parameter of this fitproblem's model 
    … …  
    122106         
    123107         
    124     def reset_model(self,model): 
    125         """  
    126             reset a model when parameter has changed 
    127             @param value: new model 
    128         """ 
    129         self.model_list[0]=model 
    130          
    131          
    132108    def schedule_tofit(self, schedule=0): 
    133109        """ 
    … …  
    136112        self.schedule=schedule 
    137113         
     114         
    138115    def get_scheduled(self): 
    139116        """ return true or false if a problem as being schedule for fitting""" 
    140117        return self.schedule 
    141118     
     119 
     120    def set_range(self, qmin=None, qmax=None): 
     121        """ 
     122            set fitting range  
     123        """ 
     124        self.qmin = qmin 
     125        self.qmax = qmax 
     126         
     127         
     128    def get_range(self): 
     129        """ 
     130            @return fitting range: 
     131        """ 
     132        return self.qmin, self.qmax 
    142133     
    143134    def clear_model_param(self): 
    … …  
    147138        self.list_param=[] 
    148139         
     140     
     141     
    149142         

  • sansview/perspectives/fitting/fitting.py

    r6859338 r2140e68  
    120120            Create a page to access simultaneous fit option 
    121121        """ 
     122        if self.sim_page !=None: 
     123            msg= "Simultaneous Fit page already opened" 
     124            wx.PostEvent(self.parent, StatusEvent(status= msg)) 
     125            return  
     126         
    122127        self.sim_page= self.fit_panel.add_sim_page() 
    123         self.sim_page.add_model(self.page_finder) 
     128        self.sim_page.draw_page(self.page_finder) 
    124129         
    125130         
    … …  
    299304             
    300305       
    301     def on_single_fit(self,id=None,qmin=None, qmax=None): 
     306    def on_single_fit(self,id=0,qmin=None, qmax=None): 
    302307        """  
    303308            perform fit for the  current page  and return chisqr,out and cov 
    … …  
    311316        self.fitter = Fit(self._fit_engine) 
    312317        #Setting an id to store model and data in fit engine 
    313         self.fit_id = 0 
    314         if id!=None: 
    315             self.fit_id = id 
    316          
     318        self.fit_id = id 
     319        
    317320        page_fitted = None 
    318         fit_problem = None 
     321         
    319322        #Get information (model , data) related to the page on  
    320323        #with the fit will be perform 
    … …  
    326329            value = self.page_finder[current_pg] 
    327330            metadata =  value.get_fit_data() 
    328             list = value.get_model() 
    329             model = list[0] 
     331            model = value.get_model() 
    330332            smearer = value.get_smearer() 
    331333            
    … …  
    394396                return 
    395397          
    396     def on_simul_fit(self, id=None,qmin=None,qmax=None): 
     398    def on_simul_fit(self, id=0,qmin=None,qmax=None): 
    397399        """  
    398400            perform fit for all the pages selected on simpage and return chisqr,out and cov 
    … …  
    404406        """ 
    405407        ##Setting an id to store model and data 
    406         self.fit_id= 0 
    407         #Setting an id to store model and data 
    408         if id!=None: 
    409              self.fit_id= id 
     408        self.fit_id= id 
     409         
    410410        ##  count the number of fitproblem schedule to fit  
    411411        fitproblem_count= 0 
    … …  
    413413            if value.get_scheduled()==1: 
    414414                fitproblem_count += 1 
     415                 
    415416        ## if simultaneous fit change automatically the engine to park 
    416417        if fitproblem_count >1: 
    … …  
    424425                if value.get_scheduled()==1: 
    425426                    metadata = value.get_fit_data() 
    426                     list = value.get_model() 
    427                     model= list[0] 
     427                    model = value.get_model() 
     428                    smearer = value.get_smearer() 
     429                    qmin , qmax = value.get_range() 
    428430                    ## store page 
    429431                    cpage= page 
    … …  
    441443                    ## create a park model and reset parameter value if constraint 
    442444                    ## is given 
    443                     new_model=Model(model) 
    444                     param=value.get_model_param() 
    445                      
     445                    new_model = Model(model) 
     446                    param = value.get_model_param() 
    446447                    if len(param)>0: 
    447448                        for item in param: 
    448449                            param_value = item[1] 
    449450                            param_name = item[0] 
    450      
    451                             new_model.parameterset[ param_name].set( param_value ) 
    452                          
    453                     self.fitter.set_model(new_model, self.fit_id, pars)  
     451                            ## check if constraint 
     452                            if param_value !=None: 
     453                                new_model.parameterset[ param_name].set( param_value ) 
     454                 
     455                    self.fitter.set_model(model= new_model, Uid=self.fit_id, pars=pars)  
    454456                    ## check that non -zero value are send as dy in the fit engine 
    455457                    dy=[] 
    … …  
    469471                                metadata.x=numpy.zeros(len(x)) 
    470472                                metadata.x=x 
    471                                  
    472                     self.fitter.set_data(metadata,self.fit_id,qmin,qmax) 
     473                               
     474                    self.fitter.set_data( data= metadata, smearer=smearer, 
     475                                         Uid=self.fit_id, qmin=qmin, qmax=qmax) 
    473476                    self.fitter.select_problem_for_fit(Uid= self.fit_id, 
    474477                                                       value= value.get_scheduled()) 
    … …  
    486489            if self.calc_fit!= None and self.calc_fit.isrunning(): 
    487490                self.calc_fit.stop() 
    488             ## perform single fit 
     491                        ## perform single fit 
    489492            if  fitproblem_count==1: 
    490493                self.calc_fit=FitThread(parent =self.parent, 
    … …  
    492495                                       qmin=qmin, 
    493496                                       qmax=qmax, 
    494                                        ymin= ymin, 
    495                                        ymax= ymax, 
    496497                                       cpage=cpage, 
    497498                                       pars= pars, 
    … …  
    505506                                       qmin=qmin, 
    506507                                       qmax=qmax, 
    507                                        ymin= ymin, 
    508                                        ymax= ymax, 
    509508                                       completefn= self._simul_fit_completed, 
    510509                                       updatefn=None) 
    … …  
    565564                           qstep=qstep) 
    566565           
    567    
    568    
     566    def _fit_helper(self, page, id ): 
     567        """ 
     568            helper for fitting 
     569        """ 
     570        #set an engine to perform fit 
     571        from sans.fit.Fitting import Fit 
     572        self.fitter = Fit(self._fit_engine) 
     573        #Setting an id to store model and data in fit engine 
     574        self.fit_id = id 
     575        page_fitted = None 
     576         
     577         
     578        
    569579    def _onSelect(self,event): 
    570580        """  
    … …  
    629639            for page, value in self.page_finder.iteritems(): 
    630640                if page==cpage : 
    631                     list = value.get_model() 
    632                     model= list[0] 
     641                    model= value.get_model() 
    633642                    break 
    634643            i = 0 
    … …  
    643652            ## plot the current model with new param 
    644653            metadata =  self.page_finder[cpage].get_fit_data() 
    645             list = self.page_finder[cpage].get_model() 
    646             model = list[0] 
     654            model = self.page_finder[cpage].get_model() 
     655            
     656            #Replot models 
     657            msg= "Single Fit completed. plotting... %s:"%model.name 
     658            wx.PostEvent(self.parent, StatusEvent(status="%s " % msg)) 
    647659            self.draw_model( model=model, data= metadata,qmin= qmin, qmax= qmax) 
    648660             
    … …  
    653665        
    654666        
    655     def _simul_fit_completed(self,result,qmin,qmax, elapsed,pars=None,cpage=None, 
    656                              xmin=None, xmax=None, ymin=None, ymax=None): 
     667    def _simul_fit_completed(self,result,qmin,qmax, elapsed=None,pars=None,cpage=None): 
    657668        """ 
    658669            Parameter estimation completed,  
    … …  
    662673        """ 
    663674        if cpage!=None: 
    664             self._single_fit_completed(result=result,pars=pars,cpage=cpage, 
    665                                        qmin=qmin,qmax=qmax, 
    666                               ymin=ymin, ymax=ymax, xmin=xmin, xmax=xmax) 
     675            self._single_fit_completed(result=result, pars=pars, cpage=cpage, 
     676                                       qmin=qmin, qmax=qmax) 
    667677            return 
    668678        else: 
    … …  
    673683                for page, value in self.page_finder.iteritems(): 
    674684                    if value.get_scheduled()==1: 
    675                         list = value.get_model() 
    676                         model= list[0] 
    677                         
     685                        model = value.get_model() 
     686                        metadata =  value.get_plotted_data() 
    678687                        small_out = [] 
    679688                        small_cov = [] 
    … …  
    692701                        page.onsetValues(result.fitness, small_out,small_cov) 
    693702                        #Replot models 
    694                         msg= "Single Fit completed plotting %s:"%model.name 
     703                        msg= "Simultaneous Fit completed. plotting... %s:"%model.name 
    695704                        wx.PostEvent(self.parent, StatusEvent(status="%s " % msg)) 
    696                         self.plot_helper(currpage= page,qmin= qmin,qmax= qmax, 
    697                                          xmin=xmin, xmax=xmax, 
    698                                          ymin=ymin, ymax=ymax)  
     705                        self.draw_model( model=model, data= metadata,qmin= qmin, qmax= qmax) 
     706                         
    699707            except: 
    700708                 msg= "Simultaneous Fit completed but Following error occurred: " 
    … …  
    755763        if model ==None: 
    756764            return 
    757         name = model.name 
     765        
    758766        sim_page=self.sim_page 
    759767        current_pg = self.fit_panel.get_current_page()  
    … …  
    762770        if current_pg != sim_page: 
    763771            
    764             if len(self.page_finder[current_pg].get_model())==0: 
     772            if self.page_finder[current_pg].get_model()== None : 
    765773                 
    766774                model.name="M"+str(self.index_model) 
    767775                self.index_model += 1   
    768776            else: 
    769                 model.name= self.page_finder[current_pg].get_model()[0].name 
    770777                 
    771             try: 
    772                 metadata=self.page_finder[current_pg].get_plotted_data() 
    773                 M_name = model.name+"= "+name+"("+metadata.name+")" 
    774             except: 
    775                 M_name = model.name+"= "+name 
     778                model.name= self.page_finder[current_pg].get_model().name 
     779                 
     780            
     781            metadata = self.page_finder[current_pg].get_plotted_data() 
     782             
    776783            # save the name containing the data name with the appropriate model 
    777             self.page_finder[current_pg].set_model(model,M_name) 
    778              
     784            self.page_finder[current_pg].set_model(model) 
    779785            # save model name 
    780786            self.draw_model( model=model, data= metadata) 
    781787             
    782788            if self.sim_page!=None: 
    783                 self.sim_page.add_model(self.page_finder) 
     789                self.sim_page.draw_page(self.page_finder) 
    784790         
    785791         
    … …  
    940946        err_image[err_image==0]= 1 
    941947        theory= Data2D(image= image , err_image= err_image) 
     948        theory.name= model.name 
    942949         
    943950        if data ==None: 
    944951            self._fill_default_model2D(theory= theory, qmax=qmax,qstep=qstep, qmin= qmin) 
    945             theory.name= model.name 
     952         
    946953        else: 
    947             theory.name= data.name 
    948954            theory.id= "Model" 
    949955            theory.group_id= "Model"+data.name 

  • sansview/perspectives/fitting/simfitpage.py

    r925a30e r2140e68  
    11 
    2 import sys,re,string, wx    
     2import sys,re,string, wx   
     3import wx.lib.newevent  
    34from sans.guicomm.events import StatusEvent     
     5 
     6class FitConstraint: 
     7    """ 
     8        Contains info on selected model to fit  
     9        via simultaneous page 
     10    """ 
     11    def __init__(self, model, page=None): 
     12        """ 
     13            initialization 
     14        """ 
     15        self.model= model 
     16        self.page = page 
     17        self.fittable_param =[] 
     18        self.selected_params=[] 
     19        self._set_fittableParam() 
     20         
     21    def _set_fittableParam(self): 
     22        """ 
     23            fill self.fittable_param 
     24        """ 
     25        for item in self.model.getParamList(): 
     26            if not item  in self.model.getDispParamList(): 
     27                self.fittable_param.append(item) 
     28        for item in self.model.fixed: 
     29            self.fittable_param.append(item) 
     30       
    431 
    532class SimultaneousFitPage(wx.ScrolledWindow): 
    … …  
    1542     
    1643     
    17     def __init__(self, parent, *args, **kwargs): 
     44    def __init__(self, parent,page_finder ={}, *args, **kwargs): 
    1845        wx.ScrolledWindow.__init__(self, parent, *args, **kwargs) 
    1946        """ 
    … …  
    2148        """ 
    2249        self.parent = parent 
    23         self.page_finder={} 
    24         self.sizer3 = wx.GridBagSizer(5,5) 
    25         self.sizer1 = wx.GridBagSizer(5,5) 
    26         self.sizer2  = wx.BoxSizer(wx.HORIZONTAL) 
     50        ## store page_finder 
     51        self.page_finder=page_finder 
     52        ## list contaning info to set constraint  
     53        ## look like self.constraint_dict[page]=[ model_name, parameter_name, constraint(string)] 
     54        self.constraint_dict={} 
     55        ## item list  self.constraints_list=[combobox1, combobox2,=,textcrtl, button ] 
     56        self.constraints_list=[] 
     57        ## list of current model  
     58        self.model_list=[] 
     59        ## selected mdoel to fit 
     60        self.model_toFit=[] 
     61         
     62         
     63        ## draw 
     64        self.define_page_structure() 
     65        self.draw_page(self.page_finder) 
     66         
     67        self.set_layout() 
     68         
     69        
     70         
     71    def define_page_structure(self): 
     72        """ 
     73            Create empty sizer for a panel 
     74        """ 
    2775        self.vbox  = wx.BoxSizer(wx.VERTICAL) 
    28         self.vbox.Add(self.sizer3) 
     76        self.sizer1 = wx.BoxSizer(wx.VERTICAL) 
     77        self.sizer2 = wx.BoxSizer(wx.VERTICAL) 
     78       
    2979        self.vbox.Add(self.sizer1) 
    3080        self.vbox.Add(self.sizer2) 
    31         id = wx.NewId() 
    32         self.btFit =wx.Button(self,id,'Constraint Fit') 
    33         self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id) 
    34         self.btFit.SetToolTipString("Perform fit.") 
    35         ix = 0 
    36         iy = 1  
    37         self.cb1 = wx.CheckBox(self, -1,'Models', (10, 10)) 
    38         self.sizer3.Add(self.cb1,(iy, ix),(1,1),\ 
    39                         wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
    40         wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_model_name) 
    41        
    42         text=wx.StaticText(self, -1, 'Constraint') 
    43         self.sizer2.Add(text,0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10) 
    44  
    45         self.ctl2 = wx.TextCtrl(self, -1, style=wx.TE_MULTILINE) 
    46         
    47         self.sizer2.Add(self.ctl2, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10) 
    48         
    49         self.sizer2.Add(self.btFit, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10) 
    50         self.params=[] 
    51         self.model_list=[] 
    52         self.model_toFit=[] 
    53          
     81         
     82    def set_scroll(self): 
     83        self.SetScrollbars(20,20,200,100) 
     84        self.Layout()   
     85          
     86    def set_layout(self): 
     87        """ 
     88             layout 
     89        """ 
    5490        self.vbox.Layout() 
    5591        self.vbox.Fit(self)  
    5692        self.SetSizer(self.vbox) 
    57         self.SetScrollbars(20,20,55,40) 
     93        
     94        self.set_scroll() 
    5895        self.Centre() 
    59          
    6096         
    6197    def onFit(self,event): 
    6298        """ signal for fitting""" 
    63         for page in self.page_finder.iterkeys(): 
    64             page.set_model_parameter() 
     99        ## making sure all parameters content a constraint 
     100        ## validity of the constraint expression is own by fit engine 
     101        self._set_constraint() 
     102        ## get the fit range of very fit problem         
     103        for page, value in self.page_finder.iteritems(): 
     104            qmin, qmax= page.get_range() 
     105            value.set_range(qmin, qmax) 
     106        ## model was actually selected from this page to be fit 
    65107        if len(self.model_toFit) >= 1 : 
    66             self._onTextEnter() 
    67             self.set_model() 
    68             self.manager._on_simul_fit() 
     108            self.manager.on_simul_fit() 
    69109        else: 
    70             wx.PostEvent(self.parent.Parent, StatusEvent(status=\ 
    71                             "Select at least one model to fit ")) 
     110            msg= "Select at least one model to fit " 
     111            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
    72112            
    73113             
    … …  
    79119        """ 
    80120        self.manager = manager 
    81         self.add_model( self.manager.page_finder) 
    82          
    83          
    84     def select_all_model_name(self,event): 
     121        self.draw_page( self.manager.page_finder) 
     122         
     123         
     124    def check_all_model_name(self,event): 
    85125        """ 
    86126            check all models names 
    … …  
    92132                self.manager.schedule_for_fit( value=1,fitproblem =item[1])  
    93133                self.model_toFit.append(item) 
     134                 
     135            ## constraint info 
     136            self._store_model() 
     137            ## display constraint fields 
     138            if self.show_constraint.GetValue(): 
     139                self.sizer_couples.Clear(True)  
     140                self._show_constraint() 
     141                return 
    94142        else: 
    95143            for item in self.model_list: 
    … …  
    98146                 
    99147            self.model_toFit=[] 
    100         
    101              
    102     def add_model(self,page_finder): 
    103         """ 
    104             Receive a dictionary containing information to display model name 
    105             @param page_finder: the dictionary containing models information 
    106         """ 
    107         if len(self.model_list)>0: 
    108             for item in self.model_list: 
    109                 item[0].SetValue(False)  
    110                 self.manager.schedule_for_fit( value=0,fitproblem =item[1]) 
    111         self.model_list=[] 
    112         self.model_toFit=[] 
    113         self.sizer1.Clear(True) 
    114         self.page_finder=page_finder 
    115         self.cb1.SetValue(False) 
    116         ix = 0 
    117         iy = 1  
    118         list=[] 
    119         for page, value in page_finder.iteritems(): 
    120             try: 
    121                 list = value.get_model() 
    122                 model = list[0] 
    123                 modelname = list[1] 
    124                 cb = wx.CheckBox(self, -1, modelname, (10, 10)) 
    125                 cb.SetValue(False) 
    126                 self.sizer1.Add( cb,( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
    127                 ix = 0 
    128                 iy += 1  
    129                 wx.EVT_CHECKBOX(self, cb.GetId(), self.select_model_name) 
    130                 self.model_list.append([cb,value,page,modelname]) 
    131             except: 
    132                 pass 
    133         iy +=1 
    134         self.sizer1.Add((20,20),( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
    135         self.sizer1.Layout()         
    136         self.vbox.Layout() 
    137          
    138          
    139     def remove_model(self,delpage): 
    140         """ 
    141              Remove  a checkbox and the name related to a model selected on page delpage 
    142              @param delpage: the page removed 
    143         """ 
    144         self.model_list=[] 
    145         self.model_toFit=[] 
    146         self.sizer1.Clear(True) 
    147         
    148         self.cb1.SetValue(False) 
    149         ix = 0 
    150         iy = 1  
    151         list=[] 
    152         for page, value in self.page_finder.iteritems(): 
    153             try: 
    154                 if page!= delpage: 
    155                     list = value.get_model() 
    156                     model = list[0] 
    157                     modelname = list[1] 
    158                     cb = wx.CheckBox(self, -1, modelname, (10, 10)) 
    159                     cb.SetValue(False) 
    160                     self.sizer1.Add( cb,( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
    161                     ix = 0 
    162                     iy += 1  
    163                     wx.EVT_CHECKBOX(self, cb.GetId(), self.select_model_name) 
    164                     self.model_list.append([cb,value,page,modelname]) 
    165             except: 
    166                 pass 
    167         iy +=1 
    168         self.sizer1.Add((20,20),( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
    169         self.sizer1.Layout()         
    170         self.vbox.Layout() 
    171          
    172          
    173     def select_model_name(self,event): 
    174         """ 
    175             Save interformation related to checkbox and their states 
     148            ##constraint info 
     149            self._hide_constraint() 
     150         
     151   
     152    def check_model_name(self,event): 
     153        """ 
     154            Save information related to checkbox and their states 
    176155        """ 
    177156        self.model_toFit=[] 
    … …  
    185164                    self.model_toFit.remove(item) 
    186165                    self.cb1.SetValue(False) 
    187                      
     166        ## display constraint fields 
     167        if len(self.model_toFit)>=2: 
     168            self._store_model() 
     169            if self.show_constraint.GetValue(): 
     170                self._show_constraint() 
     171                           
     172        
     173        ## set the value of the main check button           
    188174        if len(self.model_list)==len(self.model_toFit): 
    189175            self.cb1.SetValue(True) 
     176            ## constraint info 
     177            self._store_model() 
     178            ## display constraint fields 
     179            if self.show_constraint.GetValue(): 
     180                self.sizer_couples.Clear(True)  
     181                self._show_constraint() 
     182                return 
    190183        else: 
    191184            self.cb1.SetValue(False) 
    192       
    193          
     185            ##constraint info 
     186            self._hide_constraint() 
     187         
     188   
     189    def draw_page(self, page_finder={}):       
     190        """ 
     191            Draw a sizer containing couples of data and model  
     192        """    
     193        ## receive a new page_finder   
     194        self.page_finder = page_finder   
     195        
     196        self.model_list=[] 
     197        self.model_toFit=[] 
     198        if len(self.model_list)>0: 
     199            for item in self.model_list: 
     200                item[0].SetValue(False)  
     201                self.manager.schedule_for_fit( value=0,fitproblem =item[1]) 
     202                 
     203        self.sizer1.Clear(True) 
     204         
     205                 
     206        box_description= wx.StaticBox(self, -1,"Fit Couples") 
     207        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL) 
     208        sizer_title = wx.BoxSizer(wx.HORIZONTAL) 
     209        sizer_couples = wx.GridBagSizer(5,5) 
     210         
     211        #------------------------------------------------------ 
     212        if len(self.page_finder)==0: 
     213            sizer_title.Add(wx.StaticText(self,-1," No fit couple available !")) 
     214        else: 
     215            ## store model   
     216            self._store_model() 
     217         
     218            self.cb1 = wx.CheckBox(self, -1,'Select all Fit Couples') 
     219            self.cb1.SetValue(False) 
     220            wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.check_all_model_name) 
     221             
     222            sizer_title.Add((15,15)) 
     223            sizer_title.Add(self.cb1) 
     224            ## draw list of model name 
     225            self._fill_sizer_model_list(sizer_couples) 
     226            ## draw the sizer containing constraint info 
     227            self._fill_sizer_constraint() 
     228        #-------------------------------------------------------- 
     229        boxsizer1.Add((5, 5))   
     230        boxsizer1.Add(sizer_title) 
     231        boxsizer1.Add((5, 5))   
     232        boxsizer1.Add(sizer_couples) 
     233        boxsizer1.Add((5,5)) 
     234        
     235        self.sizer1.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10) 
     236        self.sizer1.Layout() 
     237        self.SetScrollbars(20,20,200,100) 
     238        self.AdjustScrollbars() 
     239         
     240    def _store_model(self): 
     241        """ 
     242            Store selected model 
     243        """ 
     244        if len(self.model_toFit) < 2: 
     245            return 
     246        for page, value in self.page_finder.iteritems(): 
     247            model = value.get_model() 
     248            for item in self.model_toFit: 
     249                if model in item and not model in self.constraint_dict.keys(): 
     250                    self.constraint_dict[model]=FitConstraint(model=model, page=page) 
     251                    
     252         
     253    def _display_constraint(self, event): 
     254        """ 
     255            Show fields to add constraint 
     256        """ 
     257        if len(self.model_toFit)< 2: 
     258            msg= "Select at least 2 models to add constraint " 
     259            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     260            ## hide button 
     261            self._hide_constraint() 
     262            return 
     263        if self.show_constraint.GetValue(): 
     264            self.sizer_couples.Clear(True)  
     265            self._show_constraint() 
     266            return 
     267        else: 
     268           self._hide_constraint() 
     269           return  
     270             
    194271    
    195     def set_model(self): 
    196         """ 
    197             set_model take values in self.params which are the values 
    198             entered by the user and try to assign them into the model  
    199             concerned in self.manager. page.finder 
    200         """ 
    201         if len(self.params) >0: 
    202             for item in self.model_toFit: 
    203                 list=item[1].get_model() 
    204                 model=list[0] 
    205                 param_list=model.getParamList() 
    206                 if self.params !=[]: 
    207                     for element in self.params: 
    208                         if model.name == str(element[0]): 
    209                             for item in param_list: 
    210                                 if item==str(element[1]): 
    211                                     self.manager.set_page_finder(model.name,element[1], 
    212                                                                  str(element[2])) 
    213                             
    214      
    215     def _onTextEnter(self): 
     272    def _show_constraint(self): 
     273        """ 
     274            Show constraint fields 
     275        """ 
     276        flag = False 
     277        if len(self.constraints_list)!= 0: 
     278            nb_fit_param = 0 
     279            for value in self.constraint_dict.values(): 
     280                nb_fit_param += len(value.fittable_param) 
     281            ##Don't add anymore 
     282            if len(self.constraints_list) == nb_fit_param: 
     283                flag = True 
     284        if len(self.model_toFit) < 2 or flag: 
     285            msg= "Select at least 2 model to add constraint " 
     286            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     287            self.sizer_couples.Layout() 
     288            self.sizer2.Layout() 
     289            self.SetScrollbars(20,20,200,100) 
     290            return 
     291             
     292        sizer_constraint =  wx.BoxSizer(wx.HORIZONTAL) 
     293         
     294        model_cbox = wx.ComboBox(self, -1) 
     295        model_cbox.Clear() 
     296 
     297        param_cbox = wx.ComboBox(self, -1) 
     298        wx.EVT_COMBOBOX(param_cbox,-1, self._on_select_param) 
     299        param_cbox.Hide() 
     300         
     301        ctl2 = wx.TextCtrl(self, -1) 
     302        ctl2.Bind(wx.EVT_TEXT_ENTER,self._onTextEnter) 
     303        ctl2.Hide() 
     304         
     305        egal_txt= wx.StaticText(self,-1," = ") 
     306        egal_txt.Hide() 
     307        
     308         
     309        for model, value in self.constraint_dict.iteritems(): 
     310            ## check if all parameters have been selected for constraint 
     311            ## then do not allow add constraint on parameters 
     312            if len(value.fittable_param)!= len(value.selected_params): 
     313                    ## the user can apply constraint only on model select to fit 
     314                    model_cbox.Append( str(model.name), model) 
     315             
     316            
     317        wx.EVT_COMBOBOX(model_cbox,-1, self._on_select_model) 
     318         
     319        ##[combobox1, combobox2,=,textcrtl ] 
     320        self.constraints_list.append([model_cbox, param_cbox, egal_txt, ctl2]) 
     321         
     322        sizer_constraint.Add(model_cbox) 
     323        sizer_constraint.Add((10,10)) 
     324        sizer_constraint.Add(param_cbox) 
     325        sizer_constraint.Add(egal_txt) 
     326        sizer_constraint.Add(ctl2) 
     327        sizer_constraint.Add((10,10)) 
     328         
     329         
     330        self.sizer_couples.Add(sizer_constraint) 
     331        self.sizer_couples.Add((10,10)) 
     332        self.sizer_couples.Layout() 
     333        self.sizer2.Layout() 
     334        self.SetScrollbars(20,20,200,100) 
     335         
     336    def _hide_constraint(self):  
     337        """ 
     338            hide buttons related constraint  
     339        """ 
     340        if len(self.constraint_dict)>0: 
     341            for value in self.constraint_dict.itervalues(): 
     342                value.selected_params=[] 
     343                param = value.fittable_param## list of parameter name 
     344                ## reset the constraint to None on fitproblem 
     345                for item in param: 
     346                    self.page_finder[value.page].set_model_param(item,None) 
     347                    
     348        self.btAdd.Hide() 
     349        self.constraints_list=[]          
     350        self.sizer_couples.Clear(True)  
     351        self.sizer_couples.Layout()     
     352        self.sizer2.Layout() 
     353        self.SetScrollbars(20,20,200,100) 
     354        self.AdjustScrollbars()     
     355                 
     356         
     357         
     358    def _on_select_model(self, event): 
     359        """ 
     360         fill combox box with list of parameters 
     361        """ 
     362        model = event.GetClientData() 
     363        param_list= self.constraint_dict[model].fittable_param 
     364        length = len(self.constraints_list) 
     365        if length < 1: 
     366            return  
     367         
     368        param_cbox = self.constraints_list[length-1][1] 
     369        param_cbox.Clear() 
     370        ## insert only fittable paramaters 
     371        for param in param_list: 
     372            if not param in self.constraint_dict[model].selected_params: 
     373                param_cbox.Append( str(param), model) 
     374        param_cbox.Show(True) 
     375        
     376        
     377        self.sizer2.Layout() 
     378        self.SetScrollbars(20,20,200,100) 
     379         
     380         
     381    def _on_select_param(self, event): 
     382        """ 
     383            Store the appropriate constraint in the page_finder 
     384        """ 
     385        model = event.GetClientData() 
     386        param = event.GetString() 
     387        self.constraint_dict[model].selected_params.append(param) 
     388         
     389        length = len(self.constraints_list) 
     390        if length < 1: 
     391            return  
     392        egal_txt = self.constraints_list[length-1][2] 
     393        egal_txt.Show(True)        
     394         
     395        ctl2 = self.constraints_list[length-1][3] 
     396        ctl2.Show(True) 
     397         
     398        self.btAdd.Show(True) 
     399        self.sizer2.Layout() 
     400        self.SetScrollbars(20,20,200,100) 
     401         
     402         
     403    def _onAdd_constraint(self, event):   
     404        """ 
     405            Add another line for constraint 
     406        """ 
     407        msg= " " 
     408        wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     409         
     410        ## check that a constraint is added before allow to add another cosntraint 
     411         
     412        for item in self.constraints_list: 
     413            model_cbox = item[0] 
     414            if model_cbox.GetString(0)=="": 
     415                msg= " Select a model Name! " 
     416                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     417                return  
     418            param_cbox = item[1] 
     419            if param_cbox.GetString(0)=="": 
     420                msg= " Select a parameter Name! " 
     421                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     422                return  
     423            ctl2 = item[3] 
     424            if ctl2.GetValue().lstrip().rstrip()=="": 
     425                 msg= " Enter a constraint for %s.%s! "%(model_cbox.GetString(0), 
     426                                              param_cbox.GetString(0))            
     427                 wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     428                 return  
     429        
     430        ## some model or parameters can be constrained 
     431        self._show_constraint() 
     432         
     433         
     434    
     435        
     436         
     437    def _fill_sizer_constraint(self): 
     438        """ 
     439            Fill sizer containing constraint info 
     440        """ 
     441        msg= "Select at least 2 model to add constraint " 
     442        wx.PostEvent(self.parent.Parent, StatusEvent(status= msg )) 
     443         
     444        self.sizer2.Clear(True) 
     445  
     446        box_description= wx.StaticBox(self, -1,"Fit Constraints") 
     447        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL) 
     448        sizer_title = wx.BoxSizer(wx.HORIZONTAL) 
     449        self.sizer_couples = wx.BoxSizer(wx.VERTICAL) 
     450        sizer_button = wx.BoxSizer(wx.HORIZONTAL) 
     451         
     452        self.hide_constraint = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP) 
     453        self.show_constraint = wx.RadioButton(self, -1, 'Yes', (10, 30)) 
     454         
     455        self.Bind( wx.EVT_RADIOBUTTON, self._display_constraint, 
     456                    id= self.hide_constraint.GetId() ) 
     457         
     458        self.Bind(  wx.EVT_RADIOBUTTON, self._display_constraint, 
     459                         id= self.show_constraint.GetId()    ) 
     460         
     461           
     462        
     463         
     464        sizer_title.Add( wx.StaticText(self,-1," Model") ) 
     465        sizer_title.Add(( 10,10) ) 
     466        sizer_title.Add( wx.StaticText(self,-1," Parameter") ) 
     467        sizer_title.Add(( 10,10) ) 
     468        sizer_title.Add( wx.StaticText(self,-1," Add Constraint?") ) 
     469        sizer_title.Add(( 10,10) ) 
     470        sizer_title.Add( self.show_constraint ) 
     471        sizer_title.Add( self.hide_constraint ) 
     472        sizer_title.Add(( 10,10) ) 
     473        
     474         
     475         
     476        ## Draw combobox box related to model name and model parameters 
     477        if self.show_constraint.GetValue(): 
     478            self._fill_sizer_constraint_helper(self.sizer_couples) 
     479        
     480        self.btAdd =wx.Button(self,wx.NewId(),'Add') 
     481        self.btAdd.Bind(wx.EVT_BUTTON, self._onAdd_constraint,id= self.btAdd.GetId()) 
     482        self.btAdd.SetToolTipString("Add another constraint?") 
     483        self.btAdd.Hide() 
     484         
     485        self.btFit = wx.Button(self,wx.NewId(),'Fit') 
     486        self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id= self.btFit.GetId()) 
     487        self.btFit.SetToolTipString("Perform fit.") 
     488         
     489        text_hint = wx.StaticText(self,-1,"Example: M0.paramter = M1.parameter")  
     490        sizer_button.Add(text_hint, 0 , wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
     491        sizer_button.Add(self.btAdd, 0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
     492        sizer_button.Add(self.btFit, 0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
     493        
     494        boxsizer1.Add(sizer_title) 
     495        boxsizer1.Add((10,10)) 
     496        boxsizer1.Add(self.sizer_couples) 
     497        boxsizer1.Add((10,10)) 
     498        boxsizer1.Add(sizer_button) 
     499         
     500        self.sizer2.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10) 
     501        self.sizer2.Layout() 
     502        self.SetScrollbars(20,20,200,100) 
     503         
     504         
     505         
     506    def _onTextEnter(self, event): 
     507        """ 
     508            callback function for textcrtl 
     509        """ 
     510        self.btAdd.Show(True) 
     511        self.sizer2.Layout() 
     512        self.SetScrollbars(20,20,200,100) 
     513        self._set_constraint() 
     514         
     515    def _set_constraint(self): 
    216516        """ 
    217517            get values from the constrainst textcrtl ,parses them into model name 
    … …  
    220520            uses it to reset the appropriate model and its appropriates parameters 
    221521        """ 
    222         value= self.ctl2.GetValue() 
    223         if value: 
    224             self.params=[] 
     522        for item in self.constraints_list: 
     523            model = item[0].GetClientData(0) 
     524            param = item[1].GetString(0) 
     525            constraint = item[3].GetValue().lstrip().rstrip() 
     526            if model  in self.constraint_dict.keys(): 
     527                page = self.constraint_dict[model].page 
     528                if constraint != "": 
     529                    self.page_finder[page].set_model_param(param,constraint) 
     530         
     531    
     532               
     533    def _fill_sizer_model_list(self,sizer): 
     534        """ 
     535            Receive a dictionary containing information to display model name 
     536            @param page_finder: the dictionary containing models information 
     537        """ 
     538        ix = 0 
     539        iy = 0 
     540        list=[] 
     541        sizer.Clear(True) 
     542         
     543        new_name = wx.StaticText(self, -1, 'New Model Name', style=wx.ALIGN_CENTER) 
     544        new_name.SetBackgroundColour('orange') 
     545        sizer.Add(new_name,(iy, ix),(1,1), 
     546                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
     547         
     548        ix +=2  
     549        model_type = wx.StaticText(self, -1, '  Model Type') 
     550        model_type.SetBackgroundColour('grey') 
     551        sizer.Add(model_type,(iy, ix),(1,1), 
     552                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)  
     553        ix +=1  
     554        data_used = wx.StaticText(self, -1, '  Used Data') 
     555        data_used.SetBackgroundColour('grey') 
     556        sizer.Add(data_used,(iy, ix),(1,1), 
     557                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)  
     558         
     559        for page, value in self.page_finder.iteritems(): 
    225560            try: 
    226                 expression='[\s,;]' 
    227                 if re.search(expression,value) !=None: 
    228                     word=re.split(expression,value) 
    229                     for item in word: 
    230                         self.params.append(self.parser_helper(item)) 
    231                 else: 
    232                     self.params.append(self.parser_helper(value))   
     561                ix = 0 
     562                iy += 1  
     563                model = value.get_model() 
     564                cb = wx.CheckBox(self, -1, str(model.name)) 
     565                cb.SetValue(False) 
     566                sizer.Add( cb,( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
     567                wx.EVT_CHECKBOX(self, cb.GetId(), self.check_model_name) 
     568                 
     569                
     570                ix +=2  
     571                type = model.__class__.__name__ 
     572                model_type = wx.StaticText(self, -1, str(type)) 
     573                sizer.Add(model_type,( iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
     574                 
     575                ix +=1  
     576                data = value.get_fit_data() 
     577                data_used= wx.StaticText(self, -1, str(data.name)) 
     578                sizer.Add(data_used,( iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
     579                 
     580                 
     581                self.model_list.append([cb,value,page,model]) 
     582                 
    233583            except: 
    234                 wx.PostEvent(self.parent.Parent, StatusEvent(status="Constraint Error: %s" % sys.exc_value)) 
    235        
    236          
    237     def parser_helper(self,value): 
    238         """ 
    239              @return  param:a list containing the name of a model ,its parameters  
    240              value and name extracted from the constrainst controlbox 
    241         """ 
    242         if string.find(value, "=") !=-1: 
    243             model_param= re.split("=",value) 
    244             param_name=model_param[0] 
    245             param_value=model_param[1] 
    246              
    247             if string.find(param_name,".")!=-1: 
    248                 param_names= re.split("\.",param_name) 
    249                 model_name=param_names[0] 
    250                 param_name=param_names[1] 
    251                 param=[str(model_name),param_name,str(param_value)] 
    252                 return param 
    253             else: 
    254                 wx.PostEvent(self.parent.Parent, 
    255                               StatusEvent(status="cannot evaluate this expression")) 
    256                 return 
    257         else: 
    258             wx.PostEvent(self.parent.Parent, StatusEvent(status="Missing '=' in expression")) 
    259              
     584                pass 
     585        iy +=1 
     586        sizer.Add((20,20),( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
     587        sizer.Layout()     
     588         
    260589    
     590   
     591         
     592class HelpWindow(wx.Frame): 
     593    def __init__(self, parent, id, title): 
     594        wx.Frame.__init__(self, parent, id, title, size=(570, 400)) 
     595         
     596        page_finder ={} 
     597        ## create random data 
     598        from danse.common.plottools.plottables import Data1D 
     599        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1]) 
     600        data.name="mydata.txt" 
     601        ## create model 
     602        from sans.models.CylinderModel import CylinderModel 
     603        model = CylinderModel() 
     604        model.name="M0" 
     605         
     606         
     607        from fitproblem import FitProblem 
     608        page_finder["page"]= FitProblem() 
     609        ## fill the page_finder 
     610        page_finder["page"].add_fit_data(data) 
     611        page_finder["page"].set_model(model) 
     612        self.page = SimultaneousFitPage(self, page_finder=page_finder)  
     613         
     614         
     615         
     616        self.Centre() 
     617        self.Show(True) 
     618 
     619 
     620    
     621if __name__=="__main__": 
     622    app = wx.App() 
     623    HelpWindow(None, -1, 'HelpWindow') 
     624    app.MainLoop() 
    261625     
Note: See TracChangeset for help on using the changeset viewer.