mirror of
https://github.com/andreili/klipper.git
synced 2025-08-23 19:34:06 +02:00
c09ca4cf5a
Move register_flush_callback() from mcu.py code to motion_queuing module. Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
# Helper code for low-level motion queuing and flushing
|
|
#
|
|
# Copyright (C) 2025 Kevin O'Connor <kevin@koconnor.net>
|
|
#
|
|
# This file may be distributed under the terms of the GNU GPLv3 license.
|
|
import logging
|
|
import chelper
|
|
|
|
class PrinterMotionQueuing:
|
|
def __init__(self, config):
|
|
self.printer = config.get_printer()
|
|
self.steppers = []
|
|
self.trapqs = []
|
|
self.stepcompress = []
|
|
self.steppersyncs = []
|
|
self.flush_callbacks = []
|
|
ffi_main, ffi_lib = chelper.get_ffi()
|
|
self.trapq_finalize_moves = ffi_lib.trapq_finalize_moves
|
|
def allocate_trapq(self):
|
|
ffi_main, ffi_lib = chelper.get_ffi()
|
|
trapq = ffi_main.gc(ffi_lib.trapq_alloc(), ffi_lib.trapq_free)
|
|
self.trapqs.append(trapq)
|
|
return trapq
|
|
def allocate_stepcompress(self, mcu, oid):
|
|
ffi_main, ffi_lib = chelper.get_ffi()
|
|
sc = ffi_main.gc(ffi_lib.stepcompress_alloc(oid),
|
|
ffi_lib.stepcompress_free)
|
|
self.stepcompress.append((mcu, sc))
|
|
return sc
|
|
def allocate_steppersync(self, mcu, serialqueue, move_count):
|
|
stepqueues = []
|
|
for sc_mcu, sc in self.stepcompress:
|
|
if sc_mcu is mcu:
|
|
stepqueues.append(sc)
|
|
ffi_main, ffi_lib = chelper.get_ffi()
|
|
ss = ffi_main.gc(
|
|
ffi_lib.steppersync_alloc(serialqueue, stepqueues, len(stepqueues),
|
|
move_count),
|
|
ffi_lib.steppersync_free)
|
|
self.steppersyncs.append((mcu, ss))
|
|
return ss
|
|
def register_stepper(self, config, stepper):
|
|
self.steppers.append(stepper)
|
|
def register_flush_callback(self, callback):
|
|
self.flush_callbacks.append(callback)
|
|
def flush_motion_queues(self, must_flush_time, max_step_gen_time):
|
|
for cb in self.flush_callbacks:
|
|
cb(must_flush_time)
|
|
for stepper in self.steppers:
|
|
stepper.generate_steps(max_step_gen_time)
|
|
def clean_motion_queues(self, trapq_free_time, clear_history_time):
|
|
for trapq in self.trapqs:
|
|
self.trapq_finalize_moves(trapq, trapq_free_time,
|
|
clear_history_time)
|
|
def wipe_trapq(self, trapq):
|
|
# Expire any remaining movement in the trapq (force to history list)
|
|
NEVER = 9999999999999999.
|
|
self.trapq_finalize_moves(trapq, NEVER, 0.)
|
|
def lookup_trapq_append(self):
|
|
ffi_main, ffi_lib = chelper.get_ffi()
|
|
return ffi_lib.trapq_append
|
|
|
|
def load_config(config):
|
|
return PrinterMotionQueuing(config)
|