LinearOperator.h

Go to the documentation of this file.

 
#ifndef included_AMP_LinearOperator
#define included_AMP_LinearOperator
 
#include "AMP/matrices/Matrix.h"
#include "AMP/operators/Operator.h"
#include "AMP/operators/OperatorParameters.h"
#include "AMP/vectors/Vector.h"
#include <memory>
 
namespace AMP::Operator {
 
 
class LinearOperator : public Operator
{
public:
    explicit LinearOperator( std::shared_ptr<const OperatorParameters> params );
 
    virtual ~LinearOperator() {}
 
    std::string type() const override { return "LinearOperator"; }
 
    virtual void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                        AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
    virtual std::shared_ptr<AMP::LinearAlgebra::Matrix> getMatrix();
 
    virtual void setMatrix( std::shared_ptr<AMP::LinearAlgebra::Matrix> in_mat );
 
    virtual void setVariables( std::shared_ptr<AMP::LinearAlgebra::Variable> in,
                               std::shared_ptr<AMP::LinearAlgebra::Variable> out )
    {
        d_inputVariable  = in;
        d_outputVariable = out;
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getInputVariable() const override
    {
        return d_inputVariable;
    }
    std::shared_ptr<AMP::LinearAlgebra::Variable> getOutputVariable() const override
    {
        return d_outputVariable;
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Vector> createInputVector() const override;
 
    std::shared_ptr<AMP::LinearAlgebra::Vector> createOutputVector() const override;
 
protected:
    LinearOperator();
 
    // input and output variables
    std::shared_ptr<AMP::LinearAlgebra::Variable> d_inputVariable;
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> d_outputVariable;
 
    std::shared_ptr<AMP::LinearAlgebra::Matrix> d_matrix; // The matrix shared pointer
 
private:
};
} // namespace AMP::Operator
 
#endif