changeset 117:94258a86f773

upgrade research tree visualization
author Rizmari Versfeld <rizziepit@gmail.com>
date Wed, 09 May 2012 23:22:42 +0200
parents 0cdd13622355
children bc9da66ec333
files gamelib/visualize.py
diffstat 1 files changed, 15 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/gamelib/visualize.py	Wed May 09 23:03:45 2012 +0200
+++ b/gamelib/visualize.py	Wed May 09 23:22:42 2012 +0200
@@ -9,14 +9,24 @@
         rt.add_node(research, label=research.NAME, color='green')
     for research in lab.new_research:
         for prereq in research.PREREQUISITES:
-            rt.add_edge(prereq[0], research, weight=1.0 / prereq[1], \
-                label=prereq[1], color='green')
+            prefix = prereq[0].NAME[0:3]
+            cur = prereq[0]
+            for i in range(1, prereq[1] + 1):
+                rt.add_node(prefix + str(i), color='green')
+                rt.add_edge(cur, prefix + str(i), color='green')
+                cur = prefix + str(i)
+            rt.add_edge(cur, research, color='green')
     for schematic in lab.new_schematics:
-        rt.add_node(schematic, label=schematic.NAME, color='red')
+        rt.add_node(schematic, label=schematic.NAME, color='pink')
     for schematic in lab.new_schematics:
         for prereq in schematic.PREREQUISITES:
-            rt.add_edge(prereq[0], schematic, weight=1.0 / prereq[1], \
-                label=prereq[1], color='red', size=schematic.COST / 10)
+            prefix = prereq[0].NAME[0:3]
+            cur = prereq[0]
+            for i in range(1, prereq[1] + 1):
+                rt.add_node(prefix + str(i), color='pink')
+                rt.add_edge(cur, prefix + str(i), color='pink')
+                cur = prefix + str(i)
+            rt.add_edge(cur, schematic, color='pink')
     return rt