1 | import sys, time |
---|
2 | from calcthread import CalcThread |
---|
3 | from sans.pr.invertor import Invertor |
---|
4 | import numpy |
---|
5 | from config import printEVT |
---|
6 | |
---|
7 | class CalcPr(CalcThread): |
---|
8 | """ |
---|
9 | Compute 2D model |
---|
10 | This calculation assumes a 2-fold symmetry of the model |
---|
11 | where points are computed for one half of the detector |
---|
12 | and I(qx, qy) = I(-qx, -qy) is assumed. |
---|
13 | """ |
---|
14 | |
---|
15 | def __init__(self, pr, nfunc=5, error_func=None, |
---|
16 | completefn = None, |
---|
17 | updatefn = None, |
---|
18 | yieldtime = 0.01, |
---|
19 | worktime = 0.01 |
---|
20 | ): |
---|
21 | CalcThread.__init__(self,completefn, |
---|
22 | updatefn, |
---|
23 | yieldtime, |
---|
24 | worktime) |
---|
25 | self.pr = pr |
---|
26 | self.nfunc = nfunc |
---|
27 | self.error_func = error_func |
---|
28 | self.starttime = 0 |
---|
29 | |
---|
30 | def isquit(self): |
---|
31 | try: |
---|
32 | CalcThread.isquit(self) |
---|
33 | except KeyboardInterrupt: |
---|
34 | printEVT("P(r) calc interrupted") |
---|
35 | raise KeyboardInterrupt |
---|
36 | |
---|
37 | def compute(self): |
---|
38 | import time |
---|
39 | try: |
---|
40 | self.starttime = time.time() |
---|
41 | #out, cov = self.pr.invert(self.nfunc) |
---|
42 | out, cov = self.pr.lstsq(self.nfunc) |
---|
43 | elapsed = time.time()-self.starttime |
---|
44 | self.complete(out=out, cov=cov, pr=self.pr, elapsed=elapsed) |
---|
45 | except: |
---|
46 | if not self.error_func==None: |
---|
47 | self.error_func("CalcPr.compute: %s" % sys.exc_value) |
---|
48 | |
---|
49 | class EstimatePr(CalcThread): |
---|
50 | """ |
---|
51 | Compute 2D model |
---|
52 | This calculation assumes a 2-fold symmetry of the model |
---|
53 | where points are computed for one half of the detector |
---|
54 | and I(qx, qy) = I(-qx, -qy) is assumed. |
---|
55 | """ |
---|
56 | |
---|
57 | def __init__(self, pr, nfunc=5, error_func=None, |
---|
58 | completefn = None, |
---|
59 | updatefn = None, |
---|
60 | yieldtime = 0.01, |
---|
61 | worktime = 0.01 |
---|
62 | ): |
---|
63 | CalcThread.__init__(self,completefn, |
---|
64 | updatefn, |
---|
65 | yieldtime, |
---|
66 | worktime) |
---|
67 | self.pr = pr |
---|
68 | self.nfunc = nfunc |
---|
69 | self.error_func = error_func |
---|
70 | self.starttime = 0 |
---|
71 | |
---|
72 | def isquit(self): |
---|
73 | try: |
---|
74 | CalcThread.isquit(self) |
---|
75 | except KeyboardInterrupt: |
---|
76 | printEVT("P(r) calc interrupted") |
---|
77 | raise KeyboardInterrupt |
---|
78 | |
---|
79 | def compute(self): |
---|
80 | import time |
---|
81 | try: |
---|
82 | |
---|
83 | print "Alpha Oscill" |
---|
84 | |
---|
85 | self.starttime = time.time() |
---|
86 | # If the current alpha is zero, try |
---|
87 | # another value |
---|
88 | if self.pr.alpha<=0: |
---|
89 | self.pr.alpha = 0.0001 |
---|
90 | |
---|
91 | out, cov = self.pr.lstsq(self.nfunc) |
---|
92 | elapsed = time.time()-self.starttime |
---|
93 | |
---|
94 | |
---|
95 | # Take the default and try to find |
---|
96 | # a better value |
---|
97 | best_alpha = self.pr.alpha |
---|
98 | best_osc = self.pr.oscillations(out) |
---|
99 | |
---|
100 | print best_alpha, best_osc |
---|
101 | |
---|
102 | alpha = self.pr.suggested_alpha |
---|
103 | print "initial:", alpha |
---|
104 | |
---|
105 | # Look at smaller values |
---|
106 | for i in range(5): |
---|
107 | self.pr.alpha = (0.1)**(i)*alpha |
---|
108 | out, cov = self.pr.lstsq(self.nfunc) |
---|
109 | osc = self.pr.oscillations(out) |
---|
110 | print self.pr.alpha, osc |
---|
111 | if osc < best_osc: |
---|
112 | best_osc = osc |
---|
113 | best_alpha = alpha |
---|
114 | |
---|
115 | ## Look at larger values |
---|
116 | #for i in range(4): |
---|
117 | # self.pr.alpha = (10.0)**(i+1)*alpha |
---|
118 | # out, cov = self.pr.lstsq(self.nfunc) |
---|
119 | # osc = self.pr.oscillations(out) |
---|
120 | # print self.pr.alpha, osc |
---|
121 | # if osc < best_osc: |
---|
122 | # best_osc = osc |
---|
123 | # best_alpha = alpha |
---|
124 | |
---|
125 | |
---|
126 | self.complete(alpha=best_alpha, elapsed=elapsed) |
---|
127 | |
---|
128 | |
---|
129 | |
---|
130 | |
---|
131 | except: |
---|
132 | if not self.error_func==None: |
---|
133 | printEVT("EstimatePr.compute: %s" % sys.exc_value) |
---|
134 | |
---|
135 | |
---|