From e76197faa97f2aa0d6a488823456a5fcaba5ad7b Mon Sep 17 00:00:00 2001
From: peteGSX <peteracole@outlook.com.au>
Date: Sun, 29 Jan 2023 19:26:33 +1000
Subject: [PATCH] Brief PWM start

---
 IO_EXIOExpander.h | 38 ++++++++++++++++++++++++++++++--------
 1 file changed, 30 insertions(+), 8 deletions(-)

diff --git a/IO_EXIOExpander.h b/IO_EXIOExpander.h
index 5c10d88..5476a41 100644
--- a/IO_EXIOExpander.h
+++ b/IO_EXIOExpander.h
@@ -90,6 +90,18 @@ private:
         _analoguePinBytes = _numAnaloguePins * 2;
         _analogueInputStates = (byte*) calloc(_analoguePinBytes, 1);
         _analoguePinMap = (uint8_t*) calloc(_numAnaloguePins, 1);
+      } else {
+      _command2Buffer[0] = EXIOINIT;
+      _command2Buffer[1] = _nPins;
+      // Send config, if EXIOINITA returned, we're good, setup analogue input buffer, otherwise go offline
+      I2CManager.read(_i2cAddress, _receive3Buffer, 3, _command2Buffer, 2);
+      if (_receive3Buffer[0] == EXIOINITA) {
+        _numAnaloguePins = _receive3Buffer[1];
+        _numPWMPins = _receive3Buffer[2];
+        _analoguePinBytes = _numAnaloguePins * 2;
+        _analogueInputStates = (byte*) calloc(_analoguePinBytes, 1);
+        _analoguePinMap = (uint8_t*) calloc(_numAnaloguePins, 1);
+
       } else {
         DIAG(F("ERROR configuring EX-IOExpander device, I2C:x%x"), _i2cAddress);
         _deviceState = DEVSTATE_FAILED;
@@ -123,13 +135,23 @@ private:
   // Digital input pin configuration, used to enable on EX-IOExpander device and set pullups if in use
   bool _configure(VPIN vpin, ConfigTypeEnum configType, int paramCount, int params[]) override {
     if (paramCount != 1) return false;
-    bool pullup = params[0];
     int pin = vpin - _firstVpin;
-    _digitalOutBuffer[0] = EXIODPUP;
-    _digitalOutBuffer[1] = pin;
-    _digitalOutBuffer[2] = pullup;
-    I2CManager.write(_i2cAddress, _digitalOutBuffer, 3);
-    return true;
+    if (configType == CONFIGURE_INPUT) {
+      bool pullup = params[0];
+      _digitalOutBuffer[0] = EXIODPUP;
+      _digitalOutBuffer[1] = pin;
+      _digitalOutBuffer[2] = pullup;
+      I2CManager.write(_i2cAddress, _digitalOutBuffer, 3);
+      return true;
+    } else if (configType == CONFIGURE_SERVO) {
+      DIAG(F("Configure servo at pin %d"), (int)pin);
+      for (int i = 0; i < paramCount; i++) {
+        DIAG(F("Param %d is %x"), (int)i, params[i]);
+      }
+      return true;
+    } else {
+      return false;
+    }
   }
 
   // Analogue input pin configuration, used to enable on EX-IOExpander device
@@ -189,7 +211,7 @@ private:
   }
 
   uint8_t _i2cAddress;
-  uint8_t _numAnaloguePins;
+  uint8_t _numAnaloguePins = 0;
   byte _digitalOutBuffer[3];
   uint8_t _versionBuffer[3];
   uint8_t _majorVer = 0;
@@ -201,7 +223,7 @@ private:
   uint8_t _analoguePinBytes = 0;
   byte _command1Buffer[1];
   byte _command2Buffer[2];
-  byte _receive2Buffer[2];
+  byte _receive3Buffer[3];
   uint8_t* _analoguePinMap;
   byte _command1Buffer[1];
   byte _command2Buffer[2];