def _read_stn(self, file):
network = STN()
state = ""
for line in file:
if line.startswith('#'):
if "points" in line.lower() and "num" in line.lower():
state = "NO_POINTS"
elif "edges" in line.lower() and "num" in line.lower():
state = "NO_EDGES"
elif "links" in line.lower():
state = "NO_LINKS"
elif "names" in line.lower():
state = "NAMES"
elif "edges" in line.lower():
state = "EDGES"
edge_counter = 0
elif "links" in line.lower():
state = "LINKS"
else:
pass
else:
if state == 'NO_POINTS':
num_points = int(line)
network.length = num_points
network.successor_edges = [{} for i in range(num_points)]
network.names_list = ["0" for i in range(num_points)]
elif state == 'NO_EDGES':
num_edges = int(line)
elif state == 'NO_LINKS':
# for testing, throw an error
raise Exception(
"Simple Temporal Networks do not have contingent links."
)
elif state == 'NAMES':
list_of_nodes = line.split()
if len(list_of_nodes) != num_points:
raise Exception(
"Number of names does not match the number of nodes provided"
)
for idx, node_name in enumerate(list_of_nodes):
network.names_dict[node_name] = idx
network.names_list[idx] = node_name
elif state == 'EDGES':
weights = line.split()
edge_counter += 1
# make a list of list of tuples
idx_node = network.names_dict[weights[0]]
idx_successor = network.names_dict[weights[2]]
network.successor_edges[idx_node][idx_successor] = int(
weights[1])
elif state == 'LINKS':
raise Exception(
"Simple Temporal Networks do not have contingent links."
)
else:
pass
if num_edges != edge_counter:
raise Exception(
"Number of edges does not match the number given above")
return network
def _read_stnu(self, file):
network = STN()
state = ""
for line in file:
if line.startswith('#'):
if "points" in line.lower() and "num" in line.lower():
state = "NO_POINTS"
elif "edges" in line.lower() and "num" in line.lower():
state = "NO_EDGES"
elif "links" in line.lower():
state = "NO_LINKS"
elif "names" in line.lower():
state = "NAMES"
elif "edges" in line.lower():
state = "EDGES"
elif "links" in line.lower():
state = "LINKS"
else:
raise Exception("Invalid Network Type")
else:
if state == 'NO_POINTS':
num_points = int(line)
network.length = num_points
network.successor_edges = [{} for i in range(num_points)]
network.names_list = ["0" for i in range(num_points)]
elif state == 'NO_EDGES':
num_edges = int(line)
elif state == 'NO_LINKS':
no_links = int(line)
elif state == 'NAMES':
list_of_nodes = line.split()
if len(list_of_nodes) != num_points:
raise Exception(
"Number of names does not match the number of nodes provided"
)
for idx, node_name in enumerate(list_of_nodes):
network.names_dict[node_name] = idx
network.names_list[idx] = node_name
elif state == 'EDGES':
weights = line.split()
# make a list of list of tuples
idx_node = network.names_dict[weights[0]]
idx_successor = network.names_dict[weights[2]]
network.successor_edges[idx_node][idx_successor] = int(
weights[1])
elif state == 'LINKS':
# deal with contingent links later
pass
else:
pass