aboutsummaryrefslogtreecommitdiffstats
path: root/libavcodec/vlc.h
blob: 679666801a5c6f3bd83054608db6e1fd1e9bab8e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#ifndef AVCODEC_VLC_H
#define AVCODEC_VLC_H

#include <stddef.h>
#include <stdint.h>

#include "libavutil/macros.h"

#define VLC_MULTI_MAX_SYMBOLS 6

// When changing this, be sure to also update tableprint_vlc.h accordingly.
typedef int16_t VLCBaseType;

typedef struct VLCElem {
    VLCBaseType sym, len;
} VLCElem;

typedef struct VLC {
    int bits;
    VLCElem *table;
    int table_size, table_allocated;
} VLC;

typedef struct VLC_MULTI_ELEM {
    uint8_t val[VLC_MULTI_MAX_SYMBOLS];
    int8_t len; // -31,32
    uint8_t num;
} VLC_MULTI_ELEM;

typedef struct VLC_MULTI {
    VLC_MULTI_ELEM *table;
    int table_size, table_allocated;
} VLC_MULTI;

typedef struct RL_VLC_ELEM {
    int16_t level;
    int8_t len;
    uint8_t run;
} RL_VLC_ELEM;

#define vlc_init(vlc, nb_bits, nb_codes,                \
                 bits, bits_wrap, bits_size,            \
                 codes, codes_wrap, codes_size,         \
                 flags)                                 \
    ff_vlc_init_sparse(vlc, nb_bits, nb_codes,          \
                       bits, bits_wrap, bits_size,      \
                       codes, codes_wrap, codes_size,   \
                       NULL, 0, 0, flags)

/**
 * Build VLC decoding tables suitable for use with get_vlc2().
 *
 * @param[in,out] vlc      The VLC to be initialized; table and table_allocated
 *                         must have been set when initializing a static VLC,
 *                         otherwise this will be treated as uninitialized.
 * @param[in] nb_bits      The number of bits to use for the VLC table;
 *                         higher values take up more memory and cache, but
 *                         allow to read codes with fewer reads.
 *                         Corresponds to the `bits` parameter of get_vlc2().
 * @param[in] nb_codes     The number of provided bits, codes and (if supplied)
 *                         symbol entries.
 * @param[in] bits         The lengths (in bits) of the codes. Entries > 0
 *                         correspond to valid codes; entries == 0 will be skipped.
 * @param[in] bits_wrap    Stride (in bytes) of the bits table.
 * @param[in] codes_size   Size of the bits. 1, 2 and 4 are supported.
 * @param[in] codes        Table which gives the bit pattern of of each vlc code.
 * @param[in] codes_wrap   Stride (in bytes) of the codes table.
 * @param[in] codes_size   Size of the codes. 1, 2 and 4 are supported.
 * @param[in] symbols      The symbols, i.e. what is returned from get_vlc2()
 *                         when the corresponding code is encountered.
 *                         May be NULL, then 0, 1, 2, 3, 4,... will be used.
 * @param[in] symbols_wrap Stride (in bytes) of the symbols table.
 * @param[in] symbols_size Size of the symbols. 1 and 2 are supported.
 * @param[in] flags        A combination of the VLC_INIT_* flags.
 *
 * 'wrap' and 'size' make it possible to use any memory configuration and types
 * (byte/word/int) to store the 'bits', 'codes', and 'symbols' tables.
 */
int ff_vlc_init_sparse(VLC *vlc, int nb_bits, int nb_codes,
                       const void *bits, int bits_wrap, int bits_size,
                       const void *codes, int codes_wrap, int codes_size,
                       const void *symbols, int symbols_wrap, int symbols_size,
                       int flags);

/**
 * Build VLC decoding tables suitable for use with get_vlc2()
 *
 * This function takes lengths and symbols and calculates the codes from them.
 * For this the input lengths and symbols have to be sorted according to "left
 * nodes in the corresponding tree first".
 *
 * @param[in,out] vlc      The VLC to be initialized; table and table_allocated
 *                         must have been set when initializing a static VLC,
 *                         otherwise this will be treated as uninitialized.
 * @param[in] nb_bits      The number of bits to use for the VLC table;
 *                         higher values take up more memory and cache, but
 *                         allow to read codes with fewer reads.
 * @param[in] nb_codes     The number of provided length and (if supplied) symbol
 *                         entries.
 * @param[in] lens         The lengths of the codes. Entries > 0 correspond to
 *                         valid codes; entries == 0 will be skipped and entries
 *                         with len < 0 indicate that the tree is incomplete and
 *                         has an open end of length -len at this position.
 * @param[in] lens_wrap    Stride (in bytes) of the lengths.
 * @param[in] symbols      The symbols, i.e. what is returned from get_vlc2()
 *                         when the corresponding code is encountered.
 *                         May be NULL, then 0, 1, 2, 3, 4,... will be used.
 * @param[in] symbols_wrap Stride (in bytes) of the symbols.
 * @param[in] symbols_size Size of the symbols. 1 and 2 are supported.
 * @param[in] offset       An offset to apply to all the valid symbols.
 * @param[in] flags        A combination of the VLC_INIT_* flags; notice that
 *                         VLC_INIT_INPUT_LE is pointless and ignored.
 */
