General code cleanup.
Making code more readable in different places. Change-Id: Iea92c9a35e64d257ee358879fc04fc926843d52e
This commit is contained in:
@@ -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);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,14 +1790,11 @@ 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)
|
||||||
vp9_adjust_default_coef_probs(pc);
|
vp9_adjust_default_coef_probs(pc);
|
||||||
@@ -1812,18 +1808,15 @@ 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
|
||||||
|
|
||||||
if (0) {
|
if (0) {
|
||||||
@@ -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;
|
||||||
|
|||||||
Reference in New Issue
Block a user