DynSpec.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_DYNSPEC_H_
#define DP3_STEPS_DYNSPEC_H_
 
#include "steps/Step.h"
 
#include "common/ParameterSet.h"
#include "common/Timer.h"
#include "common/DynSpecFitsWriter.h"
 
#include "model/Patch.h"
 
#include "ResultStep.h"
 
namespace dp3::steps {
 
class DynSpec : public Step {
 public:
  using DynamicSpectrumTensor =
      xt::xtensor<float, 4, xt::layout_type::column_major>;
 
  DynSpec(const common::ParameterSet&, const std::string& prefix);
 
  common::Fields getRequiredFields() const final {
    return kDataField | kUvwField | kWeightsField | kFlagsField;
  }
 
  common::Fields getProvidedFields() const final { return {}; }
 
  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:
  xt::xtensor<float, 2> ComputeAbsoluteStokesParameters(
      xt::xtensor<std::complex<float>, 2>& baseline_averaged_data) const;
 
  void WriteSpectraToDisk();
 
  size_t time_index_{0};
 
  std::string name_;
 
  std::string source_file_name_;
 
  std::string fits_prefix_;
 
  std::string model_column_;
 
  std::vector<std::shared_ptr<model::Patch>> source_list_;
 
  bool subtract_sources_{false};
  bool subtract_with_h5parmpredict_{false};
  bool subtract_model_column_{false};
 
  bool apply_calibration_solutions_{false};
 
  bool apply_beam_correction_{false};
  bool apply_beam_reweighted_{false};
 
  DynamicSpectrumTensor dynamic_spectra_;
 
  std::vector<std::shared_ptr<Step>> first_substeps_;
  std::vector<std::shared_ptr<ResultStep>> results_;
  std::shared_ptr<Step> model_step_;
  std::shared_ptr<ResultStep> model_result_;
 
  common::NSTimer total_timer_;
  common::NSTimer substep_timer_;
  common::NSTimer computation_timer_;
  common::NSTimer write_timer_;
};
 
}  // namespace dp3::steps
 
#endif