Old engine for Continuous Time Bayesian Networks. Superseded by reCTBN. 🐍
https://github.com/madlabunimib/PyCTBN
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
800 lines
27 KiB
800 lines
27 KiB
4 years ago
|
import os
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
import shlex
|
||
|
|
import time
|
||
|
|
import subprocess
|
||
|
|
from copy import copy
|
||
|
|
from distutils import ccompiler
|
||
|
|
from distutils.ccompiler import (
|
||
|
|
compiler_class, gen_lib_options, get_default_compiler, new_compiler,
|
||
|
|
CCompiler
|
||
|
|
)
|
||
|
|
from distutils.errors import (
|
||
|
|
DistutilsExecError, DistutilsModuleError, DistutilsPlatformError,
|
||
|
|
CompileError, UnknownFileError
|
||
|
|
)
|
||
|
|
from distutils.sysconfig import customize_compiler
|
||
|
|
from distutils.version import LooseVersion
|
||
|
|
|
||
|
|
from numpy.distutils import log
|
||
|
|
from numpy.distutils.exec_command import (
|
||
|
|
filepath_from_subprocess_output, forward_bytes_to_stdout
|
||
|
|
)
|
||
|
|
from numpy.distutils.misc_util import cyg2win32, is_sequence, mingw32, \
|
||
|
|
get_num_build_jobs, \
|
||
|
|
_commandline_dep_string
|
||
|
|
|
||
|
|
# globals for parallel build management
|
||
|
|
try:
|
||
|
|
import threading
|
||
|
|
except ImportError:
|
||
|
|
import dummy_threading as threading
|
||
|
|
_job_semaphore = None
|
||
|
|
_global_lock = threading.Lock()
|
||
|
|
_processing_files = set()
|
||
|
|
|
||
|
|
|
||
|
|
def _needs_build(obj, cc_args, extra_postargs, pp_opts):
|
||
|
|
"""
|
||
|
|
Check if an objects needs to be rebuild based on its dependencies
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
obj : str
|
||
|
|
object file
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
bool
|
||
|
|
"""
|
||
|
|
# defined in unixcompiler.py
|
||
|
|
dep_file = obj + '.d'
|
||
|
|
if not os.path.exists(dep_file):
|
||
|
|
return True
|
||
|
|
|
||
|
|
# dep_file is a makefile containing 'object: dependencies'
|
||
|
|
# formatted like posix shell (spaces escaped, \ line continuations)
|
||
|
|
# the last line contains the compiler commandline arguments as some
|
||
|
|
# projects may compile an extension multiple times with different
|
||
|
|
# arguments
|
||
|
|
with open(dep_file, "r") as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
|
||
|
|
cmdline =_commandline_dep_string(cc_args, extra_postargs, pp_opts)
|
||
|
|
last_cmdline = lines[-1]
|
||
|
|
if last_cmdline != cmdline:
|
||
|
|
return True
|
||
|
|
|
||
|
|
contents = ''.join(lines[:-1])
|
||
|
|
deps = [x for x in shlex.split(contents, posix=True)
|
||
|
|
if x != "\n" and not x.endswith(":")]
|
||
|
|
|
||
|
|
try:
|
||
|
|
t_obj = os.stat(obj).st_mtime
|
||
|
|
|
||
|
|
# check if any of the dependencies is newer than the object
|
||
|
|
# the dependencies includes the source used to create the object
|
||
|
|
for f in deps:
|
||
|
|
if os.stat(f).st_mtime > t_obj:
|
||
|
|
return True
|
||
|
|
except OSError:
|
||
|
|
# no object counts as newer (shouldn't happen if dep_file exists)
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def replace_method(klass, method_name, func):
|
||
|
|
# Py3k does not have unbound method anymore, MethodType does not work
|
||
|
|
m = lambda self, *args, **kw: func(self, *args, **kw)
|
||
|
|
setattr(klass, method_name, m)
|
||
|
|
|
||
|
|
|
||
|
|
######################################################################
|
||
|
|
## Method that subclasses may redefine. But don't call this method,
|
||
|
|
## it i private to CCompiler class and may return unexpected
|
||
|
|
## results if used elsewhere. So, you have been warned..
|
||
|
|
|
||
|
|
def CCompiler_find_executables(self):
|
||
|
|
"""
|
||
|
|
Does nothing here, but is called by the get_version method and can be
|
||
|
|
overridden by subclasses. In particular it is redefined in the `FCompiler`
|
||
|
|
class where more documentation can be found.
|
||
|
|
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'find_executables', CCompiler_find_executables)
|
||
|
|
|
||
|
|
|
||
|
|
# Using customized CCompiler.spawn.
|
||
|
|
def CCompiler_spawn(self, cmd, display=None):
|
||
|
|
"""
|
||
|
|
Execute a command in a sub-process.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
cmd : str
|
||
|
|
The command to execute.
|
||
|
|
display : str or sequence of str, optional
|
||
|
|
The text to add to the log file kept by `numpy.distutils`.
|
||
|
|
If not given, `display` is equal to `cmd`.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
None
|
||
|
|
|
||
|
|
Raises
|
||
|
|
------
|
||
|
|
DistutilsExecError
|
||
|
|
If the command failed, i.e. the exit status was not 0.
|
||
|
|
|
||
|
|
"""
|
||
|
|
if display is None:
|
||
|
|
display = cmd
|
||
|
|
if is_sequence(display):
|
||
|
|
display = ' '.join(list(display))
|
||
|
|
log.info(display)
|
||
|
|
try:
|
||
|
|
if self.verbose:
|
||
|
|
subprocess.check_output(cmd)
|
||
|
|
else:
|
||
|
|
subprocess.check_output(cmd, stderr=subprocess.STDOUT)
|
||
|
|
except subprocess.CalledProcessError as exc:
|
||
|
|
o = exc.output
|
||
|
|
s = exc.returncode
|
||
|
|
except OSError:
|
||
|
|
# OSError doesn't have the same hooks for the exception
|
||
|
|
# output, but exec_command() historically would use an
|
||
|
|
# empty string for EnvironmentError (base class for
|
||
|
|
# OSError)
|
||
|
|
o = b''
|
||
|
|
# status previously used by exec_command() for parent
|
||
|
|
# of OSError
|
||
|
|
s = 127
|
||
|
|
else:
|
||
|
|
# use a convenience return here so that any kind of
|
||
|
|
# caught exception will execute the default code after the
|
||
|
|
# try / except block, which handles various exceptions
|
||
|
|
return None
|
||
|
|
|
||
|
|
if is_sequence(cmd):
|
||
|
|
cmd = ' '.join(list(cmd))
|
||
|
|
|
||
|
|
if self.verbose:
|
||
|
|
forward_bytes_to_stdout(o)
|
||
|
|
|
||
|
|
if re.search(b'Too many open files', o):
|
||
|
|
msg = '\nTry rerunning setup command until build succeeds.'
|
||
|
|
else:
|
||
|
|
msg = ''
|
||
|
|
raise DistutilsExecError('Command "%s" failed with exit status %d%s' %
|
||
|
|
(cmd, s, msg))
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'spawn', CCompiler_spawn)
|
||
|
|
|
||
|
|
def CCompiler_object_filenames(self, source_filenames, strip_dir=0, output_dir=''):
|
||
|
|
"""
|
||
|
|
Return the name of the object files for the given source files.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
source_filenames : list of str
|
||
|
|
The list of paths to source files. Paths can be either relative or
|
||
|
|
absolute, this is handled transparently.
|
||
|
|
strip_dir : bool, optional
|
||
|
|
Whether to strip the directory from the returned paths. If True,
|
||
|
|
the file name prepended by `output_dir` is returned. Default is False.
|
||
|
|
output_dir : str, optional
|
||
|
|
If given, this path is prepended to the returned paths to the
|
||
|
|
object files.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
obj_names : list of str
|
||
|
|
The list of paths to the object files corresponding to the source
|
||
|
|
files in `source_filenames`.
|
||
|
|
|
||
|
|
"""
|
||
|
|
if output_dir is None:
|
||
|
|
output_dir = ''
|
||
|
|
obj_names = []
|
||
|
|
for src_name in source_filenames:
|
||
|
|
base, ext = os.path.splitext(os.path.normpath(src_name))
|
||
|
|
base = os.path.splitdrive(base)[1] # Chop off the drive
|
||
|
|
base = base[os.path.isabs(base):] # If abs, chop off leading /
|
||
|
|
if base.startswith('..'):
|
||
|
|
# Resolve starting relative path components, middle ones
|
||
|
|
# (if any) have been handled by os.path.normpath above.
|
||
|
|
i = base.rfind('..')+2
|
||
|
|
d = base[:i]
|
||
|
|
d = os.path.basename(os.path.abspath(d))
|
||
|
|
base = d + base[i:]
|
||
|
|
if ext not in self.src_extensions:
|
||
|
|
raise UnknownFileError("unknown file type '%s' (from '%s')" % (ext, src_name))
|
||
|
|
if strip_dir:
|
||
|
|
base = os.path.basename(base)
|
||
|
|
obj_name = os.path.join(output_dir, base + self.obj_extension)
|
||
|
|
obj_names.append(obj_name)
|
||
|
|
return obj_names
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'object_filenames', CCompiler_object_filenames)
|
||
|
|
|
||
|
|
def CCompiler_compile(self, sources, output_dir=None, macros=None,
|
||
|
|
include_dirs=None, debug=0, extra_preargs=None,
|
||
|
|
extra_postargs=None, depends=None):
|
||
|
|
"""
|
||
|
|
Compile one or more source files.
|
||
|
|
|
||
|
|
Please refer to the Python distutils API reference for more details.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
sources : list of str
|
||
|
|
A list of filenames
|
||
|
|
output_dir : str, optional
|
||
|
|
Path to the output directory.
|
||
|
|
macros : list of tuples
|
||
|
|
A list of macro definitions.
|
||
|
|
include_dirs : list of str, optional
|
||
|
|
The directories to add to the default include file search path for
|
||
|
|
this compilation only.
|
||
|
|
debug : bool, optional
|
||
|
|
Whether or not to output debug symbols in or alongside the object
|
||
|
|
file(s).
|
||
|
|
extra_preargs, extra_postargs : ?
|
||
|
|
Extra pre- and post-arguments.
|
||
|
|
depends : list of str, optional
|
||
|
|
A list of file names that all targets depend on.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
objects : list of str
|
||
|
|
A list of object file names, one per source file `sources`.
|
||
|
|
|
||
|
|
Raises
|
||
|
|
------
|
||
|
|
CompileError
|
||
|
|
If compilation fails.
|
||
|
|
|
||
|
|
"""
|
||
|
|
# This method is effective only with Python >=2.3 distutils.
|
||
|
|
# Any changes here should be applied also to fcompiler.compile
|
||
|
|
# method to support pre Python 2.3 distutils.
|
||
|
|
global _job_semaphore
|
||
|
|
|
||
|
|
jobs = get_num_build_jobs()
|
||
|
|
|
||
|
|
# setup semaphore to not exceed number of compile jobs when parallelized at
|
||
|
|
# extension level (python >= 3.5)
|
||
|
|
with _global_lock:
|
||
|
|
if _job_semaphore is None:
|
||
|
|
_job_semaphore = threading.Semaphore(jobs)
|
||
|
|
|
||
|
|
if not sources:
|
||
|
|
return []
|
||
|
|
from numpy.distutils.fcompiler import (FCompiler, is_f_file,
|
||
|
|
has_f90_header)
|
||
|
|
if isinstance(self, FCompiler):
|
||
|
|
display = []
|
||
|
|
for fc in ['f77', 'f90', 'fix']:
|
||
|
|
fcomp = getattr(self, 'compiler_'+fc)
|
||
|
|
if fcomp is None:
|
||
|
|
continue
|
||
|
|
display.append("Fortran %s compiler: %s" % (fc, ' '.join(fcomp)))
|
||
|
|
display = '\n'.join(display)
|
||
|
|
else:
|
||
|
|
ccomp = self.compiler_so
|
||
|
|
display = "C compiler: %s\n" % (' '.join(ccomp),)
|
||
|
|
log.info(display)
|
||
|
|
macros, objects, extra_postargs, pp_opts, build = \
|
||
|
|
self._setup_compile(output_dir, macros, include_dirs, sources,
|
||
|
|
depends, extra_postargs)
|
||
|
|
cc_args = self._get_cc_args(pp_opts, debug, extra_preargs)
|
||
|
|
display = "compile options: '%s'" % (' '.join(cc_args))
|
||
|
|
if extra_postargs:
|
||
|
|
display += "\nextra options: '%s'" % (' '.join(extra_postargs))
|
||
|
|
log.info(display)
|
||
|
|
|
||
|
|
def single_compile(args):
|
||
|
|
obj, (src, ext) = args
|
||
|
|
if not _needs_build(obj, cc_args, extra_postargs, pp_opts):
|
||
|
|
return
|
||
|
|
|
||
|
|
# check if we are currently already processing the same object
|
||
|
|
# happens when using the same source in multiple extensions
|
||
|
|
while True:
|
||
|
|
# need explicit lock as there is no atomic check and add with GIL
|
||
|
|
with _global_lock:
|
||
|
|
# file not being worked on, start working
|
||
|
|
if obj not in _processing_files:
|
||
|
|
_processing_files.add(obj)
|
||
|
|
break
|
||
|
|
# wait for the processing to end
|
||
|
|
time.sleep(0.1)
|
||
|
|
|
||
|
|
try:
|
||
|
|
# retrieve slot from our #job semaphore and build
|
||
|
|
with _job_semaphore:
|
||
|
|
self._compile(obj, src, ext, cc_args, extra_postargs, pp_opts)
|
||
|
|
finally:
|
||
|
|
# register being done processing
|
||
|
|
with _global_lock:
|
||
|
|
_processing_files.remove(obj)
|
||
|
|
|
||
|
|
|
||
|
|
if isinstance(self, FCompiler):
|
||
|
|
objects_to_build = list(build.keys())
|
||
|
|
f77_objects, other_objects = [], []
|
||
|
|
for obj in objects:
|
||
|
|
if obj in objects_to_build:
|
||
|
|
src, ext = build[obj]
|
||
|
|
if self.compiler_type=='absoft':
|
||
|
|
obj = cyg2win32(obj)
|
||
|
|
src = cyg2win32(src)
|
||
|
|
if is_f_file(src) and not has_f90_header(src):
|
||
|
|
f77_objects.append((obj, (src, ext)))
|
||
|
|
else:
|
||
|
|
other_objects.append((obj, (src, ext)))
|
||
|
|
|
||
|
|
# f77 objects can be built in parallel
|
||
|
|
build_items = f77_objects
|
||
|
|
# build f90 modules serial, module files are generated during
|
||
|
|
# compilation and may be used by files later in the list so the
|
||
|
|
# ordering is important
|
||
|
|
for o in other_objects:
|
||
|
|
single_compile(o)
|
||
|
|
else:
|
||
|
|
build_items = build.items()
|
||
|
|
|
||
|
|
if len(build) > 1 and jobs > 1:
|
||
|
|
# build parallel
|
||
|
|
import multiprocessing.pool
|
||
|
|
pool = multiprocessing.pool.ThreadPool(jobs)
|
||
|
|
pool.map(single_compile, build_items)
|
||
|
|
pool.close()
|
||
|
|
else:
|
||
|
|
# build serial
|
||
|
|
for o in build_items:
|
||
|
|
single_compile(o)
|
||
|
|
|
||
|
|
# Return *all* object filenames, not just the ones we just built.
|
||
|
|
return objects
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'compile', CCompiler_compile)
|
||
|
|
|
||
|
|
def CCompiler_customize_cmd(self, cmd, ignore=()):
|
||
|
|
"""
|
||
|
|
Customize compiler using distutils command.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
cmd : class instance
|
||
|
|
An instance inheriting from `distutils.cmd.Command`.
|
||
|
|
ignore : sequence of str, optional
|
||
|
|
List of `CCompiler` commands (without ``'set_'``) that should not be
|
||
|
|
altered. Strings that are checked for are:
|
||
|
|
``('include_dirs', 'define', 'undef', 'libraries', 'library_dirs',
|
||
|
|
'rpath', 'link_objects')``.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
None
|
||
|
|
|
||
|
|
"""
|
||
|
|
log.info('customize %s using %s' % (self.__class__.__name__,
|
||
|
|
cmd.__class__.__name__))
|
||
|
|
def allow(attr):
|
||
|
|
return getattr(cmd, attr, None) is not None and attr not in ignore
|
||
|
|
|
||
|
|
if allow('include_dirs'):
|
||
|
|
self.set_include_dirs(cmd.include_dirs)
|
||
|
|
if allow('define'):
|
||
|
|
for (name, value) in cmd.define:
|
||
|
|
self.define_macro(name, value)
|
||
|
|
if allow('undef'):
|
||
|
|
for macro in cmd.undef:
|
||
|
|
self.undefine_macro(macro)
|
||
|
|
if allow('libraries'):
|
||
|
|
self.set_libraries(self.libraries + cmd.libraries)
|
||
|
|
if allow('library_dirs'):
|
||
|
|
self.set_library_dirs(self.library_dirs + cmd.library_dirs)
|
||
|
|
if allow('rpath'):
|
||
|
|
self.set_runtime_library_dirs(cmd.rpath)
|
||
|
|
if allow('link_objects'):
|
||
|
|
self.set_link_objects(cmd.link_objects)
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'customize_cmd', CCompiler_customize_cmd)
|
||
|
|
|
||
|
|
def _compiler_to_string(compiler):
|
||
|
|
props = []
|
||
|
|
mx = 0
|
||
|
|
keys = list(compiler.executables.keys())
|
||
|
|
for key in ['version', 'libraries', 'library_dirs',
|
||
|
|
'object_switch', 'compile_switch',
|
||
|
|
'include_dirs', 'define', 'undef', 'rpath', 'link_objects']:
|
||
|
|
if key not in keys:
|
||
|
|
keys.append(key)
|
||
|
|
for key in keys:
|
||
|
|
if hasattr(compiler, key):
|
||
|
|
v = getattr(compiler, key)
|
||
|
|
mx = max(mx, len(key))
|
||
|
|
props.append((key, repr(v)))
|
||
|
|
fmt = '%-' + repr(mx+1) + 's = %s'
|
||
|
|
lines = [fmt % prop for prop in props]
|
||
|
|
return '\n'.join(lines)
|
||
|
|
|
||
|
|
def CCompiler_show_customization(self):
|
||
|
|
"""
|
||
|
|
Print the compiler customizations to stdout.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
None
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
None
|
||
|
|
|
||
|
|
Notes
|
||
|
|
-----
|
||
|
|
Printing is only done if the distutils log threshold is < 2.
|
||
|
|
|
||
|
|
"""
|
||
|
|
if 0:
|
||
|
|
for attrname in ['include_dirs', 'define', 'undef',
|
||
|
|
'libraries', 'library_dirs',
|
||
|
|
'rpath', 'link_objects']:
|
||
|
|
attr = getattr(self, attrname, None)
|
||
|
|
if not attr:
|
||
|
|
continue
|
||
|
|
log.info("compiler '%s' is set to %s" % (attrname, attr))
|
||
|
|
try:
|
||
|
|
self.get_version()
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
if log._global_log.threshold<2:
|
||
|
|
print('*'*80)
|
||
|
|
print(self.__class__)
|
||
|
|
print(_compiler_to_string(self))
|
||
|
|
print('*'*80)
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'show_customization', CCompiler_show_customization)
|
||
|
|
|
||
|
|
def CCompiler_customize(self, dist, need_cxx=0):
|
||
|
|
"""
|
||
|
|
Do any platform-specific customization of a compiler instance.
|
||
|
|
|
||
|
|
This method calls `distutils.sysconfig.customize_compiler` for
|
||
|
|
platform-specific customization, as well as optionally remove a flag
|
||
|
|
to suppress spurious warnings in case C++ code is being compiled.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
dist : object
|
||
|
|
This parameter is not used for anything.
|
||
|
|
need_cxx : bool, optional
|
||
|
|
Whether or not C++ has to be compiled. If so (True), the
|
||
|
|
``"-Wstrict-prototypes"`` option is removed to prevent spurious
|
||
|
|
warnings. Default is False.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
None
|
||
|
|
|
||
|
|
Notes
|
||
|
|
-----
|
||
|
|
All the default options used by distutils can be extracted with::
|
||
|
|
|
||
|
|
from distutils import sysconfig
|
||
|
|
sysconfig.get_config_vars('CC', 'CXX', 'OPT', 'BASECFLAGS',
|
||
|
|
'CCSHARED', 'LDSHARED', 'SO')
|
||
|
|
|
||
|
|
"""
|
||
|
|
# See FCompiler.customize for suggested usage.
|
||
|
|
log.info('customize %s' % (self.__class__.__name__))
|
||
|
|
customize_compiler(self)
|
||
|
|
if need_cxx:
|
||
|
|
# In general, distutils uses -Wstrict-prototypes, but this option is
|
||
|
|
# not valid for C++ code, only for C. Remove it if it's there to
|
||
|
|
# avoid a spurious warning on every compilation.
|
||
|
|
try:
|
||
|
|
self.compiler_so.remove('-Wstrict-prototypes')
|
||
|
|
except (AttributeError, ValueError):
|
||
|
|
pass
|
||
|
|
|
||
|
|
if hasattr(self, 'compiler') and 'cc' in self.compiler[0]:
|
||
|
|
if not self.compiler_cxx:
|
||
|
|
if self.compiler[0].startswith('gcc'):
|
||
|
|
a, b = 'gcc', 'g++'
|
||
|
|
else:
|
||
|
|
a, b = 'cc', 'c++'
|
||
|
|
self.compiler_cxx = [self.compiler[0].replace(a, b)]\
|
||
|
|
+ self.compiler[1:]
|
||
|
|
else:
|
||
|
|
if hasattr(self, 'compiler'):
|
||
|
|
log.warn("#### %s #######" % (self.compiler,))
|
||
|
|
if not hasattr(self, 'compiler_cxx'):
|
||
|
|
log.warn('Missing compiler_cxx fix for ' + self.__class__.__name__)
|
||
|
|
|
||
|
|
|
||
|
|
# check if compiler supports gcc style automatic dependencies
|
||
|
|
# run on every extension so skip for known good compilers
|
||
|
|
if hasattr(self, 'compiler') and ('gcc' in self.compiler[0] or
|
||
|
|
'g++' in self.compiler[0] or
|
||
|
|
'clang' in self.compiler[0]):
|
||
|
|
self._auto_depends = True
|
||
|
|
elif os.name == 'posix':
|
||
|
|
import tempfile
|
||
|
|
import shutil
|
||
|
|
tmpdir = tempfile.mkdtemp()
|
||
|
|
try:
|
||
|
|
fn = os.path.join(tmpdir, "file.c")
|
||
|
|
with open(fn, "w") as f:
|
||
|
|
f.write("int a;\n")
|
||
|
|
self.compile([fn], output_dir=tmpdir,
|
||
|
|
extra_preargs=['-MMD', '-MF', fn + '.d'])
|
||
|
|
self._auto_depends = True
|
||
|
|
except CompileError:
|
||
|
|
self._auto_depends = False
|
||
|
|
finally:
|
||
|
|
shutil.rmtree(tmpdir)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'customize', CCompiler_customize)
|
||
|
|
|
||
|
|
def simple_version_match(pat=r'[-.\d]+', ignore='', start=''):
|
||
|
|
"""
|
||
|
|
Simple matching of version numbers, for use in CCompiler and FCompiler.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
pat : str, optional
|
||
|
|
A regular expression matching version numbers.
|
||
|
|
Default is ``r'[-.\\d]+'``.
|
||
|
|
ignore : str, optional
|
||
|
|
A regular expression matching patterns to skip.
|
||
|
|
Default is ``''``, in which case nothing is skipped.
|
||
|
|
start : str, optional
|
||
|
|
A regular expression matching the start of where to start looking
|
||
|
|
for version numbers.
|
||
|
|
Default is ``''``, in which case searching is started at the
|
||
|
|
beginning of the version string given to `matcher`.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
matcher : callable
|
||
|
|
A function that is appropriate to use as the ``.version_match``
|
||
|
|
attribute of a `CCompiler` class. `matcher` takes a single parameter,
|
||
|
|
a version string.
|
||
|
|
|
||
|
|
"""
|
||
|
|
def matcher(self, version_string):
|
||
|
|
# version string may appear in the second line, so getting rid
|
||
|
|
# of new lines:
|
||
|
|
version_string = version_string.replace('\n', ' ')
|
||
|
|
pos = 0
|
||
|
|
if start:
|
||
|
|
m = re.match(start, version_string)
|
||
|
|
if not m:
|
||
|
|
return None
|
||
|
|
pos = m.end()
|
||
|
|
while True:
|
||
|
|
m = re.search(pat, version_string[pos:])
|
||
|
|
if not m:
|
||
|
|
return None
|
||
|
|
if ignore and re.match(ignore, m.group(0)):
|
||
|
|
pos = m.end()
|
||
|
|
continue
|
||
|
|
break
|
||
|
|
return m.group(0)
|
||
|
|
return matcher
|
||
|
|
|
||
|
|
def CCompiler_get_version(self, force=False, ok_status=[0]):
|
||
|
|
"""
|
||
|
|
Return compiler version, or None if compiler is not available.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
force : bool, optional
|
||
|
|
If True, force a new determination of the version, even if the
|
||
|
|
compiler already has a version attribute. Default is False.
|
||
|
|
ok_status : list of int, optional
|
||
|
|
The list of status values returned by the version look-up process
|
||
|
|
for which a version string is returned. If the status value is not
|
||
|
|
in `ok_status`, None is returned. Default is ``[0]``.
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
version : str or None
|
||
|
|
Version string, in the format of `distutils.version.LooseVersion`.
|
||
|
|
|
||
|
|
"""
|
||
|
|
if not force and hasattr(self, 'version'):
|
||
|
|
return self.version
|
||
|
|
self.find_executables()
|
||
|
|
try:
|
||
|
|
version_cmd = self.version_cmd
|
||
|
|
except AttributeError:
|
||
|
|
return None
|
||
|
|
if not version_cmd or not version_cmd[0]:
|
||
|
|
return None
|
||
|
|
try:
|
||
|
|
matcher = self.version_match
|
||
|
|
except AttributeError:
|
||
|
|
try:
|
||
|
|
pat = self.version_pattern
|
||
|
|
except AttributeError:
|
||
|
|
return None
|
||
|
|
def matcher(version_string):
|
||
|
|
m = re.match(pat, version_string)
|
||
|
|
if not m:
|
||
|
|
return None
|
||
|
|
version = m.group('version')
|
||
|
|
return version
|
||
|
|
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(version_cmd, stderr=subprocess.STDOUT)
|
||
|
|
except subprocess.CalledProcessError as exc:
|
||
|
|
output = exc.output
|
||
|
|
status = exc.returncode
|
||
|
|
except OSError:
|
||
|
|
# match the historical returns for a parent
|
||
|
|
# exception class caught by exec_command()
|
||
|
|
status = 127
|
||
|
|
output = b''
|
||
|
|
else:
|
||
|
|
# output isn't actually a filepath but we do this
|
||
|
|
# for now to match previous distutils behavior
|
||
|
|
output = filepath_from_subprocess_output(output)
|
||
|
|
status = 0
|
||
|
|
|
||
|
|
version = None
|
||
|
|
if status in ok_status:
|
||
|
|
version = matcher(output)
|
||
|
|
if version:
|
||
|
|
version = LooseVersion(version)
|
||
|
|
self.version = version
|
||
|
|
return version
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'get_version', CCompiler_get_version)
|
||
|
|
|
||
|
|
def CCompiler_cxx_compiler(self):
|
||
|
|
"""
|
||
|
|
Return the C++ compiler.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
None
|
||
|
|
|
||
|
|
Returns
|
||
|
|
-------
|
||
|
|
cxx : class instance
|
||
|
|
The C++ compiler, as a `CCompiler` instance.
|
||
|
|
|
||
|
|
"""
|
||
|
|
if self.compiler_type in ('msvc', 'intelw', 'intelemw'):
|
||
|
|
return self
|
||
|
|
|
||
|
|
cxx = copy(self)
|
||
|
|
cxx.compiler_so = [cxx.compiler_cxx[0]] + cxx.compiler_so[1:]
|
||
|
|
if sys.platform.startswith('aix') and 'ld_so_aix' in cxx.linker_so[0]:
|
||
|
|
# AIX needs the ld_so_aix script included with Python
|
||
|
|
cxx.linker_so = [cxx.linker_so[0], cxx.compiler_cxx[0]] \
|
||
|
|
+ cxx.linker_so[2:]
|
||
|
|
else:
|
||
|
|
cxx.linker_so = [cxx.compiler_cxx[0]] + cxx.linker_so[1:]
|
||
|
|
return cxx
|
||
|
|
|
||
|
|
replace_method(CCompiler, 'cxx_compiler', CCompiler_cxx_compiler)
|
||
|
|
|
||
|
|
compiler_class['intel'] = ('intelccompiler', 'IntelCCompiler',
|
||
|
|
"Intel C Compiler for 32-bit applications")
|
||
|
|
compiler_class['intele'] = ('intelccompiler', 'IntelItaniumCCompiler',
|
||
|
|
"Intel C Itanium Compiler for Itanium-based applications")
|
||
|
|
compiler_class['intelem'] = ('intelccompiler', 'IntelEM64TCCompiler',
|
||
|
|
"Intel C Compiler for 64-bit applications")
|
||
|
|
compiler_class['intelw'] = ('intelccompiler', 'IntelCCompilerW',
|
||
|
|
"Intel C Compiler for 32-bit applications on Windows")
|
||
|
|
compiler_class['intelemw'] = ('intelccompiler', 'IntelEM64TCCompilerW',
|
||
|
|
"Intel C Compiler for 64-bit applications on Windows")
|
||
|
|
compiler_class['pathcc'] = ('pathccompiler', 'PathScaleCCompiler',
|
||
|
|
"PathScale Compiler for SiCortex-based applications")
|
||
|
|
ccompiler._default_compilers += (('linux.*', 'intel'),
|
||
|
|
('linux.*', 'intele'),
|
||
|
|
('linux.*', 'intelem'),
|
||
|
|
('linux.*', 'pathcc'),
|
||
|
|
('nt', 'intelw'),
|
||
|
|
('nt', 'intelemw'))
|
||
|
|
|
||
|
|
if sys.platform == 'win32':
|
||
|
|
compiler_class['mingw32'] = ('mingw32ccompiler', 'Mingw32CCompiler',
|
||
|
|
"Mingw32 port of GNU C Compiler for Win32"\
|
||
|
|
"(for MSC built Python)")
|
||
|
|
if mingw32():
|
||
|
|
# On windows platforms, we want to default to mingw32 (gcc)
|
||
|
|
# because msvc can't build blitz stuff.
|
||
|
|
log.info('Setting mingw32 as default compiler for nt.')
|
||
|
|
ccompiler._default_compilers = (('nt', 'mingw32'),) \
|
||
|
|
+ ccompiler._default_compilers
|
||
|
|
|
||
|
|
|
||
|
|
_distutils_new_compiler = new_compiler
|
||
|
|
def new_compiler (plat=None,
|
||
|
|
compiler=None,
|
||
|
|
verbose=None,
|
||
|
|
dry_run=0,
|
||
|
|
force=0):
|
||
|
|
# Try first C compilers from numpy.distutils.
|
||
|
|
if verbose is None:
|
||
|
|
verbose = log.get_threshold() <= log.INFO
|
||
|
|
if plat is None:
|
||
|
|
plat = os.name
|
||
|
|
try:
|
||
|
|
if compiler is None:
|
||
|
|
compiler = get_default_compiler(plat)
|
||
|
|
(module_name, class_name, long_description) = compiler_class[compiler]
|
||
|
|
except KeyError:
|
||
|
|
msg = "don't know how to compile C/C++ code on platform '%s'" % plat
|
||
|
|
if compiler is not None:
|
||
|
|
msg = msg + " with '%s' compiler" % compiler
|
||
|
|
raise DistutilsPlatformError(msg)
|
||
|
|
module_name = "numpy.distutils." + module_name
|
||
|
|
try:
|
||
|
|
__import__ (module_name)
|
||
|
|
except ImportError as e:
|
||
|
|
msg = str(e)
|
||
|
|
log.info('%s in numpy.distutils; trying from distutils',
|
||
|
|
str(msg))
|
||
|
|
module_name = module_name[6:]
|
||
|
|
try:
|
||
|
|
__import__(module_name)
|
||
|
|
except ImportError as e:
|
||
|
|
msg = str(e)
|
||
|
|
raise DistutilsModuleError("can't compile C/C++ code: unable to load module '%s'" % \
|
||
|
|
module_name)
|
||
|
|
try:
|
||
|
|
module = sys.modules[module_name]
|
||
|
|
klass = vars(module)[class_name]
|
||
|
|
except KeyError:
|
||
|
|
raise DistutilsModuleError(("can't compile C/C++ code: unable to find class '%s' " +
|
||
|
|
"in module '%s'") % (class_name, module_name))
|
||
|
|
compiler = klass(None, dry_run, force)
|
||
|
|
compiler.verbose = verbose
|
||
|
|
log.debug('new_compiler returns %s' % (klass))
|
||
|
|
return compiler
|
||
|
|
|
||
|
|
ccompiler.new_compiler = new_compiler
|
||
|
|
|
||
|
|
_distutils_gen_lib_options = gen_lib_options
|
||
|
|
def gen_lib_options(compiler, library_dirs, runtime_library_dirs, libraries):
|
||
|
|
# the version of this function provided by CPython allows the following
|
||
|
|
# to return lists, which are unpacked automatically:
|
||
|
|
# - compiler.runtime_library_dir_option
|
||
|
|
# our version extends the behavior to:
|
||
|
|
# - compiler.library_dir_option
|
||
|
|
# - compiler.library_option
|
||
|
|
# - compiler.find_library_file
|
||
|
|
r = _distutils_gen_lib_options(compiler, library_dirs,
|
||
|
|
runtime_library_dirs, libraries)
|
||
|
|
lib_opts = []
|
||
|
|
for i in r:
|
||
|
|
if is_sequence(i):
|
||
|
|
lib_opts.extend(list(i))
|
||
|
|
else:
|
||
|
|
lib_opts.append(i)
|
||
|
|
return lib_opts
|
||
|
|
ccompiler.gen_lib_options = gen_lib_options
|
||
|
|
|
||
|
|
# Also fix up the various compiler modules, which do
|
||
|
|
# from distutils.ccompiler import gen_lib_options
|
||
|
|
# Don't bother with mwerks, as we don't support Classic Mac.
|
||
|
|
for _cc in ['msvc9', 'msvc', '_msvc', 'bcpp', 'cygwinc', 'emxc', 'unixc']:
|
||
|
|
_m = sys.modules.get('distutils.' + _cc + 'compiler')
|
||
|
|
if _m is not None:
|
||
|
|
setattr(_m, 'gen_lib_options', gen_lib_options)
|
||
|
|
|