ubodt.hpp

 
#ifndef FMM_UBODT_H_
#define FMM_UBODT_H_
 
#include "network/type.hpp"
#include "mm/transition_graph.hpp"
#include "util/debug.hpp"
 
namespace FMM {
namespace MM {
 
struct Record {
  NETWORK::NodeIndex source; 
  NETWORK::NodeIndex target; 
  NETWORK::NodeIndex first_n; 
  NETWORK::NodeIndex prev_n; 
  NETWORK::EdgeIndex next_e; 
  double cost; 
  Record *next; 
};
 
class UBODT {
 public:
  UBODT(const UBODT &) = delete;
  UBODT &operator=(const UBODT &) = delete;
  UBODT(int buckets_arg, int multiplier_arg);
  ~UBODT();
  Record *look_up(NETWORK::NodeIndex source, NETWORK::NodeIndex target) const;
 
  std::vector<NETWORK::EdgeIndex> look_sp_path(NETWORK::NodeIndex source,
      NETWORK::NodeIndex target) const;
 
  C_Path construct_complete_path(const TGOpath &path,
                                 const std::vector<NETWORK::Edge> &edges,
                                 std::vector<int> *indices) const;
  double get_delta() const;
  unsigned int cal_bucket_index(NETWORK::NodeIndex source,
      NETWORK::NodeIndex target) const;
 
  void insert(Record *r);
 
  static std::shared_ptr<UBODT> read_ubodt_file(const std::string &filename,
                                                int multiplier = 50000);
  static std::shared_ptr<UBODT> read_ubodt_csv(const std::string &filename,
                                               int multiplier = 50000);
 
  static std::shared_ptr<UBODT> read_ubodt_binary(const std::string &filename,
                                                  int multiplier = 50000);
  static long estimate_ubodt_rows(const std::string &filename);
  static int find_prime_number(double value);
  constexpr static double LOAD_FACTOR = 2.0; 
  static const int BUFFER_LINE = 1024; 
 private:
  const long long multiplier;   // multiplier to get a unique ID
  const int buckets;   // number of buckets
  double delta = 0.0;
  Record **hashtable;
};
}
}
 
#endif //FMM_SRC_FMM_FFMM_UBODT_H_