source: anuga_core/SConstruct @ 3599

Last change on this file since 3599 was 3599, checked in by jack, 18 years ago

Consolidated the SConscripts into the SConstruct.
File size: 7.1 KB
Line 
1import os
2import sys
3import py_compile
4
5# Function to build a .pyc from a .py
6def build_pyc(target, source, env):
7    python = sys.executable
8    command = "import py_compile; py_compile.compile('%s', doraise=True)"
9    if sys.platform == 'win32':
10        command %= str(source[0]).replace('\\', '\\\\')
11    else:
12        command %= str(source[0])
13    rv = os.system('%s -c "%s"' % (python, command))
14    if not rv == 0:
15        raise SyntaxError, "Could not compile %s" % str(source[0])
16    return None
17
18# Function to build a .pyo from a .py
19def build_pyo(target, source, env):
20    python = sys.executable
21    command = "import py_compile; py_compile.compile('%s', doraise=True)"
22    if sys.platform == 'win32':
23        command %= str(source[0]).replace('\\', '\\\\')
24    else:
25        command %= str(source[0])
26    options = '-' + 'O' * env['OPTIMISATION_LEVEL']
27    rv = os.system('%s %s -c "%s"' % (python, options, command))
28    if not rv == 0:
29        raise SyntaxError, "Could not compile %s" % str(source[0])
30    return None
31
32# Create the builders
33pyc_builder = Builder(action = build_pyc,
34                      suffix = '.pyc',
35                      src_suffix = '.py')
36pyo_builder = Builder(action = build_pyo,
37                      suffix = '.pyo',
38                      src_suffix = '.py')
39
40# Read in build options
41opts = Options('build_options.py')
42opts.Add('GCCFLAGS', 'Flags passed to GCC')
43opts.Add('MSVCFLAGS', 'Flags passed to the MSVC compiler')
44opts.Add('METIS_DIR', 'Location of the metis directory relative to the pymetis directory')
45opts.Add('OPTIMISATION_LEVEL', 'Optimisation level for the building of .pyo files. Must be 1 or 2')
46
47# Windows' site packages are in a different location to those on Linux
48if sys.platform == 'win32':
49    opts.Add(PathOption('PREFIX',
50                        'Location to install compiled sources',
51                        os.path.join(sys.exec_prefix, 'lib', 'site-packages')))
52else:
53    opts.Add(PathOption('PREFIX',
54                        'Location to install compiled sources',
55                        os.path.join(sys.exec_prefix, 'lib', 'python' + sys.version[:3], 'site-packages')))
56
57opts.Add(BoolOption('INSTALL_PYTHON_SOURCE',
58                    'Install the .py files as well as the .pyc/.pyo files',
59                    0))
60
61env = Environment(options = opts)
62env.Append(BUILDERS={'Pyc' : pyc_builder,
63                     'Pyo' : pyo_builder})
64
65Help(opts.GenerateHelpText(env))
66
67if not (env['OPTIMISATION_LEVEL'] == 1 or env['OPTIMISATION_LEVEL'] == 2):
68    raise ValueError, "OPTIMISATION_LEVEL must be between 1 and 2 inclusive"
69
70# Where to find the Python.h
71if sys.platform == 'win32':
72    # Prefer MinGW over MSVC
73    Tool('mingw')(env)
74
75    python_include = os.path.join(sys.exec_prefix, 'include')
76    # Windows installs need to be told the lib path and the python library name
77    # else it won't work right.
78    python_libdir = os.path.join(sys.exec_prefix, 'libs')
79    env.Append(LIBPATH=[python_libdir])
80    python_libname = 'python%d%d' % (sys.version_info[0:2])
81    env.Append(LIBS=[python_libname])
82else:
83    python_include = os.path.join(sys.exec_prefix, 'include', 'python' + sys.version[:3])
84   
85# Check existence of Python.h
86headerfile = python_include + os.sep + 'Python.h'
87try:
88    open(headerfile, 'r')
89except:
90    raise """Did not find Python header file %s.
91    Make sure files for Python C-extensions are installed.
92    In debian linux, for example, you need to install a
93    package called something like python2.3-dev""" %headerfile
94
95env.Append(CPPPATH=[python_include])
96
97# Set appropriate CCFLAGS for the compiler.
98if env['CC'] == 'gcc':
99    env.Append(CCFLAGS=['${GCCFLAGS}'])
100elif env['CC'] == 'cl':
101    env.Append(CCFLAGS=['${MSVCFLAGS}'])
102
103anuga_root = os.path.join('source', 'anuga')
104install_root = os.path.join(env['PREFIX'], 'anuga')
105
106# Build .pyc and .pyo files of every .py in here and below.
107files = []
108#dirs = filter(os.path.isdir, os.listdir('.'))
109#Only build the source/anuga directory for now
110dirs = [anuga_root]
111while(dirs != []):
112    dirs += filter(os.path.isdir, map(lambda x : os.path.join(dirs[0], x), os.listdir(dirs[0])))
113    files += filter(lambda x : x[-3:] == '.py', map(lambda x : os.path.join(dirs[0], x), os.listdir(dirs[0])))
114    dirs = dirs[1:]
115for x in files:
116    env.Pyc(x + 'c', x)
117    env.Pyo(x + 'o', x)
118    # We have to cut the first character off the result of os.path.dirname(x).replace('anuga_core, '')
119    # or else we will start taking paths relative to the root directory.
120    instdir = os.path.join(env['PREFIX'], os.path.dirname(x).replace('source', '')[1:])
121    env.Install(instdir, x + 'c')
122    env.Install(instdir, x + 'o')
123    if env['INSTALL_PYTHON_SOURCE']:
124        env.Install(instdir, x)
125
126# Define the install target
127env.Alias('install', '${PREFIX}')
128
129# Mesh_engine
130mesh_env = env.Copy()
131mesh_env.Append(CPPDEFINES=[('TRILIBRARY', 1), ('NO_TIMER', 1)])
132
133mesh_dir = os.path.join(anuga_root, 'mesh_engine')
134mesh_install_dir = os.path.join(install_root, 'mesh_engine')
135env.Install(mesh_install_dir, mesh_env.SharedLibrary(os.path.join(mesh_dir, 'triang'),
136                                            map(lambda s: os.path.join(mesh_dir, s), ['triangle.c', 'triang.c'])))
137env.Install(mesh_install_dir, mesh_env.SharedLibrary(os.path.join(mesh_dir, 'triangle'),
138                                            map(lambda s: os.path.join(mesh_dir, s), ['triangle.c'])))
139
140# Utilities
141util_dir = os.path.join(anuga_root, 'utilities')
142util_install_dir = os.path.join(install_root, 'utilities')
143env.Install(util_install_dir, env.SharedLibrary(os.path.join(util_dir, 'polygon_ext'),
144                                                map(lambda s: os.path.join(util_dir, s), ['polygon_ext.c'])))
145env.Install(util_install_dir, env.SharedLibrary(os.path.join(util_dir, 'util_ext'),
146                                                map(lambda s: os.path.join(util_dir, s), ['util_ext.c'])))
147env.Install(util_install_dir, env.SharedLibrary(os.path.join(util_dir, 'sparse_ext'),
148                                                map(lambda s: os.path.join(util_dir, s), ['sparse_ext.c'])))
149
150# Abstract_2d_finite_volumes
151a2fv_env = env.Copy()
152a2fv_env.Append(CPPPATH=[os.path.join('#', anuga_root, 'utilities')])
153
154a2fv_dir = os.path.join(anuga_root, 'abstract_2d_finite_volumes')
155a2fv_install_dir = os.path.join(install_root, 'abstract_2d_finite_volumes')
156env.Install(a2fv_install_dir, a2fv_env.SharedLibrary(os.path.join(a2fv_dir, 'quantity_ext'),
157                                                     map(lambda s: os.path.join(a2fv_dir, s), ['quantity_ext.c'])))
158env.Install(a2fv_install_dir, a2fv_env.SharedLibrary(os.path.join(a2fv_dir, 'shallow_water_kinetic'),
159                                                     map(lambda s: os.path.join(a2fv_dir, s), ['shallow_water_kinetic_ext.c'])))
160
161# Shallow_water
162sw_env = env.Copy()
163sw_env.Append(CPPPATH=[os.path.join('#', anuga_root, 'utilities')])
164
165sw_dir = os.path.join(anuga_root, 'shallow_water')
166sw_install_dir = os.path.join(install_root, 'shallow_water')
167env.Install(sw_install_dir, sw_env.SharedLibrary(os.path.join(sw_dir, 'shallow_water_ext'),
168                                                 map(lambda s: os.path.join(sw_dir, s), ['shallow_water_ext.c'])))
Note: See TracBrowser for help on using the repository browser.