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 | def set_engine(self,word): |
---|
37 | """ |
---|
38 | Select the type of Fit |
---|
39 | @param word: the keyword to select the fit type |
---|
40 | @raise: if the user does not enter 'scipy' or 'park', |
---|
41 | a valueError is rase |
---|
42 | """ |
---|
43 | if word=="scipy": |
---|
44 | self._engine=ScipyFit() |
---|
45 | elif word=="park": |
---|
46 | self._engine=ParkFit() |
---|
47 | else: |
---|
48 | raise ValueError, "enter the keyword scipy or park" |
---|
49 | |
---|
50 | def returnEngine(self): |
---|
51 | """ @return self._engine""" |
---|
52 | return self._engine |
---|
53 | |
---|
54 | def fit(self, qmin=None, qmax=None): |
---|
55 | """Perform the fit """ |
---|
56 | return self._engine.fit(qmin,qmax) |
---|
57 | |
---|
58 | def set_model(self,model,Uid,pars=[]): |
---|
59 | self._engine.set_model(model,Uid,pars) |
---|
60 | |
---|
61 | def set_data(self,data,Uid,qmin=None, qmax=None,ymin=None, ymax=None): |
---|
62 | self._engine.set_data(data,Uid,qmin,qmax,ymin,ymax) |
---|
63 | |
---|
64 | def get_model(self,Uid): |
---|
65 | """ return list of data""" |
---|
66 | self._engine.get_model(Uid) |
---|
67 | |
---|
68 | def remove_Fit_Problem(self,Uid): |
---|
69 | """remove fitarrange in Uid""" |
---|
70 | self._engine.remove_Fit_Problem(Uid) |
---|
71 | |
---|
72 | def select_problem_for_fit(self,Uid,value): |
---|
73 | """ |
---|
74 | select a couple of model and data at the Uid position in dictionary |
---|
75 | and set in self.selected value to value |
---|
76 | @param value: the value to allow fitting. can only have the value one or zero |
---|
77 | """ |
---|
78 | self._engine.select_problem_for_fit(Uid,value) |
---|
79 | |
---|
80 | def get_problem_to_fit(self,Uid): |
---|
81 | """ |
---|
82 | return the self.selected value of the fit problem of Uid |
---|
83 | @param Uid: the Uid of the problem |
---|
84 | """ |
---|
85 | return self._engine.get_problem_to_fit(Uid) |
---|