def add_edge(e, R): """adds e to R and updates statistics.""" R.add_edges_from([tuple(e)]) a=R.get_edge_data(e[0], e[1]) a['c']=cf.routingCost(R,e,storeData=True) #update_after_mod(e,R) assert a['c']>=0 #who knows..
def addListProcedure(addList,remList, R, c,i,lastAdded=None):
"""
does some stuff connected to addList
This is the thing that adds complexity. 98% of the execution time is spent here.. insane..
"""
#print "goes into loop..."
#routingcost is really expensive, procedure to minimize number of calls.
for aTmp in addList:
cTmp=cf.routingCost(R,aTmp,storeData=False, add=True)
if abs(cTmp)<c and i-aTmp[2]['i_added']>1000: addList.remove(aTmp) #strange procedure, modify later.
else: aTmp[2]['c']=cTmp
if len(addList)!=0:
addList=sorted(addList, key=lambda edge: edge[2]['c']) #first sort
if len(addList)>2: assert addList[0][2]['c']<=addList[1][2]['c']
a=addList[0]
eps=1e-9 #tolerance
if a[2]['c']<-eps and abs(a[2]['c'])>c+eps: #add edge again
if lastAdded and lastAdded==a[0:2]:
print 'noWay'
return False, addList, remList, lastAdded
lastAdded=a[0:2]
print "adds. e:", a[0:2], " ec:", c, "benefit:", a[2]['c']
addList.remove(a)
#for at in addList:
# if at[0:2]==[(120.0, 168.0), (120.0, 144.0)]: raise Exception('edge was never removed..')
a[2]['c']*=-1 #you don't get any benefit from it now.
assert a[2]['c']>=0 #easy to get it wrong..
add_edge(a, R)
remList.append(a)
#print i, "1R has (312.0, 96.0), (312.0, 120.0):", R.has_edge((312.0, 96.0), (312.0, 120.0))
return True, addList, remList, lastAdded #next while cycle
return False, addList, remList, lastAdded
def bruteForce(R, G=None, ax=None, aCap=0.25, beta=1.5, add=True):
"""
works with cycles instead of shortest paths
VERY computationally complex. The earlier algorithms could handle much bigger systems.
modifies G into R:... this is a strange procedure and should be changed.
"""
R.beta=beta
if not G: G=copy.deepcopy(R)
inf = 1e15
eps=1e-9
lastAdded=None
origin=G.origin
if not origin: raise Exception('need info about origin')
#first, modify the weight of the edges a couple of times. Warmup
warmUp(R)
paths=nx.algorithms.shortest_paths.weighted.single_source_dijkstra(R, origin)
#now, start for real and save away all kind of info about the paths.
for node in R.nodes(data=True):
p1=paths[1][node[0]]
p1.reverse()
node[1]['shortest_path']=p1
if len (p1)<=1:
node[1]['second_shortest']=p1
else:
e=[p1[0], p1[1]] #edge closest to point
e.append(R.get_edge_data(e[0], e[1]))
R.remove_edge(e[0], e[1]) #temporary remove to get loop
p2=nx.dijkstra_path(R,origin, node[0])
node[1]['second_shortest']=p2
R.add_edges_from([tuple(e)]) #reset road
#ax=testRoads(R, p1, p2, ax) #used for debugging
for path in p1,p2:
last=None
for nTmp in path:
if last:
d=R.get_edge_data(*(last, nTmp)) #should always exist if designed properly
d['visits']+=1
if not node in d['visited_from_node']: d['visited_from_node'].append(node)
last=nTmp
remList=[]
addList=[]
for eTmp in R.edges(data=False):
e=copy.copy(eTmp) #copy, so we can remove them and then add them and so on.
e2=R.get_edge_data(*e) #not a copy, reference to real dict.
e=list(e)
e.append(e2)
e[2]['origin_dist']=distToOrigin(e,R)
e[2]['c']=cf.routingCost(R, e, storeData=False)
remList.append(e)
i=1
#at this point, the visited thing should be updated
for edge in R.edges(data=True):
modifyEdge(edge,R,reset=True) #reset to real weight.
while len(remList)>0:
i=i+1
print "start"
first=True
e1=False
assert len([e for e in R.edges(data=True) if e[2]['weight']<0])==0
while first or e[0:2]!=e1[0:2]:
e1=remList[0] #takes the last item in the list.
e1[2]=R.get_edge_data(e1[0], e1[1]) #edgelist is a copy, this is not.
c=cf.routingCost(R, e1, storeData=True) #also updates e[2]['c']
print " c: ", c, " len:", len(remList), "visits:", e1[2]['visits']
#e1[2]=R.get_edge_data(e1[0], e1[1])
e1[2]['c']=c #-e1[2]['origin_dist']*0.1 #origin dist just to experiment.
if c>=inf:
remList.remove(remList[0]) #cannot be empty at this time
if len(remList)==0: break
remList=sorted(remList, key=lambda edge: edge[2]['c'])#+edge[2]['visits'])#-edge[2]['origin_dist']*0.1) #first sort
e=remList[0] # could be the same..
if first: first=False
print "the chosen one:",e[2]['c'], e[0:2]
print "areacover:", R.areaCover
e[2]=R.get_edge_data(*e) # Update again, to get "new second shortest"
if add and random.uniform(0,1)<0.33:
print "adds"
added, addList, remList, lastAdded=addListProcedure(addList,remList,R,e[2]['c'],i,lastAdded)
if added: continue #go up to while again
if len(remList)==0: break
#print "will remove:", remList[0][0:2]
remList.remove(remList[0]) #we know now that no edge is added this "round"
#print "is it still there?"
#for r in remList:
# print r[0:2]
#print "removed, next", remList[0][0:2]
print e[2]['c']
if e[2]['c']>eps and aCap and R.areaCover-go.singleRoadSegmentCoverage(e, R, remove=True)*G.Ainv<aCap:
print "tries to exit", e[0:2], "ec:", e[2]['c']
if add:
added, addList, remList, lastAdded=addListProcedure(addList,remList,R,e[2]['c'],i,lastAdded)
if added:
remList.append(e) #since we just removed it and didn't remove it from graph
continue
break #we are finished
if R.degree(e[0])>2 and R.degree(e[1])>2 and c<inf: #loop condition, at least degree 3
print "removes edge ",e[0:2]
remove_edge(e, R) #remove from R.
addList.append(e) #add to lists for potential adding again.
e[2]['c']*=-1 #reverse, we now gain c by adding it again.
assert e[2]['c']<=0
e[2]['i_added']=i
#this procedure can most certainly be speeded up, expensive operations.
update_after_mod(e,R)
for e in R.edges(data=True):
modifyEdge(e, R, reset=True)
print "construction finished."
print "road area coverage:", R.areaCover
print "total area:", G.A
return R