soc: ti: Turns on power domains in soc layer

Turns on all the power domains for devices with a power-domain device
tree property in the soc layer. Supported for the R5f0_0 core on the
AM243x EVM.

Signed-off-by: Dave Joseph <[email protected]>

Author: Kronosblaster

boards/ti/am243x_evm/am243x_evm_am2434_r5f0_0_defconfig

@@ -15,3 +15,10 @@ CONFIG_UART_CONSOLE=y
 
 # Enable MPU
 CONFIG_ARM_MPU=y
+
+#Enable runtime power management
+CONFIG_PM_DEVICE=y
+CONFIG_POWER_DOMAIN=y
+CONFIG_PM=y
+CONFIG_PM_DEVICE_RUNTIME=y
+

soc/ti/k3/am6x/r5/soc.c

@@ -6,9 +6,12 @@
 
 #include <stdint.h>
 #include <zephyr/fatal.h>
+#include <zephyr/logging/log.h>
 
 #include "soc.h"
+#include "zephyr/device.h"
 #include <common/ctrl_partitions.h>
+#include <zephyr/pm/device_runtime.h>
 
 unsigned int z_soc_irq_get_active(void)
 {
@@ -53,3 +56,34 @@ void soc_early_init_hook(void)
 {
 	k3_unlock_all_ctrl_partitions();
 }
+LOG_MODULE_REGISTER(soc, LOG_LEVEL_DBG);
+
+#if CONFIG_SOC_AM2434_R5F0_0
+int soc_power_domain_init(void)
+{
+	LOG_DBG("Starting power domain initialization\n");
+
+	int ret = 0;
+
+#define CHECK_NODE_POWER_DOMAIN(child)                                                             \
+	if (DT_NODE_HAS_PROP(child, power_domains)) {                                              \
+		const struct device *dev_##child =                                                 \
+			DEVICE_DT_GET_OR_NULL(DT_PHANDLE(child, power_domains));                   \
+		if (dev_##child) {                                                                 \
+			printf("Turning on power domain: %s\n", dev_##child->name);                \
+			int err = pm_device_runtime_get(dev_##child);                              \
+			if (err < 0) {                                                             \
+				LOG_DBG("Failed to get power domain: %s\n", dev_##child->name);    \
+				if (ret) {                                                         \
+					return ret;                                                \
+				}                                                                  \
+			}                                                                          \
+		}                                                                                  \
+	}
+
+	DT_FOREACH_CHILD(DT_ROOT, CHECK_NODE_POWER_DOMAIN);
+
+	return ret;
+}
+SYS_INIT(soc_power_domain_init, APPLICATION, 99);
+#endif /* CONFIG_SOC_AM2434_R5F0_0 */