SubsetVectorData.h

Go to the documentation of this file.

#ifndef included_AMP_SubsetVectorData
#define included_AMP_SubsetVectorData
 
#include "AMP/discretization/DOF_Manager.h"
#include "AMP/vectors/Vector.h"
#include "AMP/vectors/data/GhostDataHelper.hpp"
#include "AMP/vectors/data/VectorData.h"
 
#include <vector>
 
namespace AMP::LinearAlgebra {
 
 
class SubsetVectorParameters
{
public:
    std::shared_ptr<CommunicationList> d_CommList = nullptr;
 
    std::shared_ptr<AMP::Discretization::DOFManager> d_DOFManager = nullptr;
 
    Vector::shared_ptr d_ViewVector; // Vector we subsetted for the view
};
 
class SubsetVectorData : public GhostDataHelper<double>
{
 
public:
    std::string VectorDataName() const override
    {
        return "SubsetVectorData of " + d_ViewVector->type();
    }
    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 *in, const typeID &id ) override;
    void getRawData( void *out, const typeID &id ) const override;
    typeID getType( size_t ) const override { return d_typeID; }
    uint64_t getDataID() const override { return d_ViewVector->getDataID(); }
    size_t sizeofDataBlockType( size_t ) const override { return sizeof( double ); }
    void swapData( VectorData & ) override;
    std::shared_ptr<VectorData> cloneData( const std::string &name = "" ) const override;
    bool hasContiguousData() const override { return numberOfDataBlocks() > 1 ? false : true; }
    SubsetVectorData() {}
    const AMP_MPI &getComm() const override;
    explicit SubsetVectorData( std::shared_ptr<SubsetVectorParameters> params );
 
private:
    void *getRawDataBlockAsVoid( size_t i ) override;
    const void *getRawDataBlockAsVoid( size_t i ) const override;
 
    // Internal data
    Vector::shared_ptr d_ViewVector;                   // Vector we subsetted for the view
    size_t d_parentLocalStartID;                       // Offset for the parent
    std::vector<size_t> d_SubsetLocalIDToViewGlobalID; // The list of global ID in the parent vector
    AMP::typeID d_typeID;
    std::vector<size_t> d_dataBlockSize; // The size of the data blocks
    std::vector<void *> d_dataBlockPtr;  // The pointers to the data blocks
    std::shared_ptr<AMP::Discretization::DOFManager>
        d_DOFManager; // this will contain the subset DOF
};
 
 
} // namespace AMP::LinearAlgebra
 
 
#endif