7 #ifndef WARPX_PARTICLES_COLLISION_BINARYCOLLISION_H_
8 #define WARPX_PARTICLES_COLLISION_BINARYCOLLISION_H_
32 #include <AMReX_Config.H>
68 template <
typename CollisionFunctor,
114 " does not produce species. Thus, `product_species` should not be specified in the input script." );
148 for (
int i = 0;
i < n_product_species;
i++)
152 product_species_vector.push_back(&product);
158 copy_species1.push_back(copy_factory_species1[
i].getSmartCopy());
159 copy_species2.push_back(copy_factory_species2[
i].getSmartCopy());
165 copy_species1.end(), device_copy_species1.
begin());
167 copy_species2.end(), device_copy_species2.
begin());
169 auto *copy_species1_data = device_copy_species1.
data();
170 auto *copy_species2_data = device_copy_species2.
data();
172 auto *copy_species1_data = copy_species1.data();
173 auto *copy_species2_data = copy_species2.data();
185 for (
int lev = 0; lev <=
species1.finestLevel(); ++lev){
192 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
202 copy_species1_data, copy_species2_data);
242 using namespace amrex::literals;
253 constexpr
int getpos_offset = 0;
254 for (
int i = 0;
i < n_product_species;
i++)
256 ParticleTileType& ptile_product = product_species_vector[
i]->ParticlesAt(lev, mfi);
257 tile_products.push_back(&ptile_product);
261 products_mass.push_back(product_species_vector[
i]->getMass());
263 auto *tile_products_data = tile_products.data();
276 auto const n_cells =
static_cast<int>(bins_1.
numBins());
281 const amrex::ParticleReal q1 = species_1.
getCharge();
282 const amrex::ParticleReal m1 = species_1.
getMass();
286 #if defined WARPX_DIM_1D_Z
288 #elif defined WARPX_DIM_XZ
290 #elif defined WARPX_DIM_RZ
292 const auto lo =
lbound(cbx);
293 const auto hi =
ubound(cbx);
294 int const nz = hi.y-lo.y+1;
297 #elif defined(WARPX_DIM_3D)
305 const int n_cells_products = have_product_species ? n_cells: 0;
315 const auto n_part_in_cell = cell_offsets_1[i_cell+1] - cell_offsets_1[i_cell];
317 p_n_pairs_in_each_cell[i_cell] = (n_part_in_cell == 1)? 0: (n_part_in_cell+1)/2;
323 const index_type n_total_pairs = (n_cells_products == 0) ? 0:
325 p_n_pairs_in_each_cell, pair_offsets.data());
334 const auto n_part_in_cell = (i_cell < n_cells)? cell_offsets_1[i_cell+1] - cell_offsets_1[i_cell]: 0;
336 p_n_ind_pairs_in_each_cell[i_cell] = n_part_in_cell/2;
360 pair_reaction_weight.
dataPtr();
371 index_type const cell_start_1 = cell_offsets_1[i_cell];
372 index_type const cell_stop_1 = cell_offsets_1[i_cell+1];
373 index_type const cell_half_1 = (cell_start_1+cell_stop_1)/2;
376 if ( cell_stop_1 - cell_start_1 <= 1 ) {
return; }
380 indices_1, cell_start_1, cell_half_1, engine );
398 const int i_cell =
amrex::bisect( p_coll_offsets, 0, n_cells, ui_coll );
402 index_type const cell_start_1 = cell_offsets_1[i_cell];
403 index_type const cell_stop_1 = cell_offsets_1[i_cell+1];
404 index_type const cell_half_1 = (cell_start_1+cell_stop_1)/2;
407 const index_type coll_idx = ui_coll - p_coll_offsets[i_cell];
410 index_type const cell_start_pair = have_product_species?
411 p_pair_offsets[i_cell] : 0;
413 #if defined WARPX_DIM_RZ
414 const int ri = (i_cell - i_cell%nz) / nz;
415 auto dV = MathConst::pi*(2.0_prt*ri+1.0_prt)*dr*dr*
dz;
420 binary_collision_functor(
421 cell_start_1, cell_half_1,
422 cell_half_1, cell_stop_1,
423 indices_1, indices_1,
424 soa_1, soa_1, get_position_1, get_position_1,
425 q1, q1, m1, m1,
dt, dV, coll_idx,
426 cell_start_pair, p_mask, p_pair_indices_1, p_pair_indices_2,
427 p_pair_reaction_weight, engine);
435 product_species_vector,
438 products_mass, p_mask, products_np,
439 copy_species1, copy_species2,
440 p_pair_indices_1, p_pair_indices_2,
441 p_pair_reaction_weight);
443 for (
int i = 0;
i < n_product_species;
i++)
461 auto const n_cells =
static_cast<int>(bins_1.
numBins());
466 const amrex::ParticleReal q1 = species_1.
getCharge();
467 const amrex::ParticleReal m1 = species_1.
getMass();
473 const amrex::ParticleReal q2 = species_2.
getCharge();
474 const amrex::ParticleReal m2 = species_2.
getMass();
478 #if defined WARPX_DIM_1D_Z
480 #elif defined WARPX_DIM_XZ
482 #elif defined WARPX_DIM_RZ
484 const auto lo =
lbound(cbx);
485 const auto hi =
ubound(cbx);
486 const int nz = hi.y-lo.y+1;
489 #elif defined(WARPX_DIM_3D)
496 const int n_cells_products = have_product_species ? n_cells: 0;
506 const auto n_part_in_cell_1 = cell_offsets_1[i_cell+1] - cell_offsets_1[i_cell];
507 const auto n_part_in_cell_2 = cell_offsets_2[i_cell+1] - cell_offsets_2[i_cell];
509 if (n_part_in_cell_1 == 0 || n_part_in_cell_2 == 0) {
510 p_n_pairs_in_each_cell[i_cell] = 0;
512 p_n_pairs_in_each_cell[i_cell] =
513 amrex::max(n_part_in_cell_1,n_part_in_cell_2);
520 const index_type n_total_pairs = (n_cells_products == 0) ? 0:
522 p_n_pairs_in_each_cell, pair_offsets.data());
531 if (i_cell < n_cells)
533 const auto n_part_in_cell_1 = cell_offsets_1[i_cell+1] - cell_offsets_1[i_cell];
534 const auto n_part_in_cell_2 = cell_offsets_2[i_cell+1] - cell_offsets_2[i_cell];
535 p_n_ind_pairs_in_each_cell[i_cell] =
amrex::min(n_part_in_cell_1, n_part_in_cell_2);
539 p_n_ind_pairs_in_each_cell[i_cell] = 0;
564 pair_reaction_weight.
dataPtr();
576 index_type const cell_start_1 = cell_offsets_1[i_cell];
577 index_type const cell_stop_1 = cell_offsets_1[i_cell+1];
579 index_type const cell_start_2 = cell_offsets_2[i_cell];
580 index_type const cell_stop_2 = cell_offsets_2[i_cell+1];
587 if ( cell_stop_1 - cell_start_1 < 1 ||
588 cell_stop_2 - cell_start_2 < 1 ) {
return; }
610 const int i_cell =
amrex::bisect( p_coll_offsets, 0, n_cells, ui_coll );
614 index_type const cell_start_1 = cell_offsets_1[i_cell];
615 index_type const cell_stop_1 = cell_offsets_1[i_cell+1];
617 index_type const cell_start_2 = cell_offsets_2[i_cell];
618 index_type const cell_stop_2 = cell_offsets_2[i_cell+1];
621 const index_type coll_idx = ui_coll - p_coll_offsets[i_cell];
624 index_type const cell_start_pair = have_product_species?
625 p_pair_offsets[i_cell]: 0;
631 #if defined WARPX_DIM_RZ
632 const int ri = (i_cell - i_cell%nz) / nz;
633 auto dV = MathConst::pi*(2.0_prt*ri+1.0_prt)*dr*dr*
dz;
638 binary_collision_functor(
639 cell_start_1, cell_stop_1, cell_start_2, cell_stop_2,
640 indices_1, indices_2,
641 soa_1, soa_2, get_position_1, get_position_2,
642 q1, q2, m1, m2,
dt, dV, coll_idx,
643 cell_start_pair, p_mask, p_pair_indices_1, p_pair_indices_2,
644 p_pair_reaction_weight, engine);
652 product_species_vector,
655 products_mass, p_mask, products_np,
656 copy_species1, copy_species2,
657 p_pair_indices_1, p_pair_indices_2,
658 p_pair_reaction_weight);
660 for (
int i = 0;
i < n_product_species;
i++)
CollisionType
Definition: BinaryCollisionUtils.H:17
AMREX_GPU_HOST_DEVICE AMREX_INLINE void ShuffleFisherYates(T_index *array, T_index const is, T_index const ie, amrex::RandomEngine const &engine)
Definition: ShuffleFisherYates.H:20
void setNewParticleIDs(PTile &ptile, amrex::Long old_size, amrex::Long num_added)
Sets the ids of newly created particles to the next values.
Definition: SmartUtils.H:52
#define WARPX_ABORT_WITH_MESSAGE(MSG)
Definition: TextMsg.H:15
This class performs generic binary collisions.
Definition: BinaryCollision.H:72
bool m_have_product_species
Definition: BinaryCollision.H:672
ParticleBins::index_type index_type
Definition: BinaryCollision.H:79
CopyTransformFunctor m_copy_transform_functor
Definition: BinaryCollision.H:677
bool m_isSameSpecies
Definition: BinaryCollision.H:671
WarpXParticleContainer::ParticleType ParticleType
Definition: BinaryCollision.H:74
BinaryCollision(std::string collision_name, MultiParticleContainer const *const mypc)
Constructor of the BinaryCollision class.
Definition: BinaryCollision.H:89
~BinaryCollision() override=default
void doCollisions(amrex::Real cur_time, amrex::Real dt, MultiParticleContainer *mypc) override
Definition: BinaryCollision.H:134
void doCollisionsWithinTile(amrex::Real dt, int const lev, amrex::MFIter const &mfi, WarpXParticleContainer &species_1, WarpXParticleContainer &species_2, amrex::Vector< WarpXParticleContainer * > product_species_vector, SmartCopy *copy_species1, SmartCopy *copy_species2)
Definition: BinaryCollision.H:234
BinaryCollision & operator=(BinaryCollision const &)=default
amrex::Vector< std::string > m_product_species
Definition: BinaryCollision.H:673
CollisionFunctor m_binary_collision_functor
Definition: BinaryCollision.H:675
BinaryCollision(BinaryCollision const &)=default
BinaryCollision(BinaryCollision &&)=delete
Definition: CollisionBase.H:18
amrex::Vector< std::string > m_species_names
Definition: CollisionBase.H:36
Definition: MultiParticleContainer.H:66
WarpXParticleContainer & GetParticleContainerFromName(const std::string &name) const
Definition: MultiParticleContainer.cpp:392
This class does nothing and is used as second template parameter for binary collisions that do not cr...
Definition: ParticleCreationFunc.H:309
A factory for creating SmartCopy functors.
Definition: SmartCopy.H:133
static WarpX & GetInstance()
Definition: WarpX.cpp:239
static amrex::LayoutData< amrex::Real > * getCosts(int lev)
Definition: WarpX.cpp:3045
static short load_balance_costs_update_algo
Definition: WarpX.H:173
Definition: WarpXParticleContainer.H:111
void defineAllParticleTiles() noexcept
Definition: WarpXParticleContainer.cpp:1518
amrex::ParticleReal getCharge() const
Definition: WarpXParticleContainer.H:371
amrex::ParticleReal getMass() const
Definition: WarpXParticleContainer.H:373
const Vector< Geometry > & Geom() const noexcept
const Real * CellSize() const noexcept
index_type * offsetsPtr() noexcept
index_type * permutationPtr() noexcept
Long numBins() const noexcept
AMREX_GPU_HOST_DEVICE static constexpr AMREX_FORCE_INLINE IntVect TheZeroVector() noexcept
Box tilebox() const noexcept
iterator begin() noexcept
int queryarr(const char *name, std::vector< int > &ref, int start_ix=FIRST, int num_val=ALL) const
ParticleTile< ParticleType, NArrayReal, NArrayInt, Allocator > ParticleTileType
const ParticleTileType & ParticlesAt(int lev, int grid, int tile) const
T_ParticleType ParticleType
CollisionType get_collision_type(const std::string &collision_name, MultiParticleContainer const *const mypc)
Definition: BinaryCollisionUtils.cpp:20
Definition: ParticleUtils.cpp:26
ParticleBins findParticlesInEachCell(int lev, MFIter const &mfi, ParticleTileType &ptile)
Find the particles and count the particles that are in each cell. More specifically this function ret...
Definition: ParticleUtils.cpp:40
void synchronize() noexcept
void copyAsync(HostToDevice, InIter begin, InIter end, OutIter result) noexcept
static constexpr HostToDevice hostToDevice
void streamSynchronize() noexcept
bool notInLaunchRegion() noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void Add(T *const sum, T const value) noexcept
T ExclusiveSum(N n, T const *in, T *out, RetSum a_ret_sum=retSum)
AMREX_GPU_HOST_DEVICE constexpr AMREX_FORCE_INLINE const T & max(const T &a, const T &b) noexcept
AMREX_GPU_HOST_DEVICE constexpr AMREX_FORCE_INLINE const T & min(const T &a, const T &b) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 ubound(Array4< T > const &a) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE T bisect(T lo, T hi, F f, T tol=1e-12, int max_iter=100)
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 lbound(Array4< T > const &a) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void ignore_unused(const Ts &...)
std::enable_if_t< std::is_integral_v< T > > ParallelFor(TypeList< CTOs... >, std::array< int, sizeof...(CTOs)> const &runtime_options, T N, F &&f)
AMREX_ATTRIBUTE_FLATTEN_FOR void ParallelForRNG(T n, L const &f) noexcept
i
Definition: check_interp_points_and_weights.py:174
int dz
Definition: compute_domain.py:36
float dt
Definition: stencil.py:442
value
Definition: updateAMReX.py:141
string species1
Definition: video_yt.py:35
@ Timers
load balance according to in-code timer-based weights (i.e., with costs)
Definition: WarpXAlgorithmSelection.H:147
This is a functor for performing a "smart copy" that works in both host and device code.
Definition: SmartCopy.H:34
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE int const * dataPtr() const noexcept
MFItInfo & EnableTiling(const IntVect &ts=FabArrayBase::mfiter_tile_size) noexcept
MFItInfo & SetDynamic(bool f) noexcept
ParticleTileData< StorageParticleType, NArrayReal, NArrayInt > ParticleTileDataType
ParticleTileDataType getParticleTileData()