← Previous Change
Next Change →

fitting

Timestamp:

Mar 16, 2016 11:22:41 AM (8 years ago)

Author:

ajj

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

e5ea2d0

Parents:

50a77df

Message:

SasView? loads models from sasmodels. Uses model attributes to populate various lists and dicts. Fix function name typo. Remove some redundant comments.

Location:

src/sas/sasgui/perspectives/fitting

Files:

: 2 edited

fitpanel.py (modified) (1 diff)
models.py (modified) (8 diffs)

Legend:

: Unmodified
: Added
: Removed

src/sas/sasgui/perspectives/fitting/fitpanel.py

rd85c194	rf66d9d1
120	120	"""
121	121	"""
122		temp = self.menu_mng.pulgins_reset()
	122	temp = self.menu_mng.plugins_reset()
123	123	if len(temp):
124	124	self.model_list_box = temp

src/sas/sasgui/perspectives/fitting/models.py

-                      r50a77df
+                      rf66d9d1
 import py_compile
 import shutil
 from sas.guiframe.events import StatusEvent
+from sas.sasgui.guiframe.events import StatusEvent
 # Explicitly import from the pluginmodel module so that py2exe
 # places it in the distribution. The Model1DPlugin class is used
 …
 from sas.models.BaseComponent import BaseComponent
 from sas.sasgui.guiframe.CategoryInstaller import CategoryInstaller
 from sasmodels.sasview_model import make_class
+import sasmodels.core
 PLUGIN_DIR = 'plugin_models'
 …
     ## dictionary of structure factor models
     struct_factor_dict = {}
