source: sasview/fittingview/src/sans/perspectives/fitting/fitting.py @ 743f480

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 743f480 was 743f480, checked in by Jae Cho <jhjcho@…>, 12 years ago

read theory too.

  • Property mode set to 100644
File size: 89.2 KB
RevLine 
[6d8bad4]1"""
2    Fitting perspective
3"""
[5062bbf]4################################################################################
5#This software was developed by the University of Tennessee as part of the
6#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
[f32d144]7#project funded by the US National Science Foundation.
[5062bbf]8#
9#See the license text in license.txt
10#
11#copyright 2009, University of Tennessee
12################################################################################
[a0da535]13import re
[5536035]14import sys
[056b569]15import os
[5536035]16import wx
17import logging
18import numpy
[c3b4dcb]19import time
[edcbd467]20from copy import deepcopy
[f83b94a]21import models
22
[c21e87b]23from sans.dataloader.loader import Loader
[c3b4dcb]24from sans.guiframe.dataFitting import Data2D
25from sans.guiframe.dataFitting import Data1D
[2dbffcc]26from sans.guiframe.dataFitting import check_data_validity
[6bbeacd4]27from sans.guiframe.events import NewPlotEvent
28from sans.guiframe.events import StatusEvent 
[a0da535]29from sans.guiframe.events import EVT_SLICER_PANEL
30from sans.guiframe.events import EVT_SLICER_PARS_UPDATE
[6bbeacd4]31from sans.guiframe.gui_style import GUIFRAME_ID
32from sans.guiframe.plugin_base import PluginBase
[e68bfef]33from sans.guiframe.data_processor import BatchCell
[2dbffcc]34from sans.fit.Fitting import Fit
[cc31608]35from .console import ConsoleUpdate
[2dbffcc]36from .fitproblem import FitProblemDictionary
[a0da535]37from .fitpanel import FitPanel
[cc31608]38from .fit_thread import FitThread
[a0da535]39from .pagestate import Reader
40from .fitpage import Chi2UpdateEvent
[6f140f2]41from sans.perspectives.calculator.model_editor import TextDialog
42from sans.perspectives.calculator.model_editor import EditorWindow
[ed2ea6a]43
[c26a64b]44MAX_NBR_DATA = 4
[2c6b224]45SANS_F_TOL = 5e-05
[5062bbf]46
[f32d144]47(PageInfoEvent, EVT_PAGE_INFO) = wx.lib.newevent.NewEvent()
[92a5ac92]48
[5062bbf]49
[940aca7]50if sys.platform == "win32":
[f6aee42]51    ON_MAC = False
52else:
[f32d144]53    ON_MAC = True
54
[3658abed]55
56class Plugin(PluginBase):
[d89f09b]57    """
[f32d144]58    Fitting plugin is used to perform fit
[d89f09b]59    """
[3658abed]60    def __init__(self, standalone=False):
[908b817]61        PluginBase.__init__(self, name="Fitting", standalone=standalone)
[d89f09b]62       
[ed2ea6a]63        #list of panel to send to guiframe
[f83b94a]64        self.mypanels = []
[ed2ea6a]65        # reference to the current running thread
[f83b94a]66        self.calc_2D = None
67        self.calc_1D = None
[9f4f8f4]68       
69        self.color_dict = {}
70       
[66ff250]71        self.fit_thread_list = {}
[2296316]72        self.residuals = None
[55bb249c]73        self.weight = None
[9466f2d6]74        self.fit_panel = None
[940aca7]75        self.plot_panel = None
[d89f09b]76        # Start with a good default
77        self.elapsed = 0.022
[ed2ea6a]78        # the type of optimizer selected, park or scipy
[f32d144]79        self.fitter = None
[2296316]80        self.fit_panel = None
[c660493]81        #let fit ready
82        self.fitproblem_count = None
[ed2ea6a]83        #Flag to let the plug-in know that it is running stand alone
[f83b94a]84        self.standalone = True
[f32d144]85        ## dictionary of page closed and id
[f83b94a]86        self.closed_page_dict = {}
[d89f09b]87        ## Fit engine
[e9e914f]88        self._fit_engine = 'scipy'
[7db52f1]89        self._gui_engine = None
[2296316]90        ## Relative error desired in the sum of squares (float); scipy only
[2c6b224]91        self.ftol = SANS_F_TOL
[7db52f1]92        self.batch_reset_flag = True
[ed2ea6a]93        #List of selected data
[f83b94a]94        self.selected_data_list = []
[c3435b7f]95        ## list of slicer panel created to display slicer parameters and results
[f83b94a]96        self.slicer_panels = []
[32d802f]97        # model 2D view
[f83b94a]98        self.model2D_id = None
[ed2ea6a]99        #keep reference of the simultaneous fit page
[f83b94a]100        self.sim_page = None
[fa02d95]101        self.sim_menu = None
102        self.batch_page = None
103        self.batch_menu = None
[ae4ade7]104        self.index_model = 0
[9f4f8f4]105        self.test_model_color = None
[f83b94a]106        #Create a reader for fit page's state
[f32d144]107        self.state_reader = None
[c8deee5]108        self._extensions = '.fitv'
[0b6ae5f]109        self.scipy_id = wx.NewId()
110        self.park_id = wx.NewId()
[dafc36f]111        self.menu1 = None
[6f140f2]112        self.new_model_frame = None
[0b6ae5f]113       
[4da35bc]114        self.temp_state = []
[ef16f59]115        self.state_index = 0
[b63dc6e]116        self.sfile_ext = None
[6bbeacd4]117        # take care of saving  data, model and page associated with each other
118        self.page_finder = {}
[f83b94a]119        # Log startup
[f32d144]120        logging.info("Fitting plug-in started")
[7360816]121        self.batch_capable = self.get_batch_capable()
122   
123    def get_batch_capable(self):
124        """
125        Check if the plugin has a batch capability
126        """
127        return True
[9f4f8f4]128   
[2dbffcc]129    def create_fit_problem(self, page_id):
130        """
131        Given an ID create a fitproblem container
132        """
133        self.page_finder[page_id] = FitProblemDictionary()
134       
135    def delete_fit_problem(self, page_id):
136        """
137        Given an ID create a fitproblem container
138        """
139        if page_id in self.page_finder.iterkeys():
140            del self.page_finder[page_id]
141       
[9f4f8f4]142    def add_color(self, color, id):
143        """
144        adds a color as a key with a plot id as its value to a dictionary
145        """
146        self.color_dict[id] = color
[2a8fac1]147       
[cc31608]148    def on_batch_selection(self, flag):
149        """
150        switch the the notebook of batch mode or not
151        """
152        self.batch_on = flag
[2dbffcc]153        if self.fit_panel is not None:
154            self.fit_panel.batch_on = self.batch_on
[cc31608]155       
[cab076b]156    def populate_menu(self, owner):
[d89f09b]157        """
[5062bbf]158        Create a menu for the Fitting plug-in
159       
160        :param id: id to create a menu
161        :param owner: owner of menu
162       
163        :return: list of information to populate the main menu
164       
[d89f09b]165        """
166        #Menu for fitting
167        self.menu1 = wx.Menu()
[0b6ae5f]168        id1 = wx.NewId()
169        simul_help = "Add new fit panel"
[f32d144]170        self.menu1.Append(id1, '&New Fit Page', simul_help)
[0b6ae5f]171        wx.EVT_MENU(owner, id1, self.on_add_new_page)
[6f023e8]172        self.menu1.AppendSeparator()
[dafc36f]173        self.id_simfit = wx.NewId()
[6bbeacd4]174        simul_help = "Simultaneous Fit"
[f32d144]175        self.menu1.Append(self.id_simfit, '&Simultaneous Fit', simul_help)
[dafc36f]176        wx.EVT_MENU(owner, self.id_simfit, self.on_add_sim_page)
[fa02d95]177        self.sim_menu = self.menu1.FindItemById(self.id_simfit)
[f32d144]178        self.sim_menu.Enable(False)
[fa02d95]179        #combined Batch
180        self.id_batchfit = wx.NewId()
[7b1ca97]181        batch_help = "Combined Batch"
[fa02d95]182        self.menu1.Append(self.id_batchfit, '&Combine Batch Fit', batch_help)
[f32d144]183        wx.EVT_MENU(owner, self.id_batchfit, self.on_add_sim_page)
[fa02d95]184        self.batch_menu = self.menu1.FindItemById(self.id_batchfit)
[f32d144]185        self.batch_menu.Enable(False)
[f7e9af2]186        self.menu1.AppendSeparator()
[0b6ae5f]187        #Set park engine
[fa02d95]188        scipy_help = "Scipy Engine: Perform Simple fit. More in Help window...."
[2c6b224]189        self.menu1.AppendCheckItem(self.scipy_id, "Simple FitEngine [LeastSq]",
[f32d144]190                                   scipy_help)
191        wx.EVT_MENU(owner, self.scipy_id, self._onset_engine_scipy)
[0b6ae5f]192       
193        park_help = "Park Engine: Perform Complex fit. More in Help window...."
[2c6b224]194        self.menu1.AppendCheckItem(self.park_id, "Complex FitEngine [ParkMC]",
[f32d144]195                                   park_help)
196        wx.EVT_MENU(owner, self.park_id, self._onset_engine_park)
[0b6ae5f]197       
198        self.menu1.FindItemById(self.scipy_id).Check(True)
199        self.menu1.FindItemById(self.park_id).Check(False)
[2c6b224]200        self.menu1.AppendSeparator()
201        self.id_tol = wx.NewId()
202        ftol_help = "Change the current FTolerance (=%s) " % str(self.ftol)
[f32d144]203        ftol_help += "of Simple FitEngine..."
204        self.menu1.Append(self.id_tol, "Change FTolerance",
205                                   ftol_help)
206        wx.EVT_MENU(owner, self.id_tol, self.show_ftol_dialog)
[da6fd1a]207        self.menu1.AppendSeparator()
[2c6b224]208       
[da6fd1a]209        self.id_reset_flag = wx.NewId()
210        resetf_help = "BatchFit: If checked, the initial param values will be "
[f32d144]211        resetf_help += "propagated from the previous results. "
[da6fd1a]212        resetf_help += "Otherwise, the same initial param values will be used "
[f32d144]213        resetf_help += "for all fittings."
214        self.menu1.AppendCheckItem(self.id_reset_flag,
215                                   "Chain Fitting [BatchFit Only]",
216                                   resetf_help)
217        wx.EVT_MENU(owner, self.id_reset_flag, self.on_reset_batch_flag)
[70e439f]218        chain_menu = self.menu1.FindItemById(self.id_reset_flag)
219        chain_menu.Check(not self.batch_reset_flag)
220        chain_menu.Enable(self.batch_on)
[056b569]221       
222        self.menu1.AppendSeparator()
[5d1c1f4]223        self.edit_model_menu = wx.Menu()
224        # Find and put files name in menu
225        try:
226            self.set_edit_menu(owner=owner)
227        except:
228            raise
[056b569]229       
[5d1c1f4]230        self.id_edit = wx.NewId()
[f32d144]231        editmodel_help = "Edit customized model sample file"
232        self.menu1.AppendMenu(self.id_edit, "Edit Custom Model",
[5d1c1f4]233                              self.edit_model_menu, editmodel_help)
[3b19ac9]234        #create  menubar items
[908b817]235        return [(self.menu1, self.sub_menu)]
[056b569]236   
237    def edit_custom_model(self, event):
238        """
239        Get the python editor panel
240        """
[5d1c1f4]241        id = event.GetId()
[7c8d3093]242        label = self.edit_menu.GetLabel(id)
[5d1c1f4]243        from sans.perspectives.calculator.pyconsole import PyConsole
244        filename = os.path.join(models.find_plugins_dir(), label)
[f32d144]245        frame = PyConsole(parent=self.parent, manager=self,
246                          panel=self.fit_panel,
247                          title='Advanced Custom Model Editor',
248                          filename=filename)
[056b569]249        self.put_icon(frame)
[f32d144]250        frame.Show(True)
[7c8d3093]251   
[b25caad]252    def delete_custom_model(self, event):
253        """
254        Delete custom model file
255        """
256        id = event.GetId()
257        label = self.delete_menu.GetLabel(id)
258        toks = os.path.splitext(label)
259        path = os.path.join(models.find_plugins_dir(), toks[0])
260        try:
261            for ext in ['.py', '.pyc']:
262                p_path = path + ext
263                os.remove(p_path)
264            self.update_custom_combo()
[19e614a]265            if os.path.isfile(p_path):
266                msg = "Sorry! Could not be able to delete the default "
267                msg += "custom model... \n"
268                msg += "Please manually remove the files (.py, .pyc) "
269                msg += "in the 'plugin_models' folder \n"
270                msg += "inside of the SansView application, "
271                msg += "and try it again."
272                wx.MessageBox(msg, 'Info')
273                #wx.PostEvent(self.parent, StatusEvent(status=msg, type='stop',
274                #                                      info='warning'))
275            else:
276                self.delete_menu.Delete(id)
277                for item in self.edit_menu.GetMenuItems():
278                    if item.GetLabel() == label:
279                        self.edit_menu.DeleteItem(item)
280                        msg = "The custom model, %s, has been deleted."% label
281                        wx.PostEvent(self.parent, StatusEvent(status=msg, 
282                                                type='stop', info='info'))
283                        break
[b25caad]284        except:
285            msg ='Delete Error: \nCould not delete the file; Check if in use.'
[19e614a]286            wx.MessageBox(msg, 'Error')
[b25caad]287   
[7c8d3093]288    def make_sum_model(self, event):
289        """
290        Edit summodel template and make one
291        """
292        id = event.GetId()
293        model_manager = models.ModelManager()
294        model_list = model_manager.get_model_name_list()
[796c4d4]295        plug_dir = models.find_plugins_dir()
[fdef956]296        textdial = TextDialog(None, self, -1, 'Easy Sum/Multi(p1, p2) Editor', 
[796c4d4]297                              model_list, plug_dir)
[7c8d3093]298        self.put_icon(textdial)
[f32d144]299        textdial.ShowModal()
[7c8d3093]300        textdial.Destroy()
[6f140f2]301   
302    def make_new_model(self, event):
303        """
[f32d144]304        Make new model
[6f140f2]305        """
306        if self.new_model_frame != None and self.new_model_frame.IsShown():
307            self.new_model_frame.Show(False)
308        else:
309            id = event.GetId()
310            dir_path = models.find_plugins_dir()
311            title = "New Custom Model Function"
[f32d144]312            self.new_model_frame = EditorWindow(parent=self, base=self,
[6f140f2]313                                                path=dir_path, title=title)
314            self.put_icon(self.new_model_frame)
315        self.new_model_frame.Show(True)
[7c8d3093]316
[ed8fa6a3]317    def update_custom_combo(self):
318        """
319        Update custom model list in the fitpage combo box
320        """
321        try:
[b25caad]322            # Update edit menus
323            self.set_edit_menu_helper(self.parent, self.edit_custom_model)
324            self.set_edit_menu_helper(self.parent, self.delete_custom_model)
[ed8fa6a3]325            temp = self.fit_panel.reset_pmodel_list()
326            if temp:
327                # Set the new custom model list for all fit pages
328                for uid, page in self.fit_panel.opened_pages.iteritems():
329                    if hasattr(page, "formfactorbox"):
330                        page.model_list_box = temp
[19e614a]331                        current_val = page.formfactorbox.GetLabel()
[b9dd680]332                        if page.plugin_rbutton.GetValue():
333                            #pos = page.formfactorbox.GetSelection()
334                            page._show_combox_helper()
335                            new_val = page.formfactorbox.GetLabel()
336                            if current_val != new_val and new_val != '':
337                                page.formfactorbox.SetLabel(new_val)
338                            else:
339                                page.formfactorbox.SetLabel(current_val)
[ed8fa6a3]340        except:
341            pass
342       
343       
[f32d144]344    def set_edit_menu(self, owner):
[5d1c1f4]345        """
346        Set list of the edit model menu labels
347        """
[7c8d3093]348        id = wx.NewId()
[6f140f2]349        #new_model_menu = wx.Menu()
[f32d144]350        self.edit_model_menu.Append(id, 'New',
351                                   'Add a new model function')
352        wx.EVT_MENU(owner, id, self.make_new_model)
[6f140f2]353        id = wx.NewId()
[fdef956]354        self.edit_model_menu.Append(id, 'Sum|Multi(p1, p2)',
[f32d144]355                                    'Sum of two model functions')
356        wx.EVT_MENU(owner, id, self.make_sum_model)
[7c8d3093]357        e_id = wx.NewId()
[f32d144]358        self.edit_menu = wx.Menu()
[7c8d3093]359        self.edit_model_menu.AppendMenu(e_id, 
[b25caad]360                                    'Advanced', self.edit_menu) 
361        self.set_edit_menu_helper(owner, self.edit_custom_model)
362
363        d_id = wx.NewId()
[f32d144]364        self.delete_menu = wx.Menu()
365        self.edit_model_menu.AppendMenu(d_id,
366                                        'Delete', self.delete_menu)
[b25caad]367        self.set_edit_menu_helper(owner, self.delete_custom_model)
[7c8d3093]368   
[d83549c]369    def set_edit_menu_helper(self, owner=None, menu=None):
[7c8d3093]370        """
371        help for setting list of the edit model menu labels
372        """
[d83549c]373        if menu == None:
374            menu = self.edit_custom_model
[5d1c1f4]375        list_fnames = os.listdir(models.find_plugins_dir())
[d2843a9]376        list_fnames.sort()
377        for f_item in list_fnames:
378            name = os.path.basename(f_item)
[5d1c1f4]379            toks = os.path.splitext(name)
[f32d144]380            if toks[-1] == '.py' and not toks[0] == '__init__':
[b25caad]381                if menu == self.edit_custom_model:
[f32d144]382                    if toks[0] == 'easy_sum_of_p1_p2':
[b25caad]383                        continue
384                    submenu = self.edit_menu
385                else:
386                    submenu = self.delete_menu
[5d1c1f4]387                has_file = False
[b25caad]388                for item in submenu.GetMenuItems():
389                    if name == submenu.GetLabel(item.GetId()):
[5d1c1f4]390                        has_file = True
391                if not has_file:
392                    id = wx.NewId()
[f32d144]393                    submenu.Append(id, name)
394                    wx.EVT_MENU(owner, id, menu)
[5d1c1f4]395                    has_file = False
396
[056b569]397    def put_icon(self, frame):
398        """
399        Put icon in the frame title bar
400        """
401        if hasattr(frame, "IsIconized"):
402            if not frame.IsIconized():
403                try:
404                    icon = self.parent.GetIcon()
405                    frame.SetIcon(icon)
406                except:
[f32d144]407                    pass
408
[51d47b5]409    def on_add_sim_page(self, event):
[ed2ea6a]410        """
[5062bbf]411        Create a page to access simultaneous fit option
[ed2ea6a]412        """
[fa02d95]413        id = event.GetId()
414        caption = "Simultaneous Fit"
415        page = self.sim_page
416        if id == self.id_batchfit:
[7b1ca97]417            caption = "Combined Batch"
[fa02d95]418            page = self.batch_page
419           
420        def set_focus_page(page):
421            page.Show(True)
422            page.Refresh()
423            page.SetFocus()
[581d1d4]424            self.parent._mgr.Update()
[fa02d95]425            msg = "%s already opened\n" % str(page.window_caption)
426            wx.PostEvent(self.parent, StatusEvent(status=msg))
427           
428        if page != None:
429            return set_focus_page(page)
430        if caption == "Simultaneous Fit":
431            self.sim_page = self.fit_panel.add_sim_page(caption=caption)
432        else:
433            self.batch_page = self.fit_panel.add_sim_page(caption=caption)
[b28717b]434       
[d89f09b]435    def help(self, evt):
436        """
[f32d144]437        Show a general help dialog.
[d89f09b]438        """
[484faf7]439        from help_panel import  HelpWindow
[f32d144]440        frame = HelpWindow(None, -1, 'HelpWindow')
[34ffeed]441        if hasattr(frame, "IsIconized"):
442            if not frame.IsIconized():
443                try:
444                    icon = self.parent.GetIcon()
445                    frame.SetIcon(icon)
446                except:
[f32d144]447                    pass
[2268d10]448        frame.Show(True)
449       
[6bbeacd4]450    def get_context_menu(self, plotpanel=None):
[d89f09b]451        """
[5062bbf]452        Get the context menu items available for P(r).them allow fitting option
453        for Data2D and Data1D only.
454       
455        :param graph: the Graph object to which we attach the context menu
456       
457        :return: a list of menu items with call-back function
458       
[f32d144]459        :note: if Data1D was generated from Theory1D
[5062bbf]460                the fitting option is not allowed
461               
[d89f09b]462        """
[940aca7]463        self.plot_panel = plotpanel
[6bbeacd4]464        graph = plotpanel.graph
[484faf7]465        fit_option = "Select data for fitting"
466        fit_hint =  "Dialog with fitting parameters "
[6bbeacd4]467       
468        if graph.selected_plottable not in plotpanel.plots:
469            return []
470        item = plotpanel.plots[graph.selected_plottable]
[f32d144]471        if item.__class__.__name__ is "Data2D":
472            if hasattr(item, "is_data"):
[6bbeacd4]473                if item.is_data:
474                    return [[fit_option, fit_hint, self._onSelect]]
475                else:
476                    return [] 
477            return [[fit_option, fit_hint, self._onSelect]]
478        else:
479           
480            # if is_data is true , this in an actual data loaded
481            #else it is a data created from a theory model
[f32d144]482            if hasattr(item, "is_data"):
[6bbeacd4]483                if item.is_data:
[f32d144]484                    return [[fit_option, fit_hint, self._onSelect]]
[6bbeacd4]485                else:
[f32d144]486                    return []
[940aca7]487            return [[fit_option, fit_hint, self._onSelect]]
[f32d144]488        return []
[d89f09b]489
490    def get_panels(self, parent):
491        """
[5062bbf]492        Create and return a list of panel objects
[d89f09b]493        """
494        self.parent = parent
[75fbd17]495        #self.parent.Bind(EVT_FITSTATE_UPDATE, self.on_set_state_helper)
[d89f09b]496        # Creation of the fit panel
[6bbeacd4]497        self.fit_panel = FitPanel(parent=self.parent, manager=self)
498        self.on_add_new_page(event=None)
[f93dfcb]499        #Set the manager for the main panel
[d89f09b]500        self.fit_panel.set_manager(self)
501        # List of windows used for the perspective
502        self.perspective = []
503        self.perspective.append(self.fit_panel.window_name)
[6bbeacd4]504       
[3b19ac9]505        #index number to create random model name
506        self.index_model = 0
[f32d144]507        self.index_theory = 0
[32673ac]508        self.parent.Bind(EVT_SLICER_PANEL, self._on_slicer_event)
[848a2ef]509        self.parent.Bind(EVT_SLICER_PARS_UPDATE, self._onEVT_SLICER_PANEL)
[f32d144]510        self.parent._mgr.Bind(wx.aui.EVT_AUI_PANE_CLOSE,self._onclearslicer)
[3bf4a032]511        #Create reader when fitting panel are created
[f32d144]512        self.state_reader = Reader(self.set_state)
513        #append that reader to list of available reader
[3bf4a032]514        loader = Loader()
515        loader.associate_file_reader(".fitv", self.state_reader)
[f93dfcb]516        #Send the fitting panel to guiframe
[f32d144]517        self.mypanels.append(self.fit_panel)
[568e1a5]518        return self.mypanels
[232c3db]519   
[90a7bbd]520    def clear_panel(self):
521        """
522        """
[81f00d7]523        self.fit_panel.clear_panel()
[90a7bbd]524       
[1610976]525    def set_default_perspective(self):
526        """
[5062bbf]527        Call back method that True to notify the parent that the current plug-in
[f32d144]528        can be set as default perspective.
529        when returning False, the plug-in is not candidate for an automatic
[5062bbf]530        default perspective setting
[1610976]531        """
532        return True
[a0da535]533   
[17553ae]534    def delete_data(self, data):
[8ee56a9]535        """
536        delete  the given data from panel
537        """
538        self.fit_panel.delete_data(data)
539       
[e88ebfd]540    def set_data(self, data_list=None):
[a0da535]541        """
542        receive a list of data to fit
543        """
[b2d9826]544        if data_list is None:
545            data_list = []
[c26a64b]546        selected_data_list = []
[cc31608]547        if self.batch_on:
548            page = self.add_fit_page(data=data_list)
[c26a64b]549        else:
[cc31608]550            if len(data_list) > MAX_NBR_DATA:
551                from fitting_widgets import DataDialog
552                dlg = DataDialog(data_list=data_list, nb_data=MAX_NBR_DATA)
553                if dlg.ShowModal() == wx.ID_OK:
554                    selected_data_list = dlg.get_data()
555                dlg.Destroy()
556               
557            else:
558                selected_data_list = data_list
559            try:
[5e48acb]560                group_id = wx.NewId()
[cc31608]561                for data in selected_data_list:
[5e48acb]562                    if data is not None:
[33b7954]563                        # 2D has no same group_id
564                        if data.__class__.__name__ == 'Data2D':
565                            group_id = wx.NewId()
[5e48acb]566                        data.group_id = group_id
[14774db]567                        if group_id not in data.list_group_id:
568                            data.list_group_id.append(group_id)
569                        page = self.add_fit_page(data=[data])
[cc31608]570            except:
571                msg = "Fitting Set_data: " + str(sys.exc_value)
572                wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
[1b1bbf9]573   
574    def set_top_panel(self):
575        """
576        Close default (welcome) panel
577        """
578        if 'default' in self.parent.panels:
579            self.parent.on_close_welcome_panel()
580
[5e48acb]581       
[f32d144]582    def set_theory(self, theory_list=None):
[e88ebfd]583        """
584        """
585        #set the model state for a given theory_state:
586        for item in theory_list:
587            try:
588                _, theory_state = item
589                self.fit_panel.set_model_state(theory_state)
590            except:
591                msg = "Fitting: cannot deal with the theory received"
592                logging.error("set_theory " + msg + "\n" + str(sys.exc_value))
[f32d144]593                wx.PostEvent(self.parent,
594                             StatusEvent(status=msg, info="error"))
[a0da535]595           
[b63dc6e]596    def set_state(self, state=None, datainfo=None, format=None):
[f83b94a]597        """
[5062bbf]598        Call-back method for the fit page state reader.
[8897d66]599        This method is called when a .fitv/.svs file is loaded.
[5062bbf]600       
[8897d66]601        : param state: PageState object
602        : param datainfo: data
[f83b94a]603        """
[90a7bbd]604        #state = self.state_reader.get_state()
[b63dc6e]605        if state != None:
[4bee68d]606            state = state.clone()
[b63dc6e]607            # store fitting state in temp_state
[f32d144]608            self.temp_state.append(state)
[b63dc6e]609        else:
610            self.temp_state = []
[ef16f59]611        # index to start with for a new set_state
612        self.state_index = 0
[b63dc6e]613        # state file format
614        self.sfile_ext = format
[75fbd17]615       
616        self.on_set_state_helper(event=None)
[ef16f59]617
[f32d144]618    def  on_set_state_helper(self, event=None):
[4da35bc]619        """
[f32d144]620        Set_state_helper. This actually sets state
[0fd2f27]621        after plotting data from state file.
[ef16f59]622       
[f32d144]623        : event: FitStateUpdateEvent called
[0fd2f27]624            by dataloader.plot_data from guiframe
[4da35bc]625        """
[7a07864]626        if len(self.temp_state) == 0:
[f32d144]627            if self.state_index == 0 and len(self.mypanels) <= 0 \
628            and self.sfile_ext == '.svs':
[b63dc6e]629                self.temp_state = []
630                self.state_index = 0
[4da35bc]631            return
[3eb2811]632       
[ef16f59]633        try:
634            # Load fitting state
635            state = self.temp_state[self.state_index]
[3eb2811]636            #panel state should have model selection to set_state
637            if state.formfactorcombobox != None:
638                #set state
[75fbd17]639                data = self.parent.create_gui_data(state.data)
640                data.group_id = state.data.group_id
[f32d144]641                self.parent.add_data(data_list={data.id: data})
[75fbd17]642                wx.PostEvent(self.parent, NewPlotEvent(plot=data,
643                                        title=data.title))
[f95301b]644                #need to be fix later make sure we are sendind guiframe.data
645                #to panel
646                state.data = data
[f32d144]647                page = self.fit_panel.set_state(state)
[3eb2811]648            else:
649                #just set data because set_state won't work
[75fbd17]650                data = self.parent.create_gui_data(state.data)
651                data.group_id = state.data.group_id
[f32d144]652                self.parent.add_data(data_list={data.id: data})
[f95301b]653                wx.PostEvent(self.parent, NewPlotEvent(plot=data,
654                                        title=data.title))
[cc31608]655                page = self.add_fit_page([data])
[2dbffcc]656                caption = page.window_caption
[f32d144]657                self.store_data(uid=page.uid, data_list=page.get_data_list(),
[2dbffcc]658                        caption=caption)
[f32d144]659                self.mypanels.append(page)
[3eb2811]660               
[9b18735]661            # get ready for the next set_state
[ef16f59]662            self.state_index += 1
[9b18735]663
[f32d144]664            #reset state variables to default when all set_state is finished.
[ef16f59]665            if len(self.temp_state) == self.state_index:
[9b18735]666               
[ef16f59]667                self.temp_state = []
[b63dc6e]668                #self.state_index = 0
[ef16f59]669                # Make sure the user sees the fitting panel after loading
[f32d144]670                #self.parent.set_perspective(self.perspective)
[75fbd17]671                self.on_perspective(event=None)
[ef16f59]672        except:
[f32d144]673            self.state_index = 0
[8897d66]674            self.temp_state = []
[ef16f59]675            raise
[1b14795]676       
677    def set_param2fit(self, uid, param2fit):
678        """
679        Set the list of param names to fit for fitprobelm
680        """
681        self.page_finder[uid].set_param2fit(param2fit)
[5e48acb]682       
683    def set_graph_id(self, uid, graph_id):
684        """
685        Set graph_id for fitprobelm
686        """
687        self.page_finder[uid].set_graph_id(graph_id)
688       
689    def get_graph_id(self, uid):
690        """
691        Set graph_id for fitprobelm
692        """
[f32d144]693        return self.page_finder[uid].get_graph_id()
[5e48acb]694                         
[f32d144]695    def save_fit_state(self, filepath, fitstate):
[f83b94a]696        """
[5062bbf]697        save fit page state into file
[f83b94a]698        """
699        self.state_reader.write(filename=filepath, fitstate=fitstate)
[55bb249c]700
[f7ef313]701    def set_fit_weight(self, uid, flag, is2d=False, fid=None):
[55bb249c]702        """
703        Set the fit weights of a given page for all
[f32d144]704        its data by default. If fid is provide then set the range
[55bb249c]705        only for the data with fid as id
706        :param uid: id corresponding to a fit page
707        :param fid: id corresponding to a fit problem (data, model)
708        :param weight: current dy data
709        """
710        if uid in self.page_finder.keys():
[f7ef313]711            self.page_finder[uid].set_weight(flag=flag, is2d=is2d)
[55bb249c]712                   
[2dbffcc]713    def set_fit_range(self, uid, qmin, qmax, fid=None):
[ca7a626]714        """
[2dbffcc]715        Set the fitting range of a given page for all
[f32d144]716        its data by default. If fid is provide then set the range
[2dbffcc]717        only for the data with fid as id
718        :param uid: id corresponding to a fit page
719        :param fid: id corresponding to a fit problem (data, model)
720        :param qmin: minimum  value of the fit range
721        :param qmax: maximum  value of the fit range
[ca7a626]722        """
[2dbffcc]723        if uid in self.page_finder.keys():
[f6933b8]724            self.page_finder[uid].set_range(qmin=qmin, qmax=qmax, fid=fid)
725     
[f32d144]726    def schedule_for_fit(self, value=0, uid=None):
[948add7]727        """
[5062bbf]728        Set the fit problem field to 0 or 1 to schedule that problem to fit.
[f32d144]729        Schedule the specified fitproblem or get the fit problem related to
[5062bbf]730        the current page and set value.
[f32d144]731        :param value: integer 0 or 1
[2dbffcc]732        :param uid: the id related to a page contaning fitting information
733        """
[f32d144]734        if uid in self.page_finder.keys():
[66ff250]735            self.page_finder[uid].schedule_tofit(value)
[ca7a626]736         
[d89f09b]737    def get_page_finder(self):
[5062bbf]738        """
739        return self.page_finder used also by simfitpage.py
[f32d144]740        """
741        return self.page_finder
[d89f09b]742   
[f32d144]743    def set_page_finder(self, modelname, names, values):
[d89f09b]744        """
[5062bbf]745        Used by simfitpage.py to reset a parameter given the string constrainst.
746         
[f32d144]747        :param modelname: the name ot the model for with the parameter
[0fd2f27]748                            has to reset
[5062bbf]749        :param value: can be a string in this case.
750        :param names: the paramter name
751         
752        :note: expecting park used for fit.
753         
[f32d144]754        """
[66ff250]755        sim_page_id = self.sim_page.uid
756        for uid, value in self.page_finder.iteritems():
[fa02d95]757            if uid != sim_page_id and uid != self.batch_page.uid:
[6bbeacd4]758                list = value.get_model()
[f93dfcb]759                model = list[0]
[6bbeacd4]760                if model.name == modelname:
761                    value.set_model_param(names, values)
[d89f09b]762                    break
[2db1d66]763         
[f32d144]764    def split_string(self, item):
[d89f09b]765        """
[5062bbf]766        receive a word containing dot and split it. used to split parameterset
767        name into model name and parameter name example: ::
768       
[3b19ac9]769            paramaterset (item) = M1.A
[5062bbf]770            Will return model_name = M1 , parameter name = A
771           
[d89f09b]772        """
[6d8bad4]773        if item.find(".") >= 0:
[7c8d3093]774            param_names = re.split("\.", item)
[f32d144]775            model_name = param_names[0]
[6d91073]776            ##Assume max len is 3; eg., M0.radius.width
777            if len(param_names) == 3:
[7c8d3093]778                param_name = param_names[1] + "." + param_names[2]
[6d91073]779            else:
[f32d144]780                param_name = param_names[1]
[7c8d3093]781            return model_name, param_name
[2296316]782   
783    def set_ftol(self, ftol=None):
784        """
[f32d144]785        Set ftol: Relative error desired in the sum of chi squares.
[2296316]786        """
787        # check if it is flaot
788        try:
789            f_tol = float(ftol)
790        except:
791            # default
[2c6b224]792            f_tol = SANS_F_TOL
[2296316]793           
794        self.ftol = f_tol
[2c6b224]795        # update ftol menu help strings
796        ftol_help = "Change the current FTolerance (=%s) " % str(self.ftol)
[f32d144]797        ftol_help += "of Simple FitEngine..."
[dafc36f]798        if self.menu1 != None:
799            self.menu1.SetHelpString(self.id_tol, ftol_help)
[2c6b224]800       
801    def show_ftol_dialog(self, event=None):
802        """
803        Dialog to select ftol for Scipy
804        """
805        #if event != None:
806        #    event.Skip()
807        from ftol_dialog import ChangeFtol
808        panel = ChangeFtol(self.parent, self)
809        panel.ShowModal()
810                 
[66ff250]811    def stop_fit(self, uid):
[ed2ea6a]812        """
[5062bbf]813        Stop the fit engine
[ed2ea6a]814        """
[66ff250]815        if uid in self.fit_thread_list.keys():
816            calc_fit = self.fit_thread_list[uid]
817            if calc_fit is not  None and calc_fit.isrunning():
818                calc_fit.stop()
819                msg = "Fit stop!"
820                wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[3ddc2c4]821            del self.fit_thread_list[uid]
[66ff250]822        #set the fit button label of page when fit stop is trigger from
823        #simultaneous fit pane
[fa02d95]824        sim_flag = self.sim_page is not None and uid == self.sim_page.uid
825        batch_flag = self.batch_page is not None and uid == self.batch_page.uid
826        if sim_flag or batch_flag:
[66ff250]827            for uid, value in self.page_finder.iteritems():
828                if value.get_scheduled() == 1:
829                    if uid in self.fit_panel.opened_pages.keys():
830                        panel = self.fit_panel.opened_pages[uid]
[fa02d95]831                        panel._on_fit_complete()
[6bbeacd4]832 
[2dbffcc]833    def set_smearer(self, uid, smearer, fid, qmin=None, qmax=None, draw=True,
834                    enable_smearer=False):
835        """
836        Get a smear object and store it to a fit problem of fid as id. If proper
837        flag is enable , will plot the theory with smearing information.
838       
839        :param smearer: smear object to allow smearing data of id fid
840        :param enable_smearer: Define whether or not all (data, model) contained
841            in the structure of id uid will be smeared before fitting.
842        :param qmin: the maximum value of the theory plotting range
843        :param qmax: the maximum value of the theory plotting range
844        :param draw: Determine if the theory needs to be plot
[f32d144]845        """
[66ff250]846        if uid not in self.page_finder.keys():
[2dbffcc]847            return
848        self.page_finder[uid].enable_smearing(flag=enable_smearer)
849        self.page_finder[uid].set_smearer(smearer, fid=fid)
[6bbeacd4]850        if draw:
851            ## draw model 1D with smeared data
[f32d144]852            data = self.page_finder[uid].get_fit_data(fid=fid)
[2dbffcc]853            if data is None:
854                msg = "set_mearer requires at least data.\n"
855                msg += "Got data = %s .\n" % str(data)
[269df2b]856                return
857                #raise ValueError, msg
[2dbffcc]858            model = self.page_finder[uid].get_model(fid=fid)
[6bbeacd4]859            if model is None:
860                return
[2dbffcc]861            enable1D = issubclass(data.__class__, Data1D)
862            enable2D = issubclass(data.__class__, Data2D)
[6bbeacd4]863            ## if user has already selected a model to plot
864            ## redraw the model with data smeared
[2dbffcc]865            smear = self.page_finder[uid].get_smearer(fid=fid)
[342dc442]866
867            # compute weight for the current data
868            weight = self.page_finder[uid].get_weight(fid=fid)
869
[66ff250]870            self.draw_model(model=model, data=data, page_id=uid, smearer=smear,
[ba1f0b2]871                enable1D=enable1D, enable2D=enable2D,
[342dc442]872                qmin=qmin, qmax=qmax, weight=weight)
[f6aee42]873            self._mac_sleep(0.2)
874           
875    def _mac_sleep(self, sec=0.2):
[2dbffcc]876        """
877        Give sleep to MAC
878        """
[f6aee42]879        if ON_MAC:
[f32d144]880            time.sleep(sec)
[2dbffcc]881       
[66ff250]882    def draw_model(self, model, page_id, data=None, smearer=None,
[67e258c]883                   enable1D=True, enable2D=False,
[5ef55d2]884                   state=None,
[f64a4b7]885                   fid=None,
[fa65e99]886                   toggle_mode_on=False,
[f32d144]887                   qmin=None, qmax=None,
[e3f6ef5]888                   update_chisqr=True, weight=None, source='model'):
[ca7a626]889        """
[5062bbf]890        Draw model.
891       
892        :param model: the model to draw
893        :param name: the name of the model to draw
894        :param data: the data on which the model is based to be drawn
895        :param description: model's description
896        :param enable1D: if true enable drawing model 1D
897        :param enable2D: if true enable drawing model 2D
898        :param qmin:  Range's minimum value to draw model
899        :param qmax:  Range's maximum value to draw model
900        :param qstep: number of step to divide the x and y-axis
[2296316]901        :param update_chisqr: update chisqr [bool]
[ed2ea6a]902             
[d89f09b]903        """
[62f851f]904        #self.weight = weight
[f32d144]905        if issubclass(data.__class__, Data1D) or not enable2D:
[f72333f]906            ## draw model 1D with no loaded data
[f32d144]907            self._draw_model1D(model=model,
[67e258c]908                               data=data,
[66ff250]909                               page_id=page_id,
[f32d144]910                               enable1D=enable1D,
[67e258c]911                               smearer=smearer,
912                               qmin=qmin,
[f32d144]913                               qmax=qmax,
[f64a4b7]914                               fid=fid,
[62f851f]915                               weight=weight,
[fa65e99]916                               toggle_mode_on=toggle_mode_on,
[5ef55d2]917                               state=state,
[e3f6ef5]918                               update_chisqr=update_chisqr,
919                               source=source)
[f32d144]920        else:
[f72333f]921            ## draw model 2D with no initial data
[ba1f0b2]922            self._draw_model2D(model=model,
[66ff250]923                                page_id=page_id,
[67e258c]924                                data=data,
925                                enable2D=enable2D,
926                                smearer=smearer,
927                                qmin=qmin,
928                                qmax=qmax,
[f64a4b7]929                                fid=fid,
[d682c92]930                                weight=weight,
[5ef55d2]931                                state=state,
[fa65e99]932                                toggle_mode_on=toggle_mode_on,
[e3f6ef5]933                                update_chisqr=update_chisqr,
934                                source=source)
[2db1d66]935           
[922497f]936    def onFit(self, uid):
[ca7a626]937        """
[2dbffcc]938        Get series of data, model, associates parameters and range and send then
[f32d144]939        to  series of fit engines. Fit data and model, display result to
940        corresponding panels.
[2dbffcc]941        :param uid: id related to the panel currently calling this fit function.
[ca7a626]942        """
[31469d50]943        flag = True
[f32d144]944        ##  count the number of fitproblem schedule to fit
[67e258c]945        fitproblem_count = 0
[66ff250]946        for value in self.page_finder.values():
[67e258c]947            if value.get_scheduled() == 1:
[bb18ef1]948                fitproblem_count += 1
[f32d144]949        self._gui_engine = self._return_engine_type()
950        self.fitproblem_count = fitproblem_count
[67e258c]951        if self._fit_engine == "park":
952            engineType = "Simultaneous Fit"
[ca7a626]953        else:
[67e258c]954            engineType = "Single Fit"
[f32d144]955        fitter_list = []
956        sim_fitter = None
[284f6fe]957        is_single_fit = True
[fa02d95]958        batch_on = False
[922497f]959        if self.sim_page is not None and self.sim_page.uid == uid:
[f32d144]960            #simulatanous fit only one engine need to be created
[284f6fe]961            ## if simultaneous fit change automatically the engine to park
[f32d144]962            self._on_change_engine(engine='park')
963            sim_fitter = Fit(self._fit_engine)
[fa02d95]964            sim_fitter.fitter_id = self.sim_page.uid
[f32d144]965            fitter_list.append(sim_fitter)
[284f6fe]966            is_single_fit = False
[fa02d95]967            batch_on = self.sim_page.batch_on
[7db52f1]968           
[f32d144]969        self.fitproblem_count = fitproblem_count
[284f6fe]970        if self._fit_engine == "park":
971            engineType = "Simultaneous Fit"
972        else:
973            engineType = "Single Fit"
[922497f]974       
[67e258c]975        self.current_pg = None
[66ff250]976        list_page_id = []
[922497f]977        fit_id = 0
[33dd2e5]978        batch_inputs = {}
979        batch_outputs = {}
[66ff250]980        for page_id, value in self.page_finder.iteritems():
[dc613d6]981            # For simulfit (uid give with None), do for-loop
982            # if uid is specified (singlefit), do it only on the page.
983            if engineType == "Single Fit":
[fa02d95]984                #combine more than 1 batch page on single mode
985                if self.batch_page is None or self.batch_page.uid != uid:
986                    if page_id != uid:
987                        continue
[d89f09b]988            try:
[67e258c]989                if value.get_scheduled() == 1:
[f64a4b7]990                    value.nbr_residuals_computed = 0
[f93dfcb]991                    #Get list of parameters name to fit
[d89f09b]992                    pars = []
993                    templist = []
[66ff250]994                    page = self.fit_panel.get_page_by_id(page_id)
[f32d144]995                    self.set_fit_weight(uid=page.uid,
[7ee79cb]996                                     flag=page.get_weight_flag(),
[f32d144]997                                     is2d=page._is_2D())
[3b19ac9]998                    templist = page.get_param_list()
[f32d144]999                    flag = page._update_paramv_on_fit()
[31469d50]1000                    if not flag:
1001                        msg = "Fitting range or parameter values are"
[f32d144]1002                        msg += " invalid in %s" % \
[31469d50]1003                                    page.window_caption
[f32d144]1004                        wx.PostEvent(page.parent.parent,
1005                                     StatusEvent(status=msg, info="error",
[31469d50]1006                                     type="stop"))
1007                        return flag
[d89f09b]1008                    for element in templist:
[513115c]1009                        name = str(element[1])
[ca7a626]1010                        pars.append(name)
[7bd6cfae]1011                    fitproblem_list = value.values()
1012                    for fitproblem in  fitproblem_list:
[922497f]1013                        if sim_fitter is None:
[f32d144]1014                            fitter = Fit(self._fit_engine)
[fa02d95]1015                            fitter.fitter_id = page_id
[f32d144]1016                            self._fit_helper(fitproblem=fitproblem,
1017                                             pars=pars,
1018                                             fitter=fitter,
1019                                             fit_id=fit_id,
1020                                             batch_inputs=batch_inputs,
1021                                             batch_outputs=batch_outputs)
1022                            fitter_list.append(fitter)
[922497f]1023                        else:
1024                            fitter = sim_fitter
[f32d144]1025                            self._fit_helper(fitproblem=fitproblem,
1026                                             pars=pars,
1027                                             fitter=fitter,
1028                                             fit_id=fit_id,
1029                                             batch_inputs=batch_inputs,
1030                                             batch_outputs=batch_outputs)
[922497f]1031                        fit_id += 1
[66ff250]1032                    list_page_id.append(page_id)
1033                    current_page_id = page_id
[922497f]1034                    value.clear_model_param()
[d89f09b]1035            except:
[31469d50]1036                flag = False
[f32d144]1037                msg = "%s error: %s" % (engineType, sys.exc_value)
[ba1f0b2]1038                wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
1039                                                      type="stop"))
[31469d50]1040                return flag
[e54d2c32]1041        ## If a thread is already started, stop it
1042        #if self.calc_fit!= None and self.calc_fit.isrunning():
1043        #    self.calc_fit.stop()
[a2cb1f9]1044        msg = "Fitting is in progress..."
[f32d144]1045        wx.PostEvent(self.parent, StatusEvent(status=msg, type="progress"))
[a2cb1f9]1046       
1047        #Handler used for park engine displayed message
[66ff250]1048        handler = ConsoleUpdate(parent=self.parent,
1049                                manager=self,
1050                                improvement_delta=0.1)
[f6aee42]1051        self._mac_sleep(0.2)
[e54d2c32]1052        ## perform single fit
[05fb6d12]1053        try:
1054            page = self.fit_panel.get_page_by_id(uid)
1055            batch_on = page.batch_on
1056        except:
[fa02d95]1057            try:
[f32d144]1058                #if the id cannot be found then  we deal with a self.sim_page
[fa02d95]1059                #or a self.batch_page
1060                if self.sim_page is not None and uid == self.sim_page.uid:
[7b1ca97]1061                    batch_on = self.sim_page.batch_on
[fa02d95]1062                if self.batch_page is not None and uid == self.batch_page.uid:
[7b1ca97]1063                    batch_on = self.batch_page.batch_on
[fa02d95]1064            except:
[f32d144]1065                batch_on = False
1066
[05fb6d12]1067        # batch fit
1068        if batch_on:
[f32d144]1069            calc_fit = FitThread(handler=handler,
1070                                 fn=fitter_list,
1071                                 pars=pars,
1072                                 batch_inputs=batch_inputs,
1073                                 batch_outputs=batch_outputs,
1074                                 page_id=list_page_id,
1075                                 completefn=self._batch_fit_complete,
1076                                 ftol=self.ftol,
1077                                 reset_flag=self.batch_reset_flag)
[e54d2c32]1078        else:
[f32d144]1079            # single fit: not batch and not simul fit
[92a5ac92]1080            if not is_single_fit:
1081                current_page_id = self.sim_page.uid
[e54d2c32]1082            ## Perform more than 1 fit at the time
[58e0c83]1083            calc_fit = FitThread(handler=handler,
[cc31608]1084                                    fn=fitter_list,
[33dd2e5]1085                                    batch_inputs=batch_inputs,
1086                                    batch_outputs=batch_outputs,
[66ff250]1087                                    page_id=list_page_id,
[2296316]1088                                    updatefn=handler.update_fit,
[6a4002d]1089                                    completefn=self._fit_completed,
[2296316]1090                                    ftol=self.ftol)
[66ff250]1091        self.fit_thread_list[current_page_id] = calc_fit
[a2cb1f9]1092        calc_fit.queue()
[cc31608]1093        msg = "Fitting is in progress..."
[f32d144]1094        wx.PostEvent(self.parent, StatusEvent(status=msg, type="progress"))
[cc31608]1095       
1096        self.ready_fit(calc_fit=calc_fit)
[31469d50]1097        return flag
1098   
[662da312]1099    def ready_fit(self, calc_fit):
[c660493]1100        """
1101        Ready for another fit
1102        """
1103        if self.fitproblem_count != None and self.fitproblem_count > 1:
[662da312]1104            calc_fit.ready(2.5)
[c660493]1105        else:
1106            time.sleep(0.4)
[2f189dc]1107           
[2dbffcc]1108    def remove_plot(self, uid, fid=None, theory=False):
[2f189dc]1109        """
[5062bbf]1110        remove model plot when a fit page is closed
[2dbffcc]1111        :param uid: the id related to the fitpage to close
1112        :param fid: the id of the fitproblem(data, model, range,etc)
[2f189dc]1113        """
[2dbffcc]1114        if uid not in self.page_finder.keys():
1115            return
1116        fitproblemList = self.page_finder[uid].get_fit_problem(fid)
1117        for fitproblem in fitproblemList:
1118            data = fitproblem.get_fit_data()
1119            model = fitproblem.get_model()
1120            plot_id = None
1121            if model is not None:
[6d8bad4]1122                plot_id = data.id + model.name
[2dbffcc]1123            if theory:
[e1bc51d]1124                plot_id = data.id + model.name
[2dbffcc]1125            group_id = data.group_id
1126            wx.PostEvent(self.parent, NewPlotEvent(id=plot_id,
[f32d144]1127                                                   group_id=group_id,
1128                                                   action='remove'))
[edcbd467]1129           
[2dbffcc]1130    def store_data(self, uid, data_list=None, caption=None):
[31b0c47]1131        """
[2dbffcc]1132        Recieve a list of data and store them ans well as a caption of
1133        the fit page where they come from.
1134        :param uid: if related to a fit page
1135        :param data_list: list of data to fit
1136        :param caption: caption of the window related to these data
[31b0c47]1137        """
[cc31608]1138        if data_list is None:
1139            data_list = []
[f7ef313]1140       
[2dbffcc]1141        self.page_finder[uid].set_fit_data(data=data_list)
1142        if caption is not None:
1143            self.page_finder[uid].set_fit_tab_caption(caption=caption)
[f7ef313]1144           
[6bbeacd4]1145    def on_add_new_page(self, event=None):
[edcbd467]1146        """
[6bbeacd4]1147        ask fit panel to create a new empty page
[edcbd467]1148        """
1149        try:
[6bbeacd4]1150            page = self.fit_panel.add_empty_page()
[6450d8c]1151            # add data associated to the page created
[f32d144]1152            if page != None:
[f304194]1153                wx.PostEvent(self.parent, StatusEvent(status="Page Created",
[31b0c47]1154                                               info="info"))
[edcbd467]1155            else:
[f304194]1156                msg = "Page was already Created"
[a0da535]1157                wx.PostEvent(self.parent, StatusEvent(status=msg,
1158                                                       info="warning"))
[1b1bbf9]1159            self.set_top_panel()
[edcbd467]1160        except:
[2dbffcc]1161            msg = "Creating Fit page: %s"%sys.exc_value
1162            wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
[6bbeacd4]1163       
1164    def add_fit_page(self, data):
1165        """
1166        given a data, ask to the fitting panel to create a new fitting page,
1167        get this page and store it into the page_finder of this plug-in
[2dbffcc]1168        :param data: is a list of data
[6bbeacd4]1169        """
1170        page = self.fit_panel.set_data(data)
[7f76f89]1171        # page could be None when loading state files
1172        if page == None:
1173            return page
[6bbeacd4]1174        #append Data1D to the panel containing its theory
1175        #if theory already plotted
[66ff250]1176        if page.uid in self.page_finder:
[cc31608]1177            data = page.get_data()
[2dbffcc]1178            theory_data = self.page_finder[page.uid].get_theory_data(data.id)
[6bbeacd4]1179            if issubclass(data.__class__, Data2D):
[e88ebfd]1180                data.group_id = wx.NewId()
[a5701e6]1181                if theory_data is not None:
[66ff250]1182                    group_id = str(page.uid) + " Model1D"
[f32d144]1183                    wx.PostEvent(self.parent,
[a5701e6]1184                             NewPlotEvent(group_id=group_id,
1185                                               action="delete"))
[0fd2f27]1186                    self.parent.update_data(prev_data=theory_data,
[f32d144]1187                                             new_data=data)
[6bbeacd4]1188            else:
1189                if theory_data is not None:
[66ff250]1190                    group_id = str(page.uid) + " Model2D"
[e88ebfd]1191                    data.group_id = theory_data.group_id
[a5701e6]1192                    wx.PostEvent(self.parent, 
1193                             NewPlotEvent(group_id=group_id,
1194                                               action="delete"))
[0fd2f27]1195                    self.parent.update_data(prev_data=theory_data,
[f32d144]1196                                             new_data=data)
1197        self.store_data(uid=page.uid, data_list=page.get_data_list(),
[2dbffcc]1198                        caption=page.window_caption)
[33dd2e5]1199        if self.sim_page is not None and not self.batch_on:
[6bbeacd4]1200            self.sim_page.draw_page()
[fa02d95]1201        if self.batch_page is not None and self.batch_on:
1202            self.batch_page.draw_page()
1203           
[1b1bbf9]1204        return page
[6bbeacd4]1205           
[848a2ef]1206    def _onEVT_SLICER_PANEL(self, event):
1207        """
[f32d144]1208        receive and event telling to update a panel with a name starting with
1209        event.panel_name. this method update slicer panel
[2e08a9f]1210        for a given interactor.
[5062bbf]1211       
[f32d144]1212        :param event: contains type of slicer , paramaters for updating
[2e08a9f]1213            the panel and panel_name to find the slicer 's panel concerned.
[848a2ef]1214        """
1215        for item in self.parent.panels:
[0fd2f27]1216            name = event.panel_name
1217            if self.parent.panels[item].window_caption.startswith(name):
[848a2ef]1218                self.parent.panels[item].set_slicer(event.type, event.params)
[df4c3ad]1219               
1220        self.parent._mgr.Update()
[b324aa9]1221   
[f32d144]1222    def _closed_fitpage(self, event):
[848a2ef]1223        """
[f32d144]1224        request fitpanel to close a given page when its unique data is removed
[5062bbf]1225        from the plot. close fitpage only when the a loaded data is removed
[f32d144]1226        """
[784e2fa]1227        if event is None or event.data is None:
1228            return
[f32d144]1229        if hasattr(event.data, "is_data"):
[a0da535]1230            if not event.data.is_data or \
[66ff250]1231                event.data.__class__.__name__ == "Data1D":
[f32d144]1232                self.fit_panel.close_page_with_data(event.data)
[2dbffcc]1233 
[dc613d6]1234    def _reset_schedule_problem(self, value=0, uid=None):
[6f023e8]1235        """
[5062bbf]1236        unschedule or schedule all fitproblem to be fit
[6f023e8]1237        """
[dc613d6]1238        # case that uid is not specified
1239        if uid == None:
1240            for page_id in self.page_finder.keys():
1241                self.page_finder[page_id].schedule_tofit(value)
1242        # when uid is given
1243        else:
[2dbffcc]1244            if uid in self.page_finder.keys():
1245                self.page_finder[uid].schedule_tofit(value)
[dc613d6]1246               
[33dd2e5]1247    def _fit_helper(self, fitproblem, pars, fitter, fit_id,
1248                    batch_inputs, batch_outputs):
[2140e68]1249        """
[2dbffcc]1250        Create and set fit engine with series of data and model
1251        :param pars: list of fittable parameters
1252        :param fitter_list: list of fit engine
[f32d144]1253        :param value:  structure storing data mapped to their model, range etc.
[2140e68]1254        """
[922497f]1255        data = fitproblem.get_fit_data()
1256        model = fitproblem.get_model()
1257        smearer = fitproblem.get_smearer()
1258        qmin, qmax = fitproblem.get_range()
[29a4024]1259
[922497f]1260        #Extra list of parameters and their constraints
1261        listOfConstraint = []
1262        param = fitproblem.get_model_param()
1263        if len(param) > 0:
1264            for item in param:
1265                ## check if constraint
1266                if item[0] != None and item[1] != None:
[f32d144]1267                    listOfConstraint.append((item[0], item[1]))
1268        new_model = model
[41661a0]1269        fitter.set_model(new_model, fit_id, pars, data=data,
1270                         constraints=listOfConstraint)
[f32d144]1271        fitter.set_data(data=data, id=fit_id, smearer=smearer, qmin=qmin,
[922497f]1272                        qmax=qmax)
1273        fitter.select_problem_for_fit(id=fit_id, value=1)
[7bd6cfae]1274       
[f32d144]1275    def _onSelect(self, event):
1276        """
1277        when Select data to fit a new page is created .Its reference is
[5062bbf]1278        added to self.page_finder
[d89f09b]1279        """
[940aca7]1280        panel = self.plot_panel
1281        if panel == None:
1282            raise ValueError, "Fitting:_onSelect: NonType panel"
[67e258c]1283        Plugin.on_perspective(self, event=event)
[940aca7]1284        self.select_data(panel)
[5e48acb]1285       
1286    def select_data(self, panel):
1287        """
1288        """
[940aca7]1289        for plottable in panel.graph.plottables:
[2fff4db]1290            if plottable.__class__.__name__ in ["Data1D", "Theory1D"]:
[940aca7]1291                data_id = panel.graph.selected_plottable
1292                if plottable == panel.plots[data_id]:
[6bbeacd4]1293                    data = plottable
[cc31608]1294                    self.add_fit_page(data=[data])
[edcbd467]1295                    return
[c77d859]1296            else:
[6bbeacd4]1297                data = plottable
[cc31608]1298                self.add_fit_page(data=[data])
[1b1bbf9]1299        self.set_top_panel()
[1c1436d]1300           
[66ff250]1301    def update_fit(self, result=None, msg=""):
1302        """
1303        """
1304        print "update_fit result", result
1305       
[f32d144]1306    def _batch_fit_complete(self, result, pars, page_id,
[33dd2e5]1307                            batch_outputs, batch_inputs, elapsed=None):
[cc31608]1308        """
[f32d144]1309        Display fit result in batch
[2dbffcc]1310        :param result: list of objects received fromt fit engines
1311        :param pars: list of  fitted parameters names
1312        :param page_id: list of page ids which called fit function
1313        :param elapsed: time spent at the fitting level
[cc31608]1314        """
[92a5ac92]1315        self._mac_sleep(0.2)
1316        uid = page_id[0]
1317        if uid in self.fit_thread_list.keys():
[f32d144]1318            del self.fit_thread_list[uid]
[92a5ac92]1319         
[cc31608]1320        self._update_fit_button(page_id)
[2bb37c3]1321        t1 = time.time()
1322        str_time = time.strftime("%a, %d %b %Y %H:%M:%S ", time.localtime(t1))
1323        msg = "Fit completed on %s \n" % str_time
1324        msg += "Duration time: %s s.\n" % str(elapsed)
[cc31608]1325        wx.PostEvent(self.parent, StatusEvent(status=msg, info="info",
[f32d144]1326                                              type="stop"))
[fa02d95]1327       
[33dd2e5]1328        if batch_outputs is None:
1329            batch_outputs = {}
[fa02d95]1330       
1331        # format batch_outputs
1332        batch_outputs["Chi2"] = []
1333        #Don't like these loops
[f32d144]1334        # Need to create dictionary of all fitted parameters
[fa02d95]1335        # since the number of parameters can differ between each fit result
1336        for list_res in result:
1337            for res in list_res:
1338                model, data = res.inputs[0]
1339                if model is not None and hasattr(model, "model"):
1340                    model = model.model
1341                #get all fittable parameters of the current model
1342                for param in  model.getParamList():
1343                    if param  not in batch_outputs.keys():
1344                        batch_outputs[param] = []
1345                for param in model.getDispParamList():
1346                    if not model.is_fittable(param) and \
1347                        param in batch_outputs.keys():
1348                        del batch_outputs[param]
1349                # Add fitted parameters and their error
1350                for param in res.param_list:
1351                    if param not in batch_outputs.keys():
1352                        batch_outputs[param] = []
1353                    err_param = "error on %s" % str(param)
1354                    if err_param not in batch_inputs.keys():
1355                        batch_inputs[err_param] = []
1356        msg = ""
1357        for list_res in result:
1358            for res in list_res:
1359                pid = res.fitter_id
1360                model, data = res.inputs[0]
1361                correct_result = False
1362                if model is not None and hasattr(model, "model"):
1363                    model = model.model
1364                if data is not None and hasattr(data, "sans_data"):
1365                    data = data.sans_data
1366               
1367                is_data2d = issubclass(data.__class__, Data2D)
1368                #check consistency of arrays
1369                if not is_data2d:
1370                    if len(res.theory) == len(res.index[res.index]) and \
1371                        len(res.index) == len(data.y):
[46c4ccb]1372                        correct_result = True
[fa02d95]1373                else:
1374                    copy_data = deepcopy(data)
1375                    new_theory = copy_data.data
1376                    new_theory[res.index] = res.theory
[f32d144]1377                    new_theory[res.index == False] = numpy.nan
[fa02d95]1378                    correct_result = True
1379                #get all fittable parameters of the current model
1380                param_list = model.getParamList()
1381                for param in model.getDispParamList():
1382                    if not model.is_fittable(param) and \
1383                        param in param_list:
[f32d144]1384                        param_list.remove(param)
[fa02d95]1385                if not correct_result or res.fitness is None or \
1386                    not numpy.isfinite(res.fitness) or \
1387                    numpy.any(res.pvec == None) or not \
1388                    numpy.all(numpy.isfinite(res.pvec)):
1389                    data_name = str(None)
1390                    if data is not None:
1391                        data_name = str(data.name)
1392                    model_name = str(None)
1393                    if model is not None:
1394                        model_name = str(model.name)
[f32d144]1395                    msg += "Data %s and Model %s did not fit.\n" % (data_name,
[fa02d95]1396                                                                    model_name)
1397                    ERROR = numpy.NAN
1398                    cell = BatchCell()
1399                    cell.label = res.fitness
1400                    cell.value = res.fitness
1401                    batch_outputs["Chi2"].append(ERROR)
1402                    for param in param_list:
1403                        # save value of  fixed parameters
1404                        if param not in res.param_list:
1405                            batch_outputs[str(param)].append(ERROR)
1406                        else:
[4a52223]1407                            #save only fitted values
[fa02d95]1408                            batch_outputs[param].append(ERROR)
1409                            batch_inputs["error on %s" % str(param)].append(ERROR)
1410                else:
[f32d144]1411                    # ToDo: Why sometimes res.pvec comes with numpy.float64?
[26e1001]1412                    # Need to fix it within ScipyEngine
[f32d144]1413                    if res.pvec.__class__ == numpy.float64:
[26e1001]1414                        res.pvec = [res.pvec]
1415                       
[fa02d95]1416                    cell = BatchCell()
1417                    cell.label = res.fitness
1418                    cell.value = res.fitness
1419                    batch_outputs["Chi2"].append(cell)
1420                    # add parameters to batch_results
1421                    for param in param_list:
1422                        # save value of  fixed parameters
1423                        if param not in res.param_list:
1424                            batch_outputs[str(param)].append(model.getParam(param))
1425                        else:
1426                            index = res.param_list.index(param)
[47d9be79]1427                            #save only fitted values
[fa02d95]1428                            batch_outputs[param].append(res.pvec[index])
1429                            if res.stderr is not None and \
1430                                len(res.stderr) == len(res.param_list):
[15f68ce]1431                                item = res.stderr[index]
[fa02d95]1432                                batch_inputs["error on %s" % param].append(item)
[356d2d3]1433                            else:
1434                                batch_inputs["error on %s" % param].append('-')
[fa02d95]1435                            model.setParam(param, res.pvec[index])
[f32d144]1436                #fill the batch result with emtpy value if not in the current
[fa02d95]1437                #model
1438                EMPTY = "-"
1439                for key in batch_outputs.keys():
1440                    if key not in param_list and key not in ["Chi2", "Data"]:
1441                        batch_outputs[key].append(EMPTY)
1442                               
1443                self.page_finder[pid].set_batch_result(batch_inputs=batch_inputs,
[f32d144]1444                                                       batch_outputs=batch_outputs)
[fa02d95]1445               
1446                cpage = self.fit_panel.get_page_by_id(pid)
1447                cpage._on_fit_complete()
1448                self.page_finder[pid][data.id].set_result(res)
1449                fitproblem = self.page_finder[pid][data.id]
1450                qmin, qmax = fitproblem.get_range()
[a5f6748]1451                plot_result = False
1452                if correct_result:
[fa02d95]1453                    if not is_data2d:
[f32d144]1454                        self._complete1D(x=data.x, y=res.theory, page_id=pid,
1455                                         elapsed=None,
1456                                         index=res.index, model=model,
1457                                         weight=None, fid=data.id,
1458                                         toggle_mode_on=False, state=None,
1459                                         data=data, update_chisqr=False,
1460                                         source='fit', plot_result=plot_result)
[fa02d95]1461                    else:
1462                        self._complete2D(image=new_theory, data=data,
[f32d144]1463                                         model=model,
1464                                         page_id=pid, elapsed=None,
1465                                         index=res.index,
1466                                         qmin=qmin,
1467                                         qmax=qmax, fid=data.id, weight=None,
1468                                         toggle_mode_on=False, state=None,
1469                                         update_chisqr=False,
1470                                         source='fit', plot_result=plot_result)
1471                self.on_set_batch_result(page_id=pid,
1472                                         fid=data.id,
1473                                         batch_outputs=batch_outputs,
[fa02d95]1474                                         batch_inputs=batch_inputs)
[4a52223]1475       
[7643ba2]1476        wx.PostEvent(self.parent, StatusEvent(status=msg, error="error",
[f32d144]1477                                              type="stop"))
[1ac202f]1478        # Remove parameters that are not shown
1479        cpage = self.fit_panel.get_page_by_id(uid)
1480        tbatch_outputs = {}
1481        shownkeystr = cpage.get_copy_params()
1482        for key in batch_outputs.keys():
[f32d144]1483            if key in ["Chi2", "Data"] or shownkeystr.count(key) > 0:
[1ac202f]1484                tbatch_outputs[key] = batch_outputs[key]
1485               
[f32d144]1486        wx.CallAfter(self.parent.on_set_batch_result, tbatch_outputs,
1487                     batch_inputs, self.sub_menu)
[92a5ac92]1488       
[f64a4b7]1489    def on_set_batch_result(self, page_id, fid, batch_outputs, batch_inputs):
1490        """
1491        """
[f32d144]1492        pid = page_id
[f64a4b7]1493        if fid not in self.page_finder[pid]:
1494            return
1495        fitproblem = self.page_finder[pid][fid]
1496        index = self.page_finder[pid].nbr_residuals_computed - 1
[f32d144]1497        residuals = fitproblem.get_residuals()
[f64a4b7]1498        theory_data = fitproblem.get_theory_data()
1499        data = fitproblem.get_fit_data()
1500        model = fitproblem.get_model()
1501        #fill batch result information
1502        if "Data" not in batch_outputs.keys():
1503            batch_outputs["Data"] = []
1504        from sans.guiframe.data_processor import BatchCell
[ec9a635]1505        cell = BatchCell()
1506        cell.label = data.name
1507        cell.value = index
[6b3edc1]1508       
[ec9a635]1509        if theory_data != None:
[6b3edc1]1510            #Suucessful fit
1511            theory_data.id = wx.NewId()
[b5fe787]1512            theory_data.name = model.name + "[%s]" % str(data.name)
[e80e704]1513            if issubclass(theory_data.__class__, Data2D):
1514                group_id = wx.NewId()
1515                theory_data.group_id = group_id
1516                if group_id not in theory_data.list_group_id:
1517                    theory_data.list_group_id.append(group_id)
1518               
[ec9a635]1519            try:
1520                # associate residuals plot
[e80e704]1521                if issubclass(residuals.__class__, Data2D):
1522                    group_id = wx.NewId()
1523                    residuals.group_id = group_id
1524                    if group_id not in residuals.list_group_id:
1525                        residuals.list_group_id.append(group_id)
[ec9a635]1526                batch_outputs["Chi2"][index].object = [residuals]
1527            except:
1528                pass
1529
1530        cell.object = [data, theory_data]
1531        batch_outputs["Data"].append(cell)
1532        for key, value in data.meta_data.iteritems():
1533            if key not in batch_inputs.keys():
1534                batch_inputs[key] = []
[98fdccd]1535            #if key.lower().strip() != "loader":
1536            batch_inputs[key].append(value)
[ec9a635]1537        param = "temperature"
1538        if hasattr(data.sample, param):
1539            if param not in  batch_inputs.keys():
[f32d144]1540                batch_inputs[param] = []
[ec9a635]1541            batch_inputs[param].append(data.sample.temperature)
1542       
[6a4002d]1543    def _fit_completed(self, result, page_id, batch_outputs,
[f32d144]1544                       batch_inputs=None, pars=None, elapsed=None):
[c77d859]1545        """
[2dbffcc]1546        Display result of the fit on related panel(s).
1547        :param result: list of object generated when fit ends
1548        :param pars: list of names of parameters fitted
1549        :param page_id: list of page ids which called fit function
1550        :param elapsed: time spent at the fitting level
[c77d859]1551        """
[2bb37c3]1552        t1 = time.time()
1553        str_time = time.strftime("%a, %d %b %Y %H:%M:%S ", time.localtime(t1))
1554        msg = "Fit completed on %s \n" % str_time
1555        msg += "Duration time: %s s.\n" % str(elapsed)
1556        wx.PostEvent(self.parent, StatusEvent(status=msg, info="info",
1557                                                      type="stop"))
[bb8a180]1558        # reset fit_engine if changed by simul_fit
1559        if self._fit_engine != self._gui_engine:
1560            self._on_change_engine(self._gui_engine)
[2bb37c3]1561        self._update_fit_button(page_id)
[922497f]1562        result = result[0]
[dc613d6]1563        self.fit_thread_list = {}
[66ff250]1564        if page_id is None:
1565            page_id = []
[f32d144]1566        ## fit more than 1 model at the same time
1567        self._mac_sleep(0.2)
[ca7a626]1568        try:
[6a4002d]1569            index = 0
1570            for uid in page_id:
1571                res = result[index]
[7643ba2]1572                if res.fitness is None or \
[6a4002d]1573                    not numpy.isfinite(res.fitness) or \
1574                    numpy.any(res.pvec == None) or \
1575                    not numpy.all(numpy.isfinite(res.pvec)):
1576                    msg = "Fitting did not converge!!!"
[f32d144]1577                    wx.PostEvent(self.parent,
1578                             StatusEvent(status=msg,
[6a4002d]1579                                         info="warning",
1580                                         type="stop"))
1581                    self._update_fit_button(page_id)
1582                else:
[0438933]1583                    #set the panel when fit result are float not list
[f32d144]1584                    if res.pvec.__class__ == numpy.float64:
[0438933]1585                        pvec = [res.pvec]
1586                    else:
1587                        pvec = res.pvec
[f32d144]1588                    if res.stderr.__class__ == numpy.float64:
[0438933]1589                        stderr = [res.stderr]
1590                    else:
1591                        stderr = res.stderr
[7643ba2]1592                    cpage = self.fit_panel.get_page_by_id(uid)
[f32d144]1593                    # Make sure we got all results
[7643ba2]1594                    #(CallAfter is important to MAC)
[8b1d759]1595                    try:
1596                        #if res != None:
[f32d144]1597                        wx.CallAfter(cpage.onsetValues, res.fitness,
1598                                     res.param_list,
[8b1d759]1599                                     pvec, stderr)
1600                        index += 1
1601                        wx.CallAfter(cpage._on_fit_complete)
1602                    except:
[2d0756a5]1603                        msg = "Singular point: Fitting Error occurred."
[f32d144]1604                        wx.PostEvent(self.parent, StatusEvent(status=msg,
1605                                                              info="error",
1606                                                              type="stop"))
[8b1d759]1607                   
[6a4002d]1608        except:
[d7345b71]1609            msg = "Fit completed but Following"
[6a4002d]1610            msg += " error occurred:%s" % sys.exc_value
[2296316]1611            wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
1612                                                  type="stop"))
[6a4002d]1613           
[12cd4ec]1614    def _update_fit_button(self, page_id):
1615        """
1616        Update Fit button when fit stopped
1617       
1618        : parameter page_id: fitpage where the button is
1619        """
[37290c4]1620        if page_id.__class__.__name__ != 'list':
1621            page_id = [page_id]
[f32d144]1622        for uid in page_id:
[12cd4ec]1623            page = self.fit_panel.get_page_by_id(uid)
1624            page._on_fit_complete()
1625       
[c77d859]1626    def _on_show_panel(self, event):
[5062bbf]1627        """
1628        """
1629        pass
[da6fd1a]1630   
1631    def on_reset_batch_flag(self, event):
1632        """
1633        Set batch_reset_flag
1634        """
1635        event.Skip()
[dafc36f]1636        if self.menu1 == None:
1637            return
[da6fd1a]1638        menu_item = self.menu1.FindItemById(self.id_reset_flag)
1639        flag = menu_item.IsChecked()
1640        if not flag:
1641            menu_item.Check(False)
1642            self.batch_reset_flag = True
1643        else:
1644            menu_item.Check(True)
1645            self.batch_reset_flag = False
[5062bbf]1646       
[da6fd1a]1647        ## post a message to status bar
1648        msg = "Set Chain Fitting: %s" % str(not self.batch_reset_flag)
[f32d144]1649        wx.PostEvent(self.parent,
[da6fd1a]1650                     StatusEvent(status=msg))
1651
[f32d144]1652    def _onset_engine_park(self, event):
[c77d859]1653        """
[5062bbf]1654        set engine to park
[c77d859]1655        """
1656        self._on_change_engine('park')
1657       
[f32d144]1658    def _onset_engine_scipy(self, event):
[c77d859]1659        """
[5062bbf]1660        set engine to scipy
[c77d859]1661        """
1662        self._on_change_engine('scipy')
1663       
1664    def _on_slicer_event(self, event):
1665        """
[f32d144]1666        Receive a panel as event and send it to guiframe
[5062bbf]1667       
1668        :param event: event containing a panel
1669       
[c77d859]1670        """
[5062bbf]1671        if event.panel is not None:
[c77d859]1672            new_panel = event.panel
[c3435b7f]1673            self.slicer_panels.append(event.panel)
[c77d859]1674            # Set group ID if available
1675            event_id = self.parent.popup_panel(new_panel)
1676            new_panel.uid = event_id
[f32d144]1677            self.mypanels.append(new_panel)
[5062bbf]1678       
[848a2ef]1679    def _onclearslicer(self, event):
1680        """
[5062bbf]1681        Clear the boxslicer when close the panel associate with this slicer
[848a2ef]1682        """
[f32d144]1683        name = event.GetPane().caption
[c3435b7f]1684   
1685        for panel in self.slicer_panels:
[f32d144]1686            if panel.window_caption == name:
[c3435b7f]1687               
[df4c3ad]1688                for item in self.parent.panels:
[2dbffcc]1689                    if hasattr(self.parent.panels[item], "uid"):
[f32d144]1690                        if self.parent.panels[item].uid == panel.base.uid:
[df4c3ad]1691                            self.parent.panels[item].onClearSlicer(event)
1692                            self.parent._mgr.Update()
[f32d144]1693                            break
[c3435b7f]1694                break
[5062bbf]1695   
[707436d]1696    def _return_engine_type(self):
1697        """
[5062bbf]1698        return the current type of engine
[707436d]1699        """
1700        return self._fit_engine
1701     
[c77d859]1702    def _on_change_engine(self, engine='park'):
1703        """
[f32d144]1704        Allow to select the type of engine to perform fit
[5062bbf]1705       
1706        :param engine: the key work of the engine
1707       
[c77d859]1708        """
[77e23a2]1709        ## saving fit engine name
[c77d859]1710        self._fit_engine = engine
[707436d]1711        ## change menu item state
[2dbffcc]1712        if engine == "park":
[0b6ae5f]1713            self.menu1.FindItemById(self.park_id).Check(True)
1714            self.menu1.FindItemById(self.scipy_id).Check(False)
[707436d]1715        else:
[0b6ae5f]1716            self.menu1.FindItemById(self.park_id).Check(False)
1717            self.menu1.FindItemById(self.scipy_id).Check(True)
[77e23a2]1718        ## post a message to status bar
[a0da535]1719        msg = "Engine set to: %s" % self._fit_engine
[f32d144]1720        wx.PostEvent(self.parent,
[a0da535]1721                     StatusEvent(status=msg))
[66ff250]1722        ## send the current engine type to fitpanel
1723        self.fit_panel._on_engine_change(name=self._fit_engine)
[dc613d6]1724
[c77d859]1725    def _on_model_panel(self, evt):
1726        """
[5062bbf]1727        react to model selection on any combo box or model menu.plot the model 
1728       
1729        :param evt: wx.combobox event
1730       
[c77d859]1731        """
1732        model = evt.model
[f32d144]1733        uid = evt.uid
[6bbeacd4]1734        qmin = evt.qmin
1735        qmax = evt.qmax
[2dbffcc]1736        caption = evt.caption
1737        enable_smearer = evt.enable_smearer
[9237df4]1738        if model == None:
[c77d859]1739            return
[2dbffcc]1740        if uid not in self.page_finder.keys():
1741            return
[6bbeacd4]1742        # save the name containing the data name with the appropriate model
[66ff250]1743        self.page_finder[uid].set_model(model)
[2dbffcc]1744        self.page_finder[uid].enable_smearing(enable_smearer)
[66ff250]1745        self.page_finder[uid].set_range(qmin=qmin, qmax=qmax)
[2dbffcc]1746        self.page_finder[uid].set_fit_tab_caption(caption=caption)
[33dd2e5]1747        if self.sim_page is not None and not self.batch_on:
[6bbeacd4]1748            self.sim_page.draw_page()
[fa02d95]1749        if self.batch_page is not None and self.batch_on:
1750            self.batch_page.draw_page()
[5062bbf]1751       
[2dbffcc]1752    def _update1D(self, x, output):
[bb18ef1]1753        """
[5062bbf]1754        Update the output of plotting model 1D
[bb18ef1]1755        """
[58e0c83]1756        msg = "Plot updating ... "
[f32d144]1757        wx.PostEvent(self.parent, StatusEvent(status=msg, type="update"))
[e575db9]1758       
[ad3fa1e]1759    def _complete1D(self, x, y, page_id, elapsed, index, model,
[f64a4b7]1760                    weight=None, fid=None,
[f32d144]1761                    toggle_mode_on=False, state=None,
1762                    data=None, update_chisqr=True,
[a5f6748]1763                    source='model', plot_result=True):
[c77d859]1764        """
[5062bbf]1765        Complete plotting 1D data
[f32d144]1766        """
[c77d859]1767        try:
[1ec979d]1768            numpy.nan_to_num(y)
1769           
[6bbeacd4]1770            new_plot = Data1D(x=x, y=y)
[dce84c0]1771            new_plot.is_data = False
[55d2f7e]1772            new_plot.dy = numpy.zeros(len(y))
[6bbeacd4]1773            new_plot.symbol = GUIFRAME_ID.CURVE_SYMBOL_NUM
[f32d144]1774            _yaxis, _yunit = data.get_yaxis()
1775            _xaxis, _xunit = data.get_xaxis()
[2dbffcc]1776            new_plot.title = data.name
[5e48acb]1777
[f32d144]1778            new_plot.group_id = data.group_id
[5e48acb]1779            if new_plot.group_id == None:
1780                new_plot.group_id = data.group_id
[f32d144]1781            new_plot.id = str(page_id) + "model-" + data.name
[a582d28]1782            #if new_plot.id in self.color_dict:
[f32d144]1783            #    new_plot.custom_color = self.color_dict[new_plot.id]
[6bbeacd4]1784            #find if this theory was already plotted and replace that plot given
1785            #the same id
[2dbffcc]1786            theory_data = self.page_finder[page_id].get_theory_data(fid=data.id)
[71e5c11]1787           
1788            if data.is_data:
1789                data_name = str(data.name)
1790            else:
1791                data_name = str(model.__class__.__name__)
1792           
[f32d144]1793            new_plot.name = model.name + " [" + data_name +"]"
[6bbeacd4]1794            new_plot.xaxis(_xaxis, _xunit)
1795            new_plot.yaxis(_yaxis, _yunit)
[f32d144]1796            self.page_finder[page_id].set_theory_data(data=new_plot,
[2e08a9f]1797                                                      fid=data.id)
[2dbffcc]1798            self.parent.update_theory(data_id=data.id, theory=new_plot,
[f32d144]1799                                       state=state)
[66ff250]1800            current_pg = self.fit_panel.get_page_by_id(page_id)
[ae4ade7]1801            title = new_plot.title
[eb4f2fd]1802            batch_on = self.fit_panel.get_page_by_id(page_id).batch_on
1803            if not batch_on:
1804                wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot,
[2dbffcc]1805                                            title=str(title)))
[a5f6748]1806            elif plot_result:
[90a2358]1807                top_data_id = self.fit_panel.get_page_by_id(page_id).data.id
1808                if data.id == top_data_id:
1809                    wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot,
[f32d144]1810                                            title=str(title)))
[2dbffcc]1811            caption = current_pg.window_caption
1812            self.page_finder[page_id].set_fit_tab_caption(caption=caption)
[7643ba2]1813           
[f32d144]1814            self.page_finder[page_id].set_theory_data(data=new_plot,
[2dbffcc]1815                                                      fid=data.id)
1816            if toggle_mode_on:
[f32d144]1817                wx.PostEvent(self.parent,
[2dbffcc]1818                             NewPlotEvent(group_id=str(page_id) + " Model2D",
[f32d144]1819                                          action="Hide"))
[2296316]1820            else:
[2dbffcc]1821                if update_chisqr:
1822                    wx.PostEvent(current_pg,
[2e08a9f]1823                                 Chi2UpdateEvent(output=self._cal_chisqr(
1824                                                                data=data,
[f64a4b7]1825                                                                fid=fid,
[62f851f]1826                                                                weight=weight,
[2dbffcc]1827                                                            page_id=page_id,
1828                                                            index=index)))
1829                else:
[f64a4b7]1830                    self._plot_residuals(page_id=page_id, data=data, fid=fid,
[f32d144]1831                                         index=index, weight=weight)
[cfbd06a]1832
[d522635]1833            msg = "Computation  completed!"
[f32d144]1834            wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[c77d859]1835        except:
[6bbeacd4]1836            raise
[5062bbf]1837   
[f32d144]1838    def _update2D(self, output, time=None):
[c77d859]1839        """
[5062bbf]1840        Update the output of plotting model
[c77d859]1841        """
[58e0c83]1842        wx.PostEvent(self.parent, StatusEvent(status="Plot \
[a0da535]1843        #updating ... ", type="update"))
[a2cb1f9]1844        #self.ready_fit()
[58e0c83]1845 
[f32d144]1846    def _complete2D(self, image, data, model, page_id, elapsed, index, qmin,
1847                qmax, fid=None, weight=None, toggle_mode_on=False, state=None,
[a5f6748]1848                     update_chisqr=True, source='model', plot_result=True):
[c77d859]1849        """
[5062bbf]1850        Complete get the result of modelthread and create model 2D
1851        that can be plot.
[c77d859]1852        """
[1ec979d]1853        numpy.nan_to_num(image)
[f32d144]1854        new_plot = Data2D(image=image, err_image=data.err_data)
[743f480]1855        new_plot.name = model.name + '2d'
[818589a]1856        new_plot.title = "Analytical model 2D "
[2ee5c61]1857        new_plot.id = str(page_id) + "model-" + data.name
[f3242cf]1858        new_plot.group_id = str(page_id) + " Model2D"
[2dbffcc]1859        new_plot.detector = data.detector
1860        new_plot.source = data.source
[f32d144]1861        new_plot.is_data = False
[2dbffcc]1862        new_plot.qx_data = data.qx_data
1863        new_plot.qy_data = data.qy_data
1864        new_plot.q_data = data.q_data
1865        new_plot.mask = data.mask
1866        ## plot boundaries
1867        new_plot.ymin = data.ymin
1868        new_plot.ymax = data.ymax
1869        new_plot.xmin = data.xmin
1870        new_plot.xmax = data.xmax
1871        title = data.title
[71e5c11]1872       
[dce84c0]1873        new_plot.is_data = False
[71e5c11]1874        if data.is_data:
1875            data_name = str(data.name)
1876        else:
[743f480]1877            data_name = str(model.__class__.__name__) + '2d'
[71e5c11]1878
1879        if len(title) > 1:
[940aca7]1880            new_plot.title = "Model2D for %s "% model.name + data_name
[53cf669]1881        new_plot.name = model.name + " [" + \
[940aca7]1882                                    data_name + "]"
[fa65e99]1883        theory_data = deepcopy(new_plot)
[6bbeacd4]1884       
[f32d144]1885        self.page_finder[page_id].set_theory_data(data=theory_data,
1886                                                  fid=data.id)
1887        self.parent.update_theory(data_id=data.id,
[05319e2]1888                                       theory=new_plot,
[f32d144]1889                                       state=state)
[66ff250]1890        current_pg = self.fit_panel.get_page_by_id(page_id)
[ae4ade7]1891        title = new_plot.title
[a5f6748]1892        if not source == 'fit' and plot_result:
[763bec8]1893            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot,
[ae4ade7]1894                                               title=title))
[2dbffcc]1895        if toggle_mode_on:
[f32d144]1896            wx.PostEvent(self.parent,
[2dbffcc]1897                             NewPlotEvent(group_id=str(page_id) + " Model1D",
1898                                               action="Hide"))
[2296316]1899        else:
[2dbffcc]1900            # Chisqr in fitpage
1901            if update_chisqr:
1902                wx.PostEvent(current_pg,
1903                             Chi2UpdateEvent(output=self._cal_chisqr(data=data,
[62f851f]1904                                                                    weight=weight,
[f64a4b7]1905                                                                    fid=fid,
[2dbffcc]1906                                                         page_id=page_id,
1907                                                         index=index)))
1908            else:
[f64a4b7]1909                self._plot_residuals(page_id=page_id, data=data, fid=fid,
[62f851f]1910                                      index=index, weight=weight)
[d522635]1911        msg = "Computation  completed!"
[a0da535]1912        wx.PostEvent(self.parent, StatusEvent(status=msg, type="stop"))
[6bbeacd4]1913   
[2dbffcc]1914    def _draw_model2D(self, model, page_id, qmin,
1915                      qmax,
1916                      data=None, smearer=None,
[a0da535]1917                      description=None, enable2D=False,
[5ef55d2]1918                      state=None,
[f64a4b7]1919                      fid=None,
[62f851f]1920                      weight=None,
[fa65e99]1921                      toggle_mode_on=False,
[e3f6ef5]1922                       update_chisqr=True, source='model'):
[f93dfcb]1923        """
[5062bbf]1924        draw model in 2D
1925       
1926        :param model: instance of the model to draw
1927        :param description: the description of the model
1928        :param enable2D: when True allows to draw model 2D
1929        :param qmin: the minimum value to  draw model 2D
1930        :param qmax: the maximum value to draw model 2D
1931        :param qstep: the number of division of Qx and Qy of the model to draw
[f93dfcb]1932           
1933        """
[c77d859]1934        if not enable2D:
[2dbffcc]1935            return None
[c77d859]1936        try:
1937            from model_thread import Calc2D
1938            ## If a thread is already started, stop it
[a0da535]1939            if (self.calc_2D is not None) and self.calc_2D.isrunning():
[c77d859]1940                self.calc_2D.stop()
[f32d144]1941            self.calc_2D = Calc2D(model=model,
[a0da535]1942                                    data=data,
[66ff250]1943                                    page_id=page_id,
[a0da535]1944                                    smearer=smearer,
1945                                    qmin=qmin,
1946                                    qmax=qmax,
[62f851f]1947                                    weight=weight, 
[f64a4b7]1948                                    fid=fid,
[fa65e99]1949                                    toggle_mode_on=toggle_mode_on,
[5ef55d2]1950                                    state=state,
[2296316]1951                                    completefn=self._complete2D,
[e3f6ef5]1952                                    update_chisqr=update_chisqr, source=source)
[c77d859]1953            self.calc_2D.queue()
1954        except:
[6bbeacd4]1955            raise
[f72333f]1956
[f32d144]1957    def _draw_model1D(self, model, page_id, data,
[2dbffcc]1958                      qmin, qmax, smearer=None,
[5ef55d2]1959                state=None,
[62f851f]1960                weight=None,
[f32d144]1961                fid=None,
[e3f6ef5]1962                toggle_mode_on=False, update_chisqr=True, source='model',
[2296316]1963                enable1D=True):
[c77d859]1964        """
[5062bbf]1965        Draw model 1D from loaded data1D
1966       
1967        :param data: loaded data
1968        :param model: the model to plot
1969       
[c77d859]1970        """
1971        if not enable1D:
[f32d144]1972            return
[c77d859]1973        try:
1974            from model_thread import Calc1D
1975            ## If a thread is already started, stop it
[a0da535]1976            if (self.calc_1D is not None) and self.calc_1D.isrunning():
[c77d859]1977                self.calc_1D.stop()
[2dbffcc]1978            self.calc_1D = Calc1D(data=data,
[6bbeacd4]1979                                  model=model,
[66ff250]1980                                  page_id=page_id, 
[a0da535]1981                                  qmin=qmin,
1982                                  qmax=qmax,
1983                                  smearer=smearer,
[5ef55d2]1984                                  state=state,
[62f851f]1985                                  weight=weight,
[f64a4b7]1986                                  fid=fid,
[fa65e99]1987                                  toggle_mode_on=toggle_mode_on,
[2296316]1988                                  completefn=self._complete1D,
1989                                  #updatefn = self._update1D,
[e3f6ef5]1990                                  update_chisqr=update_chisqr,
1991                                  source=source)
[c77d859]1992            self.calc_1D.queue()
1993        except:
[a0da535]1994            msg = " Error occurred when drawing %s Model 1D: " % model.name
1995            msg += " %s" % sys.exc_value
1996            wx.PostEvent(self.parent, StatusEvent(status=msg))
[2dbffcc]1997   
[f32d144]1998    def _cal_chisqr(self, page_id, data, weight, fid=None, index=None):
[f72333f]1999        """
[5062bbf]2000        Get handy Chisqr using the output from draw1D and 2D,
2001        instead of calling expansive CalcChisqr in guithread
[f72333f]2002        """
[f32d144]2003        data_copy = deepcopy(data)
[f72333f]2004        # default chisqr
2005        chisqr = None
[2dbffcc]2006        #to compute chisq make sure data has valid data
[f72333f]2007        # return None if data == None
[df9e0fc]2008        if not check_data_validity(data_copy) or data_copy == None:
[2dbffcc]2009            return chisqr
[df9e0fc]2010
[f72333f]2011        # Get data: data I, theory I, and data dI in order
[ca2bdae]2012        if data_copy.__class__.__name__ == "Data2D":
[f32d144]2013            if index == None:
[dbb3914]2014                index = numpy.ones(len(data_copy.data), dtype=bool)
[62f851f]2015            if weight != None:
2016                data_copy.err_data = weight
[a0da535]2017            # get rid of zero error points
[f32d144]2018            index = index & (data_copy.err_data != 0)
2019            index = index & (numpy.isfinite(data_copy.data))
2020            fn = data_copy.data[index]
[ca2bdae]2021            theory_data = self.page_finder[page_id].get_theory_data(fid=data_copy.id)
[f32d144]2022            if theory_data == None:
[df9e0fc]2023                return chisqr
[6bbeacd4]2024            gn = theory_data.data[index]
[ca2bdae]2025            en = data_copy.err_data[index]
[f72333f]2026        else:
2027            # 1 d theory from model_thread is only in the range of index
[2296316]2028            if index == None: 
[dbb3914]2029                index = numpy.ones(len(data_copy.y), dtype=bool)
[62f851f]2030            if weight != None:
2031                data_copy.dy = weight
[ca2bdae]2032            if data_copy.dy == None or data_copy.dy == []:
2033                dy = numpy.ones(len(data_copy.y))
[f151ddf]2034            else:
[a0da535]2035                ## Set consitently w/AbstractFitengine:
2036                # But this should be corrected later.
[ca2bdae]2037                dy = deepcopy(data_copy.dy)
[f32d144]2038                dy[dy == 0] = 1
2039            fn = data_copy.y[index]
[eb4f2fd]2040           
[ca2bdae]2041            theory_data = self.page_finder[page_id].get_theory_data(fid=data_copy.id)
[f32d144]2042            if theory_data == None:
[df9e0fc]2043                return chisqr
[6bbeacd4]2044            gn = theory_data.y
[4d5fa8c]2045            en = dy[index]
[df9e0fc]2046           
[f72333f]2047        # residual
[2296316]2048        res = (fn - gn) / en
2049        residuals = res[numpy.isfinite(res)]
[f72333f]2050        # get chisqr only w/finite
[2296316]2051        chisqr = numpy.average(residuals * residuals)
[eb4f2fd]2052       
[f32d144]2053        self._plot_residuals(page_id=page_id, data=data_copy,
[f64a4b7]2054                             fid=fid,
[62f851f]2055                             weight=weight, index=index)
[ca2bdae]2056       
[f72333f]2057        return chisqr
2058   
[f64a4b7]2059    def _plot_residuals(self, page_id, weight, fid=None,
[f32d144]2060                        data=None, index=None):
[2296316]2061        """
2062        Plot the residuals
2063       
2064        :param data: data
[f32d144]2065        :param index: index array (bool)
[2296316]2066        : Note: this is different from the residuals in cal_chisqr()
2067        """
[ca2bdae]2068        data_copy = deepcopy(data)
[2296316]2069        # Get data: data I, theory I, and data dI in order
[ca2bdae]2070        if data_copy.__class__.__name__ == "Data2D":
[2296316]2071            # build residuals
2072            residuals = Data2D()
2073            #residuals.copy_from_datainfo(data)
2074            # Not for trunk the line below, instead use the line above
[ca2bdae]2075            data_copy.clone_without_data(len(data_copy.data), residuals)
[2296316]2076            residuals.data = None
[f32d144]2077            fn = data_copy.data
[ca2bdae]2078            theory_data = self.page_finder[page_id].get_theory_data(fid=data_copy.id)
[f32d144]2079            gn = theory_data.data
[9fa4a4d]2080            if weight == None:
2081                en = data_copy.err_data
2082            else:
2083                en = weight
[f32d144]2084            residuals.data = (fn - gn) / en
2085            residuals.qx_data = data_copy.qx_data
2086            residuals.qy_data = data_copy.qy_data
2087            residuals.q_data = data_copy.q_data
2088            residuals.err_data = numpy.ones(len(residuals.data))
[2296316]2089            residuals.xmin = min(residuals.qx_data)
2090            residuals.xmax = max(residuals.qx_data)
2091            residuals.ymin = min(residuals.qy_data)
2092            residuals.ymax = max(residuals.qy_data)
[f32d144]2093            residuals.q_data = data_copy.q_data
[ca2bdae]2094            residuals.mask = data_copy.mask
[2296316]2095            residuals.scale = 'linear'
2096            # check the lengths
2097            if len(residuals.data) != len(residuals.q_data):
2098                return
2099        else:
2100            # 1 d theory from model_thread is only in the range of index
[ca2bdae]2101            if data_copy.dy == None or data_copy.dy == []:
2102                dy = numpy.ones(len(data_copy.y))
[2296316]2103            else:
[62f851f]2104                if weight == None:
[90a1440]2105                    dy = numpy.ones(len(data_copy.y))
[f32d144]2106                ## Set consitently w/AbstractFitengine:
[2296316]2107                ## But this should be corrected later.
[90a1440]2108                else:
[f32d144]2109                    dy = weight
2110                dy[dy == 0] = 1
2111            fn = data_copy.y[index]
[ca2bdae]2112            theory_data = self.page_finder[page_id].get_theory_data(fid=data_copy.id)
[2296316]2113            gn = theory_data.y
2114            en = dy[index]
2115            # build residuals
2116            residuals = Data1D()
[ffdfd23]2117            try:
2118                residuals.y = (fn - gn) / en
2119            except:
2120                msg = "ResidualPlot Error: different # of data points in theory"
2121                wx.PostEvent(self.parent, StatusEvent(status=msg, info="error"))
2122                residuals.y = (fn - gn[index]) / en
[ca2bdae]2123            residuals.x = data_copy.x[index]
[2296316]2124            residuals.dy = numpy.ones(len(residuals.y))
2125            residuals.dx = None
2126            residuals.dxl = None
2127            residuals.dxw = None
2128            residuals.ytransform = 'y'
2129            # For latter scale changes
2130            residuals.xaxis('\\rm{Q} ', 'A^{-1}')
2131            residuals.yaxis('\\rm{Residuals} ', 'normalized')
[940aca7]2132        theory_name = str(theory_data.name.split()[0])
[2296316]2133        new_plot = residuals
[940aca7]2134        new_plot.name = "Residuals for " + str(theory_name) + "[" +\
2135                        str(data.name) +"]"
[2296316]2136        ## allow to highlight data when plotted
2137        new_plot.interactive = True
2138        ## when 2 data have the same id override the 1 st plotted
[940aca7]2139        new_plot.id = "res" + str(data_copy.id) + str(theory_name)
[2296316]2140        ##group_id specify on which panel to plot this data
[5e48acb]2141        group_id = self.page_finder[page_id].get_graph_id()
2142        if group_id == None:
2143            group_id = data.group_id
[f32d144]2144        new_plot.group_id = "res" + str(group_id)
[2296316]2145        #new_plot.is_data = True
2146        ##post data to plot
[f32d144]2147        title = new_plot.name
2148        self.page_finder[page_id].set_residuals(residuals=new_plot,
2149                                                fid=data.id)
[2de278d9]2150        self.parent.update_theory(data_id=data.id, theory=new_plot)
[f64a4b7]2151        batch_on = self.fit_panel.get_page_by_id(page_id).batch_on
[5cf1e5b]2152        if not batch_on:
2153            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title=title))
Note: See TracBrowser for help on using the repository browser.