def GetConnectionData(inputFile):
d = EPANetSimulation(inputFile)
ret, nnodes = d.ENgetcount(d.EN_NODECOUNT)
ret, llinks = d.ENgetcount(d.EN_LINKCOUNT)
Conn = []
NoConn = []
h_degree = 0
if (len(Conn) == 0):
for n in range(0, nnodes + 1):
c = []
Conn.append(c)
# generate mapping used later for resilience calculation
for i in range(0, (llinks + 1)):
if (len(Conn) == (nnodes + 1)):
nodes = d.ENgetlinknodes(i + 1)
if (nodes[0] == 0):
Conn[nodes[1]].append(i + 1)
Conn[nodes[2]].append(i + 1)
h_degree = max(len(Conn[nodes[0]]), len(Conn[nodes[1]]),
h_degree)
for idx in range(0, nnodes + 1):
NoConn.append(len(Conn[idx]))
while (len(Conn[idx]) < h_degree):
Conn[idx].append(0)
del Conn[0]
del NoConn[0]
return Conn, NoConn
