Merge "Palette: count Y colors only for screen content." into nextgenv2
This commit is contained in:
@@ -1721,12 +1721,13 @@ static int rd_pick_palette_intra_sby(
|
|||||||
int dc_mode_cost, PALETTE_MODE_INFO *palette_mode_info,
|
int dc_mode_cost, PALETTE_MODE_INFO *palette_mode_info,
|
||||||
uint8_t *best_palette_color_map, TX_SIZE *best_tx, TX_TYPE *best_tx_type,
|
uint8_t *best_palette_color_map, TX_SIZE *best_tx, TX_TYPE *best_tx_type,
|
||||||
PREDICTION_MODE *mode_selected, int64_t *best_rd) {
|
PREDICTION_MODE *mode_selected, int64_t *best_rd) {
|
||||||
|
int rate_overhead = 0;
|
||||||
|
if (cpi->common.allow_screen_content_tools) {
|
||||||
MACROBLOCKD *const xd = &x->e_mbd;
|
MACROBLOCKD *const xd = &x->e_mbd;
|
||||||
MODE_INFO *const mic = xd->mi[0];
|
MODE_INFO *const mic = xd->mi[0];
|
||||||
const int rows = 4 * num_4x4_blocks_high_lookup[bsize];
|
const int rows = 4 * num_4x4_blocks_high_lookup[bsize];
|
||||||
const int cols = 4 * num_4x4_blocks_wide_lookup[bsize];
|
const int cols = 4 * num_4x4_blocks_wide_lookup[bsize];
|
||||||
int this_rate, this_rate_tokenonly, s, colors, n;
|
int this_rate, this_rate_tokenonly, s, colors, n;
|
||||||
int rate_overhead = 0;
|
|
||||||
int64_t this_distortion, this_rd;
|
int64_t this_distortion, this_rd;
|
||||||
const int src_stride = x->plane[0].src.stride;
|
const int src_stride = x->plane[0].src.stride;
|
||||||
const uint8_t *const src = x->plane[0].src.buf;
|
const uint8_t *const src = x->plane[0].src.buf;
|
||||||
@@ -1743,7 +1744,7 @@ static int rd_pick_palette_intra_sby(
|
|||||||
mic->mbmi.ext_intra_mode_info.use_ext_intra_mode[0] = 0;
|
mic->mbmi.ext_intra_mode_info.use_ext_intra_mode[0] = 0;
|
||||||
#endif // CONFIG_EXT_INTRA
|
#endif // CONFIG_EXT_INTRA
|
||||||
|
|
||||||
if (colors > 1 && colors <= 64 && cpi->common.allow_screen_content_tools) {
|
if (colors > 1 && colors <= 64) {
|
||||||
int r, c, i, j, k;
|
int r, c, i, j, k;
|
||||||
const int max_itr = 50;
|
const int max_itr = 50;
|
||||||
int color_ctx, color_idx = 0;
|
int color_ctx, color_idx = 0;
|
||||||
@@ -1817,8 +1818,8 @@ static int rd_pick_palette_intra_sby(
|
|||||||
|
|
||||||
vp10_calc_indices(data, centroids, color_map, rows * cols, k, 1);
|
vp10_calc_indices(data, centroids, color_map, rows * cols, k, 1);
|
||||||
|
|
||||||
super_block_yrd(cpi, x, &this_rate_tokenonly, &this_distortion, &s, NULL,
|
super_block_yrd(cpi, x, &this_rate_tokenonly, &this_distortion, &s,
|
||||||
bsize, *best_rd);
|
NULL, bsize, *best_rd);
|
||||||
if (this_rate_tokenonly == INT_MAX) continue;
|
if (this_rate_tokenonly == INT_MAX) continue;
|
||||||
|
|
||||||
this_rate =
|
this_rate =
|
||||||
@@ -1827,7 +1828,8 @@ static int rd_pick_palette_intra_sby(
|
|||||||
cpi->palette_y_size_cost[bsize - BLOCK_8X8][k - 2] +
|
cpi->palette_y_size_cost[bsize - BLOCK_8X8][k - 2] +
|
||||||
write_uniform_cost(k, color_map[0]) +
|
write_uniform_cost(k, color_map[0]) +
|
||||||
vp10_cost_bit(
|
vp10_cost_bit(
|
||||||
vp10_default_palette_y_mode_prob[bsize - BLOCK_8X8][palette_ctx],
|
vp10_default_palette_y_mode_prob[bsize -
|
||||||
|
BLOCK_8X8][palette_ctx],
|
||||||
1);
|
1);
|
||||||
for (i = 0; i < rows; ++i) {
|
for (i = 0; i < rows; ++i) {
|
||||||
for (j = (i == 0 ? 1 : 0); j < cols; ++j) {
|
for (j = (i == 0 ? 1 : 0); j < cols; ++j) {
|
||||||
@@ -1856,7 +1858,7 @@ static int rd_pick_palette_intra_sby(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return rate_overhead;
|
return rate_overhead;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user