MSUpdater.h

Go to the documentation of this file.

// MSUpdater.h: DPPP step writing to an MS
// Copyright (C) 2022 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef DP3_STEPS_MSUPDATER_H_
#define DP3_STEPS_MSUPDATER_H_
 
#include <casacore/tables/Tables/ColumnDesc.h>
#include <casacore/tables/Tables/RefRows.h>
#include <casacore/tables/Tables/Table.h>
 
#include "base/StManParsetKeys.h"
#include "common/ParameterSet.h"
#include "common/Timer.h"
 
#include "OutputStep.h"
 
namespace dp3 {
namespace steps {
 
 
//
 
class MSUpdater : public OutputStep {
 public:
  explicit MSUpdater(std::string msName, const common::ParameterSet& parset,
                     const std::string& prefix, bool writeHistory = true);
 
  common::Fields getRequiredFields() const override;
 
  void SetFieldsToWrite(const common::Fields& fields) 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 showTimings(std::ostream&, double duration) const override;
 
  static bool updateAllowed(const base::DPInfo& info, casacore::String msName,
                            bool throwError = true);
 
 private:
  void putFlags(const casacore::RefRows& rowNrs,
                const casacore::Cube<bool>& flags);
 
  void putWeights(const casacore::RefRows& rowNrs,
                  const casacore::Cube<float>& weights);
 
  void putData(const casacore::RefRows& rowNrs,
               const casacore::Cube<casacore::Complex>& data);
 
  bool addColumn(const std::string& colname, const casacore::DataType dataType,
                 const casacore::ColumnDesc& cd);
 
  std::string itsName;
  std::string itsMSName;
  casacore::Table itsMS;
  const common::ParameterSet& itsParset;
  base::DPBuffer itsBuffer;
  std::string itsDataColName;
  std::string itsFlagColName;
  std::string itsWeightColName;
  unsigned int itsNrTimesFlush;  
  unsigned int itsNrDone;        
  bool itsDataColAdded;          
  bool itsFlagColAdded;          
  bool itsWeightColAdded;        
  bool itsWriteHistory;          
  common::NSTimer itsTimer;
  unsigned int itsTileSize;
  base::StManParsetKeys itsStManKeys;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif