HeatTransportBHE/LocalAssemblers/LocalDataInitializer.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#pragma once
 
#include <functional>
#include <memory>
#include <type_traits>
#include <typeindex>
#include <typeinfo>
#include <unordered_map>
 
#include "MeshLib/Elements/Elements.h"
#include "NumLib/DOF/LocalToGlobalIndexMap.h"
#include "NumLib/Fem/Integration/GaussLegendreIntegrationPolicy.h"
#include "NumLib/Fem/Integration/IntegrationMethodRegistry.h"
#include "ProcessLib/HeatTransportBHE/BHE/BHETypes.h"
#include "ProcessLib/Utils/EnabledElements.h"
 
namespace ProcessLib
{
namespace HeatTransportBHE
{
template <typename LocalAssemblerInterface,
          template <typename /* shp fct */>
          class LocalAssemblerDataSoil,
          template <typename /* shp fct */, typename /* bhe type */>
          class LocalAssemblerDataBHE,
          typename... ConstructorArgs>
class LocalDataInitializer final
{
    template <unsigned DIM>
    struct IsNDElement
    {
        template <typename ElementTraits>
        constexpr bool operator()(ElementTraits*) const
        {
            return ElementTraits::Element::dimension == DIM;
        }
    };
 
public:
    using LADataIntfPtr = std::unique_ptr<LocalAssemblerInterface>;
 
    explicit LocalDataInitializer(
        NumLib::LocalToGlobalIndexMap const& dof_table,
        NumLib::IntegrationOrder const integration_order)
        : _dof_table(dof_table)
    {
        // 3D soil elements
        using Enabled3DElementTraits =
            decltype(BaseLib::TMP::filter<EnabledElementTraitsLagrange>(
                std::declval<IsNDElement<3>>()));
 
        BaseLib::TMP::foreach<Enabled3DElementTraits>(
            [this, integration_order]<typename ET>(ET*)
            {
                using MeshElement = typename ET::Element;
                using ShapeFunction = typename ET::ShapeFunction;
 
                _builder[std::type_index(typeid(MeshElement))] =
                    makeLocalAssemblerBuilder<ShapeFunction>(integration_order);
            });
 
        // 1D BHE elements
        using Enabled1DElementTraits =
            decltype(BaseLib::TMP::filter<EnabledElementTraitsLagrange>(
                std::declval<IsNDElement<1>>()));
 
        BaseLib::TMP::foreach<Enabled1DElementTraits>(
            [this, integration_order]<typename ET>(ET*)
            {
                using MeshElement = typename ET::Element;
                using ShapeFunction = typename ET::ShapeFunction;
 
                _builder[std::type_index(typeid(MeshElement))] =
                    makeLocalAssemblerBuilderBHE<ShapeFunction>(
                        integration_order);
            });
    }

    LADataIntfPtr operator()(
        std::size_t const /*id*/,
        MeshLib::Element const& mesh_item,
        std::unordered_map<std::size_t, BHE::BHETypes*> const&
            element_to_bhe_map,
        ConstructorArgs&&... args) const
    {
        auto const type_idx = std::type_index(typeid(mesh_item));
        auto const it = _builder.find(type_idx);
 
        if (it == _builder.end())
        {
            OGS_FATAL(
                "You are trying to build a local assembler for an unknown mesh "
                "element type ({:s})."
                " Maybe you have disabled this mesh element type in your build "
                "configuration, or a mesh element order does not match shape "
                "function order given in the project file.",
                type_idx.name());
        }
 
        return it->second(mesh_item,
                          element_to_bhe_map,
                          std::forward<ConstructorArgs>(args)...);
    }
 
private:
    using LADataBuilder = std::function<LADataIntfPtr(
        MeshLib::Element const& e,
        std::unordered_map<std::size_t, BHE::BHETypes*> const&
            element_to_bhe_map,
        ConstructorArgs&&...)>;
 
    template <typename ShapeFunction>
    static LADataBuilder makeLocalAssemblerBuilder(
        NumLib::IntegrationOrder const integration_order)
    {
        return [integration_order](
                   MeshLib::Element const& e,
                   std::unordered_map<std::size_t, BHE::BHETypes*> const&
                   /* unused */,
                   ConstructorArgs&&... args) -> LADataIntfPtr
        {
            auto const& integration_method = NumLib::IntegrationMethodRegistry::
                template getIntegrationMethod<
                    typename ShapeFunction::MeshElement>(integration_order);
 
            if (e.getDimension() == 3)  // soil elements
            {
                return LADataIntfPtr{new LocalAssemblerDataSoil<ShapeFunction>{
                    e, integration_method,
                    std::forward<ConstructorArgs>(args)...}};
            }
 
            return nullptr;
        };
    }
 
    template <typename ShapeFunction>
    static LADataBuilder makeLocalAssemblerBuilderBHE(
        NumLib::IntegrationOrder const integration_order)
    {
        return [integration_order](
                   MeshLib::Element const& e,
                   std::unordered_map<std::size_t, BHE::BHETypes*> const&
                       element_to_bhe_map,
                   ConstructorArgs&&... args) -> LADataIntfPtr
        {
            auto const& integration_method = NumLib::IntegrationMethodRegistry::
                template getIntegrationMethod<
                    typename ShapeFunction::MeshElement>(integration_order);
 
            auto& bhe = *element_to_bhe_map.at(e.getID());
 
            if (std::holds_alternative<BHE::BHE_1U>(bhe))
            {
                return LADataIntfPtr{
                    new LocalAssemblerDataBHE<ShapeFunction, BHE::BHE_1U>{
                        e, integration_method, std::get<BHE::BHE_1U>(bhe),
                        std::forward<ConstructorArgs>(args)...}};
            }
 
            if (std::holds_alternative<BHE::BHE_CXA>(bhe))
            {
                return LADataIntfPtr{
                    new LocalAssemblerDataBHE<ShapeFunction, BHE::BHE_CXA>{
                        e, integration_method, std::get<BHE::BHE_CXA>(bhe),
                        std::forward<ConstructorArgs>(args)...}};
            }
 
            if (std::holds_alternative<BHE::BHE_CXC>(bhe))
            {
                return LADataIntfPtr{
                    new LocalAssemblerDataBHE<ShapeFunction, BHE::BHE_CXC>{
                        e, integration_method, std::get<BHE::BHE_CXC>(bhe),
                        std::forward<ConstructorArgs>(args)...}};
            }
 
            if (std::holds_alternative<BHE::BHE_2U>(bhe))
            {
                return LADataIntfPtr{
                    new LocalAssemblerDataBHE<ShapeFunction, BHE::BHE_2U>{
                        e, integration_method, std::get<BHE::BHE_2U>(bhe),
                        std::forward<ConstructorArgs>(args)...}};
            }
 
            if (std::holds_alternative<BHE::BHE_1P>(bhe))
            {
                return LADataIntfPtr{
                    new LocalAssemblerDataBHE<ShapeFunction, BHE::BHE_1P>{
                        e, integration_method, std::get<BHE::BHE_1P>(bhe),
                        std::forward<ConstructorArgs>(args)...}};
            }
            OGS_FATAL(
                "Trying to create local assembler for an unknown BHE type.");
        };
    }

    std::unordered_map<std::type_index, LADataBuilder> _builder;
 
    NumLib::LocalToGlobalIndexMap const& _dof_table;
};
}  // namespace HeatTransportBHE
}  // namespace ProcessLib