LIE/HydroMechanics/LocalAssembler/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/FiniteElement/LowerDimShapeTable.h"
#include "NumLib/Fem/Integration/GaussLegendreIntegrationPolicy.h"
#include "NumLib/Fem/Integration/IntegrationMethodRegistry.h"
#include "ProcessLib/Utils/EnabledElements.h"
 
namespace ProcessLib
{
namespace LIE
{
namespace HydroMechanics
{
template <
    typename LocalAssemblerInterface,
    template <typename /* shp u */, typename /* shp p */, int /* global dim */>
    class LocalAssemblerDataMatrix,
    template <typename /* shp u */, typename /* shp p */, int /* global dim */>
    class LocalAssemblerDataMatrixNearFracture,
    template <typename /* shp u */, typename /* shp p */, int /* global dim */>
    class LocalAssemblerDataFracture,
    int DisplacementDim, typename... ConstructorArgs>
class LocalDataInitializer final
{
    struct IsElementEnabled
    {
        template <typename ElementTraits>
        constexpr bool operator()(ElementTraits*) const
        {
            if constexpr (DisplacementDim < ElementTraits::ShapeFunction::DIM)
            {
                return false;
            }
 
            // exclude 0D elements
            return ElementTraits::Element::dimension >= 1 &&
                   ElementTraits::ShapeFunction::ORDER >= 1;
        }
    };
 
public:
    using LADataIntfPtr = std::unique_ptr<LocalAssemblerInterface>;
 
    explicit LocalDataInitializer(
        NumLib::LocalToGlobalIndexMap const& dof_table,
        NumLib::IntegrationOrder const integration_order)
        : _dof_table(dof_table)
    {
        using EnabledElementTraits =
            decltype(BaseLib::TMP::filter<EnabledElementTraitsLagrange>(
                std::declval<IsElementEnabled>()));
 
        BaseLib::TMP::foreach<EnabledElementTraits>(
            [this, integration_order]<typename ET>(ET*)
            {
                using MeshElement = typename ET::Element;
                using ShapeFunction = typename ET::ShapeFunction;
                using LowerOrderShapeFunction =
                    typename ET::LowerOrderShapeFunction;
 
                _builder[std::type_index(typeid(MeshElement))] =
                    makeLocalAssemblerBuilder<ShapeFunction,
                                              LowerOrderShapeFunction>(
                        integration_order);
            });
    }

    LADataIntfPtr operator()(std::size_t const id,
                             MeshLib::Element const& mesh_item,
                             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());
        }
 
        auto const n_local_dof = _dof_table.getNumberOfElementDOF(id);
        auto const varIDs = _dof_table.getElementVariableIDs(id);
        bool const isPressureDeactivated = (varIDs.front() != 0);
        std::vector<int> involved_varIDs;  // including deactivated elements
        involved_varIDs.reserve(varIDs.size() + 1);
        if (isPressureDeactivated)
        {
            involved_varIDs.push_back(0);  // always pressure come in
        }
        involved_varIDs.insert(involved_varIDs.end(), varIDs.begin(),
                               varIDs.end());
 
        std::vector<unsigned> dofIndex_to_localIndex;
 
        // matrix and fracture assemblers with enrichments
        dofIndex_to_localIndex.resize(n_local_dof);
        std::vector<unsigned> vec_n_element_nodes;
        // TODO how to get the shape function order for each variable?
        vec_n_element_nodes.push_back(
            mesh_item.getNumberOfBaseNodes());  // pressure
        auto const max_varID = *std::max_element(varIDs.begin(), varIDs.end());
        for (int i = 1; i < max_varID + 1; i++)
        {
            vec_n_element_nodes.push_back(
                mesh_item.getNumberOfNodes());  // displacements
        }
 
        unsigned local_id = 0;
        unsigned dof_id = 0;
        for (unsigned i = 0; i < involved_varIDs.size(); i++)
        {
            auto const var_id = involved_varIDs[i];
            auto const n_var_comp =
                _dof_table.getNumberOfVariableComponents(var_id);
            auto const n_var_element_nodes = vec_n_element_nodes[i];
            for (int var_comp_id = 0; var_comp_id < n_var_comp; var_comp_id++)
            {
                auto const& ms = _dof_table.getMeshSubset(var_id, var_comp_id);
                auto const mesh_id = ms.getMeshID();
                for (unsigned k = 0; k < n_var_element_nodes; k++)
                {
                    MeshLib::Location l(mesh_id,
                                        MeshLib::MeshItemType::Node,
                                        getNodeIndex(mesh_item, k));
                    auto global_index =
                        _dof_table.getGlobalIndex(l, var_id, var_comp_id);
                    if (global_index != NumLib::MeshComponentMap::nop &&
                        dof_id < n_local_dof)
                    {
                        dofIndex_to_localIndex[dof_id++] = local_id;
                    }
                    local_id++;
                }
            }
        }
 
        return it->second(mesh_item, involved_varIDs.size(), n_local_dof,
                          dofIndex_to_localIndex,
                          std::forward<ConstructorArgs>(args)...);
    }
 
private:
    using LADataBuilder = std::function<LADataIntfPtr(
        MeshLib::Element const& e,
        std::size_t const n_variables,
        std::size_t const local_matrix_size,
        std::vector<unsigned> const& dofIndex_to_localIndex,
        ConstructorArgs&&...)>;

    template <typename ShapeFunctionDisplacement,
              typename ShapeFunctionPressure>
    static LADataBuilder makeLocalAssemblerBuilder(
        NumLib::IntegrationOrder const integration_order)
    {
        return [integration_order](
                   MeshLib::Element const& e,
                   std::size_t const n_variables,
                   std::size_t const local_matrix_size,
                   std::vector<unsigned> const& dofIndex_to_localIndex,
                   ConstructorArgs&&... args)
        {
            auto const& integration_method = NumLib::IntegrationMethodRegistry::
                template getIntegrationMethod<
                    typename ShapeFunctionDisplacement::MeshElement>(
                    integration_order);
 
            if (e.getDimension() == DisplacementDim)
            {
                if (n_variables == 2)
                {
                    return LADataIntfPtr{
                        new LocalAssemblerDataMatrix<ShapeFunctionDisplacement,
                                                     ShapeFunctionPressure,
                                                     DisplacementDim>{
                            e, n_variables, local_matrix_size,
                            dofIndex_to_localIndex, integration_method,
                            std::forward<ConstructorArgs>(args)...}};
                }
                return LADataIntfPtr{new LocalAssemblerDataMatrixNearFracture<
                    ShapeFunctionDisplacement, ShapeFunctionPressure,
                    DisplacementDim>{e, n_variables, local_matrix_size,
                                     dofIndex_to_localIndex, integration_method,
                                     std::forward<ConstructorArgs>(args)...}};
            }
            return LADataIntfPtr{new LocalAssemblerDataFracture<
                ShapeFunctionDisplacement, ShapeFunctionPressure,
                DisplacementDim>{e, local_matrix_size, dofIndex_to_localIndex,
                                 integration_method,
                                 std::forward<ConstructorArgs>(args)...}};
        };
    }

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