soc: rw612: Reinitialize the GDET sensor and Voltage rails

Before re-entering Power Mode 3, we should disable the GET
sensor and reinitialize the power rails.

Signed-off-by: Mahesh Mahadevan <[email protected]>

Author: mmahadevan108

boards/nxp/frdm_rw612/init.c

@@ -36,6 +36,13 @@ void board_early_init_hook(void)
 {
 	frdm_rw612_power_init_config();
 
+	/* If this is a wakeup from PM3 then return after configuring
+	 * the power supplies.
+	 */
+	if (PMU->PWR_MODE_STATUS == 2U) {
+		return;
+	}
+
 #if CONFIG_PM
 	static struct pm_notifier frdm_rw612_pm_notifier = {
 		.state_exit = frdm_rw612_pm_state_exit,

boards/nxp/rd_rw612_bga/init.c

@@ -36,6 +36,13 @@ void board_early_init_hook(void)
 {
 	rdrw61x_power_init_config();
 
+	/* If this is a wakeup from PM3 then return after configuring
+	 * the power supplies.
+	 */
+	if (PMU->PWR_MODE_STATUS == 2U) {
+		return;
+	}
+
 #if CONFIG_PM
 	static struct pm_notifier rdrw61x_pm_notifier = {
 		.state_exit = rdrw61x_pm_state_exit,

soc/nxp/rw/power.c

@@ -17,7 +17,7 @@
 #endif
 #include <zephyr/drivers/timer/system_timer.h>
 #include <zephyr/drivers/timer/nxp_os_timer.h>
-
+#include <zephyr/platform/hooks.h>
 #include "fsl_power.h"
 
 #include <zephyr/logging/log.h>
@@ -228,6 +228,14 @@ __weak void pm_state_set(enum pm_state state, uint8_t substate_id)
 				NVIC_ClearPendingIRQ(DT_IRQN(DT_NODELABEL(rtc)));
 				sys_clock_idle_exit();
 				sys_clock_set_timeout(0, true);
+				/* GDET got enabled when exiting PM3, disable it
+				 * again before re-entering PM3.
+				 */
+				POWER_DisableGDetVSensors();
+
+				/* Reinitialize the board specific power rails */
+				board_early_init_hook();
+
 				if (!(POWER_EnterPowerMode(POWER_MODE3, &slp_cfg))) {
 					break;
 				}