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

[cclm] Add bits to RD calculation

Browse source
This commit is contained in:
Joose Sainio 2021-11-22 08:15:08 +02:00
parent 85ff5f23b1
commit 697d4c0652
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/search_intra.c
View file

@ -825,8 +825,20 @@ double kvz_chroma_mode_bits(const encoder_state_t *state, int8_t chroma_mode, in
if (chroma_mode == luma_mode) {
mode_bits = CTX_ENTROPY_FBITS(ctx, 0);
} else {
if(chroma_mode > 67) {
mode_bits = 2.0 + CTX_ENTROPY_FBITS(ctx, 1);
}
else {
ctx = &(state->cabac.ctx.cclm_model);
mode_bits = CTX_ENTROPY_FBITS(ctx, chroma_mode != 81);
if (chroma_mode != 81) mode_bits += 1;
}
}
// Technically this is encoded first but for this method of counting bits it does not matter
if(state->encoder_control->cfg.cclm) {
ctx = &(state->cabac.ctx.cclm_flag);
mode_bits += CTX_ENTROPY_FBITS(ctx, chroma_mode > 67);
}
return mode_bits;
}