PDEOps.H

Go to the documentation of this file.

#ifndef PDEOPS_H
#define PDEOPS_H
 
#include "src/core/FieldUtils.H"
#include "src/equation_systems/PDEHelpers.H"
#include "src/turbulence/TurbulenceModel.H"
#include "src/utilities/IOManager.H"
#include "src/CFDSim.H"
 
#include "AMReX_ParmParse.H"
#include "AMReX_REAL.H"
 
using namespace amrex::literals;
 
namespace kynema_sgf::pde {


template <typename PDE, typename Scheme>
struct FieldRegOp
{
    explicit FieldRegOp(CFDSim& sim_in) : sim(sim_in) {}

    PDEFields operator()(const SimTime& time)
    {
        auto fields = create_fields_instance<PDE, Scheme>(time, sim.repo());
 
        // Register solution variable for this PDE as output/restart variable
        sim.io_manager().register_io_var(fields.field.name());
        return fields;
    }
 
    CFDSim& sim;
};

template <typename PDE>
struct SrcTermOpBase
{
    explicit SrcTermOpBase(PDEFields& fields_in)
        : fields(fields_in), m_density(fields_in.repo.get_field("density"))
    {}

    void init_source_terms(const CFDSim& sim)
    {
        amrex::ParmParse pp(PDE::pde_name());
        amrex::Vector<std::string> src_terms;
        pp.queryarr("source_terms", src_terms);
 
        for (auto& src_name : src_terms) {
            // Prefer to use emplace_back here
            sources.emplace_back(PDE::SrcTerm::create(src_name, sim));
        }
    }

    void multiply_rho(const FieldState fstate)
    {
        BL_PROFILE("kynema-sgf::" + PDE::pde_name() + "::multiply_rho");
        const auto rhostate = field_impl::phi_state(fstate);
        const auto& density = m_density.state(rhostate);
 
        const int nlevels = fields.repo.num_active_levels();
        for (int lev = 0; lev < nlevels; ++lev) {
            auto& src_term = fields.src_term(lev);
            const auto& vf_arrs = src_term.arrays();
            const auto& rho_arrs = density(lev).const_arrays();
 
            amrex::ParallelFor(
                src_term, amrex::IntVect(0), fields.src_term.num_comp(),
                [=] AMREX_GPU_DEVICE(int nbx, int i, int j, int k, int n) {
                    vf_arrs[nbx](i, j, k, n) *= rho_arrs[nbx](i, j, k);
                });
        }
        amrex::Gpu::streamSynchronize();
    }

    virtual void
    operator()(const FieldState fstate, const bool /* mesh_mapping */)
    {
        // Zero out source term
        this->fields.src_term.setVal(0.0_rt);
 
        // Return early if there are no source terms to process
        if (this->sources.empty()) {
            return;
        }
 
        const int nlevels = this->fields.repo.num_active_levels();
        for (int lev = 0; lev < nlevels; ++lev) {
            auto& src_term = this->fields.src_term(lev);
 
            for (const auto& src : this->sources) {
                (*src)(lev, fstate, src_term);
            }
            amrex::Gpu::streamSynchronize();
        }
 
        if (PDE::multiply_rho) {
            this->multiply_rho(fstate);
        }
    }
 
    PDEFields& fields;
    Field& m_density;
    amrex::Vector<std::unique_ptr<typename PDE::SrcTerm>> sources;
};

template <typename PDE>
struct SrcTermOp : SrcTermOpBase<PDE>
{
    explicit SrcTermOp(PDEFields& fields_in) : SrcTermOpBase<PDE>(fields_in) {}
};
 
template <typename PDE, typename Scheme, typename = void>
struct AdvectionOp
{};
 
template <typename PDE, typename Scheme, typename = void>
struct DiffusionOp
{};

template <typename PDE>
struct TurbulenceOp
{
    // cppcheck-suppress uninitMemberVar
    TurbulenceOp(turbulence::TurbulenceModel& tmodel, PDEFields& fields)
        : m_tmodel(tmodel), m_fields(fields)
    {}
 
    void operator()()
    {
        m_tmodel.update_scalar_diff(m_fields.mueff, m_fields.field.name());
    }
 
    turbulence::TurbulenceModel& m_tmodel;
    PDEFields& m_fields;
};

template <typename PDE, typename = void>
struct BCOp;

template <typename PDE>
struct PostSolveOp
{
    explicit PostSolveOp(CFDSim& sim, PDEFields& fields)
        : m_sim(sim), m_fields(fields)
    {}
 
    void operator()(const amrex::Real time) { m_fields.field.fillpatch(time); }
 
    CFDSim& m_sim;
    PDEFields& m_fields;
};
 
} // namespace kynema_sgf::pde
 
#endif /* PDEOPS_H */