WarpXFluidContainer.H

Go to the documentation of this file.

/* Copyright 2023 Grant Johnson, Remi Lehe
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_WarpXFluidContainer_H_
#define WARPX_WarpXFluidContainer_H_
 
#include "Evolve/WarpXDtType.H"
#include "Initialization/PlasmaInjector.H"
#include "MultiFluidContainer.H"
 
#include<AMReX_MultiFab.H>
#include<AMReX_Vector.H>
 
#include <string>
 
 
class WarpXFluidContainer
{
public:
    friend MultiFluidContainer;
 
    WarpXFluidContainer (int nlevs_max, int ispecies, const std::string& name);
    ~WarpXFluidContainer() {}
 
    WarpXFluidContainer (WarpXFluidContainer const &)             = delete;
    WarpXFluidContainer& operator= (WarpXFluidContainer const & ) = delete;
    WarpXFluidContainer(WarpXFluidContainer&& )                   = default;
    WarpXFluidContainer& operator=(WarpXFluidContainer&& )        = default;
 
    void AllocateLevelMFs (int lev, const amrex::BoxArray& ba, const amrex::DistributionMapping& dm);
 
    void InitData (int lev, amrex::Box init_box, amrex::Real cur_time);
 
    void ReadParameters ();
 
    void Evolve (int lev,
        const amrex::MultiFab& Ex, const amrex::MultiFab& Ey, const amrex::MultiFab& Ez,
        const amrex::MultiFab& Bx, const amrex::MultiFab& By, const amrex::MultiFab& Bz,
        amrex::MultiFab* rho, amrex::MultiFab& jx, amrex::MultiFab& jy, amrex::MultiFab& jz,
        amrex::Real cur_time, bool skip_deposition=false);
 
    void AdvectivePush_Muscl (int lev);
 
 
    void ApplyBcFluidsAndComms (int lev);
 
#if defined(WARPX_DIM_RZ)
    void centrifugal_source_rz (int lev);
#endif
 
    void GatherAndPush (int lev,
        const amrex::MultiFab& Ex, const amrex::MultiFab& Ey, const amrex::MultiFab& Ez,
        const amrex::MultiFab& Bx, const amrex::MultiFab& By, const amrex::MultiFab& Bz,
        amrex::Real cur_time);
 
    void DepositCurrent (int lev,
        amrex::MultiFab& jx, amrex::MultiFab& jy, amrex::MultiFab& jz);
 
    void DepositCharge (int lev, amrex::MultiFab &rho, int icomp = 0);
 
    amrex::Real getCharge () const {return charge;}
    amrex::Real getMass () const {return mass;}
 
protected:
    int species_id;
    std::string species_name;
    amrex::Real charge;
    amrex::Real mass;
 
    int do_not_push = 0;
    int do_not_gather = 0;
    int do_not_deposit = 0;
    PhysicalSpecies physical_species;
 
    // Parser for external fields
    std::string m_B_ext_s = "none";
    std::string m_E_ext_s = "none";
 
    // Parser for B_external on the particle
    std::unique_ptr<amrex::Parser> m_Bx_parser;
    std::unique_ptr<amrex::Parser> m_By_parser;
    std::unique_ptr<amrex::Parser> m_Bz_parser;
    amrex::ParserExecutor<4> m_Bxfield_parser;
    amrex::ParserExecutor<4> m_Byfield_parser;
    amrex::ParserExecutor<4> m_Bzfield_parser;
 
    // Parser for E_external on the particle
    std::unique_ptr<amrex::Parser> m_Ex_parser;
    std::unique_ptr<amrex::Parser> m_Ey_parser;
    std::unique_ptr<amrex::Parser> m_Ez_parser;
    amrex::ParserExecutor<4> m_Exfield_parser;
    amrex::ParserExecutor<4> m_Eyfield_parser;
    amrex::ParserExecutor<4> m_Ezfield_parser;
 
    std::unique_ptr<InjectorDensity,InjectorDensityDeleter> h_inj_rho;
    InjectorDensity* d_inj_rho = nullptr;
    std::unique_ptr<amrex::Parser> density_parser;
 
    std::unique_ptr<InjectorMomentum,InjectorMomentumDeleter> h_inj_mom;
    InjectorMomentum* d_inj_mom = nullptr;
    std::unique_ptr<amrex::Parser> ux_parser;
    std::unique_ptr<amrex::Parser> uy_parser;
    std::unique_ptr<amrex::Parser> uz_parser;
 
    // Keep a pointer to TemperatureProperties to ensure the lifetime of the
    // contained Parser
    std::unique_ptr<TemperatureProperties> h_mom_temp;
    std::unique_ptr<VelocityProperties> h_mom_vel;
 
public:
 
   // MultiFabs that contain the density (N) and momentum density (NU) of this fluid species, for each refinement level
    amrex::Vector<            std::unique_ptr<amrex::MultiFab>      > N;
    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3 > > NU;
 
};
 
#endif