Changeset 2a8fac1 in sasview for sansview/perspectives


Ignore:
Timestamp:
Feb 23, 2009 5:27:12 PM (15 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:
18eba35
Parents:
700f9b4
Message:

refactor add et and remove error

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

Legend:

Unmodified
Added
Removed

  • sansview/perspectives/fitting/fitpanel.py

    ref8b580 r2a8fac1  
    4040        #Creating a page for simultaneous fitting 
    4141        from simfitpage import SimultaneousFitPage 
    42         self.sim_page = SimultaneousFitPage(self, -1) 
     42        self.sim_page = SimultaneousFitPage(self, id=-1) 
    4343        self.AddPage(self.sim_page,"Simultaneous Fit") 
    4444         
    45         self._mgr = wx.aui.AuiManager(self) 
     45        self._mgr = self.GetAuiManager() 
    4646 
    4747         
    … …  
    6060        self.Update() 
    6161        self.Center() 
     62         
     63         
    6264    def onClosePage(self, event): 
    6365        self.ToggleWindowStyle(wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB) 
    … …  
    102104        except: 
    103105            name = 'Fit' 
    104          
     106        
    105107        if self.fit_page_name != name: 
    106108            #self.about_page.Disable() 

  • sansview/perspectives/fitting/fitproblem.py

    r9e27de9 r2a8fac1  
    1616            and its name assign example [lineModel, M0] 
    1717        """ 
    18         self.data=None 
     18        self.fit_data=None 
    1919        self.theory_name=None 
    2020        self.model_list=[] 
    … …  
    2323        self.name_per_page=None 
    2424        self.smearer= None 
     25        self.plotted_data=None 
    2526         
    2627    def set_smearer(self, smearer): 
    … …  
    4647 
    4748   
    48     def add_data(self,data): 
     49    def add_plotted_data(self,data): 
    4950        """  
    5051            save a copy of the data select to fit 
    5152            @param data: data selected 
    5253        """ 
    53         self.data = data 
     54        self.plotted_data = data 
     55    def add_fit_data(self,data): 
     56        """  
     57            save a copy of the data select to fit 
     58            @param data: data selected 
     59        """ 
     60        self.fit_data = data 
    5461             
    5562    def get_model(self): 
    … …  
    5865        return self.model_list 
    5966      
    60     def get_data(self): 
     67    def get_plotted_data(self): 
    6168        """ @return:  list of data dList""" 
    62         return self.data 
    63        
    64        
     69        return self.plotted_data 
     70     
     71     
     72    def get_fit_data(self): 
     73        return self.fit_data 
     74     
     75     
    6576    def get_theory(self): 
    6677        """ @return the name of theory for plotting purpose""" 

  • sansview/perspectives/fitting/fitting.py

    rb5c537f r2a8fac1  
    77from danse.common.plottools.PlotPanel import PlotPanel 
    88from sans.guicomm.events import NewPlotEvent, StatusEvent   
    9 from sans.guicomm.events import EVT_SLICER_PANEL,EVT_MODEL2D_PANEL 
     9from sans.guicomm.events import EVT_SLICER_PANEL,EVT_MODEL2D_PANEL,ERR_DATA 
    1010 
    1111from sans.fit.AbstractFitEngine import Model,Data,FitData1D,FitData2D 
    … …  
    4949        self._fit_engine = 'scipy' 
    5050        self.enable_model2D=False 
    51          
     51        # list of selcted data 
     52        self.selected_data_list=[] 
    5253        # Log startup 
    5354        logging.info("Fitting plug-in started")    
    5455        # model 2D view 
    5556        self.model2D_id=None 
    56  
     57        self.err_dy={} 
     58         
     59    def _on_data_error(self, event): 
     60        self.err_dy= event.err_dy 
     61         
    5762    def populate_menu(self, id, owner): 
    5863        """ 
    … …  
    117122                          "Dialog with fitting parameters ", self._onSelect]]  
    118123            else: 
    119                 if item.name==graph.selected_plottable and\ 
    120                  item.__class__.__name__ is  "Data1D": 
     124                #if item.name==graph.selected_plottable and\ 
     125                # item.__class__.__name__ is  "Data1D": 
     126                if item.name==graph.selected_plottable : 
    121127                    return [["Select data  for Fitting", \ 
    122128                             "Dialog with fitting parameters ", self._onSelect]]  
    … …  
    142148        self.index_theory= 0 
    143149        self.parent.Bind(EVT_SLICER_PANEL, self._on_slicer_event) 
    144         
     150        self.parent.Bind( ERR_DATA, self._on_data_error) 
    145151         
    146152        #create the fitting panel 
    … …  
    202208        self.panel = event.GetEventObject() 
    203209        for item in self.panel.graph.plottables: 
     210            if item.name == self.panel.graph.selected_plottable: 
     211                if len(self.err_dy)>0: 
     212                    if item.name in  self.err_dy.iterkeys(): 
     213                        dy= self.err_dy[item.name] 
     214                        data= Data1D(x=item.x, y=item.y, dy=dy) 
     215                        data.name=item.name 
     216                else: 
     217                    if item.dy==None: 
     218                        dy= numpy.zeros(len(item.y)) 
     219                        dy[dy==0]=1 
     220                        print "dy", dy 
     221                        data= Data1D(x=item.x, y=item.y, dy=dy) 
     222                        data.name=item.name 
     223                    else: 
     224                        data= Data1D(x=item.x, y=item.y, dy=item.dy) 
     225                        data.name=item.name 
    204226            if item.name == self.panel.graph.selected_plottable or\ 
    205227                 item.__class__.__name__ is "Data2D": 
    206228                #find a name for the page created for notebook 
     229                print "fitting", self.panel 
    207230                try: 
    208231                    
    209                     page = self.fit_panel.add_fit_page(item) 
     232                    #page = self.fit_panel.add_fit_page(item) 
     233                    page = self.fit_panel.add_fit_page(data) 
    210234                    #page, model_name = self.fit_panel.add_fit_page(item) 
    211235                    # add data associated to the page created 
    … …  
    215239                        self.page_finder[page]= FitProblem() 
    216240                        #self.page_finder[page].save_model_name(model_name)   
    217                         self.page_finder[page].add_data(item) 
     241                        self.page_finder[page].add_plotted_data(item) 
     242                        self.page_finder[page].add_fit_data(data) 
     243                        
     244                             
    218245                        wx.PostEvent(self.parent, StatusEvent(status="Page Created")) 
    219246                    else: 
    220247                        wx.PostEvent(self.parent, StatusEvent(status="Page was already Created")) 
    221248                except: 
    222                     #raise 
    223                     wx.PostEvent(self.parent, StatusEvent(status="Creating Fit page: %s"\ 
    224                     %sys.exc_value)) 
     249                    raise 
     250                    #wx.PostEvent(self.parent, StatusEvent(status="Creating Fit page: %s"\ 
     251                    #%sys.exc_value)) 
     252                     
     253                     
    225254    def schedule_for_fit(self,value=0,fitproblem =None):   
    226255        """ 
    … …  
    400429        #    if  value.get_scheduled() ==1 : 
    401430            value = self.page_finder[current_pg] 
    402             metadata =  value.get_data() 
     431            metadata =  value.get_fit_data() 
    403432            list = value.get_model() 
    404433            model = list[0] 
    … …  
    477506            try: 
    478507                if value.get_scheduled()==1: 
    479                     metadata = value.get_data() 
     508                    metadata = value.get_fit_data() 
    480509                    list = value.get_model() 
    481510                    model= list[0] 
    … …  
    629658            for page in self.page_finder.iterkeys(): 
    630659                if  page==currpage :   
    631                     data=self.page_finder[page].get_data() 
     660                    data=self.page_finder[page].get_plotted_data() 
    632661                    list=self.page_finder[page].get_model() 
    633662                    model=list[0] 
    … …  
    766795            self.redraw_model(qmin,qmax) 
    767796            return  
     797        self._draw_model1D(model,name,model.description, enable1D,qmin,qmax, qstep) 
    768798        self._draw_model2D(model=model, 
    769799                           description=model.description, 
    … …  
    772802                           qmax=qmax, 
    773803                           qstep=qstep) 
    774         self._draw_model1D(model,name,model.description, enable1D,qmin,qmax, qstep) 
     804         
    775805               
    776806    def _draw_model1D(self,model,name,description=None, enable1D=True, 
    … …  
    803833            else: 
    804834                for i in range(xlen): 
    805                     y[i] = model.run(x[i]) 
    806                 #print x, y    
     835                    y[i] = model.run(x[i])  
    807836                try: 
    808837                    new_plot = Theory1D(x, y) 
    … …  
    828857        
    829858        wx.PostEvent(self.parent, StatusEvent(status="Calc \ 
    830         complete in %g sec" % elapsed, type="stop")) 
     859        complete !" , type="stop")) 
    831860        #print "complete",output, model,qmin, qmax 
    832861        data = output 
    … …  
    890919        theory.ymax= qmax#/math.sqrt(2) 
    891920         
    892         print "model draw comptele xmax",theory.xmax,model.name 
     921        #print "model draw comptele xmax",theory.xmax,model.name 
    893922        wx.PostEvent(self.parent, NewPlotEvent(plot=theory, 
    894923                         title="Analytical model 2D ", reset=True )) 
    … …  
    898927    def _draw_model2D(self,model,description=None, enable2D=False, 
    899928                      qmin=DEFAULT_QMIN, qmax=DEFAULT_QMAX, qstep=DEFAULT_NPTS): 
    900         
     929         
    901930        x=  numpy.linspace(start= -1*qmax, 
    902931                               stop= qmax, 
    … …  
    910939        lx = len(x) 
    911940        #print x 
    912         data=numpy.zeros([len(x),len(y)]) 
     941        #data=numpy.zeros([len(x),len(y)]) 
    913942        self.model= model 
    914943        if enable2D: 
    915             from sans.guiframe.model_thread import Calc2D 
    916             self.calc_thread = Calc2D(parent =self.parent,x=x, 
    917                                        y=y,model= self.model,  
    918                                        qmin=qmin, 
    919                                        qmax=qmax, 
    920                                        qstep=qstep, 
    921                             completefn=self.complete, 
    922                             updatefn=None) 
    923             self.calc_thread.queue() 
    924             self.calc_thread.ready(2.5) 
     944            try: 
     945                from sans.guiframe.model_thread import Calc2D 
     946                print "in draw model 2d ",self.model 
     947                self.calc_thread = Calc2D(parent =self.parent,x=x, 
     948                                           y=y,model= self.model,  
     949                                           qmin=qmin, 
     950                                           qmax=qmax, 
     951                                           qstep=qstep, 
     952                                completefn=self.complete, 
     953                                updatefn=None) 
     954                self.calc_thread.queue() 
     955                self.calc_thread.ready(2.5) 
     956            except: 
     957                raise 
    925958             
    926959    def show_panel2D(self, id=None ): 
Note: See TracChangeset for help on using the changeset viewer.