hashirama/uvg266
1
0
Fork 0
mirror of https://github.com/ultravideo/uvg266.git synced 2024-11-27 11:24:05 +00:00
Code Issues Actions Projects Releases Wiki Activity

Set coeff flags correctly.

Browse source 
The coeff flags are no longer propagated upwards because encode_transform_tree
is being called from depth > 0. The fix is to initialize the whole coeff flag
array when the coeff flag is set.
This commit is contained in:
Ari Koivula 2014-02-27 15:45:07 +02:00
parent eda5ade06d
commit 5f75753f85
1 changed files with 24 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/encoder.c
View file

@ -1952,7 +1952,14 @@ void encode_transform_tree(encoder_control* encoder, int32_t x, int32_t y, uint8
cur_cu->coeff_y = 1;
cbf_y = 1;
if (depth <= MAX_DEPTH) {
cur_cu->coeff_top_y[depth] = 1;
int d;
for (d = 0; d <= depth; ++d) {
cur_cu->coeff_top_y[d] = 1;
}
while (d < MAX_PU_DEPTH + 4) {
cur_cu->coeff_top_y[d] = 0;
++d;
}
} else {
int pu_index = x_pu + 2 * y_pu;
cur_cu->coeff_top_y[depth + pu_index] = 1;
@ -2016,18 +2023,30 @@ void encode_transform_tree(encoder_control* encoder, int32_t x, int32_t y, uint8
transform_chroma(encoder, cur_cu, chroma_depth, base_u, pred_u, coeff_u, scan_idx_chroma, pre_quant_coeff, block);
for (i = 0; i < chroma_size; i++) {
if (coeff_u[i] != 0) {
// Found one, we can break here
int d;
for (d = 0; d <= depth; ++d) {
cur_cu->coeff_top_u[d] = 1;
}
while (d < MAX_PU_DEPTH + 4) {
cur_cu->coeff_top_u[d] = 0;
++d;
}
cur_cu->coeff_u = 1;
cur_cu->coeff_top_u[depth] = 1;
break;
}
}
transform_chroma(encoder, cur_cu, chroma_depth, base_v, pred_v, coeff_v, scan_idx_chroma, pre_quant_coeff, block);
for (i = 0; i < chroma_size; i++) {
if (coeff_v[i] != 0) {
// Found one, we can break here
int d;
for (d = 0; d <= depth; ++d) {
cur_cu->coeff_top_v[d] = 1;
}
while (d < MAX_PU_DEPTH + 4) {
cur_cu->coeff_top_v[d] = 0;
++d;
}
cur_cu->coeff_v = 1;
cur_cu->coeff_top_v[depth] = 1;
break;
}
}