from __future__ import print_function

import os

import re

import sys

import shutil

import platform

import argparse

from glob import glob

from collections import OrderedDict

jp = os.path.join

is_win = (platform.system() == 'Windows')

is_lin = (platform.system() == 'Linux')

is_mac = (platform.system() == 'Darwin')

default_prefix = os.getenv('PREFIX', 'install_prefix')

if is_win:

default_prefix = jp(default_prefix, 'Library') # conda-specific by default on Windows

parser = argparse.ArgumentParser()

parser.add_argument('--tbbroot', default='.', help='Take Intel TBB from here')

parser.add_argument('--prefix', default=default_prefix, help='Prefix')

parser.add_argument('--prebuilt', default=[], action='append', help='Directories to find prebuilt files')

parser.add_argument('--no-rebuild', default=False, action='store_true', help='do not rebuild')

parser.add_argument('--install', default=False, action='store_true', help='install all')

parser.add_argument('--install-libs', default=False, action='store_true', help='install libs')

parser.add_argument('--install-devel', default=False, action='store_true', help='install devel')

parser.add_argument('--install-docs', default=False, action='store_true', help='install docs')

parser.add_argument('--install-python', default=False, action='store_true', help='install python module')

parser.add_argument('--make-tool', default='make', help='Use different make command instead')

parser.add_argument('--copy-tool', default=None, help='Use this command for copying ($ tool file dest-dir)')

parser.add_argument('--build-args', default="", help='specify extra build args')

parser.add_argument('--build-prefix', default='local', help='build dir prefix')

parser.add_argument('--cmake-dir', help='directory to install CMake configuration files. Default: <prefix>/lib/cmake/tbb')

if is_win:

parser.add_argument('--msbuild', default=False, action='store_true', help='Use msbuild')

parser.add_argument('--vs', default="2012", help='select VS version for build')

parser.add_argument('--vs-platform', default="x64", help='select VS platform for build')

parser.add_argument('ignore', nargs='?', help="workaround conda-build issue #2512")

args = parser.parse_args()

if args.install:

args.install_libs = True

args.install_devel = True

args.install_docs = True

args.install_python= True

def custom_cp(src, dst):

assert os.system(' '.join([args.copy_tool, src, dst])) == 0

if args.copy_tool:

install_cp = custom_cp # e.g. to use install -p -D -m 755 on Linux

else:

install_cp = shutil.copy

bin_dir = jp(args.prefix, "bin")

lib_dir = jp(args.prefix, "lib")

inc_dir = jp(args.prefix, 'include')

doc_dir = jp(args.prefix, 'share', 'doc', 'tbb')

cmake_dir = jp(args.prefix, "lib", "cmake", "tbb") if args.cmake_dir is None else args.cmake_dir

if is_win:

os.environ["OS"] = "Windows_NT" # make sure TBB will interpret it correctly

libext = '.dll'

libpref = ''

dll_dir = bin_dir

else:

libext = '.dylib' if is_mac else '.so.2'

libpref = 'lib'

dll_dir = lib_dir

tbb_names = ["tbb", "tbbmalloc", "tbbmalloc_proxy"]

def system(arg):

print('$ ', arg)

return os.system(arg)

def run_make(arg):

if system('%s -j %s'% (args.make_tool, arg)) != 0:

print("\nBummer. Running serial build in order to recover the log and have a chance to fix the build")

assert system('%s %s'% (args.make_tool, arg)) == 0

os.chdir(args.tbbroot)

if args.prebuilt:

release_dirs = sum([glob(d) for d in args.prebuilt], [])

print("Using pre-built files from ", release_dirs)

else:

if is_win and args.msbuild:

preview_release_dir = release_dir = jp(args.tbbroot, 'build', 'vs'+args.vs, args.vs_platform, 'Release')

if not args.no_rebuild or not os.path.isdir(release_dir):

assert os.system('msbuild /m /p:Platform=%s /p:Configuration=Release %s build/vs%s/makefile.sln'% \

(args.vs_platform, args.build_args, args.vs)) == 0

preview_debug_dir = debug_dir = jp(args.tbbroot, 'build', 'vs'+args.vs, args.vs_platform, 'Debug')

if not args.no_rebuild or not os.path.isdir(debug_dir):

assert os.system('msbuild /m /p:Platform=%s /p:Configuration=Debug %s build/vs%s/makefile.sln'% \

(args.vs_platform, args.build_args, args.vs)) == 0

else:

