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 80
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)
58 if (strncmp(header, Y4M_MAGIC, strlen(Y4M_MAGIC)))
61 header_end = &header[i + 1]; // Include space
62 for (tokstart = &header[strlen(Y4M_MAGIC) + 1];
63 tokstart < header_end; tokstart++) {
64 if (*tokstart == 0x20)
66 switch (*tokstart++) {
67 case 'W': // Width. Required.
68 width = strtol(tokstart, &tokend, 10);
71 case 'H': // Height. Required.
72 height = strtol(tokstart, &tokend, 10);
75 case 'C': // Color space
76 if (strncmp("420jpeg", tokstart, 7) == 0) {
77 pix_fmt = AV_PIX_FMT_YUV420P;
78 chroma_sample_location = AVCHROMA_LOC_CENTER;
79 } else if (strncmp("420mpeg2", tokstart, 8) == 0) {
80 pix_fmt = AV_PIX_FMT_YUV420P;
81 chroma_sample_location = AVCHROMA_LOC_LEFT;
82 } else if (strncmp("420paldv", tokstart, 8) == 0) {
83 pix_fmt = AV_PIX_FMT_YUV420P;
84 chroma_sample_location = AVCHROMA_LOC_TOPLEFT;
85 } else if (strncmp("420p16", tokstart, 6) == 0) {
86 pix_fmt = AV_PIX_FMT_YUV420P16;
87 } else if (strncmp("422p16", tokstart, 6) == 0) {
88 pix_fmt = AV_PIX_FMT_YUV422P16;
89 } else if (strncmp("444p16", tokstart, 6) == 0) {
90 pix_fmt = AV_PIX_FMT_YUV444P16;
91 } else if (strncmp("420p14", tokstart, 6) == 0) {
92 pix_fmt = AV_PIX_FMT_YUV420P14;
93 } else if (strncmp("422p14", tokstart, 6) == 0) {
94 pix_fmt = AV_PIX_FMT_YUV422P14;
95 } else if (strncmp("444p14", tokstart, 6) == 0) {
96 pix_fmt = AV_PIX_FMT_YUV444P14;
97 } else if (strncmp("420p12", tokstart, 6) == 0) {
98 pix_fmt = AV_PIX_FMT_YUV420P12;
99 } else if (strncmp("422p12", tokstart, 6) == 0) {
100 pix_fmt = AV_PIX_FMT_YUV422P12;
101 } else if (strncmp("444p12", tokstart, 6) == 0) {
102 pix_fmt = AV_PIX_FMT_YUV444P12;
103 } else if (strncmp("420p10", tokstart, 6) == 0) {
104 pix_fmt = AV_PIX_FMT_YUV420P10;
105 } else if (strncmp("422p10", tokstart, 6) == 0) {
106 pix_fmt = AV_PIX_FMT_YUV422P10;
107 } else if (strncmp("444p10", tokstart, 6) == 0) {
108 pix_fmt = AV_PIX_FMT_YUV444P10;
109 } else if (strncmp("420p9", tokstart, 5) == 0) {
110 pix_fmt = AV_PIX_FMT_YUV420P9;
111 } else if (strncmp("422p9", tokstart, 5) == 0) {
112 pix_fmt = AV_PIX_FMT_YUV422P9;
113 } else if (strncmp("444p9", tokstart, 5) == 0) {
114 pix_fmt = AV_PIX_FMT_YUV444P9;
115 } else if (strncmp("420", tokstart, 3) == 0) {
116 pix_fmt = AV_PIX_FMT_YUV420P;
117 chroma_sample_location = AVCHROMA_LOC_CENTER;
118 } else if (strncmp("411", tokstart, 3) == 0) {
119 pix_fmt = AV_PIX_FMT_YUV411P;
120 } else if (strncmp("422", tokstart, 3) == 0) {
121 pix_fmt = AV_PIX_FMT_YUV422P;
122 } else if (strncmp("444alpha", tokstart, 8) == 0 ) {
123 av_log(s, AV_LOG_ERROR, "Cannot handle 4:4:4:4 "
124 "YUV4MPEG stream.\n");
126 } else if (strncmp("444", tokstart, 3) == 0) {
127 pix_fmt = AV_PIX_FMT_YUV444P;
128 } else if (strncmp("mono16", tokstart, 6) == 0) {
129 pix_fmt = AV_PIX_FMT_GRAY16;
130 } else if (strncmp("mono12", tokstart, 6) == 0) {
131 pix_fmt = AV_PIX_FMT_GRAY12;
132 } else if (strncmp("mono10", tokstart, 6) == 0) {
133 pix_fmt = AV_PIX_FMT_GRAY10;
134 } else if (strncmp("mono9", tokstart, 5) == 0) {
135 pix_fmt = AV_PIX_FMT_GRAY9;
136 } else if (strncmp("mono", tokstart, 4) == 0) {
137 pix_fmt = AV_PIX_FMT_GRAY8;
139 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains an unknown "
143 while (tokstart < header_end && *tokstart != 0x20)
146 case 'I': // Interlace type
147 switch (*tokstart++){
149 field_order = AV_FIELD_UNKNOWN;
152 field_order = AV_FIELD_PROGRESSIVE;
155 field_order = AV_FIELD_TT;
158 field_order = AV_FIELD_BB;
161 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains mixed "
162 "interlaced and non-interlaced frames.\n");
164 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
165 return AVERROR(EINVAL);
168 case 'F': // Frame rate
169 sscanf(tokstart, "%d:%d", &raten, &rated); // 0:0 if unknown
170 while (tokstart < header_end && *tokstart != 0x20)
173 case 'A': // Pixel aspect
174 sscanf(tokstart, "%d:%d", &aspectn, &aspectd); // 0:0 if unknown
175 while (tokstart < header_end && *tokstart != 0x20)
178 case 'X': // Vendor extensions
179 if (strncmp("YSCSS=", tokstart, 6) == 0) {
180 // Older nonstandard pixel format representation
182 if (strncmp("420JPEG", tokstart, 7) == 0)
183 alt_pix_fmt = AV_PIX_FMT_YUV420P;
184 else if (strncmp("420MPEG2", tokstart, 8) == 0)
185 alt_pix_fmt = AV_PIX_FMT_YUV420P;
186 else if (strncmp("420PALDV", tokstart, 8) == 0)
187 alt_pix_fmt = AV_PIX_FMT_YUV420P;
188 else if (strncmp("420P9", tokstart, 5) == 0)
189 alt_pix_fmt = AV_PIX_FMT_YUV420P9;
190 else if (strncmp("422P9", tokstart, 5) == 0)
191 alt_pix_fmt = AV_PIX_FMT_YUV422P9;
192 else if (strncmp("444P9", tokstart, 5) == 0)
193 alt_pix_fmt = AV_PIX_FMT_YUV444P9;
194 else if (strncmp("420P10", tokstart, 6) == 0)
195 alt_pix_fmt = AV_PIX_FMT_YUV420P10;
196 else if (strncmp("422P10", tokstart, 6) == 0)
197 alt_pix_fmt = AV_PIX_FMT_YUV422P10;
198 else if (strncmp("444P10", tokstart, 6) == 0)
199 alt_pix_fmt = AV_PIX_FMT_YUV444P10;
200 else if (strncmp("420P12", tokstart, 6) == 0)
201 alt_pix_fmt = AV_PIX_FMT_YUV420P12;
202 else if (strncmp("422P12", tokstart, 6) == 0)
203 alt_pix_fmt = AV_PIX_FMT_YUV422P12;
204 else if (strncmp("444P12", tokstart, 6) == 0)
205 alt_pix_fmt = AV_PIX_FMT_YUV444P12;
206 else if (strncmp("420P14", tokstart, 6) == 0)
207 alt_pix_fmt = AV_PIX_FMT_YUV420P14;
208 else if (strncmp("422P14", tokstart, 6) == 0)
209 alt_pix_fmt = AV_PIX_FMT_YUV422P14;
210 else if (strncmp("444P14", tokstart, 6) == 0)
211 alt_pix_fmt = AV_PIX_FMT_YUV444P14;
212 else if (strncmp("420P16", tokstart, 6) == 0)
213 alt_pix_fmt = AV_PIX_FMT_YUV420P16;
214 else if (strncmp("422P16", tokstart, 6) == 0)
215 alt_pix_fmt = AV_PIX_FMT_YUV422P16;
216 else if (strncmp("444P16", tokstart, 6) == 0)
217 alt_pix_fmt = AV_PIX_FMT_YUV444P16;
218 else if (strncmp("411", tokstart, 3) == 0)
219 alt_pix_fmt = AV_PIX_FMT_YUV411P;
220 else if (strncmp("422", tokstart, 3) == 0)
221 alt_pix_fmt = AV_PIX_FMT_YUV422P;
222 else if (strncmp("444", tokstart, 3) == 0)
223 alt_pix_fmt = AV_PIX_FMT_YUV444P;
224 } else if (strncmp("COLORRANGE=", tokstart, 11) == 0) {
226 if (strncmp("FULL",tokstart, 4) == 0)
227 color_range = AVCOL_RANGE_JPEG;
228 else if (strncmp("LIMITED", tokstart, 7) == 0)
229 color_range = AVCOL_RANGE_MPEG;
231 while (tokstart < header_end && *tokstart != 0x20)
237 if (width == -1 || height == -1) {
238 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
242 if (pix_fmt == AV_PIX_FMT_NONE) {
243 if (alt_pix_fmt == AV_PIX_FMT_NONE)
244 pix_fmt = AV_PIX_FMT_YUV420P;
246 pix_fmt = alt_pix_fmt;
249 if (raten <= 0 || rated <= 0) {
250 // Frame rate unknown
255 if (aspectn == 0 && aspectd == 0) {
256 // Pixel aspect unknown
260 st = avformat_new_stream(s, NULL);
262 return AVERROR(ENOMEM);
263 st->codecpar->width = width;
264 st->codecpar->height = height;
265 av_reduce(&raten, &rated, raten, rated, (1UL << 31) - 1);
266 avpriv_set_pts_info(st, 64, rated, raten);
267 st->avg_frame_rate = av_inv_q(st->time_base);
268 st->codecpar->format = pix_fmt;
269 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
270 st->codecpar->codec_id = AV_CODEC_ID_RAWVIDEO;
271 st->sample_aspect_ratio = (AVRational){ aspectn, aspectd };
272 st->codecpar->chroma_location = chroma_sample_location;
273 st->codecpar->color_range = color_range;
274 st->codecpar->field_order = field_order;
275 s->packet_size = av_image_get_buffer_size(st->codecpar->format, width, height, 1) + Y4M_FRAME_MAGIC_LEN;
276 if ((int) s->packet_size < 0)
277 return s->packet_size;
278 s->internal->data_offset = avio_tell(pb);
280 st->duration = (avio_size(pb) - avio_tell(pb)) / s->packet_size;
285 static int yuv4_read_packet(AVFormatContext *s, AVPacket *pkt)
288 char header[MAX_FRAME_HEADER+1];
290 int64_t off = avio_tell(s->pb);
292 for (i = 0; i < MAX_FRAME_HEADER; i++) {
293 header[i] = avio_r8(s->pb);
294 if (header[i] == '\n') {
301 else if (s->pb->eof_reached)
303 else if (i == MAX_FRAME_HEADER)
304 return AVERROR_INVALIDDATA;
306 if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC)))
307 return AVERROR_INVALIDDATA;
309 ret = av_get_packet(s->pb, pkt, s->packet_size - Y4M_FRAME_MAGIC_LEN);
312 else if (ret != s->packet_size - Y4M_FRAME_MAGIC_LEN) {
313 av_packet_unref(pkt);
314 return s->pb->eof_reached ? AVERROR_EOF : AVERROR(EIO);
316 pkt->stream_index = 0;
317 pkt->pts = (off - s->internal->data_offset) / s->packet_size;
322 static int yuv4_read_seek(AVFormatContext *s, int stream_index,
323 int64_t pts, int flags)
327 if (flags & AVSEEK_FLAG_BACKWARD)
328 pts = FFMAX(0, pts - 1);
331 pos = pts * s->packet_size;
333 if (avio_seek(s->pb, pos + s->internal->data_offset, SEEK_SET) < 0)
338 static int yuv4_probe(const AVProbeData *pd)
340 /* check file header */
341 if (strncmp(pd->buf, Y4M_MAGIC, sizeof(Y4M_MAGIC) - 1) == 0)
342 return AVPROBE_SCORE_MAX;
347 AVInputFormat ff_yuv4mpegpipe_demuxer = {
348 .name = "yuv4mpegpipe",
349 .long_name = NULL_IF_CONFIG_SMALL("YUV4MPEG pipe"),
350 .read_probe = yuv4_probe,
351 .read_header = yuv4_read_header,
352 .read_packet = yuv4_read_packet,
353 .read_seek = yuv4_read_seek,