1 # Copyright (c) 2005 The Regents of The University of Michigan
4 # Redistribution and use in source and binary forms, with or without
5 # modification, are permitted provided that the following conditions are
6 # met: redistributions of source code must retain the above copyright
7 # notice, this list of conditions and the following disclaimer;
8 # redistributions in binary form must reproduce the above copyright
9 # notice, this list of conditions and the following disclaimer in the
10 # documentation and/or other materials provided with the distribution;
11 # neither the name of the copyright holders nor the names of its
12 # contributors may be used to endorse or promote products derived from
13 # this software without specific prior written permission.
15 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 # Authors: Nathan Binkert
30 import sys
, os
, time
, atexit
, optparse
32 # import the SWIG-wrapped main C++ functions
34 # import a few SWIG-wrapped items (those that are likely to be used
35 # directly by user scripts) completely into this module for
37 from main
import simulate
, SimLoopExitEvent
39 # import the m5 compile options
42 # define this here so we can use it right away if necessary
44 print >>sys
.stderr
, 'panic:', string
47 # Prepend given directory to system module search path. We may not
48 # need this anymore if we can structure our config library more like a
51 # if it's a relative path and we know what directory the current
52 # python script is in, make the path relative to that directory.
53 if not os
.path
.isabs(path
) and sys
.path
[0]:
54 path
= os
.path
.join(sys
.path
[0], path
)
55 path
= os
.path
.realpath(path
)
56 # sys.path[0] should always refer to the current script's directory,
57 # so place the new dir right after that.
58 sys
.path
.insert(1, path
)
61 # Callback to set trace flags. Not necessarily the best way to do
62 # things in the long run (particularly if we change how these global
63 # options are handled).
64 def setTraceFlags(option
, opt_str
, value
, parser
):
65 objects
.Trace
.flags
= value
67 # Standard optparse options. Need to be explicitly included by the
68 # user script when it calls optparse.OptionParser().
70 optparse
.make_option("--traceflags", type="string", action
="callback",
71 callback
=setTraceFlags
)
74 # make a SmartDict out of the build options for our local use
76 build_env
= smartdict
.SmartDict()
77 build_env
.update(defines
.m5_build_env
)
79 # make a SmartDict out of the OS environment too
80 env
= smartdict
.SmartDict()
81 env
.update(os
.environ
)
83 # The final hook to generate .ini files. Called from the user script
84 # once the config is built.
85 def instantiate(root
):
86 config
.ticks_per_sec
= float(root
.clock
.frequency
)
87 # ugly temporary hack to get output to config.ini
88 sys
.stdout
= file('config.ini', 'w')
90 sys
.stdout
.close() # close config.ini
91 sys
.stdout
= sys
.__stdout
__ # restore to original
92 main
.initialize() # load config.ini into C++ and process it
93 noDot
= True # temporary until we fix dot
96 instance
.outputDot(dot
)
97 dot
.orientation
= "portrait"
101 dot
.write("config.dot")
102 dot
.write_ps("config.ps")
104 # Export curTick to user script.
106 return main
.cvar
.curTick
108 # register our C++ exit callback function with Python
109 atexit
.register(main
.doExitCleanup
)
111 # This import allows user scripts to reference 'm5.objects.Foo' after
112 # just doing an 'import m5' (without an 'import m5.objects'). May not
113 # matter since most scripts will probably 'from m5.objects import *'.