Changeset df7a7e3 in sasview for fittingview/src/sans/perspectives/fitting/models.py
- Timestamp:
- Sep 17, 2012 7:11:50 PM (12 years ago)
- 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:
- b71a53b
- Parents:
- cf7653d3
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
fittingview/src/sans/perspectives/fitting/models.py
rf40f743 rdf7a7e3 18 18 # as the base class of plug-in models. 19 19 from sans.models.pluginmodel import Model1DPlugin 20 from sans.models.BaseComponent import BaseComponent 20 21 21 22 PLUGIN_DIR = 'plugin_models' 23 24 def get_model_python_path(): 25 return os.path.dirname(__file__) 22 26 23 27 … … 261 265 """ 262 266 """ 267 self.model_dictionary = {} 263 268 self.stored_plugins = {} 264 269 self._getModelList() … … 283 288 284 289 """ 290 291 285 292 # regular model names only 286 293 self.model_name_list = [] 287 294 from sans.models.SphereModel import SphereModel 295 self.model_dictionary[SphereModel.__name__] = SphereModel 288 296 self.shape_list.append(SphereModel) 289 297 self.multiplication_factor.append(SphereModel) … … 291 299 292 300 from sans.models.BinaryHSModel import BinaryHSModel 301 self.model_dictionary[BinaryHSModel.__name__] = BinaryHSModel 293 302 self.shape_list.append(BinaryHSModel) 294 303 self.model_name_list.append(BinaryHSModel.__name__) 295 304 296 305 from sans.models.FuzzySphereModel import FuzzySphereModel 306 self.model_dictionary[FuzzySphereModel.__name__] = FuzzySphereModel 297 307 self.shape_list.append(FuzzySphereModel) 298 308 self.multiplication_factor.append(FuzzySphereModel) … … 300 310 301 311 from sans.models.RaspBerryModel import RaspBerryModel 312 self.model_dictionary[RaspBerryModel.__name__] = RaspBerryModel 302 313 self.shape_list.append(RaspBerryModel) 303 314 self.model_name_list.append(RaspBerryModel.__name__) 304 315 305 316 from sans.models.CoreShellModel import CoreShellModel 317 self.model_dictionary[CoreShellModel.__name__] = CoreShellModel 306 318 self.shape_list.append(CoreShellModel) 307 319 self.multiplication_factor.append(CoreShellModel) … … 309 321 310 322 from sans.models.Core2ndMomentModel import Core2ndMomentModel 323 self.model_dictionary[Core2ndMomentModel.__name__] = Core2ndMomentModel 311 324 self.shape_list.append(Core2ndMomentModel) 312 325 self.model_name_list.append(Core2ndMomentModel.__name__) 313 326 314 327 from sans.models.CoreMultiShellModel import CoreMultiShellModel 328 self.model_dictionary[CoreMultiShellModel.__name__] = CoreMultiShellModel 315 329 self.shape_list.append(CoreMultiShellModel) 316 330 self.multiplication_factor.append(CoreMultiShellModel) … … 318 332 319 333 from sans.models.VesicleModel import VesicleModel 334 self.model_dictionary[VesicleModel.__name__] = VesicleModel 320 335 self.shape_list.append(VesicleModel) 321 336 self.multiplication_factor.append(VesicleModel) … … 323 338 324 339 from sans.models.MultiShellModel import MultiShellModel 340 self.model_dictionary[MultiShellModel.__name__] = MultiShellModel 325 341 self.shape_list.append(MultiShellModel) 326 342 self.multiplication_factor.append(MultiShellModel) … … 328 344 329 345 from sans.models.OnionExpShellModel import OnionExpShellModel 346 self.model_dictionary[OnionExpShellModel.__name__] = OnionExpShellModel 330 347 self.shape_list.append(OnionExpShellModel) 331 348 self.multiplication_factor.append(OnionExpShellModel) … … 333 350 334 351 from sans.models.SphericalSLDModel import SphericalSLDModel 352 self.model_dictionary[SphericalSLDModel.__name__] = SphericalSLDModel 335 353 self.shape_list.append(SphericalSLDModel) 336 354 self.multiplication_factor.append(SphericalSLDModel) … … 339 357 340 358 from sans.models.LinearPearlsModel import LinearPearlsModel 359 self.model_dictionary[LinearPearlsModel.__name__] = LinearPearlsModel 341 360 self.shape_list.append(LinearPearlsModel) 342 361 self.model_name_list.append(LinearPearlsModel.__name__) 343 362 344 363 from sans.models.PearlNecklaceModel import PearlNecklaceModel 364 self.model_dictionary[PearlNecklaceModel.__name__] = PearlNecklaceModel 345 365 self.shape_list.append(PearlNecklaceModel) 346 366 self.model_name_list.append(PearlNecklaceModel.__name__) … … 348 368 349 369 from sans.models.CylinderModel import CylinderModel 370 self.model_dictionary[CylinderModel.__name__] = CylinderModel 350 371 self.shape_list.append(CylinderModel) 351 372 self.multiplication_factor.append(CylinderModel) … … 353 374 354 375 from sans.models.CoreShellCylinderModel import CoreShellCylinderModel 376 self.model_dictionary[CoreShellCylinderModel.__name__] = CoreShellCylinderModel 355 377 self.shape_list.append(CoreShellCylinderModel) 356 378 self.multiplication_factor.append(CoreShellCylinderModel) … … 358 380 359 381 from sans.models.CoreShellBicelleModel import CoreShellBicelleModel 382 self.model_dictionary[CoreShellBicelleModel.__name__] = CoreShellBicelleModel 360 383 self.shape_list.append(CoreShellBicelleModel) 361 384 self.multiplication_factor.append(CoreShellBicelleModel) … … 363 386 364 387 from sans.models.HollowCylinderModel import HollowCylinderModel 388 self.model_dictionary[HollowCylinderModel.__name__] = HollowCylinderModel 365 389 self.shape_list.append(HollowCylinderModel) 366 390 self.multiplication_factor.append(HollowCylinderModel) … … 368 392 369 393 from sans.models.FlexibleCylinderModel import FlexibleCylinderModel 394 self.model_dictionary[FlexibleCylinderModel.__name__] = FlexibleCylinderModel 370 395 self.shape_list.append(FlexibleCylinderModel) 371 396 self.model_name_list.append(FlexibleCylinderModel.__name__) 372 397 373 398 from sans.models.FlexCylEllipXModel import FlexCylEllipXModel 399 self.model_dictionary[FlexCylEllipXModel.__name__] = FlexCylEllipXModel 374 400 self.shape_list.append(FlexCylEllipXModel) 375 401 self.model_name_list.append(FlexCylEllipXModel.__name__) 376 402 377 403 from sans.models.StackedDisksModel import StackedDisksModel 404 self.model_dictionary[StackedDisksModel.__name__] = StackedDisksModel 378 405 self.shape_list.append(StackedDisksModel) 379 406 self.multiplication_factor.append(StackedDisksModel) … … 381 408 382 409 from sans.models.ParallelepipedModel import ParallelepipedModel 410 self.model_dictionary[ParallelepipedModel.__name__] = ParallelepipedModel 383 411 self.shape_list.append(ParallelepipedModel) 384 412 self.multiplication_factor.append(ParallelepipedModel) … … 386 414 387 415 from sans.models.CSParallelepipedModel import CSParallelepipedModel 416 self.model_dictionary[CSParallelepipedModel.__name__] = CSParallelepipedModel 388 417 self.shape_list.append(CSParallelepipedModel) 389 418 self.multiplication_factor.append(CSParallelepipedModel) … … 391 420 392 421 from sans.models.EllipticalCylinderModel import EllipticalCylinderModel 422 self.model_dictionary[EllipticalCylinderModel.__name__] = EllipticalCylinderModel 393 423 self.shape_list.append(EllipticalCylinderModel) 394 424 self.multiplication_factor.append(EllipticalCylinderModel) … … 396 426 397 427 from sans.models.BarBellModel import BarBellModel 428 self.model_dictionary[BarBellModel.__name__] = BarBellModel 398 429 self.shape_list.append(BarBellModel) 399 430 self.model_name_list.append(BarBellModel.__name__) … … 402 433 403 434 from sans.models.CappedCylinderModel import CappedCylinderModel 435 self.model_dictionary[CappedCylinderModel.__name__] = CappedCylinderModel 404 436 self.shape_list.append(CappedCylinderModel) 405 437 self.model_name_list.append(CappedCylinderModel.__name__) … … 408 440 409 441 from sans.models.EllipsoidModel import EllipsoidModel 442 self.model_dictionary[EllipsoidModel.__name__] = EllipsoidModel 410 443 self.shape_list.append(EllipsoidModel) 411 444 self.multiplication_factor.append(EllipsoidModel) … … 413 446 414 447 from sans.models.CoreShellEllipsoidModel import CoreShellEllipsoidModel 448 self.model_dictionary[CoreShellEllipsoidModel.__name__] = CoreShellEllipsoidModel 415 449 self.shape_list.append(CoreShellEllipsoidModel) 416 450 self.multiplication_factor.append(CoreShellEllipsoidModel) … … 418 452 419 453 from sans.models.TriaxialEllipsoidModel import TriaxialEllipsoidModel 454 self.model_dictionary[TriaxialEllipsoidModel.__name__] = TriaxialEllipsoidModel 420 455 self.shape_list.append(TriaxialEllipsoidModel) 421 456 self.multiplication_factor.append(TriaxialEllipsoidModel) … … 423 458 424 459 from sans.models.LamellarModel import LamellarModel 460 self.model_dictionary[LamellarModel.__name__] = LamellarModel 425 461 self.shape_list.append(LamellarModel) 426 462 self.model_name_list.append(LamellarModel.__name__) 427 463 428 464 from sans.models.LamellarFFHGModel import LamellarFFHGModel 465 self.model_dictionary[LamellarFFHGModel.__name__] = LamellarFFHGModel 429 466 self.shape_list.append(LamellarFFHGModel) 430 467 self.model_name_list.append(LamellarFFHGModel.__name__) 431 468 432 469 from sans.models.LamellarPSModel import LamellarPSModel 470 self.model_dictionary[LamellarPSModel.__name__] = LamellarPSModel 433 471 self.shape_list.append(LamellarPSModel) 434 472 self.model_name_list.append(LamellarPSModel.__name__) 435 473 436 474 from sans.models.LamellarPSHGModel import LamellarPSHGModel 475 self.model_dictionary[LamellarPSHGModel.__name__] = LamellarPSHGModel 437 476 self.shape_list.append(LamellarPSHGModel) 438 477 self.model_name_list.append(LamellarPSHGModel.__name__) 439 478 440 479 from sans.models.LamellarPCrystalModel import LamellarPCrystalModel 480 self.model_dictionary[LamellarPCrystalModel.__name__] = LamellarPCrystalModel 441 481 self.shape_list.append(LamellarPCrystalModel) 442 482 self.model_name_list.append(LamellarPCrystalModel.__name__) 443 483 444 484 from sans.models.SCCrystalModel import SCCrystalModel 485 self.model_dictionary[SCCrystalModel.__name__] = SCCrystalModel 445 486 self.shape_list.append(SCCrystalModel) 446 487 self.model_name_list.append(SCCrystalModel.__name__) 447 488 448 489 from sans.models.FCCrystalModel import FCCrystalModel 490 self.model_dictionary[FCCrystalModel.__name__] = FCCrystalModel 449 491 self.shape_list.append(FCCrystalModel) 450 492 self.model_name_list.append(FCCrystalModel.__name__) 451 493 452 494 from sans.models.BCCrystalModel import BCCrystalModel 495 self.model_dictionary[BCCrystalModel.__name__] = BCCrystalModel 453 496 self.shape_list.append(BCCrystalModel) 454 497 self.model_name_list.append(BCCrystalModel.__name__) … … 456 499 ## Structure factor 457 500 from sans.models.SquareWellStructure import SquareWellStructure 501 self.model_dictionary[SquareWellStructure.__name__] = SquareWellStructure 458 502 self.struct_list.append(SquareWellStructure) 459 503 self.model_name_list.append(SquareWellStructure.__name__) 460 504 461 505 from sans.models.HardsphereStructure import HardsphereStructure 506 self.model_dictionary[HardsphereStructure.__name__] = HardsphereStructure 462 507 self.struct_list.append(HardsphereStructure) 463 508 self.model_name_list.append(HardsphereStructure.__name__) 464 509 465 510 from sans.models.StickyHSStructure import StickyHSStructure 511 self.model_dictionary[StickyHSStructure.__name__] = StickyHSStructure 466 512 self.struct_list.append(StickyHSStructure) 467 513 self.model_name_list.append(StickyHSStructure.__name__) 468 514 469 515 from sans.models.HayterMSAStructure import HayterMSAStructure 516 self.model_dictionary[HayterMSAStructure.__name__] = HayterMSAStructure 470 517 self.struct_list.append(HayterMSAStructure) 471 518 self.model_name_list.append(HayterMSAStructure.__name__) … … 474 521 ##shape-independent models 475 522 from sans.models.PowerLawAbsModel import PowerLawAbsModel 523 self.model_dictionary[PowerLawAbsModel.__name__] = PowerLawAbsModel 476 524 self.shape_indep_list.append(PowerLawAbsModel) 477 525 self.model_name_list.append(PowerLawAbsModel.__name__) 478 526 479 527 from sans.models.BEPolyelectrolyte import BEPolyelectrolyte 528 self.model_dictionary[BEPolyelectrolyte.__name__] = BEPolyelectrolyte 480 529 self.shape_indep_list.append(BEPolyelectrolyte) 481 530 self.model_name_list.append(BEPolyelectrolyte.__name__) … … 483 532 484 533 from sans.models.BroadPeakModel import BroadPeakModel 534 self.model_dictionary[BroadPeakModel.__name__] = BroadPeakModel 485 535 self.shape_indep_list.append(BroadPeakModel) 486 536 self.model_name_list.append(BroadPeakModel.__name__) 487 537 488 538 from sans.models.CorrLengthModel import CorrLengthModel 539 self.model_dictionary[CorrLengthModel.__name__] = CorrLengthModel 489 540 self.shape_indep_list.append(CorrLengthModel) 490 541 self.model_name_list.append(CorrLengthModel.__name__) 491 542 492 543 from sans.models.DABModel import DABModel 544 self.model_dictionary[DABModel.__name__] = DABModel 493 545 self.shape_indep_list.append(DABModel) 494 546 self.model_name_list.append(DABModel.__name__) 495 547 496 548 from sans.models.DebyeModel import DebyeModel 549 self.model_dictionary[DebyeModel.__name__] = DebyeModel 497 550 self.shape_indep_list.append(DebyeModel) 498 551 self.model_name_list.append(DebyeModel.__name__) 499 552 500 553 from sans.models.FractalModel import FractalModel 554 self.model_dictionary[FractalModel.__name__] = FractalModel 501 555 self.shape_indep_list.append(FractalModel) 502 556 self.model_name_list.append(FractalModel.__name__) 503 557 504 558 from sans.models.FractalCoreShellModel import FractalCoreShellModel 559 self.model_dictionary[FractalCoreShellModel.__name__] = FractalCoreShellModel 505 560 self.shape_indep_list.append(FractalCoreShellModel) 506 561 self.model_name_list.append(FractalCoreShellModel.__name__) 507 562 508 563 from sans.models.GaussLorentzGelModel import GaussLorentzGelModel 564 self.model_dictionary[GaussLorentzGelModel.__name__] = GaussLorentzGelModel 509 565 self.shape_indep_list.append(GaussLorentzGelModel) 510 566 self.model_name_list.append(GaussLorentzGelModel.__name__) 511 567 512 568 from sans.models.GuinierModel import GuinierModel 569 self.model_dictionary[GuinierModel.__name__] = GuinierModel 513 570 self.shape_indep_list.append(GuinierModel) 514 571 self.model_name_list.append(GuinierModel.__name__) 515 572 516 573 from sans.models.GuinierPorodModel import GuinierPorodModel 574 self.model_dictionary[GuinierPorodModel.__name__] = GuinierPorodModel 517 575 self.shape_indep_list.append(GuinierPorodModel) 518 576 self.model_name_list.append(GuinierPorodModel.__name__) 519 577 520 578 from sans.models.LorentzModel import LorentzModel 579 self.model_dictionary[LorentzModel.__name__] = LorentzModel 521 580 self.shape_indep_list.append(LorentzModel) 522 581 self.model_name_list.append(LorentzModel.__name__) 523 582 524 583 from sans.models.MassFractalModel import MassFractalModel 584 self.model_dictionary[MassFractalModel.__name__] = MassFractalModel 525 585 self.shape_indep_list.append(MassFractalModel) 526 586 self.model_name_list.append(MassFractalModel.__name__) 527 587 528 588 from sans.models.MassSurfaceFractal import MassSurfaceFractal 589 self.model_dictionary[MassSurfaceFractal.__name__] = MassSurfaceFractal 529 590 self.shape_indep_list.append(MassSurfaceFractal) 530 591 self.model_name_list.append(MassSurfaceFractal.__name__) 531 592 532 593 from sans.models.PeakGaussModel import PeakGaussModel 594 self.model_dictionary[PeakGaussModel.__name__] = PeakGaussModel 533 595 self.shape_indep_list.append(PeakGaussModel) 534 596 self.model_name_list.append(PeakGaussModel.__name__) 535 597 536 598 from sans.models.PeakLorentzModel import PeakLorentzModel 599 self.model_dictionary[PeakLorentzModel.__name__] = PeakLorentzModel 537 600 self.shape_indep_list.append(PeakLorentzModel) 538 601 self.model_name_list.append(PeakLorentzModel.__name__) 539 602 540 603 from sans.models.Poly_GaussCoil import Poly_GaussCoil 604 self.model_dictionary[Poly_GaussCoil.__name__] = Poly_GaussCoil 541 605 self.shape_indep_list.append(Poly_GaussCoil) 542 606 self.model_name_list.append(Poly_GaussCoil.__name__) 543 607 544 608 from sans.models.PolymerExclVolume import PolymerExclVolume 609 self.model_dictionary[PolymerExclVolume.__name__] = PolymerExclVolume 545 610 self.shape_indep_list.append(PolymerExclVolume) 546 611 self.model_name_list.append(PolymerExclVolume.__name__) 547 612 548 613 from sans.models.PorodModel import PorodModel 614 self.model_dictionary[PorodModel.__name__] = PorodModel 549 615 self.shape_indep_list.append(PorodModel) 550 616 self.model_name_list.append(PorodModel.__name__) 551 617 552 618 from sans.models.RPA10Model import RPA10Model 619 self.model_dictionary[RPA10Model.__name__] = RPA10Model 553 620 self.shape_indep_list.append(RPA10Model) 554 621 self.multi_func_list.append(RPA10Model) 555 622 556 623 from sans.models.StarPolymer import StarPolymer 624 self.model_dictionary[StarPolymer.__name__] = StarPolymer 557 625 self.shape_indep_list.append(StarPolymer) 558 626 self.model_name_list.append(StarPolymer.__name__) 559 627 560 628 from sans.models.SurfaceFractalModel import SurfaceFractalModel 629 self.model_dictionary[SurfaceFractalModel.__name__] = SurfaceFractalModel 561 630 self.shape_indep_list.append(SurfaceFractalModel) 562 631 self.model_name_list.append(SurfaceFractalModel.__name__) 563 632 564 633 from sans.models.TeubnerStreyModel import TeubnerStreyModel 634 self.model_dictionary[TeubnerStreyModel.__name__] = TeubnerStreyModel 565 635 self.shape_indep_list.append(TeubnerStreyModel) 566 636 self.model_name_list.append(TeubnerStreyModel.__name__) 567 637 568 638 from sans.models.TwoLorentzianModel import TwoLorentzianModel 639 self.model_dictionary[TwoLorentzianModel.__name__] = TwoLorentzianModel 569 640 self.shape_indep_list.append(TwoLorentzianModel) 570 641 self.model_name_list.append(TwoLorentzianModel.__name__) 571 642 572 643 from sans.models.TwoPowerLawModel import TwoPowerLawModel 644 self.model_dictionary[TwoPowerLawModel.__name__] = TwoPowerLawModel 573 645 self.shape_indep_list.append(TwoPowerLawModel) 574 646 self.model_name_list.append(TwoPowerLawModel.__name__) 575 647 576 648 from sans.models.UnifiedPowerRgModel import UnifiedPowerRgModel 649 self.model_dictionary[UnifiedPowerRgModel.__name__] = UnifiedPowerRgModel 577 650 self.shape_indep_list.append(UnifiedPowerRgModel) 578 651 self.multi_func_list.append(UnifiedPowerRgModel) 579 652 580 653 from sans.models.LineModel import LineModel 654 self.model_dictionary[LineModel.__name__] = LineModel 581 655 self.shape_indep_list.append(LineModel) 582 656 self.model_name_list.append(LineModel.__name__) 583 657 584 658 from sans.models.ReflectivityModel import ReflectivityModel 659 self.model_dictionary[ReflectivityModel.__name__] = ReflectivityModel 585 660 self.shape_indep_list.append(ReflectivityModel) 586 661 self.multi_func_list.append(ReflectivityModel) 587 662 588 663 from sans.models.ReflectivityIIModel import ReflectivityIIModel 664 self.model_dictionary[ReflectivityIIModel.__name__] = ReflectivityIIModel 589 665 self.shape_indep_list.append(ReflectivityIIModel) 590 666 self.multi_func_list.append(ReflectivityIIModel) 591 667 592 668 from sans.models.GelFitModel import GelFitModel 669 self.model_dictionary[GelFitModel.__name__] = GelFitModel 593 670 self.shape_indep_list.append(GelFitModel) 594 671 self.model_name_list.append(GelFitModel.__name__) 595 672 673 #from sans.models.FractalO_Z import FractalO_Z 674 #self.model_dictionary[FractalO_Z.__name__] = FractalO_Z 675 #self.shape_indep_list.append(FractalO_Z) 676 #self.model_name_list.append(FractalO_Z.__name__) 677 596 678 #Looking for plugins 597 679 self.stored_plugins = self.findModels() … … 792 874 if int(evt.GetId()) in self.form_factor_dict.keys(): 793 875 from sans.models.MultiplicationModel import MultiplicationModel 876 self.model_dictionary[MultiplicationModel.__name__] = MultiplicationModel 794 877 model1, model2 = self.form_factor_dict[int(evt.GetId())] 795 878 model = MultiplicationModel(model1, model2) … … 836 919 """ 837 920 return self.model_name_list 921 922 def get_model_dictionary(self): 923 """ 924 return dictionary linking model names to objects 925 """ 926 return self.model_dictionary 838 927 839 928 … … 873 962 def get_model_name_list(self): 874 963 return self.__modelmanager.get_model_name_list() 964 965 def get_model_dictionary(self): 966 return self.__modelmanager.get_model_dictionary()
Note: See TracChangeset
for help on using the changeset viewer.
