GetVelocity.H

Go to the documentation of this file.

/* Copyright 2021 Hannah Klion
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
 
#ifndef WARPX_GET_VELOCITY_H_
#define WARPX_GET_VELOCITY_H_
 
#include "VelocityProperties.H"

struct GetVelocity
{
    /* Type of velocity initialization */
    VelocityInitType m_type;
 
    /* Velocity direction */
    int m_dir; 
    int m_sign_dir; 

    amrex::Real m_velocity{0};
    amrex::ParserExecutor<3> m_velocity_parser;

    GetVelocity (VelocityProperties const& vel) noexcept;

    AMREX_GPU_HOST_DEVICE
    amrex::Real operator() (amrex::Real const x, amrex::Real const y, amrex::Real const z) const noexcept
    {
        switch (m_type)
        {
            case (VelConstantValue):
            {
                return m_sign_dir * m_velocity;
            }
            case (VelParserFunction):
            {
                return m_sign_dir * m_velocity_parser(x,y,z);
            }
            default:
            {
                amrex::Abort("Get initial velocity: unknown type");
                return 0.0;
            }
        }
    }

    [[nodiscard]]
    AMREX_GPU_HOST_DEVICE
    int direction () const noexcept
    {
        return m_dir;
    }
};
#endif //WARPX_GET_VELOCITY_H_