From 6e665702419f8baa4740cbd1f05c844dd34364d9 Mon Sep 17 00:00:00 2001
From: jacobeva <jacob@jacobeva.co.uk>
Date: Wed, 2 Oct 2024 19:05:21 +0100
Subject: [PATCH] [SX128x] Add setDataRate method for LoRa modem (#1251)

---
 src/modules/SX128x/SX128x.cpp | 16 ++++++++++++++++
 src/modules/SX128x/SX128x.h   |  7 +++++++
 2 files changed, 23 insertions(+)

diff --git a/src/modules/SX128x/SX128x.cpp b/src/modules/SX128x/SX128x.cpp
index 5cbcc91c..a7fe165b 100644
--- a/src/modules/SX128x/SX128x.cpp
+++ b/src/modules/SX128x/SX128x.cpp
@@ -904,6 +904,22 @@ int16_t SX128x::setPreambleLength(uint32_t preambleLength) {
   return(RADIOLIB_ERR_WRONG_MODEM);
 }
 
+int16_t SX128x::setDataRate(DataRate_t dr) {
+  // check active modem
+  uint8_t modem = getPacketType();
+  int16_t state = RADIOLIB_ERR_NONE;
+  if (modem == RADIOLIB_SX128X_PACKET_TYPE_LORA) {
+      state = this->setBandwidth(dr.lora.bandwidth);
+      RADIOLIB_ASSERT(state);
+      state = this->setSpreadingFactor(dr.lora.spreadingFactor);
+      RADIOLIB_ASSERT(state);
+      state = this->setCodingRate(dr.lora.codingRate);
+  } else {
+      return(RADIOLIB_ERR_WRONG_MODEM);
+  }
+  return(state);
+}
+
 int16_t SX128x::setBitRate(float br) {
   // check active modem
   uint8_t modem = getPacketType();
diff --git a/src/modules/SX128x/SX128x.h b/src/modules/SX128x/SX128x.h
index 02162449..6f584ead 100644
--- a/src/modules/SX128x/SX128x.h
+++ b/src/modules/SX128x/SX128x.h
@@ -679,6 +679,13 @@ class SX128x: public PhysicalLayer {
     */
     int16_t setPreambleLength(uint32_t preambleLength);
 
+    /*!
+      \brief Set data rate.
+      \param dr Data rate struct. Interpretation depends on currently active modem (FSK or LoRa).
+      \returns \ref status_codes
+    */
+    int16_t setDataRate(DataRate_t dr) override;
+
     /*!
       \brief Sets FSK or FLRC bit rate. Allowed values are 125, 250, 400, 500, 800, 1000,
       1600 and 2000 kbps (for FSK modem) or 260, 325, 520, 650, 1000 and 1300 (for FLRC modem).