89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
|
|
class Route(object):
|
|
"""docstring for Route."""
|
|
def __init__(self, route_id, contract_id, load,src,dst):
|
|
super(Route, self).__init__()
|
|
self.route_id = route_id
|
|
self.contract_id = contract_id
|
|
self.load = load
|
|
self.src = src
|
|
self.dst = dst
|
|
|
|
def __repr__(self):
|
|
return str(self.route_id)+":"+self.src+" - "+self.dst+" "+self.load
|
|
|
|
|
|
class Contract(object):
|
|
"""docstring for Contract."""
|
|
def __init__(self, contract_id, routes):
|
|
super(Contract, self).__init__()
|
|
self.contract_id = contract_id
|
|
self.routes = routes
|
|
|
|
def __repr__(self):
|
|
return "\tContract : "+str(self.contract_id)+"\n\tRoutes\n\t\t"+"\n\t\t".join(map(repr,self.routes))
|
|
|
|
class Transporter(object):
|
|
"""docstring for Transporter."""
|
|
def __init__(self, contracts):
|
|
super(Transporter, self).__init__()
|
|
self.contracts = contracts
|
|
|
|
@classmethod
|
|
def from_file(cls,c_file):
|
|
return cls(cls.read_contracts(c_file))
|
|
|
|
@staticmethod
|
|
def read_contracts(c_file):
|
|
c_lines = open(c_file,'r').readlines()
|
|
c_dict = {}
|
|
for l in c_lines:
|
|
(c_id,src,dst,load) = l.split(' ')
|
|
if c_dict.has_key(c_id):
|
|
route_id = len(c_dict[c_id])
|
|
c_dict[c_id].append(Route(route_id,int(c_id),load,src,dst))
|
|
else:
|
|
c_dict[c_id] = [Route(0,int(c_id),load,src,dst)]
|
|
contrants = []
|
|
for c_id in c_dict:
|
|
contrants.append(Contract(int(c_id),c_dict[c_id]))
|
|
return contrants
|
|
|
|
def routes(self):
|
|
all_routes = []
|
|
routes_map = {}
|
|
for c in self.contracts:
|
|
for r in c.routes:
|
|
all_routes.append(r)
|
|
if routes_map.has_key(r.src):
|
|
routes_map[r.src].append(r.dst)
|
|
else:
|
|
routes_map[r.src]= [r.dst]
|
|
return all_routes,routes_map
|
|
|
|
def is_cyclic(self):
|
|
return circular_dep(self.routes()[1])
|
|
|
|
def find_loop(elem,dep_elem,deps,checked=set()):
|
|
if dep_elem in checked or not deps.has_key(dep_elem):
|
|
return (False,len(checked))
|
|
else:
|
|
checked.add(dep_elem)
|
|
for c in deps[dep_elem]:
|
|
if elem == c:
|
|
return (True,len(checked))
|
|
for c in deps[dep_elem]:
|
|
c_loop = find_loop(elem,c,deps,checked)
|
|
if c_loop[0]:
|
|
return c_loop
|
|
return (False,len(checked))
|
|
|
|
def circular_dep(deps):
|
|
dep_state = []
|
|
for k in deps.keys():
|
|
circ_dep_k = find_loop(k,k,deps,set())
|
|
dep_state.append((k,circ_dep_k))
|
|
return dep_state
|
|
|
|
print Transporter.from_file('./contracts_test.txt').is_cyclic()
|