From 729aafbb44d284694453cf71e6ae112adf60d12f Mon Sep 17 00:00:00 2001
From: Riccardo Rizzo <r.rizzo@arduino.cc>
Date: Fri, 10 Jun 2022 15:27:25 +0200
Subject: [PATCH] Added Charging Safety Timer Management

Added Charging Safety Timer Management in order to fix https://github.com/arduino-libraries/Arduino_BQ24195/issues/11
---
 src/BQ24195.cpp | 62 ++++++++++++++++++++++++++++++++++++++++++++++++-
 src/BQ24195.h   |  2 ++
 2 files changed, 63 insertions(+), 1 deletion(-)

diff --git a/src/BQ24195.cpp b/src/BQ24195.cpp
index 04d6aeb..db18f60 100644
--- a/src/BQ24195.cpp
+++ b/src/BQ24195.cpp
@@ -459,6 +459,15 @@ bool PMICClass::enableCharging() {
     DATA = DATA & 0b11001111;
     DATA = DATA | 0b00010000;
     return writeRegister(POWERON_CONFIG_REGISTER, DATA);
+
+    // enable Charging Safety Timer
+    DATA = readRegister(CHARGE_TIMER_CONTROL_REGISTER);
+
+    if (DATA == -1) {
+        return 0;
+    }
+
+    return writeRegister(CHARGE_TIMER_CONTROL_REGISTER, (DATA | 0b00001000));
 }
 
 /*******************************************************************************
@@ -475,7 +484,18 @@ bool PMICClass::disableCharging() {
         return 0;
     }
 
-    return writeRegister(POWERON_CONFIG_REGISTER, DATA & 0xCF);
+    if (writeRegister(POWERON_CONFIG_REGISTER, DATA & 0xCF)){
+        return 0;
+    }
+
+    // disable Charging Safety Timer
+    DATA = readRegister(CHARGE_TIMER_CONTROL_REGISTER);
+
+    if (DATA == -1) {
+        return 0;
+    }
+
+    return writeRegister(CHARGE_TIMER_CONTROL_REGISTER, (DATA & 0b11110110));
 }
 
 /*******************************************************************************
@@ -773,6 +793,46 @@ bool PMICClass::disableWatchdog(void) {
     return writeRegister(CHARGE_TIMER_CONTROL_REGISTER, (DATA & 0b11001110));
 }
 
+/*******************************************************************************
+ * Function Name  : enableSafetyChargeTimer
+ * Description    : Enable Safety Charge timer
+ * Input          : NONE
+ * Return         : 0 on Error, 1 on Success
+ *******************************************************************************/
+bool PMICClass::enableSafetyChargeTimer(void) {
+
+    int DATA = readRegister(CHARGE_TIMER_CONTROL_REGISTER);
+
+    if (DATA == -1) {
+        return 0;
+    }
+
+    return writeRegister(CHARGE_TIMER_CONTROL_REGISTER, (DATA | 0b00001000));
+}
+
+
+/*******************************************************************************
+ * Function Name  : disableSafetyChargeTimer
+ * Description    : Disable Safety Charge timer
+ * Input          : NONE
+ * Return         : 0 on Error, 1 on Success
+ *******************************************************************************/
+bool PMICClass::disableSafetyChargeTimer(void) {
+    int DATA = readRegister(CHARGE_TIMER_CONTROL_REGISTER) & 0x30;
+
+    if ((DATA == 0x10) && (PMIC.chargeStatus() != CHARGE_TERMINATION_DONE)){
+        return 0;
+    }
+
+    DATA = readRegister(CHARGE_TIMER_CONTROL_REGISTER);
+
+    if (DATA == -1) {
+        return 0;
+    }
+
+    return writeRegister(CHARGE_TIMER_CONTROL_REGISTER, (DATA & 0b11110110));
+}
+
 /*******************************************************************************
  * Function Name  : setThermalRegulationTemperature
  * Description    : Sets the Thermal Regulation Threshold
diff --git a/src/BQ24195.h b/src/BQ24195.h
index 7daa0bc..cf22d36 100644
--- a/src/BQ24195.h
+++ b/src/BQ24195.h
@@ -86,6 +86,8 @@ class PMICClass {
 
     // Charge Timer Control Register
     bool disableWatchdog(void);
+    bool enableSafetyChargeTimer(void);
+    bool disableSafetyChargeTimer(void);
 
     // Misc Operation Control Register
     bool enableDPDM(void);