--- /dev/null
+use super::ivibr::{IVICodebook,IVI_CB_ZERO,RVMap,IVI_ZERO_RVMAP,IVI_RVMAPS};
+
+pub fn clip8(a: i16) -> u8 {
+ if a < 0 { 0 }
+ else if a > 255 { 255 }
+ else { a as u8 }
+}
+
+#[derive(Debug,Clone,Copy,PartialEq)]
+pub enum IVIFrameType {
+ Intra,
+ Inter,
+ Bidir,
+ Intra1,
+ InterDroppable,
+ InterScal,
+ NULL,
+ NULL2,
+}
+
+impl IVIFrameType {
+ pub fn is_intra(self) -> bool {
+ (self == IVIFrameType::Intra) || (self == IVIFrameType::Intra1)
+ }
+ pub fn is_null(self) -> bool {
+ (self == IVIFrameType::NULL) || (self == IVIFrameType::NULL2)
+ }
+ pub fn is_bidir(self) -> bool {
+ self == IVIFrameType::Bidir
+ }
+}
+
+#[derive(Clone,Copy)]
+pub struct PictureHeader {
+ pub ftype: IVIFrameType,
+ pub width: usize,
+ pub height: usize,
+ pub slice_w: usize,
+ pub slice_h: usize,
+ pub transparent: bool,
+ pub luma_bands: usize,
+ pub chroma_bands: usize,
+ pub in_q: bool,
+}
+
+impl PictureHeader {
+ pub fn new(ftype: IVIFrameType, width: usize, height: usize, slice_w: usize, slice_h: usize, transparent: bool, luma_bands: usize, chroma_bands: usize, in_q: bool) -> Self {
+ PictureHeader {
+ ftype: ftype,
+ width: width, height: height, slice_w: slice_w, slice_h: slice_h,
+ transparent: transparent,
+ luma_bands: luma_bands, chroma_bands: chroma_bands,
+ in_q: in_q,
+ }
+ }
+ pub fn new_null(ftype: IVIFrameType) -> Self {
+ PictureHeader {
+ ftype: ftype,
+ width: 0, height: 0, slice_w: 0, slice_h: 0,
+ transparent: false,
+ luma_bands: 0, chroma_bands: 0,
+ in_q: false,
+ }
+ }
+}
+
+#[derive(Debug,Clone,Copy,PartialEq)]
+pub enum TSize {
+ T8x8,
+ T4x4,
+}
+
+#[derive(Debug,Clone,Copy,PartialEq)]
+pub enum TDir {
+ TwoD,
+ Row,
+ Col,
+}
+
+#[derive(Debug,Clone,Copy,PartialEq)]
+pub enum IVITransformType {
+ Haar (TSize, TDir),
+ Slant(TSize, TDir),
+ DCT (TSize, TDir),
+ None (TSize),
+}
+
+pub type TrFunc = fn (&mut [i32; 64]);
+pub type TrFuncDC = fn (&mut [i32; 64], i32);
+
+impl IVITransformType {
+ pub fn is_8x8(&self) -> bool {
+ match *self {
+ IVITransformType::Haar (ref sz, _) => { *sz == TSize::T8x8 },
+ IVITransformType::Slant(ref sz, _) => { *sz == TSize::T8x8 },
+ IVITransformType::DCT (ref sz, _) => { *sz == TSize::T8x8 },
+ IVITransformType::None (ref sz) => { *sz == TSize::T8x8 },
+ }
+ }
+ pub fn is_2d(&self) -> bool {
+ match *self {
+ IVITransformType::Haar (_, ref dir) => { *dir == TDir::TwoD },
+ IVITransformType::Slant(_, ref dir) => { *dir == TDir::TwoD },
+ IVITransformType::DCT (_, ref dir) => { *dir == TDir::TwoD },
+ _ => { false },
+ }
+ }
+}
+
+#[allow(dead_code)]
+#[derive(Clone)]
+pub struct TxParams4x4 {
+ pub quant_intra: &'static [u16; 16],
+ pub quant_inter: &'static [u16; 16],
+ pub scan: &'static [usize; 16],
+}
+
+impl TxParams4x4 {
+ pub fn new(quant_intra: &'static [u16; 16], quant_inter: &'static [u16; 16], scan: &'static [usize; 16]) -> Self {
+ TxParams4x4 {
+ quant_intra: quant_intra, quant_inter: quant_inter, scan: scan,
+ }
+ }
+}
+
+#[allow(dead_code)]
+#[derive(Clone)]
+pub struct TxParams8x8 {
+ pub quant_intra: &'static [u16; 64],
+ pub quant_inter: &'static [u16; 64],
+ pub scan: &'static [usize; 64],
+}
+
+impl TxParams8x8 {
+ pub fn new(quant_intra: &'static [u16; 64], quant_inter: &'static [u16; 64], scan: &'static [usize; 64]) -> Self {
+ TxParams8x8 {
+ quant_intra: quant_intra, quant_inter: quant_inter, scan: scan,
+ }
+ }
+}
+
+#[derive(Clone)]
+pub enum TxType {
+ Transform4(TxParams4x4),
+ Transform8(TxParams8x8),
+ None,
+}
+
+pub const CORR_MAP_SIZE: usize = 122;
+
+#[derive(Clone)]
+pub struct BandHeader {
+ pub plane_no: usize,
+ pub band_no: usize,
+ pub empty: bool,
+ pub mb_size: usize,
+ pub blk_size: usize,
+ pub halfpel: bool,
+ pub inherit_mv: bool,
+ pub has_qdelta: bool,
+ pub inherit_qd: bool,
+ pub quant: u32,
+ pub blk_cb: IVICodebook,
+ pub rvmap: RVMap,
+ pub tr: IVITransformType,
+ pub ttype: TxType,
+}
+
+impl BandHeader {
+ pub fn new(plane_no: usize, band_no: usize, mb_size: usize, blk_size: usize, halfpel: bool, inherit_mv: bool, has_qdelta: bool, inherit_qd: bool, quant: u32, rvmap_idx: usize, num_corr: usize, corr_map: [u8; CORR_MAP_SIZE], blk_cb: IVICodebook, tr: IVITransformType, ttype: TxType) -> Self {
+ let mut rvmap = IVI_RVMAPS[rvmap_idx].clone();
+ for i in 0..num_corr {
+ let pos1 = corr_map[i * 2 + 0] as usize;
+ let pos2 = corr_map[i * 2 + 1] as usize;
+ let t = rvmap.runtab[pos1];
+ rvmap.runtab[pos1] = rvmap.runtab[pos2];
+ rvmap.runtab[pos2] = t;
+ let t = rvmap.valtab[pos1];
+ rvmap.valtab[pos1] = rvmap.valtab[pos2];
+ rvmap.valtab[pos2] = t;
+ }
+ BandHeader {
+ plane_no: plane_no, band_no: band_no,
+ empty: false, halfpel: halfpel,
+ inherit_mv: inherit_mv,
+ has_qdelta: has_qdelta, inherit_qd: inherit_qd, quant: quant,
+ mb_size: mb_size, blk_size: blk_size,
+ rvmap: rvmap, blk_cb: blk_cb,
+ tr: tr, ttype: ttype,
+ }
+ }
+ pub fn new_empty(plane_no: usize, band_no: usize) -> Self {
+ BandHeader {
+ plane_no: plane_no, band_no: band_no,
+ empty: true, halfpel: true,
+ inherit_mv: false, has_qdelta: false, inherit_qd: false, quant: 0,
+ mb_size: 0, blk_size: 0,
+ rvmap: IVI_ZERO_RVMAP, blk_cb: IVI_CB_ZERO,
+ tr: IVITransformType::None(TSize::T8x8), ttype: TxType::None,
+ }
+ }
+}
+
+#[derive(Debug,Clone,Copy,PartialEq)]
+pub enum MBType {
+ Intra,
+ Inter,
+ Backward,
+ Bidir,
+}
+
+#[derive(Clone,Copy)]
+pub struct MB {
+ pub mtype: MBType,
+ pub pos_x: usize,
+ pub pos_y: usize,
+ pub mv_x: i32,
+ pub mv_y: i32,
+ pub mv2_x: i32,
+ pub mv2_y: i32,
+ pub qd: i16,
+ pub q: u8,
+ pub cbp: u8,
+}
+
+impl MB {
+ pub fn new(x: usize, y: usize) -> Self {
+ MB {
+ mtype: MBType::Intra,
+ pos_x: x, pos_y: y,
+ mv_x: 0, mv_y: 0,
+ mv2_x: 0, mv2_y: 0,
+ cbp: 0, q: 0, qd: 0,
+ }
+ }
+}
+
+pub struct IVITile {
+ pub pos_x: usize,
+ pub pos_y: usize,
+ pub mb_w: usize,
+ pub mb_h: usize,
+ pub w: usize,
+ pub h: usize,
+ pub mb: Vec<MB>,
+}
+
+impl IVITile {
+ pub fn new(pos_x: usize, pos_y: usize, w: usize, h: usize) -> Self {
+ IVITile {
+ pos_x: pos_x, pos_y: pos_y, w: w, h: h,
+ mb_w: 0, mb_h: 0, mb: Vec::new(),
+ }
+ }
+}
+
+pub const IVI_ZIGZAG: [usize; 64] = [
+ 0, 1, 8, 16, 9, 2, 3, 10,
+ 17, 24, 32, 25, 18, 11, 4, 5,
+ 12, 19, 26, 33, 40, 48, 41, 34,
+ 27, 20, 13, 6, 7, 14, 21, 28,
+ 35, 42, 49, 56, 57, 50, 43, 36,
+ 29, 22, 15, 23, 30, 37, 44, 51,
+ 58, 59, 52, 45, 38, 31, 39, 46,
+ 53, 60, 61, 54, 47, 55, 62, 63
+];
+pub const IVI_SCAN_8X8_VER: [usize; 64] = [
+ 0, 8, 16, 24, 32, 40, 48, 56,
+ 1, 9, 17, 25, 33, 41, 49, 57,
+ 2, 10, 18, 26, 34, 42, 50, 58,
+ 3, 11, 19, 27, 35, 43, 51, 59,
+ 4, 12, 20, 28, 36, 44, 52, 60,
+ 5, 13, 21, 29, 37, 45, 53, 61,
+ 6, 14, 22, 30, 38, 46, 54, 62,
+ 7, 15, 23, 31, 39, 47, 55, 63
+];
+pub const IVI_SCAN_8X8_HOR: [usize; 64] = [
+ 0, 1, 2, 3, 4, 5, 6, 7,
+ 8, 9, 10, 11, 12, 13, 14, 15,
+ 16, 17, 18, 19, 20, 21, 22, 23,
+ 24, 25, 26, 27, 28, 29, 30, 31,
+ 32, 33, 34, 35, 36, 37, 38, 39,
+ 40, 41, 42, 43, 44, 45, 46, 47,
+ 48, 49, 50, 51, 52, 53, 54, 55,
+ 56, 57, 58, 59, 60, 61, 62, 63
+];
+pub const IVI_SCAN_4X4: [usize; 16] = [ 0, 1, 4, 8, 5, 2, 3, 6, 9, 12, 13, 10, 7, 11, 14, 15 ];
projects / nihav.git / blobdiff