1# -*- mode:python -*- 2 3# Copyright (c) 2004-2005 The Regents of The University of Michigan 4# All rights reserved. 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions are 8# met: redistributions of source code must retain the above copyright --- 24 unchanged lines hidden (view full) --- 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 = ", |
79 print >>f, source[0] 80 f.close() 81 82optionDict = dict([(opt, env[opt]) for opt in env.ExportOptions]) |
83env.Command('m5/defines.py', Value(optionDict), MakeDefinesPyFile) |
84 85# Now specify the packages & files for the zip archive. 86addPkg('m5') |
87pyzip_files.append('m5/defines.py') |
88pyzip_files.append(join(env['ROOT'], 'util/pbs/jobfile.py')) 89 |
90env.Command(['swig/main_wrap.cc', 'm5/main.py'], 91 'swig/main.i', 92 '$SWIG $SWIGFLAGS -outdir ${TARGETS[1].dir} ' 93 '-o ${TARGETS[0]} $SOURCES') 94 95pyzip_dep_files.append('m5/main.py') 96 |
97# Action function to build the zip archive. Uses the PyZipFile module 98# included in the standard Python library. 99def buildPyZip(target, source, env): 100 pzf = PyZipFile(str(target[0]), 'w') 101 for s in source: 102 pzf.writepy(str(s)) 103 104# Add the zip file target to the environment. 105env.Command('m5py.zip', pyzip_files, buildPyZip) 106env.Depends('m5py.zip', pyzip_dep_files) |