split remaining decoders and demuxer from core

[nihav.git] / nihav-commonfmt / src / codecs / pcm.rs

diff --git a/nihav-commonfmt/src/codecs/pcm.rs b/nihav-commonfmt/src/codecs/pcm.rs
new file mode 100644 (file)
index 0000000..7a71051
--- /dev/null
+++ b/nihav-commonfmt/src/codecs/pcm.rs
@@ -0,0 +1,68 @@
+use std::rc::Rc;
+use std::cell::RefCell;
+use nihav_core::formats::*;
+use nihav_core::codecs::*;
+use nihav_core::frame::*;
+
+struct PCMDecoder { chmap: NAChannelMap }
+
+impl PCMDecoder {
+    fn new() -> Self {
+        PCMDecoder { chmap: NAChannelMap::new() }
+    }
+}
+
+const CHMAP_MONO: [NAChannelType; 1] = [NAChannelType::C];
+const CHMAP_STEREO: [NAChannelType; 2] = [NAChannelType::L, NAChannelType::R];
+
+fn get_default_chmap(nch: u8) -> NAChannelMap {
+    let mut chmap = NAChannelMap::new();
+    match nch {
+        1 => chmap.add_channels(&CHMAP_MONO),
+        2 => chmap.add_channels(&CHMAP_STEREO),
+        _ => (),
+    }
+    chmap
+}
+
+fn get_duration(ainfo: &NAAudioInfo, duration: Option<u64>, data_size: usize) -> u64 {
+    if duration == None {
+        let size_bits = (data_size as u64) * 8;
+        let blk_size = (ainfo.get_channels() as u64) * (ainfo.get_format().get_bits() as u64);
+        size_bits / blk_size
+    } else {
+        duration.unwrap() as u64
+    }
+}
+
+impl NADecoder for PCMDecoder {
+    fn init(&mut self, info: Rc<NACodecInfo>) -> DecoderResult<()> {
+        if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() {
+            self.chmap = get_default_chmap(ainfo.get_channels());
+            if self.chmap.num_channels() == 0 { return Err(DecoderError::InvalidData); }
+            Ok(())
+        } else {
+            Err(DecoderError::InvalidData)
+        }
+    }
+    fn decode(&mut self, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
+        let info = pkt.get_stream().get_info();
+        if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() {
+            let duration = get_duration(&ainfo, pkt.get_duration(), pkt.get_buffer().len());
+            let pktbuf = pkt.get_buffer();
+            let mut buf: Vec<u8> = Vec::with_capacity(pktbuf.len());
+            buf.clone_from(&pktbuf);
+            let abuf = NAAudioBuffer::new_from_buf(ainfo, Rc::new(RefCell::new(buf)), self.chmap.clone());
+            let mut frm = NAFrame::new_from_pkt(pkt, info, NABufferType::AudioPacked(abuf));
+            frm.set_duration(Some(duration));
+            frm.set_keyframe(true);
+            Ok(Rc::new(RefCell::new(frm)))
+        } else {
+            Err(DecoderError::InvalidData)
+        }
+    }
+}
+
+pub fn get_decoder() -> Box<NADecoder> {
+    Box::new(PCMDecoder::new())
+}