3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #include "libavutil/imgutils.h"
28 /* Header size increased to allow room for optional flags */
29 #define MAX_YUV4_HEADER 96
30 #define MAX_FRAME_HEADER 80
32 static int yuv4_read_header(AVFormatContext *s)
34 char header[MAX_YUV4_HEADER + 10]; // Include headroom for
36 char *tokstart, *tokend, *header_end;
38 AVIOContext *pb = s->pb;
39 int width = -1, height = -1, raten = 0,
40 rated = 0, aspectn = 0, aspectd = 0;
41 enum AVPixelFormat pix_fmt = AV_PIX_FMT_NONE, alt_pix_fmt = AV_PIX_FMT_NONE;
42 enum AVChromaLocation chroma_sample_location = AVCHROMA_LOC_UNSPECIFIED;
43 enum AVFieldOrder field_order = AV_FIELD_UNKNOWN;
44 enum AVColorRange color_range = AVCOL_RANGE_UNSPECIFIED;
47 for (i = 0; i < MAX_YUV4_HEADER; i++) {
48 header[i] = avio_r8(pb);
49 if (header[i] == '\n') {
50 header[i + 1] = 0x20; // Add a space after last option.
51 // Makes parsing "444" vs "444alpha" easier.
56 if (i == MAX_YUV4_HEADER) {
57 av_log(s, AV_LOG_ERROR, "Header too large.\n");
58 return AVERROR(EINVAL);
60 if (strncmp(header, Y4M_MAGIC, strlen(Y4M_MAGIC))) {
61 av_log(s, AV_LOG_ERROR, "Invalid magic number for yuv4mpeg.\n");
62 return AVERROR(EINVAL);
65 header_end = &header[i + 1]; // Include space
66 for (tokstart = &header[strlen(Y4M_MAGIC) + 1];
67 tokstart < header_end; tokstart++) {
68 if (*tokstart == 0x20)
70 switch (*tokstart++) {
71 case 'W': // Width. Required.
72 width = strtol(tokstart, &tokend, 10);
75 case 'H': // Height. Required.
76 height = strtol(tokstart, &tokend, 10);
79 case 'C': // Color space
80 if (strncmp("420jpeg", tokstart, 7) == 0) {
81 pix_fmt = AV_PIX_FMT_YUV420P;
82 chroma_sample_location = AVCHROMA_LOC_CENTER;
83 } else if (strncmp("420mpeg2", tokstart, 8) == 0) {
84 pix_fmt = AV_PIX_FMT_YUV420P;
85 chroma_sample_location = AVCHROMA_LOC_LEFT;
86 } else if (strncmp("420paldv", tokstart, 8) == 0) {
87 pix_fmt = AV_PIX_FMT_YUV420P;
88 chroma_sample_location = AVCHROMA_LOC_TOPLEFT;
89 } else if (strncmp("420p16", tokstart, 6) == 0) {
90 pix_fmt = AV_PIX_FMT_YUV420P16;
91 } else if (strncmp("422p16", tokstart, 6) == 0) {
92 pix_fmt = AV_PIX_FMT_YUV422P16;
93 } else if (strncmp("444p16", tokstart, 6) == 0) {
94 pix_fmt = AV_PIX_FMT_YUV444P16;
95 } else if (strncmp("420p14", tokstart, 6) == 0) {
96 pix_fmt = AV_PIX_FMT_YUV420P14;
97 } else if (strncmp("422p14", tokstart, 6) == 0) {
98 pix_fmt = AV_PIX_FMT_YUV422P14;
99 } else if (strncmp("444p14", tokstart, 6) == 0) {
100 pix_fmt = AV_PIX_FMT_YUV444P14;
101 } else if (strncmp("420p12", tokstart, 6) == 0) {
102 pix_fmt = AV_PIX_FMT_YUV420P12;
103 } else if (strncmp("422p12", tokstart, 6) == 0) {
104 pix_fmt = AV_PIX_FMT_YUV422P12;
105 } else if (strncmp("444p12", tokstart, 6) == 0) {
106 pix_fmt = AV_PIX_FMT_YUV444P12;
107 } else if (strncmp("420p10", tokstart, 6) == 0) {
108 pix_fmt = AV_PIX_FMT_YUV420P10;
109 } else if (strncmp("422p10", tokstart, 6) == 0) {
110 pix_fmt = AV_PIX_FMT_YUV422P10;
111 } else if (strncmp("444p10", tokstart, 6) == 0) {
112 pix_fmt = AV_PIX_FMT_YUV444P10;
113 } else if (strncmp("420p9", tokstart, 5) == 0) {
114 pix_fmt = AV_PIX_FMT_YUV420P9;
115 } else if (strncmp("422p9", tokstart, 5) == 0) {
116 pix_fmt = AV_PIX_FMT_YUV422P9;
117 } else if (strncmp("444p9", tokstart, 5) == 0) {
118 pix_fmt = AV_PIX_FMT_YUV444P9;
119 } else if (strncmp("420", tokstart, 3) == 0) {
120 pix_fmt = AV_PIX_FMT_YUV420P;
121 chroma_sample_location = AVCHROMA_LOC_CENTER;
122 } else if (strncmp("411", tokstart, 3) == 0) {
123 pix_fmt = AV_PIX_FMT_YUV411P;
124 } else if (strncmp("422", tokstart, 3) == 0) {
125 pix_fmt = AV_PIX_FMT_YUV422P;
126 } else if (strncmp("444alpha", tokstart, 8) == 0 ) {
127 pix_fmt = AV_PIX_FMT_YUVA444P;
128 } else if (strncmp("444", tokstart, 3) == 0) {
129 pix_fmt = AV_PIX_FMT_YUV444P;
130 } else if (strncmp("mono16", tokstart, 6) == 0) {
131 pix_fmt = AV_PIX_FMT_GRAY16;
132 } else if (strncmp("mono12", tokstart, 6) == 0) {
133 pix_fmt = AV_PIX_FMT_GRAY12;
134 } else if (strncmp("mono10", tokstart, 6) == 0) {
135 pix_fmt = AV_PIX_FMT_GRAY10;
136 } else if (strncmp("mono9", tokstart, 5) == 0) {
137 pix_fmt = AV_PIX_FMT_GRAY9;
138 } else if (strncmp("mono", tokstart, 4) == 0) {
139 pix_fmt = AV_PIX_FMT_GRAY8;
141 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains an unknown "
143 return AVERROR_INVALIDDATA;
145 while (tokstart < header_end && *tokstart != 0x20)
148 case 'I': // Interlace type
149 switch (*tokstart++){
151 field_order = AV_FIELD_UNKNOWN;
154 field_order = AV_FIELD_PROGRESSIVE;
157 field_order = AV_FIELD_TT;
160 field_order = AV_FIELD_BB;
163 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains mixed "
164 "interlaced and non-interlaced frames.\n");
166 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
167 return AVERROR(EINVAL);
170 case 'F': // Frame rate
171 sscanf(tokstart, "%d:%d", &raten, &rated); // 0:0 if unknown
172 while (tokstart < header_end && *tokstart != 0x20)
175 case 'A': // Pixel aspect
176 sscanf(tokstart, "%d:%d", &aspectn, &aspectd); // 0:0 if unknown
177 while (tokstart < header_end && *tokstart != 0x20)
180 case 'X': // Vendor extensions
181 if (strncmp("YSCSS=", tokstart, 6) == 0) {
182 // Older nonstandard pixel format representation
184 if (strncmp("420JPEG", tokstart, 7) == 0)
185 alt_pix_fmt = AV_PIX_FMT_YUV420P;
186 else if (strncmp("420MPEG2", tokstart, 8) == 0)
187 alt_pix_fmt = AV_PIX_FMT_YUV420P;
188 else if (strncmp("420PALDV", tokstart, 8) == 0)
189 alt_pix_fmt = AV_PIX_FMT_YUV420P;
190 else if (strncmp("420P9", tokstart, 5) == 0)
191 alt_pix_fmt = AV_PIX_FMT_YUV420P9;
192 else if (strncmp("422P9", tokstart, 5) == 0)
193 alt_pix_fmt = AV_PIX_FMT_YUV422P9;
194 else if (strncmp("444P9", tokstart, 5) == 0)
195 alt_pix_fmt = AV_PIX_FMT_YUV444P9;
196 else if (strncmp("420P10", tokstart, 6) == 0)
197 alt_pix_fmt = AV_PIX_FMT_YUV420P10;
198 else if (strncmp("422P10", tokstart, 6) == 0)
199 alt_pix_fmt = AV_PIX_FMT_YUV422P10;
200 else if (strncmp("444P10", tokstart, 6) == 0)
201 alt_pix_fmt = AV_PIX_FMT_YUV444P10;
202 else if (strncmp("420P12", tokstart, 6) == 0)
203 alt_pix_fmt = AV_PIX_FMT_YUV420P12;
204 else if (strncmp("422P12", tokstart, 6) == 0)
205 alt_pix_fmt = AV_PIX_FMT_YUV422P12;
206 else if (strncmp("444P12", tokstart, 6) == 0)
207 alt_pix_fmt = AV_PIX_FMT_YUV444P12;
208 else if (strncmp("420P14", tokstart, 6) == 0)
209 alt_pix_fmt = AV_PIX_FMT_YUV420P14;
210 else if (strncmp("422P14", tokstart, 6) == 0)
211 alt_pix_fmt = AV_PIX_FMT_YUV422P14;
212 else if (strncmp("444P14", tokstart, 6) == 0)
213 alt_pix_fmt = AV_PIX_FMT_YUV444P14;
214 else if (strncmp("420P16", tokstart, 6) == 0)
215 alt_pix_fmt = AV_PIX_FMT_YUV420P16;
216 else if (strncmp("422P16", tokstart, 6) == 0)
217 alt_pix_fmt = AV_PIX_FMT_YUV422P16;
218 else if (strncmp("444P16", tokstart, 6) == 0)
219 alt_pix_fmt = AV_PIX_FMT_YUV444P16;
220 else if (strncmp("411", tokstart, 3) == 0)
221 alt_pix_fmt = AV_PIX_FMT_YUV411P;
222 else if (strncmp("422", tokstart, 3) == 0)
223 alt_pix_fmt = AV_PIX_FMT_YUV422P;
224 else if (strncmp("444", tokstart, 3) == 0)
225 alt_pix_fmt = AV_PIX_FMT_YUV444P;
226 } else if (strncmp("COLORRANGE=", tokstart, 11) == 0) {
228 if (strncmp("FULL",tokstart, 4) == 0)
229 color_range = AVCOL_RANGE_JPEG;
230 else if (strncmp("LIMITED", tokstart, 7) == 0)
231 color_range = AVCOL_RANGE_MPEG;
233 while (tokstart < header_end && *tokstart != 0x20)
239 if (width == -1 || height == -1) {
240 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
241 return AVERROR_INVALIDDATA;
244 if (pix_fmt == AV_PIX_FMT_NONE) {
245 if (alt_pix_fmt == AV_PIX_FMT_NONE)
246 pix_fmt = AV_PIX_FMT_YUV420P;
248 pix_fmt = alt_pix_fmt;
251 if (raten <= 0 || rated <= 0) {
252 // Frame rate unknown
257 if (aspectn == 0 && aspectd == 0) {
258 // Pixel aspect unknown
262 st = avformat_new_stream(s, NULL);
264 return AVERROR(ENOMEM);
265 st->codecpar->width = width;
266 st->codecpar->height = height;
267 av_reduce(&raten, &rated, raten, rated, (1UL << 31) - 1);
268 avpriv_set_pts_info(st, 64, rated, raten);
269 st->avg_frame_rate = av_inv_q(st->time_base);
270 st->codecpar->format = pix_fmt;
271 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
272 st->codecpar->codec_id = AV_CODEC_ID_RAWVIDEO;
273 st->sample_aspect_ratio = (AVRational){ aspectn, aspectd };
274 st->codecpar->chroma_location = chroma_sample_location;
275 st->codecpar->color_range = color_range;
276 st->codecpar->field_order = field_order;
277 s->packet_size = av_image_get_buffer_size(st->codecpar->format, width, height, 1) + Y4M_FRAME_MAGIC_LEN;
278 if ((int) s->packet_size < 0)
279 return s->packet_size;
280 s->internal->data_offset = avio_tell(pb);
282 st->duration = (avio_size(pb) - avio_tell(pb)) / s->packet_size;
287 static int yuv4_read_packet(AVFormatContext *s, AVPacket *pkt)
290 char header[MAX_FRAME_HEADER+1];
292 int64_t off = avio_tell(s->pb);
294 for (i = 0; i < MAX_FRAME_HEADER; i++) {
295 header[i] = avio_r8(s->pb);
296 if (header[i] == '\n') {
303 else if (s->pb->eof_reached)
305 else if (i == MAX_FRAME_HEADER)
306 return AVERROR_INVALIDDATA;
308 if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC)))
309 return AVERROR_INVALIDDATA;
311 ret = av_get_packet(s->pb, pkt, s->packet_size - Y4M_FRAME_MAGIC_LEN);
314 else if (ret != s->packet_size - Y4M_FRAME_MAGIC_LEN) {
315 return s->pb->eof_reached ? AVERROR_EOF : AVERROR(EIO);
317 pkt->stream_index = 0;
318 pkt->pts = (off - s->internal->data_offset) / s->packet_size;
323 static int yuv4_read_seek(AVFormatContext *s, int stream_index,
324 int64_t pts, int flags)
328 if (flags & AVSEEK_FLAG_BACKWARD)
329 pts = FFMAX(0, pts - 1);
332 pos = pts * s->packet_size;
334 if (avio_seek(s->pb, pos + s->internal->data_offset, SEEK_SET) < 0)
339 static int yuv4_probe(const AVProbeData *pd)
341 /* check file header */
342 if (strncmp(pd->buf, Y4M_MAGIC, sizeof(Y4M_MAGIC) - 1) == 0)
343 return AVPROBE_SCORE_MAX;
348 AVInputFormat ff_yuv4mpegpipe_demuxer = {
349 .name = "yuv4mpegpipe",
350 .long_name = NULL_IF_CONFIG_SMALL("YUV4MPEG pipe"),
351 .read_probe = yuv4_probe,
352 .read_header = yuv4_read_header,
353 .read_packet = yuv4_read_packet,
354 .read_seek = yuv4_read_seek,