Kieran/ffmpeg-the-third
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

codec/encoder: set the medium if it's unknown

Browse Source
This commit is contained in:
meh 2015-10-01 23:34:59 +02:00
parent 4cc4ba3579
commit e7cd9fd94f
1 changed files with 51 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
src/codec/encoder/encoder.rs
View File

@ -8,30 +8,63 @@ use super::{video, audio, subtitle};
pub struct Encoder(pub Context);
impl Encoder {
pub fn video(self) -> Result<video::Video, Error> {
if self.medium() == media::Type::Video {
Ok(video::Video(self))
}
else {
Err(Error::InvalidData)
pub fn video(mut self) -> Result<video::Video, Error> {
match self.medium() {
media::Type::Unknown => {
unsafe {
(*self.as_mut_ptr()).codec_type = media::Type::Video.into();
}
Ok(video::Video(self))
}
media::Type::Video => {
Ok(video::Video(self))
}
_ => {
Err(Error::InvalidData)
}
}
}
pub fn audio(self) -> Result<audio::Audio, Error> {
if self.medium() == media::Type::Audio {
Ok(audio::Audio(self))
}
else {
Err(Error::InvalidData)
pub fn audio(mut self) -> Result<audio::Audio, Error> {
match self.medium() {
media::Type::Unknown => {
unsafe {
(*self.as_mut_ptr()).codec_type = media::Type::Audio.into();
}
Ok(audio::Audio(self))
}
media::Type::Audio => {
Ok(audio::Audio(self))
}
_ => {
Err(Error::InvalidData)
}
}
}
pub fn subtitle(self) -> Result<subtitle::Subtitle, Error> {
if self.medium() == media::Type::Subtitle {
Ok(subtitle::Subtitle(self))
}
else {
Err(Error::InvalidData)
pub fn subtitle(mut self) -> Result<subtitle::Subtitle, Error> {
match self.medium() {
media::Type::Unknown => {
unsafe {
(*self.as_mut_ptr()).codec_type = media::Type::Subtitle.into();
}
Ok(subtitle::Subtitle(self))
}
media::Type::Subtitle => {
Ok(subtitle::Subtitle(self))
}
_ => {
Err(Error::InvalidData)
}
}
}