qnsolver.hpp

// SPDX-License-Identifier: MIT
 
#pragma once
#include "chargedensitycalculator.hpp"
#include "ddc_aliases.hpp"
#include "ipoisson_solver.hpp"
#include "iqnsolver.hpp"
 
class QNSolver : public IQNSolver
{
    using PoissonSolver = IPoissonSolver<
            IdxRangeXY,
            IdxRangeXY,
            typename Kokkos::DefaultExecutionSpace::memory_space,
            Kokkos::layout_right>;
    PoissonSolver const& m_solve_poisson;
    IChargeDensityCalculator const& m_compute_rho;
 
public:
    QNSolver(PoissonSolver const& solve_poisson, IChargeDensityCalculator const& compute_rho);
 
    ~QNSolver() override = default;
 
    void operator()(
            DFieldXY electrostatic_potential,
            DFieldXY electric_field_x,
            DFieldXY electric_field_y,
            DConstFieldSpXYVxVy allfdistribu) const override;
};