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
68
69
70
71
|
#ifndef SVDMITIGATER_H
#define SVDMITIGATER_H
#include <iostream>
#include "../../structures/image2d.h"
#include "surfacefitmethod.h"
// Needs to be included LAST
#include "../../util/f2c.h"
class SVDMitigater final : public SurfaceFitMethod {
public:
SVDMitigater();
~SVDMitigater();
virtual void Initialize(const TimeFrequencyData &data) final override {
Clear();
_data = data;
_iteration = 0;
}
void PerformFit() {
_iteration++;
RemoveSingularValues(_removeCount);
}
virtual void PerformFit(unsigned) final override { PerformFit(); }
void RemoveSingularValues(unsigned singularValueCount) {
if (!IsDecomposed()) Decompose();
for (unsigned i = 0; i < singularValueCount; ++i) SetSingularValue(i, 0.0);
Compose();
}
TimeFrequencyData Background() const { return *_background; }
enum TimeFrequencyData::ComplexRepresentation ComplexRepresentation() const {
return TimeFrequencyData::ComplexParts;
}
bool IsDecomposed() const throw() { return _singularValues != 0; }
double SingularValue(unsigned index) const throw() {
return _singularValues[index];
}
void SetRemoveCount(unsigned removeCount) throw() {
_removeCount = removeCount;
}
void SetVerbose(bool verbose) throw() { _verbose = verbose; }
static void CreateSingularValueGraph(const TimeFrequencyData &data,
class Plot2D &plot);
private:
void Clear();
void Decompose();
void Compose();
void SetSingularValue(unsigned index, double newValue) throw() {
_singularValues[index] = newValue;
}
TimeFrequencyData _data;
TimeFrequencyData *_background;
double *_singularValues;
doublecomplex *_leftSingularVectors;
doublecomplex *_rightSingularVectors;
long int _m, _n;
unsigned _iteration;
unsigned _removeCount;
bool _verbose;
};
#endif
|
