SConscript revision 9338
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
326143Snate@binkert.orgimport bisect
334762Snate@binkert.orgimport imp
345522Snate@binkert.orgimport marshal
35955SN/Aimport os
365522Snate@binkert.orgimport re
37955SN/Aimport sys
385522Snate@binkert.orgimport zlib
394202Sbinkertn@umich.edu
405742Snate@binkert.orgfrom os.path import basename, dirname, exists, isdir, isfile, join as joinpath
41955SN/A
424381Sbinkertn@umich.eduimport SCons
434381Sbinkertn@umich.edu
448334Snate@binkert.org# This file defines how to build a particular configuration of gem5
45955SN/A# based on variable settings in the 'env' build environment.
46955SN/A
474202Sbinkertn@umich.eduImport('*')
48955SN/A
494382Sbinkertn@umich.edu# Children need to see the environment
504382Sbinkertn@umich.eduExport('env')
514382Sbinkertn@umich.edu
526654Snate@binkert.orgbuild_env = [(opt, env[opt]) for opt in export_vars]
535517Snate@binkert.org
548614Sgblack@eecs.umich.edufrom m5.util import code_formatter, compareVersions
557674Snate@binkert.org
566143Snate@binkert.org########################################################################
576143Snate@binkert.org# Code for adding source files of various types
586143Snate@binkert.org#
598233Snate@binkert.org# When specifying a source file of some type, a set of guards can be
608233Snate@binkert.org# specified for that file.  When get() is used to find the files, if
618233Snate@binkert.org# get specifies a set of filters, only files that match those filters
628233Snate@binkert.org# will be accepted (unspecified filters on files are assumed to be
638233Snate@binkert.org# false).  Current filters are:
648334Snate@binkert.org#     main -- specifies the gem5 main() function
658334Snate@binkert.org#     skip_lib -- do not put this file into the gem5 library
668233Snate@binkert.org#     <unittest> -- unit tests use filters based on the unit test name
678233Snate@binkert.org#
688233Snate@binkert.org# A parent can now be specified for a source file and default filter
698233Snate@binkert.org# values will be retrieved recursively from parents (children override
708233Snate@binkert.org# parents).
718233Snate@binkert.org#
726143Snate@binkert.orgclass SourceMeta(type):
738233Snate@binkert.org    '''Meta class for source files that keeps track of all files of a
748233Snate@binkert.org    particular type and has a get function for finding all functions
758233Snate@binkert.org    of a certain type that match a set of guards'''
766143Snate@binkert.org    def __init__(cls, name, bases, dict):
776143Snate@binkert.org        super(SourceMeta, cls).__init__(name, bases, dict)
786143Snate@binkert.org        cls.all = []
796143Snate@binkert.org        
808233Snate@binkert.org    def get(cls, **guards):
818233Snate@binkert.org        '''Find all files that match the specified guards.  If a source
828233Snate@binkert.org        file does not specify a flag, the default is False'''
836143Snate@binkert.org        for src in cls.all:
848233Snate@binkert.org            for flag,value in guards.iteritems():
858233Snate@binkert.org                # if the flag is found and has a different value, skip
868233Snate@binkert.org                # this file
878233Snate@binkert.org                if src.all_guards.get(flag, False) != value:
886143Snate@binkert.org                    break
896143Snate@binkert.org            else:
906143Snate@binkert.org                yield src
914762Snate@binkert.org
926143Snate@binkert.orgclass SourceFile(object):
938233Snate@binkert.org    '''Base object that encapsulates the notion of a source file.
948233Snate@binkert.org    This includes, the source node, target node, various manipulations
958233Snate@binkert.org    of those.  A source file also specifies a set of guards which
968233Snate@binkert.org    describing which builds the source file applies to.  A parent can
978233Snate@binkert.org    also be specified to get default guards from'''
986143Snate@binkert.org    __metaclass__ = SourceMeta
998233Snate@binkert.org    def __init__(self, source, parent=None, **guards):
1008233Snate@binkert.org        self.guards = guards
1018233Snate@binkert.org        self.parent = parent
1028233Snate@binkert.org
1036143Snate@binkert.org        tnode = source
1046143Snate@binkert.org        if not isinstance(source, SCons.Node.FS.File):
1056143Snate@binkert.org            tnode = File(source)
1066143Snate@binkert.org
1076143Snate@binkert.org        self.tnode = tnode
1086143Snate@binkert.org        self.snode = tnode.srcnode()
1096143Snate@binkert.org
1106143Snate@binkert.org        for base in type(self).__mro__:
1116143Snate@binkert.org            if issubclass(base, SourceFile):
1127065Snate@binkert.org                base.all.append(self)
1136143Snate@binkert.org
1148233Snate@binkert.org    @property
1158233Snate@binkert.org    def filename(self):
1168233Snate@binkert.org        return str(self.tnode)
1178233Snate@binkert.org
1188233Snate@binkert.org    @property
1198233Snate@binkert.org    def dirname(self):
1208233Snate@binkert.org        return dirname(self.filename)
1218233Snate@binkert.org
1228233Snate@binkert.org    @property
1238233Snate@binkert.org    def basename(self):
1248233Snate@binkert.org        return basename(self.filename)
1258233Snate@binkert.org
1268233Snate@binkert.org    @property
1278233Snate@binkert.org    def extname(self):
1288233Snate@binkert.org        index = self.basename.rfind('.')
1298233Snate@binkert.org        if index <= 0:
1308233Snate@binkert.org            # dot files aren't extensions
1318233Snate@binkert.org            return self.basename, None
1328233Snate@binkert.org
1338233Snate@binkert.org        return self.basename[:index], self.basename[index+1:]
1348233Snate@binkert.org
1358233Snate@binkert.org    @property
1368233Snate@binkert.org    def all_guards(self):
1378233Snate@binkert.org        '''find all guards for this object getting default values
1388233Snate@binkert.org        recursively from its parents'''
1398233Snate@binkert.org        guards = {}
1408233Snate@binkert.org        if self.parent:
1418233Snate@binkert.org            guards.update(self.parent.guards)
1428233Snate@binkert.org        guards.update(self.guards)
1438233Snate@binkert.org        return guards
1448233Snate@binkert.org
1456143Snate@binkert.org    def __lt__(self, other): return self.filename < other.filename
1466143Snate@binkert.org    def __le__(self, other): return self.filename <= other.filename
1476143Snate@binkert.org    def __gt__(self, other): return self.filename > other.filename
1486143Snate@binkert.org    def __ge__(self, other): return self.filename >= other.filename
1496143Snate@binkert.org    def __eq__(self, other): return self.filename == other.filename
1506143Snate@binkert.org    def __ne__(self, other): return self.filename != other.filename
1516143Snate@binkert.org        
1526143Snate@binkert.orgclass Source(SourceFile):
1536143Snate@binkert.org    '''Add a c/c++ source file to the build'''
1548233Snate@binkert.org    def __init__(self, source, Werror=True, swig=False, **guards):
1558233Snate@binkert.org        '''specify the source file, and any guards'''
1568233Snate@binkert.org        super(Source, self).__init__(source, **guards)
1576143Snate@binkert.org
1586143Snate@binkert.org        self.Werror = Werror
1596143Snate@binkert.org        self.swig = swig
1606143Snate@binkert.org
1616143Snate@binkert.orgclass PySource(SourceFile):
1626143Snate@binkert.org    '''Add a python source file to the named package'''
1635522Snate@binkert.org    invalid_sym_char = re.compile('[^A-z0-9_]')
1646143Snate@binkert.org    modules = {}
1656143Snate@binkert.org    tnodes = {}
1666143Snate@binkert.org    symnames = {}
1676143Snate@binkert.org    
1688233Snate@binkert.org    def __init__(self, package, source, **guards):
1698233Snate@binkert.org        '''specify the python package, the source file, and any guards'''
1708233Snate@binkert.org        super(PySource, self).__init__(source, **guards)
1716143Snate@binkert.org
1726143Snate@binkert.org        modname,ext = self.extname
1736143Snate@binkert.org        assert ext == 'py'
1746143Snate@binkert.org
1755522Snate@binkert.org        if package:
1765522Snate@binkert.org            path = package.split('.')
1775522Snate@binkert.org        else:
1785522Snate@binkert.org            path = []
1795604Snate@binkert.org
1805604Snate@binkert.org        modpath = path[:]
1816143Snate@binkert.org        if modname != '__init__':
1826143Snate@binkert.org            modpath += [ modname ]
1834762Snate@binkert.org        modpath = '.'.join(modpath)
1844762Snate@binkert.org
1856143Snate@binkert.org        arcpath = path + [ self.basename ]
1866727Ssteve.reinhardt@amd.com        abspath = self.snode.abspath
1876727Ssteve.reinhardt@amd.com        if not exists(abspath):
1886727Ssteve.reinhardt@amd.com            abspath = self.tnode.abspath
1894762Snate@binkert.org
1906143Snate@binkert.org        self.package = package
1916143Snate@binkert.org        self.modname = modname
1926143Snate@binkert.org        self.modpath = modpath
1936143Snate@binkert.org        self.arcname = joinpath(*arcpath)
1946727Ssteve.reinhardt@amd.com        self.abspath = abspath
1956143Snate@binkert.org        self.compiled = File(self.filename + 'c')
1967674Snate@binkert.org        self.cpp = File(self.filename + '.cc')
1977674Snate@binkert.org        self.symname = PySource.invalid_sym_char.sub('_', modpath)
1985604Snate@binkert.org
1996143Snate@binkert.org        PySource.modules[modpath] = self
2006143Snate@binkert.org        PySource.tnodes[self.tnode] = self
2016143Snate@binkert.org        PySource.symnames[self.symname] = self
2024762Snate@binkert.org
2036143Snate@binkert.orgclass SimObject(PySource):
2044762Snate@binkert.org    '''Add a SimObject python file as a python source object and add
2054762Snate@binkert.org    it to a list of sim object modules'''
2064762Snate@binkert.org
2076143Snate@binkert.org    fixed = False
2086143Snate@binkert.org    modnames = []
2094762Snate@binkert.org
2108233Snate@binkert.org    def __init__(self, source, **guards):
2118233Snate@binkert.org        '''Specify the source file and any guards (automatically in
2128233Snate@binkert.org        the m5.objects package)'''
2138233Snate@binkert.org        super(SimObject, self).__init__('m5.objects', source, **guards)
2146143Snate@binkert.org        if self.fixed:
2156143Snate@binkert.org            raise AttributeError, "Too late to call SimObject now."
2164762Snate@binkert.org
2176143Snate@binkert.org        bisect.insort_right(SimObject.modnames, self.modname)
2184762Snate@binkert.org
2196143Snate@binkert.orgclass SwigSource(SourceFile):
2204762Snate@binkert.org    '''Add a swig file to build'''
2216143Snate@binkert.org
2228233Snate@binkert.org    def __init__(self, package, source, **guards):
2238233Snate@binkert.org        '''Specify the python package, the source file, and any guards'''
2248233Snate@binkert.org        super(SwigSource, self).__init__(source, **guards)
2256143Snate@binkert.org
2266143Snate@binkert.org        modname,ext = self.extname
2276143Snate@binkert.org        assert ext == 'i'
2286143Snate@binkert.org
2296143Snate@binkert.org        self.module = modname
2306143Snate@binkert.org        cc_file = joinpath(self.dirname, modname + '_wrap.cc')
2316143Snate@binkert.org        py_file = joinpath(self.dirname, modname + '.py')
2326143Snate@binkert.org
2338233Snate@binkert.org        self.cc_source = Source(cc_file, swig=True, parent=self)
2348233Snate@binkert.org        self.py_source = PySource(package, py_file, parent=self)
235955SN/A
2368235Snate@binkert.orgclass UnitTest(object):
2378235Snate@binkert.org    '''Create a UnitTest'''
2386143Snate@binkert.org
2398235Snate@binkert.org    all = []
2408235Snate@binkert.org    def __init__(self, target, *sources, **kwargs):
2418235Snate@binkert.org        '''Specify the target name and any sources.  Sources that are
2428235Snate@binkert.org        not SourceFiles are evalued with Source().  All files are
2438235Snate@binkert.org        guarded with a guard of the same name as the UnitTest
2448235Snate@binkert.org        target.'''
2458235Snate@binkert.org
2468235Snate@binkert.org        srcs = []
2478235Snate@binkert.org        for src in sources:
2488235Snate@binkert.org            if not isinstance(src, SourceFile):
2498235Snate@binkert.org                src = Source(src, skip_lib=True)
2508235Snate@binkert.org            src.guards[target] = True
2518235Snate@binkert.org            srcs.append(src)
2528235Snate@binkert.org
2538235Snate@binkert.org        self.sources = srcs
2548235Snate@binkert.org        self.target = target
2558235Snate@binkert.org        self.main = kwargs.get('main', False)
2565584Snate@binkert.org        UnitTest.all.append(self)
2574382Sbinkertn@umich.edu
2584202Sbinkertn@umich.edu# Children should have access
2594382Sbinkertn@umich.eduExport('Source')
2604382Sbinkertn@umich.eduExport('PySource')
2614382Sbinkertn@umich.eduExport('SimObject')
2625584Snate@binkert.orgExport('SwigSource')
2634382Sbinkertn@umich.eduExport('UnitTest')
2644382Sbinkertn@umich.edu
2654382Sbinkertn@umich.edu########################################################################
2668232Snate@binkert.org#
2675192Ssaidi@eecs.umich.edu# Debug Flags
2688232Snate@binkert.org#
2698232Snate@binkert.orgdebug_flags = {}
2708232Snate@binkert.orgdef DebugFlag(name, desc=None):
2715192Ssaidi@eecs.umich.edu    if name in debug_flags:
2728232Snate@binkert.org        raise AttributeError, "Flag %s already specified" % name
2735192Ssaidi@eecs.umich.edu    debug_flags[name] = (name, (), desc)
2745799Snate@binkert.org
2758232Snate@binkert.orgdef CompoundFlag(name, flags, desc=None):
2765192Ssaidi@eecs.umich.edu    if name in debug_flags:
2775192Ssaidi@eecs.umich.edu        raise AttributeError, "Flag %s already specified" % name
2785192Ssaidi@eecs.umich.edu
2798232Snate@binkert.org    compound = tuple(flags)
2805192Ssaidi@eecs.umich.edu    debug_flags[name] = (name, compound, desc)
2818232Snate@binkert.org
2825192Ssaidi@eecs.umich.eduExport('DebugFlag')
2835192Ssaidi@eecs.umich.eduExport('CompoundFlag')
2845192Ssaidi@eecs.umich.edu
2855192Ssaidi@eecs.umich.edu########################################################################
2864382Sbinkertn@umich.edu#
2874382Sbinkertn@umich.edu# Set some compiler variables
2884382Sbinkertn@umich.edu#
2892667Sstever@eecs.umich.edu
2902667Sstever@eecs.umich.edu# Include file paths are rooted in this directory.  SCons will
2912667Sstever@eecs.umich.edu# automatically expand '.' to refer to both the source directory and
2922667Sstever@eecs.umich.edu# the corresponding build directory to pick up generated include
2932667Sstever@eecs.umich.edu# files.
2942667Sstever@eecs.umich.eduenv.Append(CPPPATH=Dir('.'))
2955742Snate@binkert.org
2965742Snate@binkert.orgfor extra_dir in extras_dir_list:
2975742Snate@binkert.org    env.Append(CPPPATH=Dir(extra_dir))
2985793Snate@binkert.org
2998334Snate@binkert.org# Workaround for bug in SCons version > 0.97d20071212
3005793Snate@binkert.org# Scons bug id: 2006 gem5 Bug id: 308
3015793Snate@binkert.orgfor root, dirs, files in os.walk(base_dir, topdown=True):
3025793Snate@binkert.org    Dir(root[len(base_dir) + 1:])
3034382Sbinkertn@umich.edu
3044762Snate@binkert.org########################################################################
3055344Sstever@gmail.com#
3064382Sbinkertn@umich.edu# Walk the tree and execute all SConscripts in subdirectories
3075341Sstever@gmail.com#
3085742Snate@binkert.org
3095742Snate@binkert.orghere = Dir('.').srcnode().abspath
3105742Snate@binkert.orgfor root, dirs, files in os.walk(base_dir, topdown=True):
3115742Snate@binkert.org    if root == here:
3125742Snate@binkert.org        # we don't want to recurse back into this SConscript
3134762Snate@binkert.org        continue
3145742Snate@binkert.org
3155742Snate@binkert.org    if 'SConscript' in files:
3167722Sgblack@eecs.umich.edu        build_dir = joinpath(env['BUILDDIR'], root[len(base_dir) + 1:])
3175742Snate@binkert.org        SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
3185742Snate@binkert.org
3195742Snate@binkert.orgfor extra_dir in extras_dir_list:
3205742Snate@binkert.org    prefix_len = len(dirname(extra_dir)) + 1
3218242Sbradley.danofsky@amd.com    for root, dirs, files in os.walk(extra_dir, topdown=True):
3228242Sbradley.danofsky@amd.com        # if build lives in the extras directory, don't walk down it
3238242Sbradley.danofsky@amd.com        if 'build' in dirs:
3248242Sbradley.danofsky@amd.com            dirs.remove('build')
3255341Sstever@gmail.com
3265742Snate@binkert.org        if 'SConscript' in files:
3277722Sgblack@eecs.umich.edu            build_dir = joinpath(env['BUILDDIR'], root[prefix_len:])
3284773Snate@binkert.org            SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
3296108Snate@binkert.org
3301858SN/Afor opt in export_vars:
3311085SN/A    env.ConfigFile(opt)
3326658Snate@binkert.org
3336658Snate@binkert.orgdef makeTheISA(source, target, env):
3347673Snate@binkert.org    isas = [ src.get_contents() for src in source ]
3356658Snate@binkert.org    target_isa = env['TARGET_ISA']
3366658Snate@binkert.org    def define(isa):
3376658Snate@binkert.org        return isa.upper() + '_ISA'
3386658Snate@binkert.org    
3396658Snate@binkert.org    def namespace(isa):
3406658Snate@binkert.org        return isa[0].upper() + isa[1:].lower() + 'ISA' 
3416658Snate@binkert.org
3427673Snate@binkert.org
3437673Snate@binkert.org    code = code_formatter()
3447673Snate@binkert.org    code('''\
3457673Snate@binkert.org#ifndef __CONFIG_THE_ISA_HH__
3467673Snate@binkert.org#define __CONFIG_THE_ISA_HH__
3477673Snate@binkert.org
3487673Snate@binkert.org''')
3496658Snate@binkert.org
3507673Snate@binkert.org    for i,isa in enumerate(isas):
3517673Snate@binkert.org        code('#define $0 $1', define(isa), i + 1)
3527673Snate@binkert.org
3537673Snate@binkert.org    code('''
3547673Snate@binkert.org
3557673Snate@binkert.org#define THE_ISA ${{define(target_isa)}}
3567673Snate@binkert.org#define TheISA ${{namespace(target_isa)}}
3577673Snate@binkert.org#define THE_ISA_STR "${{target_isa}}"
3587673Snate@binkert.org
3597673Snate@binkert.org#endif // __CONFIG_THE_ISA_HH__''')
3606658Snate@binkert.org
3617756SAli.Saidi@ARM.com    code.write(str(target[0]))
3627816Ssteve.reinhardt@amd.com
3636658Snate@binkert.orgenv.Command('config/the_isa.hh', map(Value, all_isa_list),
3644382Sbinkertn@umich.edu            MakeAction(makeTheISA, Transform("CFG ISA", 0)))
3654382Sbinkertn@umich.edu
3664762Snate@binkert.org########################################################################
3674762Snate@binkert.org#
3684762Snate@binkert.org# Prevent any SimObjects from being added after this point, they
3696654Snate@binkert.org# should all have been added in the SConscripts above
3706654Snate@binkert.org#
3715517Snate@binkert.orgSimObject.fixed = True
3725517Snate@binkert.org
3735517Snate@binkert.orgclass DictImporter(object):
3745517Snate@binkert.org    '''This importer takes a dictionary of arbitrary module names that
3755517Snate@binkert.org    map to arbitrary filenames.'''
3765517Snate@binkert.org    def __init__(self, modules):
3775517Snate@binkert.org        self.modules = modules
3785517Snate@binkert.org        self.installed = set()
3795517Snate@binkert.org
3805517Snate@binkert.org    def __del__(self):
3815517Snate@binkert.org        self.unload()
3825517Snate@binkert.org
3835517Snate@binkert.org    def unload(self):
3845517Snate@binkert.org        import sys
3855517Snate@binkert.org        for module in self.installed:
3865517Snate@binkert.org            del sys.modules[module]
3875517Snate@binkert.org        self.installed = set()
3886654Snate@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'):
3976143Snate@binkert.org            return None
3986654Snate@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('.')
4136654Snate@binkert.org            return mod
4146654Snate@binkert.org
4155517Snate@binkert.org        if fullname == 'm5.defines':
4165517Snate@binkert.org            mod.__dict__['buildEnv'] = m5.util.SmartDict(build_env)
4176143Snate@binkert.org            return mod
4186143Snate@binkert.org
4196143Snate@binkert.org        source = self.modules[fullname]
4206727Ssteve.reinhardt@amd.com        if source.modname == '__init__':
4215517Snate@binkert.org            mod.__path__ = source.modpath
4226727Ssteve.reinhardt@amd.com        mod.__file__ = source.abspath
4235517Snate@binkert.org
4245517Snate@binkert.org        exec file(source.abspath, 'r') in mod.__dict__
4255517Snate@binkert.org
4266654Snate@binkert.org        return mod
4276654Snate@binkert.org
4287673Snate@binkert.orgimport m5.SimObject
4296654Snate@binkert.orgimport m5.params
4306654Snate@binkert.orgfrom m5.util import code_formatter
4316654Snate@binkert.org
4326654Snate@binkert.orgm5.SimObject.clear()
4335517Snate@binkert.orgm5.params.clear()
4345517Snate@binkert.org
4355517Snate@binkert.org# install the python importer so we can grab stuff from the source
4366143Snate@binkert.org# tree itself.  We can't have SimObjects added after this point or
4375517Snate@binkert.org# else we won't know about them for the rest of the stuff.
4384762Snate@binkert.orgimporter = DictImporter(PySource.modules)
4395517Snate@binkert.orgsys.meta_path[0:0] = [ importer ]
4405517Snate@binkert.org
4416143Snate@binkert.org# import all sim objects so we can populate the all_objects list
4426143Snate@binkert.org# make sure that we're working with a list, then let's sort it
4435517Snate@binkert.orgfor modname in SimObject.modnames:
4445517Snate@binkert.org    exec('from m5.objects import %s' % modname)
4455517Snate@binkert.org
4465517Snate@binkert.org# we need to unload all of the currently imported modules so that they
4475517Snate@binkert.org# will be re-imported the next time the sconscript is run
4485517Snate@binkert.orgimporter.unload()
4495517Snate@binkert.orgsys.meta_path.remove(importer)
4505517Snate@binkert.org
4515517Snate@binkert.orgsim_objects = m5.SimObject.allClasses
4528596Ssteve.reinhardt@amd.comall_enums = m5.params.allEnums
4538596Ssteve.reinhardt@amd.com
4548596Ssteve.reinhardt@amd.comif m5.SimObject.noCxxHeader:
4558596Ssteve.reinhardt@amd.com    print >> sys.stderr, \
4568596Ssteve.reinhardt@amd.com        "warning: At least one SimObject lacks a header specification. " \
4578596Ssteve.reinhardt@amd.com        "This can cause unexpected results in the generated SWIG " \
4588596Ssteve.reinhardt@amd.com        "wrappers."
4596143Snate@binkert.org
4605517Snate@binkert.org# Find param types that need to be explicitly wrapped with swig.
4616654Snate@binkert.org# These will be recognized because the ParamDesc will have a
4626654Snate@binkert.org# swig_decl() method.  Most param types are based on types that don't
4636654Snate@binkert.org# need this, either because they're based on native types (like Int)
4646654Snate@binkert.org# or because they're SimObjects (which get swigged independently).
4656654Snate@binkert.org# For now the only things handled here are VectorParam types.
4666654Snate@binkert.orgparams_to_swig = {}
4675517Snate@binkert.orgfor name,obj in sorted(sim_objects.iteritems()):
4685517Snate@binkert.org    for param in obj._params.local.values():
4695517Snate@binkert.org        # load the ptype attribute now because it depends on the
4708596Ssteve.reinhardt@amd.com        # current version of SimObject.allClasses, but when scons
4718596Ssteve.reinhardt@amd.com        # actually uses the value, all versions of
4724762Snate@binkert.org        # SimObject.allClasses will have been loaded
4734762Snate@binkert.org        param.ptype
4744762Snate@binkert.org
4754762Snate@binkert.org        if not hasattr(param, 'swig_decl'):
4764762Snate@binkert.org            continue
4774762Snate@binkert.org        pname = param.ptype_str
4787675Snate@binkert.org        if pname not in params_to_swig:
4794762Snate@binkert.org            params_to_swig[pname] = param
4804762Snate@binkert.org
4814762Snate@binkert.org########################################################################
4824762Snate@binkert.org#
4834382Sbinkertn@umich.edu# calculate extra dependencies
4844382Sbinkertn@umich.edu#
4855517Snate@binkert.orgmodule_depends = ["m5", "m5.SimObject", "m5.params"]
4866654Snate@binkert.orgdepends = [ PySource.modules[dep].snode for dep in module_depends ]
4875517Snate@binkert.org
4888126Sgblack@eecs.umich.edu########################################################################
4896654Snate@binkert.org#
4907673Snate@binkert.org# Commands for the basic automatically generated python files
4916654Snate@binkert.org#
4926654Snate@binkert.org
4936654Snate@binkert.org# Generate Python file containing a dict specifying the current
4946654Snate@binkert.org# buildEnv flags.
4956654Snate@binkert.orgdef makeDefinesPyFile(target, source, env):
4966654Snate@binkert.org    build_env = source[0].get_contents()
4976654Snate@binkert.org
4986669Snate@binkert.org    code = code_formatter()
4996669Snate@binkert.org    code("""
5006669Snate@binkert.orgimport m5.internal
5016669Snate@binkert.orgimport m5.util
5026669Snate@binkert.org
5036669Snate@binkert.orgbuildEnv = m5.util.SmartDict($build_env)
5046654Snate@binkert.org
5057673Snate@binkert.orgcompileDate = m5.internal.core.compileDate
5065517Snate@binkert.org_globals = globals()
5078126Sgblack@eecs.umich.edufor key,val in m5.internal.core.__dict__.iteritems():
5085798Snate@binkert.org    if key.startswith('flag_'):
5097756SAli.Saidi@ARM.com        flag = key[5:]
5107816Ssteve.reinhardt@amd.com        _globals[flag] = val
5115798Snate@binkert.orgdel _globals
5125798Snate@binkert.org""")
5135517Snate@binkert.org    code.write(target[0].abspath)
5145517Snate@binkert.org
5157673Snate@binkert.orgdefines_info = Value(build_env)
5165517Snate@binkert.org# Generate a file with all of the compile options in it
5175517Snate@binkert.orgenv.Command('python/m5/defines.py', defines_info,
5187673Snate@binkert.org            MakeAction(makeDefinesPyFile, Transform("DEFINES", 0)))
5197673Snate@binkert.orgPySource('m5', 'python/m5/defines.py')
5205517Snate@binkert.org
5215798Snate@binkert.org# Generate python file containing info about the M5 source code
5225798Snate@binkert.orgdef makeInfoPyFile(target, source, env):
5238333Snate@binkert.org    code = code_formatter()
5247816Ssteve.reinhardt@amd.com    for src in source:
5255798Snate@binkert.org        data = ''.join(file(src.srcnode().abspath, 'r').xreadlines())
5265798Snate@binkert.org        code('$src = ${{repr(data)}}')
5274762Snate@binkert.org    code.write(str(target[0]))
5284762Snate@binkert.org
5294762Snate@binkert.org# Generate a file that wraps the basic top level files
5304762Snate@binkert.orgenv.Command('python/m5/info.py',
5314762Snate@binkert.org            [ '#/COPYING', '#/LICENSE', '#/README', ],
5328596Ssteve.reinhardt@amd.com            MakeAction(makeInfoPyFile, Transform("INFO")))
5335517Snate@binkert.orgPySource('m5', 'python/m5/info.py')
5345517Snate@binkert.org
5355517Snate@binkert.org########################################################################
5365517Snate@binkert.org#
5375517Snate@binkert.org# Create all of the SimObject param headers and enum headers
5387673Snate@binkert.org#
5398596Ssteve.reinhardt@amd.com
5407673Snate@binkert.orgdef createSimObjectParamStruct(target, source, env):
5415517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5428596Ssteve.reinhardt@amd.com
5435517Snate@binkert.org    name = str(source[0].get_contents())
5445517Snate@binkert.org    obj = sim_objects[name]
5455517Snate@binkert.org
5468596Ssteve.reinhardt@amd.com    code = code_formatter()
5475517Snate@binkert.org    obj.cxx_param_decl(code)
5487673Snate@binkert.org    code.write(target[0].abspath)
5497673Snate@binkert.org
5507673Snate@binkert.orgdef createParamSwigWrapper(target, source, env):
5515517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5525517Snate@binkert.org
5535517Snate@binkert.org    name = str(source[0].get_contents())
5545517Snate@binkert.org    param = params_to_swig[name]
5555517Snate@binkert.org
5565517Snate@binkert.org    code = code_formatter()
5575517Snate@binkert.org    param.swig_decl(code)
5587673Snate@binkert.org    code.write(target[0].abspath)
5597673Snate@binkert.org
5607673Snate@binkert.orgdef createEnumStrings(target, source, env):
5615517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5628596Ssteve.reinhardt@amd.com
5635517Snate@binkert.org    name = str(source[0].get_contents())
5645517Snate@binkert.org    obj = all_enums[name]
5655517Snate@binkert.org
5665517Snate@binkert.org    code = code_formatter()
5675517Snate@binkert.org    obj.cxx_def(code)
5687673Snate@binkert.org    code.write(target[0].abspath)
5697673Snate@binkert.org
5707673Snate@binkert.orgdef createEnumDecls(target, source, env):
5715517Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5728596Ssteve.reinhardt@amd.com
5737675Snate@binkert.org    name = str(source[0].get_contents())
5747675Snate@binkert.org    obj = all_enums[name]
5757675Snate@binkert.org
5767675Snate@binkert.org    code = code_formatter()
5777675Snate@binkert.org    obj.cxx_decl(code)
5787675Snate@binkert.org    code.write(target[0].abspath)
5798596Ssteve.reinhardt@amd.com
5807675Snate@binkert.orgdef createEnumSwigWrapper(target, source, env):
5817675Snate@binkert.org    assert len(target) == 1 and len(source) == 1
5828596Ssteve.reinhardt@amd.com
5838596Ssteve.reinhardt@amd.com    name = str(source[0].get_contents())
5848596Ssteve.reinhardt@amd.com    obj = all_enums[name]
5858596Ssteve.reinhardt@amd.com
5868596Ssteve.reinhardt@amd.com    code = code_formatter()
5878596Ssteve.reinhardt@amd.com    obj.swig_decl(code)
5888596Ssteve.reinhardt@amd.com    code.write(target[0].abspath)
5898596Ssteve.reinhardt@amd.com
5908596Ssteve.reinhardt@amd.comdef createSimObjectSwigWrapper(target, source, env):
5914762Snate@binkert.org    name = source[0].get_contents()
5926143Snate@binkert.org    obj = sim_objects[name]
5936143Snate@binkert.org
5946143Snate@binkert.org    code = code_formatter()
5954762Snate@binkert.org    obj.swig_decl(code)
5964762Snate@binkert.org    code.write(target[0].abspath)
5974762Snate@binkert.org
5987756SAli.Saidi@ARM.com# Generate all of the SimObject param C++ struct header files
5998596Ssteve.reinhardt@amd.comparams_hh_files = []
6004762Snate@binkert.orgfor name,simobj in sorted(sim_objects.iteritems()):
6014762Snate@binkert.org    py_source = PySource.modules[simobj.__module__]
6028596Ssteve.reinhardt@amd.com    extra_deps = [ py_source.tnode ]
6035463Snate@binkert.org
6048596Ssteve.reinhardt@amd.com    hh_file = File('params/%s.hh' % name)
6058596Ssteve.reinhardt@amd.com    params_hh_files.append(hh_file)
6065463Snate@binkert.org    env.Command(hh_file, Value(name),
6077756SAli.Saidi@ARM.com                MakeAction(createSimObjectParamStruct, Transform("SO PARAM")))
6088596Ssteve.reinhardt@amd.com    env.Depends(hh_file, depends + extra_deps)
6094762Snate@binkert.org
6107677Snate@binkert.org# Generate any needed param SWIG wrapper files
6114762Snate@binkert.orgparams_i_files = []
6124762Snate@binkert.orgfor name,param in params_to_swig.iteritems():
6136143Snate@binkert.org    i_file = File('python/m5/internal/%s.i' % (param.swig_module_name()))
6146143Snate@binkert.org    params_i_files.append(i_file)
6156143Snate@binkert.org    env.Command(i_file, Value(name),
6164762Snate@binkert.org                MakeAction(createParamSwigWrapper, Transform("SW PARAM")))
6174762Snate@binkert.org    env.Depends(i_file, depends)
6187756SAli.Saidi@ARM.com    SwigSource('m5.internal', i_file)
6197816Ssteve.reinhardt@amd.com
6204762Snate@binkert.org# Generate all enum header files
6214762Snate@binkert.orgfor name,enum in sorted(all_enums.iteritems()):
6224762Snate@binkert.org    py_source = PySource.modules[enum.__module__]
6234762Snate@binkert.org    extra_deps = [ py_source.tnode ]
6247756SAli.Saidi@ARM.com
6258596Ssteve.reinhardt@amd.com    cc_file = File('enums/%s.cc' % name)
6264762Snate@binkert.org    env.Command(cc_file, Value(name),
6274762Snate@binkert.org                MakeAction(createEnumStrings, Transform("ENUM STR")))
6287677Snate@binkert.org    env.Depends(cc_file, depends + extra_deps)
6297756SAli.Saidi@ARM.com    Source(cc_file)
6308596Ssteve.reinhardt@amd.com
6317675Snate@binkert.org    hh_file = File('enums/%s.hh' % name)
6327677Snate@binkert.org    env.Command(hh_file, Value(name),
6335517Snate@binkert.org                MakeAction(createEnumDecls, Transform("ENUMDECL")))
6348596Ssteve.reinhardt@amd.com    env.Depends(hh_file, depends + extra_deps)
6357675Snate@binkert.org
6368596Ssteve.reinhardt@amd.com    i_file = File('python/m5/internal/enum_%s.i' % name)
6378596Ssteve.reinhardt@amd.com    env.Command(i_file, Value(name),
6388596Ssteve.reinhardt@amd.com                MakeAction(createEnumSwigWrapper, Transform("ENUMSWIG")))
6398596Ssteve.reinhardt@amd.com    env.Depends(i_file, depends + extra_deps)
6408596Ssteve.reinhardt@amd.com    SwigSource('m5.internal', i_file)
6414762Snate@binkert.org
6427674Snate@binkert.org# Generate SimObject SWIG wrapper files
6437674Snate@binkert.orgfor name,simobj in sim_objects.iteritems():
6447674Snate@binkert.org    py_source = PySource.modules[simobj.__module__]
6457674Snate@binkert.org    extra_deps = [ py_source.tnode ]
6467674Snate@binkert.org
6477674Snate@binkert.org    i_file = File('python/m5/internal/param_%s.i' % name)
6487674Snate@binkert.org    env.Command(i_file, Value(name),
6497674Snate@binkert.org                MakeAction(createSimObjectSwigWrapper, Transform("SO SWIG")))
6507674Snate@binkert.org    env.Depends(i_file, depends + extra_deps)
6517674Snate@binkert.org    SwigSource('m5.internal', i_file)
6527674Snate@binkert.org
6537674Snate@binkert.org# Generate the main swig init file
6547674Snate@binkert.orgdef makeEmbeddedSwigInit(target, source, env):
6557674Snate@binkert.org    code = code_formatter()
6567674Snate@binkert.org    module = source[0].get_contents()
6574762Snate@binkert.org    code('''\
6586143Snate@binkert.org#include "sim/init.hh"
6596143Snate@binkert.org
6607756SAli.Saidi@ARM.comextern "C" {
6617816Ssteve.reinhardt@amd.com    void init_${module}();
6628235Snate@binkert.org}
6638596Ssteve.reinhardt@amd.com
6647756SAli.Saidi@ARM.comEmbeddedSwig embed_swig_${module}(init_${module});
6657816Ssteve.reinhardt@amd.com''')
6668235Snate@binkert.org    code.write(str(target[0]))
6674382Sbinkertn@umich.edu    
6688232Snate@binkert.org# Build all swig modules
6698232Snate@binkert.orgfor swig in SwigSource.all:
6708232Snate@binkert.org    env.Command([swig.cc_source.tnode, swig.py_source.tnode], swig.tnode,
6718232Snate@binkert.org                MakeAction('$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} '
6728232Snate@binkert.org                '-o ${TARGETS[0]} $SOURCES', Transform("SWIG")))
6736229Snate@binkert.org    cc_file = str(swig.tnode)
6748232Snate@binkert.org    init_file = '%s/%s_init.cc' % (dirname(cc_file), basename(cc_file))
6758232Snate@binkert.org    env.Command(init_file, Value(swig.module),
6768232Snate@binkert.org                MakeAction(makeEmbeddedSwigInit, Transform("EMBED SW")))
6776229Snate@binkert.org    Source(init_file, **swig.guards)
6787673Snate@binkert.org
6795517Snate@binkert.org#
6805517Snate@binkert.org# Handle debug flags
6817673Snate@binkert.org#
6825517Snate@binkert.orgdef makeDebugFlagCC(target, source, env):
6835517Snate@binkert.org    assert(len(target) == 1 and len(source) == 1)
6845517Snate@binkert.org
6855517Snate@binkert.org    val = eval(source[0].get_contents())
6868232Snate@binkert.org    name, compound, desc = val
6877673Snate@binkert.org    compound = list(sorted(compound))
6887673Snate@binkert.org
6898232Snate@binkert.org    code = code_formatter()
6908232Snate@binkert.org
6918232Snate@binkert.org    # file header
6928232Snate@binkert.org    code('''
6937673Snate@binkert.org/*
6945517Snate@binkert.org * DO NOT EDIT THIS FILE! Automatically generated
6958232Snate@binkert.org */
6968232Snate@binkert.org
6978232Snate@binkert.org#include "base/debug.hh"
6988232Snate@binkert.org''')
6997673Snate@binkert.org
7008232Snate@binkert.org    for flag in compound:
7018232Snate@binkert.org        code('#include "debug/$flag.hh"')
7028232Snate@binkert.org    code()
7038232Snate@binkert.org    code('namespace Debug {')
7048232Snate@binkert.org    code()
7058232Snate@binkert.org
7067673Snate@binkert.org    if not compound:
7075517Snate@binkert.org        code('SimpleFlag $name("$name", "$desc");')
7088232Snate@binkert.org    else:
7098232Snate@binkert.org        code('CompoundFlag $name("$name", "$desc",')
7105517Snate@binkert.org        code.indent()
7117673Snate@binkert.org        last = len(compound) - 1
7125517Snate@binkert.org        for i,flag in enumerate(compound):
7138232Snate@binkert.org            if i != last:
7148232Snate@binkert.org                code('$flag,')
7155517Snate@binkert.org            else:
7168232Snate@binkert.org                code('$flag);')
7178232Snate@binkert.org        code.dedent()
7188232Snate@binkert.org
7197673Snate@binkert.org    code()
7205517Snate@binkert.org    code('} // namespace Debug')
7215517Snate@binkert.org
7227673Snate@binkert.org    code.write(str(target[0]))
7235517Snate@binkert.org
7245517Snate@binkert.orgdef makeDebugFlagHH(target, source, env):
7255517Snate@binkert.org    assert(len(target) == 1 and len(source) == 1)
7268232Snate@binkert.org
7275517Snate@binkert.org    val = eval(source[0].get_contents())
7285517Snate@binkert.org    name, compound, desc = val
7298232Snate@binkert.org
7308232Snate@binkert.org    code = code_formatter()
7315517Snate@binkert.org
7328232Snate@binkert.org    # file header boilerplate
7338232Snate@binkert.org    code('''\
7345517Snate@binkert.org/*
7358232Snate@binkert.org * DO NOT EDIT THIS FILE!
7368232Snate@binkert.org *
7378232Snate@binkert.org * Automatically generated by SCons
7385517Snate@binkert.org */
7398232Snate@binkert.org
7408232Snate@binkert.org#ifndef __DEBUG_${name}_HH__
7418232Snate@binkert.org#define __DEBUG_${name}_HH__
7428232Snate@binkert.org
7438232Snate@binkert.orgnamespace Debug {
7448232Snate@binkert.org''')
7455517Snate@binkert.org
7468232Snate@binkert.org    if compound:
7478232Snate@binkert.org        code('class CompoundFlag;')
7485517Snate@binkert.org    code('class SimpleFlag;')
7498232Snate@binkert.org
7507673Snate@binkert.org    if compound:
7515517Snate@binkert.org        code('extern CompoundFlag $name;')
7527673Snate@binkert.org        for flag in compound:
7535517Snate@binkert.org            code('extern SimpleFlag $flag;')
7548232Snate@binkert.org    else:
7558232Snate@binkert.org        code('extern SimpleFlag $name;')
7568232Snate@binkert.org
7575192Ssaidi@eecs.umich.edu    code('''
7588232Snate@binkert.org}
7598232Snate@binkert.org
7608232Snate@binkert.org#endif // __DEBUG_${name}_HH__
7618232Snate@binkert.org''')
7628232Snate@binkert.org
7635192Ssaidi@eecs.umich.edu    code.write(str(target[0]))
7647674Snate@binkert.org
7655522Snate@binkert.orgfor name,flag in sorted(debug_flags.iteritems()):
7665522Snate@binkert.org    n, compound, desc = flag
7677674Snate@binkert.org    assert n == name
7687674Snate@binkert.org
7697674Snate@binkert.org    env.Command('debug/%s.hh' % name, Value(flag),
7707674Snate@binkert.org                MakeAction(makeDebugFlagHH, Transform("TRACING", 0)))
7717674Snate@binkert.org    env.Command('debug/%s.cc' % name, Value(flag),
7727674Snate@binkert.org                MakeAction(makeDebugFlagCC, Transform("TRACING", 0)))
7737674Snate@binkert.org    Source('debug/%s.cc' % name)
7747674Snate@binkert.org
7755522Snate@binkert.org# Embed python files.  All .py files that have been indicated by a
7765522Snate@binkert.org# PySource() call in a SConscript need to be embedded into the M5
7775522Snate@binkert.org# library.  To do that, we compile the file to byte code, marshal the
7785517Snate@binkert.org# byte code, compress it, and then generate a c++ file that
7795522Snate@binkert.org# inserts the result into an array.
7805517Snate@binkert.orgdef embedPyFile(target, source, env):
7816143Snate@binkert.org    def c_str(string):
7826727Ssteve.reinhardt@amd.com        if string is None:
7835522Snate@binkert.org            return "0"
7845522Snate@binkert.org        return '"%s"' % string
7855522Snate@binkert.org
7867674Snate@binkert.org    '''Action function to compile a .py into a code object, marshal
7875517Snate@binkert.org    it, compress it, and stick it into an asm file so the code appears
7887673Snate@binkert.org    as just bytes with a label in the data section'''
7897673Snate@binkert.org
7907674Snate@binkert.org    src = file(str(source[0]), 'r').read()
7917673Snate@binkert.org
7927674Snate@binkert.org    pysource = PySource.tnodes[source[0]]
7937674Snate@binkert.org    compiled = compile(src, pysource.abspath, 'exec')
7947674Snate@binkert.org    marshalled = marshal.dumps(compiled)
7957674Snate@binkert.org    compressed = zlib.compress(marshalled)
7967674Snate@binkert.org    data = compressed
7977674Snate@binkert.org    sym = pysource.symname
7985522Snate@binkert.org
7995522Snate@binkert.org    code = code_formatter()
8007674Snate@binkert.org    code('''\
8017674Snate@binkert.org#include "sim/init.hh"
8027674Snate@binkert.org
8037674Snate@binkert.orgnamespace {
8047673Snate@binkert.org
8057674Snate@binkert.orgconst uint8_t data_${sym}[] = {
8067674Snate@binkert.org''')
8077674Snate@binkert.org    code.indent()
8087674Snate@binkert.org    step = 16
8097674Snate@binkert.org    for i in xrange(0, len(data), step):
8107674Snate@binkert.org        x = array.array('B', data[i:i+step])
8117674Snate@binkert.org        code(''.join('%d,' % d for d in x))
8127674Snate@binkert.org    code.dedent()
8137811Ssteve.reinhardt@amd.com    
8147674Snate@binkert.org    code('''};
8157673Snate@binkert.org
8165522Snate@binkert.orgEmbeddedPython embedded_${sym}(
8176143Snate@binkert.org    ${{c_str(pysource.arcname)}},
8187756SAli.Saidi@ARM.com    ${{c_str(pysource.abspath)}},
8197816Ssteve.reinhardt@amd.com    ${{c_str(pysource.modpath)}},
8207674Snate@binkert.org    data_${sym},
8214382Sbinkertn@umich.edu    ${{len(data)}},
8224382Sbinkertn@umich.edu    ${{len(marshalled)}});
8234382Sbinkertn@umich.edu
8244382Sbinkertn@umich.edu} // anonymous namespace
8254382Sbinkertn@umich.edu''')
8264382Sbinkertn@umich.edu    code.write(str(target[0]))
8274382Sbinkertn@umich.edu
8284382Sbinkertn@umich.edufor source in PySource.all:
8294382Sbinkertn@umich.edu    env.Command(source.cpp, source.tnode, 
8304382Sbinkertn@umich.edu                MakeAction(embedPyFile, Transform("EMBED PY")))
8316143Snate@binkert.org    Source(source.cpp)
832955SN/A
8332655Sstever@eecs.umich.edu########################################################################
8342655Sstever@eecs.umich.edu#
8352655Sstever@eecs.umich.edu# Define binaries.  Each different build type (debug, opt, etc.) gets
8362655Sstever@eecs.umich.edu# a slightly different build environment.
8372655Sstever@eecs.umich.edu#
8385601Snate@binkert.org
8395601Snate@binkert.org# List of constructed environments to pass back to SConstruct
8408334Snate@binkert.orgenvList = []
8418334Snate@binkert.org
8428334Snate@binkert.orgdate_source = Source('base/date.cc', skip_lib=True)
8435522Snate@binkert.org
8445863Snate@binkert.org# Function to create a new build environment as clone of current
8455601Snate@binkert.org# environment 'env' with modified object suffix and optional stripped
8465601Snate@binkert.org# binary.  Additional keyword arguments are appended to corresponding
8475601Snate@binkert.org# build environment vars.
8485863Snate@binkert.orgdef makeEnv(label, objsfx, strip = False, **kwargs):
8496143Snate@binkert.org    # SCons doesn't know to append a library suffix when there is a '.' in the
8505559Snate@binkert.org    # name.  Use '_' instead.
8515559Snate@binkert.org    libname = 'gem5_' + label
8525559Snate@binkert.org    exename = 'gem5.' + label
8535559Snate@binkert.org    secondary_exename = 'm5.' + label
8548656Sandreas.hansson@arm.com
8558614Sgblack@eecs.umich.edu    new_env = env.Clone(OBJSUFFIX=objsfx, SHOBJSUFFIX=objsfx + 's')
8568614Sgblack@eecs.umich.edu    new_env.Label = label
8578737Skoansin.tan@gmail.com    new_env.Append(**kwargs)
8588737Skoansin.tan@gmail.com
8598737Skoansin.tan@gmail.com    swig_env = new_env.Clone()
8605601Snate@binkert.org    swig_env.Append(CCFLAGS='-Werror')
8616143Snate@binkert.org    if env['GCC']:
8626143Snate@binkert.org        swig_env.Append(CCFLAGS=['-Wno-uninitialized', '-Wno-sign-compare',
8636143Snate@binkert.org                                 '-Wno-parentheses', '-Wno-unused-label',
8646143Snate@binkert.org                                 '-Wno-unused-value'])
8656143Snate@binkert.org        if compareVersions(env['GCC_VERSION'], '4.6') >= 0:
8666143Snate@binkert.org            swig_env.Append(CCFLAGS='-Wno-unused-but-set-variable')
8676143Snate@binkert.org    if env['CLANG']:
8686143Snate@binkert.org        swig_env.Append(CCFLAGS=['-Wno-unused-label', '-Wno-unused-value'])
8696143Snate@binkert.org
8706143Snate@binkert.org    werror_env = new_env.Clone()
8716143Snate@binkert.org    werror_env.Append(CCFLAGS='-Werror')
8726143Snate@binkert.org
8736143Snate@binkert.org    def make_obj(source, static, extra_deps = None):
8746143Snate@binkert.org        '''This function adds the specified source to the correct
8756143Snate@binkert.org        build environment, and returns the corresponding SCons Object
8766143Snate@binkert.org        nodes'''
8776143Snate@binkert.org
8786143Snate@binkert.org        if source.swig:
8796143Snate@binkert.org            env = swig_env
8806143Snate@binkert.org        elif source.Werror:
8816143Snate@binkert.org            env = werror_env
8826143Snate@binkert.org        else:
8836143Snate@binkert.org            env = new_env
8846143Snate@binkert.org
8856143Snate@binkert.org        if static:
8868594Snate@binkert.org            obj = env.StaticObject(source.tnode)
8878594Snate@binkert.org        else:
8888594Snate@binkert.org            obj = env.SharedObject(source.tnode)
8898594Snate@binkert.org
8906143Snate@binkert.org        if extra_deps:
8916143Snate@binkert.org            env.Depends(obj, extra_deps)
8926143Snate@binkert.org
8936143Snate@binkert.org        return obj
8946143Snate@binkert.org
8956240Snate@binkert.org    static_objs = \
8965554Snate@binkert.org        [ make_obj(s, True) for s in Source.get(main=False, skip_lib=False) ]
8975522Snate@binkert.org    shared_objs = \
8985522Snate@binkert.org        [ make_obj(s, False) for s in Source.get(main=False, skip_lib=False) ]
8995797Snate@binkert.org
9005797Snate@binkert.org    static_date = make_obj(date_source, static=True, extra_deps=static_objs)
9015522Snate@binkert.org    static_objs.append(static_date)
9025601Snate@binkert.org    
9038233Snate@binkert.org    shared_date = make_obj(date_source, static=False, extra_deps=shared_objs)
9048233Snate@binkert.org    shared_objs.append(shared_date)
9058235Snate@binkert.org
9068235Snate@binkert.org    # First make a library of everything but main() so other programs can
9078235Snate@binkert.org    # link against m5.
9088235Snate@binkert.org    static_lib = new_env.StaticLibrary(libname, static_objs)
9098235Snate@binkert.org    shared_lib = new_env.SharedLibrary(libname, shared_objs)
9108235Snate@binkert.org
9118235Snate@binkert.org    # Now link a stub with main() and the static library.
9126143Snate@binkert.org    main_objs = [ make_obj(s, True) for s in Source.get(main=True) ]
9132655Sstever@eecs.umich.edu
9146143Snate@binkert.org    for test in UnitTest.all:
9156143Snate@binkert.org        flags = { test.target : True }
9168233Snate@binkert.org        test_sources = Source.get(**flags)
9176143Snate@binkert.org        test_objs = [ make_obj(s, static=True) for s in test_sources ]
9186143Snate@binkert.org        if test.main:
9194007Ssaidi@eecs.umich.edu            test_objs += main_objs
9204596Sbinkertn@umich.edu        testname = "unittest/%s.%s" % (test.target, label)
9214007Ssaidi@eecs.umich.edu        new_env.Program(testname, test_objs + static_objs)
9224596Sbinkertn@umich.edu
9237756SAli.Saidi@ARM.com    progname = exename
9247816Ssteve.reinhardt@amd.com    if strip:
9258334Snate@binkert.org        progname += '.unstripped'
9268334Snate@binkert.org
9278334Snate@binkert.org    targets = new_env.Program(progname, main_objs + static_objs)
9288334Snate@binkert.org
9295601Snate@binkert.org    if strip:
9305601Snate@binkert.org        if sys.platform == 'sunos5':
9312655Sstever@eecs.umich.edu            cmd = 'cp $SOURCE $TARGET; strip $TARGET'
932955SN/A        else:
9333918Ssaidi@eecs.umich.edu            cmd = 'strip $SOURCE -o $TARGET'
9348737Skoansin.tan@gmail.com        targets = new_env.Command(exename, progname,
9353918Ssaidi@eecs.umich.edu                    MakeAction(cmd, Transform("STRIP")))
9363918Ssaidi@eecs.umich.edu
9373918Ssaidi@eecs.umich.edu    new_env.Command(secondary_exename, exename,
9383918Ssaidi@eecs.umich.edu            MakeAction('ln $SOURCE $TARGET', Transform("HARDLINK")))
9393918Ssaidi@eecs.umich.edu
9403918Ssaidi@eecs.umich.edu    new_env.M5Binary = targets[0]
9413918Ssaidi@eecs.umich.edu    envList.append(new_env)
9423918Ssaidi@eecs.umich.edu
9433918Ssaidi@eecs.umich.edu# Start out with the compiler flags common to all compilers,
9443918Ssaidi@eecs.umich.edu# i.e. they all use -g for opt and -g -pg for prof
9453918Ssaidi@eecs.umich.educcflags = {'debug' : [], 'opt' : ['-g'], 'fast' : [], 'prof' : ['-g', '-pg'],
9463918Ssaidi@eecs.umich.edu           'perf' : ['-g']}
9473940Ssaidi@eecs.umich.edu
9483940Ssaidi@eecs.umich.edu# Start out with the linker flags common to all linkers, i.e. -pg for
9493940Ssaidi@eecs.umich.edu# prof, and -lprofiler for perf. The -lprofile flag is surrounded by
9503942Ssaidi@eecs.umich.edu# no-as-needed and as-needed as the binutils linker is too clever and
9513940Ssaidi@eecs.umich.edu# simply doesn't link to the library otherwise.
9523515Ssaidi@eecs.umich.eduldflags = {'debug' : [], 'opt' : [], 'fast' : [], 'prof' : ['-pg'],
9533918Ssaidi@eecs.umich.edu           'perf' : ['-Wl,--no-as-needed', '-lprofiler', '-Wl,--as-needed']}
9544762Snate@binkert.org
9553515Ssaidi@eecs.umich.edu# For Link Time Optimization, the optimisation flags used to compile
9562655Sstever@eecs.umich.edu# individual files are decoupled from those used at link time
9573918Ssaidi@eecs.umich.edu# (i.e. you can compile with -O3 and perform LTO with -O0), so we need
9583619Sbinkertn@umich.edu# to also update the linker flags based on the target.
959955SN/Aif env['GCC']:
960955SN/A    if sys.platform == 'sunos5':
9612655Sstever@eecs.umich.edu        ccflags['debug'] += ['-gstabs+']
9623918Ssaidi@eecs.umich.edu    else:
9633619Sbinkertn@umich.edu        ccflags['debug'] += ['-ggdb3']
964955SN/A    ldflags['debug'] += ['-O0']
965955SN/A    # opt, fast, prof and perf all share the same cc flags, also add
9662655Sstever@eecs.umich.edu    # the optimization to the ldflags as LTO defers the optimization
9673918Ssaidi@eecs.umich.edu    # to link time
9683619Sbinkertn@umich.edu    for target in ['opt', 'fast', 'prof', 'perf']:
969955SN/A        ccflags[target] += ['-O3']
970955SN/A        ldflags[target] += ['-O3']
9712655Sstever@eecs.umich.edu
9723918Ssaidi@eecs.umich.edu    ccflags['fast'] += env['LTO_CCFLAGS']
9733683Sstever@eecs.umich.edu    ldflags['fast'] += env['LTO_LDFLAGS']
9742655Sstever@eecs.umich.edu
9751869SN/Aelif env['SUNCC']:
9761869SN/A    ccflags['debug'] += ['-g0']
977    ccflags['opt'] += ['-O']
978    for target in ['fast', 'prof', 'perf']:
979        ccflags[target] += ['-fast']
980elif env['ICC']:
981    ccflags['debug'] += ['-g', '-O0']
982    ccflags['opt'] += ['-O']
983    for target in ['fast', 'prof', 'perf']:
984        ccflags[target] += ['-fast']
985elif env['CLANG']:
986    ccflags['debug'] += ['-g', '-O0']
987    # opt, fast, prof and perf all share the same cc flags
988    for target in ['opt', 'fast', 'prof', 'perf']:
989        ccflags[target] += ['-O3']
990else:
991    print 'Unknown compiler, please fix compiler options'
992    Exit(1)
993
994
995# To speed things up, we only instantiate the build environments we
996# need.  We try to identify the needed environment for each target; if
997# we can't, we fall back on instantiating all the environments just to
998# be safe.
999target_types = ['debug', 'opt', 'fast', 'prof', 'perf']
1000obj2target = {'do': 'debug', 'o': 'opt', 'fo': 'fast', 'po': 'prof',
1001              'gpo' : 'perf'}
1002
1003def identifyTarget(t):
1004    ext = t.split('.')[-1]
1005    if ext in target_types:
1006        return ext
1007    if obj2target.has_key(ext):
1008        return obj2target[ext]
1009    match = re.search(r'/tests/([^/]+)/', t)
1010    if match and match.group(1) in target_types:
1011        return match.group(1)
1012    return 'all'
1013
1014needed_envs = [identifyTarget(target) for target in BUILD_TARGETS]
1015if 'all' in needed_envs:
1016    needed_envs += target_types
1017
1018# Debug binary
1019if 'debug' in needed_envs:
1020    makeEnv('debug', '.do',
1021            CCFLAGS = Split(ccflags['debug']),
1022            CPPDEFINES = ['DEBUG', 'TRACING_ON=1'],
1023            LINKFLAGS = Split(ldflags['debug']))
1024
1025# Optimized binary
1026if 'opt' in needed_envs:
1027    makeEnv('opt', '.o',
1028            CCFLAGS = Split(ccflags['opt']),
1029            CPPDEFINES = ['TRACING_ON=1'],
1030            LINKFLAGS = Split(ldflags['opt']))
1031
1032# "Fast" binary
1033if 'fast' in needed_envs:
1034    makeEnv('fast', '.fo', strip = True,
1035            CCFLAGS = Split(ccflags['fast']),
1036            CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'],
1037            LINKFLAGS = Split(ldflags['fast']))
1038
1039# Profiled binary using gprof
1040if 'prof' in needed_envs:
1041    makeEnv('prof', '.po',
1042            CCFLAGS = Split(ccflags['prof']),
1043            CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'],
1044            LINKFLAGS = Split(ldflags['prof']))
1045
1046# Profiled binary using google-pprof
1047if 'perf' in needed_envs:
1048    makeEnv('perf', '.gpo',
1049            CCFLAGS = Split(ccflags['perf']),
1050            CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'],
1051            LINKFLAGS = Split(ldflags['perf']))
1052
1053Return('envList')
1054