release_dir = jp(args.tbbroot, 'build', args.build_prefix+'_release')

debug_dir = jp(args.tbbroot, 'build', args.build_prefix+'_debug')

if not args.no_rebuild or not (os.path.isdir(release_dir) and os.path.isdir(debug_dir)):

run_make('tbb_build_prefix=%s %s'% (args.build_prefix, args.build_args))

preview_release_dir = jp(args.tbbroot, 'build', args.build_prefix+'_preview_release')

preview_debug_dir = jp(args.tbbroot, 'build', args.build_prefix+'_preview_debug')

if not args.no_rebuild or not (os.path.isdir(preview_release_dir) and os.path.isdir(preview_debug_dir)):

run_make('tbb_build_prefix=%s_preview %s tbb_cpf=1 tbb'% (args.build_prefix, args.build_args))

release_dirs = [release_dir, debug_dir, preview_release_dir, preview_debug_dir]

filemap = OrderedDict()

def append_files(names, dst, paths=release_dirs):

global filemap

files = sum([glob(jp(d, f)) for d in paths for f in names], [])

filemap.update(dict(zip(files, [dst]*len(files))))

if args.install_libs:

append_files([libpref+f+libext for f in tbb_names], dll_dir)

if args.install_devel:

dll_files = [libpref+f+'_debug'+libext for f in tbb_names] # adding debug libraries

if not is_win or not args.msbuild:

dll_files += [libpref+"tbb_preview"+libext, libpref+"tbb_preview_debug"+libext]

if is_win:

dll_files += ['tbb*.pdb'] # copying debug info

if is_lin:

dll_files += ['libtbb*.so'] # copying linker scripts

# symlinks .so -> .so.2 should not be created instead

# since linking with -ltbb when using links can result in

# incorrect dependence upon unversioned .so files

append_files(dll_files, dll_dir)

if is_win:

append_files(['*.lib', '*.def'], lib_dir) # copying linker libs and defs

for rootdir, dirnames, filenames in os.walk(jp(args.tbbroot,'include')):

files = [f for f in filenames if not '.html' in f]

append_files(files, jp(inc_dir, rootdir.split('include')[1][1:]), paths=(rootdir,))

# Preparing CMake configuration files

cmake_build_dir = jp(args.tbbroot, 'build', args.build_prefix+'_release', 'cmake_configs')

assert system('cmake -DINSTALL_DIR=%s -DSYSTEM_NAME=%s -DTBB_VERSION_FILE=%s -DINC_REL_PATH=%s -DLIB_REL_PATH=%s -DBIN_REL_PATH=%s -P %s' % \

(cmake_build_dir,

platform.system(),

jp(args.tbbroot, 'include', 'tbb', 'tbb_stddef.h'),

os.path.relpath(inc_dir, cmake_dir),

os.path.relpath(lib_dir, cmake_dir),

os.path.relpath(bin_dir, cmake_dir),

jp(args.tbbroot, 'cmake', 'tbb_config_installer.cmake'))) == 0

append_files(['TBBConfig.cmake', 'TBBConfigVersion.cmake'], cmake_dir, paths=[cmake_build_dir])

if args.install_python: # RML part

irml_dir = jp(args.tbbroot, 'build', args.build_prefix+'_release')

run_make('-C src tbb_build_prefix=%s %s python_rml'% (args.build_prefix, args.build_args))

if is_lin:

append_files(['libirml.so.1'], dll_dir, paths=[irml_dir])

if args.install_docs:

files = [

'CHANGES',

'LICENSE',

'README',

'README.md',

'Release_Notes.txt',

]

append_files(files, doc_dir, paths=release_dirs+[jp(args.tbbroot, d) for d in ('.', 'doc')])

for f in filemap.keys():

assert os.path.exists(f)

assert os.path.isfile(f)

if filemap:

print("Copying to prefix =", args.prefix)

for f, dest in filemap.items():

if not os.path.isdir(dest):

os.makedirs(dest)

print("+ %s to $prefix%s"%(f,dest.replace(args.prefix, '')))

install_cp(f, dest)

if args.install_python: # Python part

paths = [os.path.abspath(d) for d in [args.prefix, inc_dir, irml_dir, lib_dir]+release_dirs]

os.environ["TBBROOT"] = paths[0]

# all the paths must be relative to python/ directory or be absolute

assert system('python python/setup.py build -b%s build_ext -I%s -L%s install -f'% \

(paths[2], paths[1], ':'.join(paths[2:]))) == 0

print("done")