SolutionResampler.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_SOLUTIONRESAMPLER_H_
#define DP3_DDECAL_SOLUTIONRESAMPLER_H_
 
#include <complex>
#include <vector>
 
#include "constraints/ConstraintResult.h"
 
namespace dp3 {
namespace ddecal {
 
class SolutionResampler {
 public:
  SolutionResampler(const std::vector<size_t>& n_solutions_per_direction,
                    size_t n_antennas, size_t solution_interval);
  ~SolutionResampler() = default;
 
  std::vector<std::vector<std::vector<std::complex<double>>>> Upsample(
      const std::vector<std::vector<std::vector<std::complex<double>>>>&
          solutions,
      size_t n_pol) const;
 
  std::vector<std::vector<std::vector<ConstraintResult>>> Upsample(
      const std::vector<std::vector<std::vector<ConstraintResult>>>& solutions)
      const;
 
  size_t GetNrSubSteps() const { return n_substeps_; }
 
  std::pair<size_t, size_t> MapResampledToOriginal(size_t time_index,
                                                   size_t antenna_index,
                                                   size_t direction_index,
                                                   size_t pol_index,
                                                   size_t n_pol) const;
 
 private:
  const std::vector<size_t> n_solutions_per_direction_;
  const size_t n_sub_solutions_;
  const size_t n_directions_;
 
  const size_t n_antennas_;
  const size_t solution_interval_;
  size_t n_substeps_;
};
 
}  // namespace ddecal
}  // namespace dp3
 
#endif