LatticeElementBase.H

Go to the documentation of this file.

/* Copyright 2022 David Grote
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_ACCELERATORLATTICE_LATTICEELEMENTS_LATTICEELEMENTBASE_H_
#define WARPX_ACCELERATORLATTICE_LATTICEELEMENTS_LATTICEELEMENTBASE_H_
 
#include <AMReX_REAL.H>
#include <AMReX_ParmParse.H>
#include <AMReX_GpuContainers.H>
 
#include <string>
#include <vector>
 
struct LatticeElementBase
{
 
    LatticeElementBase (std::string const& element_name);
 
    ~LatticeElementBase () = default;
 
    void
    AddElementBase(amrex::ParmParse & pp_element, amrex::ParticleReal & z_location);
 
    void
    WriteToDeviceBase ();
 
    std::string m_element_name;
 
    int nelements = 0;
 
    // The host level copy of the data.
    std::vector<amrex::ParticleReal> h_zs;
    std::vector<amrex::ParticleReal> h_ze;
 
    // The device level copy of the data.
    amrex::Gpu::DeviceVector<amrex::ParticleReal> d_zs;
    amrex::Gpu::DeviceVector<amrex::ParticleReal> d_ze;
 
};
 
#endif // WARPX_ACCELERATORLATTICE_LATTICEELEMENTS_LATTICEELEMENTBASE_H_