MultiMsReader.h

Go to the documentation of this file.

// MultiMsReader.h: DP3 step reading from multiple MSs
// Copyright (C) 2023 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_STEPS_MULTIMSREADER_H_
#define DP3_STEPS_MULTIMSREADER_H_
 
#include <casacore/tables/Tables/TableIter.h>
#include <casacore/tables/Tables/RefRows.h>
#include <casacore/casa/Arrays/Slicer.h>
 
#include "base/UVWCalculator.h"
#include "base/FlagCounter.h"
#include "common/ParameterSet.h"
 
#include "MsReader.h"
#include "ResultStep.h"
 
namespace dp3 {
namespace steps {
 
 
class MultiMsReader final : public InputStep {
 public:
  MultiMsReader(const std::vector<std::string>& msNames,
                const common::ParameterSet& parset, const std::string& prefix);
 
  ~MultiMsReader() override;
 
  bool process(std::unique_ptr<base::DPBuffer> buffer) override;
 
  void finish() override;
 
  void updateInfo(const base::DPInfo&) override;
 
  void show(std::ostream&) const override;
 
  void showCounts(std::ostream&) const override;
 
  void showTimings(std::ostream&, double duration) const override;
 
  void setFieldsToRead(const dp3::common::Fields& fields) override;
 
  std::string msName() const override { return readers_.front().name; }
 
  const casacore::Table& table() const override {
    return readers_[first_].ms_reader->table();
  }
 
 private:
  void ValidateBands();
 
  void HandleBands();
 
  void SortBands();
 
  void FillBands();
 
  void GetWeights(std::unique_ptr<base::DPBuffer>& buffer);
 
  bool order_;              
  int first_;               
  unsigned int n_missing_;  
  struct Reader {
    std::string name;
    std::shared_ptr<MsReader> ms_reader;
    std::shared_ptr<ResultStep> result;
  };
  std::vector<Reader> readers_;
  unsigned int n_fill_channels_;  
};
 
}  // namespace steps
}  // namespace dp3
 
#endif