1#![allow(non_snake_case)]
2
3extern crate libc;
4
5#[repr(C)]
6struct vorbis_encoder_helper {
7 private_data: *mut libc::c_void,
8}
9
10impl vorbis_encoder_helper {
11 fn new() -> Self {
12 unsafe {
13 std::mem::zeroed()
14 }
15 }
16}
17
18pub struct Encoder {
19 e: vorbis_encoder_helper,
20}
21
22impl Encoder {
23 pub fn new(channels: u32, rate: u64, quality: f32) -> Result<Self, libc::c_int> {
24 let mut enc = Encoder {
25 e: vorbis_encoder_helper::new(),
26 };
27 let res = unsafe { vorbis_encoder_helper_init(
28 &mut enc.e as *mut vorbis_encoder_helper,
29 channels as libc::c_uint, rate as libc::c_ulong, quality as libc::c_float)};
30 match res {
31 0 => {
32 Ok(enc)
33 },
34 _ => {
35 Err(res)
36 }
37 }
38 }
39
40 pub fn encode(&mut self, samples: &Vec<i16>) -> Result<Vec<u8>, libc::c_int> {
41 let res = unsafe { vorbis_encoder_helper_encode(
42 &mut self.e as *mut vorbis_encoder_helper,
43 samples.as_ptr() as *const libc::int16_t,
44 samples.len() as libc::c_int)};
45 if res != 0 {
46 return Err(res);
47 }
48 let s = unsafe { vorbis_encoder_helper_get_data_length(
49 &mut self.e as *mut vorbis_encoder_helper)};
50 let mut v = vec![0u8; s as usize];
51 unsafe { vorbis_encoder_helper_get_data(
52 &mut self.e as *mut vorbis_encoder_helper,
53 v[..].as_mut_ptr() as *mut libc::c_uchar);}
54 Ok(v)
55 }
56
57 pub fn flush(&mut self) -> Result<Vec<u8>, libc::c_int> {
58 let res = unsafe { vorbis_encoder_helper_flush(
59 &mut self.e as *mut vorbis_encoder_helper)};
60 if res != 0 {
61 return Err(res);
62 }
63 let s = unsafe { vorbis_encoder_helper_get_data_length(
64 &mut self.e as *mut vorbis_encoder_helper)};
65 let mut v = vec![0u8; s as usize];
66 unsafe { vorbis_encoder_helper_get_data(
67 &mut self.e as *mut vorbis_encoder_helper,
68 v[..].as_mut_ptr() as *mut libc::c_uchar);}
69 Ok(v)
70 }
71}
72
73impl Drop for Encoder {
74 fn drop(&mut self) {
75 unsafe {
76 vorbis_encoder_helper_free(&mut self.e as *mut vorbis_encoder_helper);
77 }
78 }
79}
80
81extern "C" {
82 fn vorbis_encoder_helper_init(
83 hp: *mut vorbis_encoder_helper, ch: libc::c_uint, rt: libc::c_ulong,
84 q: libc::c_float) -> libc::c_int;
85 fn vorbis_encoder_helper_encode(hp: *mut vorbis_encoder_helper, data: *const libc::int16_t,
86 bits: libc::c_int) -> libc::c_int;
87 fn vorbis_encoder_helper_flush(hp: *mut vorbis_encoder_helper) -> libc::c_int;
88 fn vorbis_encoder_helper_get_data_length(hp: *const vorbis_encoder_helper) -> libc::c_uint;
89 fn vorbis_encoder_helper_get_data(hp: *mut vorbis_encoder_helper, data: *mut libc::c_uchar);
90 fn vorbis_encoder_helper_free(hp: *mut vorbis_encoder_helper);
91}