fix stopwatch

Author: dotcypress

examples/stopwatch.rs

@@ -31,22 +31,24 @@ fn main() -> ! {
     #[cfg(feature = "stm32g0x0")]
     let mut stopwatch = dp.TIM3.stopwatch(&mut rcc);
 
+    stopwatch.set_prescaler(1);
+
     let elapsed_us = stopwatch.trace(|| {
-        delay.delay(10.us());
+        delay.delay(100.us());
     });
-    hprintln!("Delay: 10us -> {}us", elapsed_us.0).unwrap();
+    hprintln!("Delay: 100 us -> {} us", elapsed_us.0).unwrap();
 
-    timer.start(10.us());
+    timer.start(100.us());
     let elapsed_us = stopwatch.trace(|| {
         block!(timer.wait()).unwrap();
     });
-    hprintln!("Timer: 10us -> {}us", elapsed_us.0).unwrap();
+    hprintln!("Timer: 100 us -> {} us", elapsed_us.0).unwrap();
 
     let elapsed_us = stopwatch.trace(|| {
         let x = calc_something();
         assert!(x > 0);
     });
-    hprintln!("Calc @ 16MHz: {}us", elapsed_us.0).unwrap();
+    hprintln!("Calc @ 16 MHz: {} us", elapsed_us.0).unwrap();
 
     let rcc = rcc.freeze(Config::new(SysClockSrc::PLL));
     stopwatch.set_clock(rcc.clocks.apb_tim_clk);
@@ -55,7 +57,7 @@ fn main() -> ! {
         let x = calc_something();
         assert!(x > 0);
     });
-    hprintln!("Calc @ 64MHz: {}us", elapsed_us.0).unwrap();
+    hprintln!("Calc @ 64 MHz: {} us", elapsed_us.0).unwrap();
 
     loop {}
 }

src/timer/stopwatch.rs

@@ -71,7 +71,7 @@ macro_rules! stopwatches {
                 pub fn elapsed(&self, ts: Instant) -> MicroSecond {
                     let now = self.now().0;
                     let cycles = now.wrapping_sub(ts.0);
-                    self.clk.duration(cycles)
+                    self.clk.duration(cycles << self.tim.psc.read().bits())
                 }
 
                 pub fn trace<F>(&self, mut closure: F) -> MicroSecond
@@ -81,7 +81,7 @@ macro_rules! stopwatches {
                     let started = self.now().0;
                     closure();
                     let now = self.now().0;
-                    self.clk.duration(now.wrapping_sub(started))
+                    self.clk.duration(now.wrapping_sub(started) << self.tim.psc.read().bits())
                 }
             }