Refactor naming scheme

This commit is contained in:
Johann Dreo 2014-03-22 18:23:22 +01:00
commit 35d9475033
5 changed files with 11 additions and 11 deletions

View file

View file

@ -4,7 +4,7 @@ import sys
import math import math
import random import random
from collections import Counter from collections import Counter
import path import shortpath
def LOG( *args ): def LOG( *args ):
@ -31,7 +31,7 @@ def euclidian_distance( ci, cj, graph = None):
return math.sqrt( float(ci[0] - cj[0])**2 + float(ci[1] - cj[1])**2 ) return math.sqrt( float(ci[0] - cj[0])**2 + float(ci[1] - cj[1])**2 )
def graph_distance( ci, cj, graph ): def graph_distance( ci, cj, graph ):
p,c = path.astar( graph, ci, cj ) p,c = shortpath.astar( graph, ci, cj )
return c return c

View file

@ -3,9 +3,9 @@
import sys import sys
import turtle import turtle
import lindenmayer import lindenmayer
import graph import utils
import ant_colony import ants
import path import shortpath
import uberplot import uberplot
import matplotlib.pyplot as plot import matplotlib.pyplot as plot
@ -39,7 +39,7 @@ with open("penrose_%i.segments" % depth, "w") as fd:
fd.write( str(penrose) ) fd.write( str(penrose) )
print "Convert the segment list into an adjacency list graph" print "Convert the segment list into an adjacency list graph"
G = graph.graph( penrose.segments ) G = utils.adjacency_from_set( penrose.segments )
print "Solve the TSP with an Ant Colony Algorithm" print "Solve the TSP with an Ant Colony Algorithm"
@ -52,13 +52,13 @@ w_local_phero = 0.1
c_greed = 0.9 c_greed = 0.9
w_history = 1.0 w_history = 1.0
best,phero = ant_colony.search( G, max_it, num_ants, decay, w_heur, w_local_phero, w_history, c_greed, cost_func = ant_colony.graph_distance ) best,phero = ants.search( G, max_it, num_ants, decay, w_heur, w_local_phero, w_history, c_greed, cost_func = ants.graph_distance )
print "Transform the resulting nodes permutation into a path on the graph" print "Transform the resulting nodes permutation into a path on the graph"
# by finding the shortest path between two cities. # by finding the shortest path between two cities.
traj = [] traj = []
for start,end in ant_colony.tour(best["permutation"]): for start,end in ants.tour(best["permutation"]):
p,c = path.astar( G, start, end ) p,c = shortpath.astar( G, start, end )
traj += p traj += p
print "traj",len(traj) print "traj",len(traj)
@ -83,7 +83,7 @@ for i in phero:
# uberplot.scatter_segments( ax, seg, color="red", alpha=0.5, linewidth=nph ) # uberplot.scatter_segments( ax, seg, color="red", alpha=0.5, linewidth=nph )
# best tour # best tour
uberplot.plot_segments( ax, ant_colony.tour(traj), color="red", alpha=0.9, linewidth=3 ) uberplot.plot_segments( ax, ants.tour(traj), color="red", alpha=0.9, linewidth=3 )
# tesselation # tesselation
tcol = "black" tcol = "black"

View file

@ -1,5 +1,5 @@
def graph( segments ): def adjacency_from_set( segments ):
graph = {} graph = {}
for start,end in segments: for start,end in segments:
graph[start] = graph.get( start, [] ) graph[start] = graph.get( start, [] )