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
92
93
94
95
96
97
98
|
#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
|
