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.
43 lines
1017 B
43 lines
1017 B
7 years ago
|
import os
|
||
|
|
import sys
|
||
|
|
import fcntl
|
||
|
|
import hashlib
|
||
|
|
from cffi import FFI
|
||
|
|
|
||
|
|
TMPDIR = "/tmp/ccache"
|
||
|
|
|
||
|
|
def ffi_wrap(name, c_code, c_header, tmpdir=TMPDIR):
|
||
|
|
cache = name + "_" + hashlib.sha1(c_code).hexdigest()
|
||
|
|
try:
|
||
|
|
os.mkdir(tmpdir)
|
||
|
|
except OSError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
fd = os.open(tmpdir, 0)
|
||
|
|
fcntl.flock(fd, fcntl.LOCK_EX)
|
||
|
|
try:
|
||
|
|
sys.path.append(tmpdir)
|
||
|
|
try:
|
||
|
|
mod = __import__(cache)
|
||
|
|
except Exception:
|
||
|
|
print "cache miss", cache
|
||
|
|
compile_code(cache, c_code, c_header, tmpdir)
|
||
|
|
mod = __import__(cache)
|
||
|
|
finally:
|
||
|
|
os.close(fd)
|
||
|
|
|
||
|
|
return mod.ffi, mod.lib
|
||
|
|
|
||
|
|
def compile_code(name, c_code, c_header, directory):
|
||
|
|
ffibuilder = FFI()
|
||
|
|
ffibuilder.set_source(name, c_code, source_extension='.cpp')
|
||
|
|
ffibuilder.cdef(c_header)
|
||
|
|
os.environ['OPT'] = "-fwrapv -O2 -DNDEBUG -std=c++11"
|
||
|
|
os.environ['CFLAGS'] = ""
|
||
|
|
ffibuilder.compile(verbose=True, debug=False, tmpdir=directory)
|
||
|
|
|
||
|
|
def wrap_compiled(name, directory):
|
||
|
|
sys.path.append(directory)
|
||
|
|
mod = __import__(name)
|
||
|
|
return mod.ffi, mod.lib
|