avcodec/dnxhdenc: Move allocation out of radix_sort()
[ffmpeg.git] / libavcodec / dnxhdenc.h
1 /*
2  * VC3/DNxHD encoder structure definitions and prototypes
3  * Copyright (c) 2007 Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
4  *
5  * VC-3 encoder funded by the British Broadcasting Corporation
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 #ifndef AVCODEC_DNXHDENC_H
25 #define AVCODEC_DNXHDENC_H
26
27 #include <stdint.h>
28
29 #include "mpegvideo.h"
30 #include "dnxhddata.h"
31
32 typedef struct RCCMPEntry {
33     uint16_t mb;
34     int value;
35 } RCCMPEntry;
36
37 typedef struct RCEntry {
38     int ssd;
39     int bits;
40 } RCEntry;
41
42 typedef struct DNXHDEncContext {
43     AVClass *class;
44     BlockDSPContext bdsp;
45     MpegEncContext m; ///< Used for quantization dsp functions
46
47     int cid;
48     int profile;
49     const CIDEntry *cid_table;
50     uint8_t *msip; ///< Macroblock Scan Indexes Payload
51     uint32_t *slice_size;
52     uint32_t *slice_offs;
53
54     struct DNXHDEncContext *thread[MAX_THREADS];
55
56     // Because our samples are either 8 or 16 bits for 8-bit and 10-bit
57     // encoding respectively, these refer either to bytes or to two-byte words.
58     unsigned dct_y_offset;
59     unsigned dct_uv_offset;
60     unsigned block_width_l2;
61
62     int frame_size;
63     int coding_unit_size;
64     int data_offset;
65
66     int interlaced;
67     int cur_field;
68
69     int nitris_compat;
70     unsigned min_padding;
71     int intra_quant_bias;
72
73     DECLARE_ALIGNED(16, int16_t, blocks)[8][64];
74     uint8_t edge_buf_y[256];
75     uint8_t edge_buf_uv[2][128];
76
77     int      (*qmatrix_c)     [64];
78     int      (*qmatrix_l)     [64];
79     uint16_t (*qmatrix_l16)[2][64];
80     uint16_t (*qmatrix_c16)[2][64];
81
82     unsigned frame_bits;
83     uint8_t *src[3];
84
85     uint32_t *vlc_codes;
86     uint8_t  *vlc_bits;
87     uint16_t *run_codes;
88     uint8_t  *run_bits;
89
90     /** Rate control */
91     unsigned slice_bits;
92     unsigned qscale;
93     unsigned lambda;
94
95     uint16_t *mb_bits;
96     uint8_t  *mb_qscale;
97
98     RCCMPEntry *mb_cmp;
99     RCCMPEntry *mb_cmp_tmp;
100     RCEntry    *mb_rc;
101
102     void (*get_pixels_8x4_sym)(int16_t * /* align 16 */,
103                                const uint8_t *, ptrdiff_t);
104 } DNXHDEncContext;
105
106 void ff_dnxhdenc_init_x86(DNXHDEncContext *ctx);
107
108 #endif /* AVCODEC_DNXHDENC_H */