Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * VC3/DNxHD decoder. | ||
3 | * Copyright (c) 2007 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com> | ||
4 | * Copyright (c) 2011 MirriAd Ltd | ||
5 | * Copyright (c) 2015 Christophe Gisquet | ||
6 | * | ||
7 | * 10 bit support added by MirriAd Ltd, Joseph Artsimovich <joseph@mirriad.com> | ||
8 | * Slice multithreading and MB interlaced support added by Christophe Gisquet | ||
9 | * | ||
10 | * This file is part of FFmpeg. | ||
11 | * | ||
12 | * FFmpeg is free software; you can redistribute it and/or | ||
13 | * modify it under the terms of the GNU Lesser General Public | ||
14 | * License as published by the Free Software Foundation; either | ||
15 | * version 2.1 of the License, or (at your option) any later version. | ||
16 | * | ||
17 | * FFmpeg is distributed in the hope that it will be useful, | ||
18 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
20 | * Lesser General Public License for more details. | ||
21 | * | ||
22 | * You should have received a copy of the GNU Lesser General Public | ||
23 | * License along with FFmpeg; if not, write to the Free Software | ||
24 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
25 | */ | ||
26 | |||
27 | #include "libavutil/mem_internal.h" | ||
28 | #include "libavutil/pixdesc.h" | ||
29 | |||
30 | #include "avcodec.h" | ||
31 | #include "blockdsp.h" | ||
32 | #include "codec_internal.h" | ||
33 | #include "decode.h" | ||
34 | #define UNCHECKED_BITSTREAM_READER 1 | ||
35 | #include "get_bits.h" | ||
36 | #include "dnxhddata.h" | ||
37 | #include "idctdsp.h" | ||
38 | #include "profiles.h" | ||
39 | #include "thread.h" | ||
40 | |||
41 | typedef struct RowContext { | ||
42 | DECLARE_ALIGNED(32, int16_t, blocks)[12][64]; | ||
43 | int luma_scale[64]; | ||
44 | int chroma_scale[64]; | ||
45 | GetBitContext gb; | ||
46 | int last_dc[3]; | ||
47 | int last_qscale; | ||
48 | int errors; | ||
49 | /** -1:not set yet 0:off=RGB 1:on=YUV 2:variable */ | ||
50 | int format; | ||
51 | } RowContext; | ||
52 | |||
53 | typedef struct DNXHDContext { | ||
54 | AVCodecContext *avctx; | ||
55 | RowContext *rows; | ||
56 | BlockDSPContext bdsp; | ||
57 | const uint8_t* buf; | ||
58 | int buf_size; | ||
59 | int64_t cid; ///< compression id | ||
60 | unsigned int width, height; | ||
61 | enum AVPixelFormat pix_fmt; | ||
62 | unsigned int mb_width, mb_height; | ||
63 | uint32_t mb_scan_index[512]; | ||
64 | int data_offset; // End of mb_scan_index, where macroblocks start | ||
65 | int cur_field; ///< current interlaced field | ||
66 | VLC ac_vlc, dc_vlc, run_vlc; | ||
67 | IDCTDSPContext idsp; | ||
68 | uint8_t permutated_scantable[64]; | ||
69 | const CIDEntry *cid_table; | ||
70 | int bit_depth; // 8, 10, 12 or 0 if not initialized at all. | ||
71 | int is_444; | ||
72 | int alpha; | ||
73 | int lla; | ||
74 | int mbaff; | ||
75 | int act; | ||
76 | int (*decode_dct_block)(const struct DNXHDContext *ctx, | ||
77 | RowContext *row, int n); | ||
78 | } DNXHDContext; | ||
79 | |||
80 | #define DNXHD_VLC_BITS 9 | ||
81 | #define DNXHD_DC_VLC_BITS 7 | ||
82 | |||
83 | static int dnxhd_decode_dct_block_8(const DNXHDContext *ctx, | ||
84 | RowContext *row, int n); | ||
85 | static int dnxhd_decode_dct_block_10(const DNXHDContext *ctx, | ||
86 | RowContext *row, int n); | ||
87 | static int dnxhd_decode_dct_block_10_444(const DNXHDContext *ctx, | ||
88 | RowContext *row, int n); | ||
89 | static int dnxhd_decode_dct_block_12(const DNXHDContext *ctx, | ||
90 | RowContext *row, int n); | ||
91 | static int dnxhd_decode_dct_block_12_444(const DNXHDContext *ctx, | ||
92 | RowContext *row, int n); | ||
93 | |||
94 | 149 | static av_cold int dnxhd_decode_init(AVCodecContext *avctx) | |
95 | { | ||
96 | 149 | DNXHDContext *ctx = avctx->priv_data; | |
97 | |||
98 | 149 | ctx->avctx = avctx; | |
99 | 149 | ctx->cid = -1; | |
100 |
2/2✓ Branch 0 taken 74 times.
✓ Branch 1 taken 75 times.
|
149 | if (avctx->colorspace == AVCOL_SPC_UNSPECIFIED) { |
101 | 74 | avctx->colorspace = AVCOL_SPC_BT709; | |
102 | } | ||
103 | |||
104 | 149 | avctx->coded_width = FFALIGN(avctx->width, 16); | |
105 | 149 | avctx->coded_height = FFALIGN(avctx->height, 16); | |
106 | |||
107 | 149 | ctx->rows = av_calloc(avctx->thread_count, sizeof(*ctx->rows)); | |
108 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 149 times.
|
149 | if (!ctx->rows) |
109 | ✗ | return AVERROR(ENOMEM); | |
110 | |||
111 | 149 | return 0; | |
112 | } | ||
113 | |||
114 | 502 | static int dnxhd_init_vlc(DNXHDContext *ctx, uint32_t cid, int bitdepth) | |
115 | { | ||
116 | int ret; | ||
117 |
2/2✓ Branch 0 taken 148 times.
✓ Branch 1 taken 354 times.
|
502 | if (cid != ctx->cid) { |
118 | 148 | const CIDEntry *cid_table = ff_dnxhd_get_cid_table(cid); | |
119 | |||
120 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 148 times.
|
148 | if (!cid_table) { |
121 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "unsupported cid %"PRIu32"\n", cid); | |
122 | ✗ | return AVERROR(ENOSYS); | |
123 | } | ||
124 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 144 times.
|
148 | if (cid_table->bit_depth != bitdepth && |
125 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | cid_table->bit_depth != DNXHD_VARIABLE) { |
126 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "bit depth mismatches %d %d\n", | |
127 | ✗ | cid_table->bit_depth, bitdepth); | |
128 | ✗ | return AVERROR_INVALIDDATA; | |
129 | } | ||
130 | 148 | ctx->cid_table = cid_table; | |
131 | 148 | av_log(ctx->avctx, AV_LOG_VERBOSE, "Profile cid %"PRIu32".\n", cid); | |
132 | |||
133 | 148 | ff_free_vlc(&ctx->ac_vlc); | |
134 | 148 | ff_free_vlc(&ctx->dc_vlc); | |
135 | 148 | ff_free_vlc(&ctx->run_vlc); | |
136 | |||
137 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
|
148 | if ((ret = init_vlc(&ctx->ac_vlc, DNXHD_VLC_BITS, 257, |
138 | ctx->cid_table->ac_bits, 1, 1, | ||
139 | ctx->cid_table->ac_codes, 2, 2, 0)) < 0) | ||
140 | ✗ | goto out; | |
141 |
3/4✓ Branch 0 taken 18 times.
✓ Branch 1 taken 130 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 148 times.
|
148 | if ((ret = init_vlc(&ctx->dc_vlc, DNXHD_DC_VLC_BITS, bitdepth > 8 ? 14 : 12, |
142 | ctx->cid_table->dc_bits, 1, 1, | ||
143 | ctx->cid_table->dc_codes, 1, 1, 0)) < 0) | ||
144 | ✗ | goto out; | |
145 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
|
148 | if ((ret = init_vlc(&ctx->run_vlc, DNXHD_VLC_BITS, 62, |
146 | ctx->cid_table->run_bits, 1, 1, | ||
147 | ctx->cid_table->run_codes, 2, 2, 0)) < 0) | ||
148 | ✗ | goto out; | |
149 | |||
150 | 148 | ctx->cid = cid; | |
151 | } | ||
152 | 502 | ret = 0; | |
153 | 502 | out: | |
154 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ret < 0) |
155 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "init_vlc failed\n"); | |
156 | 502 | return ret; | |
157 | } | ||
158 | |||
159 | 502 | static int dnxhd_get_profile(int cid) | |
160 | { | ||
161 |
6/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 120 times.
✓ Branch 3 taken 48 times.
✓ Branch 4 taken 111 times.
✓ Branch 5 taken 219 times.
|
502 | switch(cid) { |
162 | 2 | case 1270: | |
163 | 2 | return FF_PROFILE_DNXHR_444; | |
164 | 2 | case 1271: | |
165 | 2 | return FF_PROFILE_DNXHR_HQX; | |
166 | 120 | case 1272: | |
167 | 120 | return FF_PROFILE_DNXHR_HQ; | |
168 | 48 | case 1273: | |
169 | 48 | return FF_PROFILE_DNXHR_SQ; | |
170 | 111 | case 1274: | |
171 | 111 | return FF_PROFILE_DNXHR_LB; | |
172 | } | ||
173 | 219 | return FF_PROFILE_DNXHD; | |
174 | } | ||
175 | |||
176 | 502 | static int dnxhd_decode_header(DNXHDContext *ctx, AVFrame *frame, | |
177 | const uint8_t *buf, int buf_size, | ||
178 | int first_field) | ||
179 | { | ||
180 | int i, cid, ret; | ||
181 | 502 | int old_bit_depth = ctx->bit_depth, bitdepth; | |
182 | uint64_t header_prefix; | ||
183 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < 0x280) { |
184 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
185 | "buffer too small (%d < 640).\n", buf_size); | ||
186 | ✗ | return AVERROR_INVALIDDATA; | |
187 | } | ||
188 | |||
189 | 502 | header_prefix = ff_dnxhd_parse_header_prefix(buf); | |
190 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (header_prefix == 0) { |
191 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
192 | "unknown header 0x%02X 0x%02X 0x%02X 0x%02X 0x%02X\n", | ||
193 | ✗ | buf[0], buf[1], buf[2], buf[3], buf[4]); | |
194 | ✗ | return AVERROR_INVALIDDATA; | |
195 | } | ||
196 |
2/2✓ Branch 0 taken 134 times.
✓ Branch 1 taken 368 times.
|
502 | if (buf[5] & 2) { /* interlaced */ |
197 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 67 times.
|
134 | ctx->cur_field = first_field ? buf[5] & 1 : !ctx->cur_field; |
198 | 134 | frame->interlaced_frame = 1; | |
199 | 134 | frame->top_field_first = first_field ^ ctx->cur_field; | |
200 | 134 | av_log(ctx->avctx, AV_LOG_DEBUG, | |
201 | 134 | "interlaced %d, cur field %d\n", buf[5] & 3, ctx->cur_field); | |
202 | } else { | ||
203 | 368 | ctx->cur_field = 0; | |
204 | } | ||
205 | 502 | ctx->mbaff = (buf[0x6] >> 5) & 1; | |
206 | 502 | ctx->alpha = buf[0x7] & 1; | |
207 | 502 | ctx->lla = (buf[0x7] >> 1) & 1; | |
208 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ctx->alpha) |
209 | ✗ | avpriv_request_sample(ctx->avctx, "alpha"); | |
210 | |||
211 | 502 | ctx->height = AV_RB16(buf + 0x18); | |
212 | 502 | ctx->width = AV_RB16(buf + 0x1a); | |
213 | |||
214 |
3/4✓ Branch 0 taken 432 times.
✓ Branch 1 taken 68 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
502 | switch(buf[0x21] >> 5) { |
215 | 432 | case 1: bitdepth = 8; break; | |
216 | 68 | case 2: bitdepth = 10; break; | |
217 | 2 | case 3: bitdepth = 12; break; | |
218 | ✗ | default: | |
219 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
220 | ✗ | "Unknown bitdepth indicator (%d)\n", buf[0x21] >> 5); | |
221 | ✗ | return AVERROR_INVALIDDATA; | |
222 | } | ||
223 | |||
224 | 502 | cid = AV_RB32(buf + 0x28); | |
225 | |||
226 | 502 | ctx->avctx->profile = dnxhd_get_profile(cid); | |
227 | |||
228 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 502 times.
|
502 | if ((ret = dnxhd_init_vlc(ctx, cid, bitdepth)) < 0) |
229 | ✗ | return ret; | |
230 |
3/4✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
|
502 | if (ctx->mbaff && ctx->cid_table->cid != 1260) |
231 | ✗ | av_log(ctx->avctx, AV_LOG_WARNING, | |
232 | "Adaptive MB interlace flag in an unsupported profile.\n"); | ||
233 | |||
234 |
1/5✓ Branch 0 taken 502 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
|
502 | switch ((buf[0x2C] >> 1) & 3) { |
235 | 502 | case 0: frame->colorspace = AVCOL_SPC_BT709; break; | |
236 | ✗ | case 1: frame->colorspace = AVCOL_SPC_BT2020_NCL; break; | |
237 | ✗ | case 2: frame->colorspace = AVCOL_SPC_BT2020_CL; break; | |
238 | ✗ | case 3: frame->colorspace = AVCOL_SPC_UNSPECIFIED; break; | |
239 | } | ||
240 | |||
241 | 502 | ctx->act = buf[0x2C] & 1; | |
242 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
502 | if (ctx->act && ctx->cid_table->cid != 1256 && ctx->cid_table->cid != 1270) |
243 | ✗ | av_log(ctx->avctx, AV_LOG_WARNING, | |
244 | "Adaptive color transform in an unsupported profile.\n"); | ||
245 | |||
246 | 502 | ctx->is_444 = (buf[0x2C] >> 6) & 1; | |
247 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
|
502 | if (ctx->is_444) { |
248 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (bitdepth == 8) { |
249 | ✗ | avpriv_request_sample(ctx->avctx, "4:4:4 8 bits"); | |
250 | ✗ | return AVERROR_INVALIDDATA; | |
251 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | } else if (bitdepth == 10) { |
252 | 2 | ctx->decode_dct_block = dnxhd_decode_dct_block_10_444; | |
253 | 2 | ctx->pix_fmt = ctx->act ? AV_PIX_FMT_YUV444P10 | |
254 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | : AV_PIX_FMT_GBRP10; |
255 | } else { | ||
256 | ✗ | ctx->decode_dct_block = dnxhd_decode_dct_block_12_444; | |
257 | ✗ | ctx->pix_fmt = ctx->act ? AV_PIX_FMT_YUV444P12 | |
258 | ✗ | : AV_PIX_FMT_GBRP12; | |
259 | } | ||
260 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 498 times.
|
500 | } else if (bitdepth == 12) { |
261 | 2 | ctx->decode_dct_block = dnxhd_decode_dct_block_12; | |
262 | 2 | ctx->pix_fmt = AV_PIX_FMT_YUV422P12; | |
263 |
2/2✓ Branch 0 taken 66 times.
✓ Branch 1 taken 432 times.
|
498 | } else if (bitdepth == 10) { |
264 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
|
66 | if (ctx->avctx->profile == FF_PROFILE_DNXHR_HQX) |
265 | ✗ | ctx->decode_dct_block = dnxhd_decode_dct_block_10_444; | |
266 | else | ||
267 | 66 | ctx->decode_dct_block = dnxhd_decode_dct_block_10; | |
268 | 66 | ctx->pix_fmt = AV_PIX_FMT_YUV422P10; | |
269 | } else { | ||
270 | 432 | ctx->decode_dct_block = dnxhd_decode_dct_block_8; | |
271 | 432 | ctx->pix_fmt = AV_PIX_FMT_YUV422P; | |
272 | } | ||
273 | |||
274 | 502 | ctx->avctx->bits_per_raw_sample = ctx->bit_depth = bitdepth; | |
275 |
2/2✓ Branch 0 taken 148 times.
✓ Branch 1 taken 354 times.
|
502 | if (ctx->bit_depth != old_bit_depth) { |
276 | 148 | ff_blockdsp_init(&ctx->bdsp); | |
277 | 148 | ff_idctdsp_init(&ctx->idsp, ctx->avctx); | |
278 | 148 | ff_permute_scantable(ctx->permutated_scantable, ff_zigzag_direct, | |
279 | 148 | ctx->idsp.idct_permutation); | |
280 | } | ||
281 | |||
282 | // make sure profile size constraints are respected | ||
283 | // DNx100 allows 1920->1440 and 1280->960 subsampling | ||
284 |
2/2✓ Branch 0 taken 285 times.
✓ Branch 1 taken 217 times.
|
502 | if (ctx->width != ctx->cid_table->width && |
285 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 283 times.
|
285 | ctx->cid_table->width != DNXHD_VARIABLE) { |
286 | 2 | av_reduce(&ctx->avctx->sample_aspect_ratio.num, | |
287 | 2 | &ctx->avctx->sample_aspect_ratio.den, | |
288 | 2 | ctx->width, ctx->cid_table->width, 255); | |
289 | 2 | ctx->width = ctx->cid_table->width; | |
290 | } | ||
291 | |||
292 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < ctx->cid_table->coding_unit_size) { |
293 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "incorrect frame size (%d < %u).\n", | |
294 | ✗ | buf_size, ctx->cid_table->coding_unit_size); | |
295 | ✗ | return AVERROR_INVALIDDATA; | |
296 | } | ||
297 | |||
298 | 502 | ctx->mb_width = (ctx->width + 15)>> 4; | |
299 | 502 | ctx->mb_height = AV_RB16(buf + 0x16c); | |
300 | |||
301 |
3/4✓ Branch 0 taken 502 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 134 times.
✓ Branch 3 taken 368 times.
|
502 | if ((ctx->height + 15) >> 4 == ctx->mb_height && frame->interlaced_frame) |
302 | 134 | ctx->height <<= 1; | |
303 | |||
304 | 502 | av_log(ctx->avctx, AV_LOG_VERBOSE, "%dx%d, 4:%s %d bits, MBAFF=%d ACT=%d\n", | |
305 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
|
502 | ctx->width, ctx->height, ctx->is_444 ? "4:4" : "2:2", |
306 | ctx->bit_depth, ctx->mbaff, ctx->act); | ||
307 | |||
308 | // Newer format supports variable mb_scan_index sizes | ||
309 |
3/4✓ Branch 0 taken 108 times.
✓ Branch 1 taken 394 times.
✓ Branch 3 taken 108 times.
✗ Branch 4 not taken.
|
502 | if (ctx->mb_height > 68 && ff_dnxhd_check_header_prefix_hr(header_prefix)) { |
310 | 108 | ctx->data_offset = 0x170 + (ctx->mb_height << 2); | |
311 | } else { | ||
312 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 394 times.
|
394 | if (ctx->mb_height > 68) { |
313 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
314 | "mb height too big: %d\n", ctx->mb_height); | ||
315 | ✗ | return AVERROR_INVALIDDATA; | |
316 | } | ||
317 | 394 | ctx->data_offset = 0x280; | |
318 | } | ||
319 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if ((ctx->mb_height << frame->interlaced_frame) > (ctx->height + 15) >> 4) { |
320 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
321 | "mb height too big: %d\n", ctx->mb_height); | ||
322 | ✗ | return AVERROR_INVALIDDATA; | |
323 | } | ||
324 | |||
325 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < ctx->data_offset) { |
326 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
327 | "buffer too small (%d < %d).\n", buf_size, ctx->data_offset); | ||
328 | ✗ | return AVERROR_INVALIDDATA; | |
329 | } | ||
330 | |||
331 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ctx->mb_height > FF_ARRAY_ELEMS(ctx->mb_scan_index)) { |
332 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
333 | "mb_height too big (%d > %"SIZE_SPECIFIER").\n", ctx->mb_height, FF_ARRAY_ELEMS(ctx->mb_scan_index)); | ||
334 | ✗ | return AVERROR_INVALIDDATA; | |
335 | } | ||
336 | |||
337 |
2/2✓ Branch 0 taken 30173 times.
✓ Branch 1 taken 502 times.
|
30675 | for (i = 0; i < ctx->mb_height; i++) { |
338 | 30173 | ctx->mb_scan_index[i] = AV_RB32(buf + 0x170 + (i << 2)); | |
339 | ff_dlog(ctx->avctx, "mb scan index %d, pos %d: %"PRIu32"\n", | ||
340 | i, 0x170 + (i << 2), ctx->mb_scan_index[i]); | ||
341 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30173 times.
|
30173 | if (buf_size - ctx->data_offset < ctx->mb_scan_index[i]) { |
342 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
343 | "invalid mb scan index (%"PRIu32" vs %u).\n", | ||
344 | ✗ | ctx->mb_scan_index[i], buf_size - ctx->data_offset); | |
345 | ✗ | return AVERROR_INVALIDDATA; | |
346 | } | ||
347 | } | ||
348 | |||
349 | 502 | return 0; | |
350 | } | ||
351 | |||
352 | 34743936 | static av_always_inline int dnxhd_decode_dct_block(const DNXHDContext *ctx, | |
353 | RowContext *row, | ||
354 | int n, | ||
355 | int index_bits, | ||
356 | int level_bias, | ||
357 | int level_shift, | ||
358 | int dc_shift) | ||
359 | { | ||
360 | int i, j, index1, index2, len, flags; | ||
361 | int level, component, sign; | ||
362 | const int *scale; | ||
363 | const uint8_t *weight_matrix; | ||
364 | 34743936 | const uint8_t *ac_info = ctx->cid_table->ac_info; | |
365 | 34743936 | int16_t *block = row->blocks[n]; | |
366 | 34743936 | const int eob_index = ctx->cid_table->eob_index; | |
367 | 34743936 | int ret = 0; | |
368 | 34743936 | OPEN_READER(bs, &row->gb); | |
369 | |||
370 | 34743936 | ctx->bdsp.clear_block(block); | |
371 | |||
372 |
2/2✓ Branch 0 taken 34591296 times.
✓ Branch 1 taken 152640 times.
|
34743936 | if (!ctx->is_444) { |
373 |
2/2✓ Branch 0 taken 17295648 times.
✓ Branch 1 taken 17295648 times.
|
34591296 | if (n & 2) { |
374 | 17295648 | component = 1 + (n & 1); | |
375 | 17295648 | scale = row->chroma_scale; | |
376 | 17295648 | weight_matrix = ctx->cid_table->chroma_weight; | |
377 | } else { | ||
378 | 17295648 | component = 0; | |
379 | 17295648 | scale = row->luma_scale; | |
380 | 17295648 | weight_matrix = ctx->cid_table->luma_weight; | |
381 | } | ||
382 | } else { | ||
383 | 152640 | component = (n >> 1) % 3; | |
384 |
2/2✓ Branch 0 taken 101760 times.
✓ Branch 1 taken 50880 times.
|
152640 | if (component) { |
385 | 101760 | scale = row->chroma_scale; | |
386 | 101760 | weight_matrix = ctx->cid_table->chroma_weight; | |
387 | } else { | ||
388 | 50880 | scale = row->luma_scale; | |
389 | 50880 | weight_matrix = ctx->cid_table->luma_weight; | |
390 | } | ||
391 | } | ||
392 | |||
393 | 34743936 | UPDATE_CACHE(bs, &row->gb); | |
394 | 34743936 | GET_VLC(len, bs, &row->gb, ctx->dc_vlc.table, DNXHD_DC_VLC_BITS, 1); | |
395 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 34743936 times.
|
34743936 | if (len < 0) { |
396 | ✗ | ret = len; | |
397 | ✗ | goto error; | |
398 | } | ||
399 |
2/2✓ Branch 0 taken 32365764 times.
✓ Branch 1 taken 2378172 times.
|
34743936 | if (len) { |
400 | 32365764 | level = GET_CACHE(bs, &row->gb); | |
401 | 32365764 | LAST_SKIP_BITS(bs, &row->gb, len); | |
402 | 32365764 | sign = ~level >> 31; | |
403 | 32365764 | level = (NEG_USR32(sign ^ level, len) ^ sign) - sign; | |
404 | 32365764 | row->last_dc[component] += level * (1 << dc_shift); | |
405 | } | ||
406 | 34743936 | block[0] = row->last_dc[component]; | |
407 | |||
408 | 34743936 | i = 0; | |
409 | |||
410 | 34743936 | UPDATE_CACHE(bs, &row->gb); | |
411 |
2/2✓ Branch 1 taken 4524829 times.
✓ Branch 2 taken 30219107 times.
|
34743936 | GET_VLC(index1, bs, &row->gb, ctx->ac_vlc.table, |
412 | DNXHD_VLC_BITS, 2); | ||
413 | |||
414 |
2/2✓ Branch 0 taken 142194813 times.
✓ Branch 1 taken 34743936 times.
|
176938749 | while (index1 != eob_index) { |
415 | 142194813 | level = ac_info[2*index1+0]; | |
416 | 142194813 | flags = ac_info[2*index1+1]; | |
417 | |||
418 | 142194813 | sign = SHOW_SBITS(bs, &row->gb, 1); | |
419 | 142194813 | SKIP_BITS(bs, &row->gb, 1); | |
420 | |||
421 |
2/2✓ Branch 0 taken 2609257 times.
✓ Branch 1 taken 139585556 times.
|
142194813 | if (flags & 1) { |
422 | 2609257 | level += SHOW_UBITS(bs, &row->gb, index_bits) << 7; | |
423 | 2609257 | SKIP_BITS(bs, &row->gb, index_bits); | |
424 | } | ||
425 | |||
426 |
2/2✓ Branch 0 taken 46521239 times.
✓ Branch 1 taken 95673574 times.
|
142194813 | if (flags & 2) { |
427 | 46521239 | UPDATE_CACHE(bs, &row->gb); | |
428 |
2/2✓ Branch 1 taken 2125192 times.
✓ Branch 2 taken 44396047 times.
|
46521239 | GET_VLC(index2, bs, &row->gb, ctx->run_vlc.table, |
429 | DNXHD_VLC_BITS, 2); | ||
430 | 46521239 | i += ctx->cid_table->run[index2]; | |
431 | } | ||
432 | |||
433 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 142194813 times.
|
142194813 | if (++i > 63) { |
434 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "ac tex damaged %d, %d\n", n, i); | |
435 | ✗ | ret = -1; | |
436 | ✗ | break; | |
437 | } | ||
438 | |||
439 | 142194813 | j = ctx->permutated_scantable[i]; | |
440 | 142194813 | level *= scale[i]; | |
441 | 142194813 | level += scale[i] >> 1; | |
442 |
4/4✓ Branch 0 taken 125439973 times.
✓ Branch 1 taken 16754840 times.
✓ Branch 2 taken 100500911 times.
✓ Branch 3 taken 24939062 times.
|
142194813 | if (level_bias < 32 || weight_matrix[i] != level_bias) |
443 | 117255751 | level += level_bias; // 1<<(level_shift-1) | |
444 | 142194813 | level >>= level_shift; | |
445 | |||
446 | 142194813 | block[j] = (level ^ sign) - sign; | |
447 | |||
448 | 142194813 | UPDATE_CACHE(bs, &row->gb); | |
449 |
2/2✓ Branch 1 taken 10195080 times.
✓ Branch 2 taken 131999733 times.
|
142194813 | GET_VLC(index1, bs, &row->gb, ctx->ac_vlc.table, |
450 | DNXHD_VLC_BITS, 2); | ||
451 | } | ||
452 | 34743936 | error: | |
453 | 34743936 | CLOSE_READER(bs, &row->gb); | |
454 | 34743936 | return ret; | |
455 | } | ||
456 | |||
457 | 32375616 | static int dnxhd_decode_dct_block_8(const DNXHDContext *ctx, | |
458 | RowContext *row, int n) | ||
459 | { | ||
460 | 32375616 | return dnxhd_decode_dct_block(ctx, row, n, 4, 32, 6, 0); | |
461 | } | ||
462 | |||
463 | 2085120 | static int dnxhd_decode_dct_block_10(const DNXHDContext *ctx, | |
464 | RowContext *row, int n) | ||
465 | { | ||
466 | 2085120 | return dnxhd_decode_dct_block(ctx, row, n, 6, 8, 4, 0); | |
467 | } | ||
468 | |||
469 | 152640 | static int dnxhd_decode_dct_block_10_444(const DNXHDContext *ctx, | |
470 | RowContext *row, int n) | ||
471 | { | ||
472 | 152640 | return dnxhd_decode_dct_block(ctx, row, n, 6, 32, 6, 0); | |
473 | } | ||
474 | |||
475 | 130560 | static int dnxhd_decode_dct_block_12(const DNXHDContext *ctx, | |
476 | RowContext *row, int n) | ||
477 | { | ||
478 | 130560 | return dnxhd_decode_dct_block(ctx, row, n, 6, 8, 4, 2); | |
479 | } | ||
480 | |||
481 | ✗ | static int dnxhd_decode_dct_block_12_444(const DNXHDContext *ctx, | |
482 | RowContext *row, int n) | ||
483 | { | ||
484 | ✗ | return dnxhd_decode_dct_block(ctx, row, n, 6, 32, 4, 2); | |
485 | } | ||
486 | |||
487 | 4336632 | static int dnxhd_decode_macroblock(const DNXHDContext *ctx, RowContext *row, | |
488 | AVFrame *frame, int x, int y) | ||
489 | { | ||
490 | 4336632 | int shift1 = ctx->bit_depth >= 10; | |
491 | 4336632 | int dct_linesize_luma = frame->linesize[0]; | |
492 | 4336632 | int dct_linesize_chroma = frame->linesize[1]; | |
493 | uint8_t *dest_y, *dest_u, *dest_v; | ||
494 | int dct_y_offset, dct_x_offset; | ||
495 | int qscale, i, act; | ||
496 | 4336632 | int interlaced_mb = 0; | |
497 | |||
498 |
2/2✓ Branch 0 taken 12240 times.
✓ Branch 1 taken 4324392 times.
|
4336632 | if (ctx->mbaff) { |
499 | 12240 | interlaced_mb = get_bits1(&row->gb); | |
500 | 12240 | qscale = get_bits(&row->gb, 10); | |
501 | } else { | ||
502 | 4324392 | qscale = get_bits(&row->gb, 11); | |
503 | } | ||
504 | 4336632 | act = get_bits1(&row->gb); | |
505 |
2/2✓ Branch 0 taken 12720 times.
✓ Branch 1 taken 4323912 times.
|
4336632 | if (act) { |
506 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12720 times.
|
12720 | if (!ctx->act) { |
507 | static int act_warned; | ||
508 | ✗ | if (!act_warned) { | |
509 | ✗ | act_warned = 1; | |
510 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
511 | "ACT flag set, in violation of frame header.\n"); | ||
512 | } | ||
513 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 12718 times.
|
12720 | } else if (row->format == -1) { |
514 | 2 | row->format = act; | |
515 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12718 times.
|
12718 | } else if (row->format != act) { |
516 | ✗ | row->format = 2; // Variable | |
517 | } | ||
518 | } | ||
519 | |||
520 |
2/2✓ Branch 0 taken 354747 times.
✓ Branch 1 taken 3981885 times.
|
4336632 | if (qscale != row->last_qscale) { |
521 |
2/2✓ Branch 0 taken 22703808 times.
✓ Branch 1 taken 354747 times.
|
23058555 | for (i = 0; i < 64; i++) { |
522 | 22703808 | row->luma_scale[i] = qscale * ctx->cid_table->luma_weight[i]; | |
523 | 22703808 | row->chroma_scale[i] = qscale * ctx->cid_table->chroma_weight[i]; | |
524 | } | ||
525 | 354747 | row->last_qscale = qscale; | |
526 | } | ||
527 | |||
528 |
2/2✓ Branch 0 taken 34743936 times.
✓ Branch 1 taken 4336632 times.
|
39080568 | for (i = 0; i < 8 + 4 * ctx->is_444; i++) { |
529 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 34743936 times.
|
34743936 | if (ctx->decode_dct_block(ctx, row, i) < 0) |
530 | ✗ | return AVERROR_INVALIDDATA; | |
531 | } | ||
532 | |||
533 |
2/2✓ Branch 0 taken 546720 times.
✓ Branch 1 taken 3789912 times.
|
4336632 | if (frame->interlaced_frame) { |
534 | 546720 | dct_linesize_luma <<= 1; | |
535 | 546720 | dct_linesize_chroma <<= 1; | |
536 | } | ||
537 | |||
538 | 4336632 | dest_y = frame->data[0] + ((y * dct_linesize_luma) << 4) + (x << (4 + shift1)); | |
539 | 4336632 | dest_u = frame->data[1] + ((y * dct_linesize_chroma) << 4) + (x << (3 + shift1 + ctx->is_444)); | |
540 | 4336632 | dest_v = frame->data[2] + ((y * dct_linesize_chroma) << 4) + (x << (3 + shift1 + ctx->is_444)); | |
541 | |||
542 |
4/4✓ Branch 0 taken 546720 times.
✓ Branch 1 taken 3789912 times.
✓ Branch 2 taken 273360 times.
✓ Branch 3 taken 273360 times.
|
4336632 | if (frame->interlaced_frame && ctx->cur_field) { |
543 | 273360 | dest_y += frame->linesize[0]; | |
544 | 273360 | dest_u += frame->linesize[1]; | |
545 | 273360 | dest_v += frame->linesize[2]; | |
546 | } | ||
547 |
2/2✓ Branch 0 taken 3224 times.
✓ Branch 1 taken 4333408 times.
|
4336632 | if (interlaced_mb) { |
548 | 3224 | dct_linesize_luma <<= 1; | |
549 | 3224 | dct_linesize_chroma <<= 1; | |
550 | } | ||
551 | |||
552 |
2/2✓ Branch 0 taken 3224 times.
✓ Branch 1 taken 4333408 times.
|
4336632 | dct_y_offset = interlaced_mb ? frame->linesize[0] : (dct_linesize_luma << 3); |
553 | 4336632 | dct_x_offset = 8 << shift1; | |
554 |
2/2✓ Branch 0 taken 4323912 times.
✓ Branch 1 taken 12720 times.
|
4336632 | if (!ctx->is_444) { |
555 | 4323912 | ctx->idsp.idct_put(dest_y, dct_linesize_luma, row->blocks[0]); | |
556 | 4323912 | ctx->idsp.idct_put(dest_y + dct_x_offset, dct_linesize_luma, row->blocks[1]); | |
557 | 4323912 | ctx->idsp.idct_put(dest_y + dct_y_offset, dct_linesize_luma, row->blocks[4]); | |
558 | 4323912 | ctx->idsp.idct_put(dest_y + dct_y_offset + dct_x_offset, dct_linesize_luma, row->blocks[5]); | |
559 | |||
560 |
1/2✓ Branch 0 taken 4323912 times.
✗ Branch 1 not taken.
|
4323912 | if (!(ctx->avctx->flags & AV_CODEC_FLAG_GRAY)) { |
561 |
2/2✓ Branch 0 taken 3224 times.
✓ Branch 1 taken 4320688 times.
|
4323912 | dct_y_offset = interlaced_mb ? frame->linesize[1] : (dct_linesize_chroma << 3); |
562 | 4323912 | ctx->idsp.idct_put(dest_u, dct_linesize_chroma, row->blocks[2]); | |
563 | 4323912 | ctx->idsp.idct_put(dest_v, dct_linesize_chroma, row->blocks[3]); | |
564 | 4323912 | ctx->idsp.idct_put(dest_u + dct_y_offset, dct_linesize_chroma, row->blocks[6]); | |
565 | 4323912 | ctx->idsp.idct_put(dest_v + dct_y_offset, dct_linesize_chroma, row->blocks[7]); | |
566 | } | ||
567 | } else { | ||
568 | 12720 | ctx->idsp.idct_put(dest_y, dct_linesize_luma, row->blocks[0]); | |
569 | 12720 | ctx->idsp.idct_put(dest_y + dct_x_offset, dct_linesize_luma, row->blocks[1]); | |
570 | 12720 | ctx->idsp.idct_put(dest_y + dct_y_offset, dct_linesize_luma, row->blocks[6]); | |
571 | 12720 | ctx->idsp.idct_put(dest_y + dct_y_offset + dct_x_offset, dct_linesize_luma, row->blocks[7]); | |
572 | |||
573 |
1/2✓ Branch 0 taken 12720 times.
✗ Branch 1 not taken.
|
12720 | if (!(ctx->avctx->flags & AV_CODEC_FLAG_GRAY)) { |
574 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12720 times.
|
12720 | dct_y_offset = interlaced_mb ? frame->linesize[1] : (dct_linesize_chroma << 3); |
575 | 12720 | ctx->idsp.idct_put(dest_u, dct_linesize_chroma, row->blocks[2]); | |
576 | 12720 | ctx->idsp.idct_put(dest_u + dct_x_offset, dct_linesize_chroma, row->blocks[3]); | |
577 | 12720 | ctx->idsp.idct_put(dest_u + dct_y_offset, dct_linesize_chroma, row->blocks[8]); | |
578 | 12720 | ctx->idsp.idct_put(dest_u + dct_y_offset + dct_x_offset, dct_linesize_chroma, row->blocks[9]); | |
579 | 12720 | ctx->idsp.idct_put(dest_v, dct_linesize_chroma, row->blocks[4]); | |
580 | 12720 | ctx->idsp.idct_put(dest_v + dct_x_offset, dct_linesize_chroma, row->blocks[5]); | |
581 | 12720 | ctx->idsp.idct_put(dest_v + dct_y_offset, dct_linesize_chroma, row->blocks[10]); | |
582 | 12720 | ctx->idsp.idct_put(dest_v + dct_y_offset + dct_x_offset, dct_linesize_chroma, row->blocks[11]); | |
583 | } | ||
584 | } | ||
585 | |||
586 | 4336632 | return 0; | |
587 | } | ||
588 | |||
589 | 30173 | static int dnxhd_decode_row(AVCodecContext *avctx, void *data, | |
590 | int rownb, int threadnb) | ||
591 | { | ||
592 | 30173 | const DNXHDContext *ctx = avctx->priv_data; | |
593 | 30173 | uint32_t offset = ctx->mb_scan_index[rownb]; | |
594 | 30173 | RowContext *row = ctx->rows + threadnb; | |
595 | int x, ret; | ||
596 | |||
597 | 30173 | row->last_dc[0] = | |
598 | 30173 | row->last_dc[1] = | |
599 | 30173 | row->last_dc[2] = 1 << (ctx->bit_depth + 2); // for levels +2^(bitdepth-1) | |
600 | 30173 | ret = init_get_bits8(&row->gb, ctx->buf + offset, ctx->buf_size - offset); | |
601 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30173 times.
|
30173 | if (ret < 0) { |
602 | ✗ | row->errors++; | |
603 | ✗ | return ret; | |
604 | } | ||
605 |
2/2✓ Branch 0 taken 4336632 times.
✓ Branch 1 taken 30173 times.
|
4366805 | for (x = 0; x < ctx->mb_width; x++) { |
606 | 4336632 | int ret = dnxhd_decode_macroblock(ctx, row, data, x, rownb); | |
607 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4336632 times.
|
4336632 | if (ret < 0) { |
608 | ✗ | row->errors++; | |
609 | ✗ | return ret; | |
610 | } | ||
611 | } | ||
612 | |||
613 | 30173 | return 0; | |
614 | } | ||
615 | |||
616 | 435 | static int dnxhd_decode_frame(AVCodecContext *avctx, AVFrame *picture, | |
617 | int *got_frame, AVPacket *avpkt) | ||
618 | { | ||
619 | 435 | const uint8_t *buf = avpkt->data; | |
620 | 435 | int buf_size = avpkt->size; | |
621 | 435 | DNXHDContext *ctx = avctx->priv_data; | |
622 | 435 | int first_field = 1; | |
623 | int ret, i; | ||
624 | |||
625 | ff_dlog(avctx, "frame size %d\n", buf_size); | ||
626 | |||
627 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 435 times.
|
870 | for (i = 0; i < avctx->thread_count; i++) |
628 | 435 | ctx->rows[i].format = -1; | |
629 | |||
630 | 435 | decode_coding_unit: | |
631 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 502 times.
|
502 | if ((ret = dnxhd_decode_header(ctx, picture, buf, buf_size, first_field)) < 0) |
632 | ✗ | return ret; | |
633 | |||
634 |
3/4✓ Branch 0 taken 46 times.
✓ Branch 1 taken 456 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 46 times.
|
502 | if ((avctx->width || avctx->height) && |
635 |
2/4✓ Branch 0 taken 456 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 456 times.
|
456 | (ctx->width != avctx->width || ctx->height != avctx->height)) { |
636 | ✗ | av_log(avctx, AV_LOG_WARNING, "frame size changed: %dx%d -> %ux%u\n", | |
637 | avctx->width, avctx->height, ctx->width, ctx->height); | ||
638 | ✗ | first_field = 1; | |
639 | } | ||
640 |
3/4✓ Branch 0 taken 424 times.
✓ Branch 1 taken 78 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 424 times.
|
502 | if (avctx->pix_fmt != AV_PIX_FMT_NONE && avctx->pix_fmt != ctx->pix_fmt) { |
641 | ✗ | av_log(avctx, AV_LOG_WARNING, "pix_fmt changed: %s -> %s\n", | |
642 | av_get_pix_fmt_name(avctx->pix_fmt), av_get_pix_fmt_name(ctx->pix_fmt)); | ||
643 | ✗ | first_field = 1; | |
644 | } | ||
645 | |||
646 | 502 | avctx->pix_fmt = ctx->pix_fmt; | |
647 | 502 | ret = ff_set_dimensions(avctx, ctx->width, ctx->height); | |
648 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ret < 0) |
649 | ✗ | return ret; | |
650 | |||
651 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 67 times.
|
502 | if (first_field) { |
652 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 435 times.
|
435 | if ((ret = ff_thread_get_buffer(avctx, picture, 0)) < 0) |
653 | ✗ | return ret; | |
654 | 435 | picture->pict_type = AV_PICTURE_TYPE_I; | |
655 | 435 | picture->key_frame = 1; | |
656 | } | ||
657 | |||
658 | 502 | ctx->buf_size = buf_size - ctx->data_offset; | |
659 | 502 | ctx->buf = buf + ctx->data_offset; | |
660 | 502 | avctx->execute2(avctx, dnxhd_decode_row, picture, NULL, ctx->mb_height); | |
661 | |||
662 |
4/4✓ Branch 0 taken 435 times.
✓ Branch 1 taken 67 times.
✓ Branch 2 taken 67 times.
✓ Branch 3 taken 368 times.
|
502 | if (first_field && picture->interlaced_frame) { |
663 | 67 | buf += ctx->cid_table->coding_unit_size; | |
664 | 67 | buf_size -= ctx->cid_table->coding_unit_size; | |
665 | 67 | first_field = 0; | |
666 | 67 | goto decode_coding_unit; | |
667 | } | ||
668 | |||
669 | 435 | ret = 0; | |
670 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 435 times.
|
870 | for (i = 0; i < avctx->thread_count; i++) { |
671 | 435 | ret += ctx->rows[i].errors; | |
672 | 435 | ctx->rows[i].errors = 0; | |
673 | } | ||
674 | |||
675 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 433 times.
|
435 | if (ctx->act) { |
676 | static int act_warned; | ||
677 | 2 | int format = ctx->rows[0].format; | |
678 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | for (i = 1; i < avctx->thread_count; i++) { |
679 | ✗ | if (ctx->rows[i].format != format && | |
680 | ✗ | ctx->rows[i].format != -1 /* not run */) { | |
681 | ✗ | format = 2; | |
682 | ✗ | break; | |
683 | } | ||
684 | } | ||
685 |
1/4✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
2 | switch (format) { |
686 | ✗ | case -1: | |
687 | case 2: | ||
688 | ✗ | if (!act_warned) { | |
689 | ✗ | act_warned = 1; | |
690 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
691 | "Unsupported: variable ACT flag.\n"); | ||
692 | } | ||
693 | ✗ | break; | |
694 | ✗ | case 0: | |
695 | ✗ | ctx->pix_fmt = ctx->bit_depth==10 | |
696 | ✗ | ? AV_PIX_FMT_GBRP10 : AV_PIX_FMT_GBRP12; | |
697 | ✗ | break; | |
698 | 2 | case 1: | |
699 | 4 | ctx->pix_fmt = ctx->bit_depth==10 | |
700 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | ? AV_PIX_FMT_YUV444P10 : AV_PIX_FMT_YUV444P12; |
701 | 2 | break; | |
702 | } | ||
703 | } | ||
704 | 435 | avctx->pix_fmt = ctx->pix_fmt; | |
705 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 435 times.
|
435 | if (ret) { |
706 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "%d lines with errors\n", ret); | |
707 | ✗ | return AVERROR_INVALIDDATA; | |
708 | } | ||
709 | |||
710 | 435 | *got_frame = 1; | |
711 | 435 | return avpkt->size; | |
712 | } | ||
713 | |||
714 | 149 | static av_cold int dnxhd_decode_close(AVCodecContext *avctx) | |
715 | { | ||
716 | 149 | DNXHDContext *ctx = avctx->priv_data; | |
717 | |||
718 | 149 | ff_free_vlc(&ctx->ac_vlc); | |
719 | 149 | ff_free_vlc(&ctx->dc_vlc); | |
720 | 149 | ff_free_vlc(&ctx->run_vlc); | |
721 | |||
722 | 149 | av_freep(&ctx->rows); | |
723 | |||
724 | 149 | return 0; | |
725 | } | ||
726 | |||
727 | const FFCodec ff_dnxhd_decoder = { | ||
728 | .p.name = "dnxhd", | ||
729 | CODEC_LONG_NAME("VC3/DNxHD"), | ||
730 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
731 | .p.id = AV_CODEC_ID_DNXHD, | ||
732 | .priv_data_size = sizeof(DNXHDContext), | ||
733 | .init = dnxhd_decode_init, | ||
734 | .close = dnxhd_decode_close, | ||
735 | FF_CODEC_DECODE_CB(dnxhd_decode_frame), | ||
736 | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS | | ||
737 | AV_CODEC_CAP_SLICE_THREADS, | ||
738 | .p.profiles = NULL_IF_CONFIG_SMALL(ff_dnxhd_profiles), | ||
739 | }; | ||
740 |