/
dump_contexted_events_to_json.py
85 lines (70 loc) · 2.67 KB
/
dump_contexted_events_to_json.py
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import os
import sys
import ujson as json
from events import detect_events_given_path
from event_context import extract_event_context
from interactions import InteractionsUtil as IU
from meta_graph import convert_to_original_graph
from util import json_dump, load_json_by_line
from viz_util import add_subgraph_specific_attributes_to_graph,\
to_d3_graph
from experiment_util import get_output_path
def run_with_context(interactions_path,
candidate_tree_path,
dirname=None,
to_original_graph=False,
undirected=False):
if not os.path.exists(dirname):
os.makedirs(dirname)
try:
interactions = json.load(open(interactions_path))
except ValueError as e:
print(e)
interactions = load_json_by_line(interactions_path)
interactions = IU.clean_interactions(interactions,
undirected=undirected)
output_path = get_output_path(candidate_tree_path, dirname)
K = 5
events = detect_events_given_path(candidate_tree_path, K)
contexted_events = []
for e in events:
context_dag = extract_event_context(
interactions, e,
undirected=undirected
)
if to_original_graph:
context_dag = convert_to_original_graph(context_dag)
e = convert_to_original_graph(e)
contexted_events.append(
add_subgraph_specific_attributes_to_graph(
context_dag, [(e, {'event': True})])
)
d3_events = [to_d3_graph(ce)
for ce in contexted_events]
print('writing to {}'.format(output_path))
json_dump(d3_events, output_path)
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser('Dump contexted events to json')
parser.add_argument('--interactions_path',
required=True
)
parser.add_argument('--candidate_tree_path',
required=True
)
parser.add_argument('--dirname')
parser.add_argument('--to_original_graph',
action='store_true',
default=False)
parser.add_argument('--undirected',
action='store_true',
default=False)
args = parser.parse_args()
if args.to_original_graph and args.undirected:
print('ERROR: to_original_graph not allowed for undirected')
sys.exit(-1)
run_with_context(args.interactions_path,
args.candidate_tree_path,
args.dirname,
args.to_original_graph,
undirected=args.undirected)