Merge ktlim@zamp:/z/ktlim2/clean/newmem-merge

[gem5.git] / configs / test / test.py

diff --git a/configs/test/test.py b/configs/test/test.py
index 86a44313a5121845bdd4c8a983d689537f5c01e4..2047d65af81e17f3d54aee4f980d3f28cf2290d6 100644 (file)
--- a/configs/test/test.py
+++ b/configs/test/test.py
@@ -1,10 +1,63 @@
-from m5 import *
+# Simple test script
+#
+# Alpha: "m5 test.py"
+# MIPS: "m5 test.py -a Mips -c hello_mips"
 
-class HelloWorld(AlphaLiveProcess):
-    executable = '../configs/test/hello'
-    cmd = 'hello'
+import os, optparse, sys
+import m5
+from m5.objects import *
+from FullO3Config import *
 
+# parse command-line arguments
+parser = optparse.OptionParser(option_list=m5.standardOptions)
+
+parser.add_option("-c", "--cmd", default="hello")
+parser.add_option("-o", "--options", default="")
+parser.add_option("-i", "--input", default="")
+parser.add_option("-t", "--timing", action="store_true")
+parser.add_option("-d", "--detailed", action="store_true")
+parser.add_option("-m", "--maxtick", type="int")
+
+(options, args) = parser.parse_args()
+m5.setStandardOptions(options)
+
+if args:
+    print "Error: script doesn't take any positional arguments"
+    sys.exit(1)
+
+# build configuration
+this_dir = os.path.dirname(__file__)
+
+process = LiveProcess()
+process.executable = os.path.join(this_dir, options.cmd)
+process.cmd = options.cmd + " " + options.options
+if options.input != "":
+    process.input = options.input
+
+magicbus = Bus()
 mem = PhysicalMemory()
-cpu = SimpleCPU(workload=HelloWorld(), mem=mem)
-system = System(physmem=mem, cpu=cpu)
-root = Root(system=system)
+
+if options.timing:
+    cpu = TimingSimpleCPU()
+elif options.detailed:
+    cpu = DetailedO3CPU()
+else:
+    cpu = AtomicSimpleCPU()
+cpu.workload = process
+cpu.mem = magicbus
+
+system = System(physmem = mem, cpu = cpu)
+mem.port = magicbus.port
+root = Root(system = system)
+
+# instantiate configuration
+m5.instantiate(root)
+
+# simulate until program terminates
+if options.maxtick:
+    exit_event = m5.simulate(options.maxtick)
+else:
+    exit_event = m5.simulate()
+
+print 'Exiting @ tick', m5.curTick(), 'because', exit_event.getCause()
+