151 lines
4.1 KiB
C
151 lines
4.1 KiB
C
/*
|
|
* Copyright (c) 2010 The VP8 project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
|
|
#include "onyxd_int.h"
|
|
#include "entropymode.h"
|
|
#include "findnearmv.h"
|
|
|
|
|
|
int vp8_read_bmode(vp8_reader *bc, const vp8_prob *p)
|
|
{
|
|
const int i = vp8_treed_read(bc, vp8_bmode_tree, p);
|
|
|
|
return i;
|
|
}
|
|
|
|
|
|
int vp8_read_ymode(vp8_reader *bc, const vp8_prob *p)
|
|
{
|
|
const int i = vp8_treed_read(bc, vp8_ymode_tree, p);
|
|
|
|
return i;
|
|
}
|
|
|
|
int vp8_kfread_ymode(vp8_reader *bc, const vp8_prob *p)
|
|
{
|
|
const int i = vp8_treed_read(bc, vp8_kf_ymode_tree, p);
|
|
|
|
return i;
|
|
}
|
|
|
|
|
|
|
|
int vp8_read_uv_mode(vp8_reader *bc, const vp8_prob *p)
|
|
{
|
|
const int i = vp8_treed_read(bc, vp8_uv_mode_tree, p);
|
|
|
|
return i;
|
|
}
|
|
|
|
void vp8_read_mb_features(vp8_reader *r, MB_MODE_INFO *mi, MACROBLOCKD *x)
|
|
{
|
|
// Is segmentation enabled
|
|
if (x->segmentation_enabled && x->update_mb_segmentation_map)
|
|
{
|
|
// If so then read the segment id.
|
|
if (vp8_read(r, x->mb_segment_tree_probs[0]))
|
|
mi->segment_id = (unsigned char)(2 + vp8_read(r, x->mb_segment_tree_probs[2]));
|
|
else
|
|
mi->segment_id = (unsigned char)(vp8_read(r, x->mb_segment_tree_probs[1]));
|
|
}
|
|
}
|
|
|
|
void vp8_kfread_modes(VP8D_COMP *pbi)
|
|
{
|
|
VP8_COMMON *const cp = & pbi->common;
|
|
vp8_reader *const bc = & pbi->bc;
|
|
|
|
MODE_INFO *m = cp->mi;
|
|
const int ms = cp->mode_info_stride;
|
|
|
|
int mb_row = -1;
|
|
vp8_prob prob_skip_false = 0;
|
|
|
|
if (cp->mb_no_coeff_skip)
|
|
prob_skip_false = (vp8_prob)(vp8_read_literal(bc, 8));
|
|
|
|
while (++mb_row < cp->mb_rows)
|
|
{
|
|
int mb_col = -1;
|
|
|
|
while (++mb_col < cp->mb_cols)
|
|
{
|
|
MB_PREDICTION_MODE y_mode;
|
|
|
|
vp8dx_bool_decoder_fill(bc);
|
|
// Read the Macroblock segmentation map if it is being updated explicitly this frame (reset to 0 above by default)
|
|
// By default on a key frame reset all MBs to segment 0
|
|
m->mbmi.segment_id = 0;
|
|
|
|
if (pbi->mb.update_mb_segmentation_map)
|
|
vp8_read_mb_features(bc, &m->mbmi, &pbi->mb);
|
|
|
|
// Read the macroblock coeff skip flag if this feature is in use, else default to 0
|
|
if (cp->mb_no_coeff_skip)
|
|
m->mbmi.mb_skip_coeff = vp8_read(bc, prob_skip_false);
|
|
else
|
|
m->mbmi.mb_skip_coeff = 0;
|
|
|
|
y_mode = (MB_PREDICTION_MODE) vp8_kfread_ymode(bc, cp->kf_ymode_prob);
|
|
|
|
m->mbmi.ref_frame = INTRA_FRAME;
|
|
|
|
if ((m->mbmi.mode = y_mode) == B_PRED)
|
|
{
|
|
int i = 0;
|
|
|
|
do
|
|
{
|
|
const B_PREDICTION_MODE A = vp8_above_bmi(m, i, ms)->mode;
|
|
const B_PREDICTION_MODE L = vp8_left_bmi(m, i)->mode;
|
|
|
|
m->bmi[i].mode = (B_PREDICTION_MODE) vp8_read_bmode(bc, cp->kf_bmode_prob [A] [L]);
|
|
}
|
|
while (++i < 16);
|
|
}
|
|
else
|
|
{
|
|
int BMode;
|
|
int i = 0;
|
|
|
|
switch (y_mode)
|
|
{
|
|
case DC_PRED:
|
|
BMode = B_DC_PRED;
|
|
break;
|
|
case V_PRED:
|
|
BMode = B_VE_PRED;
|
|
break;
|
|
case H_PRED:
|
|
BMode = B_HE_PRED;
|
|
break;
|
|
case TM_PRED:
|
|
BMode = B_TM_PRED;
|
|
break;
|
|
default:
|
|
BMode = B_DC_PRED;
|
|
break;
|
|
}
|
|
|
|
do
|
|
{
|
|
m->bmi[i].mode = (B_PREDICTION_MODE)BMode;
|
|
}
|
|
while (++i < 16);
|
|
}
|
|
|
|
(m++)->mbmi.uv_mode = (MB_PREDICTION_MODE)vp8_read_uv_mode(bc, cp->kf_uv_mode_prob);
|
|
}
|
|
|
|
m++; // skip the border
|
|
}
|
|
}
|