diff options
author | 2015-08-24 13:22:48 +0300 | |
---|---|---|
committer | 2015-08-24 13:22:48 +0300 | |
commit | dab741a80699f86e86c91718872a052cca9bbb25 (patch) | |
tree | 1959c4a2cea440170a5113dcb067796cb20ffb64 /external_libs/python/pyzmq-14.7.0/buildutils/detect.py | |
parent | d3f26ece7d4383df0b22fe9c3cb3e695381ec737 (diff) |
Fixed dependencies of Control Plane to use external_lib sources
Diffstat (limited to 'external_libs/python/pyzmq-14.7.0/buildutils/detect.py')
-rw-r--r-- | external_libs/python/pyzmq-14.7.0/buildutils/detect.py | 136 |
1 files changed, 136 insertions, 0 deletions
diff --git a/external_libs/python/pyzmq-14.7.0/buildutils/detect.py b/external_libs/python/pyzmq-14.7.0/buildutils/detect.py new file mode 100644 index 00000000..7a6c115a --- /dev/null +++ b/external_libs/python/pyzmq-14.7.0/buildutils/detect.py @@ -0,0 +1,136 @@ +"""Detect zmq version""" +#----------------------------------------------------------------------------- +# Copyright (C) PyZMQ Developers +# +# This file is part of pyzmq, copied and adapted from h5py. +# h5py source used under the New BSD license +# +# h5py: <http://code.google.com/p/h5py/> +# +# Distributed under the terms of the New BSD License. The full license is in +# the file COPYING.BSD, distributed as part of this software. +#----------------------------------------------------------------------------- + +import shutil +import sys +import os +import logging +import platform +from distutils import ccompiler +from subprocess import Popen, PIPE + +from .misc import get_compiler, get_output_error +from .patch import patch_lib_paths + +pjoin = os.path.join + +#----------------------------------------------------------------------------- +# Utility functions (adapted from h5py: http://h5py.googlecode.com) +#----------------------------------------------------------------------------- + +def test_compilation(cfile, compiler=None, **compiler_attrs): + """Test simple compilation with given settings""" + cc = get_compiler(compiler, **compiler_attrs) + + efile, ext = os.path.splitext(cfile) + + cpreargs = lpreargs = None + if sys.platform == 'darwin': + # use appropriate arch for compiler + if platform.architecture()[0]=='32bit': + if platform.processor() == 'powerpc': + cpu = 'ppc' + else: + cpu = 'i386' + cpreargs = ['-arch', cpu] + lpreargs = ['-arch', cpu, '-undefined', 'dynamic_lookup'] + else: + # allow for missing UB arch, since it will still work: + lpreargs = ['-undefined', 'dynamic_lookup'] + if sys.platform == 'sunos5': + if platform.architecture()[0]=='32bit': + lpreargs = ['-m32'] + else: + lpreargs = ['-m64'] + extra = compiler_attrs.get('extra_compile_args', None) + + objs = cc.compile([cfile], extra_preargs=cpreargs, extra_postargs=extra) + cc.link_executable(objs, efile, extra_preargs=lpreargs) + return efile + +def compile_and_run(basedir, src, compiler=None, **compiler_attrs): + if not os.path.exists(basedir): + os.makedirs(basedir) + cfile = pjoin(basedir, os.path.basename(src)) + shutil.copy(src, cfile) + try: + cc = get_compiler(compiler, **compiler_attrs) + efile = test_compilation(cfile, compiler=cc) + patch_lib_paths(efile, cc.library_dirs) + result = Popen(efile, stdout=PIPE, stderr=PIPE) + so, se = result.communicate() + # for py3k: + so = so.decode() + se = se.decode() + finally: + shutil.rmtree(basedir) + + return result.returncode, so, se + + +def detect_zmq(basedir, compiler=None, **compiler_attrs): + """Compile, link & execute a test program, in empty directory `basedir`. + + The C compiler will be updated with any keywords given via setattr. + + Parameters + ---------- + + basedir : path + The location where the test program will be compiled and run + compiler : str + The distutils compiler key (e.g. 'unix', 'msvc', or 'mingw32') + **compiler_attrs : dict + Any extra compiler attributes, which will be set via ``setattr(cc)``. + + Returns + ------- + + A dict of properties for zmq compilation, with the following two keys: + + vers : tuple + The ZMQ version as a tuple of ints, e.g. (2,2,0) + settings : dict + The compiler options used to compile the test function, e.g. `include_dirs`, + `library_dirs`, `libs`, etc. + """ + + cfile = pjoin(basedir, 'vers.c') + shutil.copy(pjoin(os.path.dirname(__file__), 'vers.c'), cfile) + + # check if we need to link against Realtime Extensions library + if sys.platform.startswith('linux'): + cc = ccompiler.new_compiler(compiler=compiler) + cc.output_dir = basedir + if not cc.has_function('timer_create'): + compiler_attrs['libraries'].append('rt') + + cc = get_compiler(compiler=compiler, **compiler_attrs) + efile = test_compilation(cfile, compiler=cc) + patch_lib_paths(efile, cc.library_dirs) + + rc, so, se = get_output_error([efile]) + if rc: + msg = "Error running version detection script:\n%s\n%s" % (so,se) + logging.error(msg) + raise IOError(msg) + + handlers = {'vers': lambda val: tuple(int(v) for v in val.split('.'))} + + props = {} + for line in (x for x in so.split('\n') if x): + key, val = line.split(':') + props[key] = handlers[key](val) + + return props + |