add ability to configure loop count from 1 to infinite

2 files changed, 14 insertions, 11 deletions

diff --git a/examples/nrf/src/bin/pwm_sequence.rs b/examples/nrf/src/bin/pwm_sequence.rs
index bc356c28b..8b7aeddcd 100644
--- a/examples/nrf/src/bin/pwm_sequence.rs
+++ b/examples/nrf/src/bin/pwm_sequence.rs
@@ -7,7 +7,7 @@ mod example_common;
 use defmt::*;
 use embassy::executor::Spawner;
 use embassy::time::{Duration, Timer};
-use embassy_nrf::pwm::{CounterMode, LoopingConfig, Prescaler, Pwm, SequenceLoad};
+use embassy_nrf::pwm::{CounterMode, LoopMode, LoopingConfig, Prescaler, Pwm, SequenceLoad};
 use embassy_nrf::Peripherals;
 
 #[embassy::main]
@@ -15,26 +15,23 @@ async fn main(_spawner: Spawner, p: Peripherals) {
     let seq_values: [u16; 16] = [
         0x8000, 0, 0, 0, 0, 0x8000, 0, 0, 0, 0, 0x8000, 0, 0, 0, 0, 0x8000,
     ];
+
     let config = LoopingConfig {
         counter_mode: CounterMode::Up,
         top: 15625,
         prescaler: Prescaler::Div128,
         sequence: &seq_values,
         sequence_load: SequenceLoad::Individual,
-        repeats: 0,
+        refresh: 0,
         enddelay: 0,
+        additional_loops: LoopMode::Additional(5),
     };
 
-    let pwm = unwrap!(Pwm::simple_playback(
+    let _pwm = unwrap!(Pwm::simple_playback(
         p.PWM0, p.P0_13, p.P0_15, p.P0_16, p.P0_14, config
     ));
     info!("pwm started!");
 
-    Timer::after(Duration::from_millis(10000)).await;
-
-    pwm.stop();
-    info!("pwm stopped!");
-
     loop {
         Timer::after(Duration::from_millis(1000)).await;
     }
diff --git a/examples/nrf/src/bin/pwm_simple_sin.rs b/examples/nrf/src/bin/pwm_simple_sin.rs
index 866202a4c..3fbfc960f 100644
--- a/examples/nrf/src/bin/pwm_simple_sin.rs
+++ b/examples/nrf/src/bin/pwm_simple_sin.rs
@@ -9,7 +9,7 @@ use defmt::*;
 use embassy::executor::Spawner;
 use embassy::time::{Duration, Timer};
 use embassy_nrf::gpio::NoPin;
-use embassy_nrf::pwm::{CounterMode, LoopingConfig, Prescaler, Pwm, SequenceLoad};
+use embassy_nrf::pwm::{CounterMode, LoopMode, LoopingConfig, Prescaler, Pwm, SequenceLoad};
 use embassy_nrf::Peripherals;
 use micromath::F32Ext;
 
@@ -26,15 +26,21 @@ async fn main(_spawner: Spawner, p: Peripherals) {
         prescaler: Prescaler::Div16,
         sequence: &seq_values,
         sequence_load: SequenceLoad::Common,
-        repeats: 0,
+        refresh: 0,
         enddelay: 0,
+        additional_loops: LoopMode::Infinite,
     };
 
-    let _pwm = unwrap!(Pwm::simple_playback(
+    let pwm = unwrap!(Pwm::simple_playback(
         p.PWM0, p.P0_13, NoPin, NoPin, NoPin, config
     ));
     info!("pwm started!");
 
+    Timer::after(Duration::from_millis(20000)).await;
+
+    pwm.stop();
+    info!("pwm stopped!");
+
     loop {
         Timer::after(Duration::from_millis(1000)).await;
     }