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 lev,
        amrex::Real gamma_boost,
        const amrex::Vector<amrex::Real>& time,
        amrex::BoxArray const & ba,
        amrex::DistributionMapping const & dm);

    void UpdateElementFinder (int lev, const amrex::Vector<amrex::Real>& time);
 
    /* 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;

    [[nodiscard]] LatticeElementFinderDevice
    GetFinderDeviceInstance (WarpXParIter const& a_pti, int a_offset, const amrex::Vector<amrex::Real>& dts) const;
 
    /* All of the available lattice element types */
    Drift h_drift;
    HardEdgedQuadrupole h_quad;
    HardEdgedPlasmaLens h_plasmalens;
 
};
 
#endif // WARPX_ACCELERATORLATTICE_ACCELERATORLATTICE_H_