]> git.nihav.org Git - nihav.git/commitdiff
io: generic table codebook description reader
authorKostya Shishkov <kostya.shishkov@gmail.com>
Tue, 23 Oct 2018 17:05:02 +0000 (19:05 +0200)
committerKostya Shishkov <kostya.shishkov@gmail.com>
Tue, 23 Oct 2018 17:05:02 +0000 (19:05 +0200)
src/io/codebook.rs

index d67e790fe77a339b1be3e85af5090fcf73047f87..5bfd71991e186cfe7c855d2ce11411e1d27d8a59 100644 (file)
@@ -309,6 +309,25 @@ impl CodebookDescReader<u32> for ShortCodebookDescReader {
     fn len(&mut self) -> usize { self.data.len() }
 }
 
+pub struct TableCodebookDescReader<CodeType:'static, IndexType:'static> {
+    bits:       &'static [u8],
+    codes:      &'static [CodeType],
+    idx_map:    fn(usize) -> IndexType,
+}
+
+impl<'a, CodeType, IndexType> TableCodebookDescReader<CodeType, IndexType> {
+    pub fn new(codes: &'static [CodeType], bits: &'static [u8], idx_map: fn(usize) -> IndexType) -> Self {
+        Self { bits, codes, idx_map }
+    }
+}
+impl<CodeType: Copy+Into<u32>, IndexType> CodebookDescReader<IndexType> for TableCodebookDescReader<CodeType, IndexType>
+{
+    fn bits(&mut self, idx: usize) -> u8  { self.bits[idx] }
+    fn code(&mut self, idx: usize) -> u32 { self.codes[idx].into() }
+    fn sym (&mut self, idx: usize) -> IndexType { (self.idx_map)(idx) }
+    fn len(&mut self) -> usize { self.bits.len() }
+}
+
 #[cfg(test)]
 mod test {
     use super::*;