mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-02 01:55:39 +08:00
Zachary Pincus
17edd0cc29
(1) Now uses subprocess.call instead of os.system, so that paths with spaces
(and the like) are handled properly.
(2) Removes superfluous .c.new files after deciding whether or not to copy
them to the .c files.
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
import os
|
|
import shutil
|
|
import hashlib
|
|
import subprocess
|
|
|
|
def cython(pyx_files, working_path=''):
|
|
"""Use Cython to convert the given files to C.
|
|
|
|
Parameters
|
|
----------
|
|
pyx_files : list of str
|
|
The input .pyx files.
|
|
|
|
"""
|
|
try:
|
|
import Cython
|
|
except ImportError:
|
|
# If cython is not found, we do nothing -- the build will make use of
|
|
# the distributed .c files
|
|
pass
|
|
else:
|
|
for pyxfile in [os.path.join(working_path, f) for f in pyx_files]:
|
|
# make a backup of the good c files
|
|
c_file = pyxfile[:-4] + '.c'
|
|
c_file_new = c_file + '.new'
|
|
|
|
# run cython compiler
|
|
cmd = 'cython -o %s %s' % (c_file_new, pyxfile)
|
|
print cmd
|
|
status = subprocess.call(['cython', '-o', c_file_new, pyxfile])
|
|
# if the resulting file is small, cython compilation failed
|
|
if status != 0 or os.path.getsize(c_file_new) < 100:
|
|
print "Cython compilation of %s failed. Falling back " \
|
|
"on pre-generated file." % os.path.basename(pyxfile)
|
|
elif not same_cython(c_file_new, c_file):
|
|
# if the generated .c file differs from the one provided,
|
|
# use that one instead
|
|
shutil.copy(c_file_new, c_file)
|
|
os.remove(c_file_new)
|
|
|
|
|
|
def same_cython(f0, f1):
|
|
'''Compare two Cython generated C-files, based on their md5-sum.
|
|
|
|
Returns True if the files are identical, False if not. The first
|
|
lines are skipped, due to the timestamp printed there.
|
|
|
|
'''
|
|
def md5sum(f):
|
|
m = hashlib.new('md5')
|
|
while True:
|
|
d = f.read(8096)
|
|
if not d:
|
|
break
|
|
m.update(d)
|
|
return m.hexdigest()
|
|
|
|
if not (os.path.isfile(f0) and os.path.isfile(f1)):
|
|
return False
|
|
|
|
f0 = file(f0)
|
|
f0.readline()
|
|
|
|
f1 = file(f1)
|
|
f1.readline()
|
|
|
|
return md5sum(f0) == md5sum(f1)
|
|
|