LaserProfiles.H

Go to the documentation of this file.

/* Copyright 2019-2020 Luca Fedeli, Maxence Thevenet
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_LaserProfiles_H_
#define WARPX_LaserProfiles_H_
 
#include <AMReX_Gpu.H>
#include <AMReX_ParmParse.H>
#include <AMReX_Parser.H>
#include <AMReX_REAL.H>
#include <AMReX_Vector.H>
 
#include <functional>
#include <limits>
#include <map>
#include <memory>
#include <string>
#include <utility>
 
 
namespace WarpXLaserProfiles {
 
struct CommonLaserParameters
{
    amrex::Real wavelength; 
    amrex::Real e_max;  
    amrex::Vector<amrex::Real> p_X;// ! Polarization
    amrex::Vector<amrex::Real> nvec; 
};
 
 
class ILaserProfile
{
public:
    virtual void
    init (
        const amrex::ParmParse& ppl,
        CommonLaserParameters params) = 0;
 
    virtual void
    update (
        amrex::Real t) = 0;
 
    virtual void
    fill_amplitude (
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const = 0;
 
    virtual ~ILaserProfile(){}
};
 
class GaussianLaserProfile : public ILaserProfile
{
 
public:
    void
    init (
        const amrex::ParmParse& ppl,
        CommonLaserParameters params) override final;
 
    //No update needed
    void
    update (amrex::Real /*t */) override final {}
 
    void
    fill_amplitude (
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const override final;
 
private:
    struct {
        amrex::Real waist          = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real duration       = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real t_peak         = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real focal_distance = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real zeta           = 0;
        amrex::Real beta           = 0;
        amrex::Real phi2           = 0;
        amrex::Real phi0           = 0;
 
        amrex::Vector<amrex::Real> stc_direction; 
        amrex::Real theta_stc; 
    } m_params;
 
    CommonLaserParameters m_common_params;
};
 
class HarrisLaserProfile : public ILaserProfile
{
 
public:
    void
    init (
        const amrex::ParmParse& ppl,
        CommonLaserParameters params) override final;
 
    //No update needed
    void
    update (amrex::Real /*t */) override final {}
 
    void
    fill_amplitude (
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const override final;
 
private:
    struct {
        amrex::Real waist          = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real duration       = std::numeric_limits<amrex::Real>::quiet_NaN();
        amrex::Real focal_distance = std::numeric_limits<amrex::Real>::quiet_NaN();
    } m_params;
 
    CommonLaserParameters m_common_params;
};
 
class FieldFunctionLaserProfile : public ILaserProfile
{
 
public:
    void
    init (
        const amrex::ParmParse& ppl,
        CommonLaserParameters params) override final;
 
    //No update needed
    void
    update (amrex::Real /*t */) override final {}
 
    void
    fill_amplitude (
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const override final;
 
private:
    struct{
        std::string field_function;
    } m_params;
 
    amrex::Parser m_parser;
};
 
class FromTXYEFileLaserProfile : public ILaserProfile
{
 
public:
    void
    init (
        const amrex::ParmParse& ppl,
        CommonLaserParameters params) override final;
 
    void
    update (amrex::Real t) override final;
 
    void
    fill_amplitude (
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const override final;
 
    void internal_fill_amplitude_uniform(
        const int idx_t_left,
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const;
 
    void internal_fill_amplitude_nonuniform(
        const int idx_t_left,
        const int np,
        amrex::Real const * AMREX_RESTRICT const Xp,
        amrex::Real const * AMREX_RESTRICT const Yp,
        amrex::Real t,
        amrex::Real * AMREX_RESTRICT const amplitude) const;
 
private:
    void parse_txye_file(std::string txye_file_name);
 
    std::pair<int,int> find_left_right_time_indices(amrex::Real t) const;
 
    void read_data_t_chuck(int t_begin, int t_end);
 
    struct{
        std::string txye_file_name;
        bool is_grid_uniform = false;
        int nt, nx, ny;
        amrex::Vector<amrex::Real> t_coords;
        amrex::Vector<amrex::Real> h_x_coords;
        amrex::Gpu::DeviceVector<amrex::Real> d_x_coords;
        amrex::Vector<amrex::Real> h_y_coords;
        amrex::Gpu::DeviceVector<amrex::Real> d_y_coords;
        int time_chunk_size;
        int first_time_index;
        int last_time_index;
        amrex::Gpu::DeviceVector<amrex::Real> E_data;
        amrex::Real t_delay = amrex::Real(0.0);
 
    } m_params;
 
    CommonLaserParameters m_common_params;
};
 
const
std::map<
std::string,
std::function<std::unique_ptr<ILaserProfile>()>
>
laser_profiles_dictionary =
{
    {"gaussian",
        [] () {return std::make_unique<GaussianLaserProfile>();} },
    {"harris",
        [] () {return std::make_unique<HarrisLaserProfile>();} },
    {"parse_field_function",
        [] () {return std::make_unique<FieldFunctionLaserProfile>();} },
    {"from_txye_file",
        [] () {return std::make_unique<FromTXYEFileLaserProfile>();} }
};
 
} //WarpXLaserProfiles
 
#endif //WARPX_LaserProfiles_H_