H5ParmPredict.h

Go to the documentation of this file.

// H5ParmPredict.h: DP3 step class to H5ParmPredict visibilities from a source
// model
//
// Copyright (C) 2023 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DP3_STEPS_H5PARM_PREDICT_H_
#define DP3_STEPS_H5PARM_PREDICT_H_
 
#include "base/DP3.h"
#include "steps/Step.h"
 
#include "common/Timer.h"
 
#include "Predict.h"
#include "ResultStep.h"
 
namespace dp3 {
namespace steps {
 
class H5ParmPredict : public Step {
 public:
  H5ParmPredict(const common::ParameterSet&, const std::string& prefix);
 
  common::Fields getRequiredFields() const override {
    // Combine the result of all sub steps.
    return base::GetChainRequiredFields(itsPredictSteps.front());
  }
 
  common::Fields getProvidedFields() const override {
    // Combine the result of all sub steps.
    common::Fields fields;
    for (std::shared_ptr<Step> step = itsPredictSteps.front(); step;
         step = step->getNextStep()) {
      fields |= step->getProvidedFields();
    }
    return fields;
  }
 
  bool process(std::unique_ptr<base::DPBuffer>) override;
 
  void finish() override;
 
  void updateInfo(const base::DPInfo&) override;
 
  void show(std::ostream&) const override;
 
  void showTimings(std::ostream&, double duration) const override;
 
 private:
  std::string itsName;
 
  std::vector<std::shared_ptr<Predict>> itsPredictSteps;
  std::shared_ptr<ResultStep> itsResultStep;
 
  std::string itsH5ParmName;
  std::vector<std::string> itsDirections;
 
  common::NSTimer itsTimer;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif