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_vlc_free(&ctx->ac_vlc); | |
134 | 148 | ff_vlc_free(&ctx->dc_vlc); | |
135 | 148 | ff_vlc_free(&ctx->run_vlc); | |
136 | |||
137 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 148 times.
|
148 | if ((ret = vlc_init(&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 = vlc_init(&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 0 not taken.
✓ Branch 1 taken 148 times.
|
148 | if ((ret = ff_vlc_init_sparse(&ctx->run_vlc, DNXHD_VLC_BITS, 62, |
146 | 148 | ctx->cid_table->run_bits, 1, 1, | |
147 | 148 | ctx->cid_table->run_codes, 2, 2, | |
148 | 148 | ctx->cid_table->run, 1, 1, 0)) < 0) | |
149 | ✗ | goto out; | |
150 | |||
151 | 148 | ctx->cid = cid; | |
152 | } | ||
153 | 502 | ret = 0; | |
154 | 502 | out: | |
155 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ret < 0) |
156 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "vlc_init failed\n"); | |
157 | 502 | return ret; | |
158 | } | ||
159 | |||
160 | 502 | static int dnxhd_get_profile(int cid) | |
161 | { | ||
162 |
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) { |
163 | 2 | case 1270: | |
164 | 2 | return AV_PROFILE_DNXHR_444; | |
165 | 2 | case 1271: | |
166 | 2 | return AV_PROFILE_DNXHR_HQX; | |
167 | 120 | case 1272: | |
168 | 120 | return AV_PROFILE_DNXHR_HQ; | |
169 | 48 | case 1273: | |
170 | 48 | return AV_PROFILE_DNXHR_SQ; | |
171 | 111 | case 1274: | |
172 | 111 | return AV_PROFILE_DNXHR_LB; | |
173 | } | ||
174 | 219 | return AV_PROFILE_DNXHD; | |
175 | } | ||
176 | |||
177 | 502 | static int dnxhd_decode_header(DNXHDContext *ctx, AVFrame *frame, | |
178 | const uint8_t *buf, int buf_size, | ||
179 | int first_field) | ||
180 | { | ||
181 | int i, cid, ret; | ||
182 | 502 | int old_bit_depth = ctx->bit_depth, bitdepth; | |
183 | uint64_t header_prefix; | ||
184 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < 0x280) { |
185 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
186 | "buffer too small (%d < 640).\n", buf_size); | ||
187 | ✗ | return AVERROR_INVALIDDATA; | |
188 | } | ||
189 | |||
190 | 502 | header_prefix = ff_dnxhd_parse_header_prefix(buf); | |
191 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (header_prefix == 0) { |
192 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
193 | "unknown header 0x%02X 0x%02X 0x%02X 0x%02X 0x%02X\n", | ||
194 | ✗ | buf[0], buf[1], buf[2], buf[3], buf[4]); | |
195 | ✗ | return AVERROR_INVALIDDATA; | |
196 | } | ||
197 |
2/2✓ Branch 0 taken 134 times.
✓ Branch 1 taken 368 times.
|
502 | if (buf[5] & 2) { /* interlaced */ |
198 |
2/2✓ Branch 0 taken 67 times.
✓ Branch 1 taken 67 times.
|
134 | ctx->cur_field = first_field ? buf[5] & 1 : !ctx->cur_field; |
199 | 134 | frame->flags |= AV_FRAME_FLAG_INTERLACED; | |
200 |
1/2✓ Branch 0 taken 134 times.
✗ Branch 1 not taken.
|
134 | if (first_field ^ ctx->cur_field) |
201 | 134 | frame->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST; | |
202 | 134 | av_log(ctx->avctx, AV_LOG_DEBUG, | |
203 | 134 | "interlaced %d, cur field %d\n", buf[5] & 3, ctx->cur_field); | |
204 | } else { | ||
205 | 368 | ctx->cur_field = 0; | |
206 | } | ||
207 | 502 | ctx->mbaff = (buf[0x6] >> 5) & 1; | |
208 | 502 | ctx->alpha = buf[0x7] & 1; | |
209 | 502 | ctx->lla = (buf[0x7] >> 1) & 1; | |
210 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ctx->alpha) |
211 | ✗ | avpriv_request_sample(ctx->avctx, "alpha"); | |
212 | |||
213 | 502 | ctx->height = AV_RB16(buf + 0x18); | |
214 | 502 | ctx->width = AV_RB16(buf + 0x1a); | |
215 | |||
216 |
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) { |
217 | 432 | case 1: bitdepth = 8; break; | |
218 | 68 | case 2: bitdepth = 10; break; | |
219 | 2 | case 3: bitdepth = 12; break; | |
220 | ✗ | default: | |
221 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
222 | ✗ | "Unknown bitdepth indicator (%d)\n", buf[0x21] >> 5); | |
223 | ✗ | return AVERROR_INVALIDDATA; | |
224 | } | ||
225 | |||
226 | 502 | cid = AV_RB32(buf + 0x28); | |
227 | |||
228 | 502 | ctx->avctx->profile = dnxhd_get_profile(cid); | |
229 | |||
230 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 502 times.
|
502 | if ((ret = dnxhd_init_vlc(ctx, cid, bitdepth)) < 0) |
231 | ✗ | return ret; | |
232 |
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) |
233 | ✗ | av_log(ctx->avctx, AV_LOG_WARNING, | |
234 | "Adaptive MB interlace flag in an unsupported profile.\n"); | ||
235 | |||
236 |
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) { |
237 | 502 | case 0: frame->colorspace = AVCOL_SPC_BT709; break; | |
238 | ✗ | case 1: frame->colorspace = AVCOL_SPC_BT2020_NCL; break; | |
239 | ✗ | case 2: frame->colorspace = AVCOL_SPC_BT2020_CL; break; | |
240 | ✗ | case 3: frame->colorspace = AVCOL_SPC_UNSPECIFIED; break; | |
241 | } | ||
242 | |||
243 | 502 | ctx->act = buf[0x2C] & 1; | |
244 |
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) |
245 | ✗ | av_log(ctx->avctx, AV_LOG_WARNING, | |
246 | "Adaptive color transform in an unsupported profile.\n"); | ||
247 | |||
248 | 502 | ctx->is_444 = (buf[0x2C] >> 6) & 1; | |
249 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
|
502 | if (ctx->is_444) { |
250 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (bitdepth == 8) { |
251 | ✗ | avpriv_request_sample(ctx->avctx, "4:4:4 8 bits"); | |
252 | ✗ | return AVERROR_INVALIDDATA; | |
253 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | } else if (bitdepth == 10) { |
254 | 2 | ctx->decode_dct_block = dnxhd_decode_dct_block_10_444; | |
255 | 2 | ctx->pix_fmt = ctx->act ? AV_PIX_FMT_YUV444P10 | |
256 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | : AV_PIX_FMT_GBRP10; |
257 | } else { | ||
258 | ✗ | ctx->decode_dct_block = dnxhd_decode_dct_block_12_444; | |
259 | ✗ | ctx->pix_fmt = ctx->act ? AV_PIX_FMT_YUV444P12 | |
260 | ✗ | : AV_PIX_FMT_GBRP12; | |
261 | } | ||
262 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 498 times.
|
500 | } else if (bitdepth == 12) { |
263 | 2 | ctx->decode_dct_block = dnxhd_decode_dct_block_12; | |
264 | 2 | ctx->pix_fmt = AV_PIX_FMT_YUV422P12; | |
265 |
2/2✓ Branch 0 taken 66 times.
✓ Branch 1 taken 432 times.
|
498 | } else if (bitdepth == 10) { |
266 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
|
66 | if (ctx->avctx->profile == AV_PROFILE_DNXHR_HQX) |
267 | ✗ | ctx->decode_dct_block = dnxhd_decode_dct_block_10_444; | |
268 | else | ||
269 | 66 | ctx->decode_dct_block = dnxhd_decode_dct_block_10; | |
270 | 66 | ctx->pix_fmt = AV_PIX_FMT_YUV422P10; | |
271 | } else { | ||
272 | 432 | ctx->decode_dct_block = dnxhd_decode_dct_block_8; | |
273 | 432 | ctx->pix_fmt = AV_PIX_FMT_YUV422P; | |
274 | } | ||
275 | |||
276 | 502 | ctx->avctx->bits_per_raw_sample = ctx->bit_depth = bitdepth; | |
277 |
2/2✓ Branch 0 taken 148 times.
✓ Branch 1 taken 354 times.
|
502 | if (ctx->bit_depth != old_bit_depth) { |
278 | 148 | ff_blockdsp_init(&ctx->bdsp); | |
279 | 148 | ff_idctdsp_init(&ctx->idsp, ctx->avctx); | |
280 | 148 | ff_permute_scantable(ctx->permutated_scantable, ff_zigzag_direct, | |
281 | 148 | ctx->idsp.idct_permutation); | |
282 | } | ||
283 | |||
284 | // make sure profile size constraints are respected | ||
285 | // DNx100 allows 1920->1440 and 1280->960 subsampling | ||
286 |
2/2✓ Branch 0 taken 285 times.
✓ Branch 1 taken 217 times.
|
502 | if (ctx->width != ctx->cid_table->width && |
287 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 283 times.
|
285 | ctx->cid_table->width != DNXHD_VARIABLE) { |
288 | 2 | av_reduce(&ctx->avctx->sample_aspect_ratio.num, | |
289 | 2 | &ctx->avctx->sample_aspect_ratio.den, | |
290 | 2 | ctx->width, ctx->cid_table->width, 255); | |
291 | 2 | ctx->width = ctx->cid_table->width; | |
292 | } | ||
293 | |||
294 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < ctx->cid_table->coding_unit_size) { |
295 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "incorrect frame size (%d < %u).\n", | |
296 | ✗ | buf_size, ctx->cid_table->coding_unit_size); | |
297 | ✗ | return AVERROR_INVALIDDATA; | |
298 | } | ||
299 | |||
300 | 502 | ctx->mb_width = (ctx->width + 15)>> 4; | |
301 | 502 | ctx->mb_height = AV_RB16(buf + 0x16c); | |
302 | |||
303 |
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->flags & AV_FRAME_FLAG_INTERLACED)) |
304 | 134 | ctx->height <<= 1; | |
305 | |||
306 | 502 | av_log(ctx->avctx, AV_LOG_VERBOSE, "%dx%d, 4:%s %d bits, MBAFF=%d ACT=%d\n", | |
307 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 500 times.
|
502 | ctx->width, ctx->height, ctx->is_444 ? "4:4" : "2:2", |
308 | ctx->bit_depth, ctx->mbaff, ctx->act); | ||
309 | |||
310 | // Newer format supports variable mb_scan_index sizes | ||
311 |
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)) { |
312 | 108 | ctx->data_offset = 0x170 + (ctx->mb_height << 2); | |
313 | } else { | ||
314 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 394 times.
|
394 | if (ctx->mb_height > 68) { |
315 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
316 | "mb height too big: %d\n", ctx->mb_height); | ||
317 | ✗ | return AVERROR_INVALIDDATA; | |
318 | } | ||
319 | 394 | ctx->data_offset = 0x280; | |
320 | } | ||
321 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if ((ctx->mb_height << !!(frame->flags & AV_FRAME_FLAG_INTERLACED)) > (ctx->height + 15) >> 4) { |
322 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
323 | "mb height too big: %d\n", ctx->mb_height); | ||
324 | ✗ | return AVERROR_INVALIDDATA; | |
325 | } | ||
326 | |||
327 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (buf_size < ctx->data_offset) { |
328 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
329 | "buffer too small (%d < %d).\n", buf_size, ctx->data_offset); | ||
330 | ✗ | return AVERROR_INVALIDDATA; | |
331 | } | ||
332 | |||
333 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ctx->mb_height > FF_ARRAY_ELEMS(ctx->mb_scan_index)) { |
334 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
335 | "mb_height too big (%d > %"SIZE_SPECIFIER").\n", ctx->mb_height, FF_ARRAY_ELEMS(ctx->mb_scan_index)); | ||
336 | ✗ | return AVERROR_INVALIDDATA; | |
337 | } | ||
338 | |||
339 |
2/2✓ Branch 0 taken 30173 times.
✓ Branch 1 taken 502 times.
|
30675 | for (i = 0; i < ctx->mb_height; i++) { |
340 | 30173 | ctx->mb_scan_index[i] = AV_RB32(buf + 0x170 + (i << 2)); | |
341 | ff_dlog(ctx->avctx, "mb scan index %d, pos %d: %"PRIu32"\n", | ||
342 | i, 0x170 + (i << 2), ctx->mb_scan_index[i]); | ||
343 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30173 times.
|
30173 | if (buf_size - ctx->data_offset < ctx->mb_scan_index[i]) { |
344 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
345 | "invalid mb scan index (%"PRIu32" vs %u).\n", | ||
346 | ✗ | ctx->mb_scan_index[i], buf_size - ctx->data_offset); | |
347 | ✗ | return AVERROR_INVALIDDATA; | |
348 | } | ||
349 | } | ||
350 | |||
351 | 502 | return 0; | |
352 | } | ||
353 | |||
354 | 34743936 | static av_always_inline int dnxhd_decode_dct_block(const DNXHDContext *ctx, | |
355 | RowContext *row, | ||
356 | int n, | ||
357 | int index_bits, | ||
358 | int level_bias, | ||
359 | int level_shift, | ||
360 | int dc_shift) | ||
361 | { | ||
362 | int i, j, index1, len, flags; | ||
363 | int level, component, sign; | ||
364 | const int *scale; | ||
365 | const uint8_t *weight_matrix; | ||
366 | 34743936 | const uint8_t *ac_info = ctx->cid_table->ac_info; | |
367 | 34743936 | int16_t *block = row->blocks[n]; | |
368 | 34743936 | const int eob_index = ctx->cid_table->eob_index; | |
369 | 34743936 | int ret = 0; | |
370 | 34743936 | OPEN_READER(bs, &row->gb); | |
371 | |||
372 | 34743936 | ctx->bdsp.clear_block(block); | |
373 | |||
374 |
2/2✓ Branch 0 taken 34591296 times.
✓ Branch 1 taken 152640 times.
|
34743936 | if (!ctx->is_444) { |
375 |
2/2✓ Branch 0 taken 17295648 times.
✓ Branch 1 taken 17295648 times.
|
34591296 | if (n & 2) { |
376 | 17295648 | component = 1 + (n & 1); | |
377 | 17295648 | scale = row->chroma_scale; | |
378 | 17295648 | weight_matrix = ctx->cid_table->chroma_weight; | |
379 | } else { | ||
380 | 17295648 | component = 0; | |
381 | 17295648 | scale = row->luma_scale; | |
382 | 17295648 | weight_matrix = ctx->cid_table->luma_weight; | |
383 | } | ||
384 | } else { | ||
385 | 152640 | component = (n >> 1) % 3; | |
386 |
2/2✓ Branch 0 taken 101760 times.
✓ Branch 1 taken 50880 times.
|
152640 | if (component) { |
387 | 101760 | scale = row->chroma_scale; | |
388 | 101760 | weight_matrix = ctx->cid_table->chroma_weight; | |
389 | } else { | ||
390 | 50880 | scale = row->luma_scale; | |
391 | 50880 | weight_matrix = ctx->cid_table->luma_weight; | |
392 | } | ||
393 | } | ||
394 | |||
395 | 34743936 | UPDATE_CACHE(bs, &row->gb); | |
396 | 34743936 | GET_VLC(len, bs, &row->gb, ctx->dc_vlc.table, DNXHD_DC_VLC_BITS, 1); | |
397 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 34743936 times.
|
34743936 | if (len < 0) { |
398 | ✗ | ret = len; | |
399 | ✗ | goto error; | |
400 | } | ||
401 |
2/2✓ Branch 0 taken 32365764 times.
✓ Branch 1 taken 2378172 times.
|
34743936 | if (len) { |
402 | 32365764 | level = GET_CACHE(bs, &row->gb); | |
403 | 32365764 | LAST_SKIP_BITS(bs, &row->gb, len); | |
404 | 32365764 | sign = ~level >> 31; | |
405 | 32365764 | level = (NEG_USR32(sign ^ level, len) ^ sign) - sign; | |
406 | 32365764 | row->last_dc[component] += level * (1 << dc_shift); | |
407 | } | ||
408 | 34743936 | block[0] = row->last_dc[component]; | |
409 | |||
410 | 34743936 | i = 0; | |
411 | |||
412 | 34743936 | UPDATE_CACHE(bs, &row->gb); | |
413 |
2/2✓ Branch 1 taken 4524829 times.
✓ Branch 2 taken 30219107 times.
|
34743936 | GET_VLC(index1, bs, &row->gb, ctx->ac_vlc.table, |
414 | DNXHD_VLC_BITS, 2); | ||
415 | |||
416 |
2/2✓ Branch 0 taken 142194813 times.
✓ Branch 1 taken 34743936 times.
|
176938749 | while (index1 != eob_index) { |
417 | 142194813 | level = ac_info[2*index1+0]; | |
418 | 142194813 | flags = ac_info[2*index1+1]; | |
419 | |||
420 | 142194813 | sign = SHOW_SBITS(bs, &row->gb, 1); | |
421 | 142194813 | SKIP_BITS(bs, &row->gb, 1); | |
422 | |||
423 |
2/2✓ Branch 0 taken 2609257 times.
✓ Branch 1 taken 139585556 times.
|
142194813 | if (flags & 1) { |
424 | 2609257 | level += SHOW_UBITS(bs, &row->gb, index_bits) << 7; | |
425 | 2609257 | SKIP_BITS(bs, &row->gb, index_bits); | |
426 | } | ||
427 | |||
428 |
2/2✓ Branch 0 taken 46521239 times.
✓ Branch 1 taken 95673574 times.
|
142194813 | if (flags & 2) { |
429 | int run; | ||
430 | 46521239 | UPDATE_CACHE(bs, &row->gb); | |
431 |
2/2✓ Branch 1 taken 2125192 times.
✓ Branch 2 taken 44396047 times.
|
46521239 | GET_VLC(run, bs, &row->gb, ctx->run_vlc.table, |
432 | DNXHD_VLC_BITS, 2); | ||
433 | 46521239 | i += run; | |
434 | } | ||
435 | |||
436 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 142194813 times.
|
142194813 | if (++i > 63) { |
437 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "ac tex damaged %d, %d\n", n, i); | |
438 | ✗ | ret = -1; | |
439 | ✗ | break; | |
440 | } | ||
441 | |||
442 | 142194813 | j = ctx->permutated_scantable[i]; | |
443 | 142194813 | level *= scale[i]; | |
444 | 142194813 | level += scale[i] >> 1; | |
445 |
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) |
446 | 117255751 | level += level_bias; // 1<<(level_shift-1) | |
447 | 142194813 | level >>= level_shift; | |
448 | |||
449 | 142194813 | block[j] = (level ^ sign) - sign; | |
450 | |||
451 | 142194813 | UPDATE_CACHE(bs, &row->gb); | |
452 |
2/2✓ Branch 1 taken 10195080 times.
✓ Branch 2 taken 131999733 times.
|
142194813 | GET_VLC(index1, bs, &row->gb, ctx->ac_vlc.table, |
453 | DNXHD_VLC_BITS, 2); | ||
454 | } | ||
455 | 34743936 | error: | |
456 | 34743936 | CLOSE_READER(bs, &row->gb); | |
457 | 34743936 | return ret; | |
458 | } | ||
459 | |||
460 | 32375616 | static int dnxhd_decode_dct_block_8(const DNXHDContext *ctx, | |
461 | RowContext *row, int n) | ||
462 | { | ||
463 | 32375616 | return dnxhd_decode_dct_block(ctx, row, n, 4, 32, 6, 0); | |
464 | } | ||
465 | |||
466 | 2085120 | static int dnxhd_decode_dct_block_10(const DNXHDContext *ctx, | |
467 | RowContext *row, int n) | ||
468 | { | ||
469 | 2085120 | return dnxhd_decode_dct_block(ctx, row, n, 6, 8, 4, 0); | |
470 | } | ||
471 | |||
472 | 152640 | static int dnxhd_decode_dct_block_10_444(const DNXHDContext *ctx, | |
473 | RowContext *row, int n) | ||
474 | { | ||
475 | 152640 | return dnxhd_decode_dct_block(ctx, row, n, 6, 32, 6, 0); | |
476 | } | ||
477 | |||
478 | 130560 | static int dnxhd_decode_dct_block_12(const DNXHDContext *ctx, | |
479 | RowContext *row, int n) | ||
480 | { | ||
481 | 130560 | return dnxhd_decode_dct_block(ctx, row, n, 6, 8, 4, 2); | |
482 | } | ||
483 | |||
484 | ✗ | static int dnxhd_decode_dct_block_12_444(const DNXHDContext *ctx, | |
485 | RowContext *row, int n) | ||
486 | { | ||
487 | ✗ | return dnxhd_decode_dct_block(ctx, row, n, 6, 32, 4, 2); | |
488 | } | ||
489 | |||
490 | 4336632 | static int dnxhd_decode_macroblock(const DNXHDContext *ctx, RowContext *row, | |
491 | AVFrame *frame, int x, int y) | ||
492 | { | ||
493 | 4336632 | int shift1 = ctx->bit_depth >= 10; | |
494 | 4336632 | int dct_linesize_luma = frame->linesize[0]; | |
495 | 4336632 | int dct_linesize_chroma = frame->linesize[1]; | |
496 | uint8_t *dest_y, *dest_u, *dest_v; | ||
497 | int dct_y_offset, dct_x_offset; | ||
498 | int qscale, i, act; | ||
499 | 4336632 | int interlaced_mb = 0; | |
500 | |||
501 |
2/2✓ Branch 0 taken 12240 times.
✓ Branch 1 taken 4324392 times.
|
4336632 | if (ctx->mbaff) { |
502 | 12240 | interlaced_mb = get_bits1(&row->gb); | |
503 | 12240 | qscale = get_bits(&row->gb, 10); | |
504 | } else { | ||
505 | 4324392 | qscale = get_bits(&row->gb, 11); | |
506 | } | ||
507 | 4336632 | act = get_bits1(&row->gb); | |
508 |
2/2✓ Branch 0 taken 12720 times.
✓ Branch 1 taken 4323912 times.
|
4336632 | if (act) { |
509 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12720 times.
|
12720 | if (!ctx->act) { |
510 | static int act_warned; | ||
511 | ✗ | if (!act_warned) { | |
512 | ✗ | act_warned = 1; | |
513 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
514 | "ACT flag set, in violation of frame header.\n"); | ||
515 | } | ||
516 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 12718 times.
|
12720 | } else if (row->format == -1) { |
517 | 2 | row->format = act; | |
518 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12718 times.
|
12718 | } else if (row->format != act) { |
519 | ✗ | row->format = 2; // Variable | |
520 | } | ||
521 | } | ||
522 | |||
523 |
2/2✓ Branch 0 taken 354747 times.
✓ Branch 1 taken 3981885 times.
|
4336632 | if (qscale != row->last_qscale) { |
524 |
2/2✓ Branch 0 taken 22703808 times.
✓ Branch 1 taken 354747 times.
|
23058555 | for (i = 0; i < 64; i++) { |
525 | 22703808 | row->luma_scale[i] = qscale * ctx->cid_table->luma_weight[i]; | |
526 | 22703808 | row->chroma_scale[i] = qscale * ctx->cid_table->chroma_weight[i]; | |
527 | } | ||
528 | 354747 | row->last_qscale = qscale; | |
529 | } | ||
530 | |||
531 |
2/2✓ Branch 0 taken 34743936 times.
✓ Branch 1 taken 4336632 times.
|
39080568 | for (i = 0; i < 8 + 4 * ctx->is_444; i++) { |
532 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 34743936 times.
|
34743936 | if (ctx->decode_dct_block(ctx, row, i) < 0) |
533 | ✗ | return AVERROR_INVALIDDATA; | |
534 | } | ||
535 | |||
536 |
2/2✓ Branch 0 taken 546720 times.
✓ Branch 1 taken 3789912 times.
|
4336632 | if (frame->flags & AV_FRAME_FLAG_INTERLACED) { |
537 | 546720 | dct_linesize_luma <<= 1; | |
538 | 546720 | dct_linesize_chroma <<= 1; | |
539 | } | ||
540 | |||
541 | 4336632 | dest_y = frame->data[0] + ((y * dct_linesize_luma) << 4) + (x << (4 + shift1)); | |
542 | 4336632 | dest_u = frame->data[1] + ((y * dct_linesize_chroma) << 4) + (x << (3 + shift1 + ctx->is_444)); | |
543 | 4336632 | dest_v = frame->data[2] + ((y * dct_linesize_chroma) << 4) + (x << (3 + shift1 + ctx->is_444)); | |
544 | |||
545 |
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->flags & AV_FRAME_FLAG_INTERLACED) && ctx->cur_field) { |
546 | 273360 | dest_y += frame->linesize[0]; | |
547 | 273360 | dest_u += frame->linesize[1]; | |
548 | 273360 | dest_v += frame->linesize[2]; | |
549 | } | ||
550 |
2/2✓ Branch 0 taken 3224 times.
✓ Branch 1 taken 4333408 times.
|
4336632 | if (interlaced_mb) { |
551 | 3224 | dct_linesize_luma <<= 1; | |
552 | 3224 | dct_linesize_chroma <<= 1; | |
553 | } | ||
554 | |||
555 |
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); |
556 | 4336632 | dct_x_offset = 8 << shift1; | |
557 |
2/2✓ Branch 0 taken 4323912 times.
✓ Branch 1 taken 12720 times.
|
4336632 | if (!ctx->is_444) { |
558 | 4323912 | ctx->idsp.idct_put(dest_y, dct_linesize_luma, row->blocks[0]); | |
559 | 4323912 | ctx->idsp.idct_put(dest_y + dct_x_offset, dct_linesize_luma, row->blocks[1]); | |
560 | 4323912 | ctx->idsp.idct_put(dest_y + dct_y_offset, dct_linesize_luma, row->blocks[4]); | |
561 | 4323912 | ctx->idsp.idct_put(dest_y + dct_y_offset + dct_x_offset, dct_linesize_luma, row->blocks[5]); | |
562 | |||
563 |
1/2✓ Branch 0 taken 4323912 times.
✗ Branch 1 not taken.
|
4323912 | if (!(ctx->avctx->flags & AV_CODEC_FLAG_GRAY)) { |
564 |
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); |
565 | 4323912 | ctx->idsp.idct_put(dest_u, dct_linesize_chroma, row->blocks[2]); | |
566 | 4323912 | ctx->idsp.idct_put(dest_v, dct_linesize_chroma, row->blocks[3]); | |
567 | 4323912 | ctx->idsp.idct_put(dest_u + dct_y_offset, dct_linesize_chroma, row->blocks[6]); | |
568 | 4323912 | ctx->idsp.idct_put(dest_v + dct_y_offset, dct_linesize_chroma, row->blocks[7]); | |
569 | } | ||
570 | } else { | ||
571 | 12720 | ctx->idsp.idct_put(dest_y, dct_linesize_luma, row->blocks[0]); | |
572 | 12720 | ctx->idsp.idct_put(dest_y + dct_x_offset, dct_linesize_luma, row->blocks[1]); | |
573 | 12720 | ctx->idsp.idct_put(dest_y + dct_y_offset, dct_linesize_luma, row->blocks[6]); | |
574 | 12720 | ctx->idsp.idct_put(dest_y + dct_y_offset + dct_x_offset, dct_linesize_luma, row->blocks[7]); | |
575 | |||
576 |
1/2✓ Branch 0 taken 12720 times.
✗ Branch 1 not taken.
|
12720 | if (!(ctx->avctx->flags & AV_CODEC_FLAG_GRAY)) { |
577 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12720 times.
|
12720 | dct_y_offset = interlaced_mb ? frame->linesize[1] : (dct_linesize_chroma << 3); |
578 | 12720 | ctx->idsp.idct_put(dest_u, dct_linesize_chroma, row->blocks[2]); | |
579 | 12720 | ctx->idsp.idct_put(dest_u + dct_x_offset, dct_linesize_chroma, row->blocks[3]); | |
580 | 12720 | ctx->idsp.idct_put(dest_u + dct_y_offset, dct_linesize_chroma, row->blocks[8]); | |
581 | 12720 | ctx->idsp.idct_put(dest_u + dct_y_offset + dct_x_offset, dct_linesize_chroma, row->blocks[9]); | |
582 | 12720 | ctx->idsp.idct_put(dest_v, dct_linesize_chroma, row->blocks[4]); | |
583 | 12720 | ctx->idsp.idct_put(dest_v + dct_x_offset, dct_linesize_chroma, row->blocks[5]); | |
584 | 12720 | ctx->idsp.idct_put(dest_v + dct_y_offset, dct_linesize_chroma, row->blocks[10]); | |
585 | 12720 | ctx->idsp.idct_put(dest_v + dct_y_offset + dct_x_offset, dct_linesize_chroma, row->blocks[11]); | |
586 | } | ||
587 | } | ||
588 | |||
589 | 4336632 | return 0; | |
590 | } | ||
591 | |||
592 | 30173 | static int dnxhd_decode_row(AVCodecContext *avctx, void *data, | |
593 | int rownb, int threadnb) | ||
594 | { | ||
595 | 30173 | const DNXHDContext *ctx = avctx->priv_data; | |
596 | 30173 | uint32_t offset = ctx->mb_scan_index[rownb]; | |
597 | 30173 | RowContext *row = ctx->rows + threadnb; | |
598 | int x, ret; | ||
599 | |||
600 | 30173 | row->last_dc[0] = | |
601 | 30173 | row->last_dc[1] = | |
602 | 30173 | row->last_dc[2] = 1 << (ctx->bit_depth + 2); // for levels +2^(bitdepth-1) | |
603 | 30173 | ret = init_get_bits8(&row->gb, ctx->buf + offset, ctx->buf_size - offset); | |
604 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30173 times.
|
30173 | if (ret < 0) { |
605 | ✗ | row->errors++; | |
606 | ✗ | return ret; | |
607 | } | ||
608 |
2/2✓ Branch 0 taken 4336632 times.
✓ Branch 1 taken 30173 times.
|
4366805 | for (x = 0; x < ctx->mb_width; x++) { |
609 | 4336632 | int ret = dnxhd_decode_macroblock(ctx, row, data, x, rownb); | |
610 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4336632 times.
|
4336632 | if (ret < 0) { |
611 | ✗ | row->errors++; | |
612 | ✗ | return ret; | |
613 | } | ||
614 | } | ||
615 | |||
616 | 30173 | return 0; | |
617 | } | ||
618 | |||
619 | 435 | static int dnxhd_decode_frame(AVCodecContext *avctx, AVFrame *picture, | |
620 | int *got_frame, AVPacket *avpkt) | ||
621 | { | ||
622 | 435 | const uint8_t *buf = avpkt->data; | |
623 | 435 | int buf_size = avpkt->size; | |
624 | 435 | DNXHDContext *ctx = avctx->priv_data; | |
625 | 435 | int first_field = 1; | |
626 | int ret, i; | ||
627 | |||
628 | ff_dlog(avctx, "frame size %d\n", buf_size); | ||
629 | |||
630 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 435 times.
|
870 | for (i = 0; i < avctx->thread_count; i++) |
631 | 435 | ctx->rows[i].format = -1; | |
632 | |||
633 | 435 | decode_coding_unit: | |
634 |
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) |
635 | ✗ | return ret; | |
636 | |||
637 |
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) && |
638 |
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)) { |
639 | ✗ | av_log(avctx, AV_LOG_WARNING, "frame size changed: %dx%d -> %ux%u\n", | |
640 | avctx->width, avctx->height, ctx->width, ctx->height); | ||
641 | ✗ | first_field = 1; | |
642 | } | ||
643 |
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) { |
644 | ✗ | av_log(avctx, AV_LOG_WARNING, "pix_fmt changed: %s -> %s\n", | |
645 | av_get_pix_fmt_name(avctx->pix_fmt), av_get_pix_fmt_name(ctx->pix_fmt)); | ||
646 | ✗ | first_field = 1; | |
647 | } | ||
648 | |||
649 | 502 | avctx->pix_fmt = ctx->pix_fmt; | |
650 | 502 | ret = ff_set_dimensions(avctx, ctx->width, ctx->height); | |
651 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 502 times.
|
502 | if (ret < 0) |
652 | ✗ | return ret; | |
653 | |||
654 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 67 times.
|
502 | if (first_field) { |
655 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 435 times.
|
435 | if ((ret = ff_thread_get_buffer(avctx, picture, 0)) < 0) |
656 | ✗ | return ret; | |
657 | 435 | picture->pict_type = AV_PICTURE_TYPE_I; | |
658 | 435 | picture->flags |= AV_FRAME_FLAG_KEY; | |
659 | } | ||
660 | |||
661 | 502 | ctx->buf_size = buf_size - ctx->data_offset; | |
662 | 502 | ctx->buf = buf + ctx->data_offset; | |
663 | 502 | avctx->execute2(avctx, dnxhd_decode_row, picture, NULL, ctx->mb_height); | |
664 | |||
665 |
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->flags & AV_FRAME_FLAG_INTERLACED)) { |
666 | 67 | buf += ctx->cid_table->coding_unit_size; | |
667 | 67 | buf_size -= ctx->cid_table->coding_unit_size; | |
668 | 67 | first_field = 0; | |
669 | 67 | goto decode_coding_unit; | |
670 | } | ||
671 | |||
672 | 435 | ret = 0; | |
673 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 435 times.
|
870 | for (i = 0; i < avctx->thread_count; i++) { |
674 | 435 | ret += ctx->rows[i].errors; | |
675 | 435 | ctx->rows[i].errors = 0; | |
676 | } | ||
677 | |||
678 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 433 times.
|
435 | if (ctx->act) { |
679 | static int act_warned; | ||
680 | 2 | int format = ctx->rows[0].format; | |
681 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | for (i = 1; i < avctx->thread_count; i++) { |
682 | ✗ | if (ctx->rows[i].format != format && | |
683 | ✗ | ctx->rows[i].format != -1 /* not run */) { | |
684 | ✗ | format = 2; | |
685 | ✗ | break; | |
686 | } | ||
687 | } | ||
688 |
1/4✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
2 | switch (format) { |
689 | ✗ | case -1: | |
690 | case 2: | ||
691 | ✗ | if (!act_warned) { | |
692 | ✗ | act_warned = 1; | |
693 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, | |
694 | "Unsupported: variable ACT flag.\n"); | ||
695 | } | ||
696 | ✗ | break; | |
697 | ✗ | case 0: | |
698 | ✗ | ctx->pix_fmt = ctx->bit_depth==10 | |
699 | ✗ | ? AV_PIX_FMT_GBRP10 : AV_PIX_FMT_GBRP12; | |
700 | ✗ | break; | |
701 | 2 | case 1: | |
702 | 4 | ctx->pix_fmt = ctx->bit_depth==10 | |
703 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | ? AV_PIX_FMT_YUV444P10 : AV_PIX_FMT_YUV444P12; |
704 | 2 | break; | |
705 | } | ||
706 | } | ||
707 | 435 | avctx->pix_fmt = ctx->pix_fmt; | |
708 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 435 times.
|
435 | if (ret) { |
709 | ✗ | av_log(ctx->avctx, AV_LOG_ERROR, "%d lines with errors\n", ret); | |
710 | ✗ | return AVERROR_INVALIDDATA; | |
711 | } | ||
712 | |||
713 | 435 | *got_frame = 1; | |
714 | 435 | return avpkt->size; | |
715 | } | ||
716 | |||
717 | 149 | static av_cold int dnxhd_decode_close(AVCodecContext *avctx) | |
718 | { | ||
719 | 149 | DNXHDContext *ctx = avctx->priv_data; | |
720 | |||
721 | 149 | ff_vlc_free(&ctx->ac_vlc); | |
722 | 149 | ff_vlc_free(&ctx->dc_vlc); | |
723 | 149 | ff_vlc_free(&ctx->run_vlc); | |
724 | |||
725 | 149 | av_freep(&ctx->rows); | |
726 | |||
727 | 149 | return 0; | |
728 | } | ||
729 | |||
730 | const FFCodec ff_dnxhd_decoder = { | ||
731 | .p.name = "dnxhd", | ||
732 | CODEC_LONG_NAME("VC3/DNxHD"), | ||
733 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
734 | .p.id = AV_CODEC_ID_DNXHD, | ||
735 | .priv_data_size = sizeof(DNXHDContext), | ||
736 | .init = dnxhd_decode_init, | ||
737 | .close = dnxhd_decode_close, | ||
738 | FF_CODEC_DECODE_CB(dnxhd_decode_frame), | ||
739 | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS | | ||
740 | AV_CODEC_CAP_SLICE_THREADS, | ||
741 | .p.profiles = NULL_IF_CONFIG_SMALL(ff_dnxhd_profiles), | ||
742 | }; | ||
743 |