LocalLinearLeastSquaresExtrapolator.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 <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