diff options
Diffstat (limited to 'klippy/chelper/__init__.py')
| -rw-r--r-- | klippy/chelper/__init__.py | 95 |
1 files changed, 70 insertions, 25 deletions
diff --git a/klippy/chelper/__init__.py b/klippy/chelper/__init__.py index 2aed107a..0a40dfeb 100644 --- a/klippy/chelper/__init__.py +++ b/klippy/chelper/__init__.py @@ -12,21 +12,44 @@ import cffi ###################################################################### GCC_CMD = "gcc" -COMPILE_ARGS = ("-Wall -g -O2 -shared -fPIC" - " -flto -fwhole-program -fno-use-linker-plugin" - " -o %s %s") +COMPILE_ARGS = ( + "-Wall -g -O2 -shared -fPIC" + " -flto -fwhole-program -fno-use-linker-plugin" + " -o %s %s" +) SSE_FLAGS = "-mfpmath=sse -msse2" SOURCE_FILES = [ - 'pyhelper.c', 'serialqueue.c', 'stepcompress.c', 'itersolve.c', 'trapq.c', - 'pollreactor.c', 'msgblock.c', 'trdispatch.c', - 'kin_cartesian.c', 'kin_corexy.c', 'kin_corexz.c', 'kin_delta.c', - 'kin_deltesian.c', 'kin_polar.c', 'kin_rotary_delta.c', 'kin_winch.c', - 'kin_extruder.c', 'kin_shaper.c', 'kin_idex.c', 'kin_generic.c' + "pyhelper.c", + "serialqueue.c", + "stepcompress.c", + "itersolve.c", + "trapq.c", + "pollreactor.c", + "msgblock.c", + "trdispatch.c", + "kin_cartesian.c", + "kin_corexy.c", + "kin_corexz.c", + "kin_delta.c", + "kin_deltesian.c", + "kin_polar.c", + "kin_rotary_delta.c", + "kin_winch.c", + "kin_extruder.c", + "kin_shaper.c", + "kin_idex.c", + "kin_generic.c", ] DEST_LIB = "c_helper.so" OTHER_FILES = [ - 'list.h', 'serialqueue.h', 'stepcompress.h', 'itersolve.h', 'pyhelper.h', - 'trapq.h', 'pollreactor.h', 'msgblock.h' + "list.h", + "serialqueue.h", + "stepcompress.h", + "itersolve.h", + "pyhelper.h", + "trapq.h", + "pollreactor.h", + "msgblock.h", ] defs_stepcompress = """ @@ -226,18 +249,33 @@ defs_std = """ """ defs_all = [ - defs_pyhelper, defs_serialqueue, defs_std, defs_stepcompress, - defs_itersolve, defs_trapq, defs_trdispatch, - defs_kin_cartesian, defs_kin_corexy, defs_kin_corexz, defs_kin_delta, - defs_kin_deltesian, defs_kin_polar, defs_kin_rotary_delta, defs_kin_winch, - defs_kin_extruder, defs_kin_shaper, defs_kin_idex, + defs_pyhelper, + defs_serialqueue, + defs_std, + defs_stepcompress, + defs_itersolve, + defs_trapq, + defs_trdispatch, + defs_kin_cartesian, + defs_kin_corexy, + defs_kin_corexz, + defs_kin_delta, + defs_kin_deltesian, + defs_kin_polar, + defs_kin_rotary_delta, + defs_kin_winch, + defs_kin_extruder, + defs_kin_shaper, + defs_kin_idex, defs_kin_generic_cartesian, ] + # Update filenames to an absolute path def get_abs_files(srcdir, filelist): return [os.path.join(srcdir, fname) for fname in filelist] + # Return the list of file modification times def get_mtimes(filelist): out = [] @@ -249,19 +287,21 @@ def get_mtimes(filelist): out.append(t) return out + # Check if the code needs to be compiled def check_build_code(sources, target): src_times = get_mtimes(sources) obj_times = get_mtimes([target]) return not obj_times or max(src_times) > min(obj_times) + # Check if the current gcc version supports a particular command-line option def check_gcc_option(option): - cmd = "%s %s -S -o /dev/null -xc /dev/null > /dev/null 2>&1" % ( - GCC_CMD, option) + cmd = "%s %s -S -o /dev/null -xc /dev/null > /dev/null 2>&1" % (GCC_CMD, option) res = os.system(cmd) return res == 0 + # Check if the current gcc version supports a particular command-line option def do_build_code(cmd): res = os.system(cmd) @@ -270,14 +310,17 @@ def do_build_code(cmd): logging.error(msg) raise Exception(msg) + FFI_main = None FFI_lib = None pyhelper_logging_callback = None + # Helper invoked from C errorf() code to log errors def logging_callback(msg): logging.error(FFI_main.string(msg)) + # Return the Foreign Function Interface api to the caller def get_ffi(): global FFI_main, FFI_lib, pyhelper_logging_callback @@ -286,20 +329,21 @@ def get_ffi(): srcfiles = get_abs_files(srcdir, SOURCE_FILES) ofiles = get_abs_files(srcdir, OTHER_FILES) destlib = get_abs_files(srcdir, [DEST_LIB])[0] - if check_build_code(srcfiles+ofiles+[__file__], destlib): + if check_build_code(srcfiles + ofiles + [__file__], destlib): if check_gcc_option(SSE_FLAGS): cmd = "%s %s %s" % (GCC_CMD, SSE_FLAGS, COMPILE_ARGS) else: cmd = "%s %s" % (GCC_CMD, COMPILE_ARGS) logging.info("Building C code module %s", DEST_LIB) - do_build_code(cmd % (destlib, ' '.join(srcfiles))) + do_build_code(cmd % (destlib, " ".join(srcfiles))) FFI_main = cffi.FFI() for d in defs_all: FFI_main.cdef(d) FFI_lib = FFI_main.dlopen(destlib) # Setup error logging - pyhelper_logging_callback = FFI_main.callback("void func(const char *)", - logging_callback) + pyhelper_logging_callback = FFI_main.callback( + "void func(const char *)", logging_callback + ) FFI_lib.set_python_logging_callback(pyhelper_logging_callback) return FFI_main, FFI_lib @@ -309,11 +353,12 @@ def get_ffi(): ###################################################################### HC_COMPILE_CMD = "gcc -Wall -g -O2 -o %s %s -lusb" -HC_SOURCE_FILES = ['hub-ctrl.c'] -HC_SOURCE_DIR = '../../lib/hub-ctrl' +HC_SOURCE_FILES = ["hub-ctrl.c"] +HC_SOURCE_DIR = "../../lib/hub-ctrl" HC_TARGET = "hub-ctrl" HC_CMD = "sudo %s/hub-ctrl -h 0 -P 2 -p %d" + def run_hub_ctrl(enable_power): srcdir = os.path.dirname(os.path.realpath(__file__)) hubdir = os.path.join(srcdir, HC_SOURCE_DIR) @@ -321,9 +366,9 @@ def run_hub_ctrl(enable_power): destlib = get_abs_files(hubdir, [HC_TARGET])[0] if check_build_code(srcfiles, destlib): logging.info("Building C code module %s", HC_TARGET) - do_build_code(HC_COMPILE_CMD % (destlib, ' '.join(srcfiles))) + do_build_code(HC_COMPILE_CMD % (destlib, " ".join(srcfiles))) os.system(HC_CMD % (hubdir, enable_power)) -if __name__ == '__main__': +if __name__ == "__main__": get_ffi() |