ComputeFluxUtilities.hpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//
 
#pragma once

 
#include "shambackends/sycl.hpp"
#include "shammath/riemann.hpp"
#include "shammath/riemann_dust.hpp"
#include "shammodels/ramses/Solver.hpp"
 
namespace shammodels::basegodunov::modules {
 
    using RiemannSolverMode     = shammodels::basegodunov::RiemannSolverMode;
    using DustRiemannSolverMode = shammodels::basegodunov::DustRiemannSolverMode;
    using Direction             = shammodels::basegodunov::modules::Direction;

    template<class Tvec, Direction dir>
    inline constexpr Tvec dir_normal() {
        if constexpr (dir == Direction::xp) {
            return Tvec{1, 0, 0};
        } else if constexpr (dir == Direction::xm) {
            return Tvec{-1, 0, 0};
        } else if constexpr (dir == Direction::yp) {
            return Tvec{0, 1, 0};
        } else if constexpr (dir == Direction::ym) {
            return Tvec{0, -1, 0};
        } else if constexpr (dir == Direction::zp) {
            return Tvec{0, 0, 1};
        } else if constexpr (dir == Direction::zm) {
            return Tvec{0, 0, -1};
        } else {
            static_assert(shambase::always_false_v<decltype(dir)>, "non-exhaustive visitor!");
        }
        return Tvec{};
    }

    template<shammath::FluidStateAdiabaticSpec FSpec, RiemannSolverMode mode, Direction dir>
    inline constexpr typename FSpec::Tcons riemann_flux(
        const FSpec &fspec,
        const typename FSpec::Tprim &prim_l,
        const typename FSpec::Tprim &prim_r) {
        const typename FSpec::Tvec n = dir_normal<typename FSpec::Tvec, dir>();
 
        if constexpr (mode == RiemannSolverMode::Rusanov) {
            return shammath::rusanov_flux(fspec, prim_l, prim_r, n);
        }
        if constexpr (mode == RiemannSolverMode::HLL) {
            return shammath::hll_flux(fspec, prim_l, prim_r, n);
        }
        if constexpr (mode == RiemannSolverMode::HLLC) {
            return shammath::hllc_adiab_toro_flux(fspec, prim_l, prim_r, n);
        }
    }

    template<shammath::DustFluidStateSpec FSpec, DustRiemannSolverMode mode, Direction dir>
    inline constexpr typename FSpec::Tcons riemann_dust_flux(
        const FSpec &fspec,
        const typename FSpec::Tprim &prim_l,
        const typename FSpec::Tprim &prim_r) {
        const typename FSpec::Tvec n = dir_normal<typename FSpec::Tvec, dir>();
 
        if constexpr (mode == DustRiemannSolverMode::HB) {
            return shammath::huang_bai_flux(fspec, prim_l, prim_r, n);
        }
        if constexpr (mode == DustRiemannSolverMode::DHLL) {
            return shammath::d_hll_flux(fspec, prim_l, prim_r, n);
        }
    }
 
} // namespace shammodels::basegodunov::modules