General code cleanup.

Making code more readable in different places.

Change-Id: Iea92c9a35e64d257ee358879fc04fc926843d52e
This commit is contained in:
Dmitry Kovalev
2013-04-03 18:40:17 -07:00
parent dcd3a5c055
commit d5a017300c
3 changed files with 68 additions and 79 deletions

View File

@@ -16,13 +16,8 @@ typedef enum {
DEST = 1 DEST = 1
} BLOCKSET; } BLOCKSET;
static void setup_block(BLOCKD *b, static void setup_block(BLOCKD *b, uint8_t **base, uint8_t **base2,
int mv_stride, int stride, int offset, BLOCKSET bs) {
uint8_t **base,
uint8_t **base2,
int stride,
int offset,
BLOCKSET bs) {
if (bs == DEST) { if (bs == DEST) {
b->dst_stride = stride; b->dst_stride = stride;
b->dst = offset; b->dst = offset;
@@ -35,81 +30,84 @@ static void setup_block(BLOCKD *b,
} }
} }
static void setup_macroblock(MACROBLOCKD *xd, BLOCKSET bs) { static void setup_macroblock(MACROBLOCKD *mb, BLOCKSET bs) {
int block; BLOCKD *blockd = mb->block;
uint8_t **y, **u, **v, **y2, **u2, **v2;
uint8_t **y, **u, **v; int i, stride;
uint8_t **y2 = NULL, **u2 = NULL, **v2 = NULL;
BLOCKD *blockd = xd->block;
int stride;
if (bs == DEST) { if (bs == DEST) {
y = &xd->dst.y_buffer; y = &mb->dst.y_buffer;
u = &xd->dst.u_buffer; u = &mb->dst.u_buffer;
v = &xd->dst.v_buffer; v = &mb->dst.v_buffer;
y2 = NULL;
u2 = NULL;
v2 = NULL;
} else { } else {
y = &xd->pre.y_buffer; y = &mb->pre.y_buffer;
u = &xd->pre.u_buffer; u = &mb->pre.u_buffer;
v = &xd->pre.v_buffer; v = &mb->pre.v_buffer;
y2 = &xd->second_pre.y_buffer; y2 = &mb->second_pre.y_buffer;
u2 = &xd->second_pre.u_buffer; u2 = &mb->second_pre.u_buffer;
v2 = &xd->second_pre.v_buffer; v2 = &mb->second_pre.v_buffer;
} }
stride = xd->dst.y_stride; // luma
for (block = 0; block < 16; block++) { /* y blocks */ stride = mb->dst.y_stride;
setup_block(&blockd[block], stride, y, y2, stride, for (i = 0; i < 16; ++i) {
(block >> 2) * 4 * stride + (block & 3) * 4, bs); const int offset = (i >> 2) * 4 * stride + (i & 3) * 4;
setup_block(&blockd[i], y, y2, stride, offset, bs);
} }
stride = xd->dst.uv_stride; // chroma
for (block = 16; block < 20; block++) { /* U and V blocks */ stride = mb->dst.uv_stride;
setup_block(&blockd[block], stride, u, u2, stride, for (i = 16; i < 20; i++) {
((block - 16) >> 1) * 4 * stride + (block & 1) * 4, bs); const int offset = ((i - 16) >> 1) * 4 * stride + (i & 1) * 4;
setup_block(&blockd[i], u, u2, stride, offset, bs);
setup_block(&blockd[block + 4], stride, v, v2, stride, setup_block(&blockd[i + 4], v, v2, stride, offset, bs);
((block - 16) >> 1) * 4 * stride + (block & 1) * 4, bs);
} }
} }
void vp9_setup_block_dptrs(MACROBLOCKD *xd) { void vp9_setup_block_dptrs(MACROBLOCKD *mb) {
int r, c; int r, c;
BLOCKD *blockd = xd->block; BLOCKD *blockd = mb->block;
for (r = 0; r < 4; r++) { for (r = 0; r < 4; r++) {
for (c = 0; c < 4; c++) { for (c = 0; c < 4; c++) {
blockd[r * 4 + c].diff = &xd->diff[r * 4 * 16 + c * 4]; const int to = r * 4 + c;
blockd[r * 4 + c].predictor = xd->predictor + r * 4 * 16 + c * 4; const int from = r * 4 * 16 + c * 4;
blockd[to].diff = &mb->diff[from];
blockd[to].predictor = &mb->predictor[from];
} }
} }
for (r = 0; r < 2; r++) { for (r = 0; r < 2; r++) {
for (c = 0; c < 2; c++) { for (c = 0; c < 2; c++) {
blockd[16 + r * 2 + c].diff = &xd->diff[256 + r * 4 * 8 + c * 4]; const int to = 16 + r * 2 + c;
blockd[16 + r * 2 + c].predictor = const int from = 256 + r * 4 * 8 + c * 4;
xd->predictor + 256 + r * 4 * 8 + c * 4; blockd[to].diff = &mb->diff[from];
blockd[to].predictor = &mb->predictor[from];
} }
} }
for (r = 0; r < 2; r++) { for (r = 0; r < 2; r++) {
for (c = 0; c < 2; c++) { for (c = 0; c < 2; c++) {
blockd[20 + r * 2 + c].diff = &xd->diff[320 + r * 4 * 8 + c * 4]; const int to = 20 + r * 2 + c;
blockd[20 + r * 2 + c].predictor = const int from = 320 + r * 4 * 8 + c * 4;
xd->predictor + 320 + r * 4 * 8 + c * 4; blockd[to].diff = &mb->diff[from];
blockd[to].predictor = &mb->predictor[from];
} }
} }
for (r = 0; r < 24; r++) { for (r = 0; r < 24; r++) {
blockd[r].qcoeff = xd->qcoeff + r * 16; blockd[r].qcoeff = &mb->qcoeff[r * 16];
blockd[r].dqcoeff = xd->dqcoeff + r * 16; blockd[r].dqcoeff = &mb->dqcoeff[r * 16];
} }
} }
void vp9_build_block_doffsets(MACROBLOCKD *xd) { void vp9_build_block_doffsets(MACROBLOCKD *mb) {
/* handle the destination pitch features */ // handle the destination pitch features
setup_macroblock(xd, DEST); setup_macroblock(mb, DEST);
setup_macroblock(xd, PRED); setup_macroblock(mb, PRED);
} }

View File

@@ -15,17 +15,16 @@
void vp9_setup_intra_recon(YV12_BUFFER_CONFIG *ybf) { void vp9_setup_intra_recon(YV12_BUFFER_CONFIG *ybf) {
int i; int i;
/* set up frame new frame for intra coded blocks */ // luma
vpx_memset(ybf->y_buffer - 1 - ybf->y_stride, 127, ybf->y_width + 5); vpx_memset(ybf->y_buffer - 1 - ybf->y_stride, 127, ybf->y_width + 5);
for (i = 0; i < ybf->y_height; i++) for (i = 0; i < ybf->y_height; i++)
ybf->y_buffer[ybf->y_stride * i - 1] = (uint8_t) 129; ybf->y_buffer[ybf->y_stride * i - 1] = 129;
// chroma
vpx_memset(ybf->u_buffer - 1 - ybf->uv_stride, 127, ybf->uv_width + 5); vpx_memset(ybf->u_buffer - 1 - ybf->uv_stride, 127, ybf->uv_width + 5);
for (i = 0; i < ybf->uv_height; i++)
ybf->u_buffer[ybf->uv_stride * i - 1] = (uint8_t) 129;
vpx_memset(ybf->v_buffer - 1 - ybf->uv_stride, 127, ybf->uv_width + 5); vpx_memset(ybf->v_buffer - 1 - ybf->uv_stride, 127, ybf->uv_width + 5);
for (i = 0; i < ybf->uv_height; i++) for (i = 0; i < ybf->uv_height; i++) {
ybf->v_buffer[ybf->uv_stride * i - 1] = (uint8_t) 129; ybf->u_buffer[ybf->uv_stride * i - 1] = 129;
ybf->v_buffer[ybf->uv_stride * i - 1] = 129;
}
} }

View File

@@ -1057,7 +1057,6 @@ static void decode_sb_row(VP9D_COMP *pbi, VP9_COMMON *pc,
set_refs(pbi, 16, mb_row + y_idx, mb_col + x_idx); set_refs(pbi, 16, mb_row + y_idx, mb_col + x_idx);
decode_mb(pbi, xd, mb_row + y_idx, mb_col + x_idx, bc); decode_mb(pbi, xd, mb_row + y_idx, mb_col + x_idx, bc);
/* check if the boolean decoder has suffered an error */
xd->corrupted |= bool_error(bc); xd->corrupted |= bool_error(bc);
} }
} }
@@ -1791,13 +1790,10 @@ int vp9_decode_frame(VP9D_COMP *pbi, const uint8_t **p_data_end) {
// Read inter mode probability context updates // Read inter mode probability context updates
if (pc->frame_type != KEY_FRAME) { if (pc->frame_type != KEY_FRAME) {
int i, j; int i, j;
for (i = 0; i < INTER_MODE_CONTEXTS; i++) { for (i = 0; i < INTER_MODE_CONTEXTS; ++i)
for (j = 0; j < 4; j++) { for (j = 0; j < 4; ++j)
if (vp9_read(&header_bc, 252)) { if (vp9_read(&header_bc, 252))
pc->fc.vp9_mode_contexts[i][j] = vp9_read_prob(&header_bc); pc->fc.vp9_mode_contexts[i][j] = vp9_read_prob(&header_bc);
}
}
}
} }
#if CONFIG_MODELCOEFPROB && ADJUST_KF_COEF_PROBS #if CONFIG_MODELCOEFPROB && ADJUST_KF_COEF_PROBS
if (pc->frame_type == KEY_FRAME) if (pc->frame_type == KEY_FRAME)
@@ -1812,16 +1808,13 @@ int vp9_decode_frame(VP9D_COMP *pbi, const uint8_t **p_data_end) {
for (i = 0; i < MAX_REF_FRAMES; ++i) { for (i = 0; i < MAX_REF_FRAMES; ++i) {
// Skip the dummy entry for intra ref frame. // Skip the dummy entry for intra ref frame.
if (i == INTRA_FRAME) { if (i == INTRA_FRAME)
continue; continue;
}
// Read any updates to probabilities // Read any updates to probabilities
for (j = 0; j < MAX_MV_REF_CANDIDATES - 1; ++j) { for (j = 0; j < MAX_MV_REF_CANDIDATES - 1; ++j)
if (vp9_read(&header_bc, VP9_MVREF_UPDATE_PROB)) { if (vp9_read(&header_bc, VP9_MVREF_UPDATE_PROB))
xd->mb_mv_ref_probs[i][j] = vp9_read_prob(&header_bc); xd->mb_mv_ref_probs[i][j] = vp9_read_prob(&header_bc);
}
}
} }
} }
#endif #endif
@@ -1849,18 +1842,17 @@ int vp9_decode_frame(VP9D_COMP *pbi, const uint8_t **p_data_end) {
CHECK_MEM_ERROR(pc->last_frame_seg_map, CHECK_MEM_ERROR(pc->last_frame_seg_map,
vpx_calloc((pc->mb_rows * pc->mb_cols), 1)); vpx_calloc((pc->mb_rows * pc->mb_cols), 1));
/* set up frame new frame for intra coded blocks */ // set up frame new frame for intra coded blocks
vp9_setup_intra_recon(&pc->yv12_fb[pc->new_fb_idx]); vp9_setup_intra_recon(&pc->yv12_fb[pc->new_fb_idx]);
vp9_setup_block_dptrs(xd); vp9_setup_block_dptrs(xd);
vp9_build_block_doffsets(xd); vp9_build_block_doffsets(xd);
/* clear out the coeff buffer */ // clear out the coeff buffer
vpx_memset(xd->qcoeff, 0, sizeof(xd->qcoeff)); vpx_memset(xd->qcoeff, 0, sizeof(xd->qcoeff));
/* Read the mb_no_coeff_skip flag */ // Read the mb_no_coeff_skip flag
pc->mb_no_coeff_skip = (int)vp9_read_bit(&header_bc); pc->mb_no_coeff_skip = vp9_read_bit(&header_bc);
vp9_decode_mode_mvs_init(pbi, &header_bc); vp9_decode_mode_mvs_init(pbi, &header_bc);
@@ -1916,7 +1908,7 @@ int vp9_decode_frame(VP9D_COMP *pbi, const uint8_t **p_data_end) {
} }
#endif #endif
/* Find the end of the coded buffer */ // Find the end of the coded buffer
while (residual_bc.count > CHAR_BIT && while (residual_bc.count > CHAR_BIT &&
residual_bc.count < VP9_BD_VALUE_SIZE) { residual_bc.count < VP9_BD_VALUE_SIZE) {
residual_bc.count -= CHAR_BIT; residual_bc.count -= CHAR_BIT;