GetRowHelper.h

Go to the documentation of this file.

#ifndef included_AMP_Matrix_GetRowHelper
#define included_AMP_Matrix_GetRowHelper
 
#include "AMP/discretization/DOF_Manager.h"
 
#include <array>
#include <memory>
#include <vector>
 
 
namespace AMP::LinearAlgebra {
 
 
class GetRowHelper final
{
public:
    GetRowHelper( std::shared_ptr<const AMP::Discretization::DOFManager> leftDOF,
                  std::shared_ptr<const AMP::Discretization::DOFManager> rightDOF );
 
    ~GetRowHelper();
 
    // Copy/assignment operators
    GetRowHelper()                       = default;
    GetRowHelper( GetRowHelper && )      = default;
    GetRowHelper( const GetRowHelper & ) = delete;
    GetRowHelper &operator=( GetRowHelper && ) = default;
    GetRowHelper &operator=( const GetRowHelper & ) = delete;
 
    void deallocate();
 
    std::array<size_t, 2> NNZ( size_t row ) const;
 
    template<class INT>
    void NNZ( size_t row, INT &N_local, INT &N_remote ) const;
 
    template<class INT>
    void getRow( size_t row, INT *local, INT *remote ) const;
 
    const size_t *getLocals() const { return d_local; }
 
    const size_t *getRemotes() const { return d_remote; }
 
private: // Private routines
    std::array<size_t *, 2> getRow2( size_t row ) const;
    void reserve( size_t N );
 
 
private: // Member data
    bool d_hasFields             = true;
    std::array<size_t, 2> *d_NNZ = nullptr;
    size_t *d_local              = nullptr;
    size_t *d_remote             = nullptr;
    size_t *d_localOffset        = nullptr;
    size_t *d_remoteOffset       = nullptr;
    size_t d_size[2]             = { 0, 0 };
    size_t d_capacity[2]         = { 0, 0 };
    std::shared_ptr<const AMP::Discretization::DOFManager> d_leftDOF;
    std::shared_ptr<const AMP::Discretization::DOFManager> d_rightDOF;
};
 
 
} // namespace AMP::LinearAlgebra
 
#include "AMP/matrices/GetRowHelper.hpp"
 
#endif