driver: dma: dma_silabs_siwx91x: Add pm device support for dma driver

This commit enables the pm device driver support
for the dma_silabs_siwx91x driver.

Signed-off-by: S Mohamed Fiaz <[email protected]>

Author: fimohame

drivers/dma/dma_silabs_siwx91x.c

@@ -13,6 +13,7 @@
 #include <zephyr/drivers/dma.h>
 #include <zephyr/drivers/clock_control.h>
 #include <zephyr/logging/log.h>
+#include <zephyr/pm/device.h>
 #include <zephyr/types.h>
 #include "rsi_rom_udma.h"
 #include "rsi_rom_udma_wrapper.h"
@@ -60,6 +61,46 @@ struct dma_siwx91x_data {
 					      */
 };
 
+__maybe_unused static int dma_siwx91x_pm_action(const struct device *dev, enum pm_device_action action)
+{
+	const struct dma_siwx91x_config *cfg = dev->config;
+	struct dma_siwx91x_data *data = dev->data;
+	void *udma_handle = NULL;
+	UDMA_RESOURCES udma_resources = {
+		.reg = cfg->reg,
+		.udma_irq_num = cfg->irq_number,
+		.desc = cfg->sram_desc_addr,
+	};
+	int ret;
+
+	switch (action) {
+	case PM_DEVICE_ACTION_RESUME:
+		ret = clock_control_on(cfg->clock_dev, cfg->clock_subsys);
+		if (ret) {
+			return ret;
+		}
+
+		udma_handle = UDMAx_Initialize(&udma_resources, udma_resources.desc, NULL,
+					       (uint32_t *)&data->udma_handle);
+		if (udma_handle != &data->udma_handle) {
+			return -EINVAL;
+		}
+
+		cfg->irq_configure();
+
+		if (UDMAx_DMAEnable(&udma_resources, udma_handle) != 0) {
+			return -EBUSY;
+		}
+
+		break;
+	case PM_DEVICE_ACTION_SUSPEND:
+		return 0;
+	default:
+		return -ENOTSUP;
+	}
+	return 0;
+}
+
 static enum dma_xfer_dir siwx91x_transfer_direction(uint32_t dir)
 {
 	if (dir == MEMORY_TO_MEMORY) {
@@ -591,7 +632,7 @@ static int siwx91x_dma_init(const struct device *dev)
 		return -EBUSY;
 	}
 
-	return 0;
+	return pm_device_driver_init(dev, dma_siwx91x_pm_action);
 }
 
 static void siwx91x_dma_isr(const struct device *dev)
@@ -701,7 +742,10 @@ static DEVICE_API(dma, siwx91x_dma_api) = {
 					      (siwx91x_dma_chan_desc##inst)),                      \
 		.irq_configure = siwx91x_dma_irq_configure_##inst,                                 \
 	};                                                                                         \
-	DEVICE_DT_INST_DEFINE(inst, siwx91x_dma_init, NULL, &dma_data_##inst, &dma_cfg_##inst,     \
-			      POST_KERNEL, CONFIG_DMA_INIT_PRIORITY, &siwx91x_dma_api);
+	PM_DEVICE_DT_INST_DEFINE(inst, dma_siwx91x_pm_action);                                     \
+	DEVICE_DT_INST_DEFINE(inst, siwx91x_dma_init,  PM_DEVICE_DT_INST_GET(inst),                \
+			      &dma_data_##inst, &dma_cfg_##inst, POST_KERNEL,                      \
+			      CONFIG_DMA_INIT_PRIORITY,                                           \
+			      &siwx91x_dma_api);
 
 DT_INST_FOREACH_STATUS_OKAY(SIWX91X_DMA_INIT)