[nihav.git] / nihav-duck / src / codecs / vp78.rs

use nihav_codec_support::data::GenericCache;
use super::vpcommon::*;

pub enum VPTreeDef<T: Copy> {
    Index(u8),
    Value(T),
}

pub trait VPTreeReader {
    fn read_tree<T:Copy>(&mut self, tree_def: &[VPTreeDef<T>], tree_prob: &[u8]) -> T;
}

impl<'a> VPTreeReader for BoolCoder<'a> {
    fn read_tree<T:Copy>(&mut self, tree_def: &[VPTreeDef<T>], tree_prob: &[u8]) -> T {
        let mut idx = 0;

        loop {
            let bit = self.read_prob(tree_prob[idx >> 1]);
            match tree_def[idx + (bit as usize)] {
                VPTreeDef::Value(v) => return v,
                VPTreeDef::Index(ix) => { idx = ix as usize; },
            };
        }
    }
}

#[repr(u8)]
#[derive(Clone,Copy,PartialEq,Debug)]
pub enum PredMode {
    DCPred,
    HPred,
    VPred,
    TMPred,
    BPred,

    //sub-block prediction modes
    LDPred,
    RDPred,
    VRPred,
    VLPred,
    HDPred,
    HUPred,

    Inter,
}

impl Default for PredMode {
    fn default() -> Self { PredMode::DCPred }
}

impl PredMode {
    pub fn to_b_mode(self) -> Self {
        if self == PredMode::DCPred {
            self
        } else {
            PredMode::TMPred
        }
    }
    pub fn to_b_index(self) -> usize {
        match self {
            PredMode::DCPred    => 0,
            PredMode::TMPred    => 1,
            PredMode::VPred     => 2,
            PredMode::HPred     => 3,
            PredMode::LDPred    => 4,
            PredMode::RDPred    => 5,
            PredMode::VRPred    => 6,
            PredMode::VLPred    => 7,
            PredMode::HDPred    => 8,
            PredMode::HUPred    => 9,
            _ => unreachable!(),
        }
    }
}

#[derive(Clone,Copy,PartialEq)]
pub enum DCTToken {
    Zero,
    One,
    Two,
    Three,
    Four,
    Cat1,
    Cat2,
    Cat3,
    Cat4,
    Cat5,
    Cat6,
    EOB,
}

pub fn expand_token(bc: &mut BoolCoder, token: DCTToken) -> i16 {
    let cat;
    match token {
        DCTToken::Zero  => return 0,
        DCTToken::One   => return if bc.read_bool() { -1 } else { 1 },
        DCTToken::Two   => return if bc.read_bool() { -2 } else { 2 },
        DCTToken::Three => return if bc.read_bool() { -3 } else { 3 },
        DCTToken::Four  => return if bc.read_bool() { -4 } else { 4 },
        DCTToken::Cat1  => cat = 0,
        DCTToken::Cat2  => cat = 1,
        DCTToken::Cat3  => cat = 2,
        DCTToken::Cat4  => cat = 3,
        DCTToken::Cat5  => cat = 4,
        DCTToken::Cat6  => cat = 5,
        _ => unreachable!(),
    };
    let mut add = 0i16;
    let add_probs = &VP56_COEF_ADD_PROBS[cat];
    for prob in add_probs.iter() {
        if *prob == 128 { break; }
        add                                 = (add << 1) | (bc.read_prob(*prob) as i16);
    }
    let sign                                = bc.read_bool();
    let level = VP56_COEF_BASE[cat] + add;
    if !sign {
        level
    } else {
        -level
    }
}

#[derive(Clone,Copy,Debug,PartialEq)]
pub enum MVSplitMode {
    TopBottom,
    LeftRight,
    Quarters,
    Sixteenths,
}

#[derive(Clone,Copy,Debug,PartialEq)]
pub enum SubMVRef {
    Left,
    Above,
    New,
    Zero,
}

pub struct PredCache {
    pub y_pred:         GenericCache<u8>,
    pub u_pred:         GenericCache<u8>,
    pub v_pred:         GenericCache<u8>,
    pub y2_pred:        GenericCache<u8>,
    pub y_pred_left:    [u8; 4],
    pub u_pred_left:    [u8; 2],
    pub v_pred_left:    [u8; 2],
    pub y2_pred_left:   u8,
}

