1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-02-04 06:08:26 +02:00

avcodec/cfhddata: Avoid code duplication when creating codebooks

Reviewed-by: Paul B Mahol <onemda@gmail.com>
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2022-09-03 17:56:48 +02:00
parent 764cacfdb6
commit 320c36d9c7

View File

@ -124,38 +124,42 @@ static const CFHD_RL_ELEM table_18_vlc[NB_VLC_TABLE_18] = {
{ 1, 29, 12 }, { 1, 18, 10 }, { 1, 8, 7 }, { 1, 2, 3 },
};
av_cold int ff_cfhd_init_vlcs(CFHDContext *s)
static av_cold int cfhd_init_vlc(CFHD_RL_VLC_ELEM out[], unsigned out_size,
const CFHD_RL_ELEM table_vlc[], unsigned table_size,
VLC *vlc, void *logctx)
{
int i, j, ret = 0;
uint8_t new_cfhd_vlc_len[NB_VLC_TABLE_18 * 2];
uint16_t new_cfhd_vlc_run[NB_VLC_TABLE_18 * 2];
int16_t new_cfhd_vlc_level[NB_VLC_TABLE_18 * 2];
unsigned j;
int ret;
/** Similar to dv.c, generate signed VLC tables **/
/* Table 9 */
for (i = 0, j = 0; i < NB_VLC_TABLE_9; i++, j++) {
new_cfhd_vlc_len[j] = table_9_vlc[i].len;
new_cfhd_vlc_run[j] = table_9_vlc[i].run;
new_cfhd_vlc_level[j] = table_9_vlc[i].level;
for (unsigned i = j = 0; i < table_size; i++, j++) {
new_cfhd_vlc_len[j] = table_vlc[i].len;
new_cfhd_vlc_run[j] = table_vlc[i].run;
new_cfhd_vlc_level[j] = table_vlc[i].level;
/* Don't include the zero level nor escape bits */
if (table_9_vlc[i].level && table_9_vlc[i].run) {
if (table_vlc[i].level && table_vlc[i].run) {
new_cfhd_vlc_len[j]++;
j++;
new_cfhd_vlc_len[j] = table_9_vlc[i].len + 1;
new_cfhd_vlc_run[j] = table_9_vlc[i].run;
new_cfhd_vlc_level[j] = -table_9_vlc[i].level;
new_cfhd_vlc_len[j] = table_vlc[i].len + 1;
new_cfhd_vlc_run[j] = table_vlc[i].run;
new_cfhd_vlc_level[j] = -table_vlc[i].level;
}
}
ret = ff_init_vlc_from_lengths(&s->vlc_9, VLC_BITS, j, new_cfhd_vlc_len,
1, NULL, 0, 0, 0, 0, s->avctx);
ret = ff_init_vlc_from_lengths(vlc, VLC_BITS, j, new_cfhd_vlc_len,
1, NULL, 0, 0, 0, 0, logctx);
if (ret < 0)
return ret;
for (i = 0; i < s->vlc_9.table_size; i++) {
int code = s->vlc_9.table[i].sym;
int len = s->vlc_9.table[i].len;
av_assert0(vlc->table_size == out_size);
for (unsigned i = 0; i < out_size; i++) {
int code = vlc->table[i].sym;
int len = vlc->table[i].len;
int level, run;
if (len < 0) { // more bits needed
@ -165,49 +169,29 @@ av_cold int ff_cfhd_init_vlcs(CFHDContext *s)
run = new_cfhd_vlc_run[code];
level = new_cfhd_vlc_level[code];
}
s->table_9_rl_vlc[i].len = len;
s->table_9_rl_vlc[i].level = level;
s->table_9_rl_vlc[i].run = run;
out[i].len = len;
out[i].level = level;
out[i].run = run;
}
/* Table 18 */
for (i = 0, j = 0; i < NB_VLC_TABLE_18; i++, j++) {
new_cfhd_vlc_len[j] = table_18_vlc[i].len;
new_cfhd_vlc_run[j] = table_18_vlc[i].run;
new_cfhd_vlc_level[j] = table_18_vlc[i].level;
/* Don't include the zero level nor escape bits */
if (table_18_vlc[i].level && table_18_vlc[i].run) {
new_cfhd_vlc_len[j]++;
j++;
new_cfhd_vlc_len[j] = table_18_vlc[i].len + 1;
new_cfhd_vlc_run[j] = table_18_vlc[i].run;
new_cfhd_vlc_level[j] = -table_18_vlc[i].level;
}
}
ret = ff_init_vlc_from_lengths(&s->vlc_18, VLC_BITS, j, new_cfhd_vlc_len,
1, NULL, 0, 0, 0, 0, s->avctx);
if (ret < 0)
return ret;
av_assert0(s->vlc_18.table_size == 4572);
for (i = 0; i < s->vlc_18.table_size; i++) {
int code = s->vlc_18.table[i].sym;
int len = s->vlc_18.table[i].len;
int level, run;
if (len < 0) { // more bits needed
run = 0;
level = code;
} else {
run = new_cfhd_vlc_run[code];
level = new_cfhd_vlc_level[code];
}
s->table_18_rl_vlc[i].len = len;
s->table_18_rl_vlc[i].level = level;
s->table_18_rl_vlc[i].run = run;
}
return ret;
return 0;
}
av_cold int ff_cfhd_init_vlcs(CFHDContext *s)
{
int ret;
/* Table 9 */
ret = cfhd_init_vlc(s->table_9_rl_vlc, FF_ARRAY_ELEMS(s->table_9_rl_vlc),
table_9_vlc, FF_ARRAY_ELEMS(table_9_vlc),
&s->vlc_9, s->avctx);
if (ret < 0)
return ret;
/* Table 18 */
ret = cfhd_init_vlc(s->table_18_rl_vlc, FF_ARRAY_ELEMS(s->table_18_rl_vlc),
table_18_vlc, FF_ARRAY_ELEMS(table_18_vlc),
&s->vlc_18, s->avctx);
if (ret < 0)
return ret;
return 0;
}