From e50587344e4977fcc04c04df53caee9b6ebf85f6 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 25 Jun 2019 14:16:53 +0200
Subject: RF24: toggles, add setPayloadSize

---
 src/driver/nrf24l01.cc | 42 ++++++++++++++++++++++++++----------------
 1 file changed, 26 insertions(+), 16 deletions(-)

(limited to 'src/driver/nrf24l01.cc')

diff --git a/src/driver/nrf24l01.cc b/src/driver/nrf24l01.cc
index a678655..0bb4656 100644
--- a/src/driver/nrf24l01.cc
+++ b/src/driver/nrf24l01.cc
@@ -100,6 +100,11 @@ void Nrf24l01::setRetries(uint8_t delay, uint8_t count)
 	writeRegister(SETUP_RETR, (delay & 0xf) << ARD | (count & 0xf) << ARC);
 }
 
+void Nrf24l01::setPayloadSize(uint8_t size)
+{
+	payload_size = rf24_min(size, 32);
+}
+
 void Nrf24l01::setPALevel(uint8_t level)
 {
 	uint8_t setup = readRegister(RF_SETUP) & 0b11111000;
@@ -327,27 +332,32 @@ uint8_t Nrf24l01::writeRegister(uint8_t reg, uint8_t value)
 	return rxbuf[0];
 }
 
-void Nrf24l01::enableDynamicPayloads(void)
+void Nrf24l01::enableDynamicPayloads(const bool enabled)
 {
-	// Enable dynamic payload throughout the system
-
-	//toggle_features();
-	writeRegister(FEATURE, readRegister(FEATURE) | (1 << EN_DPL));
-
-	//IF_SERIAL_DEBUG(printf("FEATURE=%i\r\n",read_register(FEATURE)));
-
-	// Enable dynamic payload on all pipes
-	//
-	// Not sure the use case of only having dynamic payload on certain
-	// pipes, so the library does not support it.
-	writeRegister(DYNPD, readRegister(DYNPD) | (1 << DPL_P5) | (1 << DPL_P4) | (1 << DPL_P3) | (1 << DPL_P2) | (1 << DPL_P1) | (1 << DPL_P0));
+	if (enabled)
+	{
+		writeRegister(FEATURE, readRegister(FEATURE) | (1 << EN_DPL));
+		writeRegister(DYNPD, readRegister(DYNPD) | (1 << DPL_P5) | (1 << DPL_P4) | (1 << DPL_P3) | (1 << DPL_P2) | (1 << DPL_P1) | (1 << DPL_P0));
+	}
+	else
+	{
+		writeRegister(FEATURE, readRegister(FEATURE) & ~(1 << EN_DPL));
+		writeRegister(DYNPD, readRegister(DYNPD) & ~((1 << DPL_P5) | (1 << DPL_P4) | (1 << DPL_P3) | (1 << DPL_P2) | (1 << DPL_P1) | (1 << DPL_P0)));
+	}
 
-	dynamic_payloads_enabled = true;
+	dynamic_payloads_enabled = enabled;
 }
 
-void Nrf24l01::enableDynamicAck(void)
+void Nrf24l01::enableDynamicAck(const bool enabled)
 {
-	writeRegister(FEATURE, readRegister(FEATURE) | (1 << EN_DYN_ACK));
+	if (enabled)
+	{
+		writeRegister(FEATURE, readRegister(FEATURE) | (1 << EN_DYN_ACK));
+	}
+	else
+	{
+		writeRegister(FEATURE, readRegister(FEATURE) & ~(1 << EN_DYN_ACK));
+	}
 }
 
 uint8_t Nrf24l01::writePayload(const void *buf, uint8_t data_len, const uint8_t writeType)
-- 
cgit v1.2.3