CollectAndInterpolateNodalDof.h

Go to the documentation of this file.

 
#pragma once
 
#include "MathLib/LinAlg/GlobalMatrixVectorTypes.h"
#include "ProcessLib/ProcessVariable.h"
#include <Eigen/Core>
 
namespace MeshLib
{
class Element;
}  // namespace MeshLib
 
namespace ProcessLib::BoundaryConditionAndSourceTerm::Python
{
Eigen::MatrixXd collectDofsToMatrix(
    MeshLib::Element const& element,
    std::size_t const mesh_id,
    NumLib::LocalToGlobalIndexMap const& dof_table,
    GlobalVector const& x);
 
Eigen::VectorXd collectDofsToMatrixOnBaseNodesSingleComponent(
    MeshLib::Element const& element, std::size_t const mesh_id,
    NumLib::LocalToGlobalIndexMap const& dof_table, GlobalVector const& x,
    int const variable, int const component);
 
template <typename NsAndWeight>
void interpolate(
    Eigen::MatrixXd const& primary_variables_mat,
    std::vector<std::reference_wrapper<ProcessVariable>> const& pv_refs,
    NsAndWeight const& ns_and_weight,
    Eigen::Ref<Eigen::VectorXd>
        interpolated_primary_variables)
{
    Eigen::Index component_flattened = 0;
 
    // We assume that all_process_variables_for_this_process have the same
    // order as the d.o.f. table. Therefore we can iterate over
    // all_process_variables_for_this_process.
    for (auto pv_ref : pv_refs)
    {
        auto const& pv = pv_ref.get();
        auto const num_comp = pv.getNumberOfGlobalComponents();
        auto const shp_fct_order = pv.getShapeFunctionOrder();
        auto const N = ns_and_weight.N(shp_fct_order);
 
        for (auto comp = decltype(num_comp){0}; comp < num_comp; ++comp)
        {
            // This computation assumes that there are no "holes" in the
            // primary_variables_mat. I.e., all nodal d.o.f. for a certain
            // (var, comp) must be stored contiguously in the respective column
            // of primary_variables_mat.
            interpolated_primary_variables[component_flattened] =
                N *
                primary_variables_mat.col(component_flattened).head(N.size());
            component_flattened++;
        }
    }
}
}  // namespace ProcessLib::BoundaryConditionAndSourceTerm::Python