DirichletMatrixCorrection.h

Go to the documentation of this file.

 
#ifndef included_AMP_DirichletMatrixCorrection
#define included_AMP_DirichletMatrixCorrection
 
#include "AMP/operators/boundary/BoundaryOperator.h"
#include "AMP/operators/boundary/DirichletMatrixCorrectionParameters.h"
#include "AMP/operators/boundary/DirichletVectorCorrection.h"
#include "AMP/vectors/Variable.h"
 
namespace AMP::Operator {
 
class DirichletMatrixCorrection : public BoundaryOperator
{
public:
    explicit DirichletMatrixCorrection( std::shared_ptr<const OperatorParameters> params );
 
    virtual ~DirichletMatrixCorrection() {}
 
    std::string type() const override { return "DirichletMatrixCorrection"; }
 
    void setVariable( const std::shared_ptr<AMP::LinearAlgebra::Variable> &var )
    {
        d_variable = var;
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getOutputVariable() const override
    {
        return d_variable;
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getInputVariable() const override
    {
        return d_variable;
    }
 
    virtual void apply( AMP::LinearAlgebra::Vector::const_shared_ptr,
                        AMP::LinearAlgebra::Vector::shared_ptr ) override
    {
        // Do Nothing
    }
 
    void reset( std::shared_ptr<const OperatorParameters> ) override;
 
    void addRHScorrection( AMP::LinearAlgebra::Vector::shared_ptr rhs ) override;
 
    void setRHScorrection( AMP::LinearAlgebra::Vector::shared_ptr rhs ) override;
 
    const auto &getBoundaryIds() { return d_boundaryIds; }
 
    const auto &getDofIds() { return d_dofIds; }
 
 
protected:
    void parseParams( std::shared_ptr<const DirichletMatrixCorrectionParameters> );
 
 
protected:
    // This must be a simple variable not a dual or multivariable
    std::shared_ptr<AMP::LinearAlgebra::Variable> d_variable;
 
    std::vector<short int> d_boundaryIds;
 
    std::vector<std::vector<double>> d_dirichletValues;
 
    std::vector<std::vector<unsigned int>> d_dofIds;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_rhsCorrectionAdd;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_dispVals;
 
    std::shared_ptr<DirichletVectorCorrection> d_rhsCorrectionSet;
 
    bool d_symmetricCorrection = false;
 
    bool d_zeroDirichletBlock = false;
 
    bool d_skipRHSaddCorrection = false;
 
    bool d_skipRHSsetCorrection = false;
 
    bool d_computedAddRHScorrection = false;
 
    bool d_initialized = false;
 
    void initRhsCorrectionSet();
 
    void initRhsCorrectionAdd( AMP::LinearAlgebra::Vector::shared_ptr rhs );
 
    void applyMatrixCorrection();
 
    bool d_applyMatrixCorrectionWasCalled;
 
    std::shared_ptr<AMP::LinearAlgebra::Matrix> d_inputMatrix;
 
private:
};
} // namespace AMP::Operator
 
#endif