SConscript revision 8745
13752Sgblack@eecs.umich.edu# -*- mode:python -*-
23752Sgblack@eecs.umich.edu
33752Sgblack@eecs.umich.edu# Copyright (c) 2004-2005 The Regents of The University of Michigan
43752Sgblack@eecs.umich.edu# All rights reserved.
53752Sgblack@eecs.umich.edu#
63752Sgblack@eecs.umich.edu# Redistribution and use in source and binary forms, with or without
73752Sgblack@eecs.umich.edu# modification, are permitted provided that the following conditions are
83752Sgblack@eecs.umich.edu# met: redistributions of source code must retain the above copyright
93752Sgblack@eecs.umich.edu# notice, this list of conditions and the following disclaimer;
103752Sgblack@eecs.umich.edu# redistributions in binary form must reproduce the above copyright
113752Sgblack@eecs.umich.edu# notice, this list of conditions and the following disclaimer in the
123752Sgblack@eecs.umich.edu# documentation and/or other materials provided with the distribution;
133752Sgblack@eecs.umich.edu# neither the name of the copyright holders nor the names of its
143752Sgblack@eecs.umich.edu# contributors may be used to endorse or promote products derived from
153752Sgblack@eecs.umich.edu# this software without specific prior written permission.
163752Sgblack@eecs.umich.edu#
173752Sgblack@eecs.umich.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
183752Sgblack@eecs.umich.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
193752Sgblack@eecs.umich.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
203752Sgblack@eecs.umich.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
213752Sgblack@eecs.umich.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
223752Sgblack@eecs.umich.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
233752Sgblack@eecs.umich.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
243752Sgblack@eecs.umich.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
253752Sgblack@eecs.umich.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
263752Sgblack@eecs.umich.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
273752Sgblack@eecs.umich.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
283752Sgblack@eecs.umich.edu#
293752Sgblack@eecs.umich.edu# Authors: Nathan Binkert
303752Sgblack@eecs.umich.edu
313752Sgblack@eecs.umich.eduimport array
323752Sgblack@eecs.umich.eduimport bisect
333752Sgblack@eecs.umich.eduimport imp
343752Sgblack@eecs.umich.eduimport marshal
353752Sgblack@eecs.umich.eduimport os
367741Sgblack@eecs.umich.eduimport re
377741Sgblack@eecs.umich.eduimport sys
387741Sgblack@eecs.umich.eduimport zlib
397741Sgblack@eecs.umich.edu
407741Sgblack@eecs.umich.edufrom os.path import basename, dirname, exists, isdir, isfile, join as joinpath
413752Sgblack@eecs.umich.edu
427741Sgblack@eecs.umich.eduimport SCons
437741Sgblack@eecs.umich.edu
447741Sgblack@eecs.umich.edu# This file defines how to build a particular configuration of gem5
457741Sgblack@eecs.umich.edu# based on variable settings in the 'env' build environment.
463752Sgblack@eecs.umich.edu
477741Sgblack@eecs.umich.eduImport('*')
487741Sgblack@eecs.umich.edu
497741Sgblack@eecs.umich.edu# Children need to see the environment
507741Sgblack@eecs.umich.eduExport('env')
517741Sgblack@eecs.umich.edu
527741Sgblack@eecs.umich.edubuild_env = [(opt, env[opt]) for opt in export_vars]
533752Sgblack@eecs.umich.edu
543752Sgblack@eecs.umich.edufrom m5.util import code_formatter
557741Sgblack@eecs.umich.edu
56########################################################################
57# Code for adding source files of various types
58#
59# When specifying a source file of some type, a set of guards can be
60# specified for that file.  When get() is used to find the files, if
61# get specifies a set of filters, only files that match those filters
62# will be accepted (unspecified filters on files are assumed to be
63# false).  Current filters are:
64#     main -- specifies the gem5 main() function
65#     skip_lib -- do not put this file into the gem5 library
66#     <unittest> -- unit tests use filters based on the unit test name
67#
68# A parent can now be specified for a source file and default filter
69# values will be retrieved recursively from parents (children override
70# parents).
71#
72class SourceMeta(type):
73    '''Meta class for source files that keeps track of all files of a
74    particular type and has a get function for finding all functions
75    of a certain type that match a set of guards'''
76    def __init__(cls, name, bases, dict):
77        super(SourceMeta, cls).__init__(name, bases, dict)
78        cls.all = []
79        
80    def get(cls, **guards):
81        '''Find all files that match the specified guards.  If a source
82        file does not specify a flag, the default is False'''
83        for src in cls.all:
84            for flag,value in guards.iteritems():
85                # if the flag is found and has a different value, skip
86                # this file
87                if src.all_guards.get(flag, False) != value:
88                    break
89            else:
90                yield src
91
92class SourceFile(object):
93    '''Base object that encapsulates the notion of a source file.
94    This includes, the source node, target node, various manipulations
95    of those.  A source file also specifies a set of guards which
96    describing which builds the source file applies to.  A parent can
97    also be specified to get default guards from'''
98    __metaclass__ = SourceMeta
99    def __init__(self, source, parent=None, **guards):
100        self.guards = guards
101        self.parent = parent
102
103        tnode = source
104        if not isinstance(source, SCons.Node.FS.File):
105            tnode = File(source)
106
107        self.tnode = tnode
108        self.snode = tnode.srcnode()
109
110        for base in type(self).__mro__:
111            if issubclass(base, SourceFile):
112                base.all.append(self)
113
114    @property
115    def filename(self):
116        return str(self.tnode)
117
118    @property
119    def dirname(self):
120        return dirname(self.filename)
121
122    @property
123    def basename(self):
124        return basename(self.filename)
125
126    @property
127    def extname(self):
128        index = self.basename.rfind('.')
129        if index <= 0:
130            # dot files aren't extensions
131            return self.basename, None
132
133        return self.basename[:index], self.basename[index+1:]
134
135    @property
136    def all_guards(self):
137        '''find all guards for this object getting default values
138        recursively from its parents'''
139        guards = {}
140        if self.parent:
141            guards.update(self.parent.guards)
142        guards.update(self.guards)
143        return guards
144
145    def __lt__(self, other): return self.filename < other.filename
146    def __le__(self, other): return self.filename <= other.filename
147    def __gt__(self, other): return self.filename > other.filename
148    def __ge__(self, other): return self.filename >= other.filename
149    def __eq__(self, other): return self.filename == other.filename
150    def __ne__(self, other): return self.filename != other.filename
151        
152class Source(SourceFile):
153    '''Add a c/c++ source file to the build'''
154    def __init__(self, source, Werror=True, swig=False, **guards):
155        '''specify the source file, and any guards'''
156        super(Source, self).__init__(source, **guards)
157
158        self.Werror = Werror
159        self.swig = swig
160
161class PySource(SourceFile):
162    '''Add a python source file to the named package'''
163    invalid_sym_char = re.compile('[^A-z0-9_]')
164    modules = {}
165    tnodes = {}
166    symnames = {}
167    
168    def __init__(self, package, source, **guards):
169        '''specify the python package, the source file, and any guards'''
170        super(PySource, self).__init__(source, **guards)
171
172        modname,ext = self.extname
173        assert ext == 'py'
174
175        if package:
176            path = package.split('.')
177        else:
178            path = []
179
180        modpath = path[:]
181        if modname != '__init__':
182            modpath += [ modname ]
183        modpath = '.'.join(modpath)
184
185        arcpath = path + [ self.basename ]
186        abspath = self.snode.abspath
187        if not exists(abspath):
188            abspath = self.tnode.abspath
189
190        self.package = package
191        self.modname = modname
192        self.modpath = modpath
193        self.arcname = joinpath(*arcpath)
194        self.abspath = abspath
195        self.compiled = File(self.filename + 'c')
196        self.cpp = File(self.filename + '.cc')
197        self.symname = PySource.invalid_sym_char.sub('_', modpath)
198
199        PySource.modules[modpath] = self
200        PySource.tnodes[self.tnode] = self
201        PySource.symnames[self.symname] = self
202
203class SimObject(PySource):
204    '''Add a SimObject python file as a python source object and add
205    it to a list of sim object modules'''
206
207    fixed = False
208    modnames = []
209
210    def __init__(self, source, **guards):
211        '''Specify the source file and any guards (automatically in
212        the m5.objects package)'''
213        super(SimObject, self).__init__('m5.objects', source, **guards)
214        if self.fixed:
215            raise AttributeError, "Too late to call SimObject now."
216
217        bisect.insort_right(SimObject.modnames, self.modname)
218
219class SwigSource(SourceFile):
220    '''Add a swig file to build'''
221
222    def __init__(self, package, source, **guards):
223        '''Specify the python package, the source file, and any guards'''
224        super(SwigSource, self).__init__(source, **guards)
225
226        modname,ext = self.extname
227        assert ext == 'i'
228
229        self.module = modname
230        cc_file = joinpath(self.dirname, modname + '_wrap.cc')
231        py_file = joinpath(self.dirname, modname + '.py')
232
233        self.cc_source = Source(cc_file, swig=True, parent=self)
234        self.py_source = PySource(package, py_file, parent=self)
235
236class UnitTest(object):
237    '''Create a UnitTest'''
238
239    all = []
240    def __init__(self, target, *sources):
241        '''Specify the target name and any sources.  Sources that are
242        not SourceFiles are evalued with Source().  All files are
243        guarded with a guard of the same name as the UnitTest
244        target.'''
245
246        srcs = []
247        for src in sources:
248            if not isinstance(src, SourceFile):
249                src = Source(src, skip_lib=True)
250            src.guards[target] = True
251            srcs.append(src)
252
253        self.sources = srcs
254        self.target = target
255        UnitTest.all.append(self)
256
257# Children should have access
258Export('Source')
259Export('PySource')
260Export('SimObject')
261Export('SwigSource')
262Export('UnitTest')
263
264########################################################################
265#
266# Debug Flags
267#
268debug_flags = {}
269def DebugFlag(name, desc=None):
270    if name in debug_flags:
271        raise AttributeError, "Flag %s already specified" % name
272    debug_flags[name] = (name, (), desc)
273
274def CompoundFlag(name, flags, desc=None):
275    if name in debug_flags:
276        raise AttributeError, "Flag %s already specified" % name
277
278    compound = tuple(flags)
279    debug_flags[name] = (name, compound, desc)
280
281Export('DebugFlag')
282Export('CompoundFlag')
283
284########################################################################
285#
286# Set some compiler variables
287#
288
289# Include file paths are rooted in this directory.  SCons will
290# automatically expand '.' to refer to both the source directory and
291# the corresponding build directory to pick up generated include
292# files.
293env.Append(CPPPATH=Dir('.'))
294
295for extra_dir in extras_dir_list:
296    env.Append(CPPPATH=Dir(extra_dir))
297
298# Workaround for bug in SCons version > 0.97d20071212
299# Scons bug id: 2006 gem5 Bug id: 308
300for root, dirs, files in os.walk(base_dir, topdown=True):
301    Dir(root[len(base_dir) + 1:])
302
303########################################################################
304#
305# Walk the tree and execute all SConscripts in subdirectories
306#
307
308here = Dir('.').srcnode().abspath
309for root, dirs, files in os.walk(base_dir, topdown=True):
310    if root == here:
311        # we don't want to recurse back into this SConscript
312        continue
313
314    if 'SConscript' in files:
315        build_dir = joinpath(env['BUILDDIR'], root[len(base_dir) + 1:])
316        SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
317
318for extra_dir in extras_dir_list:
319    prefix_len = len(dirname(extra_dir)) + 1
320    for root, dirs, files in os.walk(extra_dir, topdown=True):
321        # if build lives in the extras directory, don't walk down it
322        if 'build' in dirs:
323            dirs.remove('build')
324
325        if 'SConscript' in files:
326            build_dir = joinpath(env['BUILDDIR'], root[prefix_len:])
327            SConscript(joinpath(root, 'SConscript'), variant_dir=build_dir)
328
329for opt in export_vars:
330    env.ConfigFile(opt)
331
332def makeTheISA(source, target, env):
333    isas = [ src.get_contents() for src in source ]
334    target_isa = env['TARGET_ISA']
335    def define(isa):
336        return isa.upper() + '_ISA'
337    
338    def namespace(isa):
339        return isa[0].upper() + isa[1:].lower() + 'ISA' 
340
341
342    code = code_formatter()
343    code('''\
344#ifndef __CONFIG_THE_ISA_HH__
345#define __CONFIG_THE_ISA_HH__
346
347''')
348
349    for i,isa in enumerate(isas):
350        code('#define $0 $1', define(isa), i + 1)
351
352    code('''
353
354#define THE_ISA ${{define(target_isa)}}
355#define TheISA ${{namespace(target_isa)}}
356
357#endif // __CONFIG_THE_ISA_HH__''')
358
359    code.write(str(target[0]))
360
361env.Command('config/the_isa.hh', map(Value, all_isa_list),
362            MakeAction(makeTheISA, Transform("CFG ISA", 0)))
363
364########################################################################
365#
366# Prevent any SimObjects from being added after this point, they
367# should all have been added in the SConscripts above
368#
369SimObject.fixed = True
370
371class DictImporter(object):
372    '''This importer takes a dictionary of arbitrary module names that
373    map to arbitrary filenames.'''
374    def __init__(self, modules):
375        self.modules = modules
376        self.installed = set()
377
378    def __del__(self):
379        self.unload()
380
381    def unload(self):
382        import sys
383        for module in self.installed:
384            del sys.modules[module]
385        self.installed = set()
386
387    def find_module(self, fullname, path):
388        if fullname == 'm5.defines':
389            return self
390
391        if fullname == 'm5.objects':
392            return self
393
394        if fullname.startswith('m5.internal'):
395            return None
396
397        source = self.modules.get(fullname, None)
398        if source is not None and fullname.startswith('m5.objects'):
399            return self
400
401        return None
402
403    def load_module(self, fullname):
404        mod = imp.new_module(fullname)
405        sys.modules[fullname] = mod
406        self.installed.add(fullname)
407
408        mod.__loader__ = self
409        if fullname == 'm5.objects':
410            mod.__path__ = fullname.split('.')
411            return mod
412
413        if fullname == 'm5.defines':
414            mod.__dict__['buildEnv'] = m5.util.SmartDict(build_env)
415            return mod
416
417        source = self.modules[fullname]
418        if source.modname == '__init__':
419            mod.__path__ = source.modpath
420        mod.__file__ = source.abspath
421
422        exec file(source.abspath, 'r') in mod.__dict__
423
424        return mod
425
426import m5.SimObject
427import m5.params
428from m5.util import code_formatter
429
430m5.SimObject.clear()
431m5.params.clear()
432
433# install the python importer so we can grab stuff from the source
434# tree itself.  We can't have SimObjects added after this point or
435# else we won't know about them for the rest of the stuff.
436importer = DictImporter(PySource.modules)
437sys.meta_path[0:0] = [ importer ]
438
439# import all sim objects so we can populate the all_objects list
440# make sure that we're working with a list, then let's sort it
441for modname in SimObject.modnames:
442    exec('from m5.objects import %s' % modname)
443
444# we need to unload all of the currently imported modules so that they
445# will be re-imported the next time the sconscript is run
446importer.unload()
447sys.meta_path.remove(importer)
448
449sim_objects = m5.SimObject.allClasses
450all_enums = m5.params.allEnums
451
452all_params = {}
453for name,obj in sorted(sim_objects.iteritems()):
454    for param in obj._params.local.values():
455        # load the ptype attribute now because it depends on the
456        # current version of SimObject.allClasses, but when scons
457        # actually uses the value, all versions of
458        # SimObject.allClasses will have been loaded
459        param.ptype
460
461        if not hasattr(param, 'swig_decl'):
462            continue
463        pname = param.ptype_str
464        if pname not in all_params:
465            all_params[pname] = param
466
467########################################################################
468#
469# calculate extra dependencies
470#
471module_depends = ["m5", "m5.SimObject", "m5.params"]
472depends = [ PySource.modules[dep].snode for dep in module_depends ]
473
474########################################################################
475#
476# Commands for the basic automatically generated python files
477#
478
479# Generate Python file containing a dict specifying the current
480# buildEnv flags.
481def makeDefinesPyFile(target, source, env):
482    build_env = source[0].get_contents()
483
484    code = code_formatter()
485    code("""
486import m5.internal
487import m5.util
488
489buildEnv = m5.util.SmartDict($build_env)
490
491compileDate = m5.internal.core.compileDate
492_globals = globals()
493for key,val in m5.internal.core.__dict__.iteritems():
494    if key.startswith('flag_'):
495        flag = key[5:]
496        _globals[flag] = val
497del _globals
498""")
499    code.write(target[0].abspath)
500
501defines_info = Value(build_env)
502# Generate a file with all of the compile options in it
503env.Command('python/m5/defines.py', defines_info,
504            MakeAction(makeDefinesPyFile, Transform("DEFINES", 0)))
505PySource('m5', 'python/m5/defines.py')
506
507# Generate python file containing info about the M5 source code
508def makeInfoPyFile(target, source, env):
509    code = code_formatter()
510    for src in source:
511        data = ''.join(file(src.srcnode().abspath, 'r').xreadlines())
512        code('$src = ${{repr(data)}}')
513    code.write(str(target[0]))
514
515# Generate a file that wraps the basic top level files
516env.Command('python/m5/info.py',
517            [ '#/COPYING', '#/LICENSE', '#/README', ],
518            MakeAction(makeInfoPyFile, Transform("INFO")))
519PySource('m5', 'python/m5/info.py')
520
521########################################################################
522#
523# Create all of the SimObject param headers and enum headers
524#
525
526def createSimObjectParam(target, source, env):
527    assert len(target) == 1 and len(source) == 1
528
529    name = str(source[0].get_contents())
530    obj = sim_objects[name]
531
532    code = code_formatter()
533    obj.cxx_decl(code)
534    code.write(target[0].abspath)
535
536def createSwigParam(target, source, env):
537    assert len(target) == 1 and len(source) == 1
538
539    name = str(source[0].get_contents())
540    param = all_params[name]
541
542    code = code_formatter()
543    code('%module(package="m5.internal") $0_${name}', param.file_ext)
544    param.swig_decl(code)
545    code.write(target[0].abspath)
546
547def createEnumStrings(target, source, env):
548    assert len(target) == 1 and len(source) == 1
549
550    name = str(source[0].get_contents())
551    obj = all_enums[name]
552
553    code = code_formatter()
554    obj.cxx_def(code)
555    code.write(target[0].abspath)
556
557def createEnumParam(target, source, env):
558    assert len(target) == 1 and len(source) == 1
559
560    name = str(source[0].get_contents())
561    obj = all_enums[name]
562
563    code = code_formatter()
564    obj.cxx_decl(code)
565    code.write(target[0].abspath)
566
567def createEnumSwig(target, source, env):
568    assert len(target) == 1 and len(source) == 1
569
570    name = str(source[0].get_contents())
571    obj = all_enums[name]
572
573    code = code_formatter()
574    code('''\
575%module(package="m5.internal") enum_$name
576
577%{
578#include "enums/$name.hh"
579%}
580
581%include "enums/$name.hh"
582''')
583    code.write(target[0].abspath)
584
585# Generate all of the SimObject param struct header files
586params_hh_files = []
587for name,simobj in sorted(sim_objects.iteritems()):
588    py_source = PySource.modules[simobj.__module__]
589    extra_deps = [ py_source.tnode ]
590
591    hh_file = File('params/%s.hh' % name)
592    params_hh_files.append(hh_file)
593    env.Command(hh_file, Value(name),
594                MakeAction(createSimObjectParam, Transform("SO PARAM")))
595    env.Depends(hh_file, depends + extra_deps)
596
597# Generate any parameter header files needed
598params_i_files = []
599for name,param in all_params.iteritems():
600    i_file = File('python/m5/internal/%s_%s.i' % (param.file_ext, name))
601    params_i_files.append(i_file)
602    env.Command(i_file, Value(name),
603                MakeAction(createSwigParam, Transform("SW PARAM")))
604    env.Depends(i_file, depends)
605    SwigSource('m5.internal', i_file)
606
607# Generate all enum header files
608for name,enum in sorted(all_enums.iteritems()):
609    py_source = PySource.modules[enum.__module__]
610    extra_deps = [ py_source.tnode ]
611
612    cc_file = File('enums/%s.cc' % name)
613    env.Command(cc_file, Value(name),
614                MakeAction(createEnumStrings, Transform("ENUM STR")))
615    env.Depends(cc_file, depends + extra_deps)
616    Source(cc_file)
617
618    hh_file = File('enums/%s.hh' % name)
619    env.Command(hh_file, Value(name),
620                MakeAction(createEnumParam, Transform("EN PARAM")))
621    env.Depends(hh_file, depends + extra_deps)
622
623    i_file = File('python/m5/internal/enum_%s.i' % name)
624    env.Command(i_file, Value(name),
625                MakeAction(createEnumSwig, Transform("ENUMSWIG")))
626    env.Depends(i_file, depends + extra_deps)
627    SwigSource('m5.internal', i_file)
628
629def buildParam(target, source, env):
630    name = source[0].get_contents()
631    obj = sim_objects[name]
632    class_path = obj.cxx_class.split('::')
633    classname = class_path[-1]
634    namespaces = class_path[:-1]
635    params = obj._params.local.values()
636
637    code = code_formatter()
638
639    code('%module(package="m5.internal") param_$name')
640    code()
641    code('%{')
642    code('#include "params/$obj.hh"')
643    for param in params:
644        param.cxx_predecls(code)
645    code('%}')
646    code()
647
648    for param in params:
649        param.swig_predecls(code)
650
651    code()
652    if obj._base:
653        code('%import "python/m5/internal/param_${{obj._base}}.i"')
654    code()
655    obj.swig_objdecls(code)
656    code()
657
658    code('%include "params/$obj.hh"')
659
660    code.write(target[0].abspath)
661
662for name in sim_objects.iterkeys():
663    params_file = File('python/m5/internal/param_%s.i' % name)
664    env.Command(params_file, Value(name),
665                MakeAction(buildParam, Transform("BLDPARAM")))
666    env.Depends(params_file, depends)
667    SwigSource('m5.internal', params_file)
668
669# Generate the main swig init file
670def makeEmbeddedSwigInit(target, source, env):
671    code = code_formatter()
672    module = source[0].get_contents()
673    code('''\
674#include "sim/init.hh"
675
676extern "C" {
677    void init_${module}();
678}
679
680EmbeddedSwig embed_swig_${module}(init_${module});
681''')
682    code.write(str(target[0]))
683    
684# Build all swig modules
685for swig in SwigSource.all:
686    env.Command([swig.cc_source.tnode, swig.py_source.tnode], swig.tnode,
687                MakeAction('$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} '
688                '-o ${TARGETS[0]} $SOURCES', Transform("SWIG")))
689    cc_file = str(swig.tnode)
690    init_file = '%s/init_%s.cc' % (dirname(cc_file), basename(cc_file))
691    env.Command(init_file, Value(swig.module),
692                MakeAction(makeEmbeddedSwigInit, Transform("EMBED SW")))
693    Source(init_file, **swig.guards)
694
695#
696# Handle debug flags
697#
698def makeDebugFlagCC(target, source, env):
699    assert(len(target) == 1 and len(source) == 1)
700
701    val = eval(source[0].get_contents())
702    name, compound, desc = val
703    compound = list(sorted(compound))
704
705    code = code_formatter()
706
707    # file header
708    code('''
709/*
710 * DO NOT EDIT THIS FILE! Automatically generated
711 */
712
713#include "base/debug.hh"
714''')
715
716    for flag in compound:
717        code('#include "debug/$flag.hh"')
718    code()
719    code('namespace Debug {')
720    code()
721
722    if not compound:
723        code('SimpleFlag $name("$name", "$desc");')
724    else:
725        code('CompoundFlag $name("$name", "$desc",')
726        code.indent()
727        last = len(compound) - 1
728        for i,flag in enumerate(compound):
729            if i != last:
730                code('$flag,')
731            else:
732                code('$flag);')
733        code.dedent()
734
735    code()
736    code('} // namespace Debug')
737
738    code.write(str(target[0]))
739
740def makeDebugFlagHH(target, source, env):
741    assert(len(target) == 1 and len(source) == 1)
742
743    val = eval(source[0].get_contents())
744    name, compound, desc = val
745
746    code = code_formatter()
747
748    # file header boilerplate
749    code('''\
750/*
751 * DO NOT EDIT THIS FILE!
752 *
753 * Automatically generated by SCons
754 */
755
756#ifndef __DEBUG_${name}_HH__
757#define __DEBUG_${name}_HH__
758
759namespace Debug {
760''')
761
762    if compound:
763        code('class CompoundFlag;')
764    code('class SimpleFlag;')
765
766    if compound:
767        code('extern CompoundFlag $name;')
768        for flag in compound:
769            code('extern SimpleFlag $flag;')
770    else:
771        code('extern SimpleFlag $name;')
772
773    code('''
774}
775
776#endif // __DEBUG_${name}_HH__
777''')
778
779    code.write(str(target[0]))
780
781for name,flag in sorted(debug_flags.iteritems()):
782    n, compound, desc = flag
783    assert n == name
784
785    env.Command('debug/%s.hh' % name, Value(flag),
786                MakeAction(makeDebugFlagHH, Transform("TRACING", 0)))
787    env.Command('debug/%s.cc' % name, Value(flag),
788                MakeAction(makeDebugFlagCC, Transform("TRACING", 0)))
789    Source('debug/%s.cc' % name)
790
791# Embed python files.  All .py files that have been indicated by a
792# PySource() call in a SConscript need to be embedded into the M5
793# library.  To do that, we compile the file to byte code, marshal the
794# byte code, compress it, and then generate a c++ file that
795# inserts the result into an array.
796def embedPyFile(target, source, env):
797    def c_str(string):
798        if string is None:
799            return "0"
800        return '"%s"' % string
801
802    '''Action function to compile a .py into a code object, marshal
803    it, compress it, and stick it into an asm file so the code appears
804    as just bytes with a label in the data section'''
805
806    src = file(str(source[0]), 'r').read()
807
808    pysource = PySource.tnodes[source[0]]
809    compiled = compile(src, pysource.abspath, 'exec')
810    marshalled = marshal.dumps(compiled)
811    compressed = zlib.compress(marshalled)
812    data = compressed
813    sym = pysource.symname
814
815    code = code_formatter()
816    code('''\
817#include "sim/init.hh"
818
819namespace {
820
821const char data_${sym}[] = {
822''')
823    code.indent()
824    step = 16
825    for i in xrange(0, len(data), step):
826        x = array.array('B', data[i:i+step])
827        code(''.join('%d,' % d for d in x))
828    code.dedent()
829    
830    code('''};
831
832EmbeddedPython embedded_${sym}(
833    ${{c_str(pysource.arcname)}},
834    ${{c_str(pysource.abspath)}},
835    ${{c_str(pysource.modpath)}},
836    data_${sym},
837    ${{len(data)}},
838    ${{len(marshalled)}});
839
840} // anonymous namespace
841''')
842    code.write(str(target[0]))
843
844for source in PySource.all:
845    env.Command(source.cpp, source.tnode, 
846                MakeAction(embedPyFile, Transform("EMBED PY")))
847    Source(source.cpp)
848
849########################################################################
850#
851# Define binaries.  Each different build type (debug, opt, etc.) gets
852# a slightly different build environment.
853#
854
855# List of constructed environments to pass back to SConstruct
856envList = []
857
858date_source = Source('base/date.cc', skip_lib=True)
859
860# Function to create a new build environment as clone of current
861# environment 'env' with modified object suffix and optional stripped
862# binary.  Additional keyword arguments are appended to corresponding
863# build environment vars.
864def makeEnv(label, objsfx, strip = False, **kwargs):
865    # SCons doesn't know to append a library suffix when there is a '.' in the
866    # name.  Use '_' instead.
867    libname = 'gem5_' + label
868    exename = 'gem5.' + label
869    secondary_exename = 'm5.' + label
870
871    new_env = env.Clone(OBJSUFFIX=objsfx, SHOBJSUFFIX=objsfx + 's')
872    new_env.Label = label
873    new_env.Append(**kwargs)
874
875    swig_env = new_env.Clone()
876    swig_env.Append(CCFLAGS='-Werror')
877    if env['GCC']:
878        swig_env.Append(CCFLAGS='-Wno-uninitialized')
879        swig_env.Append(CCFLAGS='-Wno-sign-compare')
880        swig_env.Append(CCFLAGS='-Wno-parentheses')
881
882    werror_env = new_env.Clone()
883    werror_env.Append(CCFLAGS='-Werror')
884
885    def make_obj(source, static, extra_deps = None):
886        '''This function adds the specified source to the correct
887        build environment, and returns the corresponding SCons Object
888        nodes'''
889
890        if source.swig:
891            env = swig_env
892        elif source.Werror:
893            env = werror_env
894        else:
895            env = new_env
896
897        if static:
898            obj = env.StaticObject(source.tnode)
899        else:
900            obj = env.SharedObject(source.tnode)
901
902        if extra_deps:
903            env.Depends(obj, extra_deps)
904
905        return obj
906
907    sources = Source.get(main=False, skip_lib=False)
908    static_objs = [ make_obj(s, True) for s in sources ]
909    shared_objs = [ make_obj(s, False) for s in sources ]
910
911    static_date = make_obj(date_source, static=True, extra_deps=static_objs)
912    static_objs.append(static_date)
913    
914    shared_date = make_obj(date_source, static=False, extra_deps=shared_objs)
915    shared_objs.append(shared_date)
916
917    # First make a library of everything but main() so other programs can
918    # link against m5.
919    static_lib = new_env.StaticLibrary(libname, static_objs)
920    shared_lib = new_env.SharedLibrary(libname, shared_objs)
921
922    # Now link a stub with main() and the static library.
923    main_objs = [ make_obj(s, True) for s in Source.get(main=True) ]
924
925    for test in UnitTest.all:
926        flags = { test.target : True }
927        test_sources = Source.get(**flags)
928        test_objs = [ make_obj(s, static=True) for s in test_sources ]
929        testname = "unittest/%s.%s" % (test.target, label)
930        new_env.Program(testname, main_objs + test_objs + static_objs)
931
932    progname = exename
933    if strip:
934        progname += '.unstripped'
935
936    targets = new_env.Program(progname, main_objs + static_objs)
937
938    if strip:
939        if sys.platform == 'sunos5':
940            cmd = 'cp $SOURCE $TARGET; strip $TARGET'
941        else:
942            cmd = 'strip $SOURCE -o $TARGET'
943        targets = new_env.Command(exename, progname,
944                    MakeAction(cmd, Transform("STRIP")))
945
946    new_env.Command(secondary_exename, exename,
947            MakeAction('ln $SOURCE $TARGET', Transform("HARDLINK")))
948
949    new_env.M5Binary = targets[0]
950    envList.append(new_env)
951
952# Debug binary
953ccflags = {}
954if env['GCC']:
955    if sys.platform == 'sunos5':
956        ccflags['debug'] = '-gstabs+'
957    else:
958        ccflags['debug'] = '-ggdb3'
959    ccflags['opt'] = '-g -O3'
960    ccflags['fast'] = '-O3'
961    ccflags['prof'] = '-O3 -g -pg'
962elif env['SUNCC']:
963    ccflags['debug'] = '-g0'
964    ccflags['opt'] = '-g -O'
965    ccflags['fast'] = '-fast'
966    ccflags['prof'] = '-fast -g -pg'
967elif env['ICC']:
968    ccflags['debug'] = '-g -O0'
969    ccflags['opt'] = '-g -O'
970    ccflags['fast'] = '-fast'
971    ccflags['prof'] = '-fast -g -pg'
972else:
973    print 'Unknown compiler, please fix compiler options'
974    Exit(1)
975
976makeEnv('debug', '.do',
977        CCFLAGS = Split(ccflags['debug']),
978        CPPDEFINES = ['DEBUG', 'TRACING_ON=1'])
979
980# Optimized binary
981makeEnv('opt', '.o',
982        CCFLAGS = Split(ccflags['opt']),
983        CPPDEFINES = ['TRACING_ON=1'])
984
985# "Fast" binary
986makeEnv('fast', '.fo', strip = True,
987        CCFLAGS = Split(ccflags['fast']),
988        CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'])
989
990# Profiled binary
991makeEnv('prof', '.po',
992        CCFLAGS = Split(ccflags['prof']),
993        CPPDEFINES = ['NDEBUG', 'TRACING_ON=0'],
994        LINKFLAGS = '-pg')
995
996Return('envList')
997