ISRBendElement.H

Go to the documentation of this file.

/* Copyright 2022-2023 The Regents of the University of California, through Lawrence
 *           Berkeley National Laboratory (subject to receipt of any required
 *           approvals from the U.S. Dept. of Energy). All rights reserved.
 *
 * This file is part of ImpactX.
 *
 * Authors: Alex Bojanich, Chad Mitchell, Axel Huebl
 * License: BSD-3-Clause-LBNL
 */
#ifndef IMPACTX_PARTICLES_WAKEFIELDS_ISRBENDELEMENT_H
#define IMPACTX_PARTICLES_WAKEFIELDS_ISRBENDELEMENT_H
 
#include "elements/ExactSbend.H"
#include "elements/Sbend.H"
#include "elements/CFbend.H"
#include "particles/ReferenceParticle.H"
 
#include <AMReX_REAL.H>
 
#include <cmath>
#include <tuple>
#include <variant>
 
 
namespace impactx::particles::wakefields
{
    template <typename VariantType>
    std::tuple<bool, amrex::Real>
    ISRBendElement (
        VariantType const& element_variant,
        RefPart const & refpart
    )
    {
        amrex::Real R = 0;
        bool element_has_isr = false;
 
        if (std::holds_alternative<elements::Sbend>(element_variant))
        {
            auto& element = std::get<elements::Sbend>(element_variant);
            R = std::abs(element.m_rc);
            element_has_isr = true;
        }
        else if (std::holds_alternative<elements::CFbend>(element_variant))
        {
            auto& element = std::get<elements::CFbend>(element_variant);
            R = std::abs(element.m_rc);
            element_has_isr = true;
        }
        else if (std::holds_alternative<elements::ExactSbend>(element_variant))
        {
            auto& element = std::get<elements::ExactSbend>(element_variant);
            R = std::abs(element.rc(refpart));
            element_has_isr = true;
        }
 
        return {element_has_isr, R};
    }
 
} // namespace impactx::particles::wakefields
 
#endif // IMPACTX_PARTICLES_WAKEFIELDS_ISRBENDELEMENT_H