DirichletVectorCorrection.h

Go to the documentation of this file.

#ifndef included_AMP_DirichletVectorCorrection
#define included_AMP_DirichletVectorCorrection
 
#include "AMP/operators/boundary/BoundaryOperator.h"
#include "AMP/operators/boundary/DirichletVectorCorrectionParameters.h"
 
namespace AMP::Operator {
 
class DirichletVectorCorrection : public BoundaryOperator
{
public:
    explicit DirichletVectorCorrection( std::shared_ptr<const OperatorParameters> params );
 
    virtual ~DirichletVectorCorrection() {}
 
    std::string type() const override { return "DirichletVectorCorrection"; }
 
    void setVariable( const std::shared_ptr<AMP::LinearAlgebra::Variable> &var )
    {
        d_variable = var;
    }
 
    void setDirichletValues( AMP::LinearAlgebra::Vector::shared_ptr vals )
    {
        d_dirichletValues2 = mySubsetVector( vals, d_variable );
    }
 
    void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                AMP::LinearAlgebra::Vector::shared_ptr r ) override;
 
    void applyZeroValues( AMP::LinearAlgebra::Vector::shared_ptr r );
 
    void applyNonZeroValues( AMP::LinearAlgebra::Vector::shared_ptr r );
 
    void applyResidual( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                        AMP::LinearAlgebra::Vector::shared_ptr r );
 
    void reset( std::shared_ptr<const OperatorParameters> params ) override;
 
    void setRHScorrection( AMP::LinearAlgebra::Vector::shared_ptr rhs ) override
    {
        this->applyZeroValues( rhs );
    }
 
    void modifyInitialSolutionVector( AMP::LinearAlgebra::Vector::shared_ptr sol ) override
    {
        if ( !d_setResidual ) {
            this->applyNonZeroValues( sol );
        }
    }
 
    const auto &getBoundaryIds() const { return d_boundaryIds; }
 
    const auto &getDofIds() const { return d_dofIds; }
 
protected:
    AMP::LinearAlgebra::Vector::shared_ptr
    mySubsetVector( AMP::LinearAlgebra::Vector::shared_ptr vec,
                    std::shared_ptr<AMP::LinearAlgebra::Variable> var );
 
    AMP::LinearAlgebra::Vector::const_shared_ptr
    mySubsetVector( AMP::LinearAlgebra::Vector::const_shared_ptr vec,
                    std::shared_ptr<AMP::LinearAlgebra::Variable> var );
 
    std::shared_ptr<OperatorParameters>
        getJacobianParameters( AMP::LinearAlgebra::Vector::const_shared_ptr ) override;
 
    std::vector<short int> d_boundaryIds;
 
    std::vector<std::vector<size_t>> d_dofIds;
 
    std::vector<std::vector<double>> d_dirichletValues1;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_dirichletValues2;
 
    // This must be a simple variable not a dual or multivariable
    std::shared_ptr<AMP::LinearAlgebra::Variable> d_variable;
 
    bool d_isAttachedToVolumeOperator = false;
 
    bool d_setResidual = false;
 
    bool d_skipParams = false;
 
    int d_valuesType = 0;
 
    double d_scalingFactor = 0;
 
private:
};
} // namespace AMP::Operator
 
#endif