NativeThyraVectorData.h

Go to the documentation of this file.

#ifndef included_AMP_NativeThyraVectorData
#define included_AMP_NativeThyraVectorData
 
#include "AMP/vectors/data/GhostDataHelper.hpp"
#include "AMP/vectors/data/VectorData.h"
#include "AMP/vectors/trilinos/thyra/ThyraVector.h"
 
namespace AMP::LinearAlgebra {
 
 
class NativeThyraVectorData : public GhostDataHelper<double>
{
public:
    explicit NativeThyraVectorData( Teuchos::RCP<Thyra::VectorBase<double>> vec,
                                    size_t localsize,
                                    AMP_MPI comm );
 
    virtual ~NativeThyraVectorData();
 
    std::string VectorDataName() const override { return "NativeThyraVector"; }
    size_t numberOfDataBlocks() const override;
    size_t sizeOfDataBlock( size_t i ) const override;
 
    void addValuesByLocalID( size_t, const size_t *, const void *, const typeID & ) override;
    void setValuesByLocalID( size_t, const size_t *, const void *, const typeID & ) override;
    void getValuesByLocalID( size_t, const size_t *, void *, const typeID & ) const override;
    void putRawData( const void *, const typeID & ) override;
    void getRawData( void *, const typeID & ) const override;
    uint64_t getDataID() const override
    {
        return reinterpret_cast<uint64_t>( getRawDataBlockAsVoid( 0 ) );
    }
    void *getRawDataBlockAsVoid( size_t i ) override;
    const void *getRawDataBlockAsVoid( size_t i ) const override;
    size_t sizeofDataBlockType( size_t ) const override { return sizeof( double ); }
    typeID getType( size_t ) const override { return getTypeID<double>(); }
    void swapData( VectorData & ) override;
    std::shared_ptr<VectorData> cloneData( const std::string &name = "" ) const override;
 
    Teuchos::RCP<Thyra::VectorBase<double>> getVec() { return d_thyraVec; }
    Teuchos::RCP<const Thyra::VectorBase<double>> getVec() const { return d_thyraVec; }
 
protected:
    NativeThyraVectorData();
 
 
private:
    Teuchos::RCP<Thyra::VectorBase<double>> d_thyraVec;
 
    static Teuchos::RCP<const Thyra::VectorBase<double>>
    getThyraVec( std::shared_ptr<const VectorData> vec );
 
    static Teuchos::RCP<const Thyra::VectorBase<double>> getThyraVec( const VectorData &v );
    static Teuchos::RCP<Thyra::VectorBase<double>> getThyraVec( VectorData &v );
};
 
 
} // namespace AMP::LinearAlgebra
 
#endif