FieldProbeParticleContainer.H

Go to the documentation of this file.

/* Copyright 2021 Elisa Rheaume, Axel Huebl
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
 
#ifndef WARPX_FieldProbeParticleContainer_H_
#define WARPX_FieldProbeParticleContainer_H_
 
#include <AMReX_ParIter.H>
#include <AMReX_Particles.H>
 
#include <AMReX_BaseFwd.H>
#include <AMReX_AmrCoreFwd.H>
#include <AMReX_Vector.H>

struct FieldProbePIdx
{
    enum
    {
#if !defined (WARPX_DIM_1D_Z)
        x,
#endif
#if defined (WARPX_DIM_3D)
        y,
#endif
#if !defined(WARPX_DIM_RCYLINDER) || !defined(WARPX_DIM_RSPHERE)
        z,
#endif
        Ex, Ey, Ez,
        Bx, By, Bz,
        S, 
#if defined(WARPX_DIM_RZ) || defined(WARPX_DIM_RCYLINDER) || defined(WARPX_DIM_RSPHERE)
        theta,      
 
#endif
#if defined(WARPX_DIM_RSPHERE)
        phi,      
#endif
        nattribs
    };
};

class FieldProbeParticleContainer
    : public amrex::ParticleContainerPureSoA<FieldProbePIdx::nattribs, 0>
{
public:
    static constexpr int NStructReal = 0;
    static constexpr int NStructInt = 0;
    static constexpr int NReal = FieldProbePIdx::nattribs;
    static constexpr int NInt = 0;
 
    FieldProbeParticleContainer (amrex::AmrCore* amr_core);
    ~FieldProbeParticleContainer() override = default;
 
    FieldProbeParticleContainer ( FieldProbeParticleContainer const &)             = delete;
    FieldProbeParticleContainer& operator= ( FieldProbeParticleContainer const & ) = delete;
    FieldProbeParticleContainer ( FieldProbeParticleContainer&& )                  = default;
    FieldProbeParticleContainer& operator= ( FieldProbeParticleContainer&& )       = default;

    using iterator = amrex::ParIterSoA<FieldProbePIdx::nattribs, 0>;
    using const_iterator = amrex::ParConstIterSoA<FieldProbePIdx::nattribs, 0>;

    void AddNParticles (int lev, amrex::Vector<amrex::ParticleReal> const & x, amrex::Vector<amrex::ParticleReal> const & y, amrex::Vector<amrex::ParticleReal> const & z);
};
 
#endif // WARPX_FieldProbeParticleContainer_H_