Transfer.h

Go to the documentation of this file.

// Copyright (C) 2024 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_STEPS_DATA_TRANSFER_H_
#define DP3_STEPS_DATA_TRANSFER_H_
 
#include <casacore/ms/MeasurementSets/MeasurementSet.h>
#include <casacore/tables/Tables/TableIter.h>
#include "base/DPBuffer.h"
#include "Filter.h"
#include "ResultStep.h"
#include "Step.h"
#include "common/Timer.h"
 
namespace dp3 {
namespace steps {
 
 
 
 
class Transfer final : public Step {
 public:
  explicit Transfer(const common::ParameterSet& parameter_set,
                    const std::string& prefix);
 
  common::Fields getRequiredFields() const final {
    return filter_step_->getRequiredFields();
  }
 
  common::Fields getProvidedFields() const final {
    common::Fields provided_fields{};
    provided_fields |= (transfer_data_ && output_buffer_name_.empty())
                           ? kDataField
                           : common::Fields();
    provided_fields |= transfer_flags_ ? kFlagsField : common::Fields();
    return provided_fields;
  }
 
  bool process(std::unique_ptr<base::DPBuffer> buffer) final;
 
  void finish() final;
 
  void updateInfo(const base::DPInfo&) final;
 
  void show(std::ostream&) const final;
 
  void showTimings(std::ostream&, double duration) const final;
 
 private:
  void ReadSourceMsVisibilities();
 
  void ReadSourceMsFlags();
 
  template <typename T>
  void TransferSingleTimeSlot(
      T source, T& target,
      const std::vector<common::rownr_t>& target_row_numbers,
      const std::vector<double>& target_frequencies) const;
 
  std::string name_;
  common::NSTimer timer_;
  std::string source_ms_path_;
  std::string source_data_column_;
  casacore::MeasurementSet ms_;
  casacore::TableIterator ms_iterator_;
 
  base::DPBuffer::DataType data_;
  base::DPBuffer::FlagsType flags_;
 
  bool transfer_data_;
  bool transfer_flags_;
 
  std::string output_buffer_name_;
 
  std::size_t timestep_counter_ = 0;
 
  std::size_t time_averaging_factor_;
 
  double time_interval_;
 
  std::vector<double> source_channel_upper_edges_;
 
  std::shared_ptr<Step> filter_step_;
  std::shared_ptr<ResultStep> result_step_;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif