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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#ifndef LUA_SCRIPTDATA_H
#define LUA_SCRIPTDATA_H
#include <algorithm>
#include <mutex>
#include <string>
#include <vector>
#include "../structures/timefrequencydata.h"
class ScriptData {
public:
ScriptData();
~ScriptData();
ScriptData(const ScriptData&) = delete;
ScriptData& operator=(const ScriptData&) = delete;
std::unique_ptr<class BandpassFile>& GetBandpassFile() {
return _bandpassFile;
}
std::mutex& BandpassMutex() { return _bandpassMutex; }
std::unique_ptr<class StatisticsCollection>& GetStatistics() {
return _statistics;
}
void AddVisualization(TimeFrequencyData& data, const std::string& label,
size_t sortingIndex) {
if (_canVisualize) {
if (data.PolarizationCount() == 1) {
aocommon::PolarizationEnum p = data.GetPolarization(0);
for (auto& v : _visualizationData) {
if (std::get<0>(v) == label) {
if (std::get<1>(v).HasPolarization(p)) {
// Can't merge, continue search and
// if not found add like normal
} else {
// Merge
std::get<1>(v) =
TimeFrequencyData::MakeFromPolarizationCombination(
std::get<1>(v), data);
return;
}
}
}
// Label not found, add
_visualizationData.emplace_back(label, data, sortingIndex);
} else {
_visualizationData.emplace_back(label, data, sortingIndex);
}
}
}
size_t VisualizationCount() const { return _visualizationData.size(); }
std::tuple<std::string, TimeFrequencyData, size_t> GetVisualization(
size_t index) const {
return _visualizationData[index];
}
void SetProgressListener(class ProgressListener& progressListener) {
_progressListener = &progressListener;
}
ProgressListener* Progress() { return _progressListener; }
void SetCanVisualize(bool canVisualize) { _canVisualize = canVisualize; }
void SortVisualizations() {
// Sort the visualizations on their sorting index
std::sort(_visualizationData.begin(), _visualizationData.end(),
[](const std::tuple<std::string, TimeFrequencyData, size_t>& a,
const std::tuple<std::string, TimeFrequencyData, size_t>& b) {
return std::get<2>(a) < std::get<2>(b);
});
}
void Combine(ScriptData&& other);
private:
class ProgressListener* _progressListener;
std::unique_ptr<BandpassFile> _bandpassFile;
std::mutex _bandpassMutex;
bool _canVisualize;
std::vector<std::tuple<std::string, TimeFrequencyData, size_t>>
_visualizationData;
std::unique_ptr<class StatisticsCollection> _statistics;
};
#endif
|