-    ##list of shape models -- this is superseded by categories
-#    shape_list = []
-    ## shape independent model list-- this is superseded by categories
-#    shape_indep_list = []
     ##list of structure factors
     struct_list = []
 …
         """
-        ## NOTE: as of April 26, 2014, as part of first pass on fixing categories,
-        ## all the appends to shape_list or shape_independent_list are
-        ## commented out.  They should be possible to remove.  They are in
-        ## fact a "category" of model whereas the other list are actually
-        ## "attributes" of a model.  In other words is it a structure factor
-        ## that can be used against a form factor, is it a form factor that is
-        ## knows how to be multiplied by a structure factor, does it have user
-        ## defined number of parameters, etc.
-        ##
-        ## We hope this whole list will be superseded by the new C models
-        ## structure where each model will provide a method to interrogate it
-        ## about its "attributes" -- then this long list becomes a loop reading
-        ## each model in the category list to populate the "attribute"lists.
-        ## We should also refactor the whole category vs attribute list
-        ## structure when doing this as now the attribute lists think they are
-        ## also category lists.
-        ##
-        ##   -PDB  April 26, 2014
         # regular model names only
+        base_message = "Unable to load model {0}"
         self.model_name_list = []
         #Build list automagically from sasmodels package
+        import pkgutil
+        from importlib import import_module
+        import sasmodels.models
+        for importer,modname,ispkg in pkgutil.iter_modules(sasmodels.models.__path__):
+            if not ispkg:
+                module = import_module('.'+modname,'sasmodels.models')
+                self.model_dictionary[module.oldname] = make_class(module, dtype='single',namestyle='oldname')
+                self.model_name_list.append(module.oldname)
+        ### NASTY HACK - Comment out most of these lines as models are added to sasmodels.
+        ### Still need the lines that put models into multiplication etc
+        try:
+        #     # from sas.models.SphereModel import SphereModel
+        #     from sasmodels.models import sphere
+        #     SphereModel = make_class(sphere, dtype='single',namestyle='oldname')
+        #     self.model_dictionary[SphereModel.__name__] = SphereModel
+        #     #        self.shape_list.append(SphereModel)
+             self.multiplication_factor.append(SphereModel)
+        #     self.model_name_list.append(SphereModel.__name__)
+        except:
+            pass
+        try:
+            from sas.models.BinaryHSModel import BinaryHSModel
+            self.model_dictionary[BinaryHSModel.__name__] = BinaryHSModel
+            #        self.shape_list.append(BinaryHSModel)
+            self.model_name_list.append(BinaryHSModel.__name__)
+        except:
+            logging.error(base_message.format(BinaryHSModel.__name__))
+        try:
+            from sas.models.FuzzySphereModel import FuzzySphereModel
+            self.model_dictionary[FuzzySphereModel.__name__] = FuzzySphereModel
+            #        self.shape_list.append(FuzzySphereModel)
+            self.multiplication_factor.append(FuzzySphereModel)
+            self.model_name_list.append(FuzzySphereModel.__name__)
+        except:
+            logging.error(base_message.format(FuzzySphereModel.__name__))
+        try:
+            from sas.models.RaspBerryModel import RaspBerryModel
+            self.model_dictionary[RaspBerryModel.__name__] = RaspBerryModel
+            #        self.shape_list.append(RaspBerryModel)
+            self.model_name_list.append(RaspBerryModel.__name__)
+        except:
+            logging.error(base_message.format(RaspBerryModel.__name__))
+        try:
+            from sas.models.CoreShellModel import CoreShellModel
+            self.model_dictionary[CoreShellModel.__name__] = CoreShellModel
+            #        self.shape_list.append(CoreShellModel)
+            self.multiplication_factor.append(CoreShellModel)
+            self.model_name_list.append(CoreShellModel.__name__)
+        except:
+            logging.error(base_message.format(CoreShellModel.__name__))
+        try:
+            from sas.models.Core2ndMomentModel import Core2ndMomentModel
+            self.model_dictionary[Core2ndMomentModel.__name__] = Core2ndMomentModel
+            #        self.shape_list.append(Core2ndMomentModel)
+            self.model_name_list.append(Core2ndMomentModel.__name__)
+        except:
+            logging.error(base_message.format(Core2ndMomentModel.__name__))
+        try:
+            from sas.models.CoreMultiShellModel import CoreMultiShellModel
+            self.model_dictionary[CoreMultiShellModel.__name__] = CoreMultiShellModel
+            #        self.shape_list.append(CoreMultiShellModel)
+            self.multiplication_factor.append(CoreMultiShellModel)
+            self.multi_func_list.append(CoreMultiShellModel)
+        except:
+            logging.error(base_message.format(CoreMultiShellModel.__name__))
+        try:
+            from sas.models.VesicleModel import VesicleModel
+            self.model_dictionary[VesicleModel.__name__] = VesicleModel
+            #        self.shape_list.append(VesicleModel)
+            self.multiplication_factor.append(VesicleModel)
+            self.model_name_list.append(VesicleModel.__name__)
+        except:
+            logging.error(base_message.format(VesicleModel.__name__))
+        try:
+            from sas.models.MultiShellModel import MultiShellModel
+            self.model_dictionary[MultiShellModel.__name__] = MultiShellModel
+            #        self.shape_list.append(MultiShellModel)
+            self.multiplication_factor.append(MultiShellModel)
+            self.model_name_list.append(MultiShellModel.__name__)
+        except:
+            logging.error(base_message.format(MultiShellModel.__name__))
+        try:
+            from sas.models.OnionExpShellModel import OnionExpShellModel
+            self.model_dictionary[OnionExpShellModel.__name__] = OnionExpShellModel
+            #        self.shape_list.append(OnionExpShellModel)
+            self.multiplication_factor.append(OnionExpShellModel)
+            self.multi_func_list.append(OnionExpShellModel)
+        except:
+            logging.error(base_message.format(OnionExpShellModel.__name__))
+        try:
+            from sas.models.SphericalSLDModel import SphericalSLDModel
+            self.model_dictionary[SphericalSLDModel.__name__] = SphericalSLDModel
+            #        self.shape_list.append(SphericalSLDModel)
+            self.multiplication_factor.append(SphericalSLDModel)
+            self.multi_func_list.append(SphericalSLDModel)
+        except:
+            logging.error(base_message.format(SphericalSLDModel.__name__))
+        try:
+            from sas.models.LinearPearlsModel import LinearPearlsModel
+            self.model_dictionary[LinearPearlsModel.__name__] = LinearPearlsModel
+            #        self.shape_list.append(LinearPearlsModel)
+            self.model_name_list.append(LinearPearlsModel.__name__)
+        except:
+            logging.error(base_message.format(LinearPearlsModel.__name__))
+        try:
+            from sas.models.PearlNecklaceModel import PearlNecklaceModel
+            self.model_dictionary[PearlNecklaceModel.__name__] = PearlNecklaceModel
+            #        self.shape_list.append(PearlNecklaceModel)
+            self.model_name_list.append(PearlNecklaceModel.__name__)
+        except:
+            logging.error(base_message.format(PearlNecklaceModel.__name__))
+        try:
+            from sas.models.CylinderModel import CylinderModel
+            self.model_dictionary[CylinderModel.__name__] = CylinderModel
+            #        self.shape_list.append(CylinderModel)
+            self.multiplication_factor.append(CylinderModel)
+            self.model_name_list.append(CylinderModel.__name__)
+        except:
+            logging.error(base_message.format(CylinderModel.__name__))
+        try:
+            from sas.models.CoreShellCylinderModel import CoreShellCylinderModel
+            self.model_dictionary[CoreShellCylinderModel.__name__] = CoreShellCylinderModel
+            #        self.shape_list.append(CoreShellCylinderModel)
+            self.multiplication_factor.append(CoreShellCylinderModel)
+            self.model_name_list.append(CoreShellCylinderModel.__name__)
+        except:
+            logging.error(base_message.format(CoreShellCylinderModel.__name__))
+        try:
+            from sas.models.CoreShellBicelleModel import CoreShellBicelleModel
+            self.model_dictionary[CoreShellBicelleModel.__name__] = CoreShellBicelleModel
+            #        self.shape_list.append(CoreShellBicelleModel)
+            self.multiplication_factor.append(CoreShellBicelleModel)
+            self.model_name_list.append(CoreShellBicelleModel.__name__)
+        except:
+            logging.error(base_message.format(CoreShellBicelleModel.__name__))
+        try:
+            from sas.models.HollowCylinderModel import HollowCylinderModel
+            self.model_dictionary[HollowCylinderModel.__name__] = HollowCylinderModel
+            #        self.shape_list.append(HollowCylinderModel)
+            self.multiplication_factor.append(HollowCylinderModel)
+            self.model_name_list.append(HollowCylinderModel.__name__)
+        except:
+            logging.error(base_message.format(HollowCylinderModel.__name__))
+        try:
+            from sas.models.FlexibleCylinderModel import FlexibleCylinderModel
+            self.model_dictionary[FlexibleCylinderModel.__name__] = FlexibleCylinderModel
+            #        self.shape_list.append(FlexibleCylinderModel)
+            self.model_name_list.append(FlexibleCylinderModel.__name__)
+        except:
+            logging.error(base_message.format(FlexibleCylinderModel.__name__))
+        try:
+            from sas.models.FlexCylEllipXModel import FlexCylEllipXModel
+            self.model_dictionary[FlexCylEllipXModel.__name__] = FlexCylEllipXModel
+            #        self.shape_list.append(FlexCylEllipXModel)
+            self.model_name_list.append(FlexCylEllipXModel.__name__)
+        except:
+            logging.error(base_message.format(FlexCylEllipXModel.__name__))
+        try:
+            from sas.models.StackedDisksModel import StackedDisksModel
+            self.model_dictionary[StackedDisksModel.__name__] = StackedDisksModel
+            #        self.shape_list.append(StackedDisksModel)
+            self.multiplication_factor.append(StackedDisksModel)
+            self.model_name_list.append(StackedDisksModel.__name__)
+        except:
+            logging.error(base_message.format(StackedDisksModel.__name__))
+        try:
+            from sas.models.ParallelepipedModel import ParallelepipedModel
+            self.model_dictionary[ParallelepipedModel.__name__] = ParallelepipedModel
+            #        self.shape_list.append(ParallelepipedModel)
+            self.multiplication_factor.append(ParallelepipedModel)
+            self.model_name_list.append(ParallelepipedModel.__name__)
+        except:
+            logging.error(base_message.format(ParallelepipedModel.__name__))
+        try:
+            from sas.models.CSParallelepipedModel import CSParallelepipedModel
+            self.model_dictionary[CSParallelepipedModel.__name__] = CSParallelepipedModel
+            #        self.shape_list.append(CSParallelepipedModel)
+            self.multiplication_factor.append(CSParallelepipedModel)
+            self.model_name_list.append(CSParallelepipedModel.__name__)
+        except:
+            logging.error(base_message.format(CSParallelepipedModel.__name__))
+        try:
+            from sas.models.EllipticalCylinderModel import EllipticalCylinderModel
+            self.model_dictionary[EllipticalCylinderModel.__name__] = EllipticalCylinderModel
+            #        self.shape_list.append(EllipticalCylinderModel)
+            self.multiplication_factor.append(EllipticalCylinderModel)
+            self.model_name_list.append(EllipticalCylinderModel.__name__)
+        except:
+            logging.error(base_message.format(EllipticalCylinderModel.__name__))
+        try:
+            from sas.models.BarBellModel import BarBellModel
+            self.model_dictionary[BarBellModel.__name__] = BarBellModel
+            #       self.shape_list.append(BarBellModel)
+            self.model_name_list.append(BarBellModel.__name__)
+        except:
+            logging.error(base_message.format(BarBellModel.__name__))
+        try:
+            from sas.models.CappedCylinderModel import CappedCylinderModel
+            self.model_dictionary[CappedCylinderModel.__name__] = CappedCylinderModel
+            #       self.shape_list.append(CappedCylinderModel)
+            self.model_name_list.append(CappedCylinderModel.__name__)
+        except:
+            logging.error(base_message.format(CappedCylinderModel.__name__))
+        try:
+            from sas.models.EllipsoidModel import EllipsoidModel
+            self.model_dictionary[EllipsoidModel.__name__] = EllipsoidModel
+            #        self.shape_list.append(EllipsoidModel)
+            self.multiplication_factor.append(EllipsoidModel)
+            self.model_name_list.append(EllipsoidModel.__name__)
+        except:
+            logging.error(base_message.format(EllipsoidModel.__name__))
+        try:
+            from sas.models.CoreShellEllipsoidModel import CoreShellEllipsoidModel
+            self.model_dictionary[CoreShellEllipsoidModel.__name__] = CoreShellEllipsoidModel
+            #        self.shape_list.append(CoreShellEllipsoidModel)
+            self.multiplication_factor.append(CoreShellEllipsoidModel)
+            self.model_name_list.append(CoreShellEllipsoidModel.__name__)
+        except:
+            logging.error(base_message.format(CoreShellEllipsoidModel.__name__))
+        try:
+            from sas.models.CoreShellEllipsoidXTModel import CoreShellEllipsoidXTModel
+            self.model_dictionary[CoreShellEllipsoidXTModel.__name__] = CoreShellEllipsoidXTModel
+            #        self.shape_list.append(CoreShellEllipsoidXTModel)
+            self.multiplication_factor.append(CoreShellEllipsoidXTModel)
+            self.model_name_list.append(CoreShellEllipsoidXTModel.__name__)
+        except:
+            logging.error(base_message.format(CoreShellEllipsoidXTModel.__name__))
+        try:
+            from sas.models.TriaxialEllipsoidModel import TriaxialEllipsoidModel
+            self.model_dictionary[TriaxialEllipsoidModel.__name__] = TriaxialEllipsoidModel
+            #        self.shape_list.append(TriaxialEllipsoidModel)
+            self.multiplication_factor.append(TriaxialEllipsoidModel)
+            self.model_name_list.append(TriaxialEllipsoidModel.__name__)
+        except:
+            logging.error(base_message.format(TriaxialEllipsoidModel.__name__))
+        try:
+            from sas.models.LamellarModel import LamellarModel
+            self.model_dictionary[LamellarModel.__name__] = LamellarModel
+            #        self.shape_list.append(LamellarModel)
+            self.model_name_list.append(LamellarModel.__name__)
+        except:
+            logging.error(base_message.format(LamellarModel.__name__))
+        try:
+            from sas.models.LamellarFFHGModel import LamellarFFHGModel
+            self.model_dictionary[LamellarFFHGModel.__name__] = LamellarFFHGModel
+            #        self.shape_list.append(LamellarFFHGModel)
+            self.model_name_list.append(LamellarFFHGModel.__name__)
+        except:
+            logging.error(base_message.format(LamellarFFHGModel.__name__))
+        try:
+            from sas.models.LamellarPSModel import LamellarPSModel
+            self.model_dictionary[LamellarPSModel.__name__] = LamellarPSModel
+            #        self.shape_list.append(LamellarPSModel)
+            self.model_name_list.append(LamellarPSModel.__name__)
+        except:
+            logging.error(base_message.format(LamellarPSModel.__name__))
+        try:
+            from sas.models.LamellarPSHGModel import LamellarPSHGModel
+            self.model_dictionary[LamellarPSHGModel.__name__] = LamellarPSHGModel
+            #        self.shape_list.append(LamellarPSHGModel)
+            self.model_name_list.append(LamellarPSHGModel.__name__)
+        except:
+            logging.error(base_message.format(LamellarPSHGModel.__name__))
+        try:
+            from sas.models.LamellarPCrystalModel import LamellarPCrystalModel
+            self.model_dictionary[LamellarPCrystalModel.__name__] = LamellarPCrystalModel
+            #        self.shape_list.append(LamellarPCrystalModel)
+            self.model_name_list.append(LamellarPCrystalModel.__name__)
+        except:
+            logging.error(base_message.format(LamellarPCrystalModel.__name__))
+        try:
+            from sas.models.SCCrystalModel import SCCrystalModel
+            self.model_dictionary[SCCrystalModel.__name__] = SCCrystalModel
+            #        self.shape_list.append(SCCrystalModel)
+            self.model_name_list.append(SCCrystalModel.__name__)
+        except:
+            logging.error(base_message.format(SCCrystalModel.__name__))
+        try:
+            from sas.models.FCCrystalModel import FCCrystalModel
+            self.model_dictionary[FCCrystalModel.__name__] = FCCrystalModel
+            #        self.shape_list.append(FCCrystalModel)
+            self.model_name_list.append(FCCrystalModel.__name__)
+        except:
+            logging.error(base_message.format(FCCrystalModel.__name__))
+        try:
+            from sas.models.BCCrystalModel import BCCrystalModel
+            self.model_dictionary[BCCrystalModel.__name__] = BCCrystalModel
+            #        self.shape_list.append(BCCrystalModel)
+            self.model_name_list.append(BCCrystalModel.__name__)
+        except:
+            logging.error(base_message.format(BCCrystalModel.__name__))
+        ## Structure factor
+        try:
+            from sas.models.SquareWellStructure import SquareWellStructure
+            self.model_dictionary[SquareWellStructure.__name__] = SquareWellStructure
+            self.struct_list.append(SquareWellStructure)
+            self.model_name_list.append(SquareWellStructure.__name__)
+        except:
+            logging.error(base_message.format(SquareWellStructure.__name__))
+        try:
+            from sas.models.HardsphereStructure import HardsphereStructure
+            self.model_dictionary[HardsphereStructure.__name__] = HardsphereStructure
+            self.struct_list.append(HardsphereStructure)
+            self.model_name_list.append(HardsphereStructure.__name__)
+        except:
+            logging.error(base_message.format(HardsphereStructure.__name__))
+        try:
+            from sas.models.StickyHSStructure import StickyHSStructure
+            self.model_dictionary[StickyHSStructure.__name__] = StickyHSStructure
+            self.struct_list.append(StickyHSStructure)
+            self.model_name_list.append(StickyHSStructure.__name__)
+        except:
+            logging.error(base_message.format(StickyHSStructure.__name__))
+        try:
+            from sas.models.HayterMSAStructure import HayterMSAStructure
+            self.model_dictionary[HayterMSAStructure.__name__] = HayterMSAStructure
+            self.struct_list.append(HayterMSAStructure)
+            self.model_name_list.append(HayterMSAStructure.__name__)
+        except:
+            logging.error(base_message.format(HayterMSAStructure.__name__))
+        ##shape-independent models
+        try:
+            from sas.models.PowerLawAbsModel import PowerLawAbsModel
+            self.model_dictionary[PowerLawAbsModel.__name__] = PowerLawAbsModel
+            #        self.shape_indep_list.append(PowerLawAbsModel)
+            self.model_name_list.append(PowerLawAbsModel.__name__)
+        except:
+            logging.error(base_message.format(PowerLawAbsModel.__name__))
+        try:
+            from sas.models.BEPolyelectrolyte import BEPolyelectrolyte
+            self.model_dictionary[BEPolyelectrolyte.__name__] = BEPolyelectrolyte
+            #        self.shape_indep_list.append(BEPolyelectrolyte)
+            self.model_name_list.append(BEPolyelectrolyte.__name__)
+            self.form_factor_dict[str(wx.NewId())] = [SphereModel]
+        except:
+            logging.error(base_message.format(BEPolyelectrolyte.__name__))
+        try:
+            from sas.models.BroadPeakModel import BroadPeakModel
+            self.model_dictionary[BroadPeakModel.__name__] = BroadPeakModel
+            #        self.shape_indep_list.append(BroadPeakModel)
+            self.model_name_list.append(BroadPeakModel.__name__)
+        except:
+            logging.error(base_message.format(BroadPeakModel.__name__))
+        try:
+            from sas.models.CorrLengthModel import CorrLengthModel
+            self.model_dictionary[CorrLengthModel.__name__] = CorrLengthModel
+            #        self.shape_indep_list.append(CorrLengthModel)
+            self.model_name_list.append(CorrLengthModel.__name__)
+        except:
+            logging.error(base_message.format(CorrLengthModel.__name__))
+        try:
+            from sas.models.DABModel import DABModel
+            self.model_dictionary[DABModel.__name__] = DABModel
+            #        self.shape_indep_list.append(DABModel)
+            self.model_name_list.append(DABModel.__name__)
+        except:
+            logging.error(base_message.format(DABModel.__name__))
+        try:
+            from sas.models.DebyeModel import DebyeModel
+            self.model_dictionary[DebyeModel.__name__] = DebyeModel
+            #        self.shape_indep_list.append(DebyeModel)
+            self.model_name_list.append(DebyeModel.__name__)
+        except:
+            logging.error(base_message.format(DebyeModel.__name__))
+        try:
+            from sas.models.FractalModel import FractalModel
+            self.model_dictionary[FractalModel.__name__] = FractalModel
+            #        self.shape_indep_list.append(FractalModel)
+            self.model_name_list.append(FractalModel.__name__)
+        except:
+            logging.error(base_message.format(FractalModel.__name__))
+        try:
+            from sas.models.FractalCoreShellModel import FractalCoreShellModel
+            self.model_dictionary[FractalCoreShellModel.__name__] = FractalCoreShellModel
+            #        self.shape_indep_list.append(FractalCoreShellModel)
+            self.model_name_list.append(FractalCoreShellModel.__name__)
+        except:
+            logging.error(base_message.format(FractalCoreShellModel.__name__))
+        try:
+            from sas.models.GaussLorentzGelModel import GaussLorentzGelModel
+            self.model_dictionary[GaussLorentzGelModel.__name__] = GaussLorentzGelModel
+            #        self.shape_indep_list.append(GaussLorentzGelModel)
+            self.model_name_list.append(GaussLorentzGelModel.__name__)
+        except:
+            logging.error(base_message.format(GaussLorentzGelModel.__name__))
+        try:
+            from sas.models.GuinierModel import GuinierModel
+            self.model_dictionary[GuinierModel.__name__] = GuinierModel
+            #        self.shape_indep_list.append(GuinierModel)
+            self.model_name_list.append(GuinierModel.__name__)
+        except:
+            logging.error(base_message.format(GuinierModel.__name__))
+        try:
+            from sas.models.GuinierPorodModel import GuinierPorodModel
+            self.model_dictionary[GuinierPorodModel.__name__] = GuinierPorodModel
+            #        self.shape_indep_list.append(GuinierPorodModel)
+            self.model_name_list.append(GuinierPorodModel.__name__)
+        except:
+            logging.error(base_message.format(GuinierPorodModel.__name__))
+        try:
+            from sas.models.LorentzModel import LorentzModel
+            self.model_dictionary[LorentzModel.__name__] = LorentzModel
+            #        self.shape_indep_list.append(LorentzModel)
+            self.model_name_list.append(LorentzModel.__name__)
+        except:
+            logging.error(base_message.format(LorentzModel.__name__))
+        try:
+            from sas.models.MassFractalModel import MassFractalModel
+            self.model_dictionary[MassFractalModel.__name__] = MassFractalModel
+            #        self.shape_indep_list.append(MassFractalModel)
+            self.model_name_list.append(MassFractalModel.__name__)
+        except:
+            logging.error(base_message.format(MassFractalModel.__name__))
+        try:
+            from sas.models.MassSurfaceFractal import MassSurfaceFractal
+            self.model_dictionary[MassSurfaceFractal.__name__] = MassSurfaceFractal
+            #        self.shape_indep_list.append(MassSurfaceFractal)
+            self.model_name_list.append(MassSurfaceFractal.__name__)
+        except:
+            logging.error(base_message.format(MassSurfaceFractal.__name__))
+        try:
+            from sas.models.PeakGaussModel import PeakGaussModel
+            self.model_dictionary[PeakGaussModel.__name__] = PeakGaussModel
+            #        self.shape_indep_list.append(PeakGaussModel)
+            self.model_name_list.append(PeakGaussModel.__name__)
+        except:
+            logging.error(base_message.format(PeakGaussModel.__name__))
+        try:
+            from sas.models.PeakLorentzModel import PeakLorentzModel
+            self.model_dictionary[PeakLorentzModel.__name__] = PeakLorentzModel
+            #        self.shape_indep_list.append(PeakLorentzModel)
+            self.model_name_list.append(PeakLorentzModel.__name__)
+        except:
+            logging.error(base_message.format(PeakLorentzModel.__name__))
+        try:
+            from sas.models.Poly_GaussCoil import Poly_GaussCoil
+            self.model_dictionary[Poly_GaussCoil.__name__] = Poly_GaussCoil
+            #        self.shape_indep_list.append(Poly_GaussCoil)
+            self.model_name_list.append(Poly_GaussCoil.__name__)
+        except:
+            logging.error(base_message.format(Poly_GaussCoil.__name__))
+        try:
+            from sas.models.PolymerExclVolume import PolymerExclVolume
+            self.model_dictionary[PolymerExclVolume.__name__] = PolymerExclVolume
+            #        self.shape_indep_list.append(PolymerExclVolume)
+            self.model_name_list.append(PolymerExclVolume.__name__)
+        except:
+            logging.error(base_message.format(PolymerExclVolume.__name__))
+        try:
+            from sas.models.PorodModel import PorodModel
+            self.model_dictionary[PorodModel.__name__] = PorodModel
+            #        self.shape_indep_list.append(PorodModel)
+            self.model_name_list.append(PorodModel.__name__)
+        except:
+            logging.error(base_message.format(PorodModel.__name__))
+        try:
+            from sas.models.RPA10Model import RPA10Model
+            self.model_dictionary[RPA10Model.__name__] = RPA10Model
+            #        self.shape_indep_list.append(RPA10Model)
+            self.multi_func_list.append(RPA10Model)
+        except:
+            logging.error(base_message.format(RPA10Model.__name__))
+        try:
+            from sas.models.StarPolymer import StarPolymer
+            self.model_dictionary[StarPolymer.__name__] = StarPolymer
+            #        self.shape_indep_list.append(StarPolymer)
+            self.model_name_list.append(StarPolymer.__name__)
+        except:
+            logging.error(base_message.format(StarPolymer.__name__))
+        try:
+            from sas.models.SurfaceFractalModel import SurfaceFractalModel
+            self.model_dictionary[SurfaceFractalModel.__name__] = SurfaceFractalModel
+            #        self.shape_indep_list.append(SurfaceFractalModel)
+            self.model_name_list.append(SurfaceFractalModel.__name__)
+        except:
+            logging.error(base_message.format(SurfaceFractalModel.__name__))
+        try:
+            from sas.models.TeubnerStreyModel import TeubnerStreyModel
+            self.model_dictionary[TeubnerStreyModel.__name__] = TeubnerStreyModel
+            #        self.shape_indep_list.append(TeubnerStreyModel)
+            self.model_name_list.append(TeubnerStreyModel.__name__)
+        except:
+            logging.error(base_message.format(TeubnerStreyModel.__name__))
+        try:
+            from sas.models.TwoLorentzianModel import TwoLorentzianModel
+            self.model_dictionary[TwoLorentzianModel.__name__] = TwoLorentzianModel
+            #        self.shape_indep_list.append(TwoLorentzianModel)
+            self.model_name_list.append(TwoLorentzianModel.__name__)
+        except:
+            logging.error(base_message.format(TwoLorentzianModel.__name__))
+        try:
+            from sas.models.TwoPowerLawModel import TwoPowerLawModel
+            self.model_dictionary[TwoPowerLawModel.__name__] = TwoPowerLawModel
+            #        self.shape_indep_list.append(TwoPowerLawModel)
+            self.model_name_list.append(TwoPowerLawModel.__name__)
+        except:
+            logging.error(base_message.format(TwoPowerLawModel.__name__))
+        try:
+            from sas.models.UnifiedPowerRgModel import UnifiedPowerRgModel
+            self.model_dictionary[UnifiedPowerRgModel.__name__] = UnifiedPowerRgModel
+            #        self.shape_indep_list.append(UnifiedPowerRgModel)
+            self.multi_func_list.append(UnifiedPowerRgModel)
+        except:
+            logging.error(base_message.format(UnifiedPowerRgModel.__name__))
+        try:
+            from sas.models.LineModel import LineModel
+            self.model_dictionary[LineModel.__name__] = LineModel
+            #        self.shape_indep_list.append(LineModel)
+            self.model_name_list.append(LineModel.__name__)
+        except:
+            logging.error(base_message.format(LineModel.__name__))
+        try:
+            from sas.models.ReflectivityModel import ReflectivityModel
+            self.model_dictionary[ReflectivityModel.__name__] = ReflectivityModel
+            #        self.shape_indep_list.append(ReflectivityModel)
+            self.multi_func_list.append(ReflectivityModel)
+        except:
+            logging.error(base_message.format(ReflectivityModel.__name__))
+        try:
+            from sas.models.ReflectivityIIModel import ReflectivityIIModel
+            self.model_dictionary[ReflectivityIIModel.__name__] = ReflectivityIIModel
+            #        self.shape_indep_list.append(ReflectivityIIModel)
+            self.multi_func_list.append(ReflectivityIIModel)
+        except:
+            logging.error(base_message.format(ReflectivityIIModel.__name__))
+        try:
+            from sas.models.GelFitModel import GelFitModel
+            self.model_dictionary[GelFitModel.__name__] = GelFitModel
+            #        self.shape_indep_list.append(GelFitModel)
+            self.model_name_list.append(GelFitModel.__name__)
+        except:
+            logging.error(base_message.format(GelFitModel.__name__))
+        try:
+            from sas.models.PringlesModel import PringlesModel
+            self.model_dictionary[PringlesModel.__name__] = PringlesModel
+            #        self.shape_indep_list.append(PringlesModel)
+            self.model_name_list.append(PringlesModel.__name__)
+        except:
+            logging.error(base_message.format(PringlesModel.__name__))
+        try:
+            from sas.models.RectangularPrismModel import RectangularPrismModel
+            self.model_dictionary[RectangularPrismModel.__name__] = RectangularPrismModel
+            #        self.shape_list.append(RectangularPrismModel)
+            self.multiplication_factor.append(RectangularPrismModel)
+            self.model_name_list.append(RectangularPrismModel.__name__)
+        except:
+            logging.error(base_message.format(RectangularPrismModel.__name__))
+        try:
+            from sas.models.RectangularHollowPrismInfThinWallsModel import RectangularHollowPrismInfThinWallsModel
+            self.model_dictionary[RectangularHollowPrismInfThinWallsModel.__name__] = RectangularHollowPrismInfThinWallsModel
+            #        self.shape_list.append(RectangularHollowPrismInfThinWallsModel)
+            self.multiplication_factor.append(RectangularHollowPrismInfThinWallsModel)
+            self.model_name_list.append(RectangularHollowPrismInfThinWallsModel.__name__)
+        except:
+            logging.error(base_message.format(RectangularHollowPrismInfThinWallsModel.__name__))
+        try:
+            from sas.models.RectangularHollowPrismModel import RectangularHollowPrismModel
+            self.model_dictionary[RectangularHollowPrismModel.__name__] = RectangularHollowPrismModel
+            #        self.shape_list.append(RectangularHollowPrismModel)
+            self.multiplication_factor.append(RectangularHollowPrismModel)
+            self.model_name_list.append(RectangularHollowPrismModel.__name__)
+        except:
+            logging.error(base_message.format(RectangularHollowPrismModel.__name__))
+        try:
+            from sas.models.MicelleSphCoreModel import MicelleSphCoreModel
+            self.model_dictionary[MicelleSphCoreModel.__name__] = MicelleSphCoreModel
+            #        self.shape_list.append(MicelleSphCoreModel)
+            self.multiplication_factor.append(MicelleSphCoreModel)
+            self.model_name_list.append(MicelleSphCoreModel.__name__)
+        except:
+            logging.error(base_message.format(MicelleSphCoreModel.__name__))
+        #from sas.models.FractalO_Z import FractalO_Z
+        #self.model_dictionary[FractalO_Z.__name__] = FractalO_Z
+        #self.shape_indep_list.append(FractalO_Z)
+        #self.model_name_list.append(FractalO_Z.__name__)
+        for mod_name in sasmodels.core.list_models():
+            mod_def = sasmodels.core.load_model_info(mod_name)
+            self.model_dictionary[mod_def['oldname']] = make_class(mod_def,dtype=None,namestyle='oldname')
+            self.model_name_list.append(mod_def['oldname'])
+            if mod_def['structure_factor'] == True:
+                self.struct_list.append(self.model_dictionary[mod_def['oldname']])
+            if mod_def['variant_info'] is not None:
+                self.multi_func_list.append(self.model_dictionary[mod_def['oldname']])
+            if mod_def['ER'] is not None:
+                self.multiplication_factor.append(self.model_dictionary[mod_def['oldname']])
         #Looking for plugins
 …
             return {}
     def pulgins_reset(self):
+    def plugins_reset(self):
         """
         return a dictionary of model
 …
         return self.model_combobox.get_list()
