1 |
#!/usr/bin/env python |
2 |
|
3 |
# $Id: boostgraph.py,v 1.3 2008/02/21 04:46:47 joko Exp $ |
4 |
|
5 |
# (c) 2008 Andreas Motl <andreas.motl@ilo.de> |
6 |
# (c) 2008 Sebastian Utz <su@rotamente.com> |
7 |
|
8 |
# This program is free software: you can redistribute it and/or modify |
9 |
# it under the terms of the GNU Affero General Public License as published by |
10 |
# the Free Software Foundation, either version 3 of the License, or |
11 |
# (at your option) any later version. |
12 |
# |
13 |
# This program is distributed in the hope that it will be useful, |
14 |
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
16 |
# GNU Affero General Public License for more details. |
17 |
# |
18 |
# You should have received a copy of the GNU Affero General Public License |
19 |
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
20 |
|
21 |
|
22 |
|
23 |
# Uses BGL-Python (depth_first_search) |
24 |
# |
25 |
# BGL-Python Homepage: http://osl.iu.edu/~dgregor/bgl-python/ |
26 |
# |
27 |
# Documentation (Boost): |
28 |
# - http://www.boost.org/libs/graph/doc/graph_theory_review.html |
29 |
# - http://www.boost.org/libs/graph/doc/depth_first_search.html |
30 |
# - http://www.boost.org/boost/graph/depth_first_search.hpp |
31 |
# - http://www.boost.org/libs/graph/doc/DFSVisitor.html |
32 |
# - http://www.boost.org/libs/graph/example/dfs-example.cpp |
33 |
# |
34 |
# Documentation (BGL-Python): |
35 |
# - http://osl.iu.edu/~dgregor/bgl-python/reference/boost.graph.html |
36 |
# |
37 |
# Subversion-Repository: https://svn.osl.iu.edu/svn/projects_viz/bgl-python/ |
38 |
|
39 |
|
40 |
RANDOM_MAX_NODES = 10 |
41 |
RANDOM_MAX_CHILDREN_PER_NODE = 500 |
42 |
MAX_SEARCH_DEPTH = 50 |
43 |
|
44 |
|
45 |
ENABLE_PROFILING = False |
46 |
|
47 |
if ENABLE_PROFILING: |
48 |
import profile |
49 |
from profile import Profile |
50 |
|
51 |
|
52 |
import sys, os |
53 |
import random |
54 |
|
55 |
sys.path.append('bgl_python') |
56 |
os.environ['PATH'] += ';' + './bgl_python' |
57 |
|
58 |
import boost.graph as bgl |
59 |
|
60 |
|
61 |
|
62 |
# from http://www.boost.org/libs/graph/doc/DFSVisitor.html |
63 |
class tree_edges_dfs_visitor(bgl.dfs_visitor): |
64 |
#class tree_edges_bfs_visitor(bgl.bfs_visitor): |
65 |
|
66 |
def __init__(self, startVertex, endVertex, maxdepth, color_map): |
67 |
|
68 |
#print dir(self) |
69 |
|
70 |
#bgl.dfs_visitor.__init__(self) |
71 |
#self.name_map = name_map |
72 |
|
73 |
self.startVertex = startVertex |
74 |
self.endVertex = endVertex |
75 |
|
76 |
# for recognizing path switches |
77 |
self.state = True |
78 |
|
79 |
# for tracking paths |
80 |
self.paths = [] |
81 |
self.current_path = [] |
82 |
|
83 |
# for limiting search depth |
84 |
""" |
85 |
self.color_map = color_map |
86 |
self.maxdepth = maxdepth |
87 |
self.depth = 0 |
88 |
""" |
89 |
|
90 |
self.level = 0 |
91 |
|
92 |
|
93 |
#def back_edge(self, e, g): |
94 |
# self.tree_edge(e, g, 'back_edge') |
95 |
|
96 |
#def forward_or_cross_edge(self, e, g): |
97 |
# self.tree_edge(e, g, 'forward_or_cross_edge') |
98 |
|
99 |
#def tree_edge(self, e, g): |
100 |
# self.tree_edge(e, g, 'examine_edge') |
101 |
|
102 |
def examine_edge(self, e, g): |
103 |
self._touch_edge(e, g, 'examine_edge') |
104 |
|
105 |
def _touch_edge(self, e, g, label=''): |
106 |
|
107 |
(u, v) = (g.source(e), g.target(e)) |
108 |
|
109 |
# increase current search depth (level) |
110 |
""" |
111 |
self.depth += 1 |
112 |
|
113 |
# check if maximum depth reached |
114 |
if self.depth == self.maxdepth: |
115 |
# color all succeeding vertices to black (mark as "already visited") |
116 |
# BUG!!! marks too many nodes |
117 |
for child_edge in g.out_edges(v): |
118 |
end_vertex = g.target(child_edge) |
119 |
#self.color_map[end_vertex] = bgl.Color(bgl.Color.gray) |
120 |
""" |
121 |
|
122 |
name_map = g.vertex_properties['node_id'] |
123 |
|
124 |
if label: |
125 |
print "%s:" % label, |
126 |
print "edge ", |
127 |
print name_map[u], |
128 |
print " -> ", |
129 |
print name_map[v] |
130 |
|
131 |
#self.state = True |
132 |
|
133 |
if u == self.startVertex: |
134 |
print "starting" |
135 |
self.current_path = [] |
136 |
|
137 |
# skip circular references |
138 |
if v != self.startVertex: |
139 |
self.current_path.append(e) |
140 |
|
141 |
if v == self.endVertex and self.current_path: |
142 |
print "found:" |
143 |
print self.current_path |
144 |
self.paths.append(self.current_path) |
145 |
self.current_path = [] |
146 |
|
147 |
|
148 |
""" |
149 |
def start_vertex(self, v, g): |
150 |
self._seperator('start_vertex') |
151 |
#pass |
152 |
#print 'sssss' |
153 |
|
154 |
def discover_vertex(self, v, g): |
155 |
#print '>>>' |
156 |
self._seperator('discover_vertex') |
157 |
#pass |
158 |
|
159 |
def initialize_vertex(self, v, g): |
160 |
#print '>>>' |
161 |
self._seperator('initialize_vertex') |
162 |
|
163 |
def examine_vertex(self, v, g): |
164 |
self._seperator('examine_vertex') |
165 |
|
166 |
def finish_vertex(self, v, g): |
167 |
#print '<<<' |
168 |
if self.current_path: |
169 |
self.paths.append(self.current_path) |
170 |
self.current_path = [] |
171 |
self.depth = 0 |
172 |
self._seperator('finish_vertex') |
173 |
""" |
174 |
|
175 |
def _seperator(self, label = 'unknown'): |
176 |
if 1 or self.state: |
177 |
print '-' * 21, label |
178 |
self.state = False |
179 |
|
180 |
|
181 |
def build_fixed_graph(): |
182 |
|
183 |
graph = bgl.Graph() |
184 |
index = {} |
185 |
|
186 |
# doesn't this work? see http://www.nabble.com/-Graph--Getting-PageRank-to-work-in-BGL-Python-td14207115.html |
187 |
#graph.add_vertex_property_map(name='my_name', type='float') |
188 |
|
189 |
# 'write_graphviz' requires property 'node_id' on vertices |
190 |
# see http://lists.boost.org/boost-users/2006/06/19877.php |
191 |
vmap = graph.vertex_property_map('string') |
192 |
graph.vertex_properties['node_id'] = vmap |
193 |
|
194 |
|
195 |
v1 = graph.add_vertex() |
196 |
vmap[v1] = '1' |
197 |
index['1'] = v1 |
198 |
|
199 |
v2 = graph.add_vertex() |
200 |
vmap[v2] = '2' |
201 |
index['2'] = v2 |
202 |
|
203 |
v3 = graph.add_vertex() |
204 |
vmap[v3] = '3' |
205 |
index['3'] = v3 |
206 |
|
207 |
v4 = graph.add_vertex() |
208 |
vmap[v4] = '4' |
209 |
index['4'] = v4 |
210 |
|
211 |
e1 = graph.add_edge(v1, v2) |
212 |
e2 = graph.add_edge(v1, v3) |
213 |
e3 = graph.add_edge(v3, v4) |
214 |
|
215 |
e4 = graph.add_edge(v1, v4) |
216 |
#e5 = graph.add_edge(v3, v2) |
217 |
#e6 = graph.add_edge(v2, v4) |
218 |
|
219 |
""" |
220 |
for vertex in graph.vertices: |
221 |
#print vertex.id, vertex |
222 |
print vmap[vertex], vertex |
223 |
#print vertex.__getattribute__('id') |
224 |
#print vertex['id'] |
225 |
#print vertex.get('id') |
226 |
#print |
227 |
""" |
228 |
|
229 |
graph.write_graphviz('friends.dot') |
230 |
|
231 |
return (graph, index) |
232 |
|
233 |
|
234 |
def build_random_graph(): |
235 |
|
236 |
graph = bgl.Graph() |
237 |
index = {} |
238 |
|
239 |
vmap = graph.vertex_property_map('string') |
240 |
graph.vertex_properties['node_id'] = vmap |
241 |
|
242 |
count = 0 |
243 |
for parent_id in range(1, RANDOM_MAX_NODES + 1): |
244 |
count += 1 |
245 |
if count % 100 == 0: |
246 |
sys.stderr.write('.') |
247 |
|
248 |
parent_id_str = str(parent_id) |
249 |
v1New = False |
250 |
if not index.has_key(parent_id_str): |
251 |
#print "adding v1:", parent_id_str |
252 |
myVertex1 = graph.add_vertex() |
253 |
vmap[myVertex1] = parent_id_str |
254 |
index[parent_id_str] = myVertex1 |
255 |
v1New = True |
256 |
|
257 |
|
258 |
count = 0 |
259 |
#for j in range(1, random.randint(1, RANDOM_MAX_CHILDREN_PER_NODE)): |
260 |
for j in range(1, RANDOM_MAX_CHILDREN_PER_NODE): |
261 |
|
262 |
count += 1 |
263 |
if count % 100 == 0: |
264 |
sys.stderr.write('.') |
265 |
|
266 |
parent_id_str = random.choice(index.keys()) |
267 |
|
268 |
child_id_str = parent_id_str |
269 |
while child_id_str == parent_id_str: |
270 |
child_id_str = random.choice(index.keys()) |
271 |
#print child_id_str |
272 |
|
273 |
myVertex1 = index[parent_id_str] |
274 |
myVertex2 = index[child_id_str] |
275 |
|
276 |
graph.add_edge(myVertex1, myVertex2) |
277 |
|
278 |
sys.stderr.write("\n") |
279 |
|
280 |
return (graph, index) |
281 |
|
282 |
|
283 |
def dump_track(graph, track): |
284 |
track_ids = [] |
285 |
for node in track: |
286 |
node_id = graph.vertex_properties['node_id'][node] |
287 |
track_ids.append(node_id) |
288 |
|
289 |
print ' -> '.join(track_ids) |
290 |
|
291 |
def dump_edges(g, edges): |
292 |
name_map = g.vertex_properties['node_id'] |
293 |
for e in edges: |
294 |
(u, v) = (g.source(e), g.target(e)) |
295 |
print "edge ", |
296 |
print name_map[u], |
297 |
print " -> ", |
298 |
print name_map[v] |
299 |
|
300 |
def find_path_solutions(source, target, graph, paths): |
301 |
|
302 |
print |
303 |
print "=" * 42 |
304 |
print "find_path_solutions" |
305 |
print "=" * 42 |
306 |
|
307 |
#print visitor.paths |
308 |
#(u, v) = (g.source(e), g.target(e)) |
309 |
for path in paths: |
310 |
startVertex = graph.source(path[0]) |
311 |
|
312 |
track = [] |
313 |
track.append(startVertex) |
314 |
|
315 |
for edge in path: |
316 |
endVertex = graph.target(edge) |
317 |
track.append(endVertex) |
318 |
if source == startVertex and target == endVertex: |
319 |
#print "found:", track |
320 |
dump_track(graph, track) |
321 |
|
322 |
def dump_graph(graph): |
323 |
for edge in graph.edges: |
324 |
(u, v) = (graph.source(edge), graph.target(edge)) |
325 |
startIndex = graph.vertex_properties['node_id'][u] |
326 |
endIndex = graph.vertex_properties['node_id'][v] |
327 |
print "%s -> %s" % (startIndex, endIndex) |
328 |
|
329 |
|
330 |
def main(): |
331 |
|
332 |
# Load a graph from the GraphViz file 'mst.dot' |
333 |
#graph = bgl.Graph.read_graphviz('mst.dot') |
334 |
(graph, index) = build_fixed_graph() |
335 |
#(graph, index) = build_random_graph() |
336 |
|
337 |
#dump_graph(graph) |
338 |
|
339 |
# Compute all paths rooted from each vertex |
340 |
#mst_edges = bgl.kruskal_minimum_spanning_tree(graph, weight) |
341 |
#bgl.depth_first_search(graph, root_vertex = None, visitor = None, color_map = None) |
342 |
|
343 |
""" |
344 |
for root in graph.vertices: |
345 |
print |
346 |
print '=' * 42 |
347 |
print 'Paths originating from node %s' % graph.vertex_properties['node_id'][root] |
348 |
#cmap = graph.vertex_property_map('color') |
349 |
cmap = None |
350 |
visitor = tree_edges_dfs_visitor(3, graph.vertex_properties['node_id'], cmap) |
351 |
#visitor = tree_edges_bfs_visitor(3, graph.vertex_properties['node_id'], cmap) |
352 |
bgl.depth_first_search(graph, root_vertex = root, visitor = visitor, color_map = None) |
353 |
#bgl.breadth_first_search(graph, root_vertex = root, visitor = visitor, color_map = None) |
354 |
#find_path_solutions(index['1'], index['4'], graph, visitor.paths) |
355 |
#sys.exit(0) |
356 |
""" |
357 |
|
358 |
startIndex = random.choice(index.keys()) |
359 |
endIndex = random.choice(index.keys()) |
360 |
startIndex = '1' |
361 |
endIndex = '4' |
362 |
|
363 |
print "Trying to find solution for: %s -> %s" % (startIndex, endIndex) |
364 |
|
365 |
startVertex = index[startIndex] |
366 |
endVertex = index[endIndex] |
367 |
|
368 |
def doCompute(): |
369 |
cmap = graph.vertex_property_map('color') |
370 |
visitor = tree_edges_dfs_visitor(startVertex, endVertex, MAX_SEARCH_DEPTH, cmap) |
371 |
bgl.depth_first_search(graph, root_vertex = startVertex, visitor = visitor, color_map = None) |
372 |
#bgl.depth_first_visit(graph, root_vertex = startVertex, visitor = visitor, color_map = cmap) |
373 |
#find_path_solutions(startVertex, endVertex, graph, visitor.paths) |
374 |
for path in visitor.paths: |
375 |
print '-' * 21 |
376 |
dump_edges(graph, path) |
377 |
|
378 |
if ENABLE_PROFILING: |
379 |
global paths |
380 |
paths = [] |
381 |
p = Profile() |
382 |
p.runcall(doCompute) |
383 |
p.print_stats() |
384 |
else: |
385 |
paths = doCompute() |
386 |
|
387 |
|
388 |
# STOP HERE |
389 |
sys.exit(0) |
390 |
|
391 |
|
392 |
|
393 |
|
394 |
|
395 |
# Compute the weight of the minimum spanning tree |
396 |
#print 'MST weight =',sum([weight[e] for e in mst_edges]) |
397 |
|
398 |
# Put the weights into the label. Make MST edges solid while all other |
399 |
# edges remain dashed. |
400 |
label = graph.edge_property_map('string') |
401 |
style = graph.edge_property_map('string') |
402 |
for e in graph.edges: |
403 |
label[e] = str(weight[e]) |
404 |
if e in mst_edges: |
405 |
style[e] = 'solid' |
406 |
else: |
407 |
style[e] = 'dashed' |
408 |
|
409 |
# Associate the label and style property maps with the graph for output |
410 |
graph.edge_properties['label'] = label |
411 |
graph.edge_properties['style'] = style |
412 |
|
413 |
# Write out the graph in GraphViz DOT format |
414 |
graph.write_graphviz('friends_path_2to3.dot') |
415 |
|
416 |
|
417 |
if __name__ == '__main__': |
418 |
main() |