From 96f042897a120ddf3f136cab8d0320153ee756a4 Mon Sep 17 00:00:00 2001
From: Harald Barth <haba@kth.se>
Date: Thu, 4 Aug 2022 01:21:28 +0200
Subject: [PATCH] Adopt setTrackMode to RMT channel pins

---
 DCC.cpp          |  3 ++-
 TrackManager.cpp | 10 +++++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/DCC.cpp b/DCC.cpp
index 0993f4c..3d2d056 100644
--- a/DCC.cpp
+++ b/DCC.cpp
@@ -68,8 +68,9 @@ void DCC::begin(const FSH * motorShieldName) {
   (void)EEPROM; // tell compiler not to warn this is unused
   EEStore::init();
 #endif
-
+#ifndef ARDUINO_ARCH_ESP32 /* On ESP32 started in TrackManager::setTrackMode() */
   DCCWaveform::begin();
+#endif
 }
 
 
diff --git a/TrackManager.cpp b/TrackManager.cpp
index 0aac3f7..ba3f8c0 100644
--- a/TrackManager.cpp
+++ b/TrackManager.cpp
@@ -127,7 +127,6 @@ void TrackManager::setDCSignal(int16_t cab, byte speedbyte) {
   }
 }    
 
-
 bool TrackManager::setTrackMode(byte trackToSet, TRACK_MODE mode, int16_t dcAddr) {
     if (trackToSet>lastTrack || track[trackToSet]==NULL) return false;
 
@@ -139,6 +138,11 @@ bool TrackManager::setTrackMode(byte trackToSet, TRACK_MODE mode, int16_t dcAddr
              return false; 
          }
 
+#ifdef ARDUINO_ARCH_ESP32
+    // remove pin from MUX matrix and turn it off
+    DIAG(F("Track=%c remove pin %d"),trackToSet+'A', track[trackToSet]->getSignalPin());
+    gpio_reset_pin((gpio_num_t)track[trackToSet]->getSignalPin());
+#endif
     if (mode==TRACK_MODE_PROG) {
       // only allow 1 track to be prog
       FOR_EACH_TRACK(t)
@@ -175,6 +179,7 @@ bool TrackManager::setTrackMode(byte trackToSet, TRACK_MODE mode, int16_t dcAddr
     // pin should be turned on
     track[trackToSet]->enableSignal(mode != TRACK_MODE_EXT);
 
+#ifndef ARDUINO_ARCH_ESP32
     // re-evaluate HighAccuracy mode
     // We can only do this is all main and prog tracks agree
     bool canDo=true;
@@ -205,6 +210,9 @@ bool TrackManager::setTrackMode(byte trackToSet, TRACK_MODE mode, int16_t dcAddr
       }
       DCCTimer::clearPWM(); // has to be AFTER trackPWM changes because if trackPWM==true this is undone for  that track
     }
+#else
+    DCCWaveform::begin();
+#endif
     
     // Normal running tracks are set to the global power state 
     track[trackToSet]->setPower(