BdaGroupPredict.h

Go to the documentation of this file.

// BdaGroupPredict.h: class to directly predict baseline dependent averaged
// (BDA) visibilities from a source model Copyright (C) 2020 ASTRON (Netherlands
// Institute for Radio Astronomy) SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_BDAGROUPPREDICT_H
#define DP3_BDAGROUPPREDICT_H
 
#include "InputStep.h"
 
#include "base/BdaBuffer.h"
 
#include <map>
#include <queue>
#include <utility>
 
namespace dp3 {
namespace common {
class ParameterSet;
}
 
namespace steps {
 
 
class BdaGroupPredict : public ModelDataStep {
 public:
  BdaGroupPredict(const common::ParameterSet&, const std::string& prefix);
 
  BdaGroupPredict(const common::ParameterSet&, const std::string& prefix,
                  const std::vector<std::string>& source_patterns);
 
  ~BdaGroupPredict() override;
 
  common::Fields getRequiredFields() const override;
 
  bool process(std::unique_ptr<base::BdaBuffer>) override;
 
  void finish() override;
 
  void updateInfo(const base::DPInfo&) override;
 
  void show(std::ostream&) const override;
 
  void showTimings(std::ostream&, double duration) const override;
 
  bool accepts(MsType dt) const override { return dt == MsType::kBda; }
 
  MsType outputs() const override { return MsType::kBda; }
 
  base::Direction GetFirstDirection() const override;
 
 private:
  // Need to store a reference to the parset to create the OnePredict
  // substeps in updateInfo()
  const common::ParameterSet& parset_;
  std::string name_;
 
  // Structure to keep track how many rows in a BdaBuffer have been processed
  struct BufferInfo {
    std::unique_ptr<base::BdaBuffer> buffer;
    size_t nr_rows_filled;
  };
 
  std::queue<BufferInfo> buffers_;
 
  std::vector<std::string> source_patterns_;
 
  class BaselineGroup;
 
  std::map<std::pair<int, int>, BaselineGroup> averaging_to_baseline_group_map_;
 
  std::vector<std::pair<BaselineGroup*, int>> index_to_baseline_group_map_;
 
  common::NSTimer timer_;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif