forked from respec/HSPsquared
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgraph.py
137 lines (104 loc) · 3.33 KB
/
graph.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
''' Copyright (c) 2020 by RESPEC, INC.
Author: Robert Heaphy, Ph.D.
License: LGPL2
'''
from pandas import DataFrame, read_hdf
#from networkx import DiGraph, Graph, topological_sort, connected_components
def graph_from_HDF5(hdfname):
'''
Builds a Directed Acyclic Graph (DAG) from the links data in the HDF5 file.
Parameters
----------
hdfname : string
Name of the HSP2 HDF5 file.
Returns
-------
dg : Networkx DiGraph
DAG for HDF5 file.
'''
links = read_hdf(hdfname, 'CONTROL/LINKS')
links['source'] = links.apply(lambda row: f'{row.SVOL}_{row.SVOLNO}', axis='columns')
links['destination'] = links.apply(lambda row: f'{row.TVOL}_{row.TVOLNO}', axis='columns')
links = links[['source','destination']].copy()
dg = DiGraph()
for i, (source, destination) in links.iterrows():
dg.add_edge(source, destination)
return dg
def make_opsequence(hdfname, delt=60):
'''
Puts topologically valid /CONTROL/OP_SEQUENCE table into the HSP2 HDF file.
Parameters
----------
hdfname : string
Name of HSP2 HDF file.
delt : TYPE, integer
Value for OP_SEQUENCE delt value.
The default is 60.
Returns
-------
None.
'''
data = []
for o in topological_sort(graph_from_HDF5(hdfname)):
op, seg = o.split('_')
data.append((op, seg, delt))
df = DataFrame(data, columns=['OPERATION', 'SEGMENT', 'INDELT_minutes'])
df.to_hdf(hdfname, 'CONTROL/OP_SEQUENCE', )
return
def HDF5_isconnected(hdfname):
'''
Boolean value if the DAG for the HDF5 file has only one connected subset
Parameters
----------
hdfname : string
Name of HSP2 HDF5 file.
Returns
-------
Boolean
DAG has only one connected subset.
'''
dg = graph_from_HDF5(hdfname)
a = list(connected_components(Graph(dg)))
return True if len(a) == 1 else False
def component_list(hdfname):
'''
List of connected components in HDF5 file DAG
Parameters
----------
hdfname : string
Name of HSP2 HDF5 file.
Returns
-------
List of string lists
Each element of list is a list of connected subset nodes.
'''
dg = graph_from_HDF5(hdfname)
a = connected_components(Graph(dg))
return list(a)
def color_graph(hdfname):
'''
Demonstrates how DAG can be colored to provide additional info to plot.
Parameters
----------
hdfname : string
Name of HSP2 HDF file.
Returns
-------
Tuple(Networkx DiGraph, List of string color names)
The Networkx draw_network(DAG, colorlist) displays Graph in color.
'''
dg = graph_from_HDF5(hdfname)
cm = []
for x in dg.nodes:
npred = len(list(dg.predecessors(x)))
nsucc = len(list(dg.successors(x)))
if x.startswith('RCHRES'):
if npred == 0 and nsucc == 0: cm.append('red')
elif npred == 0: cm.append('orange')
elif nsucc == 0: cm.append('yellow')
else: cm.append('blue')
elif x.startswith('PERLND'): cm.append('green')
elif x.startswith('IMPLND'): cm.append('gray')
else: cm.append('black')
return dg, cm
# smart_opseq(master, slave) will return soom