[d89f09b] | 1 | import wx |
---|
[b30f001] | 2 | import imp |
---|
[442895f] | 3 | import os,sys,math |
---|
[d89f09b] | 4 | import os.path |
---|
| 5 | |
---|
| 6 | (ModelEvent, EVT_MODEL) = wx.lib.newevent.NewEvent() |
---|
[00561739] | 7 | |
---|
[2dbb681] | 8 | |
---|
[00561739] | 9 | |
---|
| 10 | |
---|
[b30f001] | 11 | def log(message): |
---|
| 12 | print message |
---|
| 13 | out = open("plugins.log", 'a') |
---|
| 14 | out.write("%10g: %s\n" % (time.clock(), message)) |
---|
| 15 | out.close() |
---|
| 16 | |
---|
[aa92772] | 17 | def findModels(): |
---|
| 18 | print "looking for models" |
---|
[b30f001] | 19 | if os.path.isdir('plugins'): |
---|
| 20 | return _findModels('plugins') |
---|
| 21 | return [] |
---|
[aa92772] | 22 | |
---|
[1c66bc5] | 23 | def _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: |
---|
[b30f001] | 45 | log("Error accessing Model in %s\n %s" % (name, sys.exc_value)) |
---|
[1c66bc5] | 46 | finally: |
---|
| 47 | if not file==None: |
---|
| 48 | file.close() |
---|
| 49 | except: |
---|
| 50 | pass |
---|
| 51 | return plugins |
---|
[d89f09b] | 52 | class ModelManager: |
---|
| 53 | |
---|
| 54 | ## Dictionary of models |
---|
| 55 | model_list = {} |
---|
[49b7efa] | 56 | indep_model_list = {} |
---|
[d89f09b] | 57 | model_list_box = {} |
---|
[b30f001] | 58 | custom_models={} |
---|
| 59 | plugins=[] |
---|
[49b7efa] | 60 | indep_model=[] |
---|
[d89f09b] | 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 = {} |
---|
[2dbb681] | 74 | |
---|
[442895f] | 75 | |
---|
| 76 | from sans.models.SphereModel import SphereModel |
---|
[2dbb681] | 77 | self.model_list[str(wx.NewId())] = SphereModel |
---|
[442895f] | 78 | |
---|
[d89f09b] | 79 | from sans.models.CylinderModel import CylinderModel |
---|
[2dbb681] | 80 | self.model_list[str(wx.NewId())] =CylinderModel |
---|
[d89f09b] | 81 | |
---|
[442895f] | 82 | from sans.models.CoreShellModel import CoreShellModel |
---|
[2dbb681] | 83 | self.model_list[str(wx.NewId())] = CoreShellModel |
---|
[442895f] | 84 | |
---|
| 85 | from sans.models.CoreShellCylinderModel import CoreShellCylinderModel |
---|
[2dbb681] | 86 | self.model_list[str(wx.NewId())] =CoreShellCylinderModel |
---|
[442895f] | 87 | |
---|
| 88 | from sans.models.EllipticalCylinderModel import EllipticalCylinderModel |
---|
[2dbb681] | 89 | self.model_list[str(wx.NewId())] =EllipticalCylinderModel |
---|
[442895f] | 90 | |
---|
| 91 | from sans.models.EllipsoidModel import EllipsoidModel |
---|
[2dbb681] | 92 | self.model_list[str(wx.NewId())] = EllipsoidModel |
---|
[442895f] | 93 | |
---|
[d89f09b] | 94 | from sans.guitools.LineModel import LineModel |
---|
[2dbb681] | 95 | self.model_list[str(wx.NewId())] = LineModel |
---|
[442895f] | 96 | |
---|
| 97 | |
---|
[49b7efa] | 98 | model_info="shape-independent models" |
---|
[442895f] | 99 | |
---|
| 100 | from sans.models.BEPolyelectrolyte import BEPolyelectrolyte |
---|
[2dbb681] | 101 | self.indep_model.append(BEPolyelectrolyte ) |
---|
[442895f] | 102 | |
---|
| 103 | from sans.models.DABModel import DABModel |
---|
[2dbb681] | 104 | self.indep_model.append(DABModel ) |
---|
[442895f] | 105 | |
---|
| 106 | from sans.models.DebyeModel import DebyeModel |
---|
[2dbb681] | 107 | self.indep_model.append(DebyeModel ) |
---|
[442895f] | 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)) |
---|
[2dbb681] | 113 | self.indep_model.append(FractalAbsModel) |
---|
[442895f] | 114 | |
---|
| 115 | from sans.models.LorentzModel import LorentzModel |
---|
[2dbb681] | 116 | self.indep_model.append( LorentzModel) |
---|
[442895f] | 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 |
---|
[2dbb681] | 125 | self.indep_model.append( PowerLawAbsModel ) |
---|
[442895f] | 126 | from sans.models.TeubnerStreyModel import TeubnerStreyModel |
---|
[2dbb681] | 127 | self.indep_model.append(TeubnerStreyModel ) |
---|
| 128 | |
---|
[49b7efa] | 129 | #Looking for plugins |
---|
| 130 | self.plugins = findModels() |
---|
| 131 | |
---|
[d89f09b] | 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 |
---|
[b30f001] | 146 | shape_submenu= wx.Menu() |
---|
| 147 | indep_submenu = wx.Menu() |
---|
| 148 | added_models = wx.Menu() |
---|
[d89f09b] | 149 | for id_str,value in self.model_list.iteritems(): |
---|
[2dbb681] | 150 | item = self.model_list[id_str]() |
---|
| 151 | name = item.__class__.__name__ |
---|
[d89f09b] | 152 | if hasattr(item, "name"): |
---|
[2dbb681] | 153 | name = item.name |
---|
| 154 | self.model_list_box[name] =value |
---|
[b30f001] | 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() |
---|
[49b7efa] | 159 | if len(self.indep_model_list) == 0: |
---|
[2dbb681] | 160 | for items in self.indep_model: |
---|
[00561739] | 161 | #if item not in self.indep_model_list.values(): |
---|
| 162 | #self.indep_model_list[str(id)] = item |
---|
[2dbb681] | 163 | self.model_list[str(id)]=items |
---|
| 164 | item=items() |
---|
| 165 | name = item.__class__.__name__ |
---|
[00561739] | 166 | if hasattr(item, "name"): |
---|
[2dbb681] | 167 | name = item.name |
---|
[00561739] | 168 | indep_submenu.Append(id,name, name) |
---|
[2dbb681] | 169 | self.model_list_box[name] =items |
---|
[00561739] | 170 | wx.EVT_MENU(event_owner, int(id), self._on_model) |
---|
| 171 | id = wx.NewId() |
---|
[49b7efa] | 172 | modelmenu.AppendMenu(wx.NewId(), "Shape-independent...", indep_submenu, "List of shape-independent models") |
---|
| 173 | id = wx.NewId() |
---|
[b30f001] | 174 | if len(self.custom_models) == 0: |
---|
[2dbb681] | 175 | for items in self.plugins: |
---|
[00561739] | 176 | #if item not in self.custom_models.values(): |
---|
| 177 | #self.custom_models[str(id)] = item |
---|
[2dbb681] | 178 | self.model_list[str(id)]=items |
---|
[44bbf6a] | 179 | name = items.__name__ |
---|
| 180 | if hasattr(items, "name"): |
---|
| 181 | name = items.name |
---|
[00561739] | 182 | added_models.Append(id, name, name) |
---|
[2dbb681] | 183 | self.model_list_box[name] =items |
---|
[00561739] | 184 | wx.EVT_MENU(event_owner, int(id), self._on_model) |
---|
| 185 | id = wx.NewId() |
---|
[b30f001] | 186 | modelmenu.AppendMenu(wx.NewId(),"Added models...", added_models, "List of additional models") |
---|
[d89f09b] | 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 | |
---|
[49b7efa] | 201 | model = self.model_list[str(evt.GetId())] |
---|
[2dbb681] | 202 | evt = ModelEvent(model= model ) |
---|
[d89f09b] | 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 | |
---|