lavc/cbs: Add JPEG support
[ffmpeg.git] / libavcodec / cbs_jpeg_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(frame_header)(CodedBitstreamContext *ctx, RWContext *rw,
20                               JPEGRawFrameHeader *current)
21 {
22     int err, i;
23
24     HEADER("Frame Header");
25
26     u(16, Lf, 8, 8 + 3 * JPEG_MAX_COMPONENTS);
27
28     u(8,  P,  2, 16);
29     u(16, Y,  0, JPEG_MAX_HEIGHT);
30     u(16, X,  1, JPEG_MAX_WIDTH);
31     u(8,  Nf, 1, JPEG_MAX_COMPONENTS);
32
33     for (i = 0; i < current->Nf; i++) {
34         us(8, C[i],  i, 0, JPEG_MAX_COMPONENTS);
35         us(4, H[i],  i, 1, 4);
36         us(4, V[i],  i, 1, 4);
37         us(8, Tq[i], i, 0, 3);
38     }
39
40     return 0;
41 }
42
43 static int FUNC(quantisation_table)(CodedBitstreamContext *ctx, RWContext *rw,
44                                      JPEGRawQuantisationTable *current)
45 {
46     int err, i;
47
48     u(4, Pq, 0, 1);
49     u(4, Tq, 0, 3);
50
51     if (current->Pq) {
52         for (i = 0; i < 64; i++)
53             us(16, Q[i], i, 1, 255);
54     } else {
55         for (i = 0; i < 64; i++)
56             us(8,  Q[i], i, 1, 255);
57     }
58
59     return 0;
60 }
61
62 static int FUNC(dqt)(CodedBitstreamContext *ctx, RWContext *rw,
63                      JPEGRawQuantisationTableSpecification *current)
64 {
65     int err, i, n;
66
67     HEADER("Quantisation Tables");
68
69     u(16, Lq, 2, 2 + 4 * 65);
70     n = current->Lq / 65;
71
72     for (i = 0; i < n; i++)
73         CHECK(FUNC(quantisation_table)(ctx, rw, &current->table[i]));
74
75     return 0;
76 }
77
78 static int FUNC(huffman_table)(CodedBitstreamContext *ctx, RWContext *rw,
79                                JPEGRawHuffmanTable *current)
80 {
81     int err, i, j, ij;
82
83     u(4, Tc, 0, 1);
84     u(4, Th, 0, 3);
85
86     for (i = 0; i < 16; i++)
87         us(8, L[i], i, 0, 224);
88
89     ij = 0;
90     for (i = 0; i < 16; i++) {
91         for (j = 0; j < current->L[i]; j++) {
92             us(8, V[ij], ij, 0, 255);
93             ++ij;
94         }
95     }
96
97     return 0;
98 }
99
100 static int FUNC(dht)(CodedBitstreamContext *ctx, RWContext *rw,
101                      JPEGRawHuffmanTableSpecification *current)
102 {
103     int err, i, j, n;
104
105     HEADER("Huffman Tables");
106
107     u(16, Lh, 2, 2 + 8 * (1 + 16 + 256));
108
109     n = 2;
110     for (i = 0; n < current->Lh; i++) {
111         CHECK(FUNC(huffman_table)(ctx, rw, &current->table[i]));
112
113         ++n;
114         for (j = 0; j < 16; j++)
115             n += 1 + current->table[i].L[j];
116     }
117
118     return 0;
119 }
120
121 static int FUNC(scan_header)(CodedBitstreamContext *ctx, RWContext *rw,
122                              JPEGRawScanHeader *current)
123 {
124     int err, j;
125
126     HEADER("Scan");
127
128     u(16, Ls, 6, 6 + 2 * JPEG_MAX_COMPONENTS);
129
130     u(8, Ns, 1, 4);
131     for (j = 0; j < current->Ns; j++) {
132         us(8, Cs[j], j, 0, JPEG_MAX_COMPONENTS);
133         us(4, Td[j], j, 0, 3);
134         us(4, Ta[j], j, 0, 3);
135     }
136
137     u(8, Ss, 0, 63);
138     u(8, Se, 0, 63);
139     u(4, Ah, 0, 13);
140     u(4, Al, 0, 15);
141
142     return 0;
143 }
144
145 static int FUNC(application_data)(CodedBitstreamContext *ctx, RWContext *rw,
146                                   JPEGRawApplicationData *current)
147 {
148     int err, i;
149
150     HEADER("Application Data");
151
152     u(16, Lp, 2, 65535);
153
154     if (current->Lp > 2) {
155 #ifdef READ
156         current->Ap_ref = av_buffer_alloc(current->Lp - 2);
157         if (!current->Ap_ref)
158             return AVERROR(ENOMEM);
159         current->Ap = current->Ap_ref->data;
160 #endif
161
162         for (i = 0; i < current->Lp - 2; i++)
163             us(8, Ap[i], i, 0, 255);
164     }
165
166     return 0;
167 }
168
169 static int FUNC(comment)(CodedBitstreamContext *ctx, RWContext *rw,
170                          JPEGRawComment *current)
171 {
172     int err, i;
173
174     HEADER("Comment");
175
176     u(16, Lc, 2, 65535);
177
178     if (current->Lc > 2) {
179 #ifdef READ
180         current->Cm_ref = av_buffer_alloc(current->Lc - 2);
181         if (!current->Cm_ref)
182             return AVERROR(ENOMEM);
183         current->Cm = current->Cm_ref->data;
184 #endif
185
186         for (i = 0; i < current->Lc - 2; i++)
187             us(8, Cm[i], i, 0, 255);
188     }
189
190     return 0;
191 }