MSBDAWriter.h

Go to the documentation of this file.

// Copyright (C) 2022 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef DP3_STEPS_MSBDAWRITER_H_
#define DP3_STEPS_MSBDAWRITER_H_
 
#include <map>
 
#include <casacore/tables/Tables/Table.h>
 
#include "base/StManParsetKeys.h"
#include "common/ParameterSet.h"
 
#include "OutputStep.h"
 
namespace dp3 {
namespace steps {
 
class MSBDAWriter : public OutputStep {
 public:
  explicit MSBDAWriter(const std::string& out_name,
                       const common::ParameterSet& parset,
                       const std::string& prefix);
 
  common::Fields getRequiredFields() const override {
    return kDataField | kFlagsField | kWeightsField | kUvwField;
  }
 
  void updateInfo(const base::DPInfo&) override;
 
  bool process(std::unique_ptr<base::BdaBuffer>) override;
 
  void finish() override;
 
  void show(std::ostream&) const override;
 
 private:
  void CreateMS();
 
  void CreateMainTable();
 
  void CreateBDATimeAxis();
 
  void CreateBDATimeFactor();
 
  void CreateMetaDataFrequencyColumns();
 
  void WriteMetaData();
 
  void WriteTimeFactorRows(casacore::Int bda_set_id,
                           unsigned int& min_factor_time,
                           unsigned int& max_factor_time);
 
  void WriteTimeAxisRow(casacore::Int bda_set_id, unsigned int min_factor_time,
                        unsigned int max_factor_time);
 
  void OverwriteSubTables(casacore::Int bda_set_id);
 
 private:
  const std::string out_name_;
  const common::ParameterSet parset_;
  const std::string prefix_;
  const bool overwrite_;
  base::StManParsetKeys st_man_keys_;
 
  std::map<std::size_t, unsigned int> nchanToDescId;
  casacore::Table ms_;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif