mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 02:58:50 +00:00
0c49e95dfb
* Encodec model. * Fixes. * Add the padding functions. * Get the LSTM bit to work. * Get the encodec model to generate some tokens (decoder only for now). * Minor tweak. * Minor tweak.
57 lines
1.5 KiB
Rust
57 lines
1.5 KiB
Rust
use std::io::prelude::*;
|
|
|
|
pub trait Sample {
|
|
fn to_i16(&self) -> i16;
|
|
}
|
|
|
|
impl Sample for f32 {
|
|
fn to_i16(&self) -> i16 {
|
|
(self.clamp(-1.0, 1.0) * 32767.0) as i16
|
|
}
|
|
}
|
|
|
|
impl Sample for f64 {
|
|
fn to_i16(&self) -> i16 {
|
|
(self.clamp(-1.0, 1.0) * 32767.0) as i16
|
|
}
|
|
}
|
|
|
|
impl Sample for i16 {
|
|
fn to_i16(&self) -> i16 {
|
|
*self
|
|
}
|
|
}
|
|
|
|
pub fn write_pcm_as_wav<W: Write, S: Sample>(
|
|
w: &mut W,
|
|
samples: &[S],
|
|
sample_rate: u32,
|
|
) -> std::io::Result<()> {
|
|
let len = 12u32; // header
|
|
let len = len + 24u32; // fmt
|
|
let len = len + samples.len() as u32 * 2 + 8; // data
|
|
let n_channels = 1u16;
|
|
let bytes_per_second = sample_rate * 2 * n_channels as u32;
|
|
w.write_all(b"RIFF")?;
|
|
w.write_all(&(len - 8).to_le_bytes())?; // total length minus 8 bytes
|
|
w.write_all(b"WAVE")?;
|
|
|
|
// Format block
|
|
w.write_all(b"fmt ")?;
|
|
w.write_all(&16u32.to_le_bytes())?; // block len minus 8 bytes
|
|
w.write_all(&1u16.to_le_bytes())?; // PCM
|
|
w.write_all(&n_channels.to_le_bytes())?; // one channel
|
|
w.write_all(&sample_rate.to_le_bytes())?;
|
|
w.write_all(&bytes_per_second.to_le_bytes())?;
|
|
w.write_all(&2u16.to_le_bytes())?; // 2 bytes of data per sample
|
|
w.write_all(&16u16.to_le_bytes())?; // bits per sample
|
|
|
|
// Data block
|
|
w.write_all(b"data")?;
|
|
w.write_all(&(samples.len() as u32 * 2).to_le_bytes())?;
|
|
for sample in samples.iter() {
|
|
w.write_all(&sample.to_i16().to_le_bytes())?
|
|
}
|
|
Ok(())
|
|
}
|