From 04d29ef2fe889246dd578b5e718c5ff1665b95e2 Mon Sep 17 00:00:00 2001
From: mattmcw <matt@sixteenmillimeter.com>
Date: Sun, 30 Jun 2024 13:41:28 -0400
Subject: [PATCH] Fix methods

---
 TB6600MotorDriver.cpp | 4 ++--
 TB6600MotorDriver.h   | 3 +--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/TB6600MotorDriver.cpp b/TB6600MotorDriver.cpp
index 38e5936..3057cc6 100644
--- a/TB6600MotorDriver.cpp
+++ b/TB6600MotorDriver.cpp
@@ -23,10 +23,10 @@ void TB6600MotorDriver::setup() {
 	pinMode(_pulsePin, OUTPUT);
 
 	digitalWrite(_directionPin, LOW);
-	digitalWrite(_enablePin, HIGH);
+	digitalWrite(_enablePin, _enabled ? HIGH : LOW);
 }
 
-void TB6600MotorDriver::setDirection(uint8_t direction) {
+void TB6600MotorDriver::setDirection(bool direction) {
 	if (direction != _direction) {
 		_direction = direction;
 		_setDirection();
diff --git a/TB6600MotorDriver.h b/TB6600MotorDriver.h
index f13c487..8e44cd3 100644
--- a/TB6600MotorDriver.h
+++ b/TB6600MotorDriver.h
@@ -35,10 +35,9 @@ class TB6600MotorDriver {
   TB6600MotorDriver();
   TB6600MotorDriver(uint8_t enablePin, uint8_t directionPin, uint8_t pulsePin, uint32_t usPulse, uint8_t microsteps);
 
-
   void setup();
 
-  void setDirection(uint8_t direction);
+  void setDirection(bool direction);
   void setSpeed(uint16_t rpm);
 
   //full