-##   I believe the next four methods are for the old form factor GUI
-##   where the dropdown showed a list of categories which then rolled out
-##   in a second dropdown to the side. Some testing shows they indeed no longer
-##   seem to be called.  If no problems are found during testing of release we
-##   can remove this huge chunck of stuff.
-##
-##   -PDB  April 26, 2014
-#   def populate_menu(self, modelmenu, event_owner):
-#       """
-#       Populate a menu with our models
+#
-#       :param id: first menu event ID to use when binding the menu events
-#       :param modelmenu: wx.Menu object to populate
-#       :param event_owner: wx object to bind the menu events to
+#
-#       :return: the next free event ID following the new menu events
+#
-#       """
+#
-        ## Fill model lists
-#        self._getModelList()
-        ## store reference to model menu of guiframe
-#        self.modelmenu = modelmenu
-        ## guiframe reference
-#        self.event_owner = event_owner
-#        shape_submenu = wx.Menu()
-#        shape_indep_submenu = wx.Menu()
-#        structure_factor = wx.Menu()
-#        added_models = wx.Menu()
-#        multip_models = wx.Menu()
-        ## create menu with shape
-#        self._fill_simple_menu(menuinfo=["Shapes",
-#                                         shape_submenu,
-#                                         " simple shape"],
-#                         list1=self.shape_list)
-#        self._fill_simple_menu(menuinfo=["Shape-Independent",
-#                                         shape_indep_submenu,
-#                                         "List of shape-independent models"],
-#                         list1=self.shape_indep_list)
-#        self._fill_simple_menu(menuinfo=["Structure Factors",
-#                                         structure_factor,
-#                                         "List of Structure factors models"],
-#                                list1=self.struct_list)
-#        self._fill_plugin_menu(menuinfo=["Customized Models", added_models,
-#                                            "List of additional models"],
-#                                 list1=self.plugins)
-#        self._fill_menu(menuinfo=["P(Q)*S(Q)", multip_models,
-#                                  "mulplication of 2 models"],
-#                                   list1=self.multiplication_factor,
-#                                   list2=self.struct_list)
-#        return 0
-#    def _fill_plugin_menu(self, menuinfo, list1):
-#        """
-#        fill the plugin menu with costumized models
-#        """
-#        print ("got to fill plugin menu")
-#        if len(list1) == 0:
-#            id = wx.NewId()
-#            msg = "No model available check plugins.log for errors to fix problem"
-#            menuinfo[1].Append(int(id), "Empty", msg)
-#        self._fill_simple_menu(menuinfo, list1)
-#   def _fill_simple_menu(self, menuinfo, list1):
-#       """
-#       Fill the menu with list item
+#
-#       :param modelmenu: the menu to fill
-#       :param menuinfo: submenu item for the first column of this modelmenu
-#                        with info.Should be a list :
-#                        [name(string) , menu(wx.menu), help(string)]
-#       :param list1: contains item (form factor )to fill modelmenu second column
+#
-#       """
-#       if len(list1) > 0:
-#           self.model_combobox.set_list(menuinfo[0], list1)
-#            for item in list1:
-#                try:
-#                    id = wx.NewId()
-#                    struct_factor = item()
-#                    struct_name = struct_factor.__class__.__name__
-#                    if hasattr(struct_factor, "name"):
-#                        struct_name = struct_factor.name
+#
-#                    menuinfo[1].Append(int(id), struct_name, struct_name)
-#                    if not  item in self.struct_factor_dict.itervalues():
-#                        self.struct_factor_dict[str(id)] = item
-#                    wx.EVT_MENU(self.event_owner, int(id), self._on_model)
-#                except:
-#                    msg = "Error Occured: %s" % sys.exc_value
-#                    wx.PostEvent(self.event_owner, StatusEvent(status=msg))
+#
-#        id = wx.NewId()
-#        self.modelmenu.AppendMenu(id, menuinfo[0], menuinfo[1], menuinfo[2])
+#
-#    def _fill_menu(self, menuinfo, list1, list2):
-#        """
-#        Fill the menu with list item
+#
-#        :param menuinfo: submenu item for the first column of this modelmenu
-#                         with info.Should be a list :
-#                         [name(string) , menu(wx.menu), help(string)]
-#        :param list1: contains item (form factor )to fill modelmenu second column
-#        :param list2: contains item (Structure factor )to fill modelmenu
-#                third column
+#
-#        """
-#        if len(list1) > 0:
-#            self.model_combobox.set_list(menuinfo[0], list1)
+#
-#            for item in list1:
-#                form_factor = item()
-#                form_name = form_factor.__class__.__name__
-#                if hasattr(form_factor, "name"):
-#                    form_name = form_factor.name
-#                ### store form factor to return to other users
-#                newmenu = wx.Menu()
-#                if len(list2) > 0:
-#                    for model  in list2:
-#                        id = wx.NewId()
-#                        struct_factor = model()
-#                        name = struct_factor.__class__.__name__
-#                        if hasattr(struct_factor, "name"):
-#                            name = struct_factor.name
-#                        newmenu.Append(id, name, name)
-#                        wx.EVT_MENU(self.event_owner, int(id), self._on_model)
-#                        ## save form_fact and struct_fact
-#                        self.form_factor_dict[int(id)] = [form_factor,
-#                                                          struct_factor]
+#
-#                form_id = wx.NewId()
-#                menuinfo[1].AppendMenu(int(form_id), form_name,
-#                                       newmenu, menuinfo[2])
-#        id = wx.NewId()
-#        self.modelmenu.AppendMenu(id, menuinfo[0], menuinfo[1], menuinfo[2])
     def _on_model(self, evt):
         """
 …
             model = self.struct_factor_dict[str(evt.GetId())]()
-        #TODO: investigate why the following two lines were left in the code
-        #      even though the ModelEvent class doesn't exist
-        #evt = ModelEvent(model=model)
-        #wx.PostEvent(self.event_owner, evt)
     def _get_multifunc_models(self):
 …
         return self.__modelmanager.update()
     def pulgins_reset(self):
         return self.__modelmanager.pulgins_reset()
+    def plugins_reset(self):
+        return self.__modelmanager.plugins_reset()
     def populate_menu(self, modelmenu, event_owner):

Note: See TracChangeset for help on using the changeset viewer.

SasView

Changeset f66d9d1 in sasview for src/sas/sasgui/perspectives/fitting

Legend:

src/sas/sasgui/perspectives/fitting/fitpanel.py

src/sas/sasgui/perspectives/fitting/models.py

Download in other formats: