BdaExpander.h

Go to the documentation of this file.

// Copyright (C) 2021 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_STEPS_BDAEXPANDER_H_
#define DP3_STEPS_BDAEXPANDER_H_
 
#include <map>
#include <utility>
#include <vector>
 
#include "base/BdaBuffer.h"
#include "base/DPBuffer.h"
#include "steps/Step.h"
 
#include "common/Timer.h"
 
namespace dp3 {
namespace steps {
 
 
 
class BdaExpander : public Step {
 public:
  explicit BdaExpander(const std::string& prefix);
 
  common::Fields getRequiredFields() const override { return kUvwField; }
 
  common::Fields getProvidedFields() const override {
    // The BdaExpander only provides fields that are already present in the
    // input BdaBuffer -> return an empty Fields object.
    return {};
  }
 
  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; }
 
 private:
  void CopyData(const base::BdaBuffer& bda_buffer,
                const base::BdaBuffer::Row& bda_row,
                std::unique_ptr<dp3::base::DPBuffer>& buf_out,
                int time_averaging_factor = 1);
 
  struct RegularBufferElement {
    RegularBufferElement(size_t n_baselines, unsigned int n_corr,
                         unsigned int n_chan, double current_time,
                         double current_exposure);
 
    std::vector<bool> baseline;
    std::unique_ptr<dp3::base::DPBuffer> regular_buffer;
  };
 
  std::map<unsigned int, RegularBufferElement> RB_elements;
 
  std::vector<std::vector<int>> channels_mapping_;
 
  unsigned int next_time_slot_to_process_;
 
  common::NSTimer timer_;
  std::string step_name_;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif