drivers: counter: stm32: introduce counter set value api

Introduce a counter set value api to set the ticks for 32b and 64b.

Signed-off-by: Ryan McClelland <ryanmcclelland@meta.com>

Author: XenuIsWatching

drivers/counter/counter_ll_stm32_timer.c

@@ -550,6 +550,16 @@ static int counter_stm32_reset_timer(const struct device *dev)
 	return 0;
 }
 
+static int counter_stm32_set_value(const struct device *dev, uint32_t ticks)
+{
+	const struct counter_stm32_config *config = dev->config;
+	TIM_TypeDef *timer = config->timer;
+
+	LL_TIM_SetCounter(timer, ticks);
+
+	return 0;
+}
+
 static void counter_stm32_top_irq_handle(const struct device *dev)
 {
 	struct counter_stm32_data *data = dev->data;
@@ -596,6 +606,7 @@ static DEVICE_API(counter, counter_stm32_driver_api) = {
 	.set_guard_period = counter_stm32_set_guard_period,
 	.get_freq = counter_stm32_get_freq,
 	.reset = counter_stm32_reset_timer,
+	.set_value = counter_stm32_set_value,
 };
 
 #define TIM_IRQ_HANDLE_CC(timx, cc)						\

include/zephyr/drivers/counter.h

@@ -216,6 +216,10 @@ typedef int (*counter_api_get_value)(const struct device *dev,
 typedef int (*counter_api_get_value_64)(const struct device *dev,
 			uint64_t *ticks);
 typedef int (*counter_api_reset)(const struct device *dev);
+typedef int (*counter_api_set_value)(const struct device *dev,
+				     uint32_t ticks);
+typedef int (*counter_api_set_value_64)(const struct device *dev,
+					uint64_t ticks);
 typedef int (*counter_api_set_alarm)(const struct device *dev,
 				     uint8_t chan_id,
 				     const struct counter_alarm_cfg *alarm_cfg);
@@ -238,6 +242,8 @@ __subsystem struct counter_driver_api {
 	counter_api_get_value get_value;
 	counter_api_get_value_64 get_value_64;
 	counter_api_reset reset;
+	counter_api_set_value set_value;
+	counter_api_set_value_64 set_value_64;
 	counter_api_set_alarm set_alarm;
 	counter_api_cancel_alarm cancel_alarm;
 	counter_api_set_top_value set_top_value;
@@ -454,6 +460,53 @@ static inline int z_impl_counter_reset(const struct device *dev)
 	return api->reset(dev);
 }
 
+/**
+ * @brief Set current counter value.
+ * @param dev Pointer to the device structure for the driver instance.
+ * @param ticks Tick value to set
+ *
+ * @retval 0 If successful.
+ * @retval Negative error code on failure getting the counter value
+ */
+ __syscall int counter_set_value(const struct device *dev, uint32_t ticks);
+
+ static inline int z_impl_counter_set_value(const struct device *dev,
+					    uint32_t ticks)
+ {
+	 const struct counter_driver_api *api =
+				(struct counter_driver_api *)dev->api;
+ 
+	if (!api->set_value) {
+		return -ENOSYS;
+	}
+
+	return api->set_value(dev, ticks);
+ }
+ 
+ /**
+  * @brief Set current counter 64-bit value.
+  * @param dev Pointer to the device structure for the driver instance.
+  * @param ticks Tick value to set
+  *
+  * @retval 0 If successful.
+  * @retval Negative error code on failure getting the counter value
+  */
+ __syscall int counter_set_value_64(const struct device *dev, uint64_t ticks);
+ 
+ static inline int z_impl_counter_set_value_64(const struct device *dev,
+					       uint64_t ticks)
+{
+ 	const struct counter_driver_api *api =
+	 	(struct counter_driver_api *)dev->api;
+ 
+ 	if (!api->set_value_64) {
+	 	return -ENOSYS;
+ 	}
+ 
+ 	return api->set_value_64(dev, ticks);
+}
+ 
+
 /**
  * @brief Set a single shot alarm on a channel.
  *