denoize 0.15.0

Pure-Rust audio denoiser with classical DSP and optional RNNoise
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use crate::Audio;
use ogg::writing::{PacketWriteEndInfo, PacketWriter};
use opus::{Application, Bitrate, Channels, Encoder};
use std::borrow::Cow;
use std::path::Path;

use super::pcm::{downmix_to_stereo, lossy_channel_layout};
use super::DownmixMode;

pub fn write_ogg_opus(
    path: &Path,
    audio: &Audio,
    bitrate: u32,
    downmix: DownmixMode,
) -> Result<(), String> {
    if audio.frames() == 0 {
        return Err("Opus output requires at least one frame".into());
    }
    let source_channels = if audio.channels() > 2 {
        // Validate the explicit policy and known layout before rendering.
        lossy_channel_layout(audio, downmix)?;
        downmix_to_stereo(audio)?
    } else {
        if audio.channels() == 0 {
            return Err("Opus requires at least one channel".into());
        }
        audio.channels.clone()
    };
    let converted =
        crate::resample::resample_channels(&source_channels, audio.sample_rate, 48_000)?;
    let count = converted.len();
    let channels = if count == 1 {
        Channels::Mono
    } else {
        Channels::Stereo
    };
    let mut encoder = Encoder::new(48_000, channels, Application::Audio)
        .map_err(|e| format!("Opus encoder: {e}"))?;
    encoder
        .set_bitrate(Bitrate::Bits(bitrate as i32))
        .map_err(|e| format!("Opus bitrate: {e}"))?;
    let pre_skip = encoder
        .get_lookahead()
        .map_err(|e| format!("Opus lookahead: {e}"))? as u16;
    let file = std::fs::File::create(path).map_err(|e| format!("Opus create: {e}"))?;
    let mut writer = PacketWriter::new(std::io::BufWriter::new(file));
    let serial = 0x444e_5a45;
    let mut head = b"OpusHead".to_vec();
    head.extend([1, count as u8]);
    head.extend(pre_skip.to_le_bytes());
    head.extend(audio.sample_rate.to_le_bytes());
    head.extend(0_i16.to_le_bytes());
    head.push(0);
    writer
        .write_packet(Cow::Owned(head), serial, PacketWriteEndInfo::EndPage, 0)
        .map_err(|e| format!("Ogg header: {e}"))?;
    let vendor = b"denoize";
    let mut tags = b"OpusTags".to_vec();
    tags.extend((vendor.len() as u32).to_le_bytes());
    tags.extend(vendor);
    tags.extend(0_u32.to_le_bytes());
    writer
        .write_packet(Cow::Owned(tags), serial, PacketWriteEndInfo::EndPage, 0)
        .map_err(|e| format!("Ogg tags: {e}"))?;
    let frame_size = 960;
    let total = converted[0].len();
    let mut position = 0;
    let mut granule = pre_skip as u64;
    while position < total {
        let actual = (total - position).min(frame_size);
        let mut pcm = vec![0.0f32; frame_size * count];
        for frame in 0..actual {
            for channel in 0..count {
                pcm[frame * count + channel] = converted[channel][position + frame] as f32;
            }
        }
        let packet = encoder
            .encode_vec_float(&pcm, 4_000)
            .map_err(|e| format!("Opus encode: {e}"))?;
        granule += actual as u64;
        position += actual;
        let end = if position == total {
            PacketWriteEndInfo::EndStream
        } else {
            PacketWriteEndInfo::EndPage
        };
        writer
            .write_packet(Cow::Owned(packet), serial, end, granule)
            .map_err(|e| format!("Ogg write: {e}"))?;
    }
    Ok(())
}