coo.hpp

Go to the documentation of this file.

#pragma once

#include "pangolin/types.hpp"
#include "pangolin/edge_list.hpp"
#include "pangolin/dense/vector.hu"

#ifdef __CUDACC__
#define HOST   __host__
#define DEVICE __device__
#else
#define HOST
#define DEVICE
#endif 

PANGOLIN_BEGIN_NAMESPACE()

template<typename Index>
class COO;

template<typename Index>
class COOView {
    friend class COO<Index>;
private:
  uint64_t nnz_; 
  uint64_t num_rows_; 

public:
 const Index *rowPtr_; 
 const Index *rowInd_; 
 const Index *colInd_; 

 HOST DEVICE uint64_t nnz() const {return nnz_; }
 HOST DEVICE uint64_t num_rows() const {return num_rows_; }

 const Index *row_ptr() const { return rowPtr_; } 
 const Index *col_ind() const { return colInd_; } 
 const Index *row_ind() const { return rowInd_; } //<! row index array

};

template<typename Index>
class COO {
private:
  Index maxCol_;
 public:
  COO(); 
  Vector<Index> rowPtr_; 
  Vector<Index> colInd_; 
  Vector<Index> rowInd_; 
  HOST DEVICE uint64_t nnz() const {
    assert(colInd_.size() == rowInd_.size());
    return colInd_.size(); 
  } 
  uint64_t num_nodes() const; 
  HOST DEVICE uint64_t num_rows() const; 

  static COO<Index> from_edgelist(const EdgeList &es, bool (*edgeFilter)(const Edge &) = nullptr);
  COOView<Index> view() const; 

 const Index *row_ptr() { return rowPtr_.data(); } 
 const Index *col_ind() { return colInd_.data(); } 
 const Index *row_ind() { return rowInd_.data(); } //<! row index array
};



PANGOLIN_END_NAMESPACE()
    

#undef HOST 
#undef DEVICE


#include "coo-impl.hpp"