src/codecs/mod.rs

   1 #[cfg(feature="decoder_indeo2")]
   2 pub mod indeo2;
   3 #[cfg(feature="decoder_pcm")]
   4 pub mod pcm;
   5
   6 use frame::*;
   7 use std::rc::Rc;
   8 use std::cell::RefCell;
   9 use io::byteio::ByteIOError;
  10 use io::bitreader::BitReaderError;
  11 use io::codebook::CodebookError;
  12
  13 #[derive(Debug,Clone,Copy,PartialEq)]
  14 #[allow(dead_code)]
  15 pub enum DecoderError {
  16     InvalidData,
  17     ShortData,
  18     MissingReference,
  19     NotImplemented,
  20     Bug,
  21 }
  22
  23 type DecoderResult<T> = Result<T, DecoderError>;
  24
  25 impl From<ByteIOError> for DecoderError {
  26     fn from(_: ByteIOError) -> Self { DecoderError::ShortData }
  27 }
  28
  29 impl From<BitReaderError> for DecoderError {
  30     fn from(e: BitReaderError) -> Self {
  31         match e {
  32             BitReaderError::BitstreamEnd => DecoderError::ShortData,
  33             _ => DecoderError::InvalidData,
  34         }
  35     }
  36 }
  37
  38 impl From<CodebookError> for DecoderError {
  39     fn from(_: CodebookError) -> Self { DecoderError::InvalidData }
  40 }
  41
  42 #[allow(dead_code)]
  43 struct HAMShuffler {
  44     lastframe: Option<NAVideoBuffer<u8>>,
  45 }
  46
  47 impl HAMShuffler {
  48     #[allow(dead_code)]
  49     fn new() -> Self { HAMShuffler { lastframe: None } }
  50     #[allow(dead_code)]
  51     fn clear(&mut self) { self.lastframe = None; }
  52     #[allow(dead_code)]
  53     fn add_frame(&mut self, buf: NAVideoBuffer<u8>) {
  54         self.lastframe = Some(buf);
  55     }
  56     #[allow(dead_code)]
  57     fn clone_ref(&mut self) -> Option<NAVideoBuffer<u8>> {
  58         if let Some(ref mut frm) = self.lastframe {
  59             let newfrm = frm.copy_buffer();
  60             *frm = newfrm.clone();
  61             Some(newfrm)
  62         } else {
  63             None
  64         }
  65     }
  66     #[allow(dead_code)]
  67     fn get_output_frame(&mut self) -> Option<NAVideoBuffer<u8>> {
  68         match self.lastframe {
  69             Some(ref frm) => Some(frm.clone()),
  70             None => None,
  71         }
  72     }
  73 }
  74
  75 pub trait NADecoder {
  76     fn init(&mut self, info: Rc<NACodecInfo>) -> DecoderResult<()>;
  77     fn decode(&mut self, pkt: &NAPacket) -> DecoderResult<NAFrameRef>;
  78 }
  79
  80 #[derive(Clone,Copy)]
  81 pub struct DecoderInfo {
  82     name: &'static str,
  83     get_decoder: fn () -> Box<NADecoder>,
  84 }
  85
  86 const DECODERS: &[DecoderInfo] = &[
  87 #[cfg(feature="decoder_indeo2")]
  88     DecoderInfo { name: "indeo2", get_decoder: indeo2::get_decoder },
  89 #[cfg(feature="decoder_pcm")]
  90     DecoderInfo { name: "pcm", get_decoder: pcm::get_decoder },
  91 ];
  92
  93 pub fn find_decoder(name: &str) -> Option<fn () -> Box<NADecoder>> {
  94     for &dec in DECODERS {
  95         if dec.name == name {
  96             return Some(dec.get_decoder);
  97         }
  98     }
  99     None
 100 }