src/codecs/mod.rs

   1 use frame::*;
   2 use std::rc::Rc;
   3 use std::cell::RefCell;
   4 use io::byteio::ByteIOError;
   5 use io::bitreader::BitReaderError;
   6 use io::codebook::CodebookError;
   7
   8 #[derive(Debug,Clone,Copy,PartialEq)]
   9 #[allow(dead_code)]
  10 pub enum DecoderError {
  11     NoFrame,
  12     AllocError,
  13     TryAgain,
  14     InvalidData,
  15     ShortData,
  16     MissingReference,
  17     NotImplemented,
  18     Bug,
  19 }
  20
  21 pub type DecoderResult<T> = Result<T, DecoderError>;
  22
  23 impl From<ByteIOError> for DecoderError {
  24     fn from(_: ByteIOError) -> Self { DecoderError::ShortData }
  25 }
  26
  27 impl From<BitReaderError> for DecoderError {
  28     fn from(e: BitReaderError) -> Self {
  29         match e {
  30             BitReaderError::BitstreamEnd => DecoderError::ShortData,
  31             _ => DecoderError::InvalidData,
  32         }
  33     }
  34 }
  35
  36 impl From<CodebookError> for DecoderError {
  37     fn from(_: CodebookError) -> Self { DecoderError::InvalidData }
  38 }
  39
  40 impl From<AllocatorError> for DecoderError {
  41     fn from(_: AllocatorError) -> Self { DecoderError::AllocError }
  42 }
  43
  44 macro_rules! validate {
  45     ($a:expr) => { if !$a { println!("check failed at {}:{}", file!(), line!()); return Err(DecoderError::InvalidData); } };
  46 }
  47
  48 #[allow(dead_code)]
  49 struct HAMShuffler {
  50     lastframe: Option<NAVideoBuffer<u8>>,
  51 }
  52
  53 impl HAMShuffler {
  54     #[allow(dead_code)]
  55     fn new() -> Self { HAMShuffler { lastframe: None } }
  56     #[allow(dead_code)]
  57     fn clear(&mut self) { self.lastframe = None; }
  58     #[allow(dead_code)]
  59     fn add_frame(&mut self, buf: NAVideoBuffer<u8>) {
  60         self.lastframe = Some(buf);
  61     }
  62     #[allow(dead_code)]
  63     fn clone_ref(&mut self) -> Option<NAVideoBuffer<u8>> {
  64         if let Some(ref mut frm) = self.lastframe {
  65             let newfrm = frm.copy_buffer();
  66             *frm = newfrm.clone();
  67             Some(newfrm)
  68         } else {
  69             None
  70         }
  71     }
  72     #[allow(dead_code)]
  73     fn get_output_frame(&mut self) -> Option<NAVideoBuffer<u8>> {
  74         match self.lastframe {
  75             Some(ref frm) => Some(frm.clone()),
  76             None => None,
  77         }
  78     }
  79 }
  80
  81 #[allow(dead_code)]
  82 struct IPShuffler {
  83     lastframe: Option<NAVideoBuffer<u8>>,
  84 }
  85
  86 impl IPShuffler {
  87     #[allow(dead_code)]
  88     fn new() -> Self { IPShuffler { lastframe: None } }
  89     #[allow(dead_code)]
  90     fn clear(&mut self) { self.lastframe = None; }
  91     #[allow(dead_code)]
  92     fn add_frame(&mut self, buf: NAVideoBuffer<u8>) {
  93         self.lastframe = Some(buf);
  94     }
  95     #[allow(dead_code)]
  96     fn get_ref(&mut self) -> Option<NAVideoBuffer<u8>> {
  97         if let Some(ref frm) = self.lastframe {
  98             Some(frm.clone())
  99         } else {
 100             None
 101         }
 102     }
 103 }
 104
 105 pub trait NADecoder {
 106     fn init(&mut self, info: Rc<NACodecInfo>) -> DecoderResult<()>;
 107     fn decode(&mut self, pkt: &NAPacket) -> DecoderResult<NAFrameRef>;
 108 }
 109
 110 #[derive(Clone,Copy)]
 111 pub struct DecoderInfo {
 112     name: &'static str,
 113     get_decoder: fn () -> Box<NADecoder>,
 114 }
 115
 116 #[cfg(feature="h263")]
 117 mod blockdsp;
 118
 119 #[cfg(feature="decoder_gdvvid")]
 120 mod gremlinvideo;
 121 #[cfg(any(feature="decoder_indeo2", feature="decoder_indeo3", feature="decoder_indeo4", feature="decoder_indeo5", feature="decoder_imc"))]
 122 mod indeo;
 123 #[cfg(feature="h263")]
 124 mod h263;
 125
 126 #[cfg(feature="decoder_pcm")]
 127 mod pcm;
 128
 129 const DECODERS: &[DecoderInfo] = &[
 130 #[cfg(feature="decoder_gdvvid")]
 131     DecoderInfo { name: "gdv-video", get_decoder: gremlinvideo::get_decoder },
 132 #[cfg(feature="decoder_indeo2")]
 133     DecoderInfo { name: "indeo2", get_decoder: indeo::indeo2::get_decoder },
 134 #[cfg(feature="decoder_indeo3")]
 135     DecoderInfo { name: "indeo3", get_decoder: indeo::indeo3::get_decoder },
 136 #[cfg(feature="decoder_indeo4")]
 137     DecoderInfo { name: "indeo4", get_decoder: indeo::indeo4::get_decoder },
 138 #[cfg(feature="decoder_indeo5")]
 139     DecoderInfo { name: "indeo5", get_decoder: indeo::indeo5::get_decoder },
 140 #[cfg(feature="decoder_intel263")]
 141     DecoderInfo { name: "intel263", get_decoder: h263::intel263::get_decoder },
 142 #[cfg(feature="decoder_realvideo1")]
 143     DecoderInfo { name: "realvideo1", get_decoder: h263::rv10::get_decoder },
 144 #[cfg(feature="decoder_realvideo2")]
 145     DecoderInfo { name: "realvideo2", get_decoder: h263::rv20::get_decoder },
 146
 147 #[cfg(feature="decoder_pcm")]
 148     DecoderInfo { name: "pcm", get_decoder: pcm::get_decoder },
 149 #[cfg(feature="decoder_imc")]
 150     DecoderInfo { name: "imc", get_decoder: indeo::imc::get_decoder_imc },
 151 #[cfg(feature="decoder_imc")]
 152     DecoderInfo { name: "iac", get_decoder: indeo::imc::get_decoder_iac },
 153 ];
 154
 155 pub fn find_decoder(name: &str) -> Option<fn () -> Box<NADecoder>> {
 156     for &dec in DECODERS {
 157         if dec.name == name {
 158             return Some(dec.get_decoder);
 159         }
 160     }
 161     None
 162 }