1 | from sas.sascalc.data_util.calcthread import CalcThread |
---|
2 | from sas.sascalc.dataloader.data_info import Data1D |
---|
3 | from scipy.fftpack import dct |
---|
4 | from scipy.integrate import simps, trapz |
---|
5 | import numpy as np |
---|
6 | from time import sleep |
---|
7 | |
---|
8 | class FourierThread(CalcThread): |
---|
9 | def __init__(self, raw_data, extrapolated_data, bg, extrap_fn=None, |
---|
10 | updatefn=None, completefn=None): |
---|
11 | CalcThread.__init__(self, updatefn=updatefn, completefn=completefn) |
---|
12 | self.data = raw_data |
---|
13 | self.background = bg |
---|
14 | self.extrapolation = extrapolated_data |
---|
15 | self.extrap_fn = extrap_fn |
---|
16 | |
---|
17 | def compute(self): |
---|
18 | qs = self.extrapolation.x |
---|
19 | iqs = self.extrapolation.y |
---|
20 | q = self.data.x |
---|
21 | background = self.background |
---|
22 | |
---|
23 | xs = np.pi*np.arange(len(qs),dtype=np.float32)/(q[1]-q[0])/len(qs) |
---|
24 | |
---|
25 | self.ready(delay=0.0) |
---|
26 | self.update(msg="Fourier transform in progress.") |
---|
27 | self.ready(delay=0.0) |
---|
28 | |
---|
29 | if self.isquit(): |
---|
30 | return |
---|
31 | try: |
---|
32 | gamma1 = dct((iqs-background)*qs**2) |
---|
33 | gamma1 = gamma1 / gamma1.max() |
---|
34 | |
---|
35 | # gamma3(R) = 1/R int_{0}^{R} gamma1(x) dx |
---|
36 | # simps uses simpson's rule to calculate the integral |
---|
37 | gamma3 = [trapz(gamma1[:n], xs[:n])/xs[n-1] for n in range(1, len(xs+1))] |
---|
38 | gamma3 = np.array(gamma3) |
---|
39 | except Exception as e: |
---|
40 | import logging |
---|
41 | logger = logging.getLogger(__name__) |
---|
42 | logger.error(e) |
---|
43 | |
---|
44 | self.update(msg="Fourier transform failed.") |
---|
45 | self.complete(transforms=None) |
---|
46 | return |
---|
47 | if self.isquit(): |
---|
48 | return |
---|
49 | self.update(msg="Fourier transform completed.") |
---|
50 | |
---|
51 | transform1 = Data1D(xs, gamma1) |
---|
52 | transform3 = Data1D(xs, gamma3) |
---|
53 | |
---|
54 | transforms = (transform1, transform3) |
---|
55 | |
---|
56 | self.complete(transforms=transforms) |
---|
57 | |
---|
58 | class HilbertThread(CalcThread): |
---|
59 | def __init__(self, raw_data, extrapolated_data, bg, updatefn=None, |
---|
60 | completefn=None): |
---|
61 | CalcThread.__init__(self, updatefn=updatefn, completefn=completefn) |
---|
62 | self.data = raw_data |
---|
63 | self.background = bg |
---|
64 | self.extrapolation = extrapolated_data |
---|
65 | |
---|
66 | def compute(self): |
---|
67 | qs = self.extrapolation.x |
---|
68 | iqs = self.extrapolation.y |
---|
69 | q = self.data.x |
---|
70 | background = self.background |
---|
71 | |
---|
72 | self.ready(delay=0.0) |
---|
73 | self.update(msg="Starting Hilbert transform.") |
---|
74 | self.ready(delay=0.0) |
---|
75 | if self.isquit(): |
---|
76 | return |
---|
77 | |
---|
78 | # TODO: Implement hilbert transform |
---|
79 | |
---|
80 | self.update(msg="Hilbert transform completed.") |
---|
81 | |
---|
82 | self.complete(transform=None) |
---|