SolutionWriter.h

Go to the documentation of this file.

// Copyright (C) 2021 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef DP3_DDECAL_SOLUTIONWRITER_H
#define DP3_DDECAL_SOLUTIONWRITER_H
 
#include "constraints/ConstraintResult.h"
 
#include "base/CalType.h"
#include "base/Direction.h"
 
#include <schaapcommon/h5parm/h5parm.h>
 
namespace dp3 {
namespace ddecal {
 
class SolutionWriter {
 public:
  explicit SolutionWriter(const std::string& filename);
 
  void AddAntennas(
      const std::vector<std::string>& all_antenna_names,
      const std::vector<std::array<double, 3>>& all_antenna_positions);
 
  void Write(
      const std::vector<std::vector<std::vector<std::complex<double>>>>&
          solutions,
      const std::vector<std::vector<std::vector<ddecal::ConstraintResult>>>&
          constraint_solutions,
      double start_time, double end_time, double ms_timestep_duration,
      size_t n_interval_timesteps,
      const std::vector<size_t>& solutions_per_direction, base::CalType mode,
      const std::vector<std::string>& used_antenna_names,
      const std::vector<base::Direction>& source_directions,
      const std::vector<std::vector<std::string>>& directions,
      const std::vector<double>& chan_freqs,
      const std::vector<double>& chan_block_freqs, const std::string& history);
 
 private:
  void WriteDirect(
      const std::vector<std::vector<std::vector<std::complex<double>>>>&
          solutions,
      const std::vector<std::vector<std::vector<ddecal::ConstraintResult>>>&
          constraint_solutions,
      double start_time, double end_time, double ms_timestep_duration,
      double solution_interval, base::CalType mode,
      const std::vector<std::string>& used_antenna_names,
      const std::vector<base::Direction>& source_directions,
      const std::vector<std::vector<std::string>>& directions,
      const std::vector<double>& chan_freqs,
      const std::vector<double>& chan_block_freqs, const std::string& history);
 
  void WriteSolverResults(
      const std::vector<std::vector<std::vector<std::complex<double>>>>&
          solutions,
      base::CalType mode, const std::vector<std::string>& used_antenna_names,
      const std::vector<base::Direction>& source_directions,
      const std::vector<std::vector<std::string>>& directions,
      const std::vector<double>& chan_freqs,
      const std::vector<double>& chan_block_freqs, const std::string& history,
      const std::vector<double>& solution_times);
 
  void WriteConstraintResults(
      const std::vector<std::vector<std::vector<ddecal::ConstraintResult>>>&
          constraint_solutions,
      base::CalType mode, const std::vector<std::string>& used_antenna_names,
      const std::vector<base::Direction>& source_directions,
      const std::vector<std::vector<std::string>>& directions,
      const std::vector<double>& chan_freqs,
      const std::vector<double>& chan_block_freqs, const std::string& history,
      const std::vector<double>& solution_times);
 
  schaapcommon::h5parm::H5Parm h5parm_;
};
 
}  // namespace ddecal
}  // namespace dp3
 
#endif