# HG changeset patch
# User Paul Boddie <paul@boddie.org.uk>
# Date 1307057550 -7200
# Node ID b86dea2451a2c19c7c8ac16893de4e2f884eb052
# Parent  aa575fd26ceac0d0e63437d6a182fab64b978226
Added a command line option for graph production.

diff -r aa575fd26cea -r b86dea2451a2 test.py
--- a/test.py	Fri Jun 03 01:31:56 2011 +0200
+++ b/test.py	Fri Jun 03 01:32:30 2011 +0200
@@ -69,6 +69,19 @@
                 print "Generating report in", directory
                 micropython.report.report(p, directory)
 
+        if "-G" in args:
+            try:
+                graph = args[args.index("-G") + 1]
+            except IndexError:
+                print "No file specified. Not generating graph."
+            else:
+                print "Generating graph as", graph
+                f = open(graph, "w")
+                try:
+                    get_graph(p, f)
+                finally:
+                    f.close()
+
         # Build the program.
 
         if "-m" in args or "-t" in args: