From 54773297bf6698f2718b785266e90d809f2d7cb2 Mon Sep 17 00:00:00 2001
From: Mike S <CyberMerln@gmail.com>
Date: Fri, 26 Mar 2021 07:37:17 -0400
Subject: [PATCH] Minor changes to Arduino Type List & DCCWaveform

---
 DCC.h           | 12 ++++++++++--
 DCCWaveform.cpp |  7 +++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/DCC.h b/DCC.h
index a224fb9..c4a6ae3 100644
--- a/DCC.h
+++ b/DCC.h
@@ -168,8 +168,16 @@ private:
 #define ARDUINO_TYPE "MEGA"
 #elif defined(ARDUINO_ARCH_MEGAAVR)
 #define ARDUINO_TYPE "MEGAAVR"
-#elif defined(TEENSYDUINO)
-#define ARDUINO_TYPE "TEENSY"
+#elif defined(ARDUINO_TEENSY32)
+#define ARDUINO_TYPE "TEENSY32"
+#elif defined(ARDUINO_TEENSY35)
+#define ARDUINO_TYPE "TEENSY35"
+#elif defined(ARDUINO_TEENSY36)
+#define ARDUINO_TYPE "TEENSY36"
+#elif defined(ARDUINO_TEENSY40)
+#define ARDUINO_TYPE "TEENSY40"
+#elif defined(ARDUINO_TEENSY41)
+#define ARDUINO_TYPE "TEENSY41"
 #else
 #error CANNOT COMPILE - DCC++ EX ONLY WORKS WITH AN ARDUINO UNO, NANO 328, OR ARDUINO MEGA 1280/2560
 #endif
diff --git a/DCCWaveform.cpp b/DCCWaveform.cpp
index c87f768..125bc93 100644
--- a/DCCWaveform.cpp
+++ b/DCCWaveform.cpp
@@ -52,12 +52,19 @@ void DCCWaveform::begin(MotorDriver * mainDriver, MotorDriver * progDriver) {
   DCCTimer::begin(DCCWaveform::interruptHandler);     
 }
 
+#if defined(TEENSYDUINO)
 void DCCWaveform::loop(bool ackManagerActive) {
   noInterrupts();
   mainTrack.checkPowerOverload(false);
   progTrack.checkPowerOverload(ackManagerActive);
   interrupts();
 }
+#else
+void DCCWaveform::loop(bool ackManagerActive) {
+  mainTrack.checkPowerOverload(false);
+  progTrack.checkPowerOverload(ackManagerActive);
+}
+#endif
 
 void DCCWaveform::interruptHandler() {
   // call the timer edge sensitive actions for progtrack and maintrack