SmartCreate.H

Go to the documentation of this file.

/* Copyright 2019-2020 Luca Fedeli, Neil Zaim
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
 
#ifndef WARPX_SMART_CREATE_H_
#define WARPX_SMART_CREATE_H_
 
#include "DefaultInitialization.H"
 
#include <AMReX.H>
#include <AMReX_AmrCore.H>
#include <AMReX_GpuContainers.H>
#include <AMReX_ParallelDescriptor.H>
#include <AMReX_Particle.H>
#include <AMReX_ParticleTile.H>

 
struct SmartCreate
{
    const InitializationPolicy* m_policy_real;
    const InitializationPolicy* m_policy_int;
    int m_weight_index = 0;
 
    template <typename PartData>
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void operator() (
        PartData& prt, const int i_prt,
        amrex::RandomEngine const& engine,
        const amrex::Real x = 0.0,
        const amrex::Real y = 0.0,
        const amrex::Real z = 0.0,
        const int cpu = 0,
        const int id = 0) const noexcept
    {
#if defined(WARPX_DIM_3D)
        prt.m_rdata[PIdx::x][i_prt] = x;
        prt.m_rdata[PIdx::y][i_prt] = y;
        prt.m_rdata[PIdx::z][i_prt] = z;
#elif defined(WARPX_DIM_XZ) || defined(WARPX_DIM_RZ)
        prt.m_rdata[PIdx::x][i_prt] = x;
        prt.m_rdata[PIdx::z][i_prt] = z;
        amrex::ignore_unused(y);
#elif defined(WARPX_DIM_1D_Z)
        prt.m_rdata[PIdx::z][i_prt] = z;
        amrex::ignore_unused(x,y);
#elif defined(WARPX_DIM_RCYLINDER) || defined(WARPX_DIM_RSPHERE)
        prt.m_rdata[PIdx::x][i_prt] = x;
        amrex::ignore_unused(y,z);
#endif
 
        prt.m_idcpu[i_prt] = amrex::SetParticleIDandCPU(id, cpu);
 
        // initialize the real components after position
        for (int j = AMREX_SPACEDIM; j < PartData::NAR; ++j) {
            prt.m_rdata[j][i_prt] = initializeRealValue(m_policy_real[j], engine);
        }
        for (int j = 0; j < prt.m_num_runtime_real; ++j) {
            prt.m_runtime_rdata[j][i_prt] = initializeRealValue(m_policy_real[j+PartData::NAR], engine);
        }
 
        // initialize the int components
        for (int j = 0; j < PartData::NAI; ++j) {
            prt.m_idata[j][i_prt] = initializeIntValue(m_policy_int[j]);
        }
        for (int j = 0; j < prt.m_num_runtime_int; ++j) {
            prt.m_runtime_idata[j][i_prt] = initializeIntValue(m_policy_int[j+PartData::NAI]);
        }
    }
};

class SmartCreateFactory
{
    PolicyVec m_policy_real;
    PolicyVec m_policy_int;
    bool m_defined{false};
 
public:
    template <class PartTileData>
    SmartCreateFactory (const PartTileData& part) noexcept:
        m_policy_real{getPolicies(part.GetRealSoANames())},
        m_policy_int{getPolicies(part.GetIntSoANames())},
        m_defined{true}
    {}
 
    [[nodiscard]] SmartCreate getSmartCreate () const noexcept
    {
        AMREX_ASSERT(m_defined);
        return SmartCreate{m_policy_real.dataPtr(),
                         m_policy_int.dataPtr()};
    }
 
    [[nodiscard]] bool isDefined () const noexcept { return m_defined; }
};
 
#endif //WARPX_SMART_CREATE_H_