poisson_like_rhs_function.hpp

// SPDX-License-Identifier: MIT
#pragma once
 
#include <ddc/ddc.hpp>
 
#include "geometry.hpp"
#include "spline_interpolator_2d_rp.hpp"
 
 
 
template <class RadialExtrapolationRule>
class PoissonLikeRHSFunction
{
public:
    using evaluator_type = ddc::SplineEvaluator2D<
            Kokkos::DefaultHostExecutionSpace,
            Kokkos::HostSpace,
            BSplinesR,
            BSplinesTheta,
            GridR,
            GridTheta,
            RadialExtrapolationRule,
            RadialExtrapolationRule,
            ddc::PeriodicExtrapolationRule<Theta>,
            ddc::PeriodicExtrapolationRule<Theta>,
            GridR,
            GridTheta>;
 
private:
    host_t<ConstSpline2D> const m_coefs;
    evaluator_type const& m_evaluator;
 
public:
    PoissonLikeRHSFunction(host_t<ConstSpline2D> coefs, evaluator_type const& evaluator)
        : m_coefs(coefs)
        , m_evaluator(evaluator)
    {
    }
 
    double operator()(CoordRTheta const& coord_rp) const
    {
        return m_evaluator(coord_rp, m_coefs);
    }
};