ABLFieldInit.H

Go to the documentation of this file.

#ifndef ABLFIELDINIT_H
#define ABLFIELDINIT_H
 
#include "amr-wind/core/Field.H"
 
#include "AMReX_Array.H"
#include "AMReX_Array4.H"
#include "AMReX_Box.H"
#include "AMReX_Geometry.H"
#include "AMReX_REAL.H"
#include "AMReX_Vector.H"
#include "AMReX_Gpu.H"
 
using namespace amrex::literals;
 
namespace amr_wind {

class ABLFieldInit
{
    static_assert(AMREX_SPACEDIM == 3, "ABL requires 3 dimensional mesh");
 
public:
    ABLFieldInit();
 
    void initialize_from_inputfile(); // default
    void initialize_from_netcdf();    // general wind/temperature profiles
 
    void operator()(
        const amrex::Box& vbx,
        const amrex::Geometry& geom,
        const amrex::Array4<amrex::Real>& velocity,
        const amrex::Array4<amrex::Real>& density,
        const amrex::Array4<amrex::Real>& temperature) const;

    void perturb_temperature(
        int lev, const amrex::Geometry& geom, Field& temperature) const;

    [[nodiscard]] bool add_temperature_perturbations() const
    {
        return m_perturb_theta;
    }

    void init_tke(const amrex::Geometry& geom, amrex::MultiFab& tke) const;
 
private:
    amrex::Vector<amrex::Real> m_vel;

    std::string m_vel_timetable;

    amrex::Real m_vel_speed;

    amrex::Real m_vel_dir;

    amrex::Vector<amrex::Real> m_theta_heights;
    amrex::Vector<amrex::Real> m_theta_values;

    bool m_initial_wind_profile{false};
    bool m_terrain_aligned_profile{false};
    std::string m_terrain_file{"terrain.amrwind"};
    amrex::Gpu::DeviceVector<amrex::Real> m_xterrain;
    amrex::Gpu::DeviceVector<amrex::Real> m_yterrain;
    amrex::Gpu::DeviceVector<amrex::Real> m_zterrain;
    std::string m_1d_rans;
    amrex::Vector<amrex::Real> m_wind_heights;
    amrex::Vector<amrex::Real> m_u_values;
    amrex::Vector<amrex::Real> m_v_values;
    amrex::Vector<amrex::Real> m_tke_values;
 
    // Device copies of the above arrays
    amrex::Gpu::DeviceVector<amrex::Real> m_thht_d;
    amrex::Gpu::DeviceVector<amrex::Real> m_thvv_d;
 
    amrex::Gpu::DeviceVector<amrex::Real> m_windht_d;
    amrex::Gpu::DeviceVector<amrex::Real> m_prof_u_d;
    amrex::Gpu::DeviceVector<amrex::Real> m_prof_v_d;
    amrex::Gpu::DeviceVector<amrex::Real> m_prof_tke_d;
    amrex::Real m_rho;

    amrex::Real m_deltaU{1.0_rt};

    amrex::Real m_deltaV{1.0_rt};

    amrex::Real m_Uperiods{4.0_rt};

    amrex::Real m_Vperiods{4.0_rt};

    amrex::Real m_ref_height{50.0_rt};

    amrex::Real m_delta_t{0.8_rt};

    amrex::Real m_theta_gauss_mean{0.0_rt};

    amrex::Real m_theta_gauss_var{1.0_rt};

    amrex::Real m_theta_cutoff_height{1.0e16_rt};

    amrex::Real m_tke_init{0.1_rt};

    amrex::Real m_tke_init_factor{0.4_rt};

    amrex::Real m_tke_cutoff_height{250.0_rt};

    amrex::RealArray m_top_vel{20.0_rt};

    amrex::RealArray m_bottom_vel{4.0_rt};

    bool m_linear_profile{false};

    bool m_perturb_vel{true};

    bool m_perturb_theta{false};

    bool m_tke_init_profile{false};

    bool m_init_uvtheta_profile{false};
};
 
} // namespace amr_wind
 
#endif /* ABLFIELDINIT_H */