vc2enc: use 32 bits for quantized coefficients LUT

16 bits were definitely not enough and caused artifacts to appear on
images at barely compressed images.

Signed-off-by: Rostislav Pehlivanov <atomnuker@gmail.com>
This commit is contained in:
Rostislav Pehlivanov 2016-02-27 18:27:18 +00:00
parent 5cc53c2e53
commit fc1d3cbfdc
1 changed files with 9 additions and 9 deletions

View File

@ -118,7 +118,7 @@ typedef struct VC2EncContext {
uint8_t quant[MAX_DWT_LEVELS][4]; uint8_t quant[MAX_DWT_LEVELS][4];
/* Coefficient LUT */ /* Coefficient LUT */
uint16_t *coef_lut_val; uint32_t *coef_lut_val;
uint8_t *coef_lut_len; uint8_t *coef_lut_len;
int num_x; /* #slices horizontally */ int num_x; /* #slices horizontally */
@ -162,7 +162,7 @@ static av_always_inline void put_padding(PutBitContext *pb, int bytes)
put_bits(pb, bits, 0); put_bits(pb, bits, 0);
} }
static av_always_inline void put_vc2_ue_uint(PutBitContext *pb, uint16_t val) static av_always_inline void put_vc2_ue_uint(PutBitContext *pb, uint32_t val)
{ {
int i; int i;
int pbits = 0, bits = 0, topbit = 1, maxval = 1; int pbits = 0, bits = 0, topbit = 1, maxval = 1;
@ -190,7 +190,7 @@ static av_always_inline void put_vc2_ue_uint(PutBitContext *pb, uint16_t val)
put_bits(pb, bits*2 + 1, (pbits << 1) | 1); put_bits(pb, bits*2 + 1, (pbits << 1) | 1);
} }
static av_always_inline int count_vc2_ue_uint(uint16_t val) static av_always_inline int count_vc2_ue_uint(uint32_t val)
{ {
int topbit = 1, maxval = 1; int topbit = 1, maxval = 1;
@ -206,8 +206,8 @@ static av_always_inline int count_vc2_ue_uint(uint16_t val)
return ff_log2(topbit)*2 + 1; return ff_log2(topbit)*2 + 1;
} }
static av_always_inline void get_vc2_ue_uint(uint16_t val, uint8_t *nbits, static av_always_inline void get_vc2_ue_uint(int val, uint8_t *nbits,
uint16_t *eval) uint32_t *eval)
{ {
int i; int i;
int pbits = 0, bits = 0, topbit = 1, maxval = 1; int pbits = 0, bits = 0, topbit = 1, maxval = 1;
@ -569,14 +569,14 @@ static void encode_subband(VC2EncContext *s, PutBitContext *pb, int sx, int sy,
const int qfactor = ff_dirac_qscale_tab[quant]; const int qfactor = ff_dirac_qscale_tab[quant];
const uint8_t *len_lut = &s->coef_lut_len[quant*COEF_LUT_TAB]; const uint8_t *len_lut = &s->coef_lut_len[quant*COEF_LUT_TAB];
const uint16_t *val_lut = &s->coef_lut_val[quant*COEF_LUT_TAB]; const uint32_t *val_lut = &s->coef_lut_val[quant*COEF_LUT_TAB];
dwtcoef *coeff = b->buf + top * b->stride; dwtcoef *coeff = b->buf + top * b->stride;
for (y = top; y < bottom; y++) { for (y = top; y < bottom; y++) {
for (x = left; x < right; x++) { for (x = left; x < right; x++) {
const int neg = coeff[x] < 0; const int neg = coeff[x] < 0;
uint16_t c_abs = FFABS(coeff[x]); uint32_t c_abs = FFABS(coeff[x]);
if (c_abs < COEF_LUT_TAB) { if (c_abs < COEF_LUT_TAB) {
const uint8_t len = len_lut[c_abs]; const uint8_t len = len_lut[c_abs];
if (len == 1) if (len == 1)
@ -635,7 +635,7 @@ static int count_hq_slice(VC2EncContext *s, BitCostCache *cache,
for (y = top; y < bottom; y++) { for (y = top; y < bottom; y++) {
for (x = left; x < right; x++) { for (x = left; x < right; x++) {
uint16_t c_abs = FFABS(buf[x]); uint32_t c_abs = FFABS(buf[x]);
if (c_abs < COEF_LUT_TAB) { if (c_abs < COEF_LUT_TAB) {
const int len = len_lut[c_abs]; const int len = len_lut[c_abs];
bits += len + (len != 1); bits += len + (len != 1);
@ -1155,7 +1155,7 @@ static av_cold int vc2_encode_init(AVCodecContext *avctx)
for (i = 0; i < s->q_ceil; i++) { for (i = 0; i < s->q_ceil; i++) {
for (j = 0; j < COEF_LUT_TAB; j++) { for (j = 0; j < COEF_LUT_TAB; j++) {
uint8_t *len_lut = &s->coef_lut_len[i*COEF_LUT_TAB]; uint8_t *len_lut = &s->coef_lut_len[i*COEF_LUT_TAB];
uint16_t *val_lut = &s->coef_lut_val[i*COEF_LUT_TAB]; uint32_t *val_lut = &s->coef_lut_val[i*COEF_LUT_TAB];
get_vc2_ue_uint(QUANT(j, ff_dirac_qscale_tab[i]), get_vc2_ue_uint(QUANT(j, ff_dirac_qscale_tab[i]),
&len_lut[j], &val_lut[j]); &len_lut[j], &val_lut[j]);
} }