stepper: Disable "no next step" check for some count=1 queue_step commands

It's possible for a "count=1" stepper move immediately after a
set_next_step_dir or reset_step_clock command to be valid and have an
interval less than min_stop_interval.  Make sure this case does not
result in a shutdown.

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

1 files changed, 16 insertions, 5 deletions

diff --git a/src/stepper.c b/src/stepper.c
index 7b794d50..0ef81904 100644
--- a/src/stepper.c
+++ b/src/stepper.c
@@ -37,8 +37,9 @@ struct stepper {
     uint8_t flags : 8;
 };
 
-enum { MF_DIR=1 };
-enum { SF_LAST_DIR=1, SF_NEXT_DIR=2, SF_INVERT_STEP=4, SF_HAVE_ADD=8 };
+enum { MF_DIR=1<<0 };
+enum { SF_LAST_DIR=1<<0, SF_NEXT_DIR=1<<1, SF_INVERT_STEP=1<<2, SF_HAVE_ADD=1<<3,
+       SF_LAST_RESET=1<<4, SF_NO_NEXT_CHECK=1<<5 };
 
 // Setup a stepper for the next move in its queue
 static uint_fast8_t
@@ -46,7 +47,8 @@ stepper_load_next(struct stepper *s, uint32_t min_next_time)
 {
     struct move *m = s->first;
     if (!m) {
-        if (s->interval - s->add < s->min_stop_interval)
+        if (s->interval - s->add < s->min_stop_interval
+            && !(s->flags & SF_NO_NEXT_CHECK))
             shutdown("No next step");
         s->count = 0;
         return SF_DONE;
@@ -163,10 +165,16 @@ command_queue_step(uint32_t *args)
     m->flags = 0;
 
     irq_disable();
-    if (!!(s->flags & SF_LAST_DIR) != !!(s->flags & SF_NEXT_DIR)) {
-        s->flags ^= SF_LAST_DIR;
+    uint8_t flags = s->flags;
+    if (!!(flags & SF_LAST_DIR) != !!(flags & SF_NEXT_DIR)) {
+        flags ^= SF_LAST_DIR;
         m->flags |= MF_DIR;
     }
+    flags &= ~SF_NO_NEXT_CHECK;
+    if (m->count == 1 && (m->flags || flags & SF_LAST_RESET))
+        // count=1 moves after a reset or dir change can have small intervals
+        flags |= SF_NO_NEXT_CHECK;
+    s->flags = flags & ~SF_LAST_RESET;
     if (s->count) {
         if (s->first)
             *s->plast = m;
@@ -201,9 +209,12 @@ command_reset_step_clock(uint32_t *args)
 {
     struct stepper *s = lookup_oid(args[0], command_config_stepper);
     uint32_t waketime = args[1];
+    irq_disable();
     if (s->count)
         shutdown("Can't reset time when stepper active");
     s->next_step_time = waketime;
+    s->flags |= SF_LAST_RESET;
+    irq_enable();
 }
 DECL_COMMAND(command_reset_step_clock, "reset_step_clock oid=%c clock=%u");