RadiationDiffusionFDBDFWrappers.h

Go to the documentation of this file.

#ifndef RAD_DIF_FD_BDF_WRAPPERS
#define RAD_DIF_FD_BDF_WRAPPERS
 
#include "AMP/IO/AsciiWriter.h"
#include "AMP/IO/PIO.h"
#include "AMP/discretization/MultiDOF_Manager.h"
#include "AMP/discretization/boxMeshDOFManager.h"
#include "AMP/geometry/shapes/Box.h"
#include "AMP/matrices/MatrixBuilder.h"
#include "AMP/mesh/structured/BoxMesh.h"
#include "AMP/mesh/structured/structuredMeshElement.h"
#include "AMP/operators/LinearOperator.h"
#include "AMP/operators/Operator.h"
#include "AMP/operators/OperatorParameters.h"
#include "AMP/utils/AMPManager.h"
#include "AMP/vectors/MultiVector.h"
#include "AMP/vectors/Vector.h"
#include "AMP/vectors/VectorBuilder.h"
 
#include "AMP/operators/radiationDiffusionFD/RadiationDiffusionFDDiscretization.h"
 
namespace AMP::Operator {
 
// Classes declared here
class BDFRadDifOpPJac;
class BDFRadDifOp;
struct BDFRadDifOpPJacData;
 
 
class BDFRadDifOp : public AMP::Operator::Operator
{
 
public:
    double d_gamma = -1.0;
    std::shared_ptr<RadDifOp> d_RadDifOp = nullptr;
 
    BDFRadDifOp( std::shared_ptr<const AMP::Operator::OperatorParameters> params_ );
 
    virtual ~BDFRadDifOp(){};
 
    void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u_in,
                AMP::LinearAlgebra::Vector::shared_ptr r ) override;
 
    std::string type() const override { return "BDFRadDifOp"; }
 
    bool isValidVector( std::shared_ptr<const AMP::LinearAlgebra::Vector> ET ) override
    {
        return d_RadDifOp->isValidVector( ET );
    }
 
    void setGamma( AMP::Scalar gamma_ );
 
 
    void setComponentScalings( std::shared_ptr<AMP::LinearAlgebra::Vector> s,
                               std::shared_ptr<AMP::LinearAlgebra::Vector> f );
 
 
protected:
    std::shared_ptr<AMP::LinearAlgebra::Vector> d_pSolutionScaling;
    std::shared_ptr<AMP::LinearAlgebra::Vector> d_pFunctionScaling;
    std::shared_ptr<AMP::LinearAlgebra::Vector> d_pScratchSolVector;
 
    std::shared_ptr<AMP::Operator::OperatorParameters>
    getJacobianParameters( AMP::LinearAlgebra::Vector::const_shared_ptr u_in ) override;
};
 
 
class BDFRadDifOpPJac : public AMP::Operator::LinearOperator
{
 
public:
    double d_gamma = -1.0;
    std::shared_ptr<RadDifOpPJac> d_RadDifOpPJac;
    std::shared_ptr<BDFRadDifOpPJacData> d_data = nullptr;
 
 
    BDFRadDifOpPJac( std::shared_ptr<AMP::Operator::OperatorParameters> params );
 
    virtual ~BDFRadDifOpPJac(){};
 
    static std::unique_ptr<AMP::Operator::Operator>
    create( std::shared_ptr<AMP::Operator::OperatorParameters> params )
    {
        return std::make_unique<BDFRadDifOpPJac>( params );
    };
 
    void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u_in,
                AMP::LinearAlgebra::Vector::shared_ptr r ) override;
 
    void reset( std::shared_ptr<const AMP::Operator::OperatorParameters> params ) override;
 
    std::shared_ptr<AMP::LinearAlgebra::Vector> createInputVector() const override;
 
 
private:
    void setData();
};
 
 
struct BDFRadDifOpPJacData {
    std::shared_ptr<AMP::LinearAlgebra::Matrix> d_E_BDF  = nullptr;
    std::shared_ptr<AMP::LinearAlgebra::Matrix> d_T_BDF  = nullptr;
    std::shared_ptr<AMP::LinearAlgebra::Vector> r_EE_BDF = nullptr;
    std::shared_ptr<AMP::LinearAlgebra::Vector> r_ET_BDF = nullptr;
    std::shared_ptr<AMP::LinearAlgebra::Vector> r_TE_BDF = nullptr;
    std::shared_ptr<AMP::LinearAlgebra::Vector> r_TT_BDF = nullptr;
 
    BDFRadDifOpPJacData(){};
    BDFRadDifOpPJacData( std::shared_ptr<RadDifOpPJacData> data, double gamma );
};
 
} // namespace AMP::Operator
 
 
#endif