Mention BulletYAML support. (Fixes issue #6)

[python-bulletml.git] / bulletml-runner

diff --git a/bulletml-runner b/bulletml-runner
index ec6007b..9136f21 100755 (executable)
--- a/bulletml-runner
+++ b/bulletml-runner
@@ -7,6 +7,19 @@ import time
 import pygame
 
 import bulletml
+import bulletml.bulletyaml
+
+try:
+    import yaml
+except ImportError:
+    yaml = None
+
+try:
+    import psyco
+except ImportError:
+    pass
+else:
+    psyco.full()
 
 def main(argv):
     if not argv:
@@ -24,7 +37,7 @@ def main(argv):
     while True:
         filename = argv[file_idx % len(argv)]
         doc = bulletml.BulletML.FromDocument(open(filename, "rU"))
-        actions = [act([], 0.5) for act in doc.top]
+        actions = [act([], 0.5) for act in doc.actions]
         source = bulletml.Bullet(
             x=150, y=150, target=target, actions=actions, rank=0.5)
                                          
@@ -59,7 +72,7 @@ def main(argv):
                     elif event.key == pygame.K_RETURN:
                         newfile = True
                     elif event.key == pygame.K_s:
-                        actions = [act([], 0.5) for act in doc.top]
+                        actions = [act([], 0.5) for act in doc.actions]
                         source = bulletml.Bullet(
                             x=150, y=150, target=target,
                             actions=actions, rank=0.5)
@@ -88,9 +101,9 @@ def main(argv):
                     print "  Processing: %04d: %d bullets, %d active." % (
                         frames, total, count)
                     if elapsed:
+                        seconds_per_bullet = elapsed / count
+                        bullets_per_second = count / elapsed
                         print "  %g seconds per bullet (60Hz max: %g)." % (
-                            seconds_per_bullet = elapsed / count
-                            bullets_per_second = count / elapsed
                             seconds_per_bullet, bullets_per_second / 60)
 
             screen.fill([0, 0, 0])