SConscript revision 3645:2bf1f7c69254
111373Sandreas.sandberg@arm.com# -*- mode:python -*-
211373Sandreas.sandberg@arm.com
311373Sandreas.sandberg@arm.com# Copyright (c) 2004-2005 The Regents of The University of Michigan
411373Sandreas.sandberg@arm.com# All rights reserved.
511373Sandreas.sandberg@arm.com#
611373Sandreas.sandberg@arm.com# Redistribution and use in source and binary forms, with or without
711373Sandreas.sandberg@arm.com# modification, are permitted provided that the following conditions are
811373Sandreas.sandberg@arm.com# met: redistributions of source code must retain the above copyright
912163Szulian@eit.uni-kl.de# notice, this list of conditions and the following disclaimer;
1011373Sandreas.sandberg@arm.com# redistributions in binary form must reproduce the above copyright
1111373Sandreas.sandberg@arm.com# notice, this list of conditions and the following disclaimer in the
1211373Sandreas.sandberg@arm.com# documentation and/or other materials provided with the distribution;
1311373Sandreas.sandberg@arm.com# neither the name of the copyright holders nor the names of its
1411373Sandreas.sandberg@arm.com# contributors may be used to endorse or promote products derived from
1511373Sandreas.sandberg@arm.com# this software without specific prior written permission.
1611373Sandreas.sandberg@arm.com#
1712214Sgabeblack@google.com# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
1812057Sgabeblack@google.com# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
1912057Sgabeblack@google.com# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
2012057Sgabeblack@google.com# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
2112058Sgabeblack@google.com# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
2211864Snikos.nikoleris@arm.com# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
2312163Szulian@eit.uni-kl.de# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
2412163Szulian@eit.uni-kl.de# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
2512163Szulian@eit.uni-kl.de# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
2612163Szulian@eit.uni-kl.de# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28#
29# Authors: Steve Reinhardt
30#          Nathan Binkert
31
32import os, os.path, re, sys
33from zipfile import PyZipFile
34
35# handy function for path joins
36def join(*args):
37    return os.path.normpath(os.path.join(*args))
38
39Import('env')
40
41# This SConscript is in charge of collecting .py files and generating
42# a zip archive that is appended to the m5 binary.
43
44# List of files & directories to include in the zip file.  To include
45# a package, list only the root directory of the package, not any
46# internal .py files (else they will get the path stripped off when
47# they are imported into the zip file).
48pyzip_files = []
49
50# List of additional files on which the zip archive depends, but which
51# are not included in pyzip_files... i.e. individual .py files within
52# a package.
53pyzip_dep_files = []
54
55# Add the specified package to the zip archive.  Adds the directory to
56# pyzip_files and all included .py files to pyzip_dep_files.
57def addPkg(pkgdir):
58    pyzip_files.append(pkgdir)
59    origdir = os.getcwd()
60    srcdir = join(Dir('.').srcnode().abspath, pkgdir)
61    os.chdir(srcdir)
62    for path, dirs, files in os.walk('.'):
63        for i,dir in enumerate(dirs):
64            if dir == 'SCCS':
65                del dirs[i]
66                break
67
68        for f in files:
69            if f.endswith('.py'):
70                pyzip_dep_files.append(join(pkgdir, path, f))
71
72    os.chdir(origdir)
73
74# Generate Python file that contains a dict specifying the current
75# build_env flags.
76def MakeDefinesPyFile(target, source, env):
77    f = file(str(target[0]), 'w')
78    print >>f, "m5_build_env = ", source[0]
79    f.close()
80
81optionDict = dict([(opt, env[opt]) for opt in env.ExportOptions])
82env.Command('m5/defines.py', Value(optionDict), MakeDefinesPyFile)
83
84def MakeInfoPyFile(target, source, env):
85    f = file(str(target[0]), 'w')
86    for src in source:
87        data = ''.join(file(src.srcnode().abspath, 'r').xreadlines())
88        print >>f, "%s = %s" % (src, repr(data))
89    f.close()
90
91env.Command('m5/info.py',
92            [ '#/AUTHORS', '#/LICENSE', '#/README', '#/RELEASE_NOTES' ],
93            MakeInfoPyFile)
94
95# Now specify the packages & files for the zip archive.
96addPkg('m5')
97pyzip_files.append('m5/defines.py')
98pyzip_files.append('m5/info.py')
99pyzip_files.append(join(env['ROOT'], 'util/pbs/jobfile.py'))
100
101env.Command(['swig/debug_wrap.cc', 'm5/internal/debug.py'],
102            'swig/debug.i',
103            '$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} '
104            '-o ${TARGETS[0]} $SOURCES')
105
106env.Command(['swig/main_wrap.cc', 'm5/internal/main.py'],
107            'swig/main.i',
108            '$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} '
109            '-o ${TARGETS[0]} $SOURCES')
110
111pyzip_dep_files.append('m5/internal/debug.py')
112pyzip_dep_files.append('m5/internal/main.py')
113
114# Action function to build the zip archive.  Uses the PyZipFile module
115# included in the standard Python library.
116def buildPyZip(target, source, env):
117    pzf = PyZipFile(str(target[0]), 'w')
118    for s in source:
119        pzf.writepy(str(s))
120
121# Add the zip file target to the environment.
122env.Command('m5py.zip', pyzip_files, buildPyZip)
123env.Depends('m5py.zip', pyzip_dep_files)
124