codec/decoder/audio: impl Iterator for Audio::decode

examples/transcode-audio.rs

@@ -123,24 +123,25 @@ fn main() {
 	let in_time_base  = transcoder.decoder.time_base();
 	let out_time_base = octx.stream(0).unwrap().time_base();
 
-	let mut decoded = frame::Audio::empty();
 	let mut encoded = ffmpeg::Packet::empty();
 
 	for (stream, mut packet) in ictx.packets() {
 		if stream.index() == transcoder.stream {
 			packet.rescale_ts(stream.time_base(), in_time_base);
 
-			if let Ok(true) = transcoder.decoder.decode(&packet, &mut decoded) {
-				let timestamp = decoded.timestamp();
-				decoded.set_pts(timestamp);
+			for decoded in transcoder.decoder.decode_iter(&mut packet) {
+				if let Ok(Some(mut decoded)) = decoded {
+					let timestamp = decoded.timestamp();
+					decoded.set_pts(timestamp);
 
-				transcoder.filter.get("in").unwrap().source().add(&decoded).unwrap();
+					transcoder.filter.get("in").unwrap().source().add(&decoded).unwrap();
 
-				while let Ok(..) = transcoder.filter.get("out").unwrap().sink().frame(&mut decoded) {
-					if let Ok(true) = transcoder.encoder.encode(&decoded, &mut encoded) {
-						encoded.set_stream(0);
-						encoded.rescale_ts(in_time_base, out_time_base);
-						encoded.write_interleaved(&mut octx).unwrap();
+					while let Ok(..) = transcoder.filter.get("out").unwrap().sink().frame(&mut decoded) {
+						if let Ok(true) = transcoder.encoder.encode(&decoded, &mut encoded) {
+							encoded.set_stream(0);
+							encoded.rescale_ts(in_time_base, out_time_base);
+							encoded.write_interleaved(&mut octx).unwrap();
+						}
 					}
 				}
 			}
@@ -149,6 +150,8 @@ fn main() {
 
 	transcoder.filter.get("in").unwrap().source().flush().unwrap();
 
+	let mut decoded = frame::Audio::empty();
+
 	while let Ok(..) = transcoder.filter.get("out").unwrap().sink().frame(&mut decoded) {
 		if let Ok(true) = transcoder.encoder.encode(&decoded, &mut encoded) {
 			encoded.set_stream(0);

src/codec/decoder/audio.rs

@@ -4,7 +4,8 @@ use libc::c_int;
 use ffi::*;
 
 use super::Opened;
-use ::{packet, Error, AudioService, ChannelLayout};
+use ::{Error, AudioService, ChannelLayout};
+use ::packet::{self, Mut};
 use ::frame;
 use ::util::format;
 use ::codec::Context;
@@ -23,6 +24,13 @@ impl Audio {
 		}
 	}
 
+	pub fn decode_iter<'a, 'b>(&'a mut self, packet: &'b mut packet::Packet) -> AudioFrameIter<'a, 'b> {
+		AudioFrameIter {
+			audio: self,
+			packet: packet,
+		}
+	}
+
 	pub fn rate(&self) -> u32 {
 		unsafe {
 			(*self.as_ptr()).sample_rate as u32
@@ -130,3 +138,38 @@ impl AsMut<Context> for Audio {
 		&mut self.0
 	}
 }
+
+pub struct AudioFrameIter<'a, 'b> {
+	audio: &'a mut Audio,
+	packet: &'b mut packet::Packet,
+}
+
+impl<'a, 'b> Iterator for AudioFrameIter<'a, 'b> {
+    type Item = Result<Option<frame::Audio>, Error>;
+
+    fn next(&mut self) -> Option<Result<Option<frame::Audio>, Error>> {
+		unsafe {
+			if !self.packet.is_empty() {
+				let mut out = frame::Audio::empty();
+				let mut got: c_int = 0;
+				let packet = self.packet.as_mut_ptr();
+
+				match avcodec_decode_audio4(self.audio.as_mut_ptr(), out.as_mut_ptr(), &mut got, packet) {
+					e if e < 0 => Some(Err(Error::from(e))),
+					n => {
+						(*packet).data = (*packet).data.offset(n as isize);
+						(*packet).size -= n;
+
+						if got != 0 {
+							Some(Ok(Some(out)))
+						} else {
+							Some(Ok(None))
+						}
+					}
+				}
+			} else {
+				None
+			}
+		}
+	}
+}