File: antennaselector.cpp

package info (click to toggle)
aoflagger 3.1.0-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 4,868 kB
  • sloc: cpp: 52,164; python: 152; sh: 60; makefile: 17
file content (47 lines) | stat: -rw-r--r-- 1,527 bytes parent folder | download
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
#include "antennaselector.h"

#include "../../quality/statisticscollection.h"
#include "../../quality/statisticsderivator.h"

using std::size_t;

namespace rfiStrategy {

std::vector<size_t> AntennaSelector::Run(
    const StatisticsCollection& statisticsCollection) {
  std::map<size_t, DefaultStatistics> antStatistics =
      statisticsCollection.GetAntennaStatistics();

  std::vector<double> stddevs;
  std::set<size_t> badAntennas;
  for (size_t p = 0; p != 4; ++p) {
    double meanStddev = 0.0;
    stddevs.clear();
    for (const std::pair<const size_t, DefaultStatistics>& antenna : antStatistics) {
      double stddev = StatisticsDerivator::GetStatisticAmplitude(
          QualityTablesFormatter::StandardDeviationStatistic, antenna.second,
          p);
      stddevs.emplace_back(stddev);
      meanStddev += stddev;
    }
    double stddevOfStddev = 0.0;
    meanStddev /= stddevs.size();
    for (double& s : stddevs) {
      stddevOfStddev += (s - meanStddev) * (s - meanStddev);
    }
    stddevOfStddev = sqrt(stddevOfStddev / stddevs.size());

    size_t index = 0;
    double limit = _threshold * stddevOfStddev;
    for (const std::pair<const size_t, DefaultStatistics>& antenna : antStatistics) {
      if (std::fabs(stddevs[index] - meanStddev) > limit ||
          stddevs[index] == 0.0) {
        if (antenna.second.count[p] != 0) badAntennas.insert(antenna.first);
      }
      ++index;
    }
  }
  return std::vector<size_t>(badAntennas.begin(), badAntennas.end());
}

}  // namespace rfiStrategy