qnsolver.hpp

// SPDX-License-Identifier: MIT
 
#pragma once
#include "ddc_aliases.hpp"
#include "ichargedensitycalculator.hpp"
#include "ipoisson_solver.hpp"
#include "iqnsolver.hpp"
 
class QNSolver : public IQNSolver
{
    using PoissonSolver = IPoissonSolver<
            IdxRangeX,
            IdxRangeX,
            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()(
            DFieldX electrostatic_potential,
            DFieldX electric_field,
            DConstFieldSpXVx allfdistribu) const override;
};