NeumannVectorCorrection.h

Go to the documentation of this file.

 
#ifndef included_AMP_NeumannVectorCorrection
#define included_AMP_NeumannVectorCorrection
 
#include "AMP/operators/boundary/BoundaryOperator.h"
#include "AMP/operators/boundary/libmesh/NeumannVectorCorrectionParameters.h"
#include "AMP/operators/boundary/libmesh/RobinPhysicsModel.h"
 
#include "AMP/discretization/createLibmeshElements.h"
 
// Libmesh files
#include "libmesh/libmesh_config.h"
#undef LIBMESH_ENABLE_REFERENCE_COUNTING
#include "libmesh/quadrature.h"
 
#include <string>
 
namespace AMP::Operator {
 
class NeumannVectorCorrection : public BoundaryOperator
{
public:
    explicit NeumannVectorCorrection( std::shared_ptr<const OperatorParameters> params );
 
    void setVariable( const std::shared_ptr<AMP::LinearAlgebra::Variable> &var )
    {
        d_variable = var;
    }
 
    virtual ~NeumannVectorCorrection() {}
 
    std::string type() const override { return "NeumannVectorCorrection"; }
 
    void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
    void computeRHScorrection( AMP::LinearAlgebra::Vector::shared_ptr rhsCorrection );
 
    void reset( std::shared_ptr<const OperatorParameters> params ) override;
 
    void addRHScorrection( AMP::LinearAlgebra::Vector::shared_ptr rhs ) override;
 
    auto getOperatorParameters() { return d_params; }
 
    void setVariableFlux( const AMP::LinearAlgebra::Vector::shared_ptr &flux );
 
    void setFrozenVector( AMP::LinearAlgebra::Vector::shared_ptr f );
 
    std::shared_ptr<RobinPhysicsModel> getRobinPhysicsModel() { return d_robinPhysicsModel; }
 
    const auto &getBoundaryIds() const { return d_boundaryIds; }
 
    const auto &getDofIds() const { return d_dofIds; }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getOutputVariable() const override
    {
        return d_variable;
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getInputVariable() const override
    {
        return d_variable;
    }
 
protected:
    std::shared_ptr<OperatorParameters>
    getJacobianParameters( AMP::LinearAlgebra::Vector::const_shared_ptr u ) override;
 
    Discretization::createLibmeshElements d_libmeshElements;
 
    std::vector<short int> d_boundaryIds;
 
    std::vector<std::vector<double>> d_neumannValues;
 
    std::vector<std::vector<unsigned int>> d_dofIds;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_rhsCorrectionAdd;
 
    // This must be a simple variable not a dual or multivariable
    std::shared_ptr<AMP::LinearAlgebra::Variable> d_variable;
 
    bool d_isConstantFlux;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_Frozen;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_variableFlux;
 
    std::vector<bool> d_IsCoupledBoundary;
    bool d_isFluxGaussPtVector;
 
    std::shared_ptr<const NeumannVectorCorrectionParameters> d_params;
 
    std::shared_ptr<RobinPhysicsModel> d_robinPhysicsModel;
 
    std::shared_ptr<const libMesh::FEType> d_type;
    libMeshEnums::Order d_qruleOrder;
    libMeshEnums::QuadratureType d_qruleType;
 
    AMP::Mesh::MeshElementVectorPtr d_currNodes;
 
private:
};
} // namespace AMP::Operator
 
#endif