LocalLinearLeastSquaresExtrapolator.h

Go to the documentation of this file.

 
#pragma once
 
#include <map>
 
#include "Extrapolator.h"
#include "NumLib/DOF/GlobalMatrixProviders.h"
#include "NumLib/DOF/LocalToGlobalIndexMap.h"
 
namespace NumLib
{
class LocalLinearLeastSquaresExtrapolator : public Extrapolator
{
public:
    explicit LocalLinearLeastSquaresExtrapolator(
        NumLib::LocalToGlobalIndexMap const& dof_table);
 
    void extrapolate(const unsigned num_components,
                     ExtrapolatableElementCollection const& extrapolatables,
                     const double t,
                     std::vector<GlobalVector*> const& x,
                     std::vector<NumLib::LocalToGlobalIndexMap const*> const&
                         dof_table) override;
 
    void calculateResiduals(
        const unsigned num_components,
        ExtrapolatableElementCollection const& extrapolatables,
        const double t,
        std::vector<GlobalVector*> const& x,
        std::vector<NumLib::LocalToGlobalIndexMap const*> const& dof_table)
        override;
 
    GlobalVector const& getNodalValues() const override
    {
        return *_nodal_values;
    }
 
    GlobalVector const& getElementResiduals() const override
    {
        return *_residuals;
    }
 
private:
    void extrapolateElement(
        std::size_t const element_index, const unsigned num_components,
        ExtrapolatableElementCollection const& extrapolatables, const double t,
        std::vector<GlobalVector*> const& x,
        std::vector<NumLib::LocalToGlobalIndexMap const*> const& dof_table,
        GlobalVector& counts);
 
    void calculateResidualElement(
        std::size_t const element_index,
        const unsigned num_components,
        ExtrapolatableElementCollection const& extrapolatables,
        const double t,
        std::vector<GlobalVector*> const& x,
        std::vector<NumLib::LocalToGlobalIndexMap const*> const& dof_table);
 
    std::unique_ptr<GlobalVector> _nodal_values;  
    std::unique_ptr<GlobalVector> _residuals;     
 
    NumLib::LocalToGlobalIndexMap const& _dof_table_single_component;
 
    std::vector<double> _integration_point_values_cache;
 
    struct CachedData
    {
        Eigen::MatrixXd A;
 
        Eigen::MatrixXd A_pinv;
    };
 
    std::map<std::pair<unsigned, unsigned>, CachedData> _qr_decomposition_cache;
};
 
}  // namespace NumLib