1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#include "sinusfitter.h"
#include <cmath>
namespace algorithms {
SinusFitter::SinusFitter() {}
SinusFitter::~SinusFitter() {}
void SinusFitter::FindPhaseAndAmplitude(num_t& phase, num_t& amplitude,
const num_t* dataX, const num_t* dataT,
const size_t dataSize,
const num_t frequency) const throw() {
// calculate 1/N * \sum_t x(t) e^{-i * frequency * t}
num_t sumR = 0.0L, sumI = 0.0L;
for (unsigned i = 0; i < dataSize; ++i) {
const num_t t = dataT[i];
const num_t x = dataX[i];
sumR += x * std::cos(-t * frequency);
sumI += x * std::sin(-t * frequency);
}
sumR /= (num_t)dataSize;
sumI /= (num_t)dataSize;
phase = Phase(sumR, sumI);
amplitude = 2.0L * std::sqrt(sumR * sumR + sumI * sumI);
}
void SinusFitter::FindPhaseAndAmplitudeComplex(
num_t& phase, num_t& amplitude, const num_t* dataR, const num_t* dataI,
const num_t* dataT, const size_t dataSize, const num_t frequency) const
throw() {
// calculate 1/N * \sum_t x(t) e^{-i * frequency * t}
num_t sumR = 0.0L, sumI = 0.0L;
for (unsigned i = 0; i < dataSize; ++i) {
const num_t t = dataT[i];
const num_t xR = dataR[i];
const num_t xI = dataI[i];
sumR += xR * std::cos(-t * frequency);
sumR += xI * std::sin(-t * frequency);
sumI += xR * std::sin(-t * frequency);
sumI -= xI * std::cos(-t * frequency);
}
sumR /= (num_t)dataSize;
sumI /= (num_t)dataSize;
phase = Phase(sumR, sumI);
amplitude = std::sqrt(sumR * sumR + sumI * sumI);
}
num_t SinusFitter::FindMean(const num_t phase, const num_t amplitude,
const num_t* dataX, const num_t* dataT,
const size_t dataSize, const num_t frequency) {
num_t sum = 0.0L;
for (unsigned i = 0; i < dataSize; ++i) {
sum += dataX[i] - Value(phase, amplitude, dataT[i], frequency, 0.0L);
}
return sum / dataSize;
}
} // namespace algorithms
|
