source: sasview/sansview/perspectives/fitting/models.py @ d171299

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 d171299 was 44bbf6a, checked in by Gervaise Alina <gervyh@…>, 16 years ago

panel 2d changed for dispersion display

  • Property mode set to 100644
File size: 7.8 KB
Line 
1import wx
2import imp
3import os,sys,math
4import os.path
5
6(ModelEvent, EVT_MODEL) = wx.lib.newevent.NewEvent()
7
8
9   
10   
11def log(message):
12    print message
13    out = open("plugins.log", 'a')
14    out.write("%10g%s\n" % (time.clock(), message))
15    out.close()
16
17def findModels():
18    print "looking for models"
19    if os.path.isdir('plugins'):
20        return _findModels('plugins')
21    return []
22   
23def _findModels(dir):
24    # List of plugin objects
25    plugins = []
26    # Go through files in plug-in directory
27    try:
28        list = os.listdir(dir)
29        for item in list:
30            toks = os.path.splitext(os.path.basename(item))
31            if toks[1]=='.py' and not toks[0]=='__init__':
32                name = toks[0]
33           
34                path = [os.path.abspath(dir)]
35                file = None
36                try:
37                    (file, path, info) = imp.find_module(name, path)
38                    module = imp.load_module( name, file, item, info )
39                    if hasattr(module, "Model"):
40                        try:
41                            plugins.append(module.Model)
42                        except:
43                            log("Error accessing Model in %s\n  %s" % (name, sys.exc_value))
44                except:
45                    log("Error accessing Model in %s\n  %s" % (name, sys.exc_value))
46                finally:
47                    if not file==None:
48                        file.close()
49    except:
50        pass
51    return plugins
52class ModelManager:
53   
54    ## Dictionary of models
55    model_list = {}
56    indep_model_list = {}
57    model_list_box = {}
58    custom_models={}
59    plugins=[]
60    indep_model=[]
61    ## Event owner
62    event_owner = None
63   
64    def _getModelList(self):
65        """
66            List of models we want to make available by default
67            for this application
68           
69            @param id: first event ID to register the menu events with
70            @return: the next free event ID following the new menu events
71        """
72        self.model_list = {}
73        self.model_list_box = {}
74       
75       
76        from sans.models.SphereModel import SphereModel
77        self.model_list[str(wx.NewId())] =  SphereModel
78       
79        from sans.models.CylinderModel import CylinderModel
80        self.model_list[str(wx.NewId())] =CylinderModel
81     
82        from sans.models.CoreShellModel import CoreShellModel
83        self.model_list[str(wx.NewId())] = CoreShellModel
84       
85        from sans.models.CoreShellCylinderModel import CoreShellCylinderModel
86        self.model_list[str(wx.NewId())] =CoreShellCylinderModel
87       
88        from sans.models.EllipticalCylinderModel import EllipticalCylinderModel
89        self.model_list[str(wx.NewId())] =EllipticalCylinderModel
90       
91        from sans.models.EllipsoidModel import EllipsoidModel
92        self.model_list[str(wx.NewId())] = EllipsoidModel
93       
94        from sans.guitools.LineModel import LineModel
95        self.model_list[str(wx.NewId())]  = LineModel
96       
97       
98        model_info="shape-independent models"
99       
100        from sans.models.BEPolyelectrolyte import BEPolyelectrolyte
101        self.indep_model.append(BEPolyelectrolyte )
102       
103        from sans.models.DABModel import DABModel
104        self.indep_model.append(DABModel )
105       
106        from sans.models.DebyeModel import DebyeModel
107        self.indep_model.append(DebyeModel )
108       
109        from sans.models.FractalModel import FractalModel
110        class FractalAbsModel(FractalModel):
111            def _Fractal(self, x):
112                return FractalModel._Fractal(self, math.fabs(x))
113        self.indep_model.append(FractalAbsModel)
114       
115        from sans.models.LorentzModel import LorentzModel
116        self.indep_model.append( LorentzModel) 
117           
118        from sans.models.PowerLawModel import PowerLawModel
119        class PowerLawAbsModel(PowerLawModel):
120            def _PowerLaw(self, x):
121                try:
122                    return PowerLawModel._PowerLaw(self, math.fabs(x))
123                except:
124                    print sys.exc_value 
125        self.indep_model.append( PowerLawAbsModel )
126        from sans.models.TeubnerStreyModel import TeubnerStreyModel
127        self.indep_model.append(TeubnerStreyModel )
128   
129        #Looking for plugins
130        self.plugins = findModels()
131       
132        return 0
133
134   
135    def populate_menu(self, modelmenu, event_owner):
136        """
137            Populate a menu with our models
138           
139            @param id: first menu event ID to use when binding the menu events
140            @param modelmenu: wx.Menu object to populate
141            @param event_owner: wx object to bind the menu events to
142            @return: the next free event ID following the new menu events
143        """
144        self._getModelList()
145        self.event_owner = event_owner
146        shape_submenu= wx.Menu() 
147        indep_submenu = wx.Menu()
148        added_models = wx.Menu()
149        for id_str,value in self.model_list.iteritems():
150            item = self.model_list[id_str]()
151            name = item.__class__.__name__
152            if hasattr(item, "name"):
153                name = item.name
154            self.model_list_box[name] =value
155            shape_submenu.Append(int(id_str), name, name)
156            wx.EVT_MENU(event_owner, int(id_str), self._on_model)
157        modelmenu.AppendMenu(wx.NewId(), "Shapes...", shape_submenu, "List of shape-based models")
158        id = wx.NewId()
159        if len(self.indep_model_list) == 0:
160            for items in self.indep_model:
161                #if item not in self.indep_model_list.values():
162                    #self.indep_model_list[str(id)] = item
163                self.model_list[str(id)]=items
164                item=items()
165                name = item.__class__.__name__
166                if hasattr(item, "name"):
167                    name = item.name
168                indep_submenu.Append(id,name, name)
169                self.model_list_box[name] =items
170                wx.EVT_MENU(event_owner, int(id), self._on_model)
171                id = wx.NewId()         
172        modelmenu.AppendMenu(wx.NewId(), "Shape-independent...", indep_submenu, "List of shape-independent models")
173        id = wx.NewId()
174        if len(self.custom_models) == 0:
175            for items in self.plugins:
176                #if item not in self.custom_models.values():
177                    #self.custom_models[str(id)] = item
178                self.model_list[str(id)]=items
179                name = items.__name__
180                if hasattr(items, "name"):
181                    name = items.name
182                added_models.Append(id, name, name)
183                self.model_list_box[name] =items
184                wx.EVT_MENU(event_owner, int(id), self._on_model)
185                id = wx.NewId()
186        modelmenu.AppendMenu(wx.NewId(),"Added models...", added_models, "List of additional models")
187        return 0
188   
189    def _on_model(self, evt):
190        """
191            React to a model menu event
192            @param event: wx menu event
193        """
194        if str(evt.GetId()) in self.model_list.keys():
195            # Notify the application manager that a new model has been set
196            #self.app_manager.set_model(self.model_list[str(evt.GetId())]())
197           
198            #TODO: post a model event to update all panels that need
199            #evt = ModelEvent(model=self.model_list[str(evt.GetId())]())
200           
201            model = self.model_list[str(evt.GetId())]
202            evt = ModelEvent(model= model )
203            wx.PostEvent(self.event_owner, evt)
204       
205    def get_model_list(self):   
206        """ @ return dictionary of models for fitpanel use """
207        return self.model_list_box
208   
209   
210   
211 
Note: See TracBrowser for help on using the repository browser.