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