Line | Count | Source |
1 | | /* |
2 | | * Copyright © 2018-2021, VideoLAN and dav1d authors |
3 | | * Copyright © 2018, Two Orioles, LLC |
4 | | * All rights reserved. |
5 | | * |
6 | | * Redistribution and use in source and binary forms, with or without |
7 | | * modification, are permitted provided that the following conditions are met: |
8 | | * |
9 | | * 1. Redistributions of source code must retain the above copyright notice, this |
10 | | * list of conditions and the following disclaimer. |
11 | | * |
12 | | * 2. Redistributions in binary form must reproduce the above copyright notice, |
13 | | * this list of conditions and the following disclaimer in the documentation |
14 | | * and/or other materials provided with the distribution. |
15 | | * |
16 | | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
17 | | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
18 | | * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
19 | | * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR |
20 | | * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
21 | | * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
22 | | * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
23 | | * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
24 | | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
25 | | * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
26 | | */ |
27 | | |
28 | | #include "config.h" |
29 | | |
30 | | #include <errno.h> |
31 | | #include <limits.h> |
32 | | #include <stdio.h> |
33 | | |
34 | | #include "dav1d/data.h" |
35 | | |
36 | | #include "common/frame.h" |
37 | | #include "common/intops.h" |
38 | | #include "common/validate.h" |
39 | | |
40 | | #include "src/decode.h" |
41 | | #include "src/getbits.h" |
42 | | #include "src/levels.h" |
43 | | #include "src/log.h" |
44 | | #include "src/obu.h" |
45 | | #include "src/ref.h" |
46 | | #include "src/thread_task.h" |
47 | | |
48 | | static int check_trailing_bits(GetBits *const gb, |
49 | | const int strict_std_compliance) |
50 | 112k | { |
51 | 112k | const int trailing_one_bit = dav1d_get_bit(gb); |
52 | | |
53 | 112k | if (gb->error) |
54 | 10.3k | return DAV1D_ERR(EINVAL); |
55 | | |
56 | 101k | if (!strict_std_compliance) |
57 | 101k | return 0; |
58 | | |
59 | 0 | if (!trailing_one_bit || gb->state) |
60 | 0 | return DAV1D_ERR(EINVAL); |
61 | | |
62 | 0 | ptrdiff_t size = gb->ptr_end - gb->ptr; |
63 | 0 | while (size > 0 && gb->ptr[size - 1] == 0) |
64 | 0 | size--; |
65 | |
|
66 | 0 | if (size) |
67 | 0 | return DAV1D_ERR(EINVAL); |
68 | | |
69 | 0 | return 0; |
70 | 0 | } |
71 | | |
72 | | static NOINLINE int parse_seq_hdr(Dav1dSequenceHeader *const hdr, |
73 | | GetBits *const gb, |
74 | | const int strict_std_compliance) |
75 | 71.9k | { |
76 | 71.9k | #define DEBUG_SEQ_HDR 0 |
77 | | |
78 | | #if DEBUG_SEQ_HDR |
79 | | const unsigned init_bit_pos = dav1d_get_bits_pos(gb); |
80 | | #endif |
81 | | |
82 | 71.9k | memset(hdr, 0, sizeof(*hdr)); |
83 | 71.9k | hdr->profile = dav1d_get_bits(gb, 3); |
84 | 71.9k | if (hdr->profile > 2) goto error; |
85 | | #if DEBUG_SEQ_HDR |
86 | | printf("SEQHDR: post-profile: off=%u\n", |
87 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
88 | | #endif |
89 | | |
90 | 70.4k | hdr->still_picture = dav1d_get_bit(gb); |
91 | 70.4k | hdr->reduced_still_picture_header = dav1d_get_bit(gb); |
92 | 70.4k | if (hdr->reduced_still_picture_header && !hdr->still_picture) goto error; |
93 | | #if DEBUG_SEQ_HDR |
94 | | printf("SEQHDR: post-stillpicture_flags: off=%u\n", |
95 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
96 | | #endif |
97 | | |
98 | 70.3k | if (hdr->reduced_still_picture_header) { |
99 | 39.4k | hdr->num_operating_points = 1; |
100 | 39.4k | hdr->operating_points[0].major_level = dav1d_get_bits(gb, 3); |
101 | 39.4k | hdr->operating_points[0].minor_level = dav1d_get_bits(gb, 2); |
102 | 39.4k | hdr->operating_points[0].initial_display_delay = 10; |
103 | 39.4k | } else { |
104 | 30.9k | hdr->timing_info_present = dav1d_get_bit(gb); |
105 | 30.9k | if (hdr->timing_info_present) { |
106 | 3.76k | hdr->num_units_in_tick = dav1d_get_bits(gb, 32); |
107 | 3.76k | hdr->time_scale = dav1d_get_bits(gb, 32); |
108 | 3.76k | if (strict_std_compliance && (!hdr->num_units_in_tick || !hdr->time_scale)) |
109 | 0 | goto error; |
110 | 3.76k | hdr->equal_picture_interval = dav1d_get_bit(gb); |
111 | 3.76k | if (hdr->equal_picture_interval) { |
112 | 1.48k | const unsigned num_ticks_per_picture = dav1d_get_vlc(gb); |
113 | 1.48k | if (num_ticks_per_picture == UINT32_MAX) |
114 | 90 | goto error; |
115 | 1.39k | hdr->num_ticks_per_picture = num_ticks_per_picture + 1; |
116 | 1.39k | } |
117 | | |
118 | 3.67k | hdr->decoder_model_info_present = dav1d_get_bit(gb); |
119 | 3.67k | if (hdr->decoder_model_info_present) { |
120 | 2.78k | hdr->encoder_decoder_buffer_delay_length = dav1d_get_bits(gb, 5) + 1; |
121 | 2.78k | hdr->num_units_in_decoding_tick = dav1d_get_bits(gb, 32); |
122 | 2.78k | if (strict_std_compliance && !hdr->num_units_in_decoding_tick) |
123 | 0 | goto error; |
124 | 2.78k | hdr->buffer_removal_delay_length = dav1d_get_bits(gb, 5) + 1; |
125 | 2.78k | hdr->frame_presentation_delay_length = dav1d_get_bits(gb, 5) + 1; |
126 | 2.78k | } |
127 | 3.67k | } |
128 | | #if DEBUG_SEQ_HDR |
129 | | printf("SEQHDR: post-timinginfo: off=%u\n", |
130 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
131 | | #endif |
132 | | |
133 | 30.8k | hdr->display_model_info_present = dav1d_get_bit(gb); |
134 | 30.8k | hdr->num_operating_points = dav1d_get_bits(gb, 5) + 1; |
135 | 89.5k | for (int i = 0; i < hdr->num_operating_points; i++) { |
136 | 63.2k | struct Dav1dSequenceHeaderOperatingPoint *const op = |
137 | 63.2k | &hdr->operating_points[i]; |
138 | 63.2k | op->idc = dav1d_get_bits(gb, 12); |
139 | 63.2k | if (op->idc && (!(op->idc & 0xff) || !(op->idc & 0xf00))) |
140 | 4.51k | goto error; |
141 | 58.7k | op->major_level = 2 + dav1d_get_bits(gb, 3); |
142 | 58.7k | op->minor_level = dav1d_get_bits(gb, 2); |
143 | 58.7k | if (op->major_level > 3) |
144 | 18.7k | op->tier = dav1d_get_bit(gb); |
145 | 58.7k | if (hdr->decoder_model_info_present) { |
146 | 11.5k | op->decoder_model_param_present = dav1d_get_bit(gb); |
147 | 11.5k | if (op->decoder_model_param_present) { |
148 | 4.70k | struct Dav1dSequenceHeaderOperatingParameterInfo *const opi = |
149 | 4.70k | &hdr->operating_parameter_info[i]; |
150 | 4.70k | opi->decoder_buffer_delay = |
151 | 4.70k | dav1d_get_bits(gb, hdr->encoder_decoder_buffer_delay_length); |
152 | 4.70k | opi->encoder_buffer_delay = |
153 | 4.70k | dav1d_get_bits(gb, hdr->encoder_decoder_buffer_delay_length); |
154 | 4.70k | opi->low_delay_mode = dav1d_get_bit(gb); |
155 | 4.70k | } |
156 | 11.5k | } |
157 | 58.7k | if (hdr->display_model_info_present) |
158 | 12.6k | op->display_model_param_present = dav1d_get_bit(gb); |
159 | 58.7k | op->initial_display_delay = |
160 | 58.7k | op->display_model_param_present ? dav1d_get_bits(gb, 4) + 1 : 10; |
161 | 58.7k | } |
162 | | #if DEBUG_SEQ_HDR |
163 | | printf("SEQHDR: post-operating-points: off=%u\n", |
164 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
165 | | #endif |
166 | 30.8k | } |
167 | | |
168 | 65.7k | hdr->width_n_bits = dav1d_get_bits(gb, 4) + 1; |
169 | 65.7k | hdr->height_n_bits = dav1d_get_bits(gb, 4) + 1; |
170 | 65.7k | hdr->max_width = dav1d_get_bits(gb, hdr->width_n_bits) + 1; |
171 | 65.7k | hdr->max_height = dav1d_get_bits(gb, hdr->height_n_bits) + 1; |
172 | | #if DEBUG_SEQ_HDR |
173 | | printf("SEQHDR: post-size: off=%u\n", |
174 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
175 | | #endif |
176 | 65.7k | if (!hdr->reduced_still_picture_header) { |
177 | 26.3k | hdr->frame_id_numbers_present = dav1d_get_bit(gb); |
178 | 26.3k | if (hdr->frame_id_numbers_present) { |
179 | 2.43k | hdr->delta_frame_id_n_bits = dav1d_get_bits(gb, 4) + 2; |
180 | 2.43k | hdr->frame_id_n_bits = dav1d_get_bits(gb, 3) + hdr->delta_frame_id_n_bits + 1; |
181 | 2.43k | } |
182 | 26.3k | } |
183 | | #if DEBUG_SEQ_HDR |
184 | | printf("SEQHDR: post-frame-id-numbers-present: off=%u\n", |
185 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
186 | | #endif |
187 | | |
188 | 65.7k | hdr->sb128 = dav1d_get_bit(gb); |
189 | 65.7k | hdr->filter_intra = dav1d_get_bit(gb); |
190 | 65.7k | hdr->intra_edge_filter = dav1d_get_bit(gb); |
191 | 65.7k | if (hdr->reduced_still_picture_header) { |
192 | 39.4k | hdr->screen_content_tools = DAV1D_ADAPTIVE; |
193 | 39.4k | hdr->force_integer_mv = DAV1D_ADAPTIVE; |
194 | 39.4k | } else { |
195 | 26.3k | hdr->inter_intra = dav1d_get_bit(gb); |
196 | 26.3k | hdr->masked_compound = dav1d_get_bit(gb); |
197 | 26.3k | hdr->warped_motion = dav1d_get_bit(gb); |
198 | 26.3k | hdr->dual_filter = dav1d_get_bit(gb); |
199 | 26.3k | hdr->order_hint = dav1d_get_bit(gb); |
200 | 26.3k | if (hdr->order_hint) { |
201 | 17.5k | hdr->jnt_comp = dav1d_get_bit(gb); |
202 | 17.5k | hdr->ref_frame_mvs = dav1d_get_bit(gb); |
203 | 17.5k | } |
204 | 26.3k | hdr->screen_content_tools = dav1d_get_bit(gb) ? DAV1D_ADAPTIVE : dav1d_get_bit(gb); |
205 | | #if DEBUG_SEQ_HDR |
206 | | printf("SEQHDR: post-screentools: off=%u\n", |
207 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
208 | | #endif |
209 | 26.3k | hdr->force_integer_mv = hdr->screen_content_tools ? |
210 | 20.4k | dav1d_get_bit(gb) ? DAV1D_ADAPTIVE : dav1d_get_bit(gb) : 2; |
211 | 26.3k | if (hdr->order_hint) |
212 | 17.5k | hdr->order_hint_n_bits = dav1d_get_bits(gb, 3) + 1; |
213 | 26.3k | } |
214 | 65.7k | hdr->super_res = dav1d_get_bit(gb); |
215 | 65.7k | hdr->cdef = dav1d_get_bit(gb); |
216 | 65.7k | hdr->restoration = dav1d_get_bit(gb); |
217 | | #if DEBUG_SEQ_HDR |
218 | | printf("SEQHDR: post-featurebits: off=%u\n", |
219 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
220 | | #endif |
221 | | |
222 | 65.7k | hdr->hbd = dav1d_get_bit(gb); |
223 | 65.7k | if (hdr->profile == 2 && hdr->hbd) |
224 | 20.4k | hdr->hbd += dav1d_get_bit(gb); |
225 | 65.7k | if (hdr->profile != 1) |
226 | 49.8k | hdr->monochrome = dav1d_get_bit(gb); |
227 | 65.7k | hdr->color_description_present = dav1d_get_bit(gb); |
228 | 65.7k | if (hdr->color_description_present) { |
229 | 6.94k | hdr->pri = dav1d_get_bits(gb, 8); |
230 | 6.94k | hdr->trc = dav1d_get_bits(gb, 8); |
231 | 6.94k | hdr->mtrx = dav1d_get_bits(gb, 8); |
232 | 58.7k | } else { |
233 | 58.7k | hdr->pri = DAV1D_COLOR_PRI_UNKNOWN; |
234 | 58.7k | hdr->trc = DAV1D_TRC_UNKNOWN; |
235 | 58.7k | hdr->mtrx = DAV1D_MC_UNKNOWN; |
236 | 58.7k | } |
237 | 65.7k | if (hdr->monochrome) { |
238 | 20.9k | hdr->color_range = dav1d_get_bit(gb); |
239 | 20.9k | hdr->layout = DAV1D_PIXEL_LAYOUT_I400; |
240 | 20.9k | hdr->ss_hor = hdr->ss_ver = 1; |
241 | 20.9k | hdr->chr = DAV1D_CHR_UNKNOWN; |
242 | 44.7k | } else if (hdr->pri == DAV1D_COLOR_PRI_BT709 && |
243 | 2.64k | hdr->trc == DAV1D_TRC_SRGB && |
244 | 1.62k | hdr->mtrx == DAV1D_MC_IDENTITY) |
245 | 834 | { |
246 | 834 | hdr->layout = DAV1D_PIXEL_LAYOUT_I444; |
247 | 834 | hdr->color_range = 1; |
248 | 834 | if (hdr->profile != 1 && !(hdr->profile == 2 && hdr->hbd == 2)) |
249 | 369 | goto error; |
250 | 43.9k | } else { |
251 | 43.9k | hdr->color_range = dav1d_get_bit(gb); |
252 | 43.9k | switch (hdr->profile) { |
253 | 14.0k | case 0: hdr->layout = DAV1D_PIXEL_LAYOUT_I420; |
254 | 14.0k | hdr->ss_hor = hdr->ss_ver = 1; |
255 | 14.0k | break; |
256 | 15.5k | case 1: hdr->layout = DAV1D_PIXEL_LAYOUT_I444; |
257 | 15.5k | break; |
258 | 14.4k | case 2: |
259 | 14.4k | if (hdr->hbd == 2) { |
260 | 8.45k | hdr->ss_hor = dav1d_get_bit(gb); |
261 | 8.45k | if (hdr->ss_hor) |
262 | 2.43k | hdr->ss_ver = dav1d_get_bit(gb); |
263 | 8.45k | } else |
264 | 5.94k | hdr->ss_hor = 1; |
265 | 14.4k | hdr->layout = hdr->ss_hor ? |
266 | 8.37k | hdr->ss_ver ? DAV1D_PIXEL_LAYOUT_I420 : |
267 | 8.37k | DAV1D_PIXEL_LAYOUT_I422 : |
268 | 14.4k | DAV1D_PIXEL_LAYOUT_I444; |
269 | 14.4k | break; |
270 | 43.9k | } |
271 | 43.9k | hdr->chr = (hdr->ss_hor & hdr->ss_ver) ? |
272 | 28.1k | dav1d_get_bits(gb, 2) : DAV1D_CHR_UNKNOWN; |
273 | 43.9k | } |
274 | 65.3k | if (strict_std_compliance && |
275 | 0 | hdr->mtrx == DAV1D_MC_IDENTITY && hdr->layout != DAV1D_PIXEL_LAYOUT_I444) |
276 | 0 | { |
277 | 0 | goto error; |
278 | 0 | } |
279 | 65.3k | if (!hdr->monochrome) |
280 | 44.4k | hdr->separate_uv_delta_q = dav1d_get_bit(gb); |
281 | | #if DEBUG_SEQ_HDR |
282 | | printf("SEQHDR: post-colorinfo: off=%u\n", |
283 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
284 | | #endif |
285 | | |
286 | 65.3k | hdr->film_grain_present = dav1d_get_bit(gb); |
287 | | #if DEBUG_SEQ_HDR |
288 | | printf("SEQHDR: post-filmgrain: off=%u\n", |
289 | | dav1d_get_bits_pos(gb) - init_bit_pos); |
290 | | #endif |
291 | | |
292 | | // We needn't bother flushing the OBU here: we'll check we didn't |
293 | | // overrun in the caller and will then discard gb, so there's no |
294 | | // point in setting its position properly. |
295 | | |
296 | 65.3k | return check_trailing_bits(gb, strict_std_compliance); |
297 | | |
298 | 6.62k | error: |
299 | 6.62k | return DAV1D_ERR(EINVAL); |
300 | 65.3k | } |
301 | | |
302 | | int dav1d_parse_sequence_header(Dav1dSequenceHeader *const out, |
303 | | const uint8_t *const ptr, const size_t sz) |
304 | 25.7k | { |
305 | 25.7k | validate_input_or_ret(out != NULL, DAV1D_ERR(EINVAL)); |
306 | 25.7k | validate_input_or_ret(ptr != NULL, DAV1D_ERR(EINVAL)); |
307 | 25.7k | validate_input_or_ret(sz > 0 && sz <= SIZE_MAX / 2, DAV1D_ERR(EINVAL)); |
308 | | |
309 | 25.7k | GetBits gb; |
310 | 25.7k | dav1d_init_get_bits(&gb, ptr, sz); |
311 | 25.7k | int res = DAV1D_ERR(ENOENT); |
312 | | |
313 | 56.1k | do { |
314 | 56.1k | dav1d_get_bit(&gb); // obu_forbidden_bit |
315 | 56.1k | const enum Dav1dObuType type = dav1d_get_bits(&gb, 4); |
316 | 56.1k | const int has_extension = dav1d_get_bit(&gb); |
317 | 56.1k | const int has_length_field = dav1d_get_bit(&gb); |
318 | 56.1k | dav1d_get_bits(&gb, 1 + 8 * has_extension); // ignore |
319 | | |
320 | 56.1k | const uint8_t *obu_end = gb.ptr_end; |
321 | 56.1k | if (has_length_field) { |
322 | 33.9k | const size_t len = dav1d_get_uleb128(&gb); |
323 | 33.9k | if (len > (size_t)(obu_end - gb.ptr)) return DAV1D_ERR(EINVAL); |
324 | 33.5k | obu_end = gb.ptr + len; |
325 | 33.5k | } |
326 | | |
327 | 55.6k | if (type == DAV1D_OBU_SEQ_HDR) { |
328 | 25.2k | if ((res = parse_seq_hdr(out, &gb, 0)) < 0) return res; |
329 | 20.8k | if (gb.ptr > obu_end) return DAV1D_ERR(EINVAL); |
330 | 20.3k | dav1d_bytealign_get_bits(&gb); |
331 | 20.3k | } |
332 | | |
333 | 50.7k | if (gb.error) return DAV1D_ERR(EINVAL); |
334 | 50.7k | assert(gb.state == 0 && gb.bits_left == 0); |
335 | 49.9k | gb.ptr = obu_end; |
336 | 49.9k | } while (gb.ptr < gb.ptr_end); |
337 | | |
338 | 19.5k | return res; |
339 | 25.7k | } |
340 | | |
341 | | static int read_frame_size(Dav1dContext *const c, GetBits *const gb, |
342 | | const int use_ref) |
343 | 470k | { |
344 | 470k | const Dav1dSequenceHeader *const seqhdr = c->seq_hdr; |
345 | 470k | Dav1dFrameHeader *const hdr = c->frame_hdr; |
346 | | |
347 | 470k | if (use_ref) { |
348 | 167k | for (int i = 0; i < 7; i++) { |
349 | 165k | if (dav1d_get_bit(gb)) { |
350 | 85.2k | const Dav1dThreadPicture *const ref = |
351 | 85.2k | &c->refs[c->frame_hdr->refidx[i]].p; |
352 | 85.2k | if (!ref->p.frame_hdr) return -1; |
353 | 84.5k | hdr->width[1] = ref->p.frame_hdr->width[1]; |
354 | 84.5k | hdr->height = ref->p.frame_hdr->height; |
355 | 84.5k | hdr->render_width = ref->p.frame_hdr->render_width; |
356 | 84.5k | hdr->render_height = ref->p.frame_hdr->render_height; |
357 | 84.5k | hdr->super_res.enabled = seqhdr->super_res && dav1d_get_bit(gb); |
358 | 84.5k | if (hdr->super_res.enabled) { |
359 | 4.43k | const int d = hdr->super_res.width_scale_denominator = |
360 | 4.43k | 9 + dav1d_get_bits(gb, 3); |
361 | 4.43k | hdr->width[0] = imax((hdr->width[1] * 8 + (d >> 1)) / d, |
362 | 4.43k | imin(16, hdr->width[1])); |
363 | 80.0k | } else { |
364 | 80.0k | hdr->super_res.width_scale_denominator = 8; |
365 | 80.0k | hdr->width[0] = hdr->width[1]; |
366 | 80.0k | } |
367 | 84.5k | return 0; |
368 | 85.2k | } |
369 | 165k | } |
370 | 87.1k | } |
371 | | |
372 | 385k | if (hdr->frame_size_override) { |
373 | 17.9k | hdr->width[1] = dav1d_get_bits(gb, seqhdr->width_n_bits) + 1; |
374 | 17.9k | hdr->height = dav1d_get_bits(gb, seqhdr->height_n_bits) + 1; |
375 | 367k | } else { |
376 | 367k | hdr->width[1] = seqhdr->max_width; |
377 | 367k | hdr->height = seqhdr->max_height; |
378 | 367k | } |
379 | 385k | hdr->super_res.enabled = seqhdr->super_res && dav1d_get_bit(gb); |
380 | 385k | if (hdr->super_res.enabled) { |
381 | 60.7k | const int d = hdr->super_res.width_scale_denominator = 9 + dav1d_get_bits(gb, 3); |
382 | 60.7k | hdr->width[0] = imax((hdr->width[1] * 8 + (d >> 1)) / d, imin(16, hdr->width[1])); |
383 | 324k | } else { |
384 | 324k | hdr->super_res.width_scale_denominator = 8; |
385 | 324k | hdr->width[0] = hdr->width[1]; |
386 | 324k | } |
387 | 385k | hdr->have_render_size = dav1d_get_bit(gb); |
388 | 385k | if (hdr->have_render_size) { |
389 | 15.2k | hdr->render_width = dav1d_get_bits(gb, 16) + 1; |
390 | 15.2k | hdr->render_height = dav1d_get_bits(gb, 16) + 1; |
391 | 370k | } else { |
392 | 370k | hdr->render_width = hdr->width[1]; |
393 | 370k | hdr->render_height = hdr->height; |
394 | 370k | } |
395 | 385k | return 0; |
396 | 470k | } |
397 | | |
398 | 2.06M | static inline int tile_log2(const int sz, const int tgt) { |
399 | 2.06M | int k; |
400 | 3.03M | for (k = 0; (sz << k) < tgt; k++) ; |
401 | 2.06M | return k; |
402 | 2.06M | } |
403 | | |
404 | | static const Dav1dLoopfilterModeRefDeltas default_mode_ref_deltas = { |
405 | | .mode_delta = { 0, 0 }, |
406 | | .ref_delta = { 1, 0, 0, 0, -1, 0, -1, -1 }, |
407 | | }; |
408 | | |
409 | 514k | static int parse_frame_hdr(Dav1dContext *const c, GetBits *const gb) { |
410 | 514k | #define DEBUG_FRAME_HDR 0 |
411 | | |
412 | | #if DEBUG_FRAME_HDR |
413 | | const uint8_t *const init_ptr = gb->ptr; |
414 | | #endif |
415 | 514k | const Dav1dSequenceHeader *const seqhdr = c->seq_hdr; |
416 | 514k | Dav1dFrameHeader *const hdr = c->frame_hdr; |
417 | | |
418 | 514k | if (!seqhdr->reduced_still_picture_header) |
419 | 229k | hdr->show_existing_frame = dav1d_get_bit(gb); |
420 | | #if DEBUG_FRAME_HDR |
421 | | printf("HDR: post-show_existing_frame: off=%td\n", |
422 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
423 | | #endif |
424 | 514k | if (hdr->show_existing_frame) { |
425 | 38.0k | hdr->existing_frame_idx = dav1d_get_bits(gb, 3); |
426 | 38.0k | if (seqhdr->decoder_model_info_present && !seqhdr->equal_picture_interval) |
427 | 649 | hdr->frame_presentation_delay = dav1d_get_bits(gb, seqhdr->frame_presentation_delay_length); |
428 | 38.0k | if (seqhdr->frame_id_numbers_present) { |
429 | 2.47k | hdr->frame_id = dav1d_get_bits(gb, seqhdr->frame_id_n_bits); |
430 | 2.47k | Dav1dFrameHeader *const ref_frame_hdr = c->refs[hdr->existing_frame_idx].p.p.frame_hdr; |
431 | 2.47k | if (!ref_frame_hdr || ref_frame_hdr->frame_id != hdr->frame_id) goto error; |
432 | 2.47k | } |
433 | 36.8k | return 0; |
434 | 38.0k | } |
435 | | |
436 | 476k | if (seqhdr->reduced_still_picture_header) { |
437 | 284k | hdr->frame_type = DAV1D_FRAME_TYPE_KEY; |
438 | 284k | hdr->show_frame = 1; |
439 | 284k | } else { |
440 | 191k | hdr->frame_type = dav1d_get_bits(gb, 2); |
441 | 191k | hdr->show_frame = dav1d_get_bit(gb); |
442 | 191k | } |
443 | 476k | if (hdr->show_frame) { |
444 | 381k | if (seqhdr->decoder_model_info_present && !seqhdr->equal_picture_interval) |
445 | 13.7k | hdr->frame_presentation_delay = dav1d_get_bits(gb, seqhdr->frame_presentation_delay_length); |
446 | 381k | hdr->showable_frame = hdr->frame_type != DAV1D_FRAME_TYPE_KEY; |
447 | 381k | } else |
448 | 94.4k | hdr->showable_frame = dav1d_get_bit(gb); |
449 | 476k | hdr->error_resilient_mode = |
450 | 476k | (hdr->frame_type == DAV1D_FRAME_TYPE_KEY && hdr->show_frame) || |
451 | 170k | hdr->frame_type == DAV1D_FRAME_TYPE_SWITCH || |
452 | 166k | seqhdr->reduced_still_picture_header || dav1d_get_bit(gb); |
453 | | #if DEBUG_FRAME_HDR |
454 | | printf("HDR: post-frametype_bits: off=%td\n", |
455 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
456 | | #endif |
457 | 476k | hdr->disable_cdf_update = dav1d_get_bit(gb); |
458 | 476k | hdr->allow_screen_content_tools = seqhdr->screen_content_tools == DAV1D_ADAPTIVE ? |
459 | 416k | dav1d_get_bit(gb) : seqhdr->screen_content_tools; |
460 | 476k | if (hdr->allow_screen_content_tools) |
461 | 346k | hdr->force_integer_mv = seqhdr->force_integer_mv == DAV1D_ADAPTIVE ? |
462 | 268k | dav1d_get_bit(gb) : seqhdr->force_integer_mv; |
463 | | |
464 | 476k | if (IS_KEY_OR_INTRA(hdr)) |
465 | 312k | hdr->force_integer_mv = 1; |
466 | | |
467 | 476k | if (seqhdr->frame_id_numbers_present) |
468 | 5.48k | hdr->frame_id = dav1d_get_bits(gb, seqhdr->frame_id_n_bits); |
469 | | |
470 | 476k | if (!seqhdr->reduced_still_picture_header) |
471 | 191k | hdr->frame_size_override = hdr->frame_type == DAV1D_FRAME_TYPE_SWITCH ? 1 : dav1d_get_bit(gb); |
472 | | #if DEBUG_FRAME_HDR |
473 | | printf("HDR: post-frame_size_override_flag: off=%td\n", |
474 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
475 | | #endif |
476 | 476k | if (seqhdr->order_hint) |
477 | 163k | hdr->frame_offset = dav1d_get_bits(gb, seqhdr->order_hint_n_bits); |
478 | 476k | hdr->primary_ref_frame = !hdr->error_resilient_mode && IS_INTER_OR_SWITCH(hdr) ? |
479 | 318k | dav1d_get_bits(gb, 3) : DAV1D_PRIMARY_REF_NONE; |
480 | | |
481 | 476k | if (seqhdr->decoder_model_info_present) { |
482 | 14.9k | hdr->buffer_removal_time_present = dav1d_get_bit(gb); |
483 | 14.9k | if (hdr->buffer_removal_time_present) { |
484 | 12.0k | for (int i = 0; i < c->seq_hdr->num_operating_points; i++) { |
485 | 9.50k | const struct Dav1dSequenceHeaderOperatingPoint *const seqop = &seqhdr->operating_points[i]; |
486 | 9.50k | struct Dav1dFrameHeaderOperatingPoint *const op = &hdr->operating_points[i]; |
487 | 9.50k | if (seqop->decoder_model_param_present) { |
488 | 5.73k | int in_temporal_layer = (seqop->idc >> hdr->temporal_id) & 1; |
489 | 5.73k | int in_spatial_layer = (seqop->idc >> (hdr->spatial_id + 8)) & 1; |
490 | 5.73k | if (!seqop->idc || (in_temporal_layer && in_spatial_layer)) |
491 | 3.08k | op->buffer_removal_time = dav1d_get_bits(gb, seqhdr->buffer_removal_delay_length); |
492 | 5.73k | } |
493 | 9.50k | } |
494 | 2.52k | } |
495 | 14.9k | } |
496 | | |
497 | 476k | if (IS_KEY_OR_INTRA(hdr)) { |
498 | 312k | hdr->refresh_frame_flags = (hdr->frame_type == DAV1D_FRAME_TYPE_KEY && |
499 | 309k | hdr->show_frame) ? 0xff : dav1d_get_bits(gb, 8); |
500 | 312k | if (hdr->refresh_frame_flags != 0xff && hdr->error_resilient_mode && seqhdr->order_hint) |
501 | 11.1k | for (int i = 0; i < 8; i++) |
502 | 9.92k | dav1d_get_bits(gb, seqhdr->order_hint_n_bits); |
503 | 312k | if (c->strict_std_compliance && |
504 | 0 | hdr->frame_type == DAV1D_FRAME_TYPE_INTRA && hdr->refresh_frame_flags == 0xff) |
505 | 0 | { |
506 | 0 | goto error; |
507 | 0 | } |
508 | 312k | if (read_frame_size(c, gb, 0) < 0) goto error; |
509 | 312k | if (hdr->allow_screen_content_tools && !hdr->super_res.enabled) |
510 | 261k | hdr->allow_intrabc = dav1d_get_bit(gb); |
511 | 312k | } else { |
512 | 163k | hdr->refresh_frame_flags = hdr->frame_type == DAV1D_FRAME_TYPE_SWITCH ? 0xff : |
513 | 163k | dav1d_get_bits(gb, 8); |
514 | 163k | if (hdr->error_resilient_mode && seqhdr->order_hint) |
515 | 37.2k | for (int i = 0; i < 8; i++) |
516 | 33.1k | dav1d_get_bits(gb, seqhdr->order_hint_n_bits); |
517 | 163k | if (seqhdr->order_hint) { |
518 | 146k | hdr->frame_ref_short_signaling = dav1d_get_bit(gb); |
519 | 146k | if (hdr->frame_ref_short_signaling) { |
520 | 94.9k | hdr->refidx[0] = dav1d_get_bits(gb, 3); |
521 | 94.9k | hdr->refidx[1] = hdr->refidx[2] = -1; |
522 | 94.9k | hdr->refidx[3] = dav1d_get_bits(gb, 3); |
523 | | |
524 | | /* +1 allows for unconditional stores, as unused |
525 | | * values can be dumped into frame_offset[-1]. */ |
526 | 94.9k | int frame_offset_mem[8+1]; |
527 | 94.9k | int *const frame_offset = &frame_offset_mem[1]; |
528 | 94.9k | int earliest_ref = -1; |
529 | 847k | for (int i = 0, earliest_offset = INT_MAX; i < 8; i++) { |
530 | 753k | const Dav1dFrameHeader *const refhdr = c->refs[i].p.p.frame_hdr; |
531 | 753k | if (!refhdr) goto error; |
532 | 752k | const int diff = get_poc_diff(seqhdr->order_hint_n_bits, |
533 | 752k | refhdr->frame_offset, |
534 | 752k | hdr->frame_offset); |
535 | 752k | frame_offset[i] = diff; |
536 | 752k | if (diff < earliest_offset) { |
537 | 143k | earliest_offset = diff; |
538 | 143k | earliest_ref = i; |
539 | 143k | } |
540 | 752k | } |
541 | 93.8k | frame_offset[hdr->refidx[0]] = INT_MIN; // = reference frame is used |
542 | 93.8k | frame_offset[hdr->refidx[3]] = INT_MIN; |
543 | 93.8k | assert(earliest_ref >= 0); |
544 | | |
545 | 93.8k | int refidx = -1; |
546 | 844k | for (int i = 0, latest_offset = 0; i < 8; i++) { |
547 | 750k | const int hint = frame_offset[i]; |
548 | 750k | if (hint >= latest_offset) { |
549 | 330k | latest_offset = hint; |
550 | 330k | refidx = i; |
551 | 330k | } |
552 | 750k | } |
553 | 93.8k | frame_offset[refidx] = INT_MIN; |
554 | 93.8k | hdr->refidx[6] = refidx; |
555 | | |
556 | 281k | for (int i = 4; i < 6; i++) { |
557 | | /* Unsigned compares to handle negative values. */ |
558 | 187k | unsigned earliest_offset = UINT8_MAX; |
559 | 187k | refidx = -1; |
560 | 1.68M | for (int j = 0; j < 8; j++) { |
561 | 1.50M | const unsigned hint = frame_offset[j]; |
562 | 1.50M | if (hint < earliest_offset) { |
563 | 180k | earliest_offset = hint; |
564 | 180k | refidx = j; |
565 | 180k | } |
566 | 1.50M | } |
567 | 187k | frame_offset[refidx] = INT_MIN; |
568 | 187k | hdr->refidx[i] = refidx; |
569 | 187k | } |
570 | | |
571 | 657k | for (int i = 1; i < 7; i++) { |
572 | 563k | refidx = hdr->refidx[i]; |
573 | 563k | if (refidx < 0) { |
574 | 225k | unsigned latest_offset = ~UINT8_MAX; |
575 | 2.03M | for (int j = 0; j < 8; j++) { |
576 | 1.80M | const unsigned hint = frame_offset[j]; |
577 | 1.80M | if (hint >= latest_offset) { |
578 | 412k | latest_offset = hint; |
579 | 412k | refidx = j; |
580 | 412k | } |
581 | 1.80M | } |
582 | 225k | frame_offset[refidx] = INT_MIN; |
583 | 225k | hdr->refidx[i] = refidx >= 0 ? refidx : earliest_ref; |
584 | 225k | } |
585 | 563k | } |
586 | 93.8k | } |
587 | 146k | } |
588 | 1.26M | for (int i = 0; i < 7; i++) { |
589 | 1.11M | if (!hdr->frame_ref_short_signaling) |
590 | 454k | hdr->refidx[i] = dav1d_get_bits(gb, 3); |
591 | 1.11M | if (seqhdr->frame_id_numbers_present) { |
592 | 4.38k | const unsigned delta_ref_frame_id = dav1d_get_bits(gb, seqhdr->delta_frame_id_n_bits) + 1; |
593 | 4.38k | const unsigned ref_frame_id = (hdr->frame_id + (1 << seqhdr->frame_id_n_bits) - delta_ref_frame_id) & ((1 << seqhdr->frame_id_n_bits) - 1); |
594 | 4.38k | Dav1dFrameHeader *const ref_frame_hdr = c->refs[hdr->refidx[i]].p.p.frame_hdr; |
595 | 4.38k | if (!ref_frame_hdr || ref_frame_hdr->frame_id != ref_frame_id) goto error; |
596 | 4.38k | } |
597 | 1.11M | } |
598 | 158k | const int use_ref = !hdr->error_resilient_mode && |
599 | 153k | hdr->frame_size_override; |
600 | 158k | if (read_frame_size(c, gb, use_ref) < 0) goto error; |
601 | 157k | if (!hdr->force_integer_mv) |
602 | 94.4k | hdr->hp = dav1d_get_bit(gb); |
603 | 157k | hdr->subpel_filter_mode = dav1d_get_bit(gb) ? DAV1D_FILTER_SWITCHABLE : |
604 | 157k | dav1d_get_bits(gb, 2); |
605 | 157k | hdr->switchable_motion_mode = dav1d_get_bit(gb); |
606 | 157k | if (!hdr->error_resilient_mode && seqhdr->ref_frame_mvs && |
607 | 125k | seqhdr->order_hint && IS_INTER_OR_SWITCH(hdr)) |
608 | 125k | { |
609 | 125k | hdr->use_ref_frame_mvs = dav1d_get_bit(gb); |
610 | 125k | } |
611 | 157k | } |
612 | | #if DEBUG_FRAME_HDR |
613 | | printf("HDR: post-frametype-specific-bits: off=%td\n", |
614 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
615 | | #endif |
616 | | |
617 | 470k | if (!seqhdr->reduced_still_picture_header && !hdr->disable_cdf_update) |
618 | 171k | hdr->refresh_context = !dav1d_get_bit(gb); |
619 | | #if DEBUG_FRAME_HDR |
620 | | printf("HDR: post-refresh_context: off=%td\n", |
621 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
622 | | #endif |
623 | | |
624 | | // tile data |
625 | 470k | hdr->tiling.uniform = dav1d_get_bit(gb); |
626 | 470k | const int sbsz_min1 = (64 << seqhdr->sb128) - 1; |
627 | 470k | const int sbsz_log2 = 6 + seqhdr->sb128; |
628 | 470k | const int sbw = (hdr->width[0] + sbsz_min1) >> sbsz_log2; |
629 | 470k | const int sbh = (hdr->height + sbsz_min1) >> sbsz_log2; |
630 | 470k | const int max_tile_width_sb = 4096 >> sbsz_log2; |
631 | 470k | const int max_tile_area_sb = 4096 * 2304 >> (2 * sbsz_log2); |
632 | 470k | hdr->tiling.min_log2_cols = tile_log2(max_tile_width_sb, sbw); |
633 | 470k | hdr->tiling.max_log2_cols = tile_log2(1, imin(sbw, DAV1D_MAX_TILE_COLS)); |
634 | 470k | hdr->tiling.max_log2_rows = tile_log2(1, imin(sbh, DAV1D_MAX_TILE_ROWS)); |
635 | 470k | const int min_log2_tiles = imax(tile_log2(max_tile_area_sb, sbw * sbh), |
636 | 470k | hdr->tiling.min_log2_cols); |
637 | 470k | if (hdr->tiling.uniform) { |
638 | 378k | for (hdr->tiling.log2_cols = hdr->tiling.min_log2_cols; |
639 | 387k | hdr->tiling.log2_cols < hdr->tiling.max_log2_cols && dav1d_get_bit(gb); |
640 | 378k | hdr->tiling.log2_cols++) ; |
641 | 378k | const int tile_w = 1 + ((sbw - 1) >> hdr->tiling.log2_cols); |
642 | 378k | hdr->tiling.cols = 0; |
643 | 790k | for (int sbx = 0; sbx < sbw; sbx += tile_w, hdr->tiling.cols++) |
644 | 411k | hdr->tiling.col_start_sb[hdr->tiling.cols] = sbx; |
645 | 378k | hdr->tiling.min_log2_rows = |
646 | 378k | imax(min_log2_tiles - hdr->tiling.log2_cols, 0); |
647 | | |
648 | 378k | for (hdr->tiling.log2_rows = hdr->tiling.min_log2_rows; |
649 | 393k | hdr->tiling.log2_rows < hdr->tiling.max_log2_rows && dav1d_get_bit(gb); |
650 | 378k | hdr->tiling.log2_rows++) ; |
651 | 378k | const int tile_h = 1 + ((sbh - 1) >> hdr->tiling.log2_rows); |
652 | 378k | hdr->tiling.rows = 0; |
653 | 810k | for (int sby = 0; sby < sbh; sby += tile_h, hdr->tiling.rows++) |
654 | 432k | hdr->tiling.row_start_sb[hdr->tiling.rows] = sby; |
655 | 378k | } else { |
656 | 91.7k | hdr->tiling.cols = 0; |
657 | 91.7k | int widest_tile = 0, max_tile_area_sb = sbw * sbh; |
658 | 332k | for (int sbx = 0; sbx < sbw && hdr->tiling.cols < DAV1D_MAX_TILE_COLS; hdr->tiling.cols++) { |
659 | 241k | const int tile_width_sb = imin(sbw - sbx, max_tile_width_sb); |
660 | 241k | const int tile_w = (tile_width_sb > 1) ? 1 + dav1d_get_uniform(gb, tile_width_sb) : 1; |
661 | 241k | hdr->tiling.col_start_sb[hdr->tiling.cols] = sbx; |
662 | 241k | sbx += tile_w; |
663 | 241k | widest_tile = imax(widest_tile, tile_w); |
664 | 241k | } |
665 | 91.7k | hdr->tiling.log2_cols = tile_log2(1, hdr->tiling.cols); |
666 | 91.7k | if (min_log2_tiles) max_tile_area_sb >>= min_log2_tiles + 1; |
667 | 91.7k | const int max_tile_height_sb = imax(max_tile_area_sb / widest_tile, 1); |
668 | | |
669 | 91.7k | hdr->tiling.rows = 0; |
670 | 321k | for (int sby = 0; sby < sbh && hdr->tiling.rows < DAV1D_MAX_TILE_ROWS; hdr->tiling.rows++) { |
671 | 229k | const int tile_height_sb = imin(sbh - sby, max_tile_height_sb); |
672 | 229k | const int tile_h = (tile_height_sb > 1) ? 1 + dav1d_get_uniform(gb, tile_height_sb) : 1; |
673 | 229k | hdr->tiling.row_start_sb[hdr->tiling.rows] = sby; |
674 | 229k | sby += tile_h; |
675 | 229k | } |
676 | 91.7k | hdr->tiling.log2_rows = tile_log2(1, hdr->tiling.rows); |
677 | 91.7k | } |
678 | 470k | hdr->tiling.col_start_sb[hdr->tiling.cols] = sbw; |
679 | 470k | hdr->tiling.row_start_sb[hdr->tiling.rows] = sbh; |
680 | 470k | if (hdr->tiling.log2_cols || hdr->tiling.log2_rows) { |
681 | 42.0k | hdr->tiling.update = dav1d_get_bits(gb, hdr->tiling.log2_cols + hdr->tiling.log2_rows); |
682 | 42.0k | if (hdr->tiling.update >= hdr->tiling.cols * hdr->tiling.rows) |
683 | 736 | goto error; |
684 | 41.3k | hdr->tiling.n_bytes = dav1d_get_bits(gb, 2) + 1; |
685 | 41.3k | } |
686 | | #if DEBUG_FRAME_HDR |
687 | | printf("HDR: post-tiling: off=%td\n", |
688 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
689 | | #endif |
690 | | |
691 | | // quant data |
692 | 469k | hdr->quant.yac = dav1d_get_bits(gb, 8); |
693 | 469k | if (dav1d_get_bit(gb)) |
694 | 41.6k | hdr->quant.ydc_delta = dav1d_get_sbits(gb, 7); |
695 | 469k | if (!seqhdr->monochrome) { |
696 | | // If the sequence header says that delta_q might be different |
697 | | // for U, V, we must check whether it actually is for this |
698 | | // frame. |
699 | 352k | const int diff_uv_delta = seqhdr->separate_uv_delta_q ? dav1d_get_bit(gb) : 0; |
700 | 352k | if (dav1d_get_bit(gb)) |
701 | 23.4k | hdr->quant.udc_delta = dav1d_get_sbits(gb, 7); |
702 | 352k | if (dav1d_get_bit(gb)) |
703 | 20.4k | hdr->quant.uac_delta = dav1d_get_sbits(gb, 7); |
704 | 352k | if (diff_uv_delta) { |
705 | 9.66k | if (dav1d_get_bit(gb)) |
706 | 2.38k | hdr->quant.vdc_delta = dav1d_get_sbits(gb, 7); |
707 | 9.66k | if (dav1d_get_bit(gb)) |
708 | 4.08k | hdr->quant.vac_delta = dav1d_get_sbits(gb, 7); |
709 | 343k | } else { |
710 | 343k | hdr->quant.vdc_delta = hdr->quant.udc_delta; |
711 | 343k | hdr->quant.vac_delta = hdr->quant.uac_delta; |
712 | 343k | } |
713 | 352k | } |
714 | | #if DEBUG_FRAME_HDR |
715 | | printf("HDR: post-quant: off=%td\n", |
716 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
717 | | #endif |
718 | 469k | hdr->quant.qm = dav1d_get_bit(gb); |
719 | 469k | if (hdr->quant.qm) { |
720 | 38.8k | hdr->quant.qm_y = dav1d_get_bits(gb, 4); |
721 | 38.8k | hdr->quant.qm_u = dav1d_get_bits(gb, 4); |
722 | 38.8k | hdr->quant.qm_v = seqhdr->separate_uv_delta_q ? dav1d_get_bits(gb, 4) : |
723 | 38.8k | hdr->quant.qm_u; |
724 | 38.8k | } |
725 | | #if DEBUG_FRAME_HDR |
726 | | printf("HDR: post-qm: off=%td\n", |
727 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
728 | | #endif |
729 | | |
730 | | // segmentation data |
731 | 469k | hdr->segmentation.enabled = dav1d_get_bit(gb); |
732 | 469k | if (hdr->segmentation.enabled) { |
733 | 32.4k | if (hdr->primary_ref_frame == DAV1D_PRIMARY_REF_NONE) { |
734 | 10.0k | hdr->segmentation.update_map = 1; |
735 | 10.0k | hdr->segmentation.update_data = 1; |
736 | 22.4k | } else { |
737 | 22.4k | hdr->segmentation.update_map = dav1d_get_bit(gb); |
738 | 22.4k | if (hdr->segmentation.update_map) |
739 | 10.2k | hdr->segmentation.temporal = dav1d_get_bit(gb); |
740 | 22.4k | hdr->segmentation.update_data = dav1d_get_bit(gb); |
741 | 22.4k | } |
742 | | |
743 | 32.4k | if (hdr->segmentation.update_data) { |
744 | 12.0k | hdr->segmentation.seg_data.last_active_segid = -1; |
745 | 108k | for (int i = 0; i < DAV1D_MAX_SEGMENTS; i++) { |
746 | 96.3k | Dav1dSegmentationData *const seg = |
747 | 96.3k | &hdr->segmentation.seg_data.d[i]; |
748 | 96.3k | if (dav1d_get_bit(gb)) { |
749 | 19.4k | seg->delta_q = dav1d_get_sbits(gb, 9); |
750 | 19.4k | hdr->segmentation.seg_data.last_active_segid = i; |
751 | 19.4k | } |
752 | 96.3k | if (dav1d_get_bit(gb)) { |
753 | 15.7k | seg->delta_lf_y_v = dav1d_get_sbits(gb, 7); |
754 | 15.7k | hdr->segmentation.seg_data.last_active_segid = i; |
755 | 15.7k | } |
756 | 96.3k | if (dav1d_get_bit(gb)) { |
757 | 20.4k | seg->delta_lf_y_h = dav1d_get_sbits(gb, 7); |
758 | 20.4k | hdr->segmentation.seg_data.last_active_segid = i; |
759 | 20.4k | } |
760 | 96.3k | if (dav1d_get_bit(gb)) { |
761 | 19.5k | seg->delta_lf_u = dav1d_get_sbits(gb, 7); |
762 | 19.5k | hdr->segmentation.seg_data.last_active_segid = i; |
763 | 19.5k | } |
764 | 96.3k | if (dav1d_get_bit(gb)) { |
765 | 16.7k | seg->delta_lf_v = dav1d_get_sbits(gb, 7); |
766 | 16.7k | hdr->segmentation.seg_data.last_active_segid = i; |
767 | 16.7k | } |
768 | 96.3k | if (dav1d_get_bit(gb)) { |
769 | 17.7k | seg->ref = dav1d_get_bits(gb, 3); |
770 | 17.7k | hdr->segmentation.seg_data.last_active_segid = i; |
771 | 17.7k | hdr->segmentation.seg_data.preskip = 1; |
772 | 78.5k | } else { |
773 | 78.5k | seg->ref = -1; |
774 | 78.5k | } |
775 | 96.3k | if ((seg->skip = dav1d_get_bit(gb))) { |
776 | 18.4k | hdr->segmentation.seg_data.last_active_segid = i; |
777 | 18.4k | hdr->segmentation.seg_data.preskip = 1; |
778 | 18.4k | } |
779 | 96.3k | if ((seg->globalmv = dav1d_get_bit(gb))) { |
780 | 16.9k | hdr->segmentation.seg_data.last_active_segid = i; |
781 | 16.9k | hdr->segmentation.seg_data.preskip = 1; |
782 | 16.9k | } |
783 | 96.3k | } |
784 | 20.4k | } else { |
785 | | // segmentation.update_data was false so we should copy |
786 | | // segmentation data from the reference frame. |
787 | 20.4k | assert(hdr->primary_ref_frame != DAV1D_PRIMARY_REF_NONE); |
788 | 20.4k | const int pri_ref = hdr->refidx[hdr->primary_ref_frame]; |
789 | 20.4k | if (!c->refs[pri_ref].p.p.frame_hdr) goto error; |
790 | 20.1k | hdr->segmentation.seg_data = |
791 | 20.1k | c->refs[pri_ref].p.p.frame_hdr->segmentation.seg_data; |
792 | 20.1k | } |
793 | 436k | } else { |
794 | 3.93M | for (int i = 0; i < DAV1D_MAX_SEGMENTS; i++) |
795 | 3.49M | hdr->segmentation.seg_data.d[i].ref = -1; |
796 | 436k | } |
797 | | #if DEBUG_FRAME_HDR |
798 | | printf("HDR: post-segmentation: off=%td\n", |
799 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
800 | | #endif |
801 | | |
802 | | // delta q |
803 | 469k | if (hdr->quant.yac) { |
804 | 404k | hdr->delta.q.present = dav1d_get_bit(gb); |
805 | 404k | if (hdr->delta.q.present) { |
806 | 73.0k | hdr->delta.q.res_log2 = dav1d_get_bits(gb, 2); |
807 | 73.0k | if (!hdr->allow_intrabc) { |
808 | 20.0k | hdr->delta.lf.present = dav1d_get_bit(gb); |
809 | 20.0k | if (hdr->delta.lf.present) { |
810 | 10.1k | hdr->delta.lf.res_log2 = dav1d_get_bits(gb, 2); |
811 | 10.1k | hdr->delta.lf.multi = dav1d_get_bit(gb); |
812 | 10.1k | } |
813 | 20.0k | } |
814 | 73.0k | } |
815 | 404k | } |
816 | | #if DEBUG_FRAME_HDR |
817 | | printf("HDR: post-delta_q_lf_flags: off=%td\n", |
818 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
819 | | #endif |
820 | | |
821 | | // derive lossless flags |
822 | 469k | const int delta_lossless = !hdr->quant.ydc_delta && !hdr->quant.udc_delta && |
823 | 418k | !hdr->quant.uac_delta && !hdr->quant.vdc_delta && !hdr->quant.vac_delta; |
824 | 469k | hdr->all_lossless = 1; |
825 | 4.22M | for (int i = 0; i < DAV1D_MAX_SEGMENTS; i++) { |
826 | 3.75M | hdr->segmentation.qidx[i] = hdr->segmentation.enabled ? |
827 | 257k | iclip_u8(hdr->quant.yac + hdr->segmentation.seg_data.d[i].delta_q) : |
828 | 3.75M | hdr->quant.yac; |
829 | 3.75M | hdr->segmentation.lossless[i] = |
830 | 3.75M | !hdr->segmentation.qidx[i] && delta_lossless; |
831 | 3.75M | hdr->all_lossless &= hdr->segmentation.lossless[i]; |
832 | 3.75M | } |
833 | | |
834 | | // loopfilter |
835 | 469k | if (hdr->all_lossless || hdr->allow_intrabc) { |
836 | 308k | hdr->loopfilter.mode_ref_delta_enabled = 1; |
837 | 308k | hdr->loopfilter.mode_ref_delta_update = 1; |
838 | 308k | hdr->loopfilter.mode_ref_deltas = default_mode_ref_deltas; |
839 | 308k | } else { |
840 | 160k | hdr->loopfilter.level_y[0] = dav1d_get_bits(gb, 6); |
841 | 160k | hdr->loopfilter.level_y[1] = dav1d_get_bits(gb, 6); |
842 | 160k | if (!seqhdr->monochrome && |
843 | 65.3k | (hdr->loopfilter.level_y[0] || hdr->loopfilter.level_y[1])) |
844 | 37.3k | { |
845 | 37.3k | hdr->loopfilter.level_u = dav1d_get_bits(gb, 6); |
846 | 37.3k | hdr->loopfilter.level_v = dav1d_get_bits(gb, 6); |
847 | 37.3k | } |
848 | 160k | hdr->loopfilter.sharpness = dav1d_get_bits(gb, 3); |
849 | | |
850 | 160k | if (hdr->primary_ref_frame == DAV1D_PRIMARY_REF_NONE) { |
851 | 73.4k | hdr->loopfilter.mode_ref_deltas = default_mode_ref_deltas; |
852 | 86.7k | } else { |
853 | 86.7k | const int ref = hdr->refidx[hdr->primary_ref_frame]; |
854 | 86.7k | if (!c->refs[ref].p.p.frame_hdr) goto error; |
855 | 85.3k | hdr->loopfilter.mode_ref_deltas = |
856 | 85.3k | c->refs[ref].p.p.frame_hdr->loopfilter.mode_ref_deltas; |
857 | 85.3k | } |
858 | 158k | hdr->loopfilter.mode_ref_delta_enabled = dav1d_get_bit(gb); |
859 | 158k | if (hdr->loopfilter.mode_ref_delta_enabled) { |
860 | 92.4k | hdr->loopfilter.mode_ref_delta_update = dav1d_get_bit(gb); |
861 | 92.4k | if (hdr->loopfilter.mode_ref_delta_update) { |
862 | 108k | for (int i = 0; i < 8; i++) |
863 | 96.3k | if (dav1d_get_bit(gb)) |
864 | 26.9k | hdr->loopfilter.mode_ref_deltas.ref_delta[i] = |
865 | 26.9k | dav1d_get_sbits(gb, 7); |
866 | 36.1k | for (int i = 0; i < 2; i++) |
867 | 24.0k | if (dav1d_get_bit(gb)) |
868 | 4.56k | hdr->loopfilter.mode_ref_deltas.mode_delta[i] = |
869 | 4.56k | dav1d_get_sbits(gb, 7); |
870 | 12.0k | } |
871 | 92.4k | } |
872 | 158k | } |
873 | | #if DEBUG_FRAME_HDR |
874 | | printf("HDR: post-lpf: off=%td\n", |
875 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
876 | | #endif |
877 | | |
878 | | // cdef |
879 | 467k | if (!hdr->all_lossless && seqhdr->cdef && !hdr->allow_intrabc) { |
880 | 90.8k | hdr->cdef.damping = dav1d_get_bits(gb, 2) + 3; |
881 | 90.8k | hdr->cdef.n_bits = dav1d_get_bits(gb, 2); |
882 | 236k | for (int i = 0; i < (1 << hdr->cdef.n_bits); i++) { |
883 | 145k | hdr->cdef.y_strength[i] = dav1d_get_bits(gb, 6); |
884 | 145k | if (!seqhdr->monochrome) |
885 | 71.6k | hdr->cdef.uv_strength[i] = dav1d_get_bits(gb, 6); |
886 | 145k | } |
887 | 90.8k | } |
888 | | #if DEBUG_FRAME_HDR |
889 | | printf("HDR: post-cdef: off=%td\n", |
890 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
891 | | #endif |
892 | | |
893 | | // restoration |
894 | 467k | if ((!hdr->all_lossless || hdr->super_res.enabled) && |
895 | 444k | seqhdr->restoration && !hdr->allow_intrabc) |
896 | 131k | { |
897 | 131k | hdr->restoration.type[0] = dav1d_get_bits(gb, 2); |
898 | 131k | if (!seqhdr->monochrome) { |
899 | 53.7k | hdr->restoration.type[1] = dav1d_get_bits(gb, 2); |
900 | 53.7k | hdr->restoration.type[2] = dav1d_get_bits(gb, 2); |
901 | 53.7k | } |
902 | | |
903 | 131k | if (hdr->restoration.type[0] || hdr->restoration.type[1] || |
904 | 40.8k | hdr->restoration.type[2]) |
905 | 91.2k | { |
906 | | // Log2 of the restoration unit size. |
907 | 91.2k | hdr->restoration.unit_size[0] = 6 + seqhdr->sb128; |
908 | 91.2k | if (dav1d_get_bit(gb)) { |
909 | 22.0k | hdr->restoration.unit_size[0]++; |
910 | 22.0k | if (!seqhdr->sb128) |
911 | 4.51k | hdr->restoration.unit_size[0] += dav1d_get_bit(gb); |
912 | 22.0k | } |
913 | 91.2k | hdr->restoration.unit_size[1] = hdr->restoration.unit_size[0]; |
914 | 91.2k | if ((hdr->restoration.type[1] || hdr->restoration.type[2]) && |
915 | 23.0k | seqhdr->ss_hor == 1 && seqhdr->ss_ver == 1) |
916 | 9.17k | { |
917 | 9.17k | hdr->restoration.unit_size[1] -= dav1d_get_bit(gb); |
918 | 9.17k | } |
919 | 91.2k | } else { |
920 | 39.9k | hdr->restoration.unit_size[0] = 8; |
921 | 39.9k | } |
922 | 131k | } |
923 | | #if DEBUG_FRAME_HDR |
924 | | printf("HDR: post-restoration: off=%td\n", |
925 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
926 | | #endif |
927 | | |
928 | 467k | if (!hdr->all_lossless) |
929 | 411k | hdr->txfm_mode = dav1d_get_bit(gb) ? DAV1D_TX_SWITCHABLE : DAV1D_TX_LARGEST; |
930 | | #if DEBUG_FRAME_HDR |
931 | | printf("HDR: post-txfmmode: off=%td\n", |
932 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
933 | | #endif |
934 | 467k | if (IS_INTER_OR_SWITCH(hdr)) |
935 | 155k | hdr->switchable_comp_refs = dav1d_get_bit(gb); |
936 | | #if DEBUG_FRAME_HDR |
937 | | printf("HDR: post-refmode: off=%td\n", |
938 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
939 | | #endif |
940 | 467k | if (hdr->switchable_comp_refs && IS_INTER_OR_SWITCH(hdr) && seqhdr->order_hint) { |
941 | 82.0k | const int poc = hdr->frame_offset; |
942 | 82.0k | int off_before = -1, off_after = -1; |
943 | 82.0k | int off_before_idx, off_after_idx; |
944 | 653k | for (int i = 0; i < 7; i++) { |
945 | 572k | if (!c->refs[hdr->refidx[i]].p.p.frame_hdr) goto error; |
946 | 571k | const int refpoc = c->refs[hdr->refidx[i]].p.p.frame_hdr->frame_offset; |
947 | | |
948 | 571k | const int diff = get_poc_diff(seqhdr->order_hint_n_bits, refpoc, poc); |
949 | 571k | if (diff > 0) { |
950 | 224k | if (off_after < 0 || get_poc_diff(seqhdr->order_hint_n_bits, |
951 | 158k | off_after, refpoc) > 0) |
952 | 77.6k | { |
953 | 77.6k | off_after = refpoc; |
954 | 77.6k | off_after_idx = i; |
955 | 77.6k | } |
956 | 347k | } else if (diff < 0 && (off_before < 0 || |
957 | 173k | get_poc_diff(seqhdr->order_hint_n_bits, |
958 | 173k | refpoc, off_before) > 0)) |
959 | 80.9k | { |
960 | 80.9k | off_before = refpoc; |
961 | 80.9k | off_before_idx = i; |
962 | 80.9k | } |
963 | 571k | } |
964 | | |
965 | 81.6k | if ((off_before | off_after) >= 0) { |
966 | 56.5k | hdr->skip_mode_refs[0] = imin(off_before_idx, off_after_idx); |
967 | 56.5k | hdr->skip_mode_refs[1] = imax(off_before_idx, off_after_idx); |
968 | 56.5k | hdr->skip_mode_allowed = 1; |
969 | 56.5k | } else if (off_before >= 0) { |
970 | 14.0k | int off_before2 = -1; |
971 | 14.0k | int off_before2_idx; |
972 | 112k | for (int i = 0; i < 7; i++) { |
973 | 98.0k | if (!c->refs[hdr->refidx[i]].p.p.frame_hdr) goto error; |
974 | 98.0k | const int refpoc = c->refs[hdr->refidx[i]].p.p.frame_hdr->frame_offset; |
975 | 98.0k | if (get_poc_diff(seqhdr->order_hint_n_bits, |
976 | 98.0k | refpoc, off_before) < 0) { |
977 | 22.5k | if (off_before2 < 0 || get_poc_diff(seqhdr->order_hint_n_bits, |
978 | 16.2k | refpoc, off_before2) > 0) |
979 | 6.70k | { |
980 | 6.70k | off_before2 = refpoc; |
981 | 6.70k | off_before2_idx = i; |
982 | 6.70k | } |
983 | 22.5k | } |
984 | 98.0k | } |
985 | | |
986 | 14.0k | if (off_before2 >= 0) { |
987 | 6.28k | hdr->skip_mode_refs[0] = imin(off_before_idx, off_before2_idx); |
988 | 6.28k | hdr->skip_mode_refs[1] = imax(off_before_idx, off_before2_idx); |
989 | 6.28k | hdr->skip_mode_allowed = 1; |
990 | 6.28k | } |
991 | 14.0k | } |
992 | 81.6k | } |
993 | 467k | if (hdr->skip_mode_allowed) |
994 | 62.8k | hdr->skip_mode_enabled = dav1d_get_bit(gb); |
995 | | #if DEBUG_FRAME_HDR |
996 | | printf("HDR: post-extskip: off=%td\n", |
997 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
998 | | #endif |
999 | 467k | if (!hdr->error_resilient_mode && IS_INTER_OR_SWITCH(hdr) && seqhdr->warped_motion) |
1000 | 125k | hdr->warp_motion = dav1d_get_bit(gb); |
1001 | | #if DEBUG_FRAME_HDR |
1002 | | printf("HDR: post-warpmotionbit: off=%td\n", |
1003 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
1004 | | #endif |
1005 | 467k | hdr->reduced_txtp_set = dav1d_get_bit(gb); |
1006 | | #if DEBUG_FRAME_HDR |
1007 | | printf("HDR: post-reducedtxtpset: off=%td\n", |
1008 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
1009 | | #endif |
1010 | | |
1011 | 3.73M | for (int i = 0; i < 7; i++) |
1012 | 3.27M | hdr->gmv[i] = dav1d_default_wm_params; |
1013 | | |
1014 | 467k | if (IS_INTER_OR_SWITCH(hdr)) { |
1015 | 1.24M | for (int i = 0; i < 7; i++) { |
1016 | 1.08M | hdr->gmv[i].type = !dav1d_get_bit(gb) ? DAV1D_WM_TYPE_IDENTITY : |
1017 | 1.08M | dav1d_get_bit(gb) ? DAV1D_WM_TYPE_ROT_ZOOM : |
1018 | 59.4k | dav1d_get_bit(gb) ? DAV1D_WM_TYPE_TRANSLATION : |
1019 | 27.0k | DAV1D_WM_TYPE_AFFINE; |
1020 | | |
1021 | 1.08M | if (hdr->gmv[i].type == DAV1D_WM_TYPE_IDENTITY) continue; |
1022 | | |
1023 | 59.4k | const Dav1dWarpedMotionParams *ref_gmv; |
1024 | 59.4k | if (hdr->primary_ref_frame == DAV1D_PRIMARY_REF_NONE) { |
1025 | 4.34k | ref_gmv = &dav1d_default_wm_params; |
1026 | 55.0k | } else { |
1027 | 55.0k | const int pri_ref = hdr->refidx[hdr->primary_ref_frame]; |
1028 | 55.0k | if (!c->refs[pri_ref].p.p.frame_hdr) goto error; |
1029 | 54.5k | ref_gmv = &c->refs[pri_ref].p.p.frame_hdr->gmv[i]; |
1030 | 54.5k | } |
1031 | 58.9k | int32_t *const mat = hdr->gmv[i].matrix; |
1032 | 58.9k | const int32_t *const ref_mat = ref_gmv->matrix; |
1033 | 58.9k | int bits, shift; |
1034 | | |
1035 | 58.9k | if (hdr->gmv[i].type >= DAV1D_WM_TYPE_ROT_ZOOM) { |
1036 | 51.1k | mat[2] = (1 << 16) + 2 * |
1037 | 51.1k | dav1d_get_bits_subexp(gb, (ref_mat[2] - (1 << 16)) >> 1, 12); |
1038 | 51.1k | mat[3] = 2 * dav1d_get_bits_subexp(gb, ref_mat[3] >> 1, 12); |
1039 | | |
1040 | 51.1k | bits = 12; |
1041 | 51.1k | shift = 10; |
1042 | 51.1k | } else { |
1043 | 7.76k | bits = 9 - !hdr->hp; |
1044 | 7.76k | shift = 13 + !hdr->hp; |
1045 | 7.76k | } |
1046 | | |
1047 | 58.9k | if (hdr->gmv[i].type == DAV1D_WM_TYPE_AFFINE) { |
1048 | 18.8k | mat[4] = 2 * dav1d_get_bits_subexp(gb, ref_mat[4] >> 1, 12); |
1049 | 18.8k | mat[5] = (1 << 16) + 2 * |
1050 | 18.8k | dav1d_get_bits_subexp(gb, (ref_mat[5] - (1 << 16)) >> 1, 12); |
1051 | 40.0k | } else { |
1052 | 40.0k | mat[4] = -mat[3]; |
1053 | 40.0k | mat[5] = mat[2]; |
1054 | 40.0k | } |
1055 | | |
1056 | 58.9k | mat[0] = dav1d_get_bits_subexp(gb, ref_mat[0] >> shift, bits) * (1 << shift); |
1057 | 58.9k | mat[1] = dav1d_get_bits_subexp(gb, ref_mat[1] >> shift, bits) * (1 << shift); |
1058 | 58.9k | } |
1059 | 155k | } |
1060 | | #if DEBUG_FRAME_HDR |
1061 | | printf("HDR: post-gmv: off=%td\n", |
1062 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
1063 | | #endif |
1064 | | |
1065 | 466k | if (seqhdr->film_grain_present && (hdr->show_frame || hdr->showable_frame)) { |
1066 | 231k | hdr->film_grain.present = dav1d_get_bit(gb); |
1067 | 231k | if (hdr->film_grain.present) { |
1068 | 30.1k | const unsigned seed = dav1d_get_bits(gb, 16); |
1069 | 30.1k | hdr->film_grain.update = hdr->frame_type != DAV1D_FRAME_TYPE_INTER || dav1d_get_bit(gb); |
1070 | 30.1k | if (!hdr->film_grain.update) { |
1071 | 21.3k | const int refidx = dav1d_get_bits(gb, 3); |
1072 | 21.3k | int i; |
1073 | 54.0k | for (i = 0; i < 7; i++) |
1074 | 53.5k | if (hdr->refidx[i] == refidx) |
1075 | 20.8k | break; |
1076 | 21.3k | if (i == 7 || !c->refs[refidx].p.p.frame_hdr) goto error; |
1077 | 20.6k | hdr->film_grain.data = c->refs[refidx].p.p.frame_hdr->film_grain.data; |
1078 | 20.6k | hdr->film_grain.data.seed = seed; |
1079 | 20.6k | } else { |
1080 | 8.88k | Dav1dFilmGrainData *const fgd = &hdr->film_grain.data; |
1081 | 8.88k | fgd->seed = seed; |
1082 | | |
1083 | 8.88k | fgd->num_y_points = dav1d_get_bits(gb, 4); |
1084 | 8.88k | if (fgd->num_y_points > 14) goto error; |
1085 | 14.9k | for (int i = 0; i < fgd->num_y_points; i++) { |
1086 | 6.98k | fgd->y_points[i][0] = dav1d_get_bits(gb, 8); |
1087 | 6.98k | if (i && fgd->y_points[i - 1][0] >= fgd->y_points[i][0]) |
1088 | 718 | goto error; |
1089 | 6.26k | fgd->y_points[i][1] = dav1d_get_bits(gb, 8); |
1090 | 6.26k | } |
1091 | | |
1092 | 7.96k | if (!seqhdr->monochrome) |
1093 | 7.27k | fgd->chroma_scaling_from_luma = dav1d_get_bit(gb); |
1094 | 7.96k | if (seqhdr->monochrome || fgd->chroma_scaling_from_luma || |
1095 | 4.73k | (seqhdr->ss_ver == 1 && seqhdr->ss_hor == 1 && !fgd->num_y_points)) |
1096 | 3.70k | { |
1097 | 3.70k | fgd->num_uv_points[0] = fgd->num_uv_points[1] = 0; |
1098 | 10.2k | } else for (int pl = 0; pl < 2; pl++) { |
1099 | 7.47k | fgd->num_uv_points[pl] = dav1d_get_bits(gb, 4); |
1100 | 7.47k | if (fgd->num_uv_points[pl] > 10) goto error; |
1101 | 15.1k | for (int i = 0; i < fgd->num_uv_points[pl]; i++) { |
1102 | 9.21k | fgd->uv_points[pl][i][0] = dav1d_get_bits(gb, 8); |
1103 | 9.21k | if (i && fgd->uv_points[pl][i - 1][0] >= fgd->uv_points[pl][i][0]) |
1104 | 862 | goto error; |
1105 | 8.34k | fgd->uv_points[pl][i][1] = dav1d_get_bits(gb, 8); |
1106 | 8.34k | } |
1107 | 6.82k | } |
1108 | | |
1109 | 6.45k | if (seqhdr->ss_hor == 1 && seqhdr->ss_ver == 1 && |
1110 | 3.10k | !!fgd->num_uv_points[0] != !!fgd->num_uv_points[1]) |
1111 | 45 | { |
1112 | 45 | goto error; |
1113 | 45 | } |
1114 | | |
1115 | 6.40k | fgd->scaling_shift = dav1d_get_bits(gb, 2) + 8; |
1116 | 6.40k | fgd->ar_coeff_lag = dav1d_get_bits(gb, 2); |
1117 | 6.40k | const int num_y_pos = 2 * fgd->ar_coeff_lag * (fgd->ar_coeff_lag + 1); |
1118 | 6.40k | if (fgd->num_y_points) |
1119 | 23.3k | for (int i = 0; i < num_y_pos; i++) |
1120 | 21.1k | fgd->ar_coeffs_y[i] = dav1d_get_bits(gb, 8) - 128; |
1121 | 19.2k | for (int pl = 0; pl < 2; pl++) |
1122 | 12.8k | if (fgd->num_uv_points[pl] || fgd->chroma_scaling_from_luma) { |
1123 | 7.20k | const int num_uv_pos = num_y_pos + !!fgd->num_y_points; |
1124 | 65.8k | for (int i = 0; i < num_uv_pos; i++) |
1125 | 58.6k | fgd->ar_coeffs_uv[pl][i] = dav1d_get_bits(gb, 8) - 128; |
1126 | 7.20k | if (!fgd->num_y_points) |
1127 | 4.22k | fgd->ar_coeffs_uv[pl][num_uv_pos] = 0; |
1128 | 7.20k | } |
1129 | 6.40k | fgd->ar_coeff_shift = dav1d_get_bits(gb, 2) + 6; |
1130 | 6.40k | fgd->grain_scale_shift = dav1d_get_bits(gb, 2); |
1131 | 19.2k | for (int pl = 0; pl < 2; pl++) |
1132 | 12.8k | if (fgd->num_uv_points[pl]) { |
1133 | 2.11k | fgd->uv_mult[pl] = dav1d_get_bits(gb, 8) - 128; |
1134 | 2.11k | fgd->uv_luma_mult[pl] = dav1d_get_bits(gb, 8) - 128; |
1135 | 2.11k | fgd->uv_offset[pl] = dav1d_get_bits(gb, 9) - 256; |
1136 | 2.11k | } |
1137 | 6.40k | fgd->overlap_flag = dav1d_get_bit(gb); |
1138 | 6.40k | fgd->clip_to_restricted_range = dav1d_get_bit(gb); |
1139 | 6.40k | } |
1140 | 30.1k | } |
1141 | 231k | } |
1142 | | #if DEBUG_FRAME_HDR |
1143 | | printf("HDR: post-filmgrain: off=%td\n", |
1144 | | (gb->ptr - init_ptr) * 8 - gb->bits_left); |
1145 | | #endif |
1146 | | |
1147 | 463k | return 0; |
1148 | | |
1149 | 13.6k | error: |
1150 | 13.6k | dav1d_log(c, "Error parsing frame header\n"); |
1151 | 13.6k | return DAV1D_ERR(EINVAL); |
1152 | 466k | } |
1153 | | |
1154 | 454k | static void parse_tile_hdr(Dav1dContext *const c, GetBits *const gb) { |
1155 | 454k | const int n_tiles = c->frame_hdr->tiling.cols * c->frame_hdr->tiling.rows; |
1156 | 454k | const int have_tile_pos = n_tiles > 1 ? dav1d_get_bit(gb) : 0; |
1157 | | |
1158 | 454k | if (have_tile_pos) { |
1159 | 4.10k | const int n_bits = c->frame_hdr->tiling.log2_cols + |
1160 | 4.10k | c->frame_hdr->tiling.log2_rows; |
1161 | 4.10k | c->tile[c->n_tile_data].start = dav1d_get_bits(gb, n_bits); |
1162 | 4.10k | c->tile[c->n_tile_data].end = dav1d_get_bits(gb, n_bits); |
1163 | 450k | } else { |
1164 | 450k | c->tile[c->n_tile_data].start = 0; |
1165 | 450k | c->tile[c->n_tile_data].end = n_tiles - 1; |
1166 | 450k | } |
1167 | 454k | } |
1168 | | |
1169 | 668k | ptrdiff_t dav1d_parse_obus(Dav1dContext *const c, Dav1dData *const in) { |
1170 | 668k | GetBits gb; |
1171 | 668k | int res; |
1172 | | |
1173 | 668k | dav1d_init_get_bits(&gb, in->data, in->sz); |
1174 | | |
1175 | | // obu header |
1176 | 668k | const int obu_forbidden_bit = dav1d_get_bit(&gb); |
1177 | 668k | if (c->strict_std_compliance && obu_forbidden_bit) goto error; |
1178 | 668k | const enum Dav1dObuType type = dav1d_get_bits(&gb, 4); |
1179 | 668k | const int has_extension = dav1d_get_bit(&gb); |
1180 | 668k | const int has_length_field = dav1d_get_bit(&gb); |
1181 | 668k | dav1d_get_bit(&gb); // reserved |
1182 | | |
1183 | 668k | int temporal_id = 0, spatial_id = 0; |
1184 | 668k | if (has_extension) { |
1185 | 19.9k | temporal_id = dav1d_get_bits(&gb, 3); |
1186 | 19.9k | spatial_id = dav1d_get_bits(&gb, 2); |
1187 | 19.9k | dav1d_get_bits(&gb, 3); // reserved |
1188 | 19.9k | } |
1189 | | |
1190 | 668k | if (has_length_field) { |
1191 | 219k | const size_t len = dav1d_get_uleb128(&gb); |
1192 | 219k | if (len > (size_t)(gb.ptr_end - gb.ptr)) goto error; |
1193 | 212k | gb.ptr_end = gb.ptr + len; |
1194 | 212k | } |
1195 | 661k | if (gb.error) goto error; |
1196 | | |
1197 | | // We must have read a whole number of bytes at this point (1 byte |
1198 | | // for the header and whole bytes at a time when reading the |
1199 | | // leb128 length field). |
1200 | 661k | assert(gb.bits_left == 0); |
1201 | | |
1202 | | // skip obu not belonging to the selected temporal/spatial layer |
1203 | 660k | if (type != DAV1D_OBU_SEQ_HDR && type != DAV1D_OBU_TD && |
1204 | 536k | has_extension && c->operating_point_idc != 0) |
1205 | 7.62k | { |
1206 | 7.62k | const int in_temporal_layer = (c->operating_point_idc >> temporal_id) & 1; |
1207 | 7.62k | const int in_spatial_layer = (c->operating_point_idc >> (spatial_id + 8)) & 1; |
1208 | 7.62k | if (!in_temporal_layer || !in_spatial_layer) |
1209 | 1.65k | return gb.ptr_end - gb.ptr_start; |
1210 | 7.62k | } |
1211 | | |
1212 | 659k | switch (type) { |
1213 | 46.6k | case DAV1D_OBU_SEQ_HDR: { |
1214 | 46.6k | Dav1dRef *ref = dav1d_ref_create_using_pool(c->seq_hdr_pool, |
1215 | 46.6k | sizeof(Dav1dSequenceHeader)); |
1216 | 46.6k | if (!ref) return DAV1D_ERR(ENOMEM); |
1217 | 46.6k | Dav1dSequenceHeader *seq_hdr = ref->data; |
1218 | 46.6k | if ((res = parse_seq_hdr(seq_hdr, &gb, c->strict_std_compliance)) < 0) { |
1219 | 5.16k | dav1d_log(c, "Error parsing sequence header\n"); |
1220 | 5.16k | dav1d_ref_dec(&ref); |
1221 | 5.16k | goto error; |
1222 | 5.16k | } |
1223 | | |
1224 | 41.5k | const int op_idx = |
1225 | 41.5k | c->operating_point < seq_hdr->num_operating_points ? c->operating_point : 0; |
1226 | 41.5k | c->operating_point_idc = seq_hdr->operating_points[op_idx].idc; |
1227 | 41.5k | const unsigned spatial_mask = c->operating_point_idc >> 8; |
1228 | 41.5k | c->max_spatial_id = spatial_mask ? ulog2(spatial_mask) : 0; |
1229 | | |
1230 | | // If we have read a sequence header which is different from |
1231 | | // the old one, this is a new video sequence and can't use any |
1232 | | // previous state. Free that state. |
1233 | | |
1234 | 41.5k | if (!c->seq_hdr) { |
1235 | 19.0k | c->frame_hdr = NULL; |
1236 | 19.0k | c->frame_flags |= PICTURE_FLAG_NEW_SEQUENCE; |
1237 | | // see 7.5, operating_parameter_info is allowed to change in |
1238 | | // sequence headers of a single sequence |
1239 | 22.5k | } else if (memcmp(seq_hdr, c->seq_hdr, offsetof(Dav1dSequenceHeader, operating_parameter_info))) { |
1240 | 9.87k | c->frame_hdr = NULL; |
1241 | 9.87k | c->mastering_display = NULL; |
1242 | 9.87k | c->content_light = NULL; |
1243 | 9.87k | dav1d_ref_dec(&c->mastering_display_ref); |
1244 | 9.87k | dav1d_ref_dec(&c->content_light_ref); |
1245 | 88.8k | for (int i = 0; i < 8; i++) { |
1246 | 78.9k | if (c->refs[i].p.p.frame_hdr) |
1247 | 27.7k | dav1d_thread_picture_unref(&c->refs[i].p); |
1248 | 78.9k | dav1d_ref_dec(&c->refs[i].segmap); |
1249 | 78.9k | dav1d_ref_dec(&c->refs[i].refmvs); |
1250 | 78.9k | dav1d_cdf_thread_unref(&c->cdf[i]); |
1251 | 78.9k | } |
1252 | 9.87k | c->frame_flags |= PICTURE_FLAG_NEW_SEQUENCE; |
1253 | | // If operating_parameter_info changed, signal it |
1254 | 12.6k | } else if (memcmp(seq_hdr->operating_parameter_info, c->seq_hdr->operating_parameter_info, |
1255 | 12.6k | sizeof(seq_hdr->operating_parameter_info))) |
1256 | 290 | { |
1257 | 290 | c->frame_flags |= PICTURE_FLAG_NEW_OP_PARAMS_INFO; |
1258 | 290 | } |
1259 | 41.5k | dav1d_ref_dec(&c->seq_hdr_ref); |
1260 | 41.5k | c->seq_hdr_ref = ref; |
1261 | 41.5k | c->seq_hdr = seq_hdr; |
1262 | 41.5k | break; |
1263 | 46.6k | } |
1264 | 2.64k | case DAV1D_OBU_REDUNDANT_FRAME_HDR: |
1265 | 2.64k | if (c->frame_hdr) break; |
1266 | | // fall-through |
1267 | 467k | case DAV1D_OBU_FRAME: |
1268 | 514k | case DAV1D_OBU_FRAME_HDR: |
1269 | 514k | if (!c->seq_hdr) goto error; |
1270 | 514k | if (!c->frame_hdr_ref) { |
1271 | 429k | c->frame_hdr_ref = dav1d_ref_create_using_pool(c->frame_hdr_pool, |
1272 | 429k | sizeof(Dav1dFrameHeader)); |
1273 | 429k | if (!c->frame_hdr_ref) return DAV1D_ERR(ENOMEM); |
1274 | 429k | } |
1275 | 514k | #ifndef NDEBUG |
1276 | | // ensure that the reference is writable |
1277 | 514k | assert(dav1d_ref_is_writable(c->frame_hdr_ref)); |
1278 | 514k | #endif |
1279 | 514k | c->frame_hdr = c->frame_hdr_ref->data; |
1280 | 514k | memset(c->frame_hdr, 0, sizeof(*c->frame_hdr)); |
1281 | 514k | c->frame_hdr->temporal_id = temporal_id; |
1282 | 514k | c->frame_hdr->spatial_id = spatial_id; |
1283 | 514k | if ((res = parse_frame_hdr(c, &gb)) < 0) { |
1284 | 13.6k | c->frame_hdr = NULL; |
1285 | 13.6k | goto error; |
1286 | 13.6k | } |
1287 | 507k | for (int n = 0; n < c->n_tile_data; n++) |
1288 | 7.30k | dav1d_data_unref_internal(&c->tile[n].data); |
1289 | 500k | c->n_tile_data = 0; |
1290 | 500k | c->n_tiles = 0; |
1291 | 500k | if (type != DAV1D_OBU_FRAME) { |
1292 | | // This is actually a frame header OBU so read the |
1293 | | // trailing bit and check for overrun. |
1294 | 45.7k | if (check_trailing_bits(&gb, c->strict_std_compliance) < 0) { |
1295 | 6.88k | c->frame_hdr = NULL; |
1296 | 6.88k | goto error; |
1297 | 6.88k | } |
1298 | 45.7k | } |
1299 | | |
1300 | 493k | if (c->frame_size_limit && (int64_t)c->frame_hdr->width[1] * |
1301 | 493k | c->frame_hdr->height > c->frame_size_limit) |
1302 | 1.19k | { |
1303 | 1.19k | dav1d_log(c, "Frame size %dx%d exceeds limit %u\n", c->frame_hdr->width[1], |
1304 | 1.19k | c->frame_hdr->height, c->frame_size_limit); |
1305 | 1.19k | c->frame_hdr = NULL; |
1306 | 1.19k | return DAV1D_ERR(ERANGE); |
1307 | 1.19k | } |
1308 | | |
1309 | 492k | if (type != DAV1D_OBU_FRAME) |
1310 | 38.9k | break; |
1311 | | // OBU_FRAMEs shouldn't be signaled with show_existing_frame |
1312 | 453k | if (c->frame_hdr->show_existing_frame) { |
1313 | 810 | c->frame_hdr = NULL; |
1314 | 810 | goto error; |
1315 | 810 | } |
1316 | | |
1317 | | // This is the frame header at the start of a frame OBU. |
1318 | | // There's no trailing bit at the end to skip, but we do need |
1319 | | // to align to the next byte. |
1320 | 452k | dav1d_bytealign_get_bits(&gb); |
1321 | | // fall-through |
1322 | 456k | case DAV1D_OBU_TILE_GRP: { |
1323 | 456k | if (!c->frame_hdr) goto error; |
1324 | 454k | if (c->n_tile_data_alloc < c->n_tile_data + 1) { |
1325 | 17.8k | if ((c->n_tile_data + 1) > INT_MAX / (int)sizeof(*c->tile)) goto error; |
1326 | 17.8k | struct Dav1dTileGroup *tile = dav1d_realloc(ALLOC_TILE, c->tile, |
1327 | 17.8k | (c->n_tile_data + 1) * sizeof(*c->tile)); |
1328 | 17.8k | if (!tile) goto error; |
1329 | 17.8k | c->tile = tile; |
1330 | 17.8k | memset(c->tile + c->n_tile_data, 0, sizeof(*c->tile)); |
1331 | 17.8k | c->n_tile_data_alloc = c->n_tile_data + 1; |
1332 | 17.8k | } |
1333 | 454k | parse_tile_hdr(c, &gb); |
1334 | | // Align to the next byte boundary and check for overrun. |
1335 | 454k | dav1d_bytealign_get_bits(&gb); |
1336 | 454k | if (gb.error) goto error; |
1337 | | |
1338 | 430k | dav1d_data_ref(&c->tile[c->n_tile_data].data, in); |
1339 | 430k | c->tile[c->n_tile_data].data.data = gb.ptr; |
1340 | 430k | c->tile[c->n_tile_data].data.sz = (size_t)(gb.ptr_end - gb.ptr); |
1341 | | // ensure tile groups are in order and sane, see 6.10.1 |
1342 | 430k | if (c->tile[c->n_tile_data].start > c->tile[c->n_tile_data].end || |
1343 | 429k | c->tile[c->n_tile_data].start != c->n_tiles) |
1344 | 2.35k | { |
1345 | 5.08k | for (int i = 0; i <= c->n_tile_data; i++) |
1346 | 2.72k | dav1d_data_unref_internal(&c->tile[i].data); |
1347 | 2.35k | c->n_tile_data = 0; |
1348 | 2.35k | c->n_tiles = 0; |
1349 | 2.35k | goto error; |
1350 | 2.35k | } |
1351 | 427k | c->n_tiles += 1 + c->tile[c->n_tile_data].end - |
1352 | 427k | c->tile[c->n_tile_data].start; |
1353 | 427k | c->n_tile_data++; |
1354 | 427k | break; |
1355 | 430k | } |
1356 | 7.32k | case DAV1D_OBU_METADATA: { |
1357 | 7.32k | #define DEBUG_OBU_METADATA 0 |
1358 | | #if DEBUG_OBU_METADATA |
1359 | | const uint8_t *const init_ptr = gb.ptr; |
1360 | | #endif |
1361 | | // obu metadta type field |
1362 | 7.32k | const enum ObuMetaType meta_type = dav1d_get_uleb128(&gb); |
1363 | 7.32k | if (gb.error) goto error; |
1364 | | |
1365 | 6.99k | switch (meta_type) { |
1366 | 692 | case OBU_META_HDR_CLL: { |
1367 | 692 | Dav1dRef *ref = dav1d_ref_create(ALLOC_OBU_META, |
1368 | 692 | sizeof(Dav1dContentLightLevel)); |
1369 | 692 | if (!ref) return DAV1D_ERR(ENOMEM); |
1370 | 692 | Dav1dContentLightLevel *const content_light = ref->data; |
1371 | | |
1372 | 692 | content_light->max_content_light_level = dav1d_get_bits(&gb, 16); |
1373 | | #if DEBUG_OBU_METADATA |
1374 | | printf("CLLOBU: max-content-light-level: %d [off=%td]\n", |
1375 | | content_light->max_content_light_level, |
1376 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1377 | | #endif |
1378 | 692 | content_light->max_frame_average_light_level = dav1d_get_bits(&gb, 16); |
1379 | | #if DEBUG_OBU_METADATA |
1380 | | printf("CLLOBU: max-frame-average-light-level: %d [off=%td]\n", |
1381 | | content_light->max_frame_average_light_level, |
1382 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1383 | | #endif |
1384 | | |
1385 | 692 | if (check_trailing_bits(&gb, c->strict_std_compliance) < 0) { |
1386 | 309 | dav1d_ref_dec(&ref); |
1387 | 309 | goto error; |
1388 | 309 | } |
1389 | | |
1390 | 383 | dav1d_ref_dec(&c->content_light_ref); |
1391 | 383 | c->content_light = content_light; |
1392 | 383 | c->content_light_ref = ref; |
1393 | 383 | break; |
1394 | 692 | } |
1395 | 312 | case OBU_META_HDR_MDCV: { |
1396 | 312 | Dav1dRef *ref = dav1d_ref_create(ALLOC_OBU_META, |
1397 | 312 | sizeof(Dav1dMasteringDisplay)); |
1398 | 312 | if (!ref) return DAV1D_ERR(ENOMEM); |
1399 | 312 | Dav1dMasteringDisplay *const mastering_display = ref->data; |
1400 | | |
1401 | 1.24k | for (int i = 0; i < 3; i++) { |
1402 | 936 | mastering_display->primaries[i][0] = dav1d_get_bits(&gb, 16); |
1403 | 936 | mastering_display->primaries[i][1] = dav1d_get_bits(&gb, 16); |
1404 | | #if DEBUG_OBU_METADATA |
1405 | | printf("MDCVOBU: primaries[%d]: (%d, %d) [off=%td]\n", i, |
1406 | | mastering_display->primaries[i][0], |
1407 | | mastering_display->primaries[i][1], |
1408 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1409 | | #endif |
1410 | 936 | } |
1411 | 312 | mastering_display->white_point[0] = dav1d_get_bits(&gb, 16); |
1412 | | #if DEBUG_OBU_METADATA |
1413 | | printf("MDCVOBU: white-point-x: %d [off=%td]\n", |
1414 | | mastering_display->white_point[0], |
1415 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1416 | | #endif |
1417 | 312 | mastering_display->white_point[1] = dav1d_get_bits(&gb, 16); |
1418 | | #if DEBUG_OBU_METADATA |
1419 | | printf("MDCVOBU: white-point-y: %d [off=%td]\n", |
1420 | | mastering_display->white_point[1], |
1421 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1422 | | #endif |
1423 | 312 | mastering_display->max_luminance = dav1d_get_bits(&gb, 32); |
1424 | | #if DEBUG_OBU_METADATA |
1425 | | printf("MDCVOBU: max-luminance: %d [off=%td]\n", |
1426 | | mastering_display->max_luminance, |
1427 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1428 | | #endif |
1429 | 312 | mastering_display->min_luminance = dav1d_get_bits(&gb, 32); |
1430 | | #if DEBUG_OBU_METADATA |
1431 | | printf("MDCVOBU: min-luminance: %d [off=%td]\n", |
1432 | | mastering_display->min_luminance, |
1433 | | (gb.ptr - init_ptr) * 8 - gb.bits_left); |
1434 | | #endif |
1435 | 312 | if (check_trailing_bits(&gb, c->strict_std_compliance) < 0) { |
1436 | 149 | dav1d_ref_dec(&ref); |
1437 | 149 | goto error; |
1438 | 149 | } |
1439 | | |
1440 | 163 | dav1d_ref_dec(&c->mastering_display_ref); |
1441 | 163 | c->mastering_display = mastering_display; |
1442 | 163 | c->mastering_display_ref = ref; |
1443 | 163 | break; |
1444 | 312 | } |
1445 | 5.35k | case OBU_META_ITUT_T35: { |
1446 | 5.35k | ptrdiff_t payload_size = gb.ptr_end - gb.ptr; |
1447 | | // Don't take into account all the trailing bits for payload_size |
1448 | 7.30k | while (payload_size > 0 && !gb.ptr[payload_size - 1]) |
1449 | 1.94k | payload_size--; // trailing_zero_bit x 8 |
1450 | 5.35k | payload_size--; // trailing_one_bit + trailing_zero_bit x 7 |
1451 | | |
1452 | 5.35k | int country_code_extension_byte = 0; |
1453 | 5.35k | const int country_code = dav1d_get_bits(&gb, 8); |
1454 | 5.35k | payload_size--; |
1455 | 5.35k | if (country_code == 0xFF) { |
1456 | 1.65k | country_code_extension_byte = dav1d_get_bits(&gb, 8); |
1457 | 1.65k | payload_size--; |
1458 | 1.65k | } |
1459 | | |
1460 | 5.35k | if (payload_size <= 0 || gb.ptr[payload_size] != 0x80) { |
1461 | 2.31k | dav1d_log(c, "Malformed ITU-T T.35 metadata message format\n"); |
1462 | 2.31k | break; |
1463 | 2.31k | } |
1464 | | |
1465 | 3.04k | if ((c->n_itut_t35 + 1) > INT_MAX / (int)sizeof(*c->itut_t35)) goto error; |
1466 | 3.04k | struct Dav1dITUTT35 *itut_t35 = dav1d_realloc(ALLOC_OBU_META, c->itut_t35, |
1467 | 3.04k | (c->n_itut_t35 + 1) * sizeof(*c->itut_t35)); |
1468 | 3.04k | if (!itut_t35) goto error; |
1469 | 3.04k | c->itut_t35 = itut_t35; |
1470 | 3.04k | memset(c->itut_t35 + c->n_itut_t35, 0, sizeof(*c->itut_t35)); |
1471 | | |
1472 | 3.04k | struct itut_t35_ctx_context *itut_t35_ctx; |
1473 | 3.04k | if (!c->n_itut_t35) { |
1474 | 1.71k | assert(!c->itut_t35_ref); |
1475 | 1.71k | itut_t35_ctx = dav1d_malloc(ALLOC_OBU_META, sizeof(struct itut_t35_ctx_context)); |
1476 | 1.71k | if (!itut_t35_ctx) goto error; |
1477 | 1.71k | c->itut_t35_ref = dav1d_ref_init(&itut_t35_ctx->ref, c->itut_t35, |
1478 | 1.71k | dav1d_picture_free_itut_t35, itut_t35_ctx, 0); |
1479 | 1.71k | } else { |
1480 | 1.32k | assert(c->itut_t35_ref && atomic_load(&c->itut_t35_ref->ref_cnt) == 1); |
1481 | 1.32k | itut_t35_ctx = c->itut_t35_ref->user_data; |
1482 | 1.32k | c->itut_t35_ref->const_data = (uint8_t *)c->itut_t35; |
1483 | 1.32k | } |
1484 | 3.04k | itut_t35_ctx->itut_t35 = c->itut_t35; |
1485 | 3.04k | itut_t35_ctx->n_itut_t35 = c->n_itut_t35 + 1; |
1486 | | |
1487 | 3.04k | Dav1dITUTT35 *const itut_t35_metadata = &c->itut_t35[c->n_itut_t35]; |
1488 | 3.04k | itut_t35_metadata->payload = dav1d_malloc(ALLOC_OBU_META, payload_size); |
1489 | 3.04k | if (!itut_t35_metadata->payload) goto error; |
1490 | | |
1491 | 3.04k | itut_t35_metadata->country_code = country_code; |
1492 | 3.04k | itut_t35_metadata->country_code_extension_byte = country_code_extension_byte; |
1493 | 3.04k | itut_t35_metadata->payload_size = payload_size; |
1494 | | |
1495 | | // We know that we've read a whole number of bytes and that the |
1496 | | // payload is within the OBU boundaries, so just use memcpy() |
1497 | 3.04k | assert(gb.bits_left == 0); |
1498 | 3.04k | memcpy(itut_t35_metadata->payload, gb.ptr, payload_size); |
1499 | | |
1500 | 3.04k | c->n_itut_t35++; |
1501 | 3.04k | break; |
1502 | 3.04k | } |
1503 | 0 | case OBU_META_SCALABILITY: |
1504 | 8 | case OBU_META_TIMECODE: |
1505 | | // ignore metadata OBUs we don't care about |
1506 | 8 | break; |
1507 | 620 | default: |
1508 | | // print a warning but don't fail for unknown types |
1509 | 620 | if (meta_type > 31) // Types 6 to 31 are "Unregistered user private", so ignore them. |
1510 | 545 | dav1d_log(c, "Unknown Metadata OBU type %d\n", meta_type); |
1511 | 620 | break; |
1512 | 6.99k | } |
1513 | | |
1514 | 6.53k | break; |
1515 | 6.99k | } |
1516 | 77.5k | case DAV1D_OBU_TD: |
1517 | 77.5k | c->frame_flags |= PICTURE_FLAG_NEW_TEMPORAL_UNIT; |
1518 | 77.5k | break; |
1519 | 654 | case DAV1D_OBU_PADDING: |
1520 | | // ignore OBUs we don't care about |
1521 | 654 | break; |
1522 | 7.55k | default: |
1523 | | // print a warning but don't fail for unknown types |
1524 | 7.55k | dav1d_log(c, "Unknown OBU type %d of size %td\n", type, gb.ptr_end - gb.ptr); |
1525 | 7.55k | break; |
1526 | 659k | } |
1527 | | |
1528 | 601k | if (c->seq_hdr && c->frame_hdr) { |
1529 | 481k | if (c->frame_hdr->show_existing_frame) { |
1530 | 36.0k | if (!c->refs[c->frame_hdr->existing_frame_idx].p.p.frame_hdr) goto error; |
1531 | 35.1k | switch (c->refs[c->frame_hdr->existing_frame_idx].p.p.frame_hdr->frame_type) { |
1532 | 19.7k | case DAV1D_FRAME_TYPE_INTER: |
1533 | 20.3k | case DAV1D_FRAME_TYPE_SWITCH: |
1534 | 20.3k | if (c->decode_frame_type > DAV1D_DECODEFRAMETYPE_REFERENCE) |
1535 | 0 | goto skip; |
1536 | 20.3k | break; |
1537 | 20.3k | case DAV1D_FRAME_TYPE_INTRA: |
1538 | 589 | if (c->decode_frame_type > DAV1D_DECODEFRAMETYPE_INTRA) |
1539 | 0 | goto skip; |
1540 | | // fall-through |
1541 | 14.7k | default: |
1542 | 14.7k | break; |
1543 | 35.1k | } |
1544 | 35.1k | if (!c->refs[c->frame_hdr->existing_frame_idx].p.p.data[0]) goto error; |
1545 | 35.1k | if (c->strict_std_compliance && |
1546 | 0 | !c->refs[c->frame_hdr->existing_frame_idx].p.showable) |
1547 | 0 | { |
1548 | 0 | goto error; |
1549 | 0 | } |
1550 | 35.1k | if (c->n_fc == 1) { |
1551 | 5.65k | dav1d_thread_picture_ref(&c->out, |
1552 | 5.65k | &c->refs[c->frame_hdr->existing_frame_idx].p); |
1553 | 5.65k | dav1d_picture_copy_props(&c->out.p, |
1554 | 5.65k | c->content_light, c->content_light_ref, |
1555 | 5.65k | c->mastering_display, c->mastering_display_ref, |
1556 | 5.65k | c->itut_t35, c->itut_t35_ref, c->n_itut_t35, |
1557 | 5.65k | &in->m); |
1558 | | // Must be removed from the context after being attached to the frame |
1559 | 5.65k | dav1d_ref_dec(&c->itut_t35_ref); |
1560 | 5.65k | c->itut_t35 = NULL; |
1561 | 5.65k | c->n_itut_t35 = 0; |
1562 | 5.65k | c->event_flags |= dav1d_picture_get_event_flags(&c->refs[c->frame_hdr->existing_frame_idx].p); |
1563 | 29.4k | } else { |
1564 | 29.4k | pthread_mutex_lock(&c->task_thread.lock); |
1565 | | // need to append this to the frame output queue |
1566 | 29.4k | const unsigned next = c->frame_thread.next++; |
1567 | 29.4k | if (c->frame_thread.next == c->n_fc) |
1568 | 7.31k | c->frame_thread.next = 0; |
1569 | | |
1570 | 29.4k | Dav1dFrameContext *const f = &c->fc[next]; |
1571 | 32.9k | while (f->n_tile_data > 0) |
1572 | 3.47k | pthread_cond_wait(&f->task_thread.cond, |
1573 | 3.47k | &f->task_thread.ttd->lock); |
1574 | 29.4k | Dav1dThreadPicture *const out_delayed = |
1575 | 29.4k | &c->frame_thread.out_delayed[next]; |
1576 | 29.4k | if (out_delayed->p.data[0] || atomic_load(&f->task_thread.error)) { |
1577 | 28.9k | unsigned first = atomic_load(&c->task_thread.first); |
1578 | 28.9k | if (first + 1U < c->n_fc) |
1579 | 28.9k | atomic_fetch_add(&c->task_thread.first, 1U); |
1580 | 7.17k | else |
1581 | 28.9k | atomic_store(&c->task_thread.first, 0); |
1582 | 28.9k | atomic_compare_exchange_strong(&c->task_thread.reset_task_cur, |
1583 | 28.9k | &first, UINT_MAX); |
1584 | 28.9k | if (c->task_thread.cur && c->task_thread.cur < c->n_fc) |
1585 | 20.7k | c->task_thread.cur--; |
1586 | 28.9k | } |
1587 | 29.4k | const int error = f->task_thread.retval; |
1588 | 29.4k | if (error) { |
1589 | 2.75k | c->cached_error = error; |
1590 | 2.75k | f->task_thread.retval = 0; |
1591 | 2.75k | dav1d_data_props_copy(&c->cached_error_props, &out_delayed->p.m); |
1592 | 2.75k | dav1d_thread_picture_unref(out_delayed); |
1593 | 26.7k | } else if (out_delayed->p.data[0]) { |
1594 | 26.0k | const unsigned progress = atomic_load_explicit(&out_delayed->progress[1], |
1595 | 26.0k | memory_order_relaxed); |
1596 | 26.0k | if ((out_delayed->visible || c->output_invisible_frames) && |
1597 | 25.8k | progress != FRAME_ERROR) |
1598 | 10.0k | { |
1599 | 10.0k | dav1d_thread_picture_ref(&c->out, out_delayed); |
1600 | 10.0k | c->event_flags |= dav1d_picture_get_event_flags(out_delayed); |
1601 | 10.0k | } |
1602 | 26.0k | dav1d_thread_picture_unref(out_delayed); |
1603 | 26.0k | } |
1604 | 29.4k | dav1d_thread_picture_ref(out_delayed, |
1605 | 29.4k | &c->refs[c->frame_hdr->existing_frame_idx].p); |
1606 | 29.4k | out_delayed->visible = 1; |
1607 | 29.4k | dav1d_picture_copy_props(&out_delayed->p, |
1608 | 29.4k | c->content_light, c->content_light_ref, |
1609 | 29.4k | c->mastering_display, c->mastering_display_ref, |
1610 | 29.4k | c->itut_t35, c->itut_t35_ref, c->n_itut_t35, |
1611 | 29.4k | &in->m); |
1612 | | // Must be removed from the context after being attached to the frame |
1613 | 29.4k | dav1d_ref_dec(&c->itut_t35_ref); |
1614 | 29.4k | c->itut_t35 = NULL; |
1615 | 29.4k | c->n_itut_t35 = 0; |
1616 | | |
1617 | 29.4k | pthread_mutex_unlock(&c->task_thread.lock); |
1618 | 29.4k | } |
1619 | 35.1k | if (c->refs[c->frame_hdr->existing_frame_idx].p.p.frame_hdr->frame_type == DAV1D_FRAME_TYPE_KEY) { |
1620 | 14.1k | const int r = c->frame_hdr->existing_frame_idx; |
1621 | 14.1k | c->refs[r].p.showable = 0; |
1622 | 127k | for (int i = 0; i < 8; i++) { |
1623 | 113k | if (i == r) continue; |
1624 | | |
1625 | 99.3k | if (c->refs[i].p.p.frame_hdr) |
1626 | 98.8k | dav1d_thread_picture_unref(&c->refs[i].p); |
1627 | 99.3k | dav1d_thread_picture_ref(&c->refs[i].p, &c->refs[r].p); |
1628 | | |
1629 | 99.3k | dav1d_cdf_thread_unref(&c->cdf[i]); |
1630 | 99.3k | dav1d_cdf_thread_ref(&c->cdf[i], &c->cdf[r]); |
1631 | | |
1632 | 99.3k | dav1d_ref_dec(&c->refs[i].segmap); |
1633 | 99.3k | c->refs[i].segmap = c->refs[r].segmap; |
1634 | 99.3k | if (c->refs[r].segmap) |
1635 | 8.71k | dav1d_ref_inc(c->refs[r].segmap); |
1636 | 99.3k | dav1d_ref_dec(&c->refs[i].refmvs); |
1637 | 99.3k | } |
1638 | 14.1k | } |
1639 | 35.1k | c->frame_hdr = NULL; |
1640 | 445k | } else if (c->n_tiles == c->frame_hdr->tiling.cols * c->frame_hdr->tiling.rows) { |
1641 | 426k | switch (c->frame_hdr->frame_type) { |
1642 | 137k | case DAV1D_FRAME_TYPE_INTER: |
1643 | 137k | case DAV1D_FRAME_TYPE_SWITCH: |
1644 | 137k | if (c->decode_frame_type > DAV1D_DECODEFRAMETYPE_REFERENCE || |
1645 | 137k | (c->decode_frame_type == DAV1D_DECODEFRAMETYPE_REFERENCE && |
1646 | 0 | !c->frame_hdr->refresh_frame_flags)) |
1647 | 0 | goto skip; |
1648 | 137k | break; |
1649 | 137k | case DAV1D_FRAME_TYPE_INTRA: |
1650 | 1.89k | if (c->decode_frame_type > DAV1D_DECODEFRAMETYPE_INTRA || |
1651 | 1.89k | (c->decode_frame_type == DAV1D_DECODEFRAMETYPE_REFERENCE && |
1652 | 0 | !c->frame_hdr->refresh_frame_flags)) |
1653 | 0 | goto skip; |
1654 | | // fall-through |
1655 | 288k | default: |
1656 | 288k | break; |
1657 | 426k | } |
1658 | 426k | if (!c->n_tile_data) |
1659 | 0 | goto error; |
1660 | 426k | if ((res = dav1d_submit_frame(c)) < 0) |
1661 | 33.0k | return res; |
1662 | 426k | assert(!c->n_tile_data); |
1663 | 393k | c->frame_hdr = NULL; |
1664 | 393k | c->n_tiles = 0; |
1665 | 393k | } |
1666 | 481k | } |
1667 | | |
1668 | 567k | return gb.ptr_end - gb.ptr_start; |
1669 | | |
1670 | 0 | skip: |
1671 | | // update refs with only the headers in case we skip the frame |
1672 | 0 | for (int i = 0; i < 8; i++) { |
1673 | 0 | if (c->frame_hdr->refresh_frame_flags & (1 << i)) { |
1674 | 0 | dav1d_thread_picture_unref(&c->refs[i].p); |
1675 | 0 | c->refs[i].p.p.frame_hdr = c->frame_hdr; |
1676 | 0 | c->refs[i].p.p.seq_hdr = c->seq_hdr; |
1677 | 0 | c->refs[i].p.p.frame_hdr_ref = c->frame_hdr_ref; |
1678 | 0 | c->refs[i].p.p.seq_hdr_ref = c->seq_hdr_ref; |
1679 | 0 | dav1d_ref_inc(c->frame_hdr_ref); |
1680 | 0 | dav1d_ref_inc(c->seq_hdr_ref); |
1681 | 0 | } |
1682 | 0 | } |
1683 | |
|
1684 | 0 | dav1d_ref_dec(&c->frame_hdr_ref); |
1685 | 0 | c->frame_hdr = NULL; |
1686 | 0 | c->n_tiles = 0; |
1687 | |
|
1688 | 0 | return gb.ptr_end - gb.ptr_start; |
1689 | | |
1690 | 65.2k | error: |
1691 | 65.2k | dav1d_data_props_copy(&c->cached_error_props, &in->m); |
1692 | 65.2k | dav1d_log(c, gb.error ? "Overrun in OBU bit buffer\n" : |
1693 | 65.2k | "Error parsing OBU data\n"); |
1694 | 65.2k | return DAV1D_ERR(EINVAL); |
1695 | 601k | } |