Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * This file is part of FFmpeg. | ||
3 | * | ||
4 | * FFmpeg is free software; you can redistribute it and/or | ||
5 | * modify it under the terms of the GNU Lesser General Public | ||
6 | * License as published by the Free Software Foundation; either | ||
7 | * version 2.1 of the License, or (at your option) any later version. | ||
8 | * | ||
9 | * FFmpeg is distributed in the hope that it will be useful, | ||
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
12 | * Lesser General Public License for more details. | ||
13 | * | ||
14 | * You should have received a copy of the GNU Lesser General Public | ||
15 | * License along with FFmpeg; if not, write to the Free Software | ||
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
17 | */ | ||
18 | |||
19 | #include "libavutil/attributes.h" | ||
20 | #include "libavutil/avassert.h" | ||
21 | #include "libavutil/mem.h" | ||
22 | |||
23 | #include "bytestream.h" | ||
24 | #include "cbs.h" | ||
25 | #include "cbs_internal.h" | ||
26 | #include "cbs_h264.h" | ||
27 | #include "cbs_h265.h" | ||
28 | #include "cbs_h266.h" | ||
29 | #include "h264.h" | ||
30 | #include "h2645_parse.h" | ||
31 | #include "refstruct.h" | ||
32 | #include "vvc.h" | ||
33 | |||
34 | #include "hevc/hevc.h" | ||
35 | |||
36 | 301599 | static int cbs_read_ue_golomb(CodedBitstreamContext *ctx, GetBitContext *gbc, | |
37 | const char *name, const int *subscripts, | ||
38 | uint32_t *write_to, | ||
39 | uint32_t range_min, uint32_t range_max) | ||
40 | { | ||
41 | uint32_t leading_bits, value; | ||
42 | int max_length, leading_zeroes; | ||
43 | |||
44 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 301599 times.
|
301599 | CBS_TRACE_READ_START(); |
45 | |||
46 |
2/2✓ Branch 1 taken 10804 times.
✓ Branch 2 taken 290795 times.
|
301599 | max_length = FFMIN(get_bits_left(gbc), 32); |
47 | |||
48 |
1/2✓ Branch 0 taken 301599 times.
✗ Branch 1 not taken.
|
301599 | leading_bits = max_length ? show_bits_long(gbc, max_length) : 0; |
49 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 301599 times.
|
301599 | if (leading_bits == 0) { |
50 | ✗ | if (max_length >= 32) { | |
51 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid ue-golomb code at " | |
52 | "%s: more than 31 zeroes.\n", name); | ||
53 | ✗ | return AVERROR_INVALIDDATA; | |
54 | } else { | ||
55 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid ue-golomb code at " | |
56 | "%s: bitstream ended.\n", name); | ||
57 | ✗ | return AVERROR_INVALIDDATA; | |
58 | } | ||
59 | } | ||
60 | |||
61 | 301599 | leading_zeroes = max_length - 1 - av_log2(leading_bits); | |
62 | 301599 | skip_bits_long(gbc, leading_zeroes); | |
63 | |||
64 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 301599 times.
|
301599 | if (get_bits_left(gbc) < leading_zeroes + 1) { |
65 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid ue-golomb code at " | |
66 | "%s: bitstream ended.\n", name); | ||
67 | ✗ | return AVERROR_INVALIDDATA; | |
68 | } | ||
69 | |||
70 | 301599 | value = get_bits_long(gbc, leading_zeroes + 1) - 1; | |
71 | |||
72 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 301599 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
301599 | CBS_TRACE_READ_END(); |
73 | |||
74 |
2/4✓ Branch 0 taken 301599 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 301599 times.
|
301599 | if (value < range_min || value > range_max) { |
75 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: " | |
76 | "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n", | ||
77 | name, value, range_min, range_max); | ||
78 | ✗ | return AVERROR_INVALIDDATA; | |
79 | } | ||
80 | |||
81 | 301599 | *write_to = value; | |
82 | 301599 | return 0; | |
83 | } | ||
84 | |||
85 | 97980 | static int cbs_read_se_golomb(CodedBitstreamContext *ctx, GetBitContext *gbc, | |
86 | const char *name, const int *subscripts, | ||
87 | int32_t *write_to, | ||
88 | int32_t range_min, int32_t range_max) | ||
89 | { | ||
90 | uint32_t leading_bits, unsigned_value; | ||
91 | int max_length, leading_zeroes; | ||
92 | int32_t value; | ||
93 | |||
94 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 97980 times.
|
97980 | CBS_TRACE_READ_START(); |
95 | |||
96 |
2/2✓ Branch 1 taken 8452 times.
✓ Branch 2 taken 89528 times.
|
97980 | max_length = FFMIN(get_bits_left(gbc), 32); |
97 | |||
98 |
1/2✓ Branch 0 taken 97980 times.
✗ Branch 1 not taken.
|
97980 | leading_bits = max_length ? show_bits_long(gbc, max_length) : 0; |
99 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 97980 times.
|
97980 | if (leading_bits == 0) { |
100 | ✗ | if (max_length >= 32) { | |
101 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid se-golomb code at " | |
102 | "%s: more than 31 zeroes.\n", name); | ||
103 | ✗ | return AVERROR_INVALIDDATA; | |
104 | } else { | ||
105 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid se-golomb code at " | |
106 | "%s: bitstream ended.\n", name); | ||
107 | ✗ | return AVERROR_INVALIDDATA; | |
108 | } | ||
109 | } | ||
110 | |||
111 | 97980 | leading_zeroes = max_length - 1 - av_log2(leading_bits); | |
112 | 97980 | skip_bits_long(gbc, leading_zeroes); | |
113 | |||
114 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 97980 times.
|
97980 | if (get_bits_left(gbc) < leading_zeroes + 1) { |
115 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid se-golomb code at " | |
116 | "%s: bitstream ended.\n", name); | ||
117 | ✗ | return AVERROR_INVALIDDATA; | |
118 | } | ||
119 | |||
120 | 97980 | unsigned_value = get_bits_long(gbc, leading_zeroes + 1); | |
121 | |||
122 |
2/2✓ Branch 0 taken 53740 times.
✓ Branch 1 taken 44240 times.
|
97980 | if (unsigned_value & 1) |
123 | 53740 | value = -(int32_t)(unsigned_value / 2); | |
124 | else | ||
125 | 44240 | value = unsigned_value / 2; | |
126 | |||
127 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 97980 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
97980 | CBS_TRACE_READ_END(); |
128 | |||
129 |
2/4✓ Branch 0 taken 97980 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 97980 times.
|
97980 | if (value < range_min || value > range_max) { |
130 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: " | |
131 | "%"PRId32", but must be in [%"PRId32",%"PRId32"].\n", | ||
132 | name, value, range_min, range_max); | ||
133 | ✗ | return AVERROR_INVALIDDATA; | |
134 | } | ||
135 | |||
136 | 97980 | *write_to = value; | |
137 | 97980 | return 0; | |
138 | } | ||
139 | |||
140 | 154844 | static int cbs_write_ue_golomb(CodedBitstreamContext *ctx, PutBitContext *pbc, | |
141 | const char *name, const int *subscripts, | ||
142 | uint32_t value, | ||
143 | uint32_t range_min, uint32_t range_max) | ||
144 | { | ||
145 | int len; | ||
146 | |||
147 |
2/2✓ Branch 0 taken 151775 times.
✓ Branch 1 taken 3069 times.
|
154844 | CBS_TRACE_WRITE_START(); |
148 | |||
149 |
2/4✓ Branch 0 taken 154844 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 154844 times.
|
154844 | if (value < range_min || value > range_max) { |
150 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: " | |
151 | "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n", | ||
152 | name, value, range_min, range_max); | ||
153 | ✗ | return AVERROR_INVALIDDATA; | |
154 | } | ||
155 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 154844 times.
|
154844 | av_assert0(value != UINT32_MAX); |
156 | |||
157 | 154844 | len = av_log2(value + 1); | |
158 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 154844 times.
|
154844 | if (put_bits_left(pbc) < 2 * len + 1) |
159 | ✗ | return AVERROR(ENOSPC); | |
160 | |||
161 | 154844 | put_bits(pbc, len, 0); | |
162 |
1/2✓ Branch 0 taken 154844 times.
✗ Branch 1 not taken.
|
154844 | if (len + 1 < 32) |
163 | 154844 | put_bits(pbc, len + 1, value + 1); | |
164 | else | ||
165 | ✗ | put_bits32(pbc, value + 1); | |
166 | |||
167 |
3/4✓ Branch 0 taken 151775 times.
✓ Branch 1 taken 3069 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 151775 times.
|
154844 | CBS_TRACE_WRITE_END(); |
168 | |||
169 | 154844 | return 0; | |
170 | } | ||
171 | |||
172 | 84001 | static int cbs_write_se_golomb(CodedBitstreamContext *ctx, PutBitContext *pbc, | |
173 | const char *name, const int *subscripts, | ||
174 | int32_t value, | ||
175 | int32_t range_min, int32_t range_max) | ||
176 | { | ||
177 | int len; | ||
178 | uint32_t uvalue; | ||
179 | |||
180 |
2/2✓ Branch 0 taken 83431 times.
✓ Branch 1 taken 570 times.
|
84001 | CBS_TRACE_WRITE_START(); |
181 | |||
182 |
2/4✓ Branch 0 taken 84001 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 84001 times.
|
84001 | if (value < range_min || value > range_max) { |
183 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: " | |
184 | "%"PRId32", but must be in [%"PRId32",%"PRId32"].\n", | ||
185 | name, value, range_min, range_max); | ||
186 | ✗ | return AVERROR_INVALIDDATA; | |
187 | } | ||
188 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 84001 times.
|
84001 | av_assert0(value != INT32_MIN); |
189 | |||
190 |
2/2✓ Branch 0 taken 19170 times.
✓ Branch 1 taken 64831 times.
|
84001 | if (value == 0) |
191 | 19170 | uvalue = 0; | |
192 |
2/2✓ Branch 0 taken 36505 times.
✓ Branch 1 taken 28326 times.
|
64831 | else if (value > 0) |
193 | 36505 | uvalue = 2 * (uint32_t)value - 1; | |
194 | else | ||
195 | 28326 | uvalue = 2 * (uint32_t)-value; | |
196 | |||
197 | 84001 | len = av_log2(uvalue + 1); | |
198 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 84001 times.
|
84001 | if (put_bits_left(pbc) < 2 * len + 1) |
199 | ✗ | return AVERROR(ENOSPC); | |
200 | |||
201 | 84001 | put_bits(pbc, len, 0); | |
202 |
1/2✓ Branch 0 taken 84001 times.
✗ Branch 1 not taken.
|
84001 | if (len + 1 < 32) |
203 | 84001 | put_bits(pbc, len + 1, uvalue + 1); | |
204 | else | ||
205 | ✗ | put_bits32(pbc, uvalue + 1); | |
206 | |||
207 |
3/4✓ Branch 0 taken 83431 times.
✓ Branch 1 taken 570 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 83431 times.
|
84001 | CBS_TRACE_WRITE_END(); |
208 | |||
209 | 84001 | return 0; | |
210 | } | ||
211 | |||
212 | // payload_extension_present() - true if we are before the last 1-bit | ||
213 | // in the payload structure, which must be in the last byte. | ||
214 | 5 | static int cbs_h265_payload_extension_present(GetBitContext *gbc, uint32_t payload_size, | |
215 | int cur_pos) | ||
216 | { | ||
217 | 5 | int bits_left = payload_size * 8 - cur_pos; | |
218 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
5 | return (bits_left > 0 && |
219 | ✗ | (bits_left > 7 || show_bits(gbc, bits_left) & MAX_UINT_BITS(bits_left - 1))); | |
220 | } | ||
221 | |||
222 | #define HEADER(name) do { \ | ||
223 | ff_cbs_trace_header(ctx, name); \ | ||
224 | } while (0) | ||
225 | |||
226 | #define CHECK(call) do { \ | ||
227 | err = (call); \ | ||
228 | if (err < 0) \ | ||
229 | return err; \ | ||
230 | } while (0) | ||
231 | |||
232 | #define FUNC_NAME2(rw, codec, name) cbs_ ## codec ## _ ## rw ## _ ## name | ||
233 | #define FUNC_NAME1(rw, codec, name) FUNC_NAME2(rw, codec, name) | ||
234 | #define FUNC_H264(name) FUNC_NAME1(READWRITE, h264, name) | ||
235 | #define FUNC_H265(name) FUNC_NAME1(READWRITE, h265, name) | ||
236 | #define FUNC_H266(name) FUNC_NAME1(READWRITE, h266, name) | ||
237 | #define FUNC_SEI(name) FUNC_NAME1(READWRITE, sei, name) | ||
238 | |||
239 | #define SEI_FUNC(name, args) \ | ||
240 | static int FUNC(name) args; \ | ||
241 | static int FUNC(name ## _internal)(CodedBitstreamContext *ctx, \ | ||
242 | RWContext *rw, void *cur, \ | ||
243 | SEIMessageState *state) \ | ||
244 | { \ | ||
245 | return FUNC(name)(ctx, rw, cur, state); \ | ||
246 | } \ | ||
247 | static int FUNC(name) args | ||
248 | |||
249 | #define SUBSCRIPTS(subs, ...) (subs > 0 ? ((int[subs + 1]){ subs, __VA_ARGS__ }) : NULL) | ||
250 | |||
251 | #define u(width, name, range_min, range_max) \ | ||
252 | xu(width, name, current->name, range_min, range_max, 0, ) | ||
253 | #define flag(name) ub(1, name) | ||
254 | #define ue(name, range_min, range_max) \ | ||
255 | xue(name, current->name, range_min, range_max, 0, ) | ||
256 | #define i(width, name, range_min, range_max) \ | ||
257 | xi(width, name, current->name, range_min, range_max, 0, ) | ||
258 | #define ib(width, name) \ | ||
259 | xi(width, name, current->name, MIN_INT_BITS(width), MAX_INT_BITS(width), 0, ) | ||
260 | #define se(name, range_min, range_max) \ | ||
261 | xse(name, current->name, range_min, range_max, 0, ) | ||
262 | |||
263 | #define us(width, name, range_min, range_max, subs, ...) \ | ||
264 | xu(width, name, current->name, range_min, range_max, subs, __VA_ARGS__) | ||
265 | #define ubs(width, name, subs, ...) \ | ||
266 | xu(width, name, current->name, 0, MAX_UINT_BITS(width), subs, __VA_ARGS__) | ||
267 | #define flags(name, subs, ...) \ | ||
268 | xu(1, name, current->name, 0, 1, subs, __VA_ARGS__) | ||
269 | #define ues(name, range_min, range_max, subs, ...) \ | ||
270 | xue(name, current->name, range_min, range_max, subs, __VA_ARGS__) | ||
271 | #define is(width, name, range_min, range_max, subs, ...) \ | ||
272 | xi(width, name, current->name, range_min, range_max, subs, __VA_ARGS__) | ||
273 | #define ibs(width, name, subs, ...) \ | ||
274 | xi(width, name, current->name, MIN_INT_BITS(width), MAX_INT_BITS(width), subs, __VA_ARGS__) | ||
275 | #define ses(name, range_min, range_max, subs, ...) \ | ||
276 | xse(name, current->name, range_min, range_max, subs, __VA_ARGS__) | ||
277 | |||
278 | #define fixed(width, name, value) do { \ | ||
279 | av_unused uint32_t fixed_value = value; \ | ||
280 | xu(width, name, fixed_value, value, value, 0, ); \ | ||
281 | } while (0) | ||
282 | |||
283 | |||
284 | #define READ | ||
285 | #define READWRITE read | ||
286 | #define RWContext GetBitContext | ||
287 | |||
288 | #define ub(width, name) do { \ | ||
289 | uint32_t value; \ | ||
290 | CHECK(ff_cbs_read_simple_unsigned(ctx, rw, width, #name, \ | ||
291 | &value)); \ | ||
292 | current->name = value; \ | ||
293 | } while (0) | ||
294 | #define xu(width, name, var, range_min, range_max, subs, ...) do { \ | ||
295 | uint32_t value; \ | ||
296 | CHECK(ff_cbs_read_unsigned(ctx, rw, width, #name, \ | ||
297 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
298 | &value, range_min, range_max)); \ | ||
299 | var = value; \ | ||
300 | } while (0) | ||
301 | #define xue(name, var, range_min, range_max, subs, ...) do { \ | ||
302 | uint32_t value; \ | ||
303 | CHECK(cbs_read_ue_golomb(ctx, rw, #name, \ | ||
304 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
305 | &value, range_min, range_max)); \ | ||
306 | var = value; \ | ||
307 | } while (0) | ||
308 | #define xi(width, name, var, range_min, range_max, subs, ...) do { \ | ||
309 | int32_t value; \ | ||
310 | CHECK(ff_cbs_read_signed(ctx, rw, width, #name, \ | ||
311 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
312 | &value, range_min, range_max)); \ | ||
313 | var = value; \ | ||
314 | } while (0) | ||
315 | #define xse(name, var, range_min, range_max, subs, ...) do { \ | ||
316 | int32_t value; \ | ||
317 | CHECK(cbs_read_se_golomb(ctx, rw, #name, \ | ||
318 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
319 | &value, range_min, range_max)); \ | ||
320 | var = value; \ | ||
321 | } while (0) | ||
322 | |||
323 | |||
324 | #define infer(name, value) do { \ | ||
325 | current->name = value; \ | ||
326 | } while (0) | ||
327 | |||
328 | 28840 | static int cbs_h2645_read_more_rbsp_data(GetBitContext *gbc) | |
329 | { | ||
330 | 28840 | int bits_left = get_bits_left(gbc); | |
331 |
2/2✓ Branch 0 taken 21537 times.
✓ Branch 1 taken 7303 times.
|
28840 | if (bits_left > 8) |
332 | 21537 | return 1; | |
333 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7303 times.
|
7303 | if (bits_left == 0) |
334 | ✗ | return 0; | |
335 |
2/2✓ Branch 1 taken 11 times.
✓ Branch 2 taken 7292 times.
|
7303 | if (show_bits(gbc, bits_left) & MAX_UINT_BITS(bits_left - 1)) |
336 | 11 | return 1; | |
337 | 7292 | return 0; | |
338 | } | ||
339 | |||
340 | #define more_rbsp_data(var) ((var) = cbs_h2645_read_more_rbsp_data(rw)) | ||
341 | |||
342 | #define bit_position(rw) (get_bits_count(rw)) | ||
343 | #define byte_alignment(rw) (get_bits_count(rw) % 8) | ||
344 | |||
345 | /* The CBS SEI code uses the refstruct API for the allocation | ||
346 | * of its child buffers. */ | ||
347 | #define allocate(name, size) do { \ | ||
348 | name = ff_refstruct_allocz(size + \ | ||
349 | AV_INPUT_BUFFER_PADDING_SIZE); \ | ||
350 | if (!name) \ | ||
351 | return AVERROR(ENOMEM); \ | ||
352 | } while (0) | ||
353 | |||
354 | #define FUNC(name) FUNC_SEI(name) | ||
355 | #include "cbs_sei_syntax_template.c" | ||
356 | #undef FUNC | ||
357 | |||
358 | #undef allocate | ||
359 | |||
360 | /* The other code uses the refstruct API for the allocation | ||
361 | * of its child buffers. */ | ||
362 | #define allocate(name, size) do { \ | ||
363 | name ## _ref = av_buffer_allocz(size + \ | ||
364 | AV_INPUT_BUFFER_PADDING_SIZE); \ | ||
365 | if (!name ## _ref) \ | ||
366 | return AVERROR(ENOMEM); \ | ||
367 | name = name ## _ref->data; \ | ||
368 | } while (0) | ||
369 | |||
370 | #define FUNC(name) FUNC_H264(name) | ||
371 | #include "cbs_h264_syntax_template.c" | ||
372 | #undef FUNC | ||
373 | |||
374 | #define FUNC(name) FUNC_H265(name) | ||
375 | #include "cbs_h265_syntax_template.c" | ||
376 | #undef FUNC | ||
377 | |||
378 | #define FUNC(name) FUNC_H266(name) | ||
379 | #include "cbs_h266_syntax_template.c" | ||
380 | #undef FUNC | ||
381 | |||
382 | #undef READ | ||
383 | #undef READWRITE | ||
384 | #undef RWContext | ||
385 | #undef ub | ||
386 | #undef xu | ||
387 | #undef xi | ||
388 | #undef xue | ||
389 | #undef xse | ||
390 | #undef infer | ||
391 | #undef more_rbsp_data | ||
392 | #undef bit_position | ||
393 | #undef byte_alignment | ||
394 | #undef allocate | ||
395 | |||
396 | |||
397 | #define WRITE | ||
398 | #define READWRITE write | ||
399 | #define RWContext PutBitContext | ||
400 | |||
401 | #define ub(width, name) do { \ | ||
402 | uint32_t value = current->name; \ | ||
403 | CHECK(ff_cbs_write_simple_unsigned(ctx, rw, width, #name, \ | ||
404 | value)); \ | ||
405 | } while (0) | ||
406 | #define xu(width, name, var, range_min, range_max, subs, ...) do { \ | ||
407 | uint32_t value = var; \ | ||
408 | CHECK(ff_cbs_write_unsigned(ctx, rw, width, #name, \ | ||
409 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
410 | value, range_min, range_max)); \ | ||
411 | } while (0) | ||
412 | #define xue(name, var, range_min, range_max, subs, ...) do { \ | ||
413 | uint32_t value = var; \ | ||
414 | CHECK(cbs_write_ue_golomb(ctx, rw, #name, \ | ||
415 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
416 | value, range_min, range_max)); \ | ||
417 | } while (0) | ||
418 | #define xi(width, name, var, range_min, range_max, subs, ...) do { \ | ||
419 | int32_t value = var; \ | ||
420 | CHECK(ff_cbs_write_signed(ctx, rw, width, #name, \ | ||
421 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
422 | value, range_min, range_max)); \ | ||
423 | } while (0) | ||
424 | #define xse(name, var, range_min, range_max, subs, ...) do { \ | ||
425 | int32_t value = var; \ | ||
426 | CHECK(cbs_write_se_golomb(ctx, rw, #name, \ | ||
427 | SUBSCRIPTS(subs, __VA_ARGS__), \ | ||
428 | value, range_min, range_max)); \ | ||
429 | } while (0) | ||
430 | |||
431 | #define infer(name, value) do { \ | ||
432 | if (current->name != (value)) { \ | ||
433 | av_log(ctx->log_ctx, AV_LOG_ERROR, \ | ||
434 | "%s does not match inferred value: " \ | ||
435 | "%"PRId64", but should be %"PRId64".\n", \ | ||
436 | #name, (int64_t)current->name, (int64_t)(value)); \ | ||
437 | return AVERROR_INVALIDDATA; \ | ||
438 | } \ | ||
439 | } while (0) | ||
440 | |||
441 | #define more_rbsp_data(var) (var) | ||
442 | |||
443 | #define bit_position(rw) (put_bits_count(rw)) | ||
444 | #define byte_alignment(rw) (put_bits_count(rw) % 8) | ||
445 | |||
446 | #define allocate(name, size) do { \ | ||
447 | if (!name) { \ | ||
448 | av_log(ctx->log_ctx, AV_LOG_ERROR, "%s must be set " \ | ||
449 | "for writing.\n", #name); \ | ||
450 | return AVERROR_INVALIDDATA; \ | ||
451 | } \ | ||
452 | } while (0) | ||
453 | |||
454 | #define FUNC(name) FUNC_SEI(name) | ||
455 | #include "cbs_sei_syntax_template.c" | ||
456 | #undef FUNC | ||
457 | |||
458 | #define FUNC(name) FUNC_H264(name) | ||
459 | #include "cbs_h264_syntax_template.c" | ||
460 | #undef FUNC | ||
461 | |||
462 | #define FUNC(name) FUNC_H265(name) | ||
463 | #include "cbs_h265_syntax_template.c" | ||
464 | #undef FUNC | ||
465 | |||
466 | #define FUNC(name) FUNC_H266(name) | ||
467 | #include "cbs_h266_syntax_template.c" | ||
468 | #undef FUNC | ||
469 | |||
470 | #undef WRITE | ||
471 | #undef READWRITE | ||
472 | #undef RWContext | ||
473 | #undef ub | ||
474 | #undef xu | ||
475 | #undef xi | ||
476 | #undef xue | ||
477 | #undef xse | ||
478 | #undef u | ||
479 | #undef i | ||
480 | #undef flag | ||
481 | #undef ue | ||
482 | #undef se | ||
483 | #undef infer | ||
484 | #undef more_rbsp_data | ||
485 | #undef bit_position | ||
486 | #undef byte_alignment | ||
487 | #undef allocate | ||
488 | |||
489 | |||
490 | 11805 | static int cbs_h2645_fragment_add_nals(CodedBitstreamContext *ctx, | |
491 | CodedBitstreamFragment *frag, | ||
492 | const H2645Packet *packet) | ||
493 | { | ||
494 | int err, i; | ||
495 | |||
496 |
2/2✓ Branch 0 taken 36489 times.
✓ Branch 1 taken 11805 times.
|
48294 | for (i = 0; i < packet->nb_nals; i++) { |
497 | 36489 | const H2645NAL *nal = &packet->nals[i]; | |
498 | AVBufferRef *ref; | ||
499 | 36489 | size_t size = nal->size; | |
500 | 36489 | enum AVCodecID codec_id = ctx->codec->codec_id; | |
501 | |||
502 |
3/4✓ Branch 0 taken 9372 times.
✓ Branch 1 taken 27117 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 9372 times.
|
36489 | if (codec_id == AV_CODEC_ID_HEVC && nal->nuh_layer_id > 0 && |
503 | ✗ | (nal->type < HEVC_NAL_VPS || nal->type > HEVC_NAL_PPS)) | |
504 | ✗ | continue; | |
505 | |||
506 | // Remove trailing zeroes. | ||
507 |
3/4✓ Branch 0 taken 44431 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7942 times.
✓ Branch 3 taken 36489 times.
|
44431 | while (size > 0 && nal->data[size - 1] == 0) |
508 | 7942 | --size; | |
509 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 36489 times.
|
36489 | if (size == 0) { |
510 | ✗ | av_log(ctx->log_ctx, AV_LOG_VERBOSE, "Discarding empty 0 NAL unit\n"); | |
511 | ✗ | continue; | |
512 | } | ||
513 | |||
514 | 72978 | ref = (nal->data == nal->raw_data) ? frag->data_ref | |
515 |
2/2✓ Branch 0 taken 35195 times.
✓ Branch 1 taken 1294 times.
|
36489 | : packet->rbsp.rbsp_buffer_ref; |
516 | |||
517 | 36489 | err = ff_cbs_append_unit_data(frag, nal->type, | |
518 | 36489 | (uint8_t*)nal->data, size, ref); | |
519 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 36489 times.
|
36489 | if (err < 0) |
520 | ✗ | return err; | |
521 | } | ||
522 | |||
523 | 11805 | return 0; | |
524 | } | ||
525 | |||
526 | 11794 | static int cbs_h2645_split_fragment(CodedBitstreamContext *ctx, | |
527 | CodedBitstreamFragment *frag, | ||
528 | int header) | ||
529 | { | ||
530 | 11794 | enum AVCodecID codec_id = ctx->codec->codec_id; | |
531 | 11794 | CodedBitstreamH2645Context *priv = ctx->priv_data; | |
532 | GetByteContext gbc; | ||
533 | int err; | ||
534 | |||
535 |
2/4✓ Branch 0 taken 11794 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 11794 times.
|
11794 | av_assert0(frag->data && frag->nb_units == 0); |
536 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11794 times.
|
11794 | if (frag->data_size == 0) |
537 | ✗ | return 0; | |
538 | |||
539 |
6/6✓ Branch 0 taken 156 times.
✓ Branch 1 taken 11638 times.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 146 times.
✓ Branch 4 taken 9 times.
✓ Branch 5 taken 1 times.
|
11803 | if (header && frag->data[0] && codec_id == AV_CODEC_ID_H264) { |
540 | // AVCC header. | ||
541 | size_t size, start, end; | ||
542 | int i, count, version; | ||
543 | |||
544 | 9 | priv->mp4 = 1; | |
545 | |||
546 | 9 | bytestream2_init(&gbc, frag->data, frag->data_size); | |
547 | |||
548 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 9 times.
|
9 | if (bytestream2_get_bytes_left(&gbc) < 6) |
549 | ✗ | return AVERROR_INVALIDDATA; | |
550 | |||
551 | 9 | version = bytestream2_get_byte(&gbc); | |
552 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (version != 1) { |
553 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid AVCC header: " | |
554 | "first byte %u.\n", version); | ||
555 | ✗ | return AVERROR_INVALIDDATA; | |
556 | } | ||
557 | |||
558 | 9 | bytestream2_skip(&gbc, 3); | |
559 | 9 | priv->nal_length_size = (bytestream2_get_byte(&gbc) & 3) + 1; | |
560 | |||
561 | // SPS array. | ||
562 | 9 | count = bytestream2_get_byte(&gbc) & 0x1f; | |
563 | 9 | start = bytestream2_tell(&gbc); | |
564 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 9 times.
|
17 | for (i = 0; i < count; i++) { |
565 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
|
8 | if (bytestream2_get_bytes_left(&gbc) < 2 * (count - i)) |
566 | ✗ | return AVERROR_INVALIDDATA; | |
567 | 8 | size = bytestream2_get_be16(&gbc); | |
568 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
|
8 | if (bytestream2_get_bytes_left(&gbc) < size) |
569 | ✗ | return AVERROR_INVALIDDATA; | |
570 | 8 | bytestream2_skip(&gbc, size); | |
571 | } | ||
572 | 9 | end = bytestream2_tell(&gbc); | |
573 | |||
574 | 9 | err = ff_h2645_packet_split(&priv->read_packet, | |
575 | 9 | frag->data + start, end - start, | |
576 | ctx->log_ctx, 2, AV_CODEC_ID_H264, | ||
577 | H2645_FLAG_IS_NALFF | H2645_FLAG_SMALL_PADDING | H2645_FLAG_USE_REF); | ||
578 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (err < 0) { |
579 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to split AVCC SPS array.\n"); | |
580 | ✗ | return err; | |
581 | } | ||
582 | 9 | err = cbs_h2645_fragment_add_nals(ctx, frag, &priv->read_packet); | |
583 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (err < 0) |
584 | ✗ | return err; | |
585 | |||
586 | // PPS array. | ||
587 | 9 | count = bytestream2_get_byte(&gbc); | |
588 | 9 | start = bytestream2_tell(&gbc); | |
589 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 9 times.
|
17 | for (i = 0; i < count; i++) { |
590 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
|
8 | if (bytestream2_get_bytes_left(&gbc) < 2 * (count - i)) |
591 | ✗ | return AVERROR_INVALIDDATA; | |
592 | 8 | size = bytestream2_get_be16(&gbc); | |
593 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
|
8 | if (bytestream2_get_bytes_left(&gbc) < size) |
594 | ✗ | return AVERROR_INVALIDDATA; | |
595 | 8 | bytestream2_skip(&gbc, size); | |
596 | } | ||
597 | 9 | end = bytestream2_tell(&gbc); | |
598 | |||
599 | 9 | err = ff_h2645_packet_split(&priv->read_packet, | |
600 | 9 | frag->data + start, end - start, | |
601 | ctx->log_ctx, 2, AV_CODEC_ID_H264, | ||
602 | H2645_FLAG_IS_NALFF | H2645_FLAG_SMALL_PADDING | H2645_FLAG_USE_REF); | ||
603 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (err < 0) { |
604 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to split AVCC PPS array.\n"); | |
605 | ✗ | return err; | |
606 | } | ||
607 | 9 | err = cbs_h2645_fragment_add_nals(ctx, frag, &priv->read_packet); | |
608 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (err < 0) |
609 | ✗ | return err; | |
610 | |||
611 |
2/2✓ Branch 1 taken 2 times.
✓ Branch 2 taken 7 times.
|
9 | if (bytestream2_get_bytes_left(&gbc) > 0) { |
612 | 2 | av_log(ctx->log_ctx, AV_LOG_WARNING, "%u bytes left at end of AVCC " | |
613 | "header.\n", bytestream2_get_bytes_left(&gbc)); | ||
614 | } | ||
615 | |||
616 |
5/6✓ Branch 0 taken 147 times.
✓ Branch 1 taken 11638 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 146 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
|
11785 | } else if (header && frag->data[0] && codec_id == AV_CODEC_ID_HEVC) { |
617 | // HVCC header. | ||
618 | size_t size, start, end; | ||
619 | int i, j, nb_arrays, nal_unit_type, nb_nals, version; | ||
620 | |||
621 | ✗ | priv->mp4 = 1; | |
622 | |||
623 | ✗ | bytestream2_init(&gbc, frag->data, frag->data_size); | |
624 | |||
625 | ✗ | if (bytestream2_get_bytes_left(&gbc) < 23) | |
626 | ✗ | return AVERROR_INVALIDDATA; | |
627 | |||
628 | ✗ | version = bytestream2_get_byte(&gbc); | |
629 | ✗ | if (version != 1) { | |
630 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid HVCC header: " | |
631 | "first byte %u.\n", version); | ||
632 | ✗ | return AVERROR_INVALIDDATA; | |
633 | } | ||
634 | |||
635 | ✗ | bytestream2_skip(&gbc, 20); | |
636 | ✗ | priv->nal_length_size = (bytestream2_get_byte(&gbc) & 3) + 1; | |
637 | |||
638 | ✗ | nb_arrays = bytestream2_get_byte(&gbc); | |
639 | ✗ | for (i = 0; i < nb_arrays; i++) { | |
640 | ✗ | nal_unit_type = bytestream2_get_byte(&gbc) & 0x3f; | |
641 | ✗ | nb_nals = bytestream2_get_be16(&gbc); | |
642 | |||
643 | ✗ | start = bytestream2_tell(&gbc); | |
644 | ✗ | for (j = 0; j < nb_nals; j++) { | |
645 | ✗ | if (bytestream2_get_bytes_left(&gbc) < 2) | |
646 | ✗ | return AVERROR_INVALIDDATA; | |
647 | ✗ | size = bytestream2_get_be16(&gbc); | |
648 | ✗ | if (bytestream2_get_bytes_left(&gbc) < size) | |
649 | ✗ | return AVERROR_INVALIDDATA; | |
650 | ✗ | bytestream2_skip(&gbc, size); | |
651 | } | ||
652 | ✗ | end = bytestream2_tell(&gbc); | |
653 | |||
654 | ✗ | err = ff_h2645_packet_split(&priv->read_packet, | |
655 | ✗ | frag->data + start, end - start, | |
656 | ctx->log_ctx, 2, AV_CODEC_ID_HEVC, | ||
657 | H2645_FLAG_IS_NALFF | H2645_FLAG_SMALL_PADDING | H2645_FLAG_USE_REF); | ||
658 | ✗ | if (err < 0) { | |
659 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to split " | |
660 | "HVCC array %d (%d NAL units of type %d).\n", | ||
661 | i, nb_nals, nal_unit_type); | ||
662 | ✗ | return err; | |
663 | } | ||
664 | ✗ | err = cbs_h2645_fragment_add_nals(ctx, frag, &priv->read_packet); | |
665 | ✗ | if (err < 0) | |
666 | ✗ | return err; | |
667 | } | ||
668 | |||
669 |
5/6✓ Branch 0 taken 147 times.
✓ Branch 1 taken 11638 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 146 times.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
|
11786 | } else if(header && frag->data[0] && codec_id == AV_CODEC_ID_VVC) { |
670 | // VVCC header. | ||
671 | int ptl_present_flag, num_arrays; | ||
672 | int b, i, j; | ||
673 | |||
674 | 1 | priv->mp4 = 1; | |
675 | |||
676 | 1 | bytestream2_init(&gbc, frag->data, frag->data_size); | |
677 | |||
678 | 1 | b = bytestream2_get_byte(&gbc); | |
679 | 1 | priv->nal_length_size = ((b >> 1) & 3) + 1; | |
680 | 1 | ptl_present_flag = b & 1; | |
681 | |||
682 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if(ptl_present_flag) { |
683 | int num_sublayers, num_bytes_constraint_info, num_sub_profiles; | ||
684 | 1 | num_sublayers = (bytestream2_get_be16u(&gbc) >> 4) & 7; | |
685 | 1 | bytestream2_skip(&gbc, 1); | |
686 | |||
687 | // begin VvcPTLRecord(num_sublayers); | ||
688 | 1 | num_bytes_constraint_info = bytestream2_get_byte(&gbc) & 0x3f; | |
689 | 1 | bytestream2_skip(&gbc, 2 + num_bytes_constraint_info); | |
690 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if(num_sublayers > 1) { |
691 | ✗ | int count_present_flags = 0; | |
692 | ✗ | b = bytestream2_get_byte(&gbc); | |
693 | ✗ | for(i = num_sublayers - 2; i >= 0; i--) { | |
694 | ✗ | if((b >> (7 - (num_sublayers - 2 - i))) & 0x01) | |
695 | ✗ | count_present_flags++; | |
696 | } | ||
697 | ✗ | bytestream2_skip(&gbc, count_present_flags); | |
698 | } | ||
699 | 1 | num_sub_profiles = bytestream2_get_byte(&gbc); | |
700 | 1 | bytestream2_skip(&gbc, num_sub_profiles * 4); | |
701 | // end VvcPTLRecord(num_sublayers); | ||
702 | |||
703 | 1 | bytestream2_skip(&gbc, 3 * 2); | |
704 | } | ||
705 | |||
706 | 1 | num_arrays = bytestream2_get_byte(&gbc); | |
707 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 1 times.
|
4 | for(j = 0; j < num_arrays; j++) { |
708 | size_t start, end, size; | ||
709 | 3 | int nal_unit_type = bytestream2_get_byte(&gbc) & 0x1f; | |
710 | 3 | unsigned int num_nalus = 1; | |
711 |
2/4✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
✗ Branch 3 not taken.
|
3 | if(nal_unit_type != VVC_DCI_NUT && nal_unit_type != VVC_OPI_NUT) |
712 | 3 | num_nalus = bytestream2_get_be16(&gbc); | |
713 | |||
714 | 3 | start = bytestream2_tell(&gbc); | |
715 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 3 times.
|
6 | for(i = 0; i < num_nalus; i++) { |
716 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
|
3 | if (bytestream2_get_bytes_left(&gbc) < 2) |
717 | ✗ | return AVERROR_INVALIDDATA; | |
718 | 3 | size = bytestream2_get_be16(&gbc); | |
719 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
|
3 | if (bytestream2_get_bytes_left(&gbc) < size) |
720 | ✗ | return AVERROR_INVALIDDATA; | |
721 | 3 | bytestream2_skip(&gbc, size); | |
722 | } | ||
723 | 3 | end = bytestream2_tell(&gbc); | |
724 | |||
725 | 3 | err = ff_h2645_packet_split(&priv->read_packet, | |
726 | 3 | frag->data + start, end - start, | |
727 | ctx->log_ctx, 2, AV_CODEC_ID_VVC, | ||
728 | H2645_FLAG_IS_NALFF | H2645_FLAG_SMALL_PADDING | H2645_FLAG_USE_REF); | ||
729 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (err < 0) { |
730 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to split " | |
731 | "VVCC array %d (%d NAL units of type %d).\n", | ||
732 | i, num_nalus, nal_unit_type); | ||
733 | ✗ | return err; | |
734 | } | ||
735 | 3 | err = cbs_h2645_fragment_add_nals(ctx, frag, &priv->read_packet); | |
736 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (err < 0) |
737 | ✗ | return err; | |
738 | } | ||
739 | } else { | ||
740 |
2/2✓ Branch 0 taken 630 times.
✓ Branch 1 taken 11154 times.
|
11784 | int flags = (H2645_FLAG_IS_NALFF * !!priv->mp4) | H2645_FLAG_SMALL_PADDING | H2645_FLAG_USE_REF; |
741 | // Annex B, or later MP4 with already-known parameters. | ||
742 | |||
743 | 11784 | err = ff_h2645_packet_split(&priv->read_packet, | |
744 | 11784 | frag->data, frag->data_size, | |
745 | ctx->log_ctx, | ||
746 | priv->nal_length_size, | ||
747 | codec_id, flags); | ||
748 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11784 times.
|
11784 | if (err < 0) |
749 | ✗ | return err; | |
750 | |||
751 | 11784 | err = cbs_h2645_fragment_add_nals(ctx, frag, &priv->read_packet); | |
752 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11784 times.
|
11784 | if (err < 0) |
753 | ✗ | return err; | |
754 | } | ||
755 | |||
756 | 11794 | return 0; | |
757 | } | ||
758 | |||
759 | #define cbs_h2645_replace_ps(h26n, ps_name, ps_var, id_element) \ | ||
760 | static int cbs_h26 ## h26n ## _replace_ ## ps_var(CodedBitstreamContext *ctx, \ | ||
761 | CodedBitstreamUnit *unit) \ | ||
762 | { \ | ||
763 | CodedBitstreamH26 ## h26n ## Context *priv = ctx->priv_data; \ | ||
764 | H26 ## h26n ## Raw ## ps_name *ps_var = unit->content; \ | ||
765 | unsigned int id = ps_var->id_element; \ | ||
766 | int err = ff_cbs_make_unit_refcounted(ctx, unit); \ | ||
767 | if (err < 0) \ | ||
768 | return err; \ | ||
769 | if (priv->ps_var[id] == priv->active_ ## ps_var) \ | ||
770 | priv->active_ ## ps_var = NULL ; \ | ||
771 | av_assert0(unit->content_ref); \ | ||
772 | ff_refstruct_replace(&priv->ps_var[id], unit->content_ref); \ | ||
773 | return 0; \ | ||
774 | } | ||
775 | |||
776 |
4/6✗ Branch 1 not taken.
✓ Branch 2 taken 146 times.
✓ Branch 3 taken 106 times.
✓ Branch 4 taken 40 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 146 times.
|
146 | cbs_h2645_replace_ps(4, SPS, sps, seq_parameter_set_id) |
777 |
4/6✗ Branch 1 not taken.
✓ Branch 2 taken 2990 times.
✓ Branch 3 taken 2936 times.
✓ Branch 4 taken 54 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2990 times.
|
2990 | cbs_h2645_replace_ps(4, PPS, pps, pic_parameter_set_id) |
778 |
3/6✗ Branch 1 not taken.
✓ Branch 2 taken 112 times.
✓ Branch 3 taken 112 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 112 times.
|
112 | cbs_h2645_replace_ps(5, VPS, vps, vps_video_parameter_set_id) |
779 |
4/6✗ Branch 1 not taken.
✓ Branch 2 taken 116 times.
✓ Branch 3 taken 114 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 116 times.
|
116 | cbs_h2645_replace_ps(5, SPS, sps, sps_seq_parameter_set_id) |
780 |
4/6✗ Branch 1 not taken.
✓ Branch 2 taken 566 times.
✓ Branch 3 taken 386 times.
✓ Branch 4 taken 180 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 566 times.
|
566 | cbs_h2645_replace_ps(5, PPS, pps, pps_pic_parameter_set_id) |
781 | |||
782 | #define cbs_h266_replace_ps(h26n, ps_name, ps_var, id_element) \ | ||
783 | static int cbs_h26 ## h26n ## _replace_ ## ps_var(CodedBitstreamContext *ctx, \ | ||
784 | CodedBitstreamUnit *unit) \ | ||
785 | { \ | ||
786 | CodedBitstreamH26 ## h26n ## Context *priv = ctx->priv_data; \ | ||
787 | H26 ## h26n ## Raw ## ps_name *ps_var = unit->content; \ | ||
788 | unsigned int id = ps_var->id_element; \ | ||
789 | int err = ff_cbs_make_unit_refcounted(ctx, unit); \ | ||
790 | if (err < 0) \ | ||
791 | return err; \ | ||
792 | av_assert0(unit->content_ref); \ | ||
793 | ff_refstruct_replace(&priv->ps_var[id], unit->content_ref); \ | ||
794 | return 0; \ | ||
795 | } | ||
796 | |||
797 |
2/4✗ Branch 1 not taken.
✓ Branch 2 taken 19 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 19 times.
|
19 | cbs_h266_replace_ps(6, VPS, vps, vps_video_parameter_set_id) |
798 |
2/4✗ Branch 1 not taken.
✓ Branch 2 taken 1501 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1501 times.
|
1501 | cbs_h266_replace_ps(6, PPS, pps, pps_pic_parameter_set_id) |
799 | |||
800 | 1223 | static int cbs_h266_replace_sps(CodedBitstreamContext *ctx, | |
801 | CodedBitstreamUnit *unit) | ||
802 | { | ||
803 | 1223 | CodedBitstreamH266Context *priv = ctx->priv_data; | |
804 | 1223 | H266RawSPS *sps = unit->content; | |
805 | 1223 | unsigned int id = sps->sps_seq_parameter_set_id; | |
806 | 1223 | int err = ff_cbs_make_unit_refcounted(ctx, unit); | |
807 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1223 times.
|
1223 | if (err < 0) |
808 | ✗ | return err; | |
809 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1223 times.
|
1223 | av_assert0(unit->content_ref); |
810 |
4/4✓ Branch 0 taken 1017 times.
✓ Branch 1 taken 206 times.
✓ Branch 2 taken 780 times.
✓ Branch 3 taken 237 times.
|
1223 | if (priv->sps[id] && memcmp(priv->sps[id], unit->content_ref, sizeof(*priv->sps[id]))) { |
811 |
2/2✓ Branch 0 taken 49920 times.
✓ Branch 1 taken 780 times.
|
50700 | for (unsigned int i = 0; i < VVC_MAX_PPS_COUNT; i++) { |
812 |
3/4✓ Branch 0 taken 780 times.
✓ Branch 1 taken 49140 times.
✓ Branch 2 taken 780 times.
✗ Branch 3 not taken.
|
49920 | if (priv->pps[i] && priv->pps[i]->pps_seq_parameter_set_id == id) |
813 | 780 | ff_refstruct_unref(&priv->pps[i]); | |
814 | } | ||
815 | } | ||
816 | 1223 | ff_refstruct_replace(&priv->sps[id], unit->content_ref); | |
817 | 1223 | return 0; | |
818 | } | ||
819 | |||
820 | 8164 | static int cbs_h266_replace_ph(CodedBitstreamContext *ctx, | |
821 | CodedBitstreamUnit *unit, | ||
822 | H266RawPictureHeader *ph) | ||
823 | { | ||
824 | 8164 | CodedBitstreamH266Context *h266 = ctx->priv_data; | |
825 | int err; | ||
826 | |||
827 | 8164 | err = ff_cbs_make_unit_refcounted(ctx, unit); | |
828 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8164 times.
|
8164 | if (err < 0) |
829 | ✗ | return err; | |
830 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8164 times.
|
8164 | av_assert0(unit->content_ref); |
831 | 8164 | ff_refstruct_replace(&h266->ph_ref, unit->content_ref); | |
832 | 8164 | h266->ph = ph; | |
833 | 8164 | return 0; | |
834 | } | ||
835 | |||
836 | 9230 | static int cbs_h264_read_nal_unit(CodedBitstreamContext *ctx, | |
837 | CodedBitstreamUnit *unit) | ||
838 | { | ||
839 | GetBitContext gbc; | ||
840 | int err; | ||
841 | |||
842 | 9230 | err = init_get_bits(&gbc, unit->data, 8 * unit->data_size); | |
843 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9230 times.
|
9230 | if (err < 0) |
844 | ✗ | return err; | |
845 | |||
846 | 9230 | err = ff_cbs_alloc_unit_content(ctx, unit); | |
847 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9230 times.
|
9230 | if (err < 0) |
848 | ✗ | return err; | |
849 | |||
850 |
7/9✓ Branch 0 taken 73 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1495 times.
✓ Branch 3 taken 5973 times.
✓ Branch 4 taken 739 times.
✓ Branch 5 taken 913 times.
✓ Branch 6 taken 34 times.
✓ Branch 7 taken 3 times.
✗ Branch 8 not taken.
|
9230 | switch (unit->type) { |
851 | 73 | case H264_NAL_SPS: | |
852 | { | ||
853 | 73 | H264RawSPS *sps = unit->content; | |
854 | |||
855 | 73 | err = cbs_h264_read_sps(ctx, &gbc, sps); | |
856 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 73 times.
|
73 | if (err < 0) |
857 | ✗ | return err; | |
858 | |||
859 | 73 | err = cbs_h264_replace_sps(ctx, unit); | |
860 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 73 times.
|
73 | if (err < 0) |
861 | ✗ | return err; | |
862 | } | ||
863 | 73 | break; | |
864 | |||
865 | ✗ | case H264_NAL_SPS_EXT: | |
866 | { | ||
867 | ✗ | err = cbs_h264_read_sps_extension(ctx, &gbc, unit->content); | |
868 | ✗ | if (err < 0) | |
869 | ✗ | return err; | |
870 | } | ||
871 | ✗ | break; | |
872 | |||
873 | 1495 | case H264_NAL_PPS: | |
874 | { | ||
875 | 1495 | H264RawPPS *pps = unit->content; | |
876 | |||
877 | 1495 | err = cbs_h264_read_pps(ctx, &gbc, pps); | |
878 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1495 times.
|
1495 | if (err < 0) |
879 | ✗ | return err; | |
880 | |||
881 | 1495 | err = cbs_h264_replace_pps(ctx, unit); | |
882 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1495 times.
|
1495 | if (err < 0) |
883 | ✗ | return err; | |
884 | } | ||
885 | 1495 | break; | |
886 | |||
887 | 5973 | case H264_NAL_SLICE: | |
888 | case H264_NAL_IDR_SLICE: | ||
889 | case H264_NAL_AUXILIARY_SLICE: | ||
890 | { | ||
891 | 5973 | H264RawSlice *slice = unit->content; | |
892 | int pos, len; | ||
893 | |||
894 | 5973 | err = cbs_h264_read_slice_header(ctx, &gbc, &slice->header); | |
895 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5973 times.
|
5973 | if (err < 0) |
896 | ✗ | return err; | |
897 | |||
898 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 5973 times.
|
5973 | if (!cbs_h2645_read_more_rbsp_data(&gbc)) |
899 | ✗ | return AVERROR_INVALIDDATA; | |
900 | |||
901 | 5973 | pos = get_bits_count(&gbc); | |
902 | 5973 | len = unit->data_size; | |
903 | |||
904 | 5973 | slice->data_size = len - pos / 8; | |
905 | 5973 | slice->data_ref = av_buffer_ref(unit->data_ref); | |
906 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5973 times.
|
5973 | if (!slice->data_ref) |
907 | ✗ | return AVERROR(ENOMEM); | |
908 | 5973 | slice->data = unit->data + pos / 8; | |
909 | 5973 | slice->data_bit_start = pos % 8; | |
910 | } | ||
911 | 5973 | break; | |
912 | |||
913 | 739 | case H264_NAL_AUD: | |
914 | { | ||
915 | 739 | err = cbs_h264_read_aud(ctx, &gbc, unit->content); | |
916 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
|
739 | if (err < 0) |
917 | ✗ | return err; | |
918 | } | ||
919 | 739 | break; | |
920 | |||
921 | 913 | case H264_NAL_SEI: | |
922 | { | ||
923 | 913 | err = cbs_h264_read_sei(ctx, &gbc, unit->content); | |
924 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 913 times.
|
913 | if (err < 0) |
925 | ✗ | return err; | |
926 | } | ||
927 | 913 | break; | |
928 | |||
929 | 34 | case H264_NAL_FILLER_DATA: | |
930 | { | ||
931 | 34 | err = cbs_h264_read_filler(ctx, &gbc, unit->content); | |
932 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 33 times.
|
34 | if (err < 0) |
933 | 1 | return err; | |
934 | } | ||
935 | 33 | break; | |
936 | |||
937 | 3 | case H264_NAL_END_SEQUENCE: | |
938 | case H264_NAL_END_STREAM: | ||
939 | { | ||
940 | 3 | err = (unit->type == H264_NAL_END_SEQUENCE ? | |
941 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
|
3 | cbs_h264_read_end_of_sequence : |
942 | 3 | cbs_h264_read_end_of_stream)(ctx, &gbc, unit->content); | |
943 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (err < 0) |
944 | ✗ | return err; | |
945 | } | ||
946 | 3 | break; | |
947 | |||
948 | ✗ | default: | |
949 | ✗ | return AVERROR(ENOSYS); | |
950 | } | ||
951 | |||
952 | 9229 | return 0; | |
953 | } | ||
954 | |||
955 | 9372 | static int cbs_h265_read_nal_unit(CodedBitstreamContext *ctx, | |
956 | CodedBitstreamUnit *unit) | ||
957 | { | ||
958 | GetBitContext gbc; | ||
959 | int err; | ||
960 | |||
961 | 9372 | err = init_get_bits(&gbc, unit->data, 8 * unit->data_size); | |
962 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9372 times.
|
9372 | if (err < 0) |
963 | ✗ | return err; | |
964 | |||
965 | 9372 | err = ff_cbs_alloc_unit_content(ctx, unit); | |
966 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9372 times.
|
9372 | if (err < 0) |
967 | ✗ | return err; | |
968 | |||
969 |
6/7✓ Branch 0 taken 56 times.
✓ Branch 1 taken 58 times.
✓ Branch 2 taken 283 times.
✓ Branch 3 taken 7188 times.
✓ Branch 4 taken 60 times.
✓ Branch 5 taken 1727 times.
✗ Branch 6 not taken.
|
9372 | switch (unit->type) { |
970 | 56 | case HEVC_NAL_VPS: | |
971 | { | ||
972 | 56 | H265RawVPS *vps = unit->content; | |
973 | |||
974 | 56 | err = cbs_h265_read_vps(ctx, &gbc, vps); | |
975 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (err < 0) |
976 | ✗ | return err; | |
977 | |||
978 | 56 | err = cbs_h265_replace_vps(ctx, unit); | |
979 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (err < 0) |
980 | ✗ | return err; | |
981 | } | ||
982 | 56 | break; | |
983 | 58 | case HEVC_NAL_SPS: | |
984 | { | ||
985 | 58 | H265RawSPS *sps = unit->content; | |
986 | |||
987 | 58 | err = cbs_h265_read_sps(ctx, &gbc, sps); | |
988 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 58 times.
|
58 | if (err < 0) |
989 | ✗ | return err; | |
990 | |||
991 | 58 | err = cbs_h265_replace_sps(ctx, unit); | |
992 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 58 times.
|
58 | if (err < 0) |
993 | ✗ | return err; | |
994 | } | ||
995 | 58 | break; | |
996 | |||
997 | 283 | case HEVC_NAL_PPS: | |
998 | { | ||
999 | 283 | H265RawPPS *pps = unit->content; | |
1000 | |||
1001 | 283 | err = cbs_h265_read_pps(ctx, &gbc, pps); | |
1002 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 283 times.
|
283 | if (err < 0) |
1003 | ✗ | return err; | |
1004 | |||
1005 | 283 | err = cbs_h265_replace_pps(ctx, unit); | |
1006 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 283 times.
|
283 | if (err < 0) |
1007 | ✗ | return err; | |
1008 | } | ||
1009 | 283 | break; | |
1010 | |||
1011 | 7188 | case HEVC_NAL_TRAIL_N: | |
1012 | case HEVC_NAL_TRAIL_R: | ||
1013 | case HEVC_NAL_TSA_N: | ||
1014 | case HEVC_NAL_TSA_R: | ||
1015 | case HEVC_NAL_STSA_N: | ||
1016 | case HEVC_NAL_STSA_R: | ||
1017 | case HEVC_NAL_RADL_N: | ||
1018 | case HEVC_NAL_RADL_R: | ||
1019 | case HEVC_NAL_RASL_N: | ||
1020 | case HEVC_NAL_RASL_R: | ||
1021 | case HEVC_NAL_BLA_W_LP: | ||
1022 | case HEVC_NAL_BLA_W_RADL: | ||
1023 | case HEVC_NAL_BLA_N_LP: | ||
1024 | case HEVC_NAL_IDR_W_RADL: | ||
1025 | case HEVC_NAL_IDR_N_LP: | ||
1026 | case HEVC_NAL_CRA_NUT: | ||
1027 | { | ||
1028 | 7188 | H265RawSlice *slice = unit->content; | |
1029 | int pos, len; | ||
1030 | |||
1031 | 7188 | err = cbs_h265_read_slice_segment_header(ctx, &gbc, &slice->header); | |
1032 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7188 times.
|
7188 | if (err < 0) |
1033 | ✗ | return err; | |
1034 | |||
1035 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 7188 times.
|
7188 | if (!cbs_h2645_read_more_rbsp_data(&gbc)) |
1036 | ✗ | return AVERROR_INVALIDDATA; | |
1037 | |||
1038 | 7188 | pos = get_bits_count(&gbc); | |
1039 | 7188 | len = unit->data_size; | |
1040 | |||
1041 | 7188 | slice->data_size = len - pos / 8; | |
1042 | 7188 | slice->data_ref = av_buffer_ref(unit->data_ref); | |
1043 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7188 times.
|
7188 | if (!slice->data_ref) |
1044 | ✗ | return AVERROR(ENOMEM); | |
1045 | 7188 | slice->data = unit->data + pos / 8; | |
1046 | 7188 | slice->data_bit_start = pos % 8; | |
1047 | } | ||
1048 | 7188 | break; | |
1049 | |||
1050 | 60 | case HEVC_NAL_AUD: | |
1051 | { | ||
1052 | 60 | err = cbs_h265_read_aud(ctx, &gbc, unit->content); | |
1053 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
|
60 | if (err < 0) |
1054 | ✗ | return err; | |
1055 | } | ||
1056 | 60 | break; | |
1057 | |||
1058 | 1727 | case HEVC_NAL_SEI_PREFIX: | |
1059 | case HEVC_NAL_SEI_SUFFIX: | ||
1060 | { | ||
1061 | 1727 | err = cbs_h265_read_sei(ctx, &gbc, unit->content, | |
1062 | 1727 | unit->type == HEVC_NAL_SEI_PREFIX); | |
1063 | |||
1064 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1727 times.
|
1727 | if (err < 0) |
1065 | ✗ | return err; | |
1066 | } | ||
1067 | 1727 | break; | |
1068 | |||
1069 | ✗ | default: | |
1070 | ✗ | return AVERROR(ENOSYS); | |
1071 | } | ||
1072 | |||
1073 | 9372 | return 0; | |
1074 | } | ||
1075 | |||
1076 | 14302 | static int cbs_h266_read_nal_unit(CodedBitstreamContext *ctx, | |
1077 | CodedBitstreamUnit *unit) | ||
1078 | { | ||
1079 | GetBitContext gbc; | ||
1080 | int err; | ||
1081 | |||
1082 | 14302 | err = init_get_bits8(&gbc, unit->data, unit->data_size); | |
1083 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14302 times.
|
14302 | if (err < 0) |
1084 | ✗ | return err; | |
1085 | |||
1086 | 14302 | err = ff_cbs_alloc_unit_content(ctx, unit); | |
1087 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14302 times.
|
14302 | if (err < 0) |
1088 | ✗ | return err; | |
1089 | |||
1090 |
10/11✓ Branch 0 taken 4 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 888 times.
✓ Branch 4 taken 1110 times.
✓ Branch 5 taken 543 times.
✓ Branch 6 taken 177 times.
✓ Branch 7 taken 8355 times.
✓ Branch 8 taken 49 times.
✓ Branch 9 taken 3159 times.
✗ Branch 10 not taken.
|
14302 | switch (unit->type) { |
1091 | 4 | case VVC_DCI_NUT: | |
1092 | { | ||
1093 | 4 | err = cbs_h266_read_dci(ctx, &gbc, unit->content); | |
1094 | |||
1095 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (err < 0) |
1096 | ✗ | return err; | |
1097 | } | ||
1098 | 4 | break; | |
1099 | 2 | case VVC_OPI_NUT: | |
1100 | { | ||
1101 | 2 | err = cbs_h266_read_opi(ctx, &gbc, unit->content); | |
1102 | |||
1103 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (err < 0) |
1104 | ✗ | return err; | |
1105 | } | ||
1106 | 2 | break; | |
1107 | 15 | case VVC_VPS_NUT: | |
1108 | { | ||
1109 | 15 | H266RawVPS *vps = unit->content; | |
1110 | |||
1111 | 15 | err = cbs_h266_read_vps(ctx, &gbc, vps); | |
1112 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | if (err < 0) |
1113 | ✗ | return err; | |
1114 | |||
1115 | 15 | err = cbs_h266_replace_vps(ctx, unit); | |
1116 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | if (err < 0) |
1117 | ✗ | return err; | |
1118 | } | ||
1119 | 15 | break; | |
1120 | 888 | case VVC_SPS_NUT: | |
1121 | { | ||
1122 | 888 | H266RawSPS *sps = unit->content; | |
1123 | |||
1124 | 888 | err = cbs_h266_read_sps(ctx, &gbc, sps); | |
1125 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 888 times.
|
888 | if (err < 0) |
1126 | ✗ | return err; | |
1127 | |||
1128 | 888 | err = cbs_h266_replace_sps(ctx, unit); | |
1129 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 888 times.
|
888 | if (err < 0) |
1130 | ✗ | return err; | |
1131 | } | ||
1132 | 888 | break; | |
1133 | |||
1134 | 1110 | case VVC_PPS_NUT: | |
1135 | { | ||
1136 | 1110 | H266RawPPS *pps = unit->content; | |
1137 | |||
1138 | 1110 | err = cbs_h266_read_pps(ctx, &gbc, pps); | |
1139 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1110 times.
|
1110 | if (err < 0) |
1140 | ✗ | return err; | |
1141 | |||
1142 | 1110 | err = cbs_h266_replace_pps(ctx, unit); | |
1143 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1110 times.
|
1110 | if (err < 0) |
1144 | ✗ | return err; | |
1145 | } | ||
1146 | 1110 | break; | |
1147 | |||
1148 | 543 | case VVC_PREFIX_APS_NUT: | |
1149 | case VVC_SUFFIX_APS_NUT: | ||
1150 | { | ||
1151 | 543 | err = cbs_h266_read_aps(ctx, &gbc, unit->content, | |
1152 | 543 | unit->type == VVC_PREFIX_APS_NUT); | |
1153 | |||
1154 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 543 times.
|
543 | if (err < 0) |
1155 | ✗ | return err; | |
1156 | } | ||
1157 | 543 | break; | |
1158 | 177 | case VVC_PH_NUT: | |
1159 | { | ||
1160 | 177 | H266RawPH *ph = unit->content; | |
1161 | 177 | err = cbs_h266_read_ph(ctx, &gbc, ph); | |
1162 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 177 times.
|
177 | if (err < 0) |
1163 | ✗ | return err; | |
1164 | 177 | err = cbs_h266_replace_ph(ctx, unit, &ph->ph_picture_header); | |
1165 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 177 times.
|
177 | if (err < 0) |
1166 | ✗ | return err; | |
1167 | } | ||
1168 | 177 | break; | |
1169 | |||
1170 | 8355 | case VVC_TRAIL_NUT: | |
1171 | case VVC_STSA_NUT: | ||
1172 | case VVC_RADL_NUT: | ||
1173 | case VVC_RASL_NUT: | ||
1174 | case VVC_IDR_W_RADL: | ||
1175 | case VVC_IDR_N_LP: | ||
1176 | case VVC_CRA_NUT: | ||
1177 | case VVC_GDR_NUT: | ||
1178 | { | ||
1179 | 8355 | H266RawSlice *slice = unit->content; | |
1180 | int pos, len; | ||
1181 | |||
1182 | 8355 | err = cbs_h266_read_slice_header(ctx, &gbc, &slice->header); | |
1183 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8355 times.
|
8355 | if (err < 0) |
1184 | ✗ | return err; | |
1185 | |||
1186 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 8355 times.
|
8355 | if (!cbs_h2645_read_more_rbsp_data(&gbc)) |
1187 | ✗ | return AVERROR_INVALIDDATA; | |
1188 | |||
1189 | 8355 | pos = get_bits_count(&gbc); | |
1190 | 8355 | len = unit->data_size; | |
1191 | |||
1192 |
2/2✓ Branch 0 taken 5857 times.
✓ Branch 1 taken 2498 times.
|
8355 | if (slice->header.sh_picture_header_in_slice_header_flag) { |
1193 | 5857 | err = cbs_h266_replace_ph(ctx, unit, &slice->header.sh_picture_header); | |
1194 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5857 times.
|
5857 | if (err < 0) |
1195 | ✗ | return err; | |
1196 | } | ||
1197 | |||
1198 | 8355 | slice->header_size = pos / 8; | |
1199 | 8355 | slice->data_size = len - pos / 8; | |
1200 | 8355 | slice->data_ref = av_buffer_ref(unit->data_ref); | |
1201 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8355 times.
|
8355 | if (!slice->data_ref) |
1202 | ✗ | return AVERROR(ENOMEM); | |
1203 | 8355 | slice->data = unit->data + pos / 8; | |
1204 | 8355 | slice->data_bit_start = pos % 8; | |
1205 | } | ||
1206 | 8355 | break; | |
1207 | |||
1208 | 49 | case VVC_AUD_NUT: | |
1209 | { | ||
1210 | 49 | err = cbs_h266_read_aud(ctx, &gbc, unit->content); | |
1211 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
|
49 | if (err < 0) |
1212 | ✗ | return err; | |
1213 | } | ||
1214 | 49 | break; | |
1215 | |||
1216 | 3159 | case VVC_PREFIX_SEI_NUT: | |
1217 | case VVC_SUFFIX_SEI_NUT: | ||
1218 | { | ||
1219 | 3159 | err = cbs_h266_read_sei(ctx, &gbc, unit->content, | |
1220 | 3159 | unit->type == VVC_PREFIX_SEI_NUT); | |
1221 | |||
1222 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3159 times.
|
3159 | if (err < 0) |
1223 | ✗ | return err; | |
1224 | } | ||
1225 | 3159 | break; | |
1226 | |||
1227 | ✗ | default: | |
1228 | ✗ | return AVERROR(ENOSYS); | |
1229 | } | ||
1230 | 14302 | return 0; | |
1231 | } | ||
1232 | |||
1233 | 14194 | static int cbs_h2645_write_slice_data(CodedBitstreamContext *ctx, | |
1234 | PutBitContext *pbc, const uint8_t *data, | ||
1235 | size_t data_size, int data_bit_start) | ||
1236 | { | ||
1237 | 14194 | size_t rest = data_size - (data_bit_start + 7) / 8; | |
1238 | 14194 | const uint8_t *pos = data + data_bit_start / 8; | |
1239 | |||
1240 |
2/4✓ Branch 0 taken 14194 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 14194 times.
|
14194 | av_assert0(data_bit_start >= 0 && |
1241 | data_size > data_bit_start / 8); | ||
1242 | |||
1243 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 14194 times.
|
14194 | if (data_size * 8 + 8 > put_bits_left(pbc)) |
1244 | ✗ | return AVERROR(ENOSPC); | |
1245 | |||
1246 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 14193 times.
|
14194 | if (!rest) |
1247 | 1 | goto rbsp_stop_one_bit; | |
1248 | |||
1249 | // First copy the remaining bits of the first byte | ||
1250 | // The above check ensures that we do not accidentally | ||
1251 | // copy beyond the rbsp_stop_one_bit. | ||
1252 |
2/2✓ Branch 0 taken 3729 times.
✓ Branch 1 taken 10464 times.
|
14193 | if (data_bit_start % 8) |
1253 | 3729 | put_bits(pbc, 8 - data_bit_start % 8, | |
1254 | 3729 | *pos++ & MAX_UINT_BITS(8 - data_bit_start % 8)); | |
1255 | |||
1256 |
2/2✓ Branch 1 taken 14084 times.
✓ Branch 2 taken 109 times.
|
14193 | if (put_bits_count(pbc) % 8 == 0) { |
1257 | // If the writer is aligned at this point, | ||
1258 | // memcpy can be used to improve performance. | ||
1259 | // This happens normally for CABAC. | ||
1260 | 14084 | flush_put_bits(pbc); | |
1261 | 14084 | memcpy(put_bits_ptr(pbc), pos, rest); | |
1262 | 14084 | skip_put_bytes(pbc, rest); | |
1263 | } else { | ||
1264 | // If not, we have to copy manually. | ||
1265 | // rbsp_stop_one_bit forces us to special-case | ||
1266 | // the last byte. | ||
1267 | uint8_t temp; | ||
1268 | int i; | ||
1269 | |||
1270 |
2/2✓ Branch 0 taken 98762 times.
✓ Branch 1 taken 109 times.
|
98871 | for (; rest > 4; rest -= 4, pos += 4) |
1271 | 98762 | put_bits32(pbc, AV_RB32(pos)); | |
1272 | |||
1273 |
2/2✓ Branch 0 taken 165 times.
✓ Branch 1 taken 109 times.
|
274 | for (; rest > 1; rest--, pos++) |
1274 | 165 | put_bits(pbc, 8, *pos); | |
1275 | |||
1276 | 109 | rbsp_stop_one_bit: | |
1277 |
2/2✓ Branch 0 taken 109 times.
✓ Branch 1 taken 1 times.
|
110 | temp = rest ? *pos : *pos & MAX_UINT_BITS(8 - data_bit_start % 8); |
1278 | |||
1279 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 110 times.
|
110 | av_assert0(temp); |
1280 | 110 | i = ff_ctz(*pos); | |
1281 | 110 | temp = temp >> i; | |
1282 |
2/2✓ Branch 0 taken 109 times.
✓ Branch 1 taken 1 times.
|
110 | i = rest ? (8 - i) : (8 - i - data_bit_start % 8); |
1283 | 110 | put_bits(pbc, i, temp); | |
1284 |
2/2✓ Branch 1 taken 92 times.
✓ Branch 2 taken 18 times.
|
110 | if (put_bits_count(pbc) % 8) |
1285 | 92 | put_bits(pbc, 8 - put_bits_count(pbc) % 8, 0); | |
1286 | } | ||
1287 | |||
1288 | 14194 | return 0; | |
1289 | } | ||
1290 | |||
1291 | 7953 | static int cbs_h264_write_nal_unit(CodedBitstreamContext *ctx, | |
1292 | CodedBitstreamUnit *unit, | ||
1293 | PutBitContext *pbc) | ||
1294 | { | ||
1295 | int err; | ||
1296 | |||
1297 |
7/10✓ Branch 0 taken 73 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1495 times.
✓ Branch 3 taken 5575 times.
✓ Branch 4 taken 302 times.
✓ Branch 5 taken 505 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 1 times.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
7953 | switch (unit->type) { |
1298 | 73 | case H264_NAL_SPS: | |
1299 | { | ||
1300 | 73 | H264RawSPS *sps = unit->content; | |
1301 | |||
1302 | 73 | err = cbs_h264_write_sps(ctx, pbc, sps); | |
1303 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 73 times.
|
73 | if (err < 0) |
1304 | ✗ | return err; | |
1305 | |||
1306 | 73 | err = cbs_h264_replace_sps(ctx, unit); | |
1307 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 73 times.
|
73 | if (err < 0) |
1308 | ✗ | return err; | |
1309 | } | ||
1310 | 73 | break; | |
1311 | |||
1312 | ✗ | case H264_NAL_SPS_EXT: | |
1313 | { | ||
1314 | ✗ | H264RawSPSExtension *sps_ext = unit->content; | |
1315 | |||
1316 | ✗ | err = cbs_h264_write_sps_extension(ctx, pbc, sps_ext); | |
1317 | ✗ | if (err < 0) | |
1318 | ✗ | return err; | |
1319 | } | ||
1320 | ✗ | break; | |
1321 | |||
1322 | 1495 | case H264_NAL_PPS: | |
1323 | { | ||
1324 | 1495 | H264RawPPS *pps = unit->content; | |
1325 | |||
1326 | 1495 | err = cbs_h264_write_pps(ctx, pbc, pps); | |
1327 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1495 times.
|
1495 | if (err < 0) |
1328 | ✗ | return err; | |
1329 | |||
1330 | 1495 | err = cbs_h264_replace_pps(ctx, unit); | |
1331 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1495 times.
|
1495 | if (err < 0) |
1332 | ✗ | return err; | |
1333 | } | ||
1334 | 1495 | break; | |
1335 | |||
1336 | 5575 | case H264_NAL_SLICE: | |
1337 | case H264_NAL_IDR_SLICE: | ||
1338 | case H264_NAL_AUXILIARY_SLICE: | ||
1339 | { | ||
1340 | 5575 | H264RawSlice *slice = unit->content; | |
1341 | |||
1342 | 5575 | err = cbs_h264_write_slice_header(ctx, pbc, &slice->header); | |
1343 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5575 times.
|
5575 | if (err < 0) |
1344 | ✗ | return err; | |
1345 | |||
1346 |
1/2✓ Branch 0 taken 5575 times.
✗ Branch 1 not taken.
|
5575 | if (slice->data) { |
1347 | 5575 | err = cbs_h2645_write_slice_data(ctx, pbc, slice->data, | |
1348 | slice->data_size, | ||
1349 | slice->data_bit_start); | ||
1350 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5575 times.
|
5575 | if (err < 0) |
1351 | ✗ | return err; | |
1352 | } else { | ||
1353 | // No slice data - that was just the header. | ||
1354 | // (Bitstream may be unaligned!) | ||
1355 | } | ||
1356 | } | ||
1357 | 5575 | break; | |
1358 | |||
1359 | 302 | case H264_NAL_AUD: | |
1360 | { | ||
1361 | 302 | err = cbs_h264_write_aud(ctx, pbc, unit->content); | |
1362 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 302 times.
|
302 | if (err < 0) |
1363 | ✗ | return err; | |
1364 | } | ||
1365 | 302 | break; | |
1366 | |||
1367 | 505 | case H264_NAL_SEI: | |
1368 | { | ||
1369 | 505 | err = cbs_h264_write_sei(ctx, pbc, unit->content); | |
1370 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 505 times.
|
505 | if (err < 0) |
1371 | ✗ | return err; | |
1372 | } | ||
1373 | 505 | break; | |
1374 | |||
1375 | ✗ | case H264_NAL_FILLER_DATA: | |
1376 | { | ||
1377 | ✗ | err = cbs_h264_write_filler(ctx, pbc, unit->content); | |
1378 | ✗ | if (err < 0) | |
1379 | ✗ | return err; | |
1380 | } | ||
1381 | ✗ | break; | |
1382 | |||
1383 | 1 | case H264_NAL_END_SEQUENCE: | |
1384 | { | ||
1385 | 1 | err = cbs_h264_write_end_of_sequence(ctx, pbc, unit->content); | |
1386 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (err < 0) |
1387 | ✗ | return err; | |
1388 | } | ||
1389 | 1 | break; | |
1390 | |||
1391 | 2 | case H264_NAL_END_STREAM: | |
1392 | { | ||
1393 | 2 | err = cbs_h264_write_end_of_stream(ctx, pbc, unit->content); | |
1394 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (err < 0) |
1395 | ✗ | return err; | |
1396 | } | ||
1397 | 2 | break; | |
1398 | |||
1399 | ✗ | default: | |
1400 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Write unimplemented for " | |
1401 | "NAL unit type %"PRIu32".\n", unit->type); | ||
1402 | ✗ | return AVERROR_PATCHWELCOME; | |
1403 | } | ||
1404 | |||
1405 | 7953 | return 0; | |
1406 | } | ||
1407 | |||
1408 | 8050 | static int cbs_h265_write_nal_unit(CodedBitstreamContext *ctx, | |
1409 | CodedBitstreamUnit *unit, | ||
1410 | PutBitContext *pbc) | ||
1411 | { | ||
1412 | int err; | ||
1413 | |||
1414 |
6/7✓ Branch 0 taken 56 times.
✓ Branch 1 taken 58 times.
✓ Branch 2 taken 283 times.
✓ Branch 3 taken 6031 times.
✓ Branch 4 taken 60 times.
✓ Branch 5 taken 1562 times.
✗ Branch 6 not taken.
|
8050 | switch (unit->type) { |
1415 | 56 | case HEVC_NAL_VPS: | |
1416 | { | ||
1417 | 56 | H265RawVPS *vps = unit->content; | |
1418 | |||
1419 | 56 | err = cbs_h265_write_vps(ctx, pbc, vps); | |
1420 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (err < 0) |
1421 | ✗ | return err; | |
1422 | |||
1423 | 56 | err = cbs_h265_replace_vps(ctx, unit); | |
1424 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (err < 0) |
1425 | ✗ | return err; | |
1426 | } | ||
1427 | 56 | break; | |
1428 | |||
1429 | 58 | case HEVC_NAL_SPS: | |
1430 | { | ||
1431 | 58 | H265RawSPS *sps = unit->content; | |
1432 | |||
1433 | 58 | err = cbs_h265_write_sps(ctx, pbc, sps); | |
1434 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 58 times.
|
58 | if (err < 0) |
1435 | ✗ | return err; | |
1436 | |||
1437 | 58 | err = cbs_h265_replace_sps(ctx, unit); | |
1438 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 58 times.
|
58 | if (err < 0) |
1439 | ✗ | return err; | |
1440 | } | ||
1441 | 58 | break; | |
1442 | |||
1443 | 283 | case HEVC_NAL_PPS: | |
1444 | { | ||
1445 | 283 | H265RawPPS *pps = unit->content; | |
1446 | |||
1447 | 283 | err = cbs_h265_write_pps(ctx, pbc, pps); | |
1448 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 283 times.
|
283 | if (err < 0) |
1449 | ✗ | return err; | |
1450 | |||
1451 | 283 | err = cbs_h265_replace_pps(ctx, unit); | |
1452 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 283 times.
|
283 | if (err < 0) |
1453 | ✗ | return err; | |
1454 | } | ||
1455 | 283 | break; | |
1456 | |||
1457 | 6031 | case HEVC_NAL_TRAIL_N: | |
1458 | case HEVC_NAL_TRAIL_R: | ||
1459 | case HEVC_NAL_TSA_N: | ||
1460 | case HEVC_NAL_TSA_R: | ||
1461 | case HEVC_NAL_STSA_N: | ||
1462 | case HEVC_NAL_STSA_R: | ||
1463 | case HEVC_NAL_RADL_N: | ||
1464 | case HEVC_NAL_RADL_R: | ||
1465 | case HEVC_NAL_RASL_N: | ||
1466 | case HEVC_NAL_RASL_R: | ||
1467 | case HEVC_NAL_BLA_W_LP: | ||
1468 | case HEVC_NAL_BLA_W_RADL: | ||
1469 | case HEVC_NAL_BLA_N_LP: | ||
1470 | case HEVC_NAL_IDR_W_RADL: | ||
1471 | case HEVC_NAL_IDR_N_LP: | ||
1472 | case HEVC_NAL_CRA_NUT: | ||
1473 | { | ||
1474 | 6031 | H265RawSlice *slice = unit->content; | |
1475 | |||
1476 | 6031 | err = cbs_h265_write_slice_segment_header(ctx, pbc, &slice->header); | |
1477 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6031 times.
|
6031 | if (err < 0) |
1478 | ✗ | return err; | |
1479 | |||
1480 |
1/2✓ Branch 0 taken 6031 times.
✗ Branch 1 not taken.
|
6031 | if (slice->data) { |
1481 | 6031 | err = cbs_h2645_write_slice_data(ctx, pbc, slice->data, | |
1482 | slice->data_size, | ||
1483 | slice->data_bit_start); | ||
1484 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6031 times.
|
6031 | if (err < 0) |
1485 | ✗ | return err; | |
1486 | } else { | ||
1487 | // No slice data - that was just the header. | ||
1488 | } | ||
1489 | } | ||
1490 | 6031 | break; | |
1491 | |||
1492 | 60 | case HEVC_NAL_AUD: | |
1493 | { | ||
1494 | 60 | err = cbs_h265_write_aud(ctx, pbc, unit->content); | |
1495 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
|
60 | if (err < 0) |
1496 | ✗ | return err; | |
1497 | } | ||
1498 | 60 | break; | |
1499 | |||
1500 | 1562 | case HEVC_NAL_SEI_PREFIX: | |
1501 | case HEVC_NAL_SEI_SUFFIX: | ||
1502 | { | ||
1503 | 1562 | err = cbs_h265_write_sei(ctx, pbc, unit->content, | |
1504 | 1562 | unit->type == HEVC_NAL_SEI_PREFIX); | |
1505 | |||
1506 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1562 times.
|
1562 | if (err < 0) |
1507 | ✗ | return err; | |
1508 | } | ||
1509 | 1562 | break; | |
1510 | |||
1511 | ✗ | default: | |
1512 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Write unimplemented for " | |
1513 | "NAL unit type %"PRIu32".\n", unit->type); | ||
1514 | ✗ | return AVERROR_PATCHWELCOME; | |
1515 | } | ||
1516 | |||
1517 | 8050 | return 0; | |
1518 | } | ||
1519 | |||
1520 | 756 | static int cbs_h264_discarded_nal_unit(CodedBitstreamContext *ctx, | |
1521 | const CodedBitstreamUnit *unit, | ||
1522 | enum AVDiscard skip) | ||
1523 | { | ||
1524 | H264RawNALUnitHeader *header; | ||
1525 | H264RawSliceHeader *slice; | ||
1526 | int slice_type_i, slice_type_b, slice_type_si; | ||
1527 | |||
1528 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 756 times.
|
756 | if (skip <= AVDISCARD_DEFAULT) |
1529 | ✗ | return 0; | |
1530 | |||
1531 | // keep non-VCL | ||
1532 |
2/2✓ Branch 0 taken 256 times.
✓ Branch 1 taken 500 times.
|
756 | if (unit->type != H264_NAL_SLICE && |
1533 |
2/2✓ Branch 0 taken 252 times.
✓ Branch 1 taken 4 times.
|
256 | unit->type != H264_NAL_IDR_SLICE && |
1534 |
1/2✓ Branch 0 taken 252 times.
✗ Branch 1 not taken.
|
252 | unit->type != H264_NAL_AUXILIARY_SLICE) |
1535 | 252 | return 0; | |
1536 | |||
1537 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 504 times.
|
504 | if (skip >= AVDISCARD_ALL) |
1538 | ✗ | return 1; | |
1539 | |||
1540 |
4/4✓ Branch 0 taken 126 times.
✓ Branch 1 taken 378 times.
✓ Branch 2 taken 125 times.
✓ Branch 3 taken 1 times.
|
504 | if (skip >= AVDISCARD_NONKEY && unit->type != H264_NAL_IDR_SLICE) |
1541 | 125 | return 1; | |
1542 | |||
1543 | 379 | header = (H264RawNALUnitHeader *)unit->content; | |
1544 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 379 times.
|
379 | if (!header) { |
1545 | ✗ | av_log(ctx->log_ctx, AV_LOG_WARNING, | |
1546 | "h264 nal unit header is null, missing decompose?\n"); | ||
1547 | ✗ | return 0; | |
1548 | } | ||
1549 | |||
1550 |
3/4✓ Branch 0 taken 379 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 183 times.
✓ Branch 3 taken 196 times.
|
379 | if (skip >= AVDISCARD_NONREF && !header->nal_ref_idc) |
1551 | 183 | return 1; | |
1552 | |||
1553 | 196 | slice = (H264RawSliceHeader *)unit->content; | |
1554 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 196 times.
|
196 | if (!slice) { |
1555 | ✗ | av_log(ctx->log_ctx, AV_LOG_WARNING, | |
1556 | "h264 slice header is null, missing decompose?\n"); | ||
1557 | ✗ | return 0; | |
1558 | } | ||
1559 | |||
1560 | 196 | slice_type_i = slice->slice_type % 5 == 2; | |
1561 | 196 | slice_type_b = slice->slice_type % 5 == 1; | |
1562 | 196 | slice_type_si = slice->slice_type % 5 == 4; | |
1563 | |||
1564 |
4/4✓ Branch 0 taken 131 times.
✓ Branch 1 taken 65 times.
✓ Branch 2 taken 60 times.
✓ Branch 3 taken 71 times.
|
196 | if (skip >= AVDISCARD_BIDIR && slice_type_b) |
1565 | 60 | return 1; | |
1566 |
5/6✓ Branch 0 taken 36 times.
✓ Branch 1 taken 100 times.
✓ Branch 2 taken 29 times.
✓ Branch 3 taken 7 times.
✓ Branch 4 taken 29 times.
✗ Branch 5 not taken.
|
136 | if (skip >= AVDISCARD_NONINTRA && !slice_type_i && !slice_type_si) |
1567 | 29 | return 1; | |
1568 | |||
1569 | 107 | return 0; | |
1570 | } | ||
1571 | |||
1572 | 536 | static int cbs_h265_discarded_nal_unit(CodedBitstreamContext *ctx, | |
1573 | const CodedBitstreamUnit *unit, | ||
1574 | enum AVDiscard skip) | ||
1575 | { | ||
1576 | H265RawSliceHeader *slice; | ||
1577 | |||
1578 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 536 times.
|
536 | if (skip <= AVDISCARD_DEFAULT) |
1579 | ✗ | return 0; | |
1580 | |||
1581 |
3/3✓ Branch 0 taken 4 times.
✓ Branch 1 taken 328 times.
✓ Branch 2 taken 204 times.
|
536 | switch (unit->type) { |
1582 | 4 | case HEVC_NAL_BLA_W_LP: | |
1583 | case HEVC_NAL_BLA_W_RADL: | ||
1584 | case HEVC_NAL_BLA_N_LP: | ||
1585 | case HEVC_NAL_IDR_W_RADL: | ||
1586 | case HEVC_NAL_IDR_N_LP: | ||
1587 | case HEVC_NAL_CRA_NUT: | ||
1588 | // IRAP slice | ||
1589 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (skip < AVDISCARD_ALL) |
1590 | 4 | return 0; | |
1591 | ✗ | break; | |
1592 | |||
1593 | 328 | case HEVC_NAL_TRAIL_R: | |
1594 | case HEVC_NAL_TRAIL_N: | ||
1595 | case HEVC_NAL_TSA_N: | ||
1596 | case HEVC_NAL_TSA_R: | ||
1597 | case HEVC_NAL_STSA_N: | ||
1598 | case HEVC_NAL_STSA_R: | ||
1599 | case HEVC_NAL_RADL_N: | ||
1600 | case HEVC_NAL_RADL_R: | ||
1601 | case HEVC_NAL_RASL_N: | ||
1602 | case HEVC_NAL_RASL_R: | ||
1603 | // Slice | ||
1604 | 328 | break; | |
1605 | 204 | default: | |
1606 | // Don't discard non-slice nal. | ||
1607 | 204 | return 0; | |
1608 | } | ||
1609 | |||
1610 |
2/2✓ Branch 0 taken 47 times.
✓ Branch 1 taken 281 times.
|
328 | if (skip >= AVDISCARD_NONKEY) |
1611 | 47 | return 1; | |
1612 | |||
1613 | 281 | slice = (H265RawSliceHeader *)unit->content; | |
1614 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 281 times.
|
281 | if (!slice) { |
1615 | ✗ | av_log(ctx->log_ctx, AV_LOG_WARNING, | |
1616 | "h265 slice header is null, missing decompose?\n"); | ||
1617 | ✗ | return 0; | |
1618 | } | ||
1619 | |||
1620 |
3/4✓ Branch 0 taken 47 times.
✓ Branch 1 taken 234 times.
✓ Branch 2 taken 47 times.
✗ Branch 3 not taken.
|
281 | if (skip >= AVDISCARD_NONINTRA && slice->slice_type != HEVC_SLICE_I) |
1621 | 47 | return 1; | |
1622 |
3/4✓ Branch 0 taken 47 times.
✓ Branch 1 taken 187 times.
✓ Branch 2 taken 47 times.
✗ Branch 3 not taken.
|
234 | if (skip >= AVDISCARD_BIDIR && slice->slice_type == HEVC_SLICE_B) |
1623 | 47 | return 1; | |
1624 | |||
1625 |
1/2✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
|
187 | if (skip >= AVDISCARD_NONREF) { |
1626 |
2/2✓ Branch 0 taken 24 times.
✓ Branch 1 taken 163 times.
|
187 | switch (unit->type) { |
1627 | 24 | case HEVC_NAL_TRAIL_N: | |
1628 | case HEVC_NAL_TSA_N: | ||
1629 | case HEVC_NAL_STSA_N: | ||
1630 | case HEVC_NAL_RADL_N: | ||
1631 | case HEVC_NAL_RASL_N: | ||
1632 | case HEVC_NAL_VCL_N10: | ||
1633 | case HEVC_NAL_VCL_N12: | ||
1634 | case HEVC_NAL_VCL_N14: | ||
1635 | // non-ref | ||
1636 | 24 | return 1; | |
1637 | 163 | default: | |
1638 | 163 | break; | |
1639 | } | ||
1640 | } | ||
1641 | |||
1642 | 163 | return 0; | |
1643 | } | ||
1644 | |||
1645 | 5777 | static int cbs_h266_write_nal_unit(CodedBitstreamContext *ctx, | |
1646 | CodedBitstreamUnit *unit, | ||
1647 | PutBitContext *pbc) | ||
1648 | { | ||
1649 | int err; | ||
1650 | |||
1651 |
10/11✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 335 times.
✓ Branch 4 taken 391 times.
✓ Branch 5 taken 228 times.
✓ Branch 6 taken 27 times.
✓ Branch 7 taken 2588 times.
✓ Branch 8 taken 12 times.
✓ Branch 9 taken 2190 times.
✗ Branch 10 not taken.
|
5777 | switch (unit->type) { |
1652 | 1 | case VVC_DCI_NUT: | |
1653 | { | ||
1654 | 1 | H266RawDCI *dci = unit->content; | |
1655 | |||
1656 | 1 | err = cbs_h266_write_dci(ctx, pbc, dci); | |
1657 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (err < 0) |
1658 | ✗ | return err; | |
1659 | } | ||
1660 | 1 | break; | |
1661 | 1 | case VVC_OPI_NUT: | |
1662 | { | ||
1663 | 1 | H266RawOPI *opi = unit->content; | |
1664 | |||
1665 | 1 | err = cbs_h266_write_opi(ctx, pbc, opi); | |
1666 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (err < 0) |
1667 | ✗ | return err; | |
1668 | } | ||
1669 | 1 | break; | |
1670 | 4 | case VVC_VPS_NUT: | |
1671 | { | ||
1672 | 4 | H266RawVPS *vps = unit->content; | |
1673 | |||
1674 | 4 | err = cbs_h266_write_vps(ctx, pbc, vps); | |
1675 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (err < 0) |
1676 | ✗ | return err; | |
1677 | |||
1678 | 4 | err = cbs_h266_replace_vps(ctx, unit); | |
1679 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (err < 0) |
1680 | ✗ | return err; | |
1681 | } | ||
1682 | 4 | break; | |
1683 | 335 | case VVC_SPS_NUT: | |
1684 | { | ||
1685 | 335 | H266RawSPS *sps = unit->content; | |
1686 | |||
1687 | 335 | err = cbs_h266_write_sps(ctx, pbc, sps); | |
1688 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 335 times.
|
335 | if (err < 0) |
1689 | ✗ | return err; | |
1690 | |||
1691 | 335 | err = cbs_h266_replace_sps(ctx, unit); | |
1692 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 335 times.
|
335 | if (err < 0) |
1693 | ✗ | return err; | |
1694 | } | ||
1695 | 335 | break; | |
1696 | |||
1697 | 391 | case VVC_PPS_NUT: | |
1698 | { | ||
1699 | 391 | H266RawPPS *pps = unit->content; | |
1700 | |||
1701 | 391 | err = cbs_h266_write_pps(ctx, pbc, pps); | |
1702 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 391 times.
|
391 | if (err < 0) |
1703 | ✗ | return err; | |
1704 | |||
1705 | 391 | err = cbs_h266_replace_pps(ctx, unit); | |
1706 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 391 times.
|
391 | if (err < 0) |
1707 | ✗ | return err; | |
1708 | } | ||
1709 | 391 | break; | |
1710 | |||
1711 | 228 | case VVC_PREFIX_APS_NUT: | |
1712 | case VVC_SUFFIX_APS_NUT: | ||
1713 | { | ||
1714 | 228 | err = cbs_h266_write_aps(ctx, pbc, unit->content, | |
1715 | 228 | unit->type == VVC_PREFIX_APS_NUT); | |
1716 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 228 times.
|
228 | if (err < 0) |
1717 | ✗ | return err; | |
1718 | } | ||
1719 | 228 | break; | |
1720 | 27 | case VVC_PH_NUT: | |
1721 | { | ||
1722 | 27 | H266RawPH *ph = unit->content; | |
1723 | 27 | err = cbs_h266_write_ph(ctx, pbc, ph); | |
1724 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
|
27 | if (err < 0) |
1725 | ✗ | return err; | |
1726 | |||
1727 | 27 | err = cbs_h266_replace_ph(ctx, unit, &ph->ph_picture_header); | |
1728 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
|
27 | if (err < 0) |
1729 | ✗ | return err; | |
1730 | } | ||
1731 | 27 | break; | |
1732 | |||
1733 | 2588 | case VVC_TRAIL_NUT: | |
1734 | case VVC_STSA_NUT: | ||
1735 | case VVC_RADL_NUT: | ||
1736 | case VVC_RASL_NUT: | ||
1737 | case VVC_IDR_W_RADL: | ||
1738 | case VVC_IDR_N_LP: | ||
1739 | case VVC_CRA_NUT: | ||
1740 | case VVC_GDR_NUT: | ||
1741 | { | ||
1742 | 2588 | H266RawSlice *slice = unit->content; | |
1743 | |||
1744 | 2588 | err = cbs_h266_write_slice_header(ctx, pbc, &slice->header); | |
1745 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2588 times.
|
2588 | if (err < 0) |
1746 | ✗ | return err; | |
1747 | |||
1748 |
2/2✓ Branch 0 taken 2103 times.
✓ Branch 1 taken 485 times.
|
2588 | if (slice->header.sh_picture_header_in_slice_header_flag) { |
1749 | 2103 | err = cbs_h266_replace_ph(ctx, unit, &slice->header.sh_picture_header); | |
1750 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2103 times.
|
2103 | if (err < 0) |
1751 | ✗ | return err; | |
1752 | } | ||
1753 | |||
1754 |
1/2✓ Branch 0 taken 2588 times.
✗ Branch 1 not taken.
|
2588 | if (slice->data) { |
1755 | 2588 | err = cbs_h2645_write_slice_data(ctx, pbc, slice->data, | |
1756 | slice->data_size, | ||
1757 | slice->data_bit_start); | ||
1758 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2588 times.
|
2588 | if (err < 0) |
1759 | ✗ | return err; | |
1760 | } else { | ||
1761 | // No slice data - that was just the header. | ||
1762 | } | ||
1763 | } | ||
1764 | 2588 | break; | |
1765 | |||
1766 | 12 | case VVC_AUD_NUT: | |
1767 | { | ||
1768 | 12 | err = cbs_h266_write_aud(ctx, pbc, unit->content); | |
1769 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if (err < 0) |
1770 | ✗ | return err; | |
1771 | } | ||
1772 | 12 | break; | |
1773 | |||
1774 | 2190 | case VVC_PREFIX_SEI_NUT: | |
1775 | case VVC_SUFFIX_SEI_NUT: | ||
1776 | { | ||
1777 | 2190 | err = cbs_h266_write_sei(ctx, pbc, unit->content, | |
1778 | 2190 | unit->type == VVC_PREFIX_SEI_NUT); | |
1779 | |||
1780 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2190 times.
|
2190 | if (err < 0) |
1781 | ✗ | return err; | |
1782 | } | ||
1783 | 2190 | break; | |
1784 | |||
1785 | ✗ | default: | |
1786 | ✗ | av_log(ctx->log_ctx, AV_LOG_ERROR, "Write unimplemented for " | |
1787 | "NAL unit type %"PRIu32".\n", unit->type); | ||
1788 | ✗ | return AVERROR_PATCHWELCOME; | |
1789 | } | ||
1790 | |||
1791 | 5777 | return 0; | |
1792 | } | ||
1793 | |||
1794 | 21780 | static int cbs_h2645_unit_requires_zero_byte(enum AVCodecID codec_id, | |
1795 | CodedBitstreamUnitType type, | ||
1796 | int nal_unit_index) | ||
1797 | { | ||
1798 | // Section B.1.2 in H.264, section B.2.2 in H.265, H.266. | ||
1799 |
2/2✓ Branch 0 taken 7246 times.
✓ Branch 1 taken 14534 times.
|
21780 | if (nal_unit_index == 0) { |
1800 | // Assume that this is the first NAL unit in an access unit. | ||
1801 | 7246 | return 1; | |
1802 | } | ||
1803 |
2/2✓ Branch 0 taken 5022 times.
✓ Branch 1 taken 9512 times.
|
14534 | if (codec_id == AV_CODEC_ID_H264) |
1804 |
4/4✓ Branch 0 taken 5014 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 448 times.
✓ Branch 3 taken 4566 times.
|
5022 | return type == H264_NAL_SPS || type == H264_NAL_PPS; |
1805 |
2/2✓ Branch 0 taken 5867 times.
✓ Branch 1 taken 3645 times.
|
9512 | if (codec_id == AV_CODEC_ID_HEVC) |
1806 |
6/6✓ Branch 0 taken 5864 times.
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 5806 times.
✓ Branch 3 taken 58 times.
✓ Branch 4 taken 56 times.
✓ Branch 5 taken 5750 times.
|
5867 | return type == HEVC_NAL_VPS || type == HEVC_NAL_SPS || type == HEVC_NAL_PPS; |
1807 |
1/2✓ Branch 0 taken 3645 times.
✗ Branch 1 not taken.
|
3645 | if (codec_id == AV_CODEC_ID_VVC) |
1808 |
4/4✓ Branch 0 taken 2642 times.
✓ Branch 1 taken 1003 times.
✓ Branch 2 taken 438 times.
✓ Branch 3 taken 2204 times.
|
3645 | return type >= VVC_OPI_NUT && type <= VVC_SUFFIX_APS_NUT; |
1809 | ✗ | return 0; | |
1810 | } | ||
1811 | |||
1812 | 7247 | static int cbs_h2645_assemble_fragment(CodedBitstreamContext *ctx, | |
1813 | CodedBitstreamFragment *frag) | ||
1814 | { | ||
1815 | uint8_t *data; | ||
1816 | size_t max_size, dp, sp; | ||
1817 | int err, i, zero_run; | ||
1818 | |||
1819 |
2/2✓ Branch 0 taken 21780 times.
✓ Branch 1 taken 7247 times.
|
29027 | for (i = 0; i < frag->nb_units; i++) { |
1820 | // Data should already all have been written when we get here. | ||
1821 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 21780 times.
|
21780 | av_assert0(frag->units[i].data); |
1822 | } | ||
1823 | |||
1824 | 7247 | max_size = 0; | |
1825 |
2/2✓ Branch 0 taken 21780 times.
✓ Branch 1 taken 7247 times.
|
29027 | for (i = 0; i < frag->nb_units; i++) { |
1826 | // Start code + content with worst-case emulation prevention. | ||
1827 | 21780 | max_size += 4 + frag->units[i].data_size * 3 / 2; | |
1828 | } | ||
1829 | |||
1830 | 7247 | data = av_realloc(NULL, max_size + AV_INPUT_BUFFER_PADDING_SIZE); | |
1831 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7247 times.
|
7247 | if (!data) |
1832 | ✗ | return AVERROR(ENOMEM); | |
1833 | |||
1834 | 7247 | dp = 0; | |
1835 |
2/2✓ Branch 0 taken 21780 times.
✓ Branch 1 taken 7247 times.
|
29027 | for (i = 0; i < frag->nb_units; i++) { |
1836 | 21780 | CodedBitstreamUnit *unit = &frag->units[i]; | |
1837 | |||
1838 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 21780 times.
|
21780 | if (unit->data_bit_padding > 0) { |
1839 | ✗ | if (i < frag->nb_units - 1) | |
1840 | ✗ | av_log(ctx->log_ctx, AV_LOG_WARNING, "Probably invalid " | |
1841 | "unaligned padding on non-final NAL unit.\n"); | ||
1842 | else | ||
1843 | ✗ | frag->data_bit_padding = unit->data_bit_padding; | |
1844 | } | ||
1845 | |||
1846 |
2/2✓ Branch 1 taken 8257 times.
✓ Branch 2 taken 13523 times.
|
21780 | if (cbs_h2645_unit_requires_zero_byte(ctx->codec->codec_id, unit->type, i)) { |
1847 | // zero_byte | ||
1848 | 8257 | data[dp++] = 0; | |
1849 | } | ||
1850 | // start_code_prefix_one_3bytes | ||
1851 | 21780 | data[dp++] = 0; | |
1852 | 21780 | data[dp++] = 0; | |
1853 | 21780 | data[dp++] = 1; | |
1854 | |||
1855 | 21780 | zero_run = 0; | |
1856 |
2/2✓ Branch 0 taken 16462517 times.
✓ Branch 1 taken 21780 times.
|
16484297 | for (sp = 0; sp < unit->data_size; sp++) { |
1857 |
2/2✓ Branch 0 taken 16455111 times.
✓ Branch 1 taken 7406 times.
|
16462517 | if (zero_run < 2) { |
1858 |
2/2✓ Branch 0 taken 147107 times.
✓ Branch 1 taken 16308004 times.
|
16455111 | if (unit->data[sp] == 0) |
1859 | 147107 | ++zero_run; | |
1860 | else | ||
1861 | 16308004 | zero_run = 0; | |
1862 | } else { | ||
1863 |
2/2✓ Branch 0 taken 1287 times.
✓ Branch 1 taken 6119 times.
|
7406 | if ((unit->data[sp] & ~3) == 0) { |
1864 | // emulation_prevention_three_byte | ||
1865 | 1287 | data[dp++] = 3; | |
1866 | } | ||
1867 | 7406 | zero_run = unit->data[sp] == 0; | |
1868 | } | ||
1869 | 16462517 | data[dp++] = unit->data[sp]; | |
1870 | } | ||
1871 | } | ||
1872 | |||
1873 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7247 times.
|
7247 | av_assert0(dp <= max_size); |
1874 | 7247 | err = av_reallocp(&data, dp + AV_INPUT_BUFFER_PADDING_SIZE); | |
1875 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7247 times.
|
7247 | if (err) |
1876 | ✗ | return err; | |
1877 | 7247 | memset(data + dp, 0, AV_INPUT_BUFFER_PADDING_SIZE); | |
1878 | |||
1879 | 7247 | frag->data_ref = av_buffer_create(data, dp + AV_INPUT_BUFFER_PADDING_SIZE, | |
1880 | NULL, NULL, 0); | ||
1881 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7247 times.
|
7247 | if (!frag->data_ref) { |
1882 | ✗ | av_freep(&data); | |
1883 | ✗ | return AVERROR(ENOMEM); | |
1884 | } | ||
1885 | |||
1886 | 7247 | frag->data = data; | |
1887 | 7247 | frag->data_size = dp; | |
1888 | |||
1889 | 7247 | return 0; | |
1890 | } | ||
1891 | |||
1892 | ✗ | static void cbs_h264_flush(CodedBitstreamContext *ctx) | |
1893 | { | ||
1894 | ✗ | CodedBitstreamH264Context *h264 = ctx->priv_data; | |
1895 | |||
1896 | ✗ | for (int i = 0; i < FF_ARRAY_ELEMS(h264->sps); i++) | |
1897 | ✗ | ff_refstruct_unref(&h264->sps[i]); | |
1898 | ✗ | for (int i = 0; i < FF_ARRAY_ELEMS(h264->pps); i++) | |
1899 | ✗ | ff_refstruct_unref(&h264->pps[i]); | |
1900 | |||
1901 | ✗ | h264->active_sps = NULL; | |
1902 | ✗ | h264->active_pps = NULL; | |
1903 | ✗ | h264->last_slice_nal_unit_type = 0; | |
1904 | ✗ | } | |
1905 | |||
1906 | 44 | static void cbs_h264_close(CodedBitstreamContext *ctx) | |
1907 | { | ||
1908 | 44 | CodedBitstreamH264Context *h264 = ctx->priv_data; | |
1909 | int i; | ||
1910 | |||
1911 | 44 | ff_h2645_packet_uninit(&h264->common.read_packet); | |
1912 | |||
1913 |
2/2✓ Branch 0 taken 1408 times.
✓ Branch 1 taken 44 times.
|
1452 | for (i = 0; i < FF_ARRAY_ELEMS(h264->sps); i++) |
1914 | 1408 | ff_refstruct_unref(&h264->sps[i]); | |
1915 |
2/2✓ Branch 0 taken 11264 times.
✓ Branch 1 taken 44 times.
|
11308 | for (i = 0; i < FF_ARRAY_ELEMS(h264->pps); i++) |
1916 | 11264 | ff_refstruct_unref(&h264->pps[i]); | |
1917 | 44 | } | |
1918 | |||
1919 | ✗ | static void cbs_h265_flush(CodedBitstreamContext *ctx) | |
1920 | { | ||
1921 | ✗ | CodedBitstreamH265Context *h265 = ctx->priv_data; | |
1922 | |||
1923 | ✗ | for (int i = 0; i < FF_ARRAY_ELEMS(h265->vps); i++) | |
1924 | ✗ | ff_refstruct_unref(&h265->vps[i]); | |
1925 | ✗ | for (int i = 0; i < FF_ARRAY_ELEMS(h265->sps); i++) | |
1926 | ✗ | ff_refstruct_unref(&h265->sps[i]); | |
1927 | ✗ | for (int i = 0; i < FF_ARRAY_ELEMS(h265->pps); i++) | |
1928 | ✗ | ff_refstruct_unref(&h265->pps[i]); | |
1929 | |||
1930 | ✗ | h265->active_vps = NULL; | |
1931 | ✗ | h265->active_sps = NULL; | |
1932 | ✗ | h265->active_pps = NULL; | |
1933 | ✗ | } | |
1934 | |||
1935 | 44 | static void cbs_h265_close(CodedBitstreamContext *ctx) | |
1936 | { | ||
1937 | 44 | CodedBitstreamH265Context *h265 = ctx->priv_data; | |
1938 | int i; | ||
1939 | |||
1940 | 44 | ff_h2645_packet_uninit(&h265->common.read_packet); | |
1941 | |||
1942 |
2/2✓ Branch 0 taken 704 times.
✓ Branch 1 taken 44 times.
|
748 | for (i = 0; i < FF_ARRAY_ELEMS(h265->vps); i++) |
1943 | 704 | ff_refstruct_unref(&h265->vps[i]); | |
1944 |
2/2✓ Branch 0 taken 704 times.
✓ Branch 1 taken 44 times.
|
748 | for (i = 0; i < FF_ARRAY_ELEMS(h265->sps); i++) |
1945 | 704 | ff_refstruct_unref(&h265->sps[i]); | |
1946 |
2/2✓ Branch 0 taken 2816 times.
✓ Branch 1 taken 44 times.
|
2860 | for (i = 0; i < FF_ARRAY_ELEMS(h265->pps); i++) |
1947 | 2816 | ff_refstruct_unref(&h265->pps[i]); | |
1948 | 44 | } | |
1949 | |||
1950 | 186 | static void cbs_h266_flush(CodedBitstreamContext *ctx) | |
1951 | { | ||
1952 | 186 | CodedBitstreamH266Context *h266 = ctx->priv_data; | |
1953 | |||
1954 |
2/2✓ Branch 0 taken 2976 times.
✓ Branch 1 taken 186 times.
|
3162 | for (int i = 0; i < FF_ARRAY_ELEMS(h266->vps); i++) |
1955 | 2976 | ff_refstruct_unref(&h266->vps[i]); | |
1956 |
2/2✓ Branch 0 taken 2976 times.
✓ Branch 1 taken 186 times.
|
3162 | for (int i = 0; i < FF_ARRAY_ELEMS(h266->sps); i++) |
1957 | 2976 | ff_refstruct_unref(&h266->sps[i]); | |
1958 |
2/2✓ Branch 0 taken 11904 times.
✓ Branch 1 taken 186 times.
|
12090 | for (int i = 0; i < FF_ARRAY_ELEMS(h266->pps); i++) |
1959 | 11904 | ff_refstruct_unref(&h266->pps[i]); | |
1960 | 186 | ff_refstruct_unref(&h266->ph_ref); | |
1961 | 186 | } | |
1962 | |||
1963 | 186 | static void cbs_h266_close(CodedBitstreamContext *ctx) | |
1964 | { | ||
1965 | 186 | CodedBitstreamH266Context *h266 = ctx->priv_data; | |
1966 | |||
1967 | 186 | cbs_h266_flush(ctx); | |
1968 | 186 | ff_h2645_packet_uninit(&h266->common.read_packet); | |
1969 | 186 | } | |
1970 | |||
1971 | 913 | static void cbs_h264_free_sei(FFRefStructOpaque unused, void *content) | |
1972 | { | ||
1973 | 913 | H264RawSEI *sei = content; | |
1974 | 913 | ff_cbs_sei_free_message_list(&sei->message_list); | |
1975 | 913 | } | |
1976 | |||
1977 | static const CodedBitstreamUnitTypeDescriptor cbs_h264_unit_types[] = { | ||
1978 | CBS_UNIT_TYPE_POD(H264_NAL_SPS, H264RawSPS), | ||
1979 | CBS_UNIT_TYPE_POD(H264_NAL_SPS_EXT, H264RawSPSExtension), | ||
1980 | |||
1981 | CBS_UNIT_TYPE_INTERNAL_REF(H264_NAL_PPS, H264RawPPS, slice_group_id), | ||
1982 | |||
1983 | CBS_UNIT_TYPES_INTERNAL_REF((H264_NAL_IDR_SLICE, | ||
1984 | H264_NAL_SLICE, | ||
1985 | H264_NAL_AUXILIARY_SLICE), H264RawSlice, data), | ||
1986 | |||
1987 | CBS_UNIT_TYPE_POD(H264_NAL_AUD, H264RawAUD), | ||
1988 | CBS_UNIT_TYPE_POD(H264_NAL_FILLER_DATA, H264RawFiller), | ||
1989 | CBS_UNIT_TYPE_POD(H264_NAL_END_SEQUENCE, H264RawNALUnitHeader), | ||
1990 | CBS_UNIT_TYPE_POD(H264_NAL_END_STREAM, H264RawNALUnitHeader), | ||
1991 | |||
1992 | CBS_UNIT_TYPE_COMPLEX(H264_NAL_SEI, H264RawSEI, &cbs_h264_free_sei), | ||
1993 | |||
1994 | CBS_UNIT_TYPE_END_OF_LIST | ||
1995 | }; | ||
1996 | |||
1997 | 1727 | static void cbs_h265_free_sei(FFRefStructOpaque unused, void *content) | |
1998 | { | ||
1999 | 1727 | H265RawSEI *sei = content; | |
2000 | 1727 | ff_cbs_sei_free_message_list(&sei->message_list); | |
2001 | 1727 | } | |
2002 | |||
2003 | static const CodedBitstreamUnitTypeDescriptor cbs_h265_unit_types[] = { | ||
2004 | CBS_UNIT_TYPE_INTERNAL_REF(HEVC_NAL_VPS, H265RawVPS, extension_data.data), | ||
2005 | CBS_UNIT_TYPE_INTERNAL_REF(HEVC_NAL_SPS, H265RawSPS, extension_data.data), | ||
2006 | CBS_UNIT_TYPE_INTERNAL_REF(HEVC_NAL_PPS, H265RawPPS, extension_data.data), | ||
2007 | |||
2008 | CBS_UNIT_TYPE_POD(HEVC_NAL_AUD, H265RawAUD), | ||
2009 | |||
2010 | // Slices of non-IRAP pictures. | ||
2011 | CBS_UNIT_RANGE_INTERNAL_REF(HEVC_NAL_TRAIL_N, HEVC_NAL_RASL_R, | ||
2012 | H265RawSlice, data), | ||
2013 | // Slices of IRAP pictures. | ||
2014 | CBS_UNIT_RANGE_INTERNAL_REF(HEVC_NAL_BLA_W_LP, HEVC_NAL_CRA_NUT, | ||
2015 | H265RawSlice, data), | ||
2016 | |||
2017 | CBS_UNIT_TYPES_COMPLEX((HEVC_NAL_SEI_PREFIX, HEVC_NAL_SEI_SUFFIX), | ||
2018 | H265RawSEI, cbs_h265_free_sei), | ||
2019 | |||
2020 | CBS_UNIT_TYPE_END_OF_LIST | ||
2021 | }; | ||
2022 | |||
2023 | 3159 | static void cbs_h266_free_sei(FFRefStructOpaque unused, void *content) | |
2024 | { | ||
2025 | 3159 | H266RawSEI *sei = content; | |
2026 | 3159 | ff_cbs_sei_free_message_list(&sei->message_list); | |
2027 | 3159 | } | |
2028 | |||
2029 | static const CodedBitstreamUnitTypeDescriptor cbs_h266_unit_types[] = { | ||
2030 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_DCI_NUT, H266RawDCI, extension_data.data), | ||
2031 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_OPI_NUT, H266RawOPI, extension_data.data), | ||
2032 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_VPS_NUT, H266RawVPS, extension_data.data), | ||
2033 | { | ||
2034 | .nb_unit_types = 1, | ||
2035 | .unit_type.list[0] = VVC_SPS_NUT, | ||
2036 | .content_type = CBS_CONTENT_TYPE_INTERNAL_REFS, | ||
2037 | .content_size = sizeof(H266RawSPS), | ||
2038 | .type.ref = { | ||
2039 | .nb_offsets = 2, | ||
2040 | .offsets = { offsetof(H266RawSPS, extension_data.data), | ||
2041 | offsetof(H266RawSPS, vui.extension_data.data) } | ||
2042 | }, | ||
2043 | }, | ||
2044 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_PPS_NUT, H266RawPPS, extension_data.data), | ||
2045 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_PREFIX_APS_NUT, H266RawAPS, extension_data.data), | ||
2046 | CBS_UNIT_TYPE_INTERNAL_REF(VVC_SUFFIX_APS_NUT, H266RawAPS, extension_data.data), | ||
2047 | |||
2048 | CBS_UNIT_TYPE_POD(VVC_PH_NUT , H266RawPH), | ||
2049 | CBS_UNIT_TYPE_POD(VVC_AUD_NUT, H266RawAUD), | ||
2050 | |||
2051 | CBS_UNIT_RANGE_INTERNAL_REF(VVC_TRAIL_NUT, VVC_RASL_NUT, | ||
2052 | H266RawSlice, data), | ||
2053 | |||
2054 | CBS_UNIT_RANGE_INTERNAL_REF(VVC_IDR_W_RADL, VVC_GDR_NUT, | ||
2055 | H266RawSlice, data), | ||
2056 | |||
2057 | CBS_UNIT_TYPES_COMPLEX((VVC_PREFIX_SEI_NUT, VVC_SUFFIX_SEI_NUT), | ||
2058 | H266RawSEI, cbs_h266_free_sei), | ||
2059 | |||
2060 | CBS_UNIT_TYPE_END_OF_LIST | ||
2061 | }; | ||
2062 | |||
2063 | const CodedBitstreamType ff_cbs_type_h264 = { | ||
2064 | .codec_id = AV_CODEC_ID_H264, | ||
2065 | |||
2066 | .priv_data_size = sizeof(CodedBitstreamH264Context), | ||
2067 | |||
2068 | .unit_types = cbs_h264_unit_types, | ||
2069 | |||
2070 | .split_fragment = &cbs_h2645_split_fragment, | ||
2071 | .read_unit = &cbs_h264_read_nal_unit, | ||
2072 | .write_unit = &cbs_h264_write_nal_unit, | ||
2073 | .discarded_unit = &cbs_h264_discarded_nal_unit, | ||
2074 | .assemble_fragment = &cbs_h2645_assemble_fragment, | ||
2075 | |||
2076 | .flush = &cbs_h264_flush, | ||
2077 | .close = &cbs_h264_close, | ||
2078 | }; | ||
2079 | |||
2080 | const CodedBitstreamType ff_cbs_type_h265 = { | ||
2081 | .codec_id = AV_CODEC_ID_HEVC, | ||
2082 | |||
2083 | .priv_data_size = sizeof(CodedBitstreamH265Context), | ||
2084 | |||
2085 | .unit_types = cbs_h265_unit_types, | ||
2086 | |||
2087 | .split_fragment = &cbs_h2645_split_fragment, | ||
2088 | .read_unit = &cbs_h265_read_nal_unit, | ||
2089 | .write_unit = &cbs_h265_write_nal_unit, | ||
2090 | .discarded_unit = &cbs_h265_discarded_nal_unit, | ||
2091 | .assemble_fragment = &cbs_h2645_assemble_fragment, | ||
2092 | |||
2093 | .flush = &cbs_h265_flush, | ||
2094 | .close = &cbs_h265_close, | ||
2095 | }; | ||
2096 | |||
2097 | const CodedBitstreamType ff_cbs_type_h266 = { | ||
2098 | .codec_id = AV_CODEC_ID_VVC, | ||
2099 | |||
2100 | .priv_data_size = sizeof(CodedBitstreamH266Context), | ||
2101 | |||
2102 | .unit_types = cbs_h266_unit_types, | ||
2103 | |||
2104 | .split_fragment = &cbs_h2645_split_fragment, | ||
2105 | .read_unit = &cbs_h266_read_nal_unit, | ||
2106 | .write_unit = &cbs_h266_write_nal_unit, | ||
2107 | .assemble_fragment = &cbs_h2645_assemble_fragment, | ||
2108 | |||
2109 | .flush = &cbs_h266_flush, | ||
2110 | .close = &cbs_h266_close, | ||
2111 | }; | ||
2112 | |||
2113 | // Macro for the read/write pair. | ||
2114 | #define SEI_MESSAGE_RW(codec, name) \ | ||
2115 | .read = cbs_ ## codec ## _read_ ## name ## _internal, \ | ||
2116 | .write = cbs_ ## codec ## _write_ ## name ## _internal | ||
2117 | |||
2118 | static const SEIMessageTypeDescriptor cbs_sei_common_types[] = { | ||
2119 | { | ||
2120 | SEI_TYPE_FILLER_PAYLOAD, | ||
2121 | 1, 1, | ||
2122 | sizeof(SEIRawFillerPayload), | ||
2123 | SEI_MESSAGE_RW(sei, filler_payload), | ||
2124 | }, | ||
2125 | { | ||
2126 | SEI_TYPE_USER_DATA_REGISTERED_ITU_T_T35, | ||
2127 | 1, 1, | ||
2128 | sizeof(SEIRawUserDataRegistered), | ||
2129 | SEI_MESSAGE_RW(sei, user_data_registered), | ||
2130 | }, | ||
2131 | { | ||
2132 | SEI_TYPE_USER_DATA_UNREGISTERED, | ||
2133 | 1, 1, | ||
2134 | sizeof(SEIRawUserDataUnregistered), | ||
2135 | SEI_MESSAGE_RW(sei, user_data_unregistered), | ||
2136 | }, | ||
2137 | { | ||
2138 | SEI_TYPE_FRAME_PACKING_ARRANGEMENT, | ||
2139 | 1, 0, | ||
2140 | sizeof(SEIRawFramePackingArrangement), | ||
2141 | SEI_MESSAGE_RW(sei, frame_packing_arrangement), | ||
2142 | }, | ||
2143 | { | ||
2144 | SEI_TYPE_DECODED_PICTURE_HASH, | ||
2145 | 0, 1, | ||
2146 | sizeof(SEIRawDecodedPictureHash), | ||
2147 | SEI_MESSAGE_RW(sei, decoded_picture_hash), | ||
2148 | }, | ||
2149 | { | ||
2150 | SEI_TYPE_MASTERING_DISPLAY_COLOUR_VOLUME, | ||
2151 | 1, 0, | ||
2152 | sizeof(SEIRawMasteringDisplayColourVolume), | ||
2153 | SEI_MESSAGE_RW(sei, mastering_display_colour_volume), | ||
2154 | }, | ||
2155 | { | ||
2156 | SEI_TYPE_CONTENT_LIGHT_LEVEL_INFO, | ||
2157 | 1, 0, | ||
2158 | sizeof(SEIRawContentLightLevelInfo), | ||
2159 | SEI_MESSAGE_RW(sei, content_light_level_info), | ||
2160 | }, | ||
2161 | { | ||
2162 | SEI_TYPE_ALTERNATIVE_TRANSFER_CHARACTERISTICS, | ||
2163 | 1, 0, | ||
2164 | sizeof(SEIRawAlternativeTransferCharacteristics), | ||
2165 | SEI_MESSAGE_RW(sei, alternative_transfer_characteristics), | ||
2166 | }, | ||
2167 | { | ||
2168 | SEI_TYPE_AMBIENT_VIEWING_ENVIRONMENT, | ||
2169 | 1, 0, | ||
2170 | sizeof(SEIRawAmbientViewingEnvironment), | ||
2171 | SEI_MESSAGE_RW(sei, ambient_viewing_environment), | ||
2172 | }, | ||
2173 | SEI_MESSAGE_TYPE_END, | ||
2174 | }; | ||
2175 | |||
2176 | static const SEIMessageTypeDescriptor cbs_sei_h264_types[] = { | ||
2177 | { | ||
2178 | SEI_TYPE_BUFFERING_PERIOD, | ||
2179 | 1, 0, | ||
2180 | sizeof(H264RawSEIBufferingPeriod), | ||
2181 | SEI_MESSAGE_RW(h264, sei_buffering_period), | ||
2182 | }, | ||
2183 | { | ||
2184 | SEI_TYPE_PIC_TIMING, | ||
2185 | 1, 0, | ||
2186 | sizeof(H264RawSEIPicTiming), | ||
2187 | SEI_MESSAGE_RW(h264, sei_pic_timing), | ||
2188 | }, | ||
2189 | { | ||
2190 | SEI_TYPE_PAN_SCAN_RECT, | ||
2191 | 1, 0, | ||
2192 | sizeof(H264RawSEIPanScanRect), | ||
2193 | SEI_MESSAGE_RW(h264, sei_pan_scan_rect), | ||
2194 | }, | ||
2195 | { | ||
2196 | SEI_TYPE_RECOVERY_POINT, | ||
2197 | 1, 0, | ||
2198 | sizeof(H264RawSEIRecoveryPoint), | ||
2199 | SEI_MESSAGE_RW(h264, sei_recovery_point), | ||
2200 | }, | ||
2201 | { | ||
2202 | SEI_TYPE_FILM_GRAIN_CHARACTERISTICS, | ||
2203 | 1, 0, | ||
2204 | sizeof(H264RawFilmGrainCharacteristics), | ||
2205 | SEI_MESSAGE_RW(h264, film_grain_characteristics), | ||
2206 | }, | ||
2207 | { | ||
2208 | SEI_TYPE_FRAME_PACKING_ARRANGEMENT, | ||
2209 | 1, 0, | ||
2210 | sizeof(H264RawSEIFramePackingArrangement), | ||
2211 | SEI_MESSAGE_RW(h264, sei_frame_packing_arrangement), | ||
2212 | }, | ||
2213 | { | ||
2214 | SEI_TYPE_DISPLAY_ORIENTATION, | ||
2215 | 1, 0, | ||
2216 | sizeof(H264RawSEIDisplayOrientation), | ||
2217 | SEI_MESSAGE_RW(h264, sei_display_orientation), | ||
2218 | }, | ||
2219 | SEI_MESSAGE_TYPE_END | ||
2220 | }; | ||
2221 | |||
2222 | static const SEIMessageTypeDescriptor cbs_sei_h265_types[] = { | ||
2223 | { | ||
2224 | SEI_TYPE_BUFFERING_PERIOD, | ||
2225 | 1, 0, | ||
2226 | sizeof(H265RawSEIBufferingPeriod), | ||
2227 | SEI_MESSAGE_RW(h265, sei_buffering_period), | ||
2228 | }, | ||
2229 | { | ||
2230 | SEI_TYPE_PIC_TIMING, | ||
2231 | 1, 0, | ||
2232 | sizeof(H265RawSEIPicTiming), | ||
2233 | SEI_MESSAGE_RW(h265, sei_pic_timing), | ||
2234 | }, | ||
2235 | { | ||
2236 | SEI_TYPE_PAN_SCAN_RECT, | ||
2237 | 1, 0, | ||
2238 | sizeof(H265RawSEIPanScanRect), | ||
2239 | SEI_MESSAGE_RW(h265, sei_pan_scan_rect), | ||
2240 | }, | ||
2241 | { | ||
2242 | SEI_TYPE_RECOVERY_POINT, | ||
2243 | 1, 0, | ||
2244 | sizeof(H265RawSEIRecoveryPoint), | ||
2245 | SEI_MESSAGE_RW(h265, sei_recovery_point), | ||
2246 | }, | ||
2247 | { | ||
2248 | SEI_TYPE_FILM_GRAIN_CHARACTERISTICS, | ||
2249 | 1, 0, | ||
2250 | sizeof(H265RawFilmGrainCharacteristics), | ||
2251 | SEI_MESSAGE_RW(h265, film_grain_characteristics), | ||
2252 | }, | ||
2253 | { | ||
2254 | SEI_TYPE_DISPLAY_ORIENTATION, | ||
2255 | 1, 0, | ||
2256 | sizeof(H265RawSEIDisplayOrientation), | ||
2257 | SEI_MESSAGE_RW(h265, sei_display_orientation), | ||
2258 | }, | ||
2259 | { | ||
2260 | SEI_TYPE_ACTIVE_PARAMETER_SETS, | ||
2261 | 1, 0, | ||
2262 | sizeof(H265RawSEIActiveParameterSets), | ||
2263 | SEI_MESSAGE_RW(h265, sei_active_parameter_sets), | ||
2264 | }, | ||
2265 | { | ||
2266 | SEI_TYPE_DECODED_PICTURE_HASH, | ||
2267 | 0, 1, | ||
2268 | sizeof(H265RawSEIDecodedPictureHash), | ||
2269 | SEI_MESSAGE_RW(h265, sei_decoded_picture_hash), | ||
2270 | }, | ||
2271 | { | ||
2272 | SEI_TYPE_TIME_CODE, | ||
2273 | 1, 0, | ||
2274 | sizeof(H265RawSEITimeCode), | ||
2275 | SEI_MESSAGE_RW(h265, sei_time_code), | ||
2276 | }, | ||
2277 | { | ||
2278 | SEI_TYPE_ALPHA_CHANNEL_INFO, | ||
2279 | 1, 0, | ||
2280 | sizeof(H265RawSEIAlphaChannelInfo), | ||
2281 | SEI_MESSAGE_RW(h265, sei_alpha_channel_info), | ||
2282 | }, | ||
2283 | { | ||
2284 | SEI_TYPE_THREE_DIMENSIONAL_REFERENCE_DISPLAYS_INFO, | ||
2285 | 1, 0, | ||
2286 | sizeof(H265RawSEI3DReferenceDisplaysInfo), | ||
2287 | SEI_MESSAGE_RW(h265, sei_3d_reference_displays_info), | ||
2288 | }, | ||
2289 | SEI_MESSAGE_TYPE_END | ||
2290 | }; | ||
2291 | |||
2292 | static const SEIMessageTypeDescriptor cbs_sei_h266_types[] = { | ||
2293 | SEI_MESSAGE_TYPE_END | ||
2294 | }; | ||
2295 | |||
2296 | 14373 | const SEIMessageTypeDescriptor *ff_cbs_sei_find_type(CodedBitstreamContext *ctx, | |
2297 | int payload_type) | ||
2298 | { | ||
2299 | const SEIMessageTypeDescriptor *codec_list; | ||
2300 | int i; | ||
2301 | |||
2302 |
3/4✓ Branch 0 taken 1983 times.
✓ Branch 1 taken 4851 times.
✓ Branch 2 taken 7539 times.
✗ Branch 3 not taken.
|
14373 | switch (ctx->codec->codec_id) { |
2303 | 1983 | case AV_CODEC_ID_H264: | |
2304 | 1983 | codec_list = cbs_sei_h264_types; | |
2305 | 1983 | break; | |
2306 | 4851 | case AV_CODEC_ID_H265: | |
2307 | 4851 | codec_list = cbs_sei_h265_types; | |
2308 | 4851 | break; | |
2309 | 7539 | case AV_CODEC_ID_H266: | |
2310 | 7539 | codec_list = cbs_sei_h266_types; | |
2311 | 7539 | break; | |
2312 | ✗ | default: | |
2313 | ✗ | return NULL; | |
2314 | } | ||
2315 | |||
2316 |
2/2✓ Branch 0 taken 40628 times.
✓ Branch 1 taken 7692 times.
|
48320 | for (i = 0; codec_list[i].type >= 0; i++) { |
2317 |
2/2✓ Branch 0 taken 6681 times.
✓ Branch 1 taken 33947 times.
|
40628 | if (codec_list[i].type == payload_type) |
2318 | 6681 | return &codec_list[i]; | |
2319 | } | ||
2320 | |||
2321 |
2/2✓ Branch 0 taken 39194 times.
✓ Branch 1 taken 269 times.
|
39463 | for (i = 0; cbs_sei_common_types[i].type >= 0; i++) { |
2322 |
2/2✓ Branch 0 taken 7423 times.
✓ Branch 1 taken 31771 times.
|
39194 | if (cbs_sei_common_types[i].type == payload_type) |
2323 | 7423 | return &cbs_sei_common_types[i]; | |
2324 | } | ||
2325 | |||
2326 | 269 | return NULL; | |
2327 | } | ||
2328 |