changeset 118:bc9da66ec333

merge
author Rizmari Versfeld <rizziepit@gmail.com>
date Wed, 09 May 2012 23:23:00 +0200
parents 94258a86f773 (diff) a8a46c14d467 (current diff)
children 9a30162f2a9c
files
diffstat 1 files changed, 15 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/gamelib/visualize.py	Wed May 09 23:20:04 2012 +0200
+++ b/gamelib/visualize.py	Wed May 09 23:23:00 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