Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F88623724
SConstruct
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sat, Oct 19, 19:51
Size
5 KB
Mime Type
text/x-python
Expires
Mon, Oct 21, 19:51 (1 d, 23 h)
Engine
blob
Format
Raw Data
Handle
21711306
Attached To
rTAMAAS tamaas
SConstruct
View Options
from __future__ import print_function
import os
from os.path import join, abspath
from version import write_info_file
# Compilation colors
colors = {
'cyan': '\033[96m',
'purple': '\033[95m',
'blue': '\033[94m',
'green': '\033[92m',
'yellow': '\033[93m',
'red': '\033[91m',
'end': '\033[0m'
}
# Inherit all environment variables (for CXX detection, etc.)
main_env = Environment(ENV=os.environ)
main_env['COLOR_DICT'] = colors
# Compiler detection
compiler_default = 'g++'
if 'CXX' in os.environ:
compiler_default = os.environ['CXX']
# Build variables
vars = Variables('build-setup.conf')
vars.Add(EnumVariable('build_type', 'Build type', 'release',
allowed_values=('release', 'profiling', 'debug'),
ignorecase=2))
vars.Add('prefix', 'Prefix where to install', '/usr/local')
vars.Add('CXX', 'Compiler', compiler_default)
vars.Add('py_exec', 'Python executable', 'python')
vars.Add(BoolVariable('timer', 'Activate the timer possibilities', False))
vars.Add(BoolVariable('verbose', 'Activate verbosity', False))
vars.Add(BoolVariable('build_doc', 'Build documentation', False))
vars.Add(BoolVariable('color', 'Color the non-verbose compilation output', False))
vars.Update(main_env)
Help(vars.GenerateHelpText(main_env))
# Save all options, not just those that differ from default
with open('build-setup.conf', 'w') as setup:
for key in vars.keys():
setup.write("{} = '{}'\n".format(key, main_env[key]))
build_type = main_env['build_type']
build_dir = 'build-' + main_env['build_type']
print("Building in " + build_dir)
verbose = main_env['verbose']
# Remove colors if not set
if not main_env['color']:
for key in colors:
colors[key] = ''
if not verbose:
main_env['SHCXXCOMSTR'] = u'{0}[Compiling] {1}$SOURCE{2}'.format(colors['green'],
colors['blue'],
colors['end'])
main_env['SHLINKCOMSTR'] = u'{0}[Linking] {1}$TARGET{2}'.format(colors['purple'],
colors['blue'],
colors['end'])
main_env['SWIGCOMSTR'] = u'{0}[Swig] {1}$SOURCE{2}'.format(colors['yellow'],
colors['blue'],
colors['end'])
# Include paths
main_env.AppendUnique(CPPPATH=['#/src',
'#/src/core',
'#/src/surface',
'#/src/bem',
'#/src/python',
'#/python'])
# Treating Intel compiler for OpenMP
if main_env['CXX'] != 'icpc':
omp_libs = ['gomp']
omp_flag = '-fopenmp'
else:
omp_libs = ['']
omp_flag = '-qopenmp'
main_env.AppendUnique(LIBS=omp_libs)
main_env.AppendUnique(LINKFLAGS=[omp_flag])
# Flags and options
main_env.AppendUnique(CXXFLAGS=['-std=c++11',
'-Wall',
omp_flag])
# Adding compile flags defined in evironment
if 'CXXFLAGS' in os.environ:
main_env.AppendUnique(CXXFLAGS=Split(os.environ['CXXFLAGS']))
if main_env['timer']:
main_env.AppendUnique(CPPDEFINES=['USING_TIMER'])
cxxflags_dict = {
"debug": Split("-g -O0 -DTAMAAS_DEBUG"),
"profiling": Split("-g -pg -O2"),
"release": Split("-O3")
}
shlinkflags_dict = {
"debug": [],
"profiling": ['-pg'],
"release": []
}
main_env.AppendUnique(CXXFLAGS=cxxflags_dict[build_type])
main_env.AppendUnique(SHLINKFLAGS=shlinkflags_dict[build_type])
main_env['LIBPATH'] = [abspath(join(build_dir, 'src'))]
main_env['RPATH'] = "$LIBPATH"
fftw_include = ""
fftw_library = ""
# If FFTW is provided by module system (on clusters)
if 'FFTW_ROOT' in main_env['ENV']:
fftw_include = join(main_env['ENV']['FFTW_ROOT'], 'include')
fftw_library = join(main_env['ENV']['FFTW_ROOT'], 'lib')
main_env = main_env.Clone(
tools=[fftw],
FFTW_LIBRARY_WISH=['main', 'omp'],
FFTW_INCLUDE_DIR=fftw_include,
FFTW_LIBRARY_DIR=fftw_library,
)
# Writing information file
write_info_file("src/tamaas_info.cpp")
Export('main_env')
# Compiling sources
SConscript('src/SConscript', variant_dir=join(build_dir, 'src'), duplicate=False)
# Building python wrapper
SConscript('python/SConscript', variant_dir=join(build_dir, 'python'), duplicate=False)
# Copying tests
SConscript('tests/SConscript', variant_dir=join(build_dir, 'tests'), duplicate=False)
# Building documentation
if main_env['build_doc']:
SConscript('doc/SConscript', variant_dir=join(build_dir, 'doc'), duplicate=False)
# Saving the env file
env_content = """export PYTHONPATH=$PYTHONPATH:{0}/python
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:{0}/src
"""
def write_env_file(target, source, env):
with open(str(target[0]), 'w') as env_file:
env_file.write(env_content.format(abspath(build_dir)))
def print_write_env_file(command, target, source, env):
colors = env['COLOR_DICT']
print("{}[Writing] {}{}{}".format(colors['cyan'],
colors['blue'],
target[0],
colors['end']))
env_file_env = main_env.Clone(PRINT_CMD_LINE_FUNC=print_write_env_file)
# Need to have a command and manage tamaas_environement.sh as target because
# the build directory does not always exist
env_file_env.Command(join(build_dir, 'tamaas_environement.sh'), None, write_env_file)
Event Timeline
Log In to Comment