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 SMART_CREATE_H_
#define SMART_CREATE_H_
 
#include "DefaultInitialization.H"
 
#include <AMReX.H>
#include <AMReX_AmrCore.H>
#include <AMReX_GpuContainers.H>
#include <AMReX_ParallelDescriptor.H>
 
struct SmartCreate
{
    const InitializationPolicy* m_policy_real;
    const InitializationPolicy* m_policy_int;
    const 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_aos[i_prt].pos(0) = x;
        prt.m_aos[i_prt].pos(1) = y;
        prt.m_aos[i_prt].pos(2) = z;
#elif defined(WARPX_DIM_XZ) || defined(WARPX_DIM_RZ)
        prt.m_aos[i_prt].pos(0) = x;
        prt.m_aos[i_prt].pos(1) = z;
        amrex::ignore_unused(y);
#else
        prt.m_aos[i_prt].pos(0) = z;
        amrex::ignore_unused(x,y);
#endif
 
        prt.m_aos[i_prt].cpu() = cpu;
        prt.m_aos[i_prt].id() = id;
 
        // initialize the real components
        for (int j = 0; 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.getParticleComps())},
        m_policy_int{getPolicies(part.getParticleiComps())},
        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 //SMART_CREATE_H_