mirror of https://github.com/zeromq/pyzmq.git
141 lines
4.6 KiB
Python
141 lines
4.6 KiB
Python
"""Detect zmq version"""
|
|
#-----------------------------------------------------------------------------
|
|
# Copyright (C) 2011 Brian Granger, Min Ragan-Kelley
|
|
#
|
|
# 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 distutils.sysconfig import customize_compiler
|
|
from subprocess import Popen, PIPE
|
|
|
|
from .misc import customize_mingw
|
|
|
|
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"""
|
|
if compiler is None or isinstance(compiler, str):
|
|
cc = ccompiler.new_compiler(compiler=compiler)
|
|
customize_compiler(cc)
|
|
if cc.compiler_type == 'mingw32':
|
|
customize_mingw(cc)
|
|
else:
|
|
cc = compiler
|
|
|
|
for name, val in compiler_attrs.items():
|
|
setattr(cc, name, val)
|
|
|
|
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']
|
|
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:
|
|
efile = test_compilation(cfile, compiler=compiler, **compiler_attrs)
|
|
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')
|
|
|
|
efile = test_compilation(cfile, compiler=compiler, **compiler_attrs)
|
|
|
|
result = Popen(efile, stdout=PIPE, stderr=PIPE)
|
|
so, se = result.communicate()
|
|
# for py3k:
|
|
so = so.decode()
|
|
se = se.decode()
|
|
if result.returncode:
|
|
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
|
|
|