# HG changeset patch
# User Rizmari Versfeld <rizziepit@gmail.com>
# Date 1336598562 -7200
# Node ID 94258a86f77381714d5bfe19bc8c17d9ed7bfffd
# Parent  0cdd136223550f32a5b1e2a16cda493980cca85e
upgrade research tree visualization

diff -r 0cdd13622355 -r 94258a86f773 gamelib/visualize.py
--- 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