LinearTimeOperator.h

Go to the documentation of this file.

#ifndef included_AMP_LinearTimeOperator
#define included_AMP_LinearTimeOperator
 
#include "AMP/operators/LinearOperator.h"
#include "AMP/operators/OperatorParameters.h"
#include "AMP/utils/Utilities.h"
#include "AMP/vectors/Vector.h"
#include <memory>
 
 
namespace AMP::TimeIntegrator {
 
class LinearTimeOperator : public AMP::Operator::LinearOperator
{
public:
    explicit LinearTimeOperator( std::shared_ptr<const AMP::Operator::OperatorParameters> params );
    virtual ~LinearTimeOperator();
 
    void reset( std::shared_ptr<const AMP::Operator::OperatorParameters> params ) override;
 
    void registerRhsOperator( std::shared_ptr<AMP::Operator::LinearOperator> op )
    {
        d_pRhsOperator = op;
    }
    void registerMassOperator( std::shared_ptr<AMP::Operator::LinearOperator> op )
    {
        d_pMassOperator = op;
    }
 
    std::shared_ptr<Operator> getRhsOperator( void ) { return d_pRhsOperator; }
    std::shared_ptr<Operator> getMassOperator( void ) { return d_pMassOperator; }
 
    void setDt( double dt ) { d_dCurrentDt = dt; }
 
    // added by JL
    void setScalingFactor( double scalingFactor ) { d_dScalingFactor = scalingFactor; }
 
    // added by JL //correction by RS
    std::shared_ptr<AMP::LinearAlgebra::Variable> getInputVariable() const override
    {
        return d_pRhsOperator->getInputVariable();
    }
 
    std::shared_ptr<AMP::LinearAlgebra::Variable> getOutputVariable() const override
    {
        return d_pRhsOperator->getOutputVariable();
    }
    // JL
    void registerCurrentTime( double currentTime ) { d_current_time = currentTime; }
 
    std::shared_ptr<AMP::Operator::OperatorParameters>
    getParameters( const std::string &type,
                   AMP::LinearAlgebra::Vector::const_shared_ptr u,
                   std::shared_ptr<AMP::Operator::OperatorParameters> params = nullptr ) override;
 
protected:
    LinearTimeOperator();
 
    void getFromInput( std::shared_ptr<AMP::Database> db );
 
    bool d_bModifyRhsOperatorMatrix;
 
    bool d_bAlgebraicComponent;
 
    double d_dScalingFactor;
    double d_dCurrentDt;
 
    std::shared_ptr<AMP::Operator::LinearOperator> d_pRhsOperator;
 
    std::shared_ptr<AMP::Operator::LinearOperator> d_pMassOperator;
 
    std::shared_ptr<AMP::LinearAlgebra::Vector> d_pScratchVector;
 
    double d_current_time;
    double d_beta;
 
private:
};
} // namespace AMP::TimeIntegrator
 
#endif