1use std::fmt;
2use std::sync::atomic::{AtomicU64, Ordering};
3
4use super::tick;
5
6pub struct SystemClock {
11 pub sample_rate: f32,
13 position: AtomicU64,
15 bpm: AtomicU64,
17}
18
19impl SystemClock {
20 pub fn new(sample_rate: f32, initial_bpm: f64) -> Self {
22 Self {
23 sample_rate,
24 position: AtomicU64::new(0),
25 bpm: AtomicU64::new(initial_bpm.to_bits()),
26 }
27 }
28
29 pub fn with_sample_rate(sample_rate: f32) -> Self {
31 Self::new(sample_rate, 120.0)
32 }
33
34 pub fn next_tick(&mut self, block_size: usize) -> tick::ClockTick {
36 let samples = block_size as u32;
37 let pos = self.position.fetch_add(samples as u64, Ordering::Relaxed);
38
39 tick::ClockTick {
40 sample_pos: pos,
41 samples_since_last: samples,
42 is_new_block: true,
43 sample_rate: self.sample_rate,
44 tempo: Some(self.bpm() as f32),
45 source: String::new(),
46 speed_ratio: 1.0,
47 is_final: true,
48 }
49 }
50
51 pub fn bpm(&self) -> f64 {
53 f64::from_bits(self.bpm.load(Ordering::Relaxed))
54 }
55
56 pub fn set_bpm(&self, bpm: f64) {
58 self.bpm.store(bpm.to_bits(), Ordering::Relaxed);
59 }
60
61 pub fn position(&self) -> u64 {
63 self.position.load(Ordering::Relaxed)
64 }
65
66 pub fn reset(&self) {
68 self.position.store(0, Ordering::Relaxed);
69 }
70}
71
72impl fmt::Debug for SystemClock {
73 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
74 f.debug_struct("SystemClock")
75 .field("sample_rate", &self.sample_rate)
76 .field("position", &self.position.load(Ordering::Relaxed))
77 .field("bpm", &self.bpm())
78 .finish()
79 }
80}
81
82#[cfg(test)]
83mod tests {
84 use super::*;
85
86 #[test]
87 fn test_system_clock() {
88 let mut clock = SystemClock::new(44100.0, 120.0);
89
90 let tick = clock.next_tick(64);
91 assert_eq!(tick.sample_pos, 0);
92 assert_eq!(tick.samples_since_last, 64);
93 assert_eq!(tick.sample_rate, 44100.0);
94 assert_eq!(tick.tempo, Some(120.0));
95
96 let tick = clock.next_tick(64);
97 assert_eq!(tick.sample_pos, 64);
98 }
99}