impl PredCache {
    pub fn new() -> Self {
        Self {
            y_pred:         GenericCache::new(1, 1, 0),
            u_pred:         GenericCache::new(1, 1, 0),
            v_pred:         GenericCache::new(1, 1, 0),
            y2_pred:        GenericCache::new(1, 1, 0),
            y_pred_left:    [0; 4],
            u_pred_left:    [0; 2],
            v_pred_left:    [0; 2],
            y2_pred_left:   0,
        }
    }
    pub fn resize(&mut self, mb_w: usize) {
        self.y_pred     = GenericCache::new(4, mb_w * 4 + 1, 0);
        self.u_pred     = GenericCache::new(2, mb_w * 2 + 1, 0);
        self.v_pred     = GenericCache::new(2, mb_w * 2 + 1, 0);
        self.y2_pred    = GenericCache::new(1, mb_w     + 1, 0);
    }
    pub fn reset(&mut self) {
        self.y_pred.reset();
        self.u_pred.reset();
        self.v_pred.reset();
        self.y2_pred.reset();
        self.y_pred_left = [0; 4];
        self.u_pred_left = [0; 2];
        self.v_pred_left = [0; 2];
        self.y2_pred_left = 0;
    }
    pub fn update_row(&mut self) {
        self.y_pred.update_row();
        self.u_pred.update_row();
        self.v_pred.update_row();
        self.y2_pred.update_row();
    }
}

pub const Y_MODE_TREE: &[VPTreeDef<PredMode>] = &[
    VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
    VPTreeDef::Index(4),                VPTreeDef::Index(6),
    VPTreeDef::Value(PredMode::VPred),  VPTreeDef::Value(PredMode::HPred),
    VPTreeDef::Value(PredMode::TMPred), VPTreeDef::Value(PredMode::BPred),
];
pub const KF_Y_MODE_TREE: &[VPTreeDef<PredMode>] = &[
    VPTreeDef::Value(PredMode::BPred),  VPTreeDef::Index(2),
    VPTreeDef::Index(4),                VPTreeDef::Index(6),
    VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Value(PredMode::VPred),
    VPTreeDef::Value(PredMode::HPred),  VPTreeDef::Value(PredMode::TMPred),
];
pub const UV_MODE_TREE: &[VPTreeDef<PredMode>] = &[
    VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
    VPTreeDef::Value(PredMode::VPred),  VPTreeDef::Index(4),
    VPTreeDef::Value(PredMode::HPred),  VPTreeDef::Value(PredMode::TMPred)
];
pub const B_MODE_TREE: &[VPTreeDef<PredMode>] = &[
    VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
    VPTreeDef::Value(PredMode::TMPred), VPTreeDef::Index(4),
    VPTreeDef::Value(PredMode::VPred),  VPTreeDef::Index(6),
    VPTreeDef::Index(8),                VPTreeDef::Index(12),
    VPTreeDef::Value(PredMode::HPred),  VPTreeDef::Index(10),
    VPTreeDef::Value(PredMode::RDPred), VPTreeDef::Value(PredMode::VRPred),
    VPTreeDef::Value(PredMode::LDPred), VPTreeDef::Index(14),
    VPTreeDef::Value(PredMode::VLPred), VPTreeDef::Index(16),
    VPTreeDef::Value(PredMode::HDPred), VPTreeDef::Value(PredMode::HUPred)
];

pub const FEATURE_TREE: &[VPTreeDef<usize>] = &[
    VPTreeDef::Index(2), VPTreeDef::Index(4),
    VPTreeDef::Value(0), VPTreeDef::Value(1),
    VPTreeDef::Value(2), VPTreeDef::Value(3)
];

pub const COEF_TREE: &[VPTreeDef<DCTToken>] = &[
    VPTreeDef::Value(DCTToken::EOB),    VPTreeDef::Index(2),
    VPTreeDef::Value(DCTToken::Zero),   VPTreeDef::Index(4),
    VPTreeDef::Value(DCTToken::One),    VPTreeDef::Index(6),
    VPTreeDef::Index(8),                VPTreeDef::Index(12),
    VPTreeDef::Value(DCTToken::Two),    VPTreeDef::Index(10),
    VPTreeDef::Value(DCTToken::Three),  VPTreeDef::Value(DCTToken::Four),
    VPTreeDef::Index(14),               VPTreeDef::Index(16),
    VPTreeDef::Value(DCTToken::Cat1),   VPTreeDef::Value(DCTToken::Cat2),
    VPTreeDef::Index(18),               VPTreeDef::Index(20),
    VPTreeDef::Value(DCTToken::Cat3),   VPTreeDef::Value(DCTToken::Cat4),
    VPTreeDef::Value(DCTToken::Cat5),   VPTreeDef::Value(DCTToken::Cat6)
];

pub const MV_REF_TREE: &[VPTreeDef<VPMBType>] = &[
    VPTreeDef::Value(VPMBType::InterNoMV),      VPTreeDef::Index(2),
    VPTreeDef::Value(VPMBType::InterNearest),   VPTreeDef::Index(4),
    VPTreeDef::Value(VPMBType::InterNear),      VPTreeDef::Index(6),
    VPTreeDef::Value(VPMBType::InterMV),        VPTreeDef::Value(VPMBType::InterFourMV)
];
pub const SMALL_MV_TREE: &[VPTreeDef<i16>] = &[
    VPTreeDef::Index(2),    VPTreeDef::Index(8),
    VPTreeDef::Index(4),    VPTreeDef::Index(6),
    VPTreeDef::Value(0),    VPTreeDef::Value(1),
    VPTreeDef::Value(2),    VPTreeDef::Value(3),
    VPTreeDef::Index(10),   VPTreeDef::Index(12),
    VPTreeDef::Value(4),    VPTreeDef::Value(5),
    VPTreeDef::Value(6),    VPTreeDef::Value(7)
];
pub const MV_SPLIT_MODE_TREE: &[VPTreeDef<MVSplitMode>] = &[
    VPTreeDef::Value(MVSplitMode::Sixteenths),  VPTreeDef::Index(2),
    VPTreeDef::Value(MVSplitMode::Quarters),    VPTreeDef::Index(4),
    VPTreeDef::Value(MVSplitMode::TopBottom),   VPTreeDef::Value(MVSplitMode::LeftRight)
];
pub const SUB_MV_REF_TREE: &[VPTreeDef<SubMVRef>] = &[
    VPTreeDef::Value(SubMVRef::Left),   VPTreeDef::Index(2),
    VPTreeDef::Value(SubMVRef::Above),  VPTreeDef::Index(4),
    VPTreeDef::Value(SubMVRef::Zero),   VPTreeDef::Value(SubMVRef::New)
];
Commit	Line	Data
0cc75664 KS	1	use nihav_codec_support::data::GenericCache;
	2	use super::vpcommon::*;
	3
	4	pub enum VPTreeDef<T: Copy> {
	5	Index(u8),
	6	Value(T),
	7	}
	8
	9	pub trait VPTreeReader {
	10	fn read_tree<T:Copy>(&mut self, tree_def: &[VPTreeDef<T>], tree_prob: &[u8]) -> T;
	11	}
	12
	13	impl<'a> VPTreeReader for BoolCoder<'a> {
	14	fn read_tree<T:Copy>(&mut self, tree_def: &[VPTreeDef<T>], tree_prob: &[u8]) -> T {
	15	let mut idx = 0;
	16
	17	loop {
	18	let bit = self.read_prob(tree_prob[idx >> 1]);
	19	match tree_def[idx + (bit as usize)] {
	20	VPTreeDef::Value(v) => return v,
	21	VPTreeDef::Index(ix) => { idx = ix as usize; },
	22	};
	23	}
	24	}
	25	}
	26
	27	#[repr(u8)]
	28	#[derive(Clone,Copy,PartialEq,Debug)]
	29	pub enum PredMode {
	30	DCPred,
	31	HPred,
	32	VPred,
	33	TMPred,
	34	BPred,
	35
	36	//sub-block prediction modes
	37	LDPred,
	38	RDPred,
	39	VRPred,
	40	VLPred,
	41	HDPred,
	42	HUPred,
	43
	44	Inter,
	45	}
	46
	47	impl Default for PredMode {
	48	fn default() -> Self { PredMode::DCPred }
	49	}
	50
	51	impl PredMode {
	52	pub fn to_b_mode(self) -> Self {
	53	if self == PredMode::DCPred {
	54	self
	55	} else {
	56	PredMode::TMPred
	57	}
	58	}
	59	pub fn to_b_index(self) -> usize {
	60	match self {
	61	PredMode::DCPred => 0,
	62	PredMode::TMPred => 1,
	63	PredMode::VPred => 2,
	64	PredMode::HPred => 3,
65	PredMode::LDPred => 4,
66	PredMode::RDPred => 5,
67	PredMode::VRPred => 6,
68	PredMode::VLPred => 7,
69	PredMode::HDPred => 8,
70	PredMode::HUPred => 9,
71	_ => unreachable!(),
72	}
73	}
74	}
75
76	#[derive(Clone,Copy,PartialEq)]
77	pub enum DCTToken {
78	Zero,
79	One,
80	Two,
81	Three,
82	Four,
83	Cat1,
84	Cat2,
85	Cat3,
86	Cat4,
87	Cat5,
88	Cat6,
89	EOB,
90	}
91
92	pub fn expand_token(bc: &mut BoolCoder, token: DCTToken) -> i16 {
93	let cat;
94	match token {
95	DCTToken::Zero => return 0,
96	DCTToken::One => return if bc.read_bool() { -1 } else { 1 },
97	DCTToken::Two => return if bc.read_bool() { -2 } else { 2 },
98	DCTToken::Three => return if bc.read_bool() { -3 } else { 3 },
99	DCTToken::Four => return if bc.read_bool() { -4 } else { 4 },
100	DCTToken::Cat1 => cat = 0,
101	DCTToken::Cat2 => cat = 1,
102	DCTToken::Cat3 => cat = 2,
103	DCTToken::Cat4 => cat = 3,
104	DCTToken::Cat5 => cat = 4,
105	DCTToken::Cat6 => cat = 5,
106	_ => unreachable!(),
107	};
108	let mut add = 0i16;
109	let add_probs = &VP56_COEF_ADD_PROBS[cat];
110	for prob in add_probs.iter() {
111	if *prob == 128 { break; }
112	add = (add << 1) \| (bc.read_prob(*prob) as i16);
113	}
114	let sign = bc.read_bool();
115	let level = VP56_COEF_BASE[cat] + add;
116	if !sign {
117	level
118	} else {
119	-level
120	}
121	}
122
123	#[derive(Clone,Copy,Debug,PartialEq)]
124	pub enum MVSplitMode {
125	TopBottom,
126	LeftRight,
127	Quarters,
128	Sixteenths,
129	}
130
131	#[derive(Clone,Copy,Debug,PartialEq)]
132	pub enum SubMVRef {
133	Left,
134	Above,
135	New,
136	Zero,
137	}
138
139	pub struct PredCache {
140	pub y_pred: GenericCache<u8>,
141	pub u_pred: GenericCache<u8>,
142	pub v_pred: GenericCache<u8>,
143	pub y2_pred: GenericCache<u8>,
144	pub y_pred_left: [u8; 4],
145	pub u_pred_left: [u8; 2],
146	pub v_pred_left: [u8; 2],
147	pub y2_pred_left: u8,
148	}
149
150	impl PredCache {
151	pub fn new() -> Self {
152	Self {
153	y_pred: GenericCache::new(1, 1, 0),
154	u_pred: GenericCache::new(1, 1, 0),
155	v_pred: GenericCache::new(1, 1, 0),
156	y2_pred: GenericCache::new(1, 1, 0),
157	y_pred_left: [0; 4],
158	u_pred_left: [0; 2],
159	v_pred_left: [0; 2],
160	y2_pred_left: 0,
161	}
162	}
163	pub fn resize(&mut self, mb_w: usize) {
164	self.y_pred = GenericCache::new(4, mb_w * 4 + 1, 0);
165	self.u_pred = GenericCache::new(2, mb_w * 2 + 1, 0);
166	self.v_pred = GenericCache::new(2, mb_w * 2 + 1, 0);
167	self.y2_pred = GenericCache::new(1, mb_w + 1, 0);
168	}
169	pub fn reset(&mut self) {
170	self.y_pred.reset();
171	self.u_pred.reset();
172	self.v_pred.reset();
173	self.y2_pred.reset();
174	self.y_pred_left = [0; 4];
175	self.u_pred_left = [0; 2];
176	self.v_pred_left = [0; 2];
177	self.y2_pred_left = 0;
178	}
179	pub fn update_row(&mut self) {
180	self.y_pred.update_row();
181	self.u_pred.update_row();
182	self.v_pred.update_row();
183	self.y2_pred.update_row();
184	}
185	}
186
187	pub const Y_MODE_TREE: &[VPTreeDef<PredMode>] = &[
188	VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
189	VPTreeDef::Index(4), VPTreeDef::Index(6),
190	VPTreeDef::Value(PredMode::VPred), VPTreeDef::Value(PredMode::HPred),
191	VPTreeDef::Value(PredMode::TMPred), VPTreeDef::Value(PredMode::BPred),
192	];
193	pub const KF_Y_MODE_TREE: &[VPTreeDef<PredMode>] = &[
194	VPTreeDef::Value(PredMode::BPred), VPTreeDef::Index(2),
195	VPTreeDef::Index(4), VPTreeDef::Index(6),
196	VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Value(PredMode::VPred),
197	VPTreeDef::Value(PredMode::HPred), VPTreeDef::Value(PredMode::TMPred),
198	];
199	pub const UV_MODE_TREE: &[VPTreeDef<PredMode>] = &[
200	VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
201	VPTreeDef::Value(PredMode::VPred), VPTreeDef::Index(4),
202	VPTreeDef::Value(PredMode::HPred), VPTreeDef::Value(PredMode::TMPred)
203	];
204	pub const B_MODE_TREE: &[VPTreeDef<PredMode>] = &[
205	VPTreeDef::Value(PredMode::DCPred), VPTreeDef::Index(2),
206	VPTreeDef::Value(PredMode::TMPred), VPTreeDef::Index(4),
207	VPTreeDef::Value(PredMode::VPred), VPTreeDef::Index(6),
208	VPTreeDef::Index(8), VPTreeDef::Index(12),
209	VPTreeDef::Value(PredMode::HPred), VPTreeDef::Index(10),
210	VPTreeDef::Value(PredMode::RDPred), VPTreeDef::Value(PredMode::VRPred),
211	VPTreeDef::Value(PredMode::LDPred), VPTreeDef::Index(14),
212	VPTreeDef::Value(PredMode::VLPred), VPTreeDef::Index(16),
213	VPTreeDef::Value(PredMode::HDPred), VPTreeDef::Value(PredMode::HUPred)
214	];
215
216	pub const FEATURE_TREE: &[VPTreeDef<usize>] = &[
217	VPTreeDef::Index(2), VPTreeDef::Index(4),
218	VPTreeDef::Value(0), VPTreeDef::Value(1),
219	VPTreeDef::Value(2), VPTreeDef::Value(3)
220	];
221
222	pub const COEF_TREE: &[VPTreeDef<DCTToken>] = &[
223	VPTreeDef::Value(DCTToken::EOB), VPTreeDef::Index(2),
224	VPTreeDef::Value(DCTToken::Zero), VPTreeDef::Index(4),
225	VPTreeDef::Value(DCTToken::One), VPTreeDef::Index(6),
226	VPTreeDef::Index(8), VPTreeDef::Index(12),
227	VPTreeDef::Value(DCTToken::Two), VPTreeDef::Index(10),
228	VPTreeDef::Value(DCTToken::Three), VPTreeDef::Value(DCTToken::Four),
229	VPTreeDef::Index(14), VPTreeDef::Index(16),
230	VPTreeDef::Value(DCTToken::Cat1), VPTreeDef::Value(DCTToken::Cat2),
231	VPTreeDef::Index(18), VPTreeDef::Index(20),
232	VPTreeDef::Value(DCTToken::Cat3), VPTreeDef::Value(DCTToken::Cat4),
233	VPTreeDef::Value(DCTToken::Cat5), VPTreeDef::Value(DCTToken::Cat6)
234	];
235
236	pub const MV_REF_TREE: &[VPTreeDef<VPMBType>] = &[
237	VPTreeDef::Value(VPMBType::InterNoMV), VPTreeDef::Index(2),
238	VPTreeDef::Value(VPMBType::InterNearest), VPTreeDef::Index(4),
239	VPTreeDef::Value(VPMBType::InterNear), VPTreeDef::Index(6),
240	VPTreeDef::Value(VPMBType::InterMV), VPTreeDef::Value(VPMBType::InterFourMV)
241	];
242	pub const SMALL_MV_TREE: &[VPTreeDef<i16>] = &[
243	VPTreeDef::Index(2), VPTreeDef::Index(8),
244	VPTreeDef::Index(4), VPTreeDef::Index(6),
245	VPTreeDef::Value(0), VPTreeDef::Value(1),
246	VPTreeDef::Value(2), VPTreeDef::Value(3),
247	VPTreeDef::Index(10), VPTreeDef::Index(12),
248	VPTreeDef::Value(4), VPTreeDef::Value(5),
249	VPTreeDef::Value(6), VPTreeDef::Value(7)
250	];
251	pub const MV_SPLIT_MODE_TREE: &[VPTreeDef<MVSplitMode>] = &[
252	VPTreeDef::Value(MVSplitMode::Sixteenths), VPTreeDef::Index(2),
253	VPTreeDef::Value(MVSplitMode::Quarters), VPTreeDef::Index(4),
254	VPTreeDef::Value(MVSplitMode::TopBottom), VPTreeDef::Value(MVSplitMode::LeftRight)
255	];
256	pub const SUB_MV_REF_TREE: &[VPTreeDef<SubMVRef>] = &[
257	VPTreeDef::Value(SubMVRef::Left), VPTreeDef::Index(2),
258	VPTreeDef::Value(SubMVRef::Above), VPTreeDef::Index(4),
259	VPTreeDef::Value(SubMVRef::Zero), VPTreeDef::Value(SubMVRef::New)
260	];
261