ffmpeg_next/codec/encoder/
encoder.rs

1use std::ops::{Deref, DerefMut};
2use std::ptr;
3
4use ffi::*;
5use libc::c_int;
6
7use super::{audio, subtitle, video};
8use codec::Context;
9use {media, packet, Error, Frame};
10
11pub struct Encoder(pub Context);
12
13impl Encoder {
14    pub fn video(mut self) -> Result<video::Video, Error> {
15        match self.medium() {
16            media::Type::Unknown => {
17                unsafe {
18                    (*self.as_mut_ptr()).codec_type = media::Type::Video.into();
19                }
20
21                Ok(video::Video(self))
22            }
23
24            media::Type::Video => Ok(video::Video(self)),
25
26            _ => Err(Error::InvalidData),
27        }
28    }
29
30    pub fn audio(mut self) -> Result<audio::Audio, Error> {
31        match self.medium() {
32            media::Type::Unknown => {
33                unsafe {
34                    (*self.as_mut_ptr()).codec_type = media::Type::Audio.into();
35                }
36
37                Ok(audio::Audio(self))
38            }
39
40            media::Type::Audio => Ok(audio::Audio(self)),
41
42            _ => Err(Error::InvalidData),
43        }
44    }
45
46    pub fn subtitle(mut self) -> Result<subtitle::Subtitle, Error> {
47        match self.medium() {
48            media::Type::Unknown => {
49                unsafe {
50                    (*self.as_mut_ptr()).codec_type = media::Type::Subtitle.into();
51                }
52
53                Ok(subtitle::Subtitle(self))
54            }
55
56            media::Type::Subtitle => Ok(subtitle::Subtitle(self)),
57
58            _ => Err(Error::InvalidData),
59        }
60    }
61
62    pub fn send_frame(&mut self, frame: &Frame) -> Result<(), Error> {
63        unsafe {
64            match avcodec_send_frame(self.as_mut_ptr(), frame.as_ptr()) {
65                e if e < 0 => Err(Error::from(e)),
66                _ => Ok(()),
67            }
68        }
69    }
70
71    /// Sends a NULL packet to the encoder to signal end of stream and enter
72    /// draining mode.
73    pub fn send_eof(&mut self) -> Result<(), Error> {
74        unsafe { self.send_frame(&Frame::wrap(ptr::null_mut())) }
75    }
76
77    pub fn receive_packet<P: packet::Mut>(&mut self, packet: &mut P) -> Result<(), Error> {
78        unsafe {
79            match avcodec_receive_packet(self.as_mut_ptr(), packet.as_mut_ptr()) {
80                e if e < 0 => Err(Error::from(e)),
81                _ => Ok(()),
82            }
83        }
84    }
85
86    pub fn set_bit_rate(&mut self, value: usize) {
87        unsafe {
88            (*self.as_mut_ptr()).bit_rate = value as i64;
89        }
90    }
91
92    pub fn set_max_bit_rate(&mut self, value: usize) {
93        unsafe {
94            (*self.as_mut_ptr()).rc_max_rate = value as i64;
95        }
96    }
97
98    pub fn set_tolerance(&mut self, value: usize) {
99        unsafe {
100            (*self.as_mut_ptr()).bit_rate_tolerance = value as c_int;
101        }
102    }
103
104    pub fn set_quality(&mut self, value: usize) {
105        unsafe {
106            (*self.as_mut_ptr()).global_quality = value as c_int;
107        }
108    }
109
110    pub fn set_compression(&mut self, value: Option<usize>) {
111        unsafe {
112            if let Some(value) = value {
113                (*self.as_mut_ptr()).compression_level = value as c_int;
114            } else {
115                (*self.as_mut_ptr()).compression_level = -1;
116            }
117        }
118    }
119}
120
121impl Deref for Encoder {
122    type Target = Context;
123
124    fn deref(&self) -> &<Self as Deref>::Target {
125        &self.0
126    }
127}
128
129impl DerefMut for Encoder {
130    fn deref_mut(&mut self) -> &mut <Self as Deref>::Target {
131        &mut self.0
132    }
133}
134
135impl AsRef<Context> for Encoder {
136    fn as_ref(&self) -> &Context {
137        self
138    }
139}
140
141impl AsMut<Context> for Encoder {
142    fn as_mut(&mut self) -> &mut Context {
143        &mut *self
144    }
145}
ffmpeg_next/codec/encoder/encoder.rs

ffmpeg_next/codec/encoder/
encoder.rs
