Création du module pmoaudio

pmoaudio/src/nodes/source_node.rs

@@ -0,0 +1,169 @@
+use crate::{AudioChunk, nodes::{AudioError, MultiSubscriberNode}};
+use std::sync::Arc;
+use tokio::sync::mpsc;
+
+/// SourceNode - Génère ou lit des chunks audio depuis une source
+///
+/// Ce node est la source du pipeline. Version mock pour tests.
+pub struct SourceNode {
+    subscribers: MultiSubscriberNode,
+}
+
+impl SourceNode {
+    pub fn new() -> Self {
+        Self {
+            subscribers: MultiSubscriberNode::new(),
+        }
+    }
+
+    pub fn add_subscriber(&mut self, tx: mpsc::Sender<Arc<AudioChunk>>) {
+        self.subscribers.add_subscriber(tx);
+    }
+
+    /// Génère un chunk de test avec une forme d'onde sinusoïdale
+    pub fn generate_test_chunk(
+        order: u64,
+        size: usize,
+        sample_rate: u32,
+        frequency: f32,
+    ) -> AudioChunk {
+        let mut left = Vec::with_capacity(size);
+        let mut right = Vec::with_capacity(size);
+
+        for i in 0..size {
+            let t = (order * size as u64 + i as u64) as f32 / sample_rate as f32;
+            let sample = (2.0 * std::f32::consts::PI * frequency * t).sin();
+            left.push(sample);
+            right.push(sample * 0.8); // Légèrement différent pour la stéréo
+        }
+
+        AudioChunk::new(order, left, right, sample_rate)
+    }
+
+    /// Génère et envoie des chunks de test
+    pub async fn generate_chunks(
+        &self,
+        count: u64,
+        chunk_size: usize,
+        sample_rate: u32,
+        frequency: f32,
+    ) -> Result<(), AudioError> {
+        for i in 0..count {
+            let chunk = Self::generate_test_chunk(i, chunk_size, sample_rate, frequency);
+            self.subscribers.push(Arc::new(chunk)).await?;
+        }
+        Ok(())
+    }
+
+    /// Génère des chunks silencieux
+    pub async fn generate_silence(
+        &self,
+        count: u64,
+        chunk_size: usize,
+        sample_rate: u32,
+    ) -> Result<(), AudioError> {
+        for i in 0..count {
+            let chunk = AudioChunk::new(
+                i,
+                vec![0.0; chunk_size],
+                vec![0.0; chunk_size],
+                sample_rate,
+            );
+            self.subscribers.push(Arc::new(chunk)).await?;
+        }
+        Ok(())
+    }
+
+    /// Version streaming : génère des chunks continuellement avec délai
+    pub async fn stream_chunks(
+        &self,
+        chunk_size: usize,
+        sample_rate: u32,
+        frequency: f32,
+        duration_ms: u64,
+    ) -> Result<(), AudioError> {
+        let chunk_duration_ms = (chunk_size as f64 / sample_rate as f64 * 1000.0) as u64;
+        let mut order = 0u64;
+
+        let start = tokio::time::Instant::now();
+        let duration = tokio::time::Duration::from_millis(duration_ms);
+
+        while start.elapsed() < duration {
+            let chunk = Self::generate_test_chunk(order, chunk_size, sample_rate, frequency);
+            self.subscribers.push(Arc::new(chunk)).await?;
+
+            order += 1;
+
+            // Attendre pour simuler le timing réel
+            tokio::time::sleep(tokio::time::Duration::from_millis(chunk_duration_ms)).await;
+        }
+
+        Ok(())
+    }
+}
+
+impl Default for SourceNode {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[tokio::test]
+    async fn test_source_node_generation() {
+        let mut source = SourceNode::new();
+        let (tx, mut rx) = mpsc::channel(10);
+
+        source.add_subscriber(tx);
+
+        // Générer 3 chunks
+        source.generate_chunks(3, 100, 48000, 440.0).await.unwrap();
+
+        // Vérifier la réception
+        for i in 0..3 {
+            let chunk = rx.recv().await.unwrap();
+            assert_eq!(chunk.order, i);
+            assert_eq!(chunk.len(), 100);
+            assert_eq!(chunk.sample_rate, 48000);
+        }
+    }
+
+    #[test]
+    fn test_sine_wave_generation() {
+        let chunk = SourceNode::generate_test_chunk(0, 48000, 48000, 440.0);
+
+        // Vérifier qu'on a bien une sinusoïde
+        // À 440 Hz avec 48000 samples/s, on devrait avoir 440 cycles
+        let left = &*chunk.left;
+
+        // Trouver les passages par zéro
+        let mut zero_crossings = 0;
+        for i in 1..left.len() {
+            if (left[i - 1] < 0.0 && left[i] >= 0.0) || (left[i - 1] >= 0.0 && left[i] < 0.0) {
+                zero_crossings += 1;
+            }
+        }
+
+        // 440 cycles = 880 passages par zéro (approximativement)
+        assert!(zero_crossings > 850 && zero_crossings < 910);
+    }
+
+    #[tokio::test]
+    async fn test_source_node_silence() {
+        let mut source = SourceNode::new();
+        let (tx, mut rx) = mpsc::channel(10);
+
+        source.add_subscriber(tx);
+
+        source.generate_silence(2, 100, 48000).await.unwrap();
+
+        for _ in 0..2 {
+            let chunk = rx.recv().await.unwrap();
+            assert!(chunk.left.iter().all(|&x| x == 0.0));
+            assert!(chunk.right.iter().all(|&x| x == 0.0));
+        }
+    }
+}