AcceleratorLattice.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_ACCELERATORLATTICE_H_
#define WARPX_ACCELERATORLATTICE_ACCELERATORLATTICE_H_
 
#include "LatticeElementFinder.H"
#include "LatticeElements/Drift.H"
#include "LatticeElements/HardEdgedQuadrupole.H"
#include "LatticeElements/HardEdgedPlasmaLens.H"
 
#include <memory>
#include <string>
 
/* \brief AcceleratorLattice contains the lattice elements
 */
class AcceleratorLattice
{
public:
 
    AcceleratorLattice ();
 
    void
    ReadLattice (std::string const & root_name, amrex::ParticleReal & z_location);
 
    /* Refinement level at which this accelerator lattice is defined */
    int m_level;
 
    /* Flags where any lattice elements have been specified by the user */
    bool m_lattice_defined = false;
 
    void InitElementFinder (int const lev, amrex::BoxArray const & ba, amrex::DistributionMapping const & dm);
 
    void UpdateElementFinder (int const lev);
 
    /* The lattice element finder handles the lookup that finds the elements at the particle locations.
     * It should follow the same grid layout as the main grids.
     */
    std::unique_ptr<amrex::LayoutData<LatticeElementFinder>> m_element_finder;
 
    LatticeElementFinderDevice GetFinderDeviceInstance (WarpXParIter const& a_pti, int const a_offset) const;
 
    /* All of the available lattice element types */
    Drift h_drift;
    HardEdgedQuadrupole h_quad;
    HardEdgedPlasmaLens h_plasmalens;
 
};
 
#endif // WARPX_ACCELERATORLATTICE_ACCELERATORLATTICE_H_