SConscript revision 8334
1955SN/A# -*- mode:python -*-
2955SN/A
31762SN/A# Copyright (c) 2004-2005 The Regents of The University of Michigan
4955SN/A# All rights reserved.
5955SN/A#
6955SN/A# Redistribution and use in source and binary forms, with or without
7955SN/A# modification, are permitted provided that the following conditions are
8955SN/A# met: redistributions of source code must retain the above copyright
9955SN/A# notice, this list of conditions and the following disclaimer;
10955SN/A# redistributions in binary form must reproduce the above copyright
11955SN/A# notice, this list of conditions and the following disclaimer in the
12955SN/A# documentation and/or other materials provided with the distribution;
13955SN/A# neither the name of the copyright holders nor the names of its
14955SN/A# contributors may be used to endorse or promote products derived from
15955SN/A# this software without specific prior written permission.
16955SN/A#
17955SN/A# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18955SN/A# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19955SN/A# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20955SN/A# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21955SN/A# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22955SN/A# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23955SN/A# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24955SN/A# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25955SN/A# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26955SN/A# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27955SN/A# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
282665Ssaidi@eecs.umich.edu#
294762Snate@binkert.org# Authors: Nathan Binkert
30955SN/A
315522Snate@binkert.orgimport array
324762Snate@binkert.orgimport bisect
335522Snate@binkert.orgimport imp
34955SN/Aimport marshal
355522Snate@binkert.orgimport os
36955SN/Aimport re
375522Snate@binkert.orgimport sys
384202Sbinkertn@umich.eduimport zlib
395342Sstever@gmail.com
40955SN/Afrom os.path import basename, dirname, exists, isdir, isfile, join as joinpath
414381Sbinkertn@umich.edu
424381Sbinkertn@umich.eduimport SCons
43955SN/A
44955SN/A# This file defines how to build a particular configuration of gem5
45955SN/A# based on variable settings in the 'env' build environment.
464202Sbinkertn@umich.edu
47955SN/AImport('*')
484382Sbinkertn@umich.edu
494382Sbinkertn@umich.edu# Children need to see the environment
504382Sbinkertn@umich.eduExport('env')
515517Snate@binkert.org
525517Snate@binkert.orgbuild_env = [(opt, env[opt]) for opt in export_vars]
534762Snate@binkert.org
544762Snate@binkert.orgfrom m5.util import code_formatter
554762Snate@binkert.org
564762Snate@binkert.org########################################################################
574762Snate@binkert.org# Code for adding source files of various types
584762Snate@binkert.org#
594762Snate@binkert.org# When specifying a source file of some type, a set of guards can be
604762Snate@binkert.org# specified for that file.  When get() is used to find the files, if
614762Snate@binkert.org# get specifies a set of filters, only files that match those filters
624762Snate@binkert.org# will be accepted (unspecified filters on files are assumed to be
635522Snate@binkert.org# false).  Current filters are:
644762Snate@binkert.org#     main -- specifies the gem5 main() function
654762Snate@binkert.org#     skip_lib -- do not put this file into the gem5 library
664762Snate@binkert.org#     <unittest> -- unit tests use filters based on the unit test name
674762Snate@binkert.org#
684762Snate@binkert.org# A parent can now be specified for a source file and default filter
695522Snate@binkert.org# values will be retrieved recursively from parents (children override
705522Snate@binkert.org# parents).
715522Snate@binkert.org#
725522Snate@binkert.orgclass SourceMeta(type):
734762Snate@binkert.org    '''Meta class for source files that keeps track of all files of a
744762Snate@binkert.org    particular type and has a get function for finding all functions
754762Snate@binkert.org    of a certain type that match a set of guards'''
764762Snate@binkert.org    def __init__(cls, name, bases, dict):
774762Snate@binkert.org        super(SourceMeta, cls).__init__(name, bases, dict)
785522Snate@binkert.org        cls.all = []
794762Snate@binkert.org        
804762Snate@binkert.org    def get(cls, **guards):
815522Snate@binkert.org        '''Find all files that match the specified guards.  If a source
825522Snate@binkert.org        file does not specify a flag, the default is False'''
834762Snate@binkert.org        for src in cls.all:
844762Snate@binkert.org            for flag,value in guards.iteritems():
854762Snate@binkert.org                # if the flag is found and has a different value, skip
864762Snate@binkert.org                # this file
874762Snate@binkert.org                if src.all_guards.get(flag, False) != value:
884762Snate@binkert.org                    break
895522Snate@binkert.org            else:
905522Snate@binkert.org                yield src
915522Snate@binkert.org
924762Snate@binkert.orgclass SourceFile(object):
934382Sbinkertn@umich.edu    '''Base object that encapsulates the notion of a source file.
944762Snate@binkert.org    This includes, the source node, target node, various manipulations
954382Sbinkertn@umich.edu    of those.  A source file also specifies a set of guards which
965522Snate@binkert.org    describing which builds the source file applies to.  A parent can
974381Sbinkertn@umich.edu    also be specified to get default guards from'''
985522Snate@binkert.org    __metaclass__ = SourceMeta
994762Snate@binkert.org    def __init__(self, source, parent=None, **guards):
1004762Snate@binkert.org        self.guards = guards
1014762Snate@binkert.org        self.parent = parent
1025522Snate@binkert.org
1035522Snate@binkert.org        tnode = source
1045522Snate@binkert.org        if not isinstance(source, SCons.Node.FS.File):
1055522Snate@binkert.org            tnode = File(source)
1065522Snate@binkert.org
1075522Snate@binkert.org        self.tnode = tnode
1085522Snate@binkert.org        self.snode = tnode.srcnode()
1095522Snate@binkert.org
1105522Snate@binkert.org        for base in type(self).__mro__:
1114762Snate@binkert.org            if issubclass(base, SourceFile):
1124762Snate@binkert.org                base.all.append(self)
1134762Snate@binkert.org
1144762Snate@binkert.org    @property
1154762Snate@binkert.org    def filename(self):
1164762Snate@binkert.org        return str(self.tnode)
1174762Snate@binkert.org
1184762Snate@binkert.org    @property
1194762Snate@binkert.org    def dirname(self):
1204762Snate@binkert.org        return dirname(self.filename)
1214762Snate@binkert.org
1224762Snate@binkert.org    @property
1234762Snate@binkert.org    def basename(self):
1244762Snate@binkert.org        return basename(self.filename)
1254762Snate@binkert.org
1264762Snate@binkert.org    @property
1274762Snate@binkert.org    def extname(self):
1284762Snate@binkert.org        index = self.basename.rfind('.')
1294762Snate@binkert.org        if index <= 0:
1304762Snate@binkert.org            # dot files aren't extensions
1314762Snate@binkert.org            return self.basename, None
1324762Snate@binkert.org
1334762Snate@binkert.org        return self.basename[:index], self.basename[index+1:]
1344762Snate@binkert.org
1354762Snate@binkert.org    @property
1364762Snate@binkert.org    def all_guards(self):
1374762Snate@binkert.org        '''find all guards for this object getting default values
1384762Snate@binkert.org        recursively from its parents'''
1394762Snate@binkert.org        guards = {}
1404762Snate@binkert.org        if self.parent:
1414762Snate@binkert.org            guards.update(self.parent.guards)
1424762Snate@binkert.org        guards.update(self.guards)
1434762Snate@binkert.org        return guards
1444762Snate@binkert.org
1454762Snate@binkert.org    def __lt__(self, other): return self.filename < other.filename
146955SN/A    def __le__(self, other): return self.filename <= other.filename
1475584Snate@binkert.org    def __gt__(self, other): return self.filename > other.filename
1485584Snate@binkert.org    def __ge__(self, other): return self.filename >= other.filename
1495584Snate@binkert.org    def __eq__(self, other): return self.filename == other.filename
1505584Snate@binkert.org    def __ne__(self, other): return self.filename != other.filename
1515584Snate@binkert.org        
1525584Snate@binkert.orgclass Source(SourceFile):
1535584Snate@binkert.org    '''Add a c/c++ source file to the build'''
1545584Snate@binkert.org    def __init__(self, source, Werror=True, swig=False, **guards):
1555584Snate@binkert.org        '''specify the source file, and any guards'''
1565584Snate@binkert.org        super(Source, self).__init__(source, **guards)
1575584Snate@binkert.org
1585584Snate@binkert.org        self.Werror = Werror
1595584Snate@binkert.org        self.swig = swig
1604382Sbinkertn@umich.edu
1614202Sbinkertn@umich.educlass PySource(SourceFile):
1625522Snate@binkert.org    '''Add a python source file to the named package'''
1634382Sbinkertn@umich.edu    invalid_sym_char = re.compile('[^A-z0-9_]')
1644382Sbinkertn@umich.edu    modules = {}
1654382Sbinkertn@umich.edu    tnodes = {}
1665584Snate@binkert.org    symnames = {}
1674382Sbinkertn@umich.edu    
1684382Sbinkertn@umich.edu    def __init__(self, package, source, **guards):
1694382Sbinkertn@umich.edu        '''specify the python package, the source file, and any guards'''
1705192Ssaidi@eecs.umich.edu        super(PySource, self).__init__(source, **guards)
1715192Ssaidi@eecs.umich.edu
1725192Ssaidi@eecs.umich.edu        modname,ext = self.extname
1735192Ssaidi@eecs.umich.edu        assert ext == 'py'
1745192Ssaidi@eecs.umich.edu
1755192Ssaidi@eecs.umich.edu        if package:
1765192Ssaidi@eecs.umich.edu            path = package.split('.')
1775192Ssaidi@eecs.umich.edu        else:
1785192Ssaidi@eecs.umich.edu            path = []
1795192Ssaidi@eecs.umich.edu
1805192Ssaidi@eecs.umich.edu        modpath = path[:]
1815192Ssaidi@eecs.umich.edu        if modname != '__init__':
1825192Ssaidi@eecs.umich.edu            modpath += [ modname ]
1835192Ssaidi@eecs.umich.edu        modpath = '.'.join(modpath)
1845192Ssaidi@eecs.umich.edu
1855192Ssaidi@eecs.umich.edu        arcpath = path + [ self.basename ]
1865192Ssaidi@eecs.umich.edu        abspath = self.snode.abspath
1875192Ssaidi@eecs.umich.edu        if not exists(abspath):
1885192Ssaidi@eecs.umich.edu            abspath = self.tnode.abspath
1895192Ssaidi@eecs.umich.edu
1905192Ssaidi@eecs.umich.edu        self.package = package
1915192Ssaidi@eecs.umich.edu        self.modname = modname
1925192Ssaidi@eecs.umich.edu        self.modpath = modpath
1935192Ssaidi@eecs.umich.edu        self.arcname = joinpath(*arcpath)
1945192Ssaidi@eecs.umich.edu        self.abspath = abspath
1955192Ssaidi@eecs.umich.edu        self.compiled = File(self.filename + 'c')
1965192Ssaidi@eecs.umich.edu        self.cpp = File(self.filename + '.cc')
1975192Ssaidi@eecs.umich.edu        self.symname = PySource.invalid_sym_char.sub('_', modpath)
1985192Ssaidi@eecs.umich.edu
1995192Ssaidi@eecs.umich.edu        PySource.modules[modpath] = self
2005192Ssaidi@eecs.umich.edu        PySource.tnodes[self.tnode] = self
2015192Ssaidi@eecs.umich.edu        PySource.symnames[self.symname] = self
2024382Sbinkertn@umich.edu
2034382Sbinkertn@umich.educlass SimObject(PySource):
2044382Sbinkertn@umich.edu    '''Add a SimObject python file as a python source object and add
2052667Sstever@eecs.umich.edu    it to a list of sim object modules'''
2062667Sstever@eecs.umich.edu
2072667Sstever@eecs.umich.edu    fixed = False
2082667Sstever@eecs.umich.edu    modnames = []
2092667Sstever@eecs.umich.edu
2102667Sstever@eecs.umich.edu    def __init__(self, source, **guards):
2112037SN/A        '''Specify the source file and any guards (automatically in
2122037SN/A        the m5.objects package)'''
2132037SN/A        super(SimObject, self).__init__('m5.objects', source, **guards)
2144382Sbinkertn@umich.edu        if self.fixed:
2154762Snate@binkert.org            raise AttributeError, "Too late to call SimObject now."
2165344Sstever@gmail.com
2174382Sbinkertn@umich.edu        bisect.insort_right(SimObject.modnames, self.modname)
2185341Sstever@gmail.com
2195341Sstever@gmail.comclass SwigSource(SourceFile):
2205341Sstever@gmail.com    '''Add a swig file to build'''
2215344Sstever@gmail.com
2225341Sstever@gmail.com    def __init__(self, package, source, **guards):
2235341Sstever@gmail.com        '''Specify the python package, the source file, and any guards'''
2245341Sstever@gmail.com        super(SwigSource, self).__init__(source, **guards)
2254762Snate@binkert.org
2265341Sstever@gmail.com        modname,ext = self.extname
2275344Sstever@gmail.com        assert ext == 'i'
2285341Sstever@gmail.com
2294773Snate@binkert.org        self.module = modname
2301858SN/A        cc_file = joinpath(self.dirname, modname + '_wrap.cc')
2311858SN/A        py_file = joinpath(self.dirname, modname + '.py')
2321085SN/A
2334382Sbinkertn@umich.edu        self.cc_source = Source(cc_file, swig=True, parent=self)
2344382Sbinkertn@umich.edu        self.py_source = PySource(package, py_file, parent=self)
2354762Snate@binkert.org
2364762Snate@binkert.orgclass UnitTest(object):
2374762Snate@binkert.org    '''Create a UnitTest'''
2385517Snate@binkert.org
2395517Snate@binkert.org    all = []
2405517Snate@binkert.org    def __init__(self, target, *sources):
2415517Snate@binkert.org        '''Specify the target name and any sources.  Sources that are
2425517Snate@binkert.org        not SourceFiles are evalued with Source().  All files are
2435517Snate@binkert.org        guarded with a guard of the same name as the UnitTest
2445517Snate@binkert.org        target.'''
2455517Snate@binkert.org
2465517Snate@binkert.org        srcs = []
2475517Snate@binkert.org        for src in sources:
2485517Snate@binkert.org            if not isinstance(src, SourceFile):
2495517Snate@binkert.org                src = Source(src, skip_lib=True)
2505517Snate@binkert.org            src.guards[target] = True
2515517Snate@binkert.org            srcs.append(src)
2525517Snate@binkert.org
2535517Snate@binkert.org        self.sources = srcs
2545517Snate@binkert.org        self.target = target
2555517Snate@binkert.org        UnitTest.all.append(self)
2565517Snate@binkert.org
2575517Snate@binkert.org# Children should have access
2585517Snate@binkert.orgExport('Source')
2595517Snate@binkert.orgExport('PySource')
2605517Snate@binkert.orgExport('SimObject')
2615517Snate@binkert.orgExport('SwigSource')
2625517Snate@binkert.orgExport('UnitTest')
2635517Snate@binkert.org
2645517Snate@binkert.org########################################################################
2655517Snate@binkert.org#
2665517Snate@binkert.org# Debug Flags
2675517Snate@binkert.org#
2685517Snate@binkert.orgdebug_flags = {}
2695517Snate@binkert.orgdef DebugFlag(name, desc=None):
2705517Snate@binkert.org    if name in debug_flags:
2715517Snate@binkert.org        raise AttributeError, "Flag %s already specified" % name
2725517Snate@binkert.org    debug_flags[name] = (name, (), desc)
2735517Snate@binkert.orgTraceFlag = DebugFlag
2745517Snate@binkert.org
2755517Snate@binkert.orgdef CompoundFlag(name, flags, desc=None):
2765517Snate@binkert.org    if name in debug_flags:
2775517Snate@binkert.org        raise AttributeError, "Flag %s already specified" % name
2785517Snate@binkert.org
2795517Snate@binkert.org    compound = tuple(flags)
2805517Snate@binkert.org    debug_flags[name] = (name, compound, desc)
2815517Snate@binkert.org
2825517Snate@binkert.orgExport('DebugFlag')
2835517Snate@binkert.orgExport('TraceFlag')
2845517Snate@binkert.orgExport('CompoundFlag')
2855517Snate@binkert.org
2865517Snate@binkert.org########################################################################
2875517Snate@binkert.org#
2885517Snate@binkert.org# Set some compiler variables
2895517Snate@binkert.org#
2905517Snate@binkert.org
2915517Snate@binkert.org# Include file paths are rooted in this directory.  SCons will
2925517Snate@binkert.org# automatically expand '.' to refer to both the source directory and
2935517Snate@binkert.org# the corresponding build directory to pick up generated include
2945522Snate@binkert.org# files.
2955517Snate@binkert.orgenv.Append(CPPPATH=Dir('.'))
2965517Snate@binkert.org
2975517Snate@binkert.orgfor extra_dir in extras_dir_list:
2985517Snate@binkert.org    env.Append(CPPPATH=Dir(extra_dir))
2994762Snate@binkert.org
3005517Snate@binkert.org# Workaround for bug in SCons version > 0.97d20071212
3015517Snate@binkert.org# Scons bug id: 2006 gem5 Bug id: 308
3024762Snate@binkert.orgfor root, dirs, files in os.walk(base_dir, topdown=True):
3035517Snate@binkert.org    Dir(root[len(base_dir) + 1:])
3044762Snate@binkert.org
3055517Snate@binkert.org########################################################################
3065517Snate@binkert.org#
3075517Snate@binkert.org# Walk the tree and execute all SConscripts in subdirectories
3085517Snate@binkert.org#
3095517Snate@binkert.org
3105517Snate@binkert.orghere = Dir('.').srcnode().abspath
3115517Snate@binkert.orgfor root, dirs, files in os.walk(base_dir, topdown=True):
3125517Snate@binkert.org    if root == here:
3135517Snate@binkert.org        # we don't want to recurse back into this SConscript
3145517Snate@binkert.org        continue
3155517Snate@binkert.org
3165517Snate@binkert.org    if 'SConscript' in files:
3175517Snate@binkert.org        build_dir = joinpath(env['BUILDDIR'], root[len(base_dir) + 1:])
3185517Snate@binkert.org        SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
3195517Snate@binkert.org
3205517Snate@binkert.orgfor extra_dir in extras_dir_list:
3215517Snate@binkert.org    prefix_len = len(dirname(extra_dir)) + 1
3225517Snate@binkert.org    for root, dirs, files in os.walk(extra_dir, topdown=True):
3235517Snate@binkert.org        # if build lives in the extras directory, don't walk down it
3245517Snate@binkert.org        if 'build' in dirs:
3255517Snate@binkert.org            dirs.remove('build')
3265517Snate@binkert.org
3275517Snate@binkert.org        if 'SConscript' in files:
3284762Snate@binkert.org            build_dir = joinpath(env['BUILDDIR'], root[prefix_len:])
3294762Snate@binkert.org            SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
3304762Snate@binkert.org
3314762Snate@binkert.orgfor opt in export_vars:
3324762Snate@binkert.org    env.ConfigFile(opt)
3334762Snate@binkert.org
3345517Snate@binkert.orgdef makeTheISA(source, target, env):
3354762Snate@binkert.org    isas = [ src.get_contents() for src in source ]
3364762Snate@binkert.org    target_isa = env['TARGET_ISA']
3374762Snate@binkert.org    def define(isa):
3384762Snate@binkert.org        return isa.upper() + '_ISA'
3394382Sbinkertn@umich.edu    
3404382Sbinkertn@umich.edu    def namespace(isa):
3415517Snate@binkert.org        return isa[0].upper() + isa[1:].lower() + 'ISA' 
3425517Snate@binkert.org
3435517Snate@binkert.org
3445517Snate@binkert.org    code = code_formatter()
3455517Snate@binkert.org    code('''\
3465517Snate@binkert.org#ifndef __CONFIG_THE_ISA_HH__
3475517Snate@binkert.org#define __CONFIG_THE_ISA_HH__
3485517Snate@binkert.org
3495517Snate@binkert.org''')
3505517Snate@binkert.org
3515517Snate@binkert.org    for i,isa in enumerate(isas):
3525517Snate@binkert.org        code('#define $0 $1', define(isa), i + 1)
3535517Snate@binkert.org
3545517Snate@binkert.org    code('''
3555517Snate@binkert.org
3565517Snate@binkert.org#define THE_ISA ${{define(target_isa)}}
3575517Snate@binkert.org#define TheISA ${{namespace(target_isa)}}
3585517Snate@binkert.org
3595517Snate@binkert.org#endif // __CONFIG_THE_ISA_HH__''')
3605517Snate@binkert.org
3615517Snate@binkert.org    code.write(str(target[0]))
3625517Snate@binkert.org
3635517Snate@binkert.orgenv.Command('config/the_isa.hh', map(Value, all_isa_list),
3645517Snate@binkert.org            MakeAction(makeTheISA, Transform("CFG ISA", 0)))
3654762Snate@binkert.org
3665517Snate@binkert.org########################################################################
3674382Sbinkertn@umich.edu#
3684382Sbinkertn@umich.edu# Prevent any SimObjects from being added after this point, they
3694762Snate@binkert.org# should all have been added in the SConscripts above
3704382Sbinkertn@umich.edu#
3714382Sbinkertn@umich.eduSimObject.fixed = True
3725517Snate@binkert.org
3734382Sbinkertn@umich.educlass DictImporter(object):
3744382Sbinkertn@umich.edu    '''This importer takes a dictionary of arbitrary module names that
3754762Snate@binkert.org    map to arbitrary filenames.'''
3764382Sbinkertn@umich.edu    def __init__(self, modules):
3774762Snate@binkert.org        self.modules = modules
3785517Snate@binkert.org        self.installed = set()
3794382Sbinkertn@umich.edu
3804382Sbinkertn@umich.edu    def __del__(self):
3814762Snate@binkert.org        self.unload()
3824762Snate@binkert.org
3834762Snate@binkert.org    def unload(self):
3844762Snate@binkert.org        import sys
3854762Snate@binkert.org        for module in self.installed:
3865517Snate@binkert.org            del sys.modules[module]
3875517Snate@binkert.org        self.installed = set()
3885517Snate@binkert.org
3895517Snate@binkert.org    def find_module(self, fullname, path):
3905517Snate@binkert.org        if fullname == 'm5.defines':
3915517Snate@binkert.org            return self
3925517Snate@binkert.org
3935517Snate@binkert.org        if fullname == 'm5.objects':
3945517Snate@binkert.org            return self
3955517Snate@binkert.org
3965517Snate@binkert.org        if fullname.startswith('m5.internal'):
3975517Snate@binkert.org            return None
3985517Snate@binkert.org
3995517Snate@binkert.org        source = self.modules.get(fullname, None)
4005517Snate@binkert.org        if source is not None and fullname.startswith('m5.objects'):
4015517Snate@binkert.org            return self
4025517Snate@binkert.org
4035517Snate@binkert.org        return None
4045517Snate@binkert.org
4055517Snate@binkert.org    def load_module(self, fullname):
4065517Snate@binkert.org        mod = imp.new_module(fullname)
4075517Snate@binkert.org        sys.modules[fullname] = mod
4085517Snate@binkert.org        self.installed.add(fullname)
4095517Snate@binkert.org
4105517Snate@binkert.org        mod.__loader__ = self
4115517Snate@binkert.org        if fullname == 'm5.objects':
4125517Snate@binkert.org            mod.__path__ = fullname.split('.')
4135517Snate@binkert.org            return mod
4145517Snate@binkert.org
4155517Snate@binkert.org        if fullname == 'm5.defines':
4165517Snate@binkert.org            mod.__dict__['buildEnv'] = m5.util.SmartDict(build_env)
4175517Snate@binkert.org            return mod
4185517Snate@binkert.org
4195517Snate@binkert.org        source = self.modules[fullname]
4205517Snate@binkert.org        if source.modname == '__init__':
4215517Snate@binkert.org            mod.__path__ = source.modpath
4225517Snate@binkert.org        mod.__file__ = source.abspath
4235517Snate@binkert.org
4244762Snate@binkert.org        exec file(source.abspath, 'r') in mod.__dict__
4254762Snate@binkert.org
4265517Snate@binkert.org        return mod
4275517Snate@binkert.org
4284762Snate@binkert.orgimport m5.SimObject
4294762Snate@binkert.orgimport m5.params
4304762Snate@binkert.orgfrom m5.util import code_formatter
4315517Snate@binkert.org
4324762Snate@binkert.orgm5.SimObject.clear()
4334762Snate@binkert.orgm5.params.clear()
4344762Snate@binkert.org
4355463Snate@binkert.org# install the python importer so we can grab stuff from the source
4365517Snate@binkert.org# tree itself.  We can't have SimObjects added after this point or
4374762Snate@binkert.org# else we won't know about them for the rest of the stuff.
4384762Snate@binkert.orgimporter = DictImporter(PySource.modules)
4394762Snate@binkert.orgsys.meta_path[0:0] = [ importer ]
4404762Snate@binkert.org
4414762Snate@binkert.org# import all sim objects so we can populate the all_objects list
4424762Snate@binkert.org# make sure that we're working with a list, then let's sort it
4435463Snate@binkert.orgfor modname in SimObject.modnames:
4445517Snate@binkert.org    exec('from m5.objects import %s' % modname)
4454762Snate@binkert.org
4464762Snate@binkert.org# we need to unload all of the currently imported modules so that they
4474762Snate@binkert.org# will be re-imported the next time the sconscript is run
4485517Snate@binkert.orgimporter.unload()
4495517Snate@binkert.orgsys.meta_path.remove(importer)
4504762Snate@binkert.org
4514762Snate@binkert.orgsim_objects = m5.SimObject.allClasses
4525517Snate@binkert.orgall_enums = m5.params.allEnums
4534762Snate@binkert.org
4544762Snate@binkert.orgall_params = {}
4554762Snate@binkert.orgfor name,obj in sorted(sim_objects.iteritems()):
4564762Snate@binkert.org    for param in obj._params.local.values():
4575517Snate@binkert.org        # load the ptype attribute now because it depends on the
4584762Snate@binkert.org        # current version of SimObject.allClasses, but when scons
4594762Snate@binkert.org        # actually uses the value, all versions of
4604762Snate@binkert.org        # SimObject.allClasses will have been loaded
4614762Snate@binkert.org        param.ptype
4625517Snate@binkert.org
4635517Snate@binkert.org        if not hasattr(param, 'swig_decl'):
4645517Snate@binkert.org            continue
4655517Snate@binkert.org        pname = param.ptype_str
4665517Snate@binkert.org        if pname not in all_params:
4675517Snate@binkert.org            all_params[pname] = param
4685517Snate@binkert.org
4695517Snate@binkert.org########################################################################
4705517Snate@binkert.org#
4715517Snate@binkert.org# calculate extra dependencies
4725517Snate@binkert.org#
4735517Snate@binkert.orgmodule_depends = ["m5", "m5.SimObject", "m5.params"]
4745517Snate@binkert.orgdepends = [ PySource.modules[dep].snode for dep in module_depends ]
4755517Snate@binkert.org
4765517Snate@binkert.org########################################################################
4775517Snate@binkert.org#
4785517Snate@binkert.org# Commands for the basic automatically generated python files
4795517Snate@binkert.org#
4805517Snate@binkert.org
4815517Snate@binkert.org# Generate Python file containing a dict specifying the current
4825517Snate@binkert.org# buildEnv flags.
4835517Snate@binkert.orgdef makeDefinesPyFile(target, source, env):
4845517Snate@binkert.org    build_env = source[0].get_contents()
4855517Snate@binkert.org
4865517Snate@binkert.org    code = code_formatter()
4875517Snate@binkert.org    code("""
4885517Snate@binkert.orgimport m5.internal
4895517Snate@binkert.orgimport m5.util
4905517Snate@binkert.org
4915517Snate@binkert.orgbuildEnv = m5.util.SmartDict($build_env)
4925517Snate@binkert.org
4935517Snate@binkert.orgcompileDate = m5.internal.core.compileDate
4945517Snate@binkert.org_globals = globals()
4955517Snate@binkert.orgfor key,val in m5.internal.core.__dict__.iteritems():
4965517Snate@binkert.org    if key.startswith('flag_'):
4975517Snate@binkert.org        flag = key[5:]
4985517Snate@binkert.org        _globals[flag] = val
4995517Snate@binkert.orgdel _globals
5005517Snate@binkert.org""")
5015517Snate@binkert.org    code.write(target[0].abspath)
5025517Snate@binkert.org
5035517Snate@binkert.orgdefines_info = Value(build_env)
5045517Snate@binkert.org# Generate a file with all of the compile options in it
5055517Snate@binkert.orgenv.Command('python/m5/defines.py', defines_info,
5065517Snate@binkert.org            MakeAction(makeDefinesPyFile, Transform("DEFINES", 0)))
5075517Snate@binkert.orgPySource('m5', 'python/m5/defines.py')
5085517Snate@binkert.org
5095517Snate@binkert.org# Generate python file containing info about the M5 source code
5105517Snate@binkert.orgdef makeInfoPyFile(target, source, env):
5115517Snate@binkert.org    code = code_formatter()
5125517Snate@binkert.org    for src in source:
5135517Snate@binkert.org        data = ''.join(file(src.srcnode().abspath, 'r').xreadlines())
5145517Snate@binkert.org        code('$src = ${{repr(data)}}')
5155517Snate@binkert.org    code.write(str(target[0]))
5165517Snate@binkert.org
5175517Snate@binkert.org# Generate a file that wraps the basic top level files
5185517Snate@binkert.orgenv.Command('python/m5/info.py',
5195517Snate@binkert.org            [ '#/COPYING', '#/LICENSE', '#/README', ],
5205517Snate@binkert.org            MakeAction(makeInfoPyFile, Transform("INFO")))
5215517Snate@binkert.orgPySource('m5', 'python/m5/info.py')
5225517Snate@binkert.org
5235517Snate@binkert.org########################################################################
5245517Snate@binkert.org#
5255517Snate@binkert.org# Create all of the SimObject param headers and enum headers
5265517Snate@binkert.org#
5275517Snate@binkert.org
5285517Snate@binkert.orgdef createSimObjectParam(target, source, env):
5295517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5305517Snate@binkert.org
5315517Snate@binkert.org    name = str(source[0].get_contents())
5325517Snate@binkert.org    obj = sim_objects[name]
5335517Snate@binkert.org
5345517Snate@binkert.org    code = code_formatter()
5355517Snate@binkert.org    obj.cxx_decl(code)
5365517Snate@binkert.org    code.write(target[0].abspath)
5375517Snate@binkert.org
5385517Snate@binkert.orgdef createSwigParam(target, source, env):
5395517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5405517Snate@binkert.org
5415517Snate@binkert.org    name = str(source[0].get_contents())
5425517Snate@binkert.org    param = all_params[name]
5435517Snate@binkert.org
5445517Snate@binkert.org    code = code_formatter()
5455517Snate@binkert.org    code('%module(package="m5.internal") $0_${name}', param.file_ext)
5465517Snate@binkert.org    param.swig_decl(code)
5475517Snate@binkert.org    code.write(target[0].abspath)
5485517Snate@binkert.org
5495517Snate@binkert.orgdef createEnumStrings(target, source, env):
5505517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5515517Snate@binkert.org
5524762Snate@binkert.org    name = str(source[0].get_contents())
5535517Snate@binkert.org    obj = all_enums[name]
5545517Snate@binkert.org
5555463Snate@binkert.org    code = code_formatter()
5564762Snate@binkert.org    obj.cxx_def(code)
5574762Snate@binkert.org    code.write(target[0].abspath)
5584762Snate@binkert.org
5594382Sbinkertn@umich.edudef createEnumParam(target, source, env):
5605554Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5614762Snate@binkert.org
5624382Sbinkertn@umich.edu    name = str(source[0].get_contents())
5634762Snate@binkert.org    obj = all_enums[name]
5644382Sbinkertn@umich.edu
5654762Snate@binkert.org    code = code_formatter()
5664762Snate@binkert.org    obj.cxx_decl(code)
5674762Snate@binkert.org    code.write(target[0].abspath)
5684762Snate@binkert.org
5694382Sbinkertn@umich.edudef createEnumSwig(target, source, env):
5704382Sbinkertn@umich.edu    assert len(target) == 1 and len(source) == 1
5714382Sbinkertn@umich.edu
5724382Sbinkertn@umich.edu    name = str(source[0].get_contents())
5734382Sbinkertn@umich.edu    obj = all_enums[name]
5744382Sbinkertn@umich.edu
5754762Snate@binkert.org    code = code_formatter()
5764382Sbinkertn@umich.edu    code('''\
5775554Snate@binkert.org%module(package="m5.internal") enum_$name
5784382Sbinkertn@umich.edu
5794382Sbinkertn@umich.edu%{
5804762Snate@binkert.org#include "enums/$name.hh"
5815517Snate@binkert.org%}
5825517Snate@binkert.org
5835517Snate@binkert.org%include "enums/$name.hh"
5845517Snate@binkert.org''')
5855517Snate@binkert.org    code.write(target[0].abspath)
5865517Snate@binkert.org
5875522Snate@binkert.org# Generate all of the SimObject param struct header files
5885517Snate@binkert.orgparams_hh_files = []
5895517Snate@binkert.orgfor name,simobj in sorted(sim_objects.iteritems()):
5905517Snate@binkert.org    py_source = PySource.modules[simobj.__module__]
5915517Snate@binkert.org    extra_deps = [ py_source.tnode ]
5925517Snate@binkert.org
5935522Snate@binkert.org    hh_file = File('params/%s.hh' % name)
5945522Snate@binkert.org    params_hh_files.append(hh_file)
5954382Sbinkertn@umich.edu    env.Command(hh_file, Value(name),
5965192Ssaidi@eecs.umich.edu                MakeAction(createSimObjectParam, Transform("SO PARAM")))
5975517Snate@binkert.org    env.Depends(hh_file, depends + extra_deps)
5985517Snate@binkert.org
5995517Snate@binkert.org# Generate any parameter header files needed
6005517Snate@binkert.orgparams_i_files = []
6015517Snate@binkert.orgfor name,param in all_params.iteritems():
6025517Snate@binkert.org    i_file = File('python/m5/internal/%s_%s.i' % (param.file_ext, name))
6035517Snate@binkert.org    params_i_files.append(i_file)
6045517Snate@binkert.org    env.Command(i_file, Value(name),
6055517Snate@binkert.org                MakeAction(createSwigParam, Transform("SW PARAM")))
6065517Snate@binkert.org    env.Depends(i_file, depends)
6075517Snate@binkert.org    SwigSource('m5.internal', i_file)
6085517Snate@binkert.org
6095517Snate@binkert.org# Generate all enum header files
6105517Snate@binkert.orgfor name,enum in sorted(all_enums.iteritems()):
6115517Snate@binkert.org    py_source = PySource.modules[enum.__module__]
6125517Snate@binkert.org    extra_deps = [ py_source.tnode ]
6135517Snate@binkert.org
6145517Snate@binkert.org    cc_file = File('enums/%s.cc' % name)
6155517Snate@binkert.org    env.Command(cc_file, Value(name),
6165517Snate@binkert.org                MakeAction(createEnumStrings, Transform("ENUM STR")))
6175517Snate@binkert.org    env.Depends(cc_file, depends + extra_deps)
6185517Snate@binkert.org    Source(cc_file)
6195517Snate@binkert.org
6205517Snate@binkert.org    hh_file = File('enums/%s.hh' % name)
6215517Snate@binkert.org    env.Command(hh_file, Value(name),
6225517Snate@binkert.org                MakeAction(createEnumParam, Transform("EN PARAM")))
6235517Snate@binkert.org    env.Depends(hh_file, depends + extra_deps)
6245517Snate@binkert.org
6255517Snate@binkert.org    i_file = File('python/m5/internal/enum_%s.i' % name)
6265517Snate@binkert.org    env.Command(i_file, Value(name),
6275517Snate@binkert.org                MakeAction(createEnumSwig, Transform("ENUMSWIG")))
6285517Snate@binkert.org    env.Depends(i_file, depends + extra_deps)
6295517Snate@binkert.org    SwigSource('m5.internal', i_file)
6305517Snate@binkert.org
6315517Snate@binkert.orgdef buildParam(target, source, env):
6325517Snate@binkert.org    name = source[0].get_contents()
6335517Snate@binkert.org    obj = sim_objects[name]
6345517Snate@binkert.org    class_path = obj.cxx_class.split('::')
6355517Snate@binkert.org    classname = class_path[-1]
6365517Snate@binkert.org    namespaces = class_path[:-1]
6375517Snate@binkert.org    params = obj._params.local.values()
6385517Snate@binkert.org
6395517Snate@binkert.org    code = code_formatter()
6405517Snate@binkert.org
6415517Snate@binkert.org    code('%module(package="m5.internal") param_$name')
6425517Snate@binkert.org    code()
6435517Snate@binkert.org    code('%{')
6445517Snate@binkert.org    code('#include "params/$obj.hh"')
6455517Snate@binkert.org    for param in params:
6465517Snate@binkert.org        param.cxx_predecls(code)
6475517Snate@binkert.org    code('%}')
6485517Snate@binkert.org    code()
6495517Snate@binkert.org
6505517Snate@binkert.org    for param in params:
6515517Snate@binkert.org        param.swig_predecls(code)
6525517Snate@binkert.org
6535517Snate@binkert.org    code()
6545517Snate@binkert.org    if obj._base:
6555517Snate@binkert.org        code('%import "python/m5/internal/param_${{obj._base}}.i"')
6565517Snate@binkert.org    code()
6575517Snate@binkert.org    obj.swig_objdecls(code)
6585517Snate@binkert.org    code()
6595517Snate@binkert.org
6605517Snate@binkert.org    code('%include "params/$obj.hh"')
6615517Snate@binkert.org
6625517Snate@binkert.org    code.write(target[0].abspath)
6635517Snate@binkert.org
6645517Snate@binkert.orgfor name in sim_objects.iterkeys():
6655517Snate@binkert.org    params_file = File('python/m5/internal/param_%s.i' % name)
6665517Snate@binkert.org    env.Command(params_file, Value(name),
6675517Snate@binkert.org                MakeAction(buildParam, Transform("BLDPARAM")))
6685517Snate@binkert.org    env.Depends(params_file, depends)
6695517Snate@binkert.org    SwigSource('m5.internal', params_file)
6705517Snate@binkert.org
6715517Snate@binkert.org# Generate the main swig init file
6725517Snate@binkert.orgdef makeEmbeddedSwigInit(target, source, env):
6735517Snate@binkert.org    code = code_formatter()
6745517Snate@binkert.org    module = source[0].get_contents()
6755517Snate@binkert.org    code('''\
6765517Snate@binkert.org#include "sim/init.hh"
6775517Snate@binkert.org
6785517Snate@binkert.orgextern "C" {
6795517Snate@binkert.org    void init_${module}();
6805517Snate@binkert.org}
6815517Snate@binkert.org
6825517Snate@binkert.orgEmbeddedSwig embed_swig_${module}(init_${module});
6835517Snate@binkert.org''')
6845517Snate@binkert.org    code.write(str(target[0]))
6855517Snate@binkert.org    
6865517Snate@binkert.org# Build all swig modules
6875517Snate@binkert.orgfor swig in SwigSource.all:
6885517Snate@binkert.org    env.Command([swig.cc_source.tnode, swig.py_source.tnode], swig.tnode,
6895517Snate@binkert.org                MakeAction('$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} '
6905517Snate@binkert.org                '-o ${TARGETS[0]} $SOURCES', Transform("SWIG")))
6915517Snate@binkert.org    cc_file = str(swig.tnode)
6925517Snate@binkert.org    init_file = '%s/init_%s.cc' % (dirname(cc_file), basename(cc_file))
6935517Snate@binkert.org    env.Command(init_file, Value(swig.module),
6945517Snate@binkert.org                MakeAction(makeEmbeddedSwigInit, Transform("EMBED SW")))
6955517Snate@binkert.org    Source(init_file, **swig.guards)
6965517Snate@binkert.org
6975517Snate@binkert.org#
6985517Snate@binkert.org# Handle debug flags
6995517Snate@binkert.org#
7005517Snate@binkert.orgdef makeDebugFlagCC(target, source, env):
7015517Snate@binkert.org    assert(len(target) == 1 and len(source) == 1)
7025517Snate@binkert.org
7035517Snate@binkert.org    val = eval(source[0].get_contents())
7045517Snate@binkert.org    name, compound, desc = val
7055517Snate@binkert.org    compound = list(sorted(compound))
7065517Snate@binkert.org
7075517Snate@binkert.org    code = code_formatter()
7085517Snate@binkert.org
7095517Snate@binkert.org    # file header
7105517Snate@binkert.org    code('''
7115517Snate@binkert.org/*
7125517Snate@binkert.org * DO NOT EDIT THIS FILE! Automatically generated
7135517Snate@binkert.org */
7145517Snate@binkert.org
7155517Snate@binkert.org#include "base/debug.hh"
7165517Snate@binkert.org''')
7175517Snate@binkert.org
7185517Snate@binkert.org    for flag in compound:
7195517Snate@binkert.org        code('#include "debug/$flag.hh"')
7205517Snate@binkert.org    code()
7215517Snate@binkert.org    code('namespace Debug {')
7225517Snate@binkert.org    code()
7235517Snate@binkert.org
7245517Snate@binkert.org    if not compound:
7255517Snate@binkert.org        code('SimpleFlag $name("$name", "$desc");')
7265517Snate@binkert.org    else:
7275517Snate@binkert.org        code('CompoundFlag $name("$name", "$desc",')
7285517Snate@binkert.org        code.indent()
7295517Snate@binkert.org        last = len(compound) - 1
7305517Snate@binkert.org        for i,flag in enumerate(compound):
7315517Snate@binkert.org            if i != last:
7325517Snate@binkert.org                code('$flag,')
7335517Snate@binkert.org            else:
7345517Snate@binkert.org                code('$flag);')
7355517Snate@binkert.org        code.dedent()
7365517Snate@binkert.org
7375517Snate@binkert.org    code()
7385517Snate@binkert.org    code('} // namespace Debug')
7395517Snate@binkert.org
7405517Snate@binkert.org    code.write(str(target[0]))
7415517Snate@binkert.org
7425517Snate@binkert.orgdef makeDebugFlagHH(target, source, env):
7435517Snate@binkert.org    assert(len(target) == 1 and len(source) == 1)
7445517Snate@binkert.org
7455517Snate@binkert.org    val = eval(source[0].get_contents())
7465517Snate@binkert.org    name, compound, desc = val
7475517Snate@binkert.org
7485517Snate@binkert.org    code = code_formatter()
7495517Snate@binkert.org
7505517Snate@binkert.org    # file header boilerplate
7515517Snate@binkert.org    code('''\
7525517Snate@binkert.org/*
7535517Snate@binkert.org * DO NOT EDIT THIS FILE!
7545517Snate@binkert.org *
7555517Snate@binkert.org * Automatically generated by SCons
7565517Snate@binkert.org */
7575517Snate@binkert.org
7585517Snate@binkert.org#ifndef __DEBUG_${name}_HH__
7595517Snate@binkert.org#define __DEBUG_${name}_HH__
7605517Snate@binkert.org
7615517Snate@binkert.orgnamespace Debug {
7625517Snate@binkert.org''')
7635517Snate@binkert.org
7645517Snate@binkert.org    if compound:
7655517Snate@binkert.org        code('class CompoundFlag;')
7665517Snate@binkert.org    code('class SimpleFlag;')
7675517Snate@binkert.org
7685517Snate@binkert.org    if compound:
7695517Snate@binkert.org        code('extern CompoundFlag $name;')
7705517Snate@binkert.org        for flag in compound:
7715517Snate@binkert.org            code('extern SimpleFlag $flag;')
7725517Snate@binkert.org    else:
7735517Snate@binkert.org        code('extern SimpleFlag $name;')
7745517Snate@binkert.org
7755517Snate@binkert.org    code('''
7765517Snate@binkert.org}
7775517Snate@binkert.org
7785517Snate@binkert.org#endif // __DEBUG_${name}_HH__
7795517Snate@binkert.org''')
7805517Snate@binkert.org
7815517Snate@binkert.org    code.write(str(target[0]))
7825517Snate@binkert.org
7835517Snate@binkert.orgfor name,flag in sorted(debug_flags.iteritems()):
7845517Snate@binkert.org    n, compound, desc = flag
7855517Snate@binkert.org    assert n == name
7865517Snate@binkert.org
7875517Snate@binkert.org    env.Command('debug/%s.hh' % name, Value(flag),
7885192Ssaidi@eecs.umich.edu                MakeAction(makeDebugFlagHH, Transform("TRACING", 0)))
7895517Snate@binkert.org    env.Command('debug/%s.cc' % name, Value(flag),
7905192Ssaidi@eecs.umich.edu                MakeAction(makeDebugFlagCC, Transform("TRACING", 0)))
7915192Ssaidi@eecs.umich.edu    Source('debug/%s.cc' % name)
7925517Snate@binkert.org
7935517Snate@binkert.org# Embed python files.  All .py files that have been indicated by a
7945192Ssaidi@eecs.umich.edu# PySource() call in a SConscript need to be embedded into the M5
7955192Ssaidi@eecs.umich.edu# library.  To do that, we compile the file to byte code, marshal the
7965456Ssaidi@eecs.umich.edu# byte code, compress it, and then generate a c++ file that
7975517Snate@binkert.org# inserts the result into an array.
7985517Snate@binkert.orgdef embedPyFile(target, source, env):
7995517Snate@binkert.org    def c_str(string):
8005517Snate@binkert.org        if string is None:
8015517Snate@binkert.org            return "0"
8025517Snate@binkert.org        return '"%s"' % string
8035517Snate@binkert.org
8045517Snate@binkert.org    '''Action function to compile a .py into a code object, marshal
8055517Snate@binkert.org    it, compress it, and stick it into an asm file so the code appears
8065517Snate@binkert.org    as just bytes with a label in the data section'''
8075517Snate@binkert.org
8085517Snate@binkert.org    src = file(str(source[0]), 'r').read()
8095517Snate@binkert.org
8105517Snate@binkert.org    pysource = PySource.tnodes[source[0]]
8115517Snate@binkert.org    compiled = compile(src, pysource.abspath, 'exec')
8125517Snate@binkert.org    marshalled = marshal.dumps(compiled)
8135517Snate@binkert.org    compressed = zlib.compress(marshalled)
8145517Snate@binkert.org    data = compressed
8155517Snate@binkert.org    sym = pysource.symname
8165517Snate@binkert.org
8175517Snate@binkert.org    code = code_formatter()
8185517Snate@binkert.org    code('''\
8195517Snate@binkert.org#include "sim/init.hh"
8205517Snate@binkert.org
8215517Snate@binkert.orgnamespace {
8225517Snate@binkert.org
8235517Snate@binkert.orgconst char data_${sym}[] = {
8245517Snate@binkert.org''')
8255517Snate@binkert.org    code.indent()
8265517Snate@binkert.org    step = 16
8275517Snate@binkert.org    for i in xrange(0, len(data), step):
8285517Snate@binkert.org        x = array.array('B', data[i:i+step])
8295456Ssaidi@eecs.umich.edu        code(''.join('%d,' % d for d in x))
8305461Snate@binkert.org    code.dedent()
8315517Snate@binkert.org    
8325456Ssaidi@eecs.umich.edu    code('''};
8335522Snate@binkert.org
8345522Snate@binkert.orgEmbeddedPython embedded_${sym}(
8355522Snate@binkert.org    ${{c_str(pysource.arcname)}},
8365522Snate@binkert.org    ${{c_str(pysource.abspath)}},
8375522Snate@binkert.org    ${{c_str(pysource.modpath)}},
8385522Snate@binkert.org    data_${sym},
8395522Snate@binkert.org    ${{len(data)}},
8405522Snate@binkert.org    ${{len(marshalled)}});
8415522Snate@binkert.org
8425517Snate@binkert.org} // anonymous namespace
8435522Snate@binkert.org''')
8445522Snate@binkert.org    code.write(str(target[0]))
8455522Snate@binkert.org
8465522Snate@binkert.orgfor source in PySource.all:
8475517Snate@binkert.org    env.Command(source.cpp, source.tnode, 
8485522Snate@binkert.org                MakeAction(embedPyFile, Transform("EMBED PY")))
8495522Snate@binkert.org    Source(source.cpp)
8505517Snate@binkert.org
8515522Snate@binkert.org########################################################################
8525522Snate@binkert.org#
8535522Snate@binkert.org# Define binaries.  Each different build type (debug, opt, etc.) gets
8545522Snate@binkert.org# a slightly different build environment.
8555522Snate@binkert.org#
8565517Snate@binkert.org
8575522Snate@binkert.org# List of constructed environments to pass back to SConstruct
8585522Snate@binkert.orgenvList = []
8595522Snate@binkert.org
8605522Snate@binkert.orgdate_source = Source('base/date.cc', skip_lib=True)
8615522Snate@binkert.org
8625522Snate@binkert.org# Function to create a new build environment as clone of current
8635522Snate@binkert.org# environment 'env' with modified object suffix and optional stripped
8645522Snate@binkert.org# binary.  Additional keyword arguments are appended to corresponding
8655522Snate@binkert.org# build environment vars.
8665522Snate@binkert.orgdef makeEnv(label, objsfx, strip = False, **kwargs):
8675522Snate@binkert.org    # SCons doesn't know to append a library suffix when there is a '.' in the
8685522Snate@binkert.org    # name.  Use '_' instead.
8695522Snate@binkert.org    libname = 'gem5_' + label
8705522Snate@binkert.org    exename = 'gem5.' + label
8715522Snate@binkert.org    secondary_exename = 'm5.' + label
8725522Snate@binkert.org
8735522Snate@binkert.org    new_env = env.Clone(OBJSUFFIX=objsfx, SHOBJSUFFIX=objsfx + 's')
8745522Snate@binkert.org    new_env.Label = label
8755522Snate@binkert.org    new_env.Append(**kwargs)
8764382Sbinkertn@umich.edu
8775522Snate@binkert.org    swig_env = new_env.Clone()
8785522Snate@binkert.org    swig_env.Append(CCFLAGS='-Werror')
8794382Sbinkertn@umich.edu    if env['GCC']:
8805522Snate@binkert.org        swig_env.Append(CCFLAGS='-Wno-uninitialized')
8815522Snate@binkert.org        swig_env.Append(CCFLAGS='-Wno-sign-compare')
8825522Snate@binkert.org        swig_env.Append(CCFLAGS='-Wno-parentheses')
8835522Snate@binkert.org
8845522Snate@binkert.org    werror_env = new_env.Clone()
8855522Snate@binkert.org    werror_env.Append(CCFLAGS='-Werror')
8865522Snate@binkert.org
8875522Snate@binkert.org    def make_obj(source, static, extra_deps = None):
8885522Snate@binkert.org        '''This function adds the specified source to the correct
8895522Snate@binkert.org        build environment, and returns the corresponding SCons Object
8904382Sbinkertn@umich.edu        nodes'''
8915522Snate@binkert.org
8925522Snate@binkert.org        if source.swig:
8935522Snate@binkert.org            env = swig_env
8945522Snate@binkert.org        elif source.Werror:
8955522Snate@binkert.org            env = werror_env
8965522Snate@binkert.org        else:
8975522Snate@binkert.org            env = new_env
8985522Snate@binkert.org
8995522Snate@binkert.org        if static:
9005522Snate@binkert.org            obj = env.StaticObject(source.tnode)
9015522Snate@binkert.org        else:
9025522Snate@binkert.org            obj = env.SharedObject(source.tnode)
9035522Snate@binkert.org
9045522Snate@binkert.org        if extra_deps:
9055522Snate@binkert.org            env.Depends(obj, extra_deps)
9065522Snate@binkert.org
9075522Snate@binkert.org        return obj
9085522Snate@binkert.org
9095522Snate@binkert.org    sources = Source.get(main=False, skip_lib=False)
9105522Snate@binkert.org    static_objs = [ make_obj(s, True) for s in sources ]
9115522Snate@binkert.org    shared_objs = [ make_obj(s, False) for s in sources ]
9125522Snate@binkert.org
9135522Snate@binkert.org    static_date = make_obj(date_source, static=True, extra_deps=static_objs)
9145522Snate@binkert.org    static_objs.append(static_date)
9155522Snate@binkert.org    
9165522Snate@binkert.org    shared_date = make_obj(date_source, static=False, extra_deps=shared_objs)
9175522Snate@binkert.org    shared_objs.append(shared_date)
9185522Snate@binkert.org
9195522Snate@binkert.org    # First make a library of everything but main() so other programs can
9205522Snate@binkert.org    # link against m5.
9215522Snate@binkert.org    static_lib = new_env.StaticLibrary(libname, static_objs)
9224382Sbinkertn@umich.edu    shared_lib = new_env.SharedLibrary(libname, shared_objs)
9234382Sbinkertn@umich.edu
9244382Sbinkertn@umich.edu    # Now link a stub with main() and the static library.
9254382Sbinkertn@umich.edu    main_objs = [ make_obj(s, True) for s in Source.get(main=True) ]
9264382Sbinkertn@umich.edu
9274382Sbinkertn@umich.edu    for test in UnitTest.all:
9284382Sbinkertn@umich.edu        flags = { test.target : True }
9294382Sbinkertn@umich.edu        test_sources = Source.get(**flags)
9304382Sbinkertn@umich.edu        test_objs = [ make_obj(s, static=True) for s in test_sources ]
9314382Sbinkertn@umich.edu        testname = "unittest/%s.%s" % (test.target, label)
932955SN/A        new_env.Program(testname, main_objs + test_objs + static_objs)
933955SN/A
934955SN/A    progname = exename
935955SN/A    if strip:
9361108SN/A        progname += '.unstripped'
937955SN/A
938955SN/A    targets = new_env.Program(progname, main_objs + static_objs)
9395456Ssaidi@eecs.umich.edu
940955SN/A    if strip:
941955SN/A        if sys.platform == 'sunos5':
942955SN/A            cmd = 'cp $SOURCE $TARGET; strip $TARGET'
9435456Ssaidi@eecs.umich.edu        else:
9445456Ssaidi@eecs.umich.edu            cmd = 'strip $SOURCE -o $TARGET'
9455456Ssaidi@eecs.umich.edu        targets = new_env.Command(exename, progname,
9465456Ssaidi@eecs.umich.edu                    MakeAction(cmd, Transform("STRIP")))
9475456Ssaidi@eecs.umich.edu
9485456Ssaidi@eecs.umich.edu    new_env.Command(secondary_exename, exename,
949955SN/A            MakeAction('ln $SOURCE $TARGET', Transform("HARDLINK")))
9505456Ssaidi@eecs.umich.edu
9515456Ssaidi@eecs.umich.edu    new_env.M5Binary = targets[0]
952955SN/A    envList.append(new_env)
953955SN/A
9542655Sstever@eecs.umich.edu# Debug binary
9552655Sstever@eecs.umich.educcflags = {}
9562655Sstever@eecs.umich.eduif env['GCC']:
9572655Sstever@eecs.umich.edu    if sys.platform == 'sunos5':
9582655Sstever@eecs.umich.edu        ccflags['debug'] = '-gstabs+'
9592655Sstever@eecs.umich.edu    else:
9602655Sstever@eecs.umich.edu        ccflags['debug'] = '-ggdb3'
9612655Sstever@eecs.umich.edu    ccflags['opt'] = '-g -O3'
9625522Snate@binkert.org    ccflags['fast'] = '-O3'
9635554Snate@binkert.org    ccflags['prof'] = '-O3 -g -pg'
9645559Snate@binkert.orgelif env['SUNCC']:
9655559Snate@binkert.org    ccflags['debug'] = '-g0'
9665559Snate@binkert.org    ccflags['opt'] = '-g -O'
9675559Snate@binkert.org    ccflags['fast'] = '-fast'
9685554Snate@binkert.org    ccflags['prof'] = '-fast -g -pg'
9695554Snate@binkert.orgelif env['ICC']:
9705522Snate@binkert.org    ccflags['debug'] = '-g -O0'
9715522Snate@binkert.org    ccflags['opt'] = '-g -O'
9725522Snate@binkert.org    ccflags['fast'] = '-fast'
9735522Snate@binkert.org    ccflags['prof'] = '-fast -g -pg'
9745522Snate@binkert.orgelse:
9755554Snate@binkert.org    print 'Unknown compiler, please fix compiler options'
9765554Snate@binkert.org    Exit(1)
9775554Snate@binkert.org
9785522Snate@binkert.orgmakeEnv('debug', '.do',
9795584Snate@binkert.org        CCFLAGS = Split(ccflags['debug']),
9805584Snate@binkert.org        CPPDEFINES = ['DEBUG', 'TRACING_ON=1'])
9815584Snate@binkert.org
9825584Snate@binkert.org# Optimized binary
9835522Snate@binkert.orgmakeEnv('opt', '.o',
9842655Sstever@eecs.umich.edu        CCFLAGS = Split(ccflags['opt']),
9855522Snate@binkert.org        CPPDEFINES = ['TRACING_ON=1'])
9862655Sstever@eecs.umich.edu
9875522Snate@binkert.org# "Fast" binary
9885522Snate@binkert.orgmakeEnv('fast', '.fo', strip = True,
9894007Ssaidi@eecs.umich.edu        CCFLAGS = Split(ccflags['fast']),
9904596Sbinkertn@umich.edu        CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'])
9914007Ssaidi@eecs.umich.edu
9924596Sbinkertn@umich.edu# Profiled binary
9935522Snate@binkert.orgmakeEnv('prof', '.po',
9945522Snate@binkert.org        CCFLAGS = Split(ccflags['prof']),
9955522Snate@binkert.org        CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'],
9965522Snate@binkert.org        LINKFLAGS = '-pg')
9972655Sstever@eecs.umich.edu
9982655Sstever@eecs.umich.eduReturn('envList')
9992655Sstever@eecs.umich.edu