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_