int ff_vlc_init_from_lengths(VLC *vlc, int nb_bits, int nb_codes,
                             const int8_t *lens, int lens_wrap,
                             const void *symbols, int symbols_wrap, int symbols_size,
                             int offset, int flags, void *logctx);

/**
 * Build VLC decoding tables suitable for use with get_vlc_multi()
 *
 * This function takes lengths and symbols and calculates the codes from them.
 * For this the input lengths and symbols have to be sorted according to "left
 * nodes in the corresponding tree first".
 *
 * @param[in,out] vlc      The VLC to be initialized; table and table_allocated
 *                         must have been set when initializing a static VLC,
 *                         otherwise this will be treated as uninitialized.
 * @param[in,out] multi    The VLC_MULTI to be initialized; table and table_allocated
 *                         must have been set when initializing a static VLC,
 *                         otherwise this will be treated as uninitialized.
 * @param[in] nb_bits      The number of bits to use for the VLC table;
 *                         higher values take up more memory and cache, but
 *                         allow to read codes with fewer reads.
 * @param[in] nb_elems     The max possible number of elements.
 * @param[in] nb_codes     The number of provided length and (if supplied) symbol
 *                         entries.
 * @param[in] lens         The lengths of the codes. Entries > 0 correspond to
 *                         valid codes; entries == 0 will be skipped and entries
 *                         with len < 0 indicate that the tree is incomplete and
 *                         has an open end of length -len at this position.
 * @param[in] lens_wrap    Stride (in bytes) of the lengths.
 * @param[in] symbols      The symbols, i.e. what is returned from get_vlc2()
 *                         when the corresponding code is encountered.
 *                         May be NULL, then 0, 1, 2, 3, 4,... will be used.
 * @param[in] symbols_wrap Stride (in bytes) of the symbols.
 * @param[in] symbols_size Size of the symbols. 1 and 2 are supported.
 * @param[in] offset       An offset to apply to all the valid symbols.
 * @param[in] flags        A combination of the VLC_INIT_* flags; notice that
 *                         VLC_INIT_INPUT_LE is pointless and ignored.
 */
int ff_vlc_init_multi_from_lengths(VLC *vlc, VLC_MULTI *multi, int nb_bits, int nb_elems,
                                   int nb_codes, const int8_t *lens, int lens_wrap,
                                   const void *symbols, int symbols_wrap, int symbols_size,
                                   int offset, int flags, void *logctx);


void ff_vlc_free_multi(VLC_MULTI *vlc);
void ff_vlc_free(VLC *vlc);

#define VLC_INIT_USE_STATIC     1
#define VLC_INIT_STATIC_OVERLONG (2 | VLC_INIT_USE_STATIC)
/* If VLC_INIT_INPUT_LE is set, the LSB bit of the codes used to
 * initialize the VLC table is the first bit to be read. */
#define VLC_INIT_INPUT_LE       4
/* If set the VLC is intended for a little endian bitstream reader. */
#define VLC_INIT_OUTPUT_LE      8
#define VLC_INIT_LE             (VLC_INIT_INPUT_LE | VLC_INIT_OUTPUT_LE)

#define VLC_INIT_CUSTOM_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g,      \
                                      h, i, j, flags, static_size)         \
    do {                                                                   \
        static VLCElem table[static_size];                                 \
        (vlc)->table           = table;                                    \
        (vlc)->table_allocated = static_size;                              \
        ff_vlc_init_sparse(vlc, bits, a, b, c, d, e, f, g, h, i, j,        \
                           flags | VLC_INIT_USE_STATIC);                   \
    } while (0)

#define VLC_INIT_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g, h, i, j, static_size) \
    VLC_INIT_CUSTOM_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g,          \
                                  h, i, j, 0, static_size)

#define VLC_INIT_LE_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g, h, i, j, static_size) \
    VLC_INIT_CUSTOM_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g,          \
                                  h, i, j, VLC_INIT_LE, static_size)

#define VLC_INIT_CUSTOM_STATIC(vlc, bits, a, b, c, d, e, f, g, flags, static_size) \
    VLC_INIT_CUSTOM_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g,          \
                                  NULL, 0, 0, flags, static_size)

#define VLC_INIT_STATIC(vlc, bits, a, b, c, d, e, f, g, static_size)       \
    VLC_INIT_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g, NULL, 0, 0, static_size)

#define VLC_INIT_LE_STATIC(vlc, bits, a, b, c, d, e, f, g, static_size) \
    VLC_INIT_LE_SPARSE_STATIC(vlc, bits, a, b, c, d, e, f, g, NULL, 0, 0, static_size)

#define VLC_INIT_STATIC_FROM_LENGTHS(vlc, bits, nb_codes, lens, len_wrap,  \
                                     symbols, symbols_wrap, symbols_size,  \
                                     offset, flags, static_size)           \
    do {                                                                   \
        static VLCElem table[static_size];                                 \
        (vlc)->table           = table;                                    \
        (vlc)->table_allocated = static_size;                              \
        ff_vlc_init_from_lengths(vlc, bits, nb_codes, lens, len_wrap,      \
                                 symbols, symbols_wrap, symbols_size,      \
                                 offset, flags | VLC_INIT_USE_STATIC,      \
                                 NULL);                                    \
    } while (0)

/**
 * For static VLCs, the number of bits can often be hardcoded
 * at each get_vlc2() callsite. Then using a full VLC would be uneconomical,
 * because only VLC.table would ever be accessed after initialization.
 * The following functions provide wrappers around the relevant ff_vlc_init_*
 * functions suitable for said task.
 *
 * The ff_vlc_init_tables_* functions are intended to be used for initializing
 * a series of VLCs. The user initializes a VLCInitState with the details
 * about the underlying array of VLCElem; it is automatically updated by
 * the ff_vlc_init_tables_* functions (i.e. table is incremented and size
 * decremented by the number of elements of the current table).
 * The VLC_INIT_STATIC_OVERLONG flag is also automatically added.
 * These functions return a pointer to the table just initialized,
 * potentially to be used in arrays of pointer to VLC tables.
 *
 * The ff_vlc_init_table_* functions are intended to be used for initializing
 * a single VLC table, given by table and table_size. The VLC_INIT_USE_STATIC
 * flag is automatically added.
 */

typedef struct VLCInitState {
    VLCElem *table;  ///< points to where the next VLC table will be placed
    unsigned size;   ///< remaining number of elements in table
} VLCInitState;

#define VLC_INIT_STATE(_table) { .table = (_table), .size = FF_ARRAY_ELEMS(_table) }

void ff_vlc_init_table_from_lengths(VLCElem table[], int table_size,
                                    int nb_bits, int nb_codes,
                                    const int8_t *lens, int lens_wrap,
                                    const void *symbols, int symbols_wrap, int symbols_size,
                                    int offset, int flags);

const VLCElem *ff_vlc_init_tables_from_lengths(VLCInitState *state,
                                               int nb_bits, int nb_codes,
                                               const int8_t *lens, int lens_wrap,
                                               const void *symbols, int symbols_wrap, int symbols_size,
                                               int offset, int flags);

void ff_vlc_init_table_sparse(VLCElem table[], int table_size,
                              int nb_bits, int nb_codes,
                              const void *bits, int bits_wrap, int bits_size,
                              const void *codes, int codes_wrap, int codes_size,
                              const void *symbols, int symbols_wrap, int symbols_size,
                              int flags);

const VLCElem *ff_vlc_init_tables_sparse(VLCInitState *state,
                                         int nb_bits, int nb_codes,
                                         const void *bits, int bits_wrap, int bits_size,
                                         const void *codes, int codes_wrap, int codes_size,
                                         const void *symbols, int symbols_wrap, int symbols_size,
                                         int flags);

static inline
const VLCElem *ff_vlc_init_tables(VLCInitState *state,
                                  int nb_bits, int nb_codes,
                                  const void *bits, int bits_wrap, int bits_size,
                                  const void *codes, int codes_wrap, int codes_size,
                                  int flags)
{
    return ff_vlc_init_tables_sparse(state, nb_bits, nb_codes,
                                     bits, bits_wrap, bits_size,
                                     codes, codes_wrap, codes_size,
                                     NULL, 0, 0, flags);
}

#define VLC_INIT_STATIC_SPARSE_TABLE(vlc_table, nb_bits, nb_codes,         \
                                     bits, bits_wrap, bits_size,           \
                                     codes, codes_wrap, codes_size,        \
                                     symbols, symbols_wrap, symbols_size,  \
                                     flags)                                \
    ff_vlc_init_table_sparse(vlc_table, FF_ARRAY_ELEMS(vlc_table),         \
                             (nb_bits), (nb_codes),                        \
                             (bits), (bits_wrap), (bits_size),             \
                             (codes), (codes_wrap), (codes_size),          \
                             (symbols), (symbols_wrap), (symbols_size),    \
                             (flags))

#define VLC_INIT_STATIC_TABLE(vlc_table, nb_bits, nb_codes,                \
                              bits, bits_wrap, bits_size,                  \
                              codes, codes_wrap, codes_size,               \
                              flags)                                       \
    ff_vlc_init_table_sparse(vlc_table, FF_ARRAY_ELEMS(vlc_table),         \
                             (nb_bits), (nb_codes),                        \
                             (bits), (bits_wrap), (bits_size),             \
                             (codes), (codes_wrap), (codes_size),          \
                             NULL, 0, 0, (flags))

#define VLC_INIT_STATIC_TABLE_FROM_LENGTHS(vlc_table, nb_bits, nb_codes,   \
                                           lens, lens_wrap,                \
                                           syms, syms_wrap, syms_size,     \
                                           offset, flags)                  \
    ff_vlc_init_table_from_lengths(vlc_table, FF_ARRAY_ELEMS(vlc_table),   \
                                   (nb_bits), (nb_codes),                  \
                                   (lens), (lens_wrap),                    \
                                   (syms), (syms_wrap), (syms_size),       \
                                   (offset), (flags))

#endif /* AVCODEC_VLC_H */