Potentially resample when writing opus files.

.gitignore

@@ -16,3 +16,5 @@ Cargo.lock
 __pycache__
 
 bria.mp3
+bria.wav
+bria.opus

src/lib.rs

@@ -191,21 +191,25 @@ fn write_wav(
 }
 
 #[pyfunction]
-#[pyo3(signature = (filename, data, sample_rate))]
+#[pyo3(signature = (filename, data, sample_rate, resample_to=None))]
 fn write_opus(
     filename: std::path::PathBuf,
     data: numpy::PyReadonlyArray1<f32>,
     sample_rate: u32,
+    resample_to: Option<u32>,
 ) -> PyResult<()> {
     let w = std::fs::File::create(&filename).w_f(filename.as_path())?;
     let mut w = std::io::BufWriter::new(w);
     let data = data.as_array();
     match data.as_slice() {
         None => {
             let data = data.to_vec();
-            opus::write_ogg(&mut w, data.as_ref(), sample_rate).w_f(filename.as_path())?
+            opus::write_ogg(&mut w, data.as_ref(), sample_rate, resample_to)
+                .w_f(filename.as_path())?
+        }
+        Some(data) => {
+            opus::write_ogg(&mut w, data, sample_rate, resample_to).w_f(filename.as_path())?
         }
-        Some(data) => opus::write_ogg(&mut w, data, sample_rate).w_f(filename.as_path())?,
     }
     Ok(())
 }

src/opus.rs

@@ -120,7 +120,7 @@ fn write_opus_tags<W: std::io::Write>(w: &mut W) -> std::io::Result<()> {
     Ok(())
 }
 
-pub fn write_ogg<W: std::io::Write>(w: &mut W, pcm: &[f32], sample_rate: u32) -> Result<()> {
+fn write_ogg_<W: std::io::Write>(w: &mut W, pcm: &[f32], sample_rate: u32) -> Result<()> {
     let mut pw = ogg::PacketWriter::new(w);
 
     // Write the opus headers and tags
@@ -152,3 +152,18 @@ pub fn write_ogg<W: std::io::Write>(w: &mut W, pcm: &[f32], sample_rate: u32) ->
 
     Ok(())
 }
+
+pub fn write_ogg<W: std::io::Write>(
+    w: &mut W,
+    pcm: &[f32],
+    sample_rate: u32,
+    resample_to: Option<u32>,
+) -> Result<()> {
+    match resample_to {
+        None => write_ogg_(w, pcm, sample_rate),
+        Some(resample_to) => {
+            let pcm = crate::audio::resample(pcm, sample_rate as usize, resample_to as usize)?;
+            write_ogg_(w, &pcm, resample_to)
+        }
+    }
+}

test/basic.py

@@ -14,3 +14,4 @@
 print(data.shape, sr)
 
 sphn.write_wav("bria.wav", data[0], sr)
+sphn.write_opus("bria.opus", data[0], sr, resample_to=48000)