1 | """ |
---|
2 | @organization: Class Fit contains ScipyFit and ParkFit methods declaration |
---|
3 | allows to create instance of type ScipyFit or ParkFit to perform either |
---|
4 | a park fit or a scipy fit. |
---|
5 | """ |
---|
6 | #from sans.guitools.plottables import Data1D |
---|
7 | #from danse.common.plottools.plottables import Data1D |
---|
8 | #from Loader import Load |
---|
9 | from scipy import optimize |
---|
10 | from ScipyFitting import ScipyFit |
---|
11 | from ParkFitting import ParkFit |
---|
12 | |
---|
13 | |
---|
14 | class Fit: |
---|
15 | """ |
---|
16 | Wrap class that allows to select the fitting type.this class |
---|
17 | can be used as follow : |
---|
18 | |
---|
19 | from sans.fit.Fitting import Fit |
---|
20 | fitter= Fit() |
---|
21 | fitter.fit_engine('scipy') or fitter.fit_engine('park') |
---|
22 | engine = fitter.returnEngine() |
---|
23 | engine.set_data(data,Uid) |
---|
24 | engine.set_param( model,model.name, pars) |
---|
25 | engine.set_model(model,Uid) |
---|
26 | |
---|
27 | chisqr1, out1, cov1=engine.fit(pars,qmin,qmax) |
---|
28 | """ |
---|
29 | def __init__(self, engine='scipy'): |
---|
30 | """ |
---|
31 | self._engine will contain an instance of ScipyFit or ParkFit |
---|
32 | """ |
---|
33 | self._engine = None |
---|
34 | self.set_engine(engine) |
---|
35 | |
---|
36 | |
---|
37 | def set_engine(self,word): |
---|
38 | """ |
---|
39 | Select the type of Fit |
---|
40 | @param word: the keyword to select the fit type |
---|
41 | @raise: if the user does not enter 'scipy' or 'park', |
---|
42 | a valueError is rase |
---|
43 | """ |
---|
44 | if word=="scipy": |
---|
45 | self._engine=ScipyFit() |
---|
46 | elif word=="park": |
---|
47 | self._engine=ParkFit() |
---|
48 | else: |
---|
49 | raise ValueError, "enter the keyword scipy or park" |
---|
50 | |
---|
51 | |
---|
52 | |
---|
53 | def fit(self,handler=None): |
---|
54 | """Perform the fit """ |
---|
55 | |
---|
56 | return self._engine.fit(handler) |
---|
57 | |
---|
58 | |
---|
59 | def set_model(self,model,Uid,pars=[]): |
---|
60 | """ |
---|
61 | store a model model to fit at the position Uid of the fit engine |
---|
62 | """ |
---|
63 | self._engine.set_model(model,Uid,pars) |
---|
64 | |
---|
65 | |
---|
66 | def set_data(self,data,Uid,smearer=None,qmin=None, qmax=None): |
---|
67 | """ |
---|
68 | Store data to fit at the psotion Uid of the fit engine |
---|
69 | @param data: data to fit |
---|
70 | @param smearer: smearerobject to smear data |
---|
71 | @param qmin: the minimum q range to fit |
---|
72 | @param qmax: the minimum q range to fit |
---|
73 | """ |
---|
74 | self._engine.set_data(data,Uid,smearer,qmin, qmax) |
---|
75 | |
---|
76 | |
---|
77 | def get_model(self,Uid): |
---|
78 | """ return list of data""" |
---|
79 | self._engine.get_model(Uid) |
---|
80 | |
---|
81 | |
---|
82 | def remove_Fit_Problem(self,Uid): |
---|
83 | """remove fitarrange in Uid""" |
---|
84 | self._engine.remove_Fit_Problem(Uid) |
---|
85 | |
---|
86 | |
---|
87 | def select_problem_for_fit(self,Uid,value): |
---|
88 | """ |
---|
89 | select a couple of model and data at the Uid position in dictionary |
---|
90 | and set in self.selected value to value |
---|
91 | @param value: the value to allow fitting. can only have the value one or zero |
---|
92 | """ |
---|
93 | self._engine.select_problem_for_fit(Uid,value) |
---|
94 | |
---|
95 | |
---|
96 | def get_problem_to_fit(self,Uid): |
---|
97 | """ |
---|
98 | return the self.selected value of the fit problem of Uid |
---|
99 | @param Uid: the Uid of the problem |
---|
100 | """ |
---|
101 | return self._engine.get_problem_to_fit(Uid) |
---|