From aebca646d9c7af5ed659089ce08f8b6db113c7d5 Mon Sep 17 00:00:00 2001
From: Harald Barth <haba@kth.se>
Date: Sat, 28 Sep 2024 22:10:46 +0200
Subject: [PATCH] Move railcom decisions into same block

---
 DCCWaveform.cpp | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

diff --git a/DCCWaveform.cpp b/DCCWaveform.cpp
index d61de9d..89d0804 100644
--- a/DCCWaveform.cpp
+++ b/DCCWaveform.cpp
@@ -149,26 +149,28 @@ void DCCWaveform::interrupt2() {
   if (remainingPreambles > 0 ) {
     state=WAVE_MID_1;  // switch state to trigger LOW on next interrupt
     
-    // predict railcom cutout on next interrupt 
-    cutoutNextTime= remainingPreambles==requiredPreambles
-           && railcomActive 
-           && isMainTrack;
-
     remainingPreambles--;
   
     // As we get to the end of the preambles, open the reminder window.
     // This delays any reminder insertion until the last moment so
     // that the reminder doesn't block a more urgent packet. 
     reminderWindowOpen=transmitRepeats==0 && remainingPreambles<4 && remainingPreambles>1;
-    if (remainingPreambles==1) promotePendingPacket();
+    if (remainingPreambles==1)
+      promotePendingPacket();
     else if (isMainTrack && railcomActive) {
-      // cutout has ended so its now possible to poll the railcom detectors
-      // requiredPreambles is one higher that preamble length so
-      // if preamble length is 16 then this evaluates to 5
-      if (remainingPreambles==(requiredPreambles-12)) railcomSampleWindow=true;
-      // cutout can be ended when read
-      // see above for requiredPreambles
-      else if (remainingPreambles==(requiredPreambles-3)) DCCTimer::ackRailcomTimer();
+      if (remainingPreambles==(requiredPreambles-1)) {
+        // First look if we need to start a railcom cutout on next interrupt
+        cutoutNextTime= true;
+      } else if (remainingPreambles==(requiredPreambles-12)) {
+        // cutout has ended so its now possible to poll the railcom detectors
+        // requiredPreambles is one higher that preamble length so
+        // if preamble length is 16 then this evaluates to 5
+        railcomSampleWindow=true;
+      } else if (remainingPreambles==(requiredPreambles-3)) {
+        // cutout can be ended when read
+        // see above for requiredPreambles
+        DCCTimer::ackRailcomTimer();
+      }
     }
     // Update free memory diagnostic as we don't have anything else to do this time.
     // Allow for checkAck and its called functions using 22 bytes more.