drivers: spi: cc23x0: Add power management

Add PM support to cc23x0 SPI module.

Signed-off-by: Julien Panis <jpanis@baylibre.com>

Author: jpanisbl

drivers/spi/spi_cc23x0.c

@@ -14,6 +14,9 @@ LOG_MODULE_REGISTER(spi_cc23x0, CONFIG_SPI_LOG_LEVEL);
 #include <zephyr/drivers/pinctrl.h>
 #include <zephyr/drivers/spi.h>
 #include <zephyr/irq.h>
+#include <zephyr/pm/device.h>
+#include <zephyr/pm/device_runtime.h>
+#include <zephyr/pm/policy.h>
 #include <zephyr/sys/util.h>
 
 #include <driverlib/clkctl.h>
@@ -63,13 +66,31 @@ struct spi_cc23x0_data {
 #endif
 };
 
+static inline void spi_cc23x0_pm_policy_state_lock_get(void)
+{
+#ifdef CONFIG_PM_DEVICE
+	pm_policy_state_lock_get(PM_STATE_RUNTIME_IDLE, PM_ALL_SUBSTATES);
+	pm_policy_state_lock_get(PM_STATE_STANDBY, PM_ALL_SUBSTATES);
+#endif
+}
+
+static inline void spi_cc23x0_pm_policy_state_lock_put(void)
+{
+#ifdef CONFIG_PM_DEVICE
+	pm_policy_state_lock_put(PM_STATE_STANDBY, PM_ALL_SUBSTATES);
+	pm_policy_state_lock_put(PM_STATE_RUNTIME_IDLE, PM_ALL_SUBSTATES);
+#endif
+}
+
 static void spi_cc23x0_isr(const struct device *dev)
 {
 	const struct spi_cc23x0_config *cfg = dev->config;
 	struct spi_cc23x0_data *data = dev->data;
 	struct spi_context *ctx = &data->ctx;
 	uint32_t status;
-#ifndef CONFIG_SPI_CC23X0_DMA_DRIVEN
+#ifdef CONFIG_SPI_CC23X0_DMA_DRIVEN
+	int ret;
+#else
 	uint32_t txd = 0, rxd;
 #endif
 
@@ -79,6 +100,13 @@ static void spi_cc23x0_isr(const struct device *dev)
 #ifdef CONFIG_SPI_CC23X0_DMA_DRIVEN
 	if (status & SPI_DMA_DONE_RX) {
 		SPIClearInt(cfg->base, SPI_DMA_DONE_RX);
+
+		ret = pm_device_runtime_put(cfg->dma_dev);
+		if (ret) {
+			LOG_ERR("Failed to suspend DMA (%d)", ret);
+			return;
+		}
+
 		spi_context_complete(ctx, dev, 0);
 	}
 #else
@@ -307,6 +335,8 @@ static int spi_cc23x0_transceive(const struct device *dev,
 	};
 #endif
 
+	spi_cc23x0_pm_policy_state_lock_get();
+
 	spi_context_lock(ctx, false, NULL, NULL, config);
 
 	ret = spi_cc23x0_configure(dev, config);
@@ -339,6 +369,12 @@ static int spi_cc23x0_transceive(const struct device *dev,
 	block_cfg_rx.dest_addr_adj = DMA_ADDR_ADJ_INCREMENT;
 	block_cfg_rx.block_size = SPI_CC23_DFS * data->tx_len_left;
 
+	ret = pm_device_runtime_get(cfg->dma_dev);
+	if (ret) {
+		LOG_ERR("Failed to resume DMA");
+		goto int_disable;
+	}
+
 	ret = dma_config(cfg->dma_dev, cfg->dma_channel_tx, &dma_cfg_tx);
 	if (ret) {
 		LOG_ERR("Failed to configure DMA TX channel");
@@ -388,6 +424,7 @@ static int spi_cc23x0_transceive(const struct device *dev,
 
 ctx_release:
 	spi_context_release(ctx, ret);
+	spi_cc23x0_pm_policy_state_lock_put();
 	return ret;
 }
 
@@ -434,6 +471,12 @@ static int spi_cc23x0_init(const struct device *dev)
 		LOG_ERR("DMA not ready");
 		return -ENODEV;
 	}
+
+	ret = pm_device_runtime_enable(cfg->dma_dev);
+	if (ret) {
+		LOG_ERR("Failed to enable DMA runtime PM");
+		return ret;
+	}
 #endif
 
 	ret = spi_context_cs_configure_all(&data->ctx);
@@ -446,6 +489,29 @@ static int spi_cc23x0_init(const struct device *dev)
 	return 0;
 }
 
+#ifdef CONFIG_PM_DEVICE
+
+static int spi_cc23x0_pm_action(const struct device *dev, enum pm_device_action action)
+{
+	const struct spi_cc23x0_config *cfg = dev->config;
+	struct spi_cc23x0_data *data = dev->data;
+
+	switch (action) {
+	case PM_DEVICE_ACTION_SUSPEND:
+		SPIDisable(cfg->base);
+		CLKCTLDisable(CLKCTL_BASE, CLKCTL_SPI0);
+		return 0;
+	case PM_DEVICE_ACTION_RESUME:
+		/* Force SPI to be reconfigured at next transfer */
+		data->ctx.config = NULL;
+		return 0;
+	default:
+		return -ENOTSUP;
+	}
+}
+
+#endif /* CONFIG_PM_DEVICE */
+
 #ifdef CONFIG_SPI_CC23X0_DMA_DRIVEN
 #define SPI_CC23X0_DMA_INIT(n)						\
 	.dma_dev = DEVICE_DT_GET(TI_CC23X0_DT_INST_DMA_CTLR(n, tx)),	\
@@ -459,6 +525,7 @@ static int spi_cc23x0_init(const struct device *dev)
 
 #define SPI_CC23X0_INIT(n)						\
 	PINCTRL_DT_INST_DEFINE(n);					\
+	PM_DEVICE_DT_INST_DEFINE(n, spi_cc23x0_pm_action);		\
 									\
 	static void spi_irq_config_func_##n(void)			\
 	{								\
@@ -484,7 +551,7 @@ static int spi_cc23x0_init(const struct device *dev)
 									\
 	DEVICE_DT_INST_DEFINE(n,					\
 			      spi_cc23x0_init,				\
-			      NULL,					\
+			      PM_DEVICE_DT_INST_GET(n),			\
 			      &spi_cc23x0_data_##n,			\
 			      &spi_cc23x0_config_##n,			\
 			      POST_KERNEL,				\