LoadVector.h

Go to the documentation of this file.

 
#ifndef PBAT_FEM_LOAD_VECTOR_H
#define PBAT_FEM_LOAD_VECTOR_H
 
#include "Concepts.h"
#include "ShapeFunctions.h"
#include "pbat/profiling/Profiling.h"
 
#include <Eigen/Core>
 
namespace pbat::fem {

template <
    CElement TElement,
    class TDerivedE,
    class TDerivedeg,
    class TDerivedwg,
    class TDerivedNeg,
    class TDerivedFeg>
auto LoadVectors(
    Eigen::DenseBase<TDerivedE> const& E,
    Eigen::Index nNodes,
    Eigen::DenseBase<TDerivedeg> const& eg,
    Eigen::DenseBase<TDerivedwg> const& wg,
    Eigen::MatrixBase<TDerivedNeg> const& Neg,
    Eigen::MatrixBase<TDerivedFeg> const& Feg)
    -> Eigen::Matrix<typename TDerivedFeg::Scalar, TDerivedFeg::RowsAtCompileTime, Eigen::Dynamic>
{
    PBAT_PROFILE_NAMED_SCOPE("pbat.fem.LoadVector");
    using MatrixType =
        Eigen::Matrix<typename TDerivedFeg::Scalar, TDerivedFeg::RowsAtCompileTime, Eigen::Dynamic>;
    MatrixType F = MatrixType::Zero(Feg.rows(), nNodes);
    for (auto g = 0; g < wg.size(); ++g)
    {
        auto const nodes = E.col(eg(g));
        F(Eigen::placeholders::all, nodes) += wg(g) * Feg.col(g) * Neg.col(g).transpose();
    }
    return F;
}
 
} // namespace pbat::fem
 
#endif // PBAT_FEM_LOAD_VECTOR_H