csr_kb.h

Go to the documentation of this file.

#ifndef _CSR_KB_H
#define _CSR_KB_H

#include <map>
#include <list>
#include <vector>
#include <string>

namespace freeling {
  

  class csr_kb {

  public:  
    csr_kb(const std::wstring &, int, double, double);      
    size_t get_vertex(const std::wstring &) const;

    void pagerank(std::vector<double> &) const;

    size_t size() const;

    static const size_t VERTEX_NOT_FOUND;
      
  private:
    int MaxIterations;
    double Threshold;
    double Damping;

    std::map<std::wstring,size_t> vertex_index;
    std::vector<double> out_coef;

    std::vector<size_t> first_edge;
    std::vector<size_t> num_edges;
    std::vector<size_t> edges;
    size_t num_vertices;

    size_t add_vertex(const std::wstring &);
    void fill_CSR_tables(size_t, std::list<std::pair<size_t,size_t> > &);
  };
  
  
} // namespace freeling

#endif