Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
484 additions
and
42 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,230 @@ | ||
| """Detect zmq version""" | ||
| # | ||
| # Copyright (c) 2011 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/> | ||
| # BSD license: <http://www.opensource.org/licenses/bsd-license.php> | ||
| # | ||
| # pyzmq is free software; you can redistribute it and/or modify it under | ||
| # the terms of the Lesser GNU General Public License as published by | ||
| # the Free Software Foundation; either version 3 of the License, or | ||
| # (at your option) any later version. | ||
| # | ||
| # pyzmq is distributed in the hope that it will be useful, | ||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| # Lesser GNU General Public License for more details. | ||
| # | ||
| # You should have received a copy of the Lesser GNU General Public License | ||
| # along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
|
|
||
| import sys | ||
| import os | ||
| import logging | ||
|
|
||
| try: | ||
| from configparser import ConfigParser | ||
| except: | ||
| from ConfigParser import ConfigParser | ||
|
|
||
| pjoin = os.path.join | ||
|
|
||
| #----------------------------------------------------------------------------- | ||
| # Logging (adapted from h5py: http://h5py.googlecode.com) | ||
| #----------------------------------------------------------------------------- | ||
| logger = logging.getLogger() | ||
| logger.addHandler(logging.StreamHandler(sys.stderr)) | ||
|
|
||
| def debug(what): | ||
| pass | ||
|
|
||
| def fatal(instring, code=1): | ||
| logger.error("Fatal: "+instring) | ||
| exit(code) | ||
|
|
||
| def warn(instring): | ||
| logger.error("Warning: "+instring) | ||
|
|
||
|
|
||
| #----------------------------------------------------------------------------- | ||
| # Utility functions (adapted from h5py: http://h5py.googlecode.com) | ||
| #----------------------------------------------------------------------------- | ||
|
|
||
| def detect_zmq(basedir, **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_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) | ||
| options : dict | ||
| The compiler options used to compile the test function, e.g. `include_dirs`, | ||
| `library_dirs`, `libs`, etc. | ||
| """ | ||
|
|
||
| from distutils import ccompiler | ||
| import subprocess | ||
|
|
||
| cc = ccompiler.new_compiler() | ||
| for name, val in compiler_attrs.items(): | ||
| setattr(cc, name, val) | ||
|
|
||
| cfile = pjoin(basedir, 'vers.c') | ||
| efile = pjoin(basedir, 'vers') | ||
|
|
||
| f = open(cfile, 'w') | ||
| try: | ||
| f.write( | ||
| r""" | ||
| #include <stdio.h> | ||
| #include "zmq.h" | ||
| int main(){ | ||
| unsigned int major, minor, patch; | ||
| zmq_version(&major, &minor, &patch); | ||
| fprintf(stdout, "vers: %d.%d.%d\n", major, minor, patch); | ||
| return 0; | ||
| } | ||
| """) | ||
| finally: | ||
| f.close() | ||
|
|
||
| if sys.platform == 'darwin': | ||
| # allow for missing UB arch, since it will still work: | ||
| preargs = ['-undefined', 'dynamic_lookup'] | ||
| else: | ||
| preargs = None | ||
|
|
||
| objs = cc.compile([cfile]) | ||
| cc.link_executable(objs, efile, extra_preargs=preargs) | ||
|
|
||
| result = subprocess.Popen(efile, | ||
| stdout=subprocess.PIPE, stderr=subprocess.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) | ||
|
|
||
| props['options'] = compiler_attrs | ||
| return props | ||
|
|
||
| def localpath(*args): | ||
| return os.path.abspath(reduce(pjoin, (os.path.dirname(__file__),)+args)) | ||
|
|
||
| def loadpickle(name): | ||
| """ Load object from pickle file, or None if it can't be opened """ | ||
| import pickle | ||
| name = pjoin('conf', name) | ||
| try: | ||
| f = open(name,'rb') | ||
| except IOError: | ||
| # raise | ||
| return None | ||
| try: | ||
| return pickle.load(f) | ||
| except Exception: | ||
| # raise | ||
| return None | ||
| finally: | ||
| f.close() | ||
|
|
||
| def savepickle(name, data): | ||
| """ Save to pickle file, exiting if it can't be written """ | ||
| import pickle | ||
| if not os.path.exists('conf'): | ||
| os.mkdir('conf') | ||
| name = pjoin('conf', name) | ||
| try: | ||
| f = open(name, 'wb') | ||
| except IOError: | ||
| fatal("Can't open pickle file \"%s\" for writing" % name) | ||
| try: | ||
| pickle.dump(data, f, 0) | ||
| finally: | ||
| f.close() | ||
|
|
||
| def v_str(v_tuple): | ||
| """turn (2,0,1) into '2.0.1'.""" | ||
| return ".".join(str(x) for x in v_tuple) | ||
|
|
||
| def get_eargs(): | ||
| """ Look for options in environment vars """ | ||
|
|
||
| settings = {} | ||
|
|
||
| zmq = os.environ.get("ZMQ_DIR", '') | ||
| if zmq != '': | ||
| debug("Found environ var ZMQ_DIR=%s" % zmq) | ||
| settings['zmq'] = zmq | ||
|
|
||
| return settings | ||
|
|
||
| def get_cfg_args(): | ||
| """ Look for options in setup.cfg """ | ||
|
|
||
| settings = {} | ||
| zmq = '' | ||
| if not os.path.exists('setup.cfg'): | ||
| return settings | ||
| cfg = ConfigParser() | ||
| cfg.read('setup.cfg') | ||
| if 'build_ext' in cfg.sections() and \ | ||
| cfg.has_option('build_ext', 'include_dirs'): | ||
| includes = cfg.get('build_ext', 'include_dirs') | ||
| include = includes.split(os.pathsep)[0] | ||
| if include.endswith('include') and os.path.isdir(include): | ||
| zmq = include[:-8] | ||
| if zmq != '': | ||
| debug("Found ZMQ=%s in setup.cfg" % zmq) | ||
| settings['zmq'] = zmq | ||
|
|
||
| return settings | ||
|
|
||
| def get_cargs(): | ||
| """ Look for global options in the command line """ | ||
| settings = loadpickle('buildconf.pickle') | ||
| if settings is None: settings = {} | ||
| for arg in sys.argv[:]: | ||
| if arg.find('--zmq=') == 0: | ||
| zmq = arg.split('=')[-1] | ||
| if zmq.lower() == 'default': | ||
| settings.pop('zmq', None) | ||
| else: | ||
| settings['zmq'] = zmq | ||
| sys.argv.remove(arg) | ||
| savepickle('buildconf.pickle', settings) | ||
| return settings | ||
|
|
||
| def discover_settings(): | ||
| """ Discover custom settings for ZMQ path""" | ||
| settings = get_cfg_args() # lowest priority | ||
| settings.update(get_eargs()) | ||
| settings.update(get_cargs()) # highest priority | ||
| return settings.get('zmq') |
Oops, something went wrong.