FlagCounter.h

Go to the documentation of this file.

// FlagCounter.h: Class to keep counts of nr of flagged points
// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_FLAGCOUNTER_H_
#define DP3_FLAGCOUNTER_H_
 
#include <casacore/casa/Arrays/Vector.h>
 
#include <cstdint>
#include <ostream>
 
namespace dp3 {
namespace common {
class ParameterSet;
}
 
namespace base {
class DPInfo;
 
 
 
class FlagCounter {
 public:
  FlagCounter() = default;
 
  FlagCounter(const common::ParameterSet&, const std::string& prefix);
 
  void init(const DPInfo& info);
 
  void incrBaseline(unsigned int bl) { base_line_counts_[bl]++; }
 
  void incrChannel(unsigned int chan) { channel_counts_[chan]++; }
 
  void incrCorrelation(unsigned int corr) { correlation_counts_[corr]++; }
 
  void add(const FlagCounter& that);
 
  const std::vector<int64_t>& baselineCounts() const {
    return base_line_counts_;
  }
  const std::vector<int64_t>& channelCounts() const { return channel_counts_; }
  const std::vector<int64_t>& correlationCounts() const {
    return correlation_counts_;
  }
 
  void showBaseline(std::ostream& os, int64_t ntimes) const;
  void showChannel(std::ostream& os, int64_t ntimes) const;
  void showCorrelation(std::ostream& os, int64_t ntimes) const;
 
  void showStation(std::ostream& os, int64_t ntimes) const;
 
  static void showPerc1(std::ostream&, double value, double total);
 
  static void showPerc3(std::ostream&, double value, double total);
 
 private:
  void saveStation(int64_t npoints, const casacore::Vector<int64_t>& nused,
                   const casacore::Vector<int64_t>& count) const;
 
  void saveChannel(int64_t npoints, const std::vector<int64_t>& count) const;
 
  const DPInfo* info_{nullptr};
  std::string save_filename_{};
  double warning_percentage_{0.0};
  bool show_fully_flagged_{false};
  bool save_{false};
  std::string path_{};
  std::string name_{};
  std::vector<int64_t> base_line_counts_{};
  std::vector<int64_t> channel_counts_{};
  std::vector<int64_t> correlation_counts_{};
};
 
}  // namespace base
}  // namespace dp3
 
#endif