idle_timeout: Add printing/ready/idle tracking

Internally track the overall printer state.  Generate events on state
transitions.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 65 insertions, 21 deletions

diff --git a/klippy/extras/idle_timeout.py b/klippy/extras/idle_timeout.py
index 39563b12..af3e773c 100644
--- a/klippy/extras/idle_timeout.py
+++ b/klippy/extras/idle_timeout.py
@@ -9,41 +9,85 @@ TURN_OFF_HEATERS
 M84
 """
 
+PIN_MIN_TIME = 0.100
+READY_TIMEOUT = .500
+
 class IdleTimeout:
     def __init__(self, config):
         self.printer = config.get_printer()
-        self.toolhead = None
-        self.last_timeout = 0.
+        self.reactor = self.printer.get_reactor()
+        self.gcode = self.printer.lookup_object('gcode')
+        self.toolhead = self.timeout_timer = None
+        self.state = "Idle"
         self.idle_timeout = config.getfloat('timeout', 600., above=0.)
         self.idle_gcode = config.get('gcode', DEFAULT_IDLE_GCODE).split('\n')
     def printer_state(self, state):
         if state == 'ready':
             self.toolhead = self.printer.lookup_object('toolhead')
-            reactor = self.printer.get_reactor()
-            reactor.register_timer(self.timeout_handler, reactor.NOW)
-    def run_idle_script(self):
-        gcode = self.printer.lookup_object('gcode')
+            self.timeout_timer = self.reactor.register_timer(
+                self.timeout_handler)
+            self.printer.register_event_handler("toolhead:sync_print_time",
+                                                self.handle_sync_print_time)
+    def transition_idle_state(self, eventtime):
+        self.state = "Printing"
         try:
-            res = gcode.process_batch(self.idle_gcode)
+            res = self.gcode.process_batch(self.idle_gcode)
         except:
             logging.exception("idle timeout gcode execution")
-            return
+            return eventtime + 1.
         if not res:
             # Raced with incoming g-code commands
-            return
-        self.last_timeout = self.toolhead.get_last_move_time()
-    def timeout_handler(self, eventtime):
-        info = self.toolhead.get_status(eventtime)
-        print_time = info['print_time']
-        status = info['status']
-        if print_time <= self.last_timeout or status == 'Printing':
+            return eventtime + 1.
+        print_time = self.toolhead.get_last_move_time()
+        self.state = "Idle"
+        self.printer.send_event("idle_timeout:idle", print_time)
+        return self.reactor.NEVER
+    def check_idle_timeout(self, eventtime):
+        # Make sure toolhead class isn't busy
+        print_time, est_print_time, lookahead_empty = self.toolhead.check_busy(
+            eventtime)
+        idle_time = est_print_time - print_time
+        if not lookahead_empty or idle_time < 1.:
+            # Toolhead is busy
             return eventtime + self.idle_timeout
-        estimated_print_time = info['estimated_print_time']
-        elapsed_time = estimated_print_time - print_time
-        if elapsed_time < self.idle_timeout:
-            return eventtime + self.idle_timeout - elapsed_time
-        self.run_idle_script()
-        return eventtime + 1.
+        if idle_time < self.idle_timeout:
+            # Wait for idle timeout
+            return eventtime + self.idle_timeout - idle_time
+        if not self.gcode.process_batch([]):
+            # Gcode class busy
+            return eventtime + 1.
+        # Idle timeout has elapsed
+        return self.transition_idle_state(eventtime)
+    def timeout_handler(self, eventtime):
+        if self.state == "Ready":
+            return self.check_idle_timeout(eventtime)
+        # Check if need to transition to "ready" state
+        print_time, est_print_time, lookahead_empty = self.toolhead.check_busy(
+            eventtime)
+        buffer_time = min(2., print_time - est_print_time)
+        if not lookahead_empty:
+            # Toolhead is busy
+            return eventtime + READY_TIMEOUT + max(0., buffer_time)
+        if buffer_time > -READY_TIMEOUT:
+            # Wait for ready timeout
+            return eventtime + READY_TIMEOUT + buffer_time
+        if not self.gcode.process_batch([]):
+            # Gcode class busy
+            return eventtime + READY_TIMEOUT
+        # Transition to "ready" state
+        self.state = "Ready"
+        self.printer.send_event("idle_timeout:ready",
+                                est_print_time + PIN_MIN_TIME)
+        return eventtime + self.idle_timeout
+    def handle_sync_print_time(self, curtime, print_time, est_print_time):
+        if self.state == "Printing":
+            return
+        # Transition to "printing" state
+        self.state = "Printing"
+        check_time = READY_TIMEOUT + print_time - est_print_time
+        self.reactor.update_timer(self.timeout_timer, curtime + check_time)
+        self.printer.send_event("idle_timeout:printing",
+                                est_print_time + PIN_MIN_TIME)
 
 def load_config(config):
     return IdleTimeout(config)