IterativeScalarSolver.h

Go to the documentation of this file.

// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef DDECAL_ITERATIVE_SCALAR_SOLVER_H
#define DDECAL_ITERATIVE_SCALAR_SOLVER_H
 
#include "SolverBase.h"
#include "SolveData.h"
 
namespace dp3 {
namespace ddecal {
 
template <typename VisMatrix>
class IterativeScalarSolver final : public SolverBase {
 public:
  using ChannelBlockData = typename SolveData<VisMatrix>::ChannelBlockData;
  SolveResult Solve(const SolveData<VisMatrix>& data,
                    std::vector<std::vector<DComplex>>& solutions,
                    double time) override;
 
  size_t NSolutionPolarizations() const override { return 1; }
 
  bool SupportsDdSolutionIntervals() const override { return true; }
 
 private:
  void PerformIteration(size_t ch_block, const ChannelBlockData& cb_data,
                        std::vector<VisMatrix>& v_residual,
                        const std::vector<DComplex>& solutions,
                        SolutionTensor& next_solutions);
 
  template <bool Add>
  void AddOrSubtractDirection(const ChannelBlockData& cb_data,
                              std::vector<VisMatrix>& v_residual,
                              size_t direction,
                              const std::vector<DComplex>& solutions);
 
  void SolveDirection(size_t ch_block, const ChannelBlockData& cb_data,
                      const std::vector<VisMatrix>& v_residual,
                      size_t direction, const std::vector<DComplex>& solutions,
                      SolutionTensor& next_solutions);
};
 
extern template class IterativeScalarSolver<std::complex<float>>;
extern template class IterativeScalarSolver<aocommon::MC2x2F>;
extern template class IterativeScalarSolver<aocommon::MC2x2FDiag>;
 
}  // namespace ddecal
}  // namespace dp3
 
#endif  // DDECAL_ITERATIVE_SCALAR_SOLVER_H