NativePetscMatrixData.h

Go to the documentation of this file.

#ifndef included_AMP_Petsc_MatrixData
#define included_AMP_Petsc_MatrixData
 
// AMP files
#include "AMP/matrices/data/MatrixData.h"
#include "petscmat.h"
 
namespace AMP::LinearAlgebra {
 
class NativePetscMatrixData : public MatrixData
{
public:
    NativePetscMatrixData();
 
    explicit NativePetscMatrixData( std::shared_ptr<MatrixParametersBase> params );
 
    explicit NativePetscMatrixData( Mat m, bool dele = false );
 
    virtual ~NativePetscMatrixData();
 
    static std::shared_ptr<MatrixData> duplicateMat( Mat m );
 
    void copyFromMat( Mat m );
 
    virtual std::string type() const override { return "NativePetscMatrixData"; }
 
    std::shared_ptr<MatrixData> cloneMatrixData() const override;
 
    std::shared_ptr<MatrixData> transpose() const override;
 
    void removeRange( AMP::Scalar, AMP::Scalar ) override { AMP_ERROR( "Not implemented" ); }
 
    void addValuesByGlobalID(
        size_t, size_t, const size_t *, const size_t *, const void *, const typeID & ) override;
    void setValuesByGlobalID(
        size_t, size_t, const size_t *, const size_t *, const void *, const typeID & ) override;
    void getValuesByGlobalID(
        size_t, size_t, const size_t *, const size_t *, void *, const typeID & ) const override;
    void getRowByGlobalID( size_t, std::vector<size_t> &, std::vector<double> & ) const override;
 
    size_t numberColumnIDs( size_t row ) const override;
    std::vector<size_t> getColumnIDs( size_t ) const override;
 
    void makeConsistent( AMP::LinearAlgebra::ScatterType t ) override;
 
    std::shared_ptr<Vector> createInputVector() const;
    std::shared_ptr<Vector> createOutputVector() const;
    std::shared_ptr<Discretization::DOFManager> getRightDOFManager() const override;
    std::shared_ptr<Discretization::DOFManager> getLeftDOFManager() const override;
 
    size_t numGlobalRows() const override;
    size_t numGlobalColumns() const override;
 
    Mat getMat() { return d_Mat; }
 
    void setMat( Mat mat, bool manage = true );
 
    AMP_MPI getComm() const override;
 
    typeID getCoeffType() const override
    {
        constexpr auto type = getTypeID<PetscScalar>();
        return type;
    }
 
private:
    Mat d_Mat                   = nullptr;
    bool d_MatCreatedInternally = false;
};
 
 
} // namespace AMP::LinearAlgebra
 
 
#endif