-
Notifications
You must be signed in to change notification settings - Fork 0
/
build.h
67 lines (62 loc) · 2.01 KB
/
build.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
//Contributors: Sibo Wang, Renchi Yang
#ifndef BUILD_H
#define BUILD_H
#include "algo.h"
#include "graph.h"
#include "heap.h"
#include "config.h"
#include <boost/serialization/serialization.hpp>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/archive/binary_iarchive.hpp>
#include <boost/serialization/vector.hpp>
#include <boost/serialization/map.hpp>
#include <boost/serialization/utility.hpp>
inline string get_exact_topk_ppr_file(){
if(!boost::algorithm::ends_with(config.exact_pprs_folder, FILESEP))
config.exact_pprs_folder += FILESEP;
return config.exact_pprs_folder+config.graph_alias+".topk.pprs";
}
inline string get_self_ppr_file(){
if(!boost::algorithm::ends_with(config.exact_pprs_folder, FILESEP))
config.exact_pprs_folder += FILESEP;
return config.exact_pprs_folder+config.graph_alias+".self.pprs";
}
inline void save_exact_topk_ppr(){
string filename = get_exact_topk_ppr_file();
std::ofstream ofs(filename);
boost::archive::text_oarchive oa(ofs);
oa << exact_topk_pprs;
}
inline void save_self_ppr(const map<int ,double> &ppr_self){
string filename = get_self_ppr_file();
std::ofstream ofs(filename);
boost::archive::text_oarchive oa(ofs);
oa << ppr_self;
}
inline void load_exact_topk_ppr(){
string filename = get_exact_topk_ppr_file();
if(!exists_test(filename)){
INFO("No exact topk ppr file", filename);
return;
}
std::ifstream ifs(filename);
boost::archive::text_iarchive ia(ifs);
ia >> exact_topk_pprs;
INFO(exact_topk_pprs.size());
}
inline map<int ,double> load_exact_self_ppr(){
map<int ,double> ppr_self;
string filename = get_self_ppr_file();
if(!exists_test(filename)){
INFO("No exact self ppr file", filename);
return ppr_self;
}
std::ifstream ifs(filename);
boost::archive::text_iarchive ia(ifs);
ia >> ppr_self;
INFO(ppr_self.size());
return ppr_self;
}
#endif