lavc/j2kenc: Support AV_PIX_FMT_PAL8.
[ffmpeg.git] / libavcodec / cbs_mpeg2_syntax_template.c
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 static int FUNC(sequence_header)(CodedBitstreamContext *ctx, RWContext *rw,
20                                  MPEG2RawSequenceHeader *current)
21 {
22     CodedBitstreamMPEG2Context *mpeg2 = ctx->priv_data;
23     int err, i;
24
25     HEADER("Sequence Header");
26
27     ui(8,  sequence_header_code);
28
29     ui(12, horizontal_size_value);
30     ui(12, vertical_size_value);
31
32     mpeg2->horizontal_size = current->horizontal_size_value;
33     mpeg2->vertical_size   = current->vertical_size_value;
34
35     ui(4,  aspect_ratio_information);
36     ui(4,  frame_rate_code);
37     ui(18, bit_rate_value);
38
39     marker_bit();
40
41     ui(10, vbv_buffer_size_value);
42     ui(1,  constrained_parameters_flag);
43
44     ui(1, load_intra_quantiser_matrix);
45     if (current->load_intra_quantiser_matrix) {
46         for (i = 0; i < 64; i++)
47             uis(8, intra_quantiser_matrix[i], 1, i);
48     }
49
50     ui(1, load_non_intra_quantiser_matrix);
51     if (current->load_non_intra_quantiser_matrix) {
52         for (i = 0; i < 64; i++)
53             uis(8, non_intra_quantiser_matrix[i], 1, i);
54     }
55
56     return 0;
57 }
58
59 static int FUNC(user_data)(CodedBitstreamContext *ctx, RWContext *rw,
60                            MPEG2RawUserData *current)
61 {
62     size_t k;
63     int err;
64
65     HEADER("User Data");
66
67     ui(8, user_data_start_code);
68
69 #ifdef READ
70     k = get_bits_left(rw);
71     av_assert0(k % 8 == 0);
72     current->user_data_length = k /= 8;
73     if (k > 0) {
74         current->user_data_ref = av_buffer_alloc(k);
75         if (!current->user_data_ref)
76             return AVERROR(ENOMEM);
77         current->user_data = current->user_data_ref->data;
78     }
79 #endif
80
81     for (k = 0; k < current->user_data_length; k++)
82         xui(8, user_data, current->user_data[k], 0);
83
84     return 0;
85 }
86
87 static int FUNC(sequence_extension)(CodedBitstreamContext *ctx, RWContext *rw,
88                                     MPEG2RawSequenceExtension *current)
89 {
90     CodedBitstreamMPEG2Context *mpeg2 = ctx->priv_data;
91     int err;
92
93     HEADER("Sequence Extension");
94
95     ui(8,  profile_and_level_indication);
96     ui(1,  progressive_sequence);
97     ui(2,  chroma_format);
98     ui(2,  horizontal_size_extension);
99     ui(2,  vertical_size_extension);
100
101     mpeg2->horizontal_size = (mpeg2->horizontal_size & 0xfff) |
102         current->horizontal_size_extension << 12;
103     mpeg2->vertical_size = (mpeg2->vertical_size & 0xfff) |
104         current->vertical_size_extension << 12;
105     mpeg2->progressive_sequence = current->progressive_sequence;
106
107     ui(12, bit_rate_extension);
108     marker_bit();
109     ui(8,  vbv_buffer_size_extension);
110     ui(1,  low_delay);
111     ui(2,  frame_rate_extension_n);
112     ui(5,  frame_rate_extension_d);
113
114     return 0;
115 }
116
117 static int FUNC(sequence_display_extension)(CodedBitstreamContext *ctx, RWContext *rw,
118                                             MPEG2RawSequenceDisplayExtension *current)
119 {
120     int err;
121
122     HEADER("Sequence Display Extension");
123
124     ui(3, video_format);
125
126     ui(1, colour_description);
127     if (current->colour_description) {
128         ui(8, colour_primaries);
129         ui(8, transfer_characteristics);
130         ui(8, matrix_coefficients);
131     }
132
133     ui(14, display_horizontal_size);
134     marker_bit();
135     ui(14, display_vertical_size);
136
137     return 0;
138 }
139
140 static int FUNC(group_of_pictures_header)(CodedBitstreamContext *ctx, RWContext *rw,
141                                           MPEG2RawGroupOfPicturesHeader *current)
142 {
143     int err;
144
145     HEADER("Group of Pictures Header");
146
147     ui(8,  group_start_code);
148
149     ui(25, time_code);
150     ui(1,  closed_gop);
151     ui(1,  broken_link);
152
153     return 0;
154 }
155
156 static int FUNC(picture_header)(CodedBitstreamContext *ctx, RWContext *rw,
157                                 MPEG2RawPictureHeader *current)
158 {
159     int err;
160
161     HEADER("Picture Header");
162
163     ui(8,  picture_start_code);
164
165     ui(10, temporal_reference);
166     ui(3,  picture_coding_type);
167     ui(16, vbv_delay);
168
169     if (current->picture_coding_type == 2 ||
170         current->picture_coding_type == 3) {
171         ui(1, full_pel_forward_vector);
172         ui(3, forward_f_code);
173     }
174
175     if (current->picture_coding_type == 3) {
176         ui(1, full_pel_backward_vector);
177         ui(3, backward_f_code);
178     }
179
180     ui(1, extra_bit_picture);
181
182     return 0;
183 }
184
185 static int FUNC(picture_coding_extension)(CodedBitstreamContext *ctx, RWContext *rw,
186                                           MPEG2RawPictureCodingExtension *current)
187 {
188     CodedBitstreamMPEG2Context *mpeg2 = ctx->priv_data;
189     int err;
190
191     HEADER("Picture Coding Extension");
192
193     ui(4, f_code[0][0]);
194     ui(4, f_code[0][1]);
195     ui(4, f_code[1][0]);
196     ui(4, f_code[1][1]);
197
198     ui(2, intra_dc_precision);
199     ui(2, picture_structure);
200     ui(1, top_field_first);
201     ui(1, frame_pred_frame_dct);
202     ui(1, concealment_motion_vectors);
203     ui(1, q_scale_type);
204     ui(1, intra_vlc_format);
205     ui(1, alternate_scan);
206     ui(1, repeat_first_field);
207     ui(1, chroma_420_type);
208     ui(1, progressive_frame);
209
210     if (mpeg2->progressive_sequence) {
211         if (current->repeat_first_field) {
212             if (current->top_field_first)
213                 mpeg2->number_of_frame_centre_offsets = 3;
214             else
215                 mpeg2->number_of_frame_centre_offsets = 2;
216         } else {
217             mpeg2->number_of_frame_centre_offsets = 1;
218         }
219     } else {
220         if (current->picture_structure == 1 || // Top field.
221             current->picture_structure == 2) { // Bottom field.
222             mpeg2->number_of_frame_centre_offsets = 1;
223         } else {
224             if (current->repeat_first_field)
225                 mpeg2->number_of_frame_centre_offsets = 3;
226             else
227                 mpeg2->number_of_frame_centre_offsets = 2;
228         }
229     }
230
231     ui(1, composite_display_flag);
232     if (current->composite_display_flag) {
233         ui(1, v_axis);
234         ui(3, field_sequence);
235         ui(1, sub_carrier);
236         ui(7, burst_amplitude);
237         ui(8, sub_carrier_phase);
238     }
239
240     return 0;
241 }
242
243 static int FUNC(quant_matrix_extension)(CodedBitstreamContext *ctx, RWContext *rw,
244                                         MPEG2RawQuantMatrixExtension *current)
245 {
246     int err, i;
247
248     HEADER("Quant Matrix Extension");
249
250     ui(1, load_intra_quantiser_matrix);
251     if (current->load_intra_quantiser_matrix) {
252         for (i = 0; i < 64; i++)
253             uis(8, intra_quantiser_matrix[i], 1, i);
254     }
255
256     ui(1, load_non_intra_quantiser_matrix);
257     if (current->load_non_intra_quantiser_matrix) {
258         for (i = 0; i < 64; i++)
259             uis(8, non_intra_quantiser_matrix[i], 1, i);
260     }
261
262     ui(1, load_chroma_intra_quantiser_matrix);
263     if (current->load_chroma_intra_quantiser_matrix) {
264         for (i = 0; i < 64; i++)
265             uis(8, intra_quantiser_matrix[i], 1, i);
266     }
267
268     ui(1, load_chroma_non_intra_quantiser_matrix);
269     if (current->load_chroma_non_intra_quantiser_matrix) {
270         for (i = 0; i < 64; i++)
271             uis(8, chroma_non_intra_quantiser_matrix[i], 1, i);
272     }
273
274     return 0;
275 }
276
277 static int FUNC(picture_display_extension)(CodedBitstreamContext *ctx, RWContext *rw,
278                                            MPEG2RawPictureDisplayExtension *current)
279 {
280     CodedBitstreamMPEG2Context *mpeg2 = ctx->priv_data;
281     int err, i;
282
283     HEADER("Picture Display Extension");
284
285     for (i = 0; i < mpeg2->number_of_frame_centre_offsets; i++) {
286         ui(16, frame_centre_horizontal_offset[i]);
287         marker_bit();
288         ui(16, frame_centre_vertical_offset[i]);
289         marker_bit();
290     }
291
292     return 0;
293 }
294
295 static int FUNC(extension_data)(CodedBitstreamContext *ctx, RWContext *rw,
296                                 MPEG2RawExtensionData *current)
297 {
298     int err;
299
300     HEADER("Extension Data");
301
302     ui(8, extension_start_code);
303     ui(4, extension_start_code_identifier);
304
305     switch (current->extension_start_code_identifier) {
306     case 1:
307         return FUNC(sequence_extension)
308             (ctx, rw, &current->data.sequence);
309     case 2:
310         return FUNC(sequence_display_extension)
311             (ctx, rw, &current->data.sequence_display);
312     case 3:
313         return FUNC(quant_matrix_extension)
314             (ctx, rw, &current->data.quant_matrix);
315     case 7:
316         return FUNC(picture_display_extension)
317             (ctx, rw, &current->data.picture_display);
318     case 8:
319         return FUNC(picture_coding_extension)
320             (ctx, rw, &current->data.picture_coding);
321     default:
322         av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid extension ID %d.\n",
323                current->extension_start_code_identifier);
324         return AVERROR_INVALIDDATA;
325     }
326 }
327
328 static int FUNC(slice_header)(CodedBitstreamContext *ctx, RWContext *rw,
329                               MPEG2RawSliceHeader *current)
330 {
331     CodedBitstreamMPEG2Context *mpeg2 = ctx->priv_data;
332     int err;
333
334     HEADER("Slice Header");
335
336     ui(8, slice_vertical_position);
337
338     if (mpeg2->vertical_size > 2800)
339         ui(3, slice_vertical_position_extension);
340     if (mpeg2->scalable) {
341         if (mpeg2->scalable_mode == 0)
342             ui(7, priority_breakpoint);
343     }
344
345     ui(5, quantiser_scale_code);
346
347     if (nextbits(1, 1, current->slice_extension_flag)) {
348         ui(1, slice_extension_flag);
349         ui(1, intra_slice);
350         ui(1, slice_picture_id_enable);
351         ui(6, slice_picture_id);
352
353         {
354             size_t k;
355 #ifdef READ
356             GetBitContext start;
357             uint8_t bit;
358             start = *rw;
359             for (k = 0; nextbits(1, 1, bit); k++)
360                 skip_bits(rw, 8);
361             current->extra_information_length = k;
362             if (k > 0) {
363                 *rw = start;
364                 current->extra_information =
365                     av_malloc(current->extra_information_length);
366                 if (!current->extra_information)
367                     return AVERROR(ENOMEM);
368                 for (k = 0; k < current->extra_information_length; k++) {
369                     xui(1, extra_bit_slice, bit, 0);
370                     xui(8, extra_information_slice[k],
371                         current->extra_information[k], 1, k);
372                 }
373             }
374 #else
375             for (k = 0; k < current->extra_information_length; k++) {
376                 xui(1, extra_bit_slice, 1, 0);
377                 xui(8, extra_information_slice[k],
378                     current->extra_information[k], 1, k);
379             }
380 #endif
381         }
382     }
383     ui(1, extra_bit_slice);
384
385     return 0;
386 }