1 /*
2 * Copyright (C) 2007 Marco Gerards <marco@gnu.org>
3 * Copyright (C) 2009 David Conrad
4 * Copyright (C) 2011 Jordi Ortiz
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 /**
24 * @file
25 * Dirac Decoder
26 * @author Marco Gerards <marco@gnu.org>, David Conrad, Jordi Ortiz <nenjordi@gmail.com>
27 */
28
29 #include "config.h"
30
32
37
38 #if CONFIG_DIRAC_PARSE
39
40 typedef struct dirac_source_params {
43 uint8_t chroma_format; ///< 0: 444 1: 422 2: 420
44
46 uint8_t top_field_first;
47
48 uint8_t frame_rate_index; ///< index into dirac_frame_rate[]
49 uint8_t aspect_ratio_index; ///< index into dirac_aspect_ratio[]
50
51 uint16_t clean_width;
52 uint16_t clean_height;
53 uint16_t clean_left_offset;
54 uint16_t clean_right_offset;
55
56 uint8_t pixel_range_index; ///< index into dirac_pixel_range_presets[]
57 uint8_t color_spec_index; ///< index into dirac_color_spec_presets[]
58 } dirac_source_params;
59
60 /* defaults for source parameters */
61 static const dirac_source_params dirac_source_parameters_defaults[] = {
62 { 640, 480, 2, 0, 0, 1, 1, 640, 480, 0, 0, 1, 0 },
63 { 176, 120, 2, 0, 0, 9, 2, 176, 120, 0, 0, 1, 1 },
64 { 176, 144, 2, 0, 1, 10, 3, 176, 144, 0, 0, 1, 2 },
65 { 352, 240, 2, 0, 0, 9, 2, 352, 240, 0, 0, 1, 1 },
66 { 352, 288, 2, 0, 1, 10, 3, 352, 288, 0, 0, 1, 2 },
67 { 704, 480, 2, 0, 0, 9, 2, 704, 480, 0, 0, 1, 1 },
68 { 704, 576, 2, 0, 1, 10, 3, 704, 576, 0, 0, 1, 2 },
69 { 720, 480, 1, 1, 0, 4, 2, 704, 480, 8, 0, 3, 1 },
70 { 720, 576, 1, 1, 1, 3, 3, 704, 576, 8, 0, 3, 2 },
71
72 { 1280, 720, 1, 0, 1, 7, 1, 1280, 720, 0, 0, 3, 3 },
73 { 1280, 720, 1, 0, 1, 6, 1, 1280, 720, 0, 0, 3, 3 },
74 { 1920, 1080, 1, 1, 1, 4, 1, 1920, 1080, 0, 0, 3, 3 },
75 { 1920, 1080, 1, 1, 1, 3, 1, 1920, 1080, 0, 0, 3, 3 },
76 { 1920, 1080, 1, 0, 1, 7, 1, 1920, 1080, 0, 0, 3, 3 },
77 { 1920, 1080, 1, 0, 1, 6, 1, 1920, 1080, 0, 0, 3, 3 },
78 { 2048, 1080, 0, 0, 1, 2, 1, 2048, 1080, 0, 0, 4, 4 },
79 { 4096, 2160, 0, 0, 1, 2, 1, 4096, 2160, 0, 0, 4, 4 },
80
81 { 3840, 2160, 1, 0, 1, 7, 1, 3840, 2160, 0, 0, 3, 3 },
82 { 3840, 2160, 1, 0, 1, 6, 1, 3840, 2160, 0, 0, 3, 3 },
83 { 7680, 4320, 1, 0, 1, 7, 1, 3840, 2160, 0, 0, 3, 3 },
84 { 7680, 4320, 1, 0, 1, 6, 1, 3840, 2160, 0, 0, 3, 3 },
85 };
86
87 /* [DIRAC_STD] Table 10.4 - Available preset pixel aspect ratio values */
88 static const AVRational dirac_preset_aspect_ratios[] = {
89 { 1, 1 },
90 { 10, 11 },
91 { 12, 11 },
92 { 40, 33 },
93 { 16, 11 },
94 { 4, 3 },
95 };
96
97 /* [DIRAC_STD] Values 9,10 of 10.3.5 Frame Rate.
98 * Table 10.3 Available preset frame rate values
99 */
100 static const AVRational dirac_frame_rate[] = {
101 { 15000, 1001 },
102 { 25, 2 },
103 };
104
105 /* [DIRAC_STD] This should be equivalent to Table 10.5 Available signal
106 * range presets */
107 static const struct {
110 } pixel_range_presets[] = {
115 };
116
121 };
122
123 static const struct {
127 } dirac_color_presets[] = {
133 };
134
135 /* [DIRAC_STD] Table 10.2 Supported chroma sampling formats */
140 };
141
142 /* [DIRAC_STD] 10.3 Parse Source Parameters.
143 * source_parameters(base_video_format) */
145 void *log_ctx)
146 {
148 unsigned luma_depth = 8, luma_offset = 16;
149 int idx;
150 int chroma_x_shift, chroma_y_shift;
152
153 /* [DIRAC_STD] 10.3.2 Frame size. frame_size(video_params) */
154 /* [DIRAC_STD] custom_dimensions_flag */
158 }
159
160 /* [DIRAC_STD] 10.3.3 Chroma Sampling Format.
161 * chroma_sampling_format(video_params) */
162 /* [DIRAC_STD] custom_chroma_format_flag */
164 /* [DIRAC_STD] CHROMA_FORMAT_INDEX */
167 if (log_ctx)
171 }
172
173 /* [DIRAC_STD] 10.3.4 Scan Format. scan_format(video_params) */
174 /* [DIRAC_STD] custom_scan_format_flag */
176 /* [DIRAC_STD] SOURCE_SAMPLING */
180
181 /* [DIRAC_STD] 10.3.5 Frame Rate. frame_rate(video_params) */
182 if (
get_bits1(gb)) {
/* [DIRAC_STD] custom_frame_rate_flag */
184
187
189 /* [DIRAC_STD] FRAME_RATE_NUMER */
191 /* [DIRAC_STD] FRAME_RATE_DENOM */
193 }
194 }
195 /* [DIRAC_STD] preset_frame_rate(video_params, index) */
199 else
200 /* [DIRAC_STD] Table 10.3 values 9-10 */
202 }
204
205 /* [DIRAC_STD] 10.3.6 Pixel Aspect Ratio.
206 * pixel_aspect_ratio(video_params) */
207 if (
get_bits1(gb)) {
/* [DIRAC_STD] custom_pixel_aspect_ratio_flag */
208 /* [DIRAC_STD] index */
210
213
217 }
218 }
219 /* [DIRAC_STD] Take value from Table 10.4 Available preset pixel
220 * aspect ratio values */
224
225 /* [DIRAC_STD] 10.3.7 Clean area. clean_area(video_params) */
226 if (
get_bits1(gb)) {
/* [DIRAC_STD] custom_clean_area_flag */
227 /* [DIRAC_STD] CLEAN_WIDTH */
229 /* [DIRAC_STD] CLEAN_HEIGHT */
231 /* [DIRAC_STD] CLEAN_LEFT_OFFSET */
233 /* [DIRAC_STD] CLEAN_RIGHT_OFFSET */
235 }
236
237 /* [DIRAC_STD] 10.3.8 Signal range. signal_range(video_params)
238 * WARNING: Some adaptation seems to be done using the
239 * AVCOL_RANGE_MPEG/JPEG values */
240 if (
get_bits1(gb)) {
/* [DIRAC_STD] custom_signal_range_flag */
241 /* [DIRAC_STD] index */
243
246
247 /* This assumes either fullrange or MPEG levels only */
255 }
256 }
257 /* [DIRAC_STD] Table 10.5
258 * Available signal range presets <--> pixel_range_presets */
261 luma_depth = pixel_range_presets[idx].bitdepth;
262 dsh->
color_range = pixel_range_presets[idx].color_range;
263 }
264
266
267 /* Full range 8 bts uses the same pix_fmts as limited range 8 bits */
269
272
277
278 if ((dsh->
width % (1<<chroma_x_shift)) || (dsh->
height % (1<<chroma_y_shift))) {
279 if (log_ctx)
280 av_log(log_ctx,
AV_LOG_ERROR,
"Dimensions must be an integer multiple of the chroma subsampling\n");
282 }
283
284 /* [DIRAC_STD] 10.3.9 Colour specification. colour_spec(video_params) */
285 if (
get_bits1(gb)) {
/* [DIRAC_STD] custom_colour_spec_flag */
286 /* [DIRAC_STD] index */
288
291
293 dsh->
colorspace = dirac_color_presets[idx].colorspace;
294 dsh->
color_trc = dirac_color_presets[idx].color_trc;
295
297 /* [DIRAC_STD] 10.3.9.1 Colour primaries */
302 }
303 /* [DIRAC_STD] 10.3.9.2 Colour matrix */
306 if (!idx)
308 else if (idx == 1)
310 }
311 /* [DIRAC_STD] 10.3.9.3 Transfer function */
314 }
315 } else {
318 dsh->
colorspace = dirac_color_presets[idx].colorspace;
319 dsh->
color_trc = dirac_color_presets[idx].color_trc;
320 }
321
322 return 0;
323 }
324
325 /* [DIRAC_STD] 10. Sequence Header. sequence_header() */
327 const uint8_t *buf, size_t buf_size,
328 void *log_ctx)
329 {
332 unsigned video_format, picture_coding_mode;
334
336 if (!dsh)
338
342
343 /* [DIRAC_SPEC] 10.1 Parse Parameters. parse_parameters() */
348 /* [DIRAC_SPEC] sequence_header() -> base_video_format as defined in
349 * 10.2 Base Video Format, table 10.1 Dirac predefined video formats */
351
356
357 if (video_format > 20
U) {
360 }
361
362 /* Fill in defaults for the source parameters. */
363 dsh->
width = dirac_source_parameters_defaults[video_format].width;
364 dsh->
height = dirac_source_parameters_defaults[video_format].height;
365 dsh->
chroma_format = dirac_source_parameters_defaults[video_format].chroma_format;
366 dsh->
interlaced = dirac_source_parameters_defaults[video_format].interlaced;
367 dsh->
top_field_first = dirac_source_parameters_defaults[video_format].top_field_first;
368 dsh->
frame_rate_index = dirac_source_parameters_defaults[video_format].frame_rate_index;
369 dsh->
aspect_ratio_index = dirac_source_parameters_defaults[video_format].aspect_ratio_index;
370 dsh->
clean_width = dirac_source_parameters_defaults[video_format].clean_width;
371 dsh->
clean_height = dirac_source_parameters_defaults[video_format].clean_height;
372 dsh->
clean_left_offset = dirac_source_parameters_defaults[video_format].clean_left_offset;
373 dsh->
clean_right_offset = dirac_source_parameters_defaults[video_format].clean_right_offset;
374 dsh->
pixel_range_index = dirac_source_parameters_defaults[video_format].pixel_range_index;
375 dsh->
color_spec_index = dirac_source_parameters_defaults[video_format].color_spec_index;
376
377 /* [DIRAC_STD] 10.3 Source Parameters
378 * Override the defaults. */
379 ret = parse_source_parameters(dsh, &gb, log_ctx);
382
383 /* [DIRAC_STD] picture_coding_mode shall be 0 for fields and 1 for frames
384 * currently only used to signal field coding */
386 if (picture_coding_mode != 0) {
387 if (log_ctx) {
389 picture_coding_mode);
390 }
393 }
394
395 *pdsh = dsh;
396 return 0;
401 }
402 #else
404 const uint8_t *buf, size_t buf_size,
405 void *log_ctx)
406 {
408 }
409 #endif