From fd803c4b1166e75f88625c435693fb56d4a00c7f Mon Sep 17 00:00:00 2001
From: Martino Facchin <m.facchin@arduino.cc>
Date: Tue, 3 Sep 2024 13:22:49 +0200
Subject: [PATCH 1/2] distance: allow both VL53L4CD and VL53L4ED

---
 src/Modulino.h | 85 +++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 73 insertions(+), 12 deletions(-)

diff --git a/src/Modulino.h b/src/Modulino.h
index ee29261..f790346 100644
--- a/src/Modulino.h
+++ b/src/Modulino.h
@@ -4,6 +4,7 @@
 #include "Wire.h"
 #include <vector>
 #include <vl53l4cd_class.h>  // from stm32duino
+#include <vl53l4ed_class.h>  // from stm32duino
 #include "Arduino_LSM6DSOX.h"
 #include <Arduino_LPS22HB.h>
 #include <Arduino_HS300x.h>
@@ -396,6 +397,52 @@ class ModulinoLight : public Module {
 
 };
 
+class _distance_api {
+public:
+  _distance_api(VL53L4CD* sensor) : sensor(sensor) {
+    isVL53L4CD = true;
+  };
+  _distance_api(VL53L4ED* sensor) : sensor(sensor) {};
+  uint8_t setRangeTiming(uint32_t timing_budget_ms, uint32_t inter_measurement_ms) {
+    if (isVL53L4CD) {
+      return ((VL53L4CD*)sensor)->VL53L4CD_SetRangeTiming(timing_budget_ms, inter_measurement_ms);
+    } else {
+      return ((VL53L4ED*)sensor)->VL53L4ED_SetRangeTiming(timing_budget_ms, inter_measurement_ms);
+    }
+  }
+  uint8_t startRanging() {
+    if (isVL53L4CD) {
+      return ((VL53L4CD*)sensor)->VL53L4CD_StartRanging();
+    } else {
+      return ((VL53L4ED*)sensor)->VL53L4ED_StartRanging();
+    }
+  }
+  uint8_t checkForDataReady(uint8_t* p_is_data_ready) {
+    if (isVL53L4CD) {
+      return ((VL53L4CD*)sensor)->VL53L4CD_CheckForDataReady(p_is_data_ready);
+    } else {
+      return ((VL53L4ED*)sensor)->VL53L4ED_CheckForDataReady(p_is_data_ready);
+    }
+  }
+  uint8_t clearInterrupt() {
+    if (isVL53L4CD) {
+      return ((VL53L4CD*)sensor)->VL53L4CD_ClearInterrupt();
+    } else {
+      return ((VL53L4ED*)sensor)->VL53L4ED_ClearInterrupt();
+    }
+  }
+  uint8_t getResult(void* result) {
+    if (isVL53L4CD) {
+      return ((VL53L4CD*)sensor)->VL53L4CD_GetResult((VL53L4CD_Result_t*)result);
+    } else {
+      return ((VL53L4ED*)sensor)->VL53L4ED_GetResult((VL53L4ED_ResultsData_t*)result);
+    }
+  }
+private:
+  void* sensor;
+  bool isVL53L4CD = false;
+};
+
 class ModulinoDistance : public Module {
 public:
   bool begin() {
@@ -406,29 +453,40 @@ class ModulinoDistance : public Module {
     }
     tof_sensor = new VL53L4CD((TwoWire*)getWire(), -1);
     auto ret = tof_sensor->InitSensor();
-    __increaseI2CPriority();
-    if (ret == VL53L4CD_ERROR_NONE) {
-      tof_sensor->VL53L4CD_SetRangeTiming(20, 0);
-      tof_sensor->VL53L4CD_StartRanging();
-      return true;
-    } else {
+    if (ret != VL53L4CD_ERROR_NONE) {
+      delete tof_sensor;
       tof_sensor = nullptr;
-      return false;
+      tof_sensor_alt = new VL53L4ED((TwoWire*)getWire(), -1);
+      ret = tof_sensor_alt->InitSensor();
+      if (ret == VL53L4ED_ERROR_NONE) {
+        api = new _distance_api(tof_sensor_alt);
+      } else {
+        delete tof_sensor_alt;
+        tof_sensor_alt = nullptr;
+        return false;
+      }
+    } else {
+      api = new _distance_api(tof_sensor);
     }
+
+    __increaseI2CPriority();
+    api->setRangeTiming(20, 0);
+    api->startRanging();
+    return true;
   }
   operator bool() {
-    return (tof_sensor != nullptr);
+    return (api != nullptr);
   }
   bool available() {
-    if (tof_sensor == nullptr) {
+    if (api == nullptr) {
       return false;
     }
     float ret = internal;
     uint8_t NewDataReady = 0;
-    tof_sensor->VL53L4CD_CheckForDataReady(&NewDataReady);
+    api->checkForDataReady(&NewDataReady);
     if (NewDataReady) {
-      tof_sensor->VL53L4CD_ClearInterrupt();
-      tof_sensor->VL53L4CD_GetResult(&results);
+      api->clearInterrupt();
+      api->getResult(&results);
     }
     if (results.range_status == 0) {
       internal = results.distance_mm;
@@ -442,6 +500,9 @@ class ModulinoDistance : public Module {
   }
 private:
   VL53L4CD* tof_sensor = nullptr;
+  VL53L4ED* tof_sensor_alt = nullptr;
   VL53L4CD_Result_t results;
+  //VL53L4ED_ResultsData_t results;
   float internal = NAN;
+  _distance_api* api = nullptr;
 };
\ No newline at end of file

From b7a2d495a2b3ed1061f9b588eb8d0329caa08d95 Mon Sep 17 00:00:00 2001
From: Martino Facchin <m.facchin@arduino.cc>
Date: Tue, 3 Sep 2024 14:35:42 +0200
Subject: [PATCH 2/2] gh: add new dependency

---
 .github/workflows/compile-examples.yml | 1 +
 library.properties                     | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/compile-examples.yml b/.github/workflows/compile-examples.yml
index 7c9ed7c..c058661 100644
--- a/.github/workflows/compile-examples.yml
+++ b/.github/workflows/compile-examples.yml
@@ -47,6 +47,7 @@ jobs:
             - source-path: ./
             # Install library dependencies.
             - name: STM32duino VL53L4CD
+            - name: STM32duino VL53L4ED
             - name: Arduino_LSM6DSOX
             - name: Arduino_LPS22HB
             - name: Arduino_HS300x
diff --git a/library.properties b/library.properties
index 6234a79..3141f03 100644
--- a/library.properties
+++ b/library.properties
@@ -8,4 +8,4 @@ category=Communication
 url=https://github.com/arduino-libraries/Modulino
 architectures=*
 includes=Modulino.h
-depends=STM32duino VL53L4CD,Arduino_LSM6DSOX,Arduino_LPS22HB,Arduino_HS300x
+depends=STM32duino VL53L4CD,STM32duino VL53L4ED,Arduino_LSM6DSOX,Arduino_LPS22HB,Arduino_HS300x