4 #ifdef ERF_USE_PARTICLES
7 #include <AMReX_Particles.H>
9 struct ERFParticlesIntIdxAoS
17 struct ERFParticlesRealIdxAoS
24 struct ERFParticlesIntIdxSoA
31 struct ERFParticlesRealIdxSoA
43 namespace ERFParticleInitializations
46 const std::string init_box_uniform =
"box";
49 namespace ERFParticleNames
51 const std::string tracers =
"tracer_particles";
52 const std::string hydro =
"hydro_particles";
55 struct ERFParticlesAssignor
59 amrex::IntVect operator() ( P
const& p,
60 amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>
const& plo,
61 amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>
const& dxi,
62 const amrex::Box& domain )
const noexcept
65 AMREX_D_DECL(
int(amrex::Math::floor((p.pos(0)-plo[0])*dxi[0])),
66 int(amrex::Math::floor((p.pos(1)-plo[1])*dxi[1])),
67 p.idata(ERFParticlesIntIdxAoS::k) ) );
68 iv[0] += domain.smallEnd()[0];
69 iv[1] += domain.smallEnd()[1];
75 AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
76 void update_location_idata ( P& a_p,
77 amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>
const& a_plo,
78 amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>
const& a_dxi,
79 const amrex::Array4<amrex::Real const>& a_height_arr )
81 amrex::IntVect iv(
int(amrex::Math::floor((a_p.pos(0)-a_plo[0])*a_dxi[0])),
82 int(amrex::Math::floor((a_p.pos(1)-a_plo[1])*a_dxi[1])),
83 a_p.idata(ERFParticlesIntIdxAoS::k) );
86 amrex::Real lx = (a_p.pos(0)-a_plo[0])*a_dxi[0] -
static_cast<amrex::Real
>(iv[0]);
87 amrex::Real ly = (a_p.pos(1)-a_plo[1])*a_dxi[1] -
static_cast<amrex::Real
>(iv[1]);
88 auto zlo = a_height_arr(iv[0] ,iv[1] ,iv[2] ) * (1.0-lx) * (1.0-ly) +
89 a_height_arr(iv[0]+1,iv[1] ,iv[2] ) * lx * (1.0-ly) +
90 a_height_arr(iv[0] ,iv[1]+1,iv[2] ) * (1.0-lx) * ly +
91 a_height_arr(iv[0]+1,iv[1]+1,iv[2] ) * lx * ly;
92 auto zhi = a_height_arr(iv[0] ,iv[1] ,iv[2]+1) * (1.0-lx) * (1.0-ly) +
93 a_height_arr(iv[0]+1,iv[1] ,iv[2]+1) * lx * (1.0-ly) +
94 a_height_arr(iv[0] ,iv[1]+1,iv[2]+1) * (1.0-lx) * ly +
95 a_height_arr(iv[0]+1,iv[1]+1,iv[2]+1) * lx * ly;
97 if (a_p.pos(2) > zhi) {
98 a_p.idata(ERFParticlesIntIdxAoS::k) += 1;
99 }
else if (a_p.pos(2) <= zlo) {
100 a_p.idata(ERFParticlesIntIdxAoS::k) -= 1;
105 class ERFPC :
public amrex::ParticleContainer< ERFParticlesRealIdxAoS::ncomps,
106 ERFParticlesIntIdxAoS::ncomps,
107 ERFParticlesRealIdxSoA::ncomps,
108 ERFParticlesIntIdxSoA::ncomps,
109 amrex::DefaultAllocator,
110 ERFParticlesAssignor >
115 ERFPC ( amrex::ParGDBBase* a_gdb,
116 const std::string& a_name =
"particles" )
117 :
amrex::ParticleContainer< ERFParticlesRealIdxAoS::ncomps,
118 ERFParticlesIntIdxAoS::ncomps,
119 ERFParticlesRealIdxSoA::ncomps,
120 ERFParticlesIntIdxSoA::ncomps,
121 amrex::DefaultAllocator,
122 ERFParticlesAssignor> (a_gdb)
124 BL_PROFILE(
"ERFPCPC::ERFPC()");
130 ERFPC (
const amrex::Geometry& a_geom,
131 const amrex::DistributionMapping& a_dmap,
132 const amrex::BoxArray& a_ba,
133 const std::string& a_name =
"particles" )
134 :
amrex::ParticleContainer< ERFParticlesRealIdxAoS::ncomps,
135 ERFParticlesIntIdxAoS::ncomps,
136 ERFParticlesRealIdxSoA::ncomps,
137 ERFParticlesIntIdxSoA::ncomps,
138 amrex::DefaultAllocator,
139 ERFParticlesAssignor> ( a_geom, a_dmap, a_ba )
141 BL_PROFILE(
"ERFPCPC::ERFPC()");
147 virtual void InitializeParticles (
const std::unique_ptr<amrex::MultiFab>& a_ptr =
nullptr);
150 virtual void EvolveParticles (
int,
152 amrex::Vector<amrex::Vector<amrex::MultiFab>>&,
153 const amrex::Vector<std::unique_ptr<amrex::MultiFab>>& );
156 virtual amrex::Vector<std::string> varNames ()
const
158 BL_PROFILE(
"ERFPCPC::varNames()");
159 return {AMREX_D_DECL(
"xvel",
"yvel",
"zvel"),
"mass",
"temperature"};
163 virtual amrex::Vector<std::string> meshPlotVarNames ()
const
165 BL_PROFILE(
"ERFPCPC::varNames()");
166 return {
"mass_density"};
170 virtual void AdvectWithFlow ( amrex::MultiFab*,
173 const std::unique_ptr<amrex::MultiFab>& );
176 virtual void AdvectWithGravity (
int,
178 const std::unique_ptr<amrex::MultiFab>& );
181 virtual void ComputeTemperature (
const amrex::MultiFab&,
184 const std::unique_ptr<amrex::MultiFab>& );
187 virtual void massDensity ( amrex::MultiFab&,
const int&,
const int& a_comp = 0)
const;
190 virtual void computeMeshVar(
const std::string& a_var_name,
191 amrex::MultiFab& a_mf,
192 const int a_lev)
const
194 if (a_var_name ==
"mass_density") {
195 massDensity( a_mf, a_lev );
202 inline void setAdvectWithFlow (
bool a_flag)
204 BL_PROFILE(
"ERFPCPC::setAdvectWithFlow()");
205 m_advect_w_flow = a_flag;
208 inline void setAdvectWithGravity (
bool a_flag)
210 BL_PROFILE(
"ERFPCPC::setAdvectWithGravity()");
211 m_advect_w_gravity = a_flag;
218 void initializeParticlesUniformDistributionInBox (
const std::unique_ptr<amrex::MultiFab>& a_ptr,
219 const amrex::RealBox& particle_box);
223 bool m_advect_w_flow;
224 bool m_advect_w_gravity;
226 amrex::RealBox m_particle_box;
230 std::string m_initialization_type;
233 bool m_stable_redistribute;
236 virtual void readInputs ();
240 bool place_randomly_in_cells;
Definition: ERF_ConsoleIO.cpp:12