FieldVariant.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 "shambase/exception.hpp"
#include "shambackends/typeAliasVec.hpp"
#include <variant>
 
namespace shamrock::patch {
 
    template<template<class> class Container>
    class FieldVariant {
 
        public:
        using var_t_template = std::variant<
            Container<f32>,
            Container<f32_2>,
            Container<f32_3>,
            Container<f32_4>,
            Container<f32_8>,
            Container<f32_16>,
            Container<f64>,
            Container<f64_2>,
            Container<f64_3>,
            Container<f64_4>,
            Container<f64_8>,
            Container<f64_16>,
            Container<u32>,
            Container<u64>,
            Container<u32_3>,
            Container<u64_3>,
            Container<i64_3>,
            Container<i64>>;
 
        var_t_template value; 
        template<class T>
        explicit FieldVariant(Container<T> &&val) : value(std::move(val)) {}
 
        template<class T>
        Container<T> &get_if_ref_throw() {
            if (Container<T> *pval = std::get_if<Container<T>>(&value)) {
                return *pval;
            }
            throw shambase::make_except_with_loc<std::invalid_argument>(
                "the type asked is not correct");
        }
 
        template<class Func>
        void visit(Func &&f) {
            std::visit(
                [&](auto &arg) {
                    f(arg);
                },
                value);
        }
 
        template<class Func>
        auto visit_return(Func &&f) {
            return std::visit(
                [&](auto &arg) {
                    return f(arg);
                },
                value);
        }
 
        template<class Func>
        void visit(Func &&f) const {
            std::visit(
                [&](auto &arg) {
                    f(arg);
                },
                value);
        }
 
        template<class Func>
        auto visit_return(Func &&f) const {
            return std::visit(
                [&](auto &arg) {
                    return f(arg);
                },
                value);
        }
 
        template<template<class> class Container2, class Func>
        FieldVariant<Container2> convert(Func &&f) {
            return std::visit(
                [&](auto &arg) {
                    return f(arg);
                },
                value);
        }
    };
 
    template<template<class> class Container>
    using var_t_template = typename FieldVariant<Container>::var_t_template;
 
} // namespace shamrock::patch