+ #[allow(clippy::comparison_chain)]
+ fn decode_intra_mb_pred_quant(&mut self, bd: &mut dyn BlockDecoder, bdsp: &dyn BlockDSP, mb_pos: usize, binfo: &BlockInfo, sstate: &SliceState, apply_acpred: bool) -> DecoderResult<()> {
+ for i in 0..6 {
+ bd.decode_block_intra(&binfo, &sstate, binfo.get_q(), i, (binfo.cbp & (1 << (5 - i))) != 0, &mut self.blk[i])?;
+ let q = if binfo.get_q() < 8 {
+ 8
+ } else {
+ i16::from(binfo.get_q() * 2)
+ };
+ let qadd = (q + 1) >> 1;
+ let quant_gray = 1024 / q;
+ if apply_acpred && (binfo.acpred != ACPredMode::None) {
+ let has_b = (i == 1) || (i == 3) || !sstate.first_mb;
+ let has_a = (i == 2) || (i == 3) || !sstate.first_line;
+ let (b_mb, b_blk) = if has_b {
+ if (i == 1) || (i == 3) {
+ (mb_pos, i - 1)
+ } else if i < 4 {
+ (mb_pos - 1, i + 1)
+ } else {
+ (mb_pos - 1, i)
+ }
+ } else { (0, 0) };
+ let (a_mb, a_blk) = if has_a {
+ if (i == 2) || (i == 3) {
+ (mb_pos, i - 2)
+ } else if i < 4 {
+ (mb_pos - self.mb_w, i + 2)
+ } else {
+ (mb_pos - self.mb_w, i)
+ }
+ } else { (0, 0) };
+ let dc_a = if has_a && self.pred_coeffs[a_mb].ver[a_blk][0] != 0x7FFF {
+ self.pred_coeffs[a_mb].ver[a_blk][0]
+ } else {
+ quant_gray
+ };
+ let dc_b = if has_b && self.pred_coeffs[b_mb].hor[b_blk][0] != 0x7FFF {
+ self.pred_coeffs[b_mb].hor[b_blk][0]
+ } else {
+ quant_gray
+ };
+ let dc = match binfo.acpred {
+ ACPredMode::DC => {
+ if has_a && has_b {
+ (dc_a + dc_b + 1) >> 1
+ } else if has_a {
+ dc_a
+ } else if has_b {
+ dc_b
+ } else {
+ quant_gray
+ }
+ },
+ ACPredMode::Hor => {
+ if has_b {
+ for k in 1..8 {
+ self.blk[i][k * 8] += self.pred_coeffs[b_mb].hor[b_blk][k];
+ }
+ for k in 1..8 {
+ self.blk[i][k * 8] = clip_ac(self.blk[i][k * 8]);
+ }
+ dc_b
+ } else {
+ quant_gray
+ }
+ },
+ ACPredMode::Ver => {
+ if has_a {
+ for k in 1..8 {
+ self.blk[i][k] += self.pred_coeffs[a_mb].ver[a_blk][k];
+ }
+ for k in 1..8 {
+ self.blk[i][k] = clip_ac(self.blk[i][k]);
+ }
+ dc_a
+ } else {
+ quant_gray
+ }
+ },
+ ACPredMode::None => { 0 },
+ };
+ self.blk[i][0] += dc;
+ for t in 0..8 { self.pred_coeffs[mb_pos].hor[i][t] = self.blk[i][t * 8]; }
+ for t in 0..8 { self.pred_coeffs[mb_pos].ver[i][t] = self.blk[i][t]; }
+ }
+ if apply_acpred {
+ let start = if binfo.get_q() < 8 {
+ self.blk[i][0] <<= 3;
+ 1
+ } else {
+ 0
+ };
+ for el in self.blk[i].iter_mut().skip(start) {
+ if *el > 0 {
+ *el = *el * q + qadd;
+ } else if *el < 0 {
+ *el = *el * q - qadd;
+ }
+ }
+ }
+ bdsp.idct(&mut self.blk[i]);
+ }
+ Ok(())
+ }