yuv4mpeg: split the demuxer and muxer into separate files
[ffmpeg.git] / libavformat / yuv4mpegenc.c
1 /*
2  * YUV4MPEG muxer
3  * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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.
11  *
12  * Libav 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.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/pixdesc.h"
23 #include "avformat.h"
24 #include "internal.h"
25 #include "yuv4mpeg.h"
26
27 #define Y4M_LINE_MAX 256
28
29 static int yuv4_generate_header(AVFormatContext *s, char* buf)
30 {
31     AVStream *st;
32     int width, height;
33     int raten, rated, aspectn, aspectd, n;
34     char inter;
35     const char *colorspace = "";
36
37     st     = s->streams[0];
38     width  = st->codec->width;
39     height = st->codec->height;
40
41     av_reduce(&raten, &rated, st->codec->time_base.den,
42               st->codec->time_base.num, (1UL << 31) - 1);
43
44     aspectn = st->sample_aspect_ratio.num;
45     aspectd = st->sample_aspect_ratio.den;
46
47     if (aspectn == 0 && aspectd == 1)
48         aspectd = 0;  // 0:0 means unknown
49
50     inter = 'p'; /* progressive is the default */
51     if (st->codec->coded_frame && st->codec->coded_frame->interlaced_frame)
52         inter = st->codec->coded_frame->top_field_first ? 't' : 'b';
53
54     switch (st->codec->pix_fmt) {
55     case AV_PIX_FMT_GRAY8:
56         colorspace = " Cmono";
57         break;
58     case AV_PIX_FMT_YUV411P:
59         colorspace = " C411 XYSCSS=411";
60         break;
61     case AV_PIX_FMT_YUV420P:
62         switch (st->codec->chroma_sample_location) {
63         case AVCHROMA_LOC_TOPLEFT: colorspace = " C420paldv XYSCSS=420PALDV"; break;
64         case AVCHROMA_LOC_LEFT:    colorspace = " C420mpeg2 XYSCSS=420MPEG2"; break;
65         default:                   colorspace = " C420jpeg XYSCSS=420JPEG";   break;
66         }
67         break;
68     case AV_PIX_FMT_YUV422P:
69         colorspace = " C422 XYSCSS=422";
70         break;
71     case AV_PIX_FMT_YUV444P:
72         colorspace = " C444 XYSCSS=444";
73         break;
74     }
75
76     /* construct stream header, if this is the first frame */
77     n = snprintf(buf, Y4M_LINE_MAX, "%s W%d H%d F%d:%d I%c A%d:%d%s\n",
78                  Y4M_MAGIC, width, height, raten, rated, inter,
79                  aspectn, aspectd, colorspace);
80
81     return n;
82 }
83
84 static int yuv4_write_packet(AVFormatContext *s, AVPacket *pkt)
85 {
86     AVStream *st = s->streams[pkt->stream_index];
87     AVIOContext *pb = s->pb;
88     AVPicture *picture;
89     int* first_pkt = s->priv_data;
90     int width, height, h_chroma_shift, v_chroma_shift;
91     int i;
92     char buf2[Y4M_LINE_MAX + 1];
93     char buf1[20];
94     uint8_t *ptr, *ptr1, *ptr2;
95
96     picture = (AVPicture *)pkt->data;
97
98     /* for the first packet we have to output the header as well */
99     if (*first_pkt) {
100         *first_pkt = 0;
101         if (yuv4_generate_header(s, buf2) < 0) {
102             av_log(s, AV_LOG_ERROR,
103                    "Error. YUV4MPEG stream header write failed.\n");
104             return AVERROR(EIO);
105         } else {
106             avio_write(pb, buf2, strlen(buf2));
107         }
108     }
109
110     /* construct frame header */
111
112     snprintf(buf1, sizeof(buf1), "%s\n", Y4M_FRAME_MAGIC);
113     avio_write(pb, buf1, strlen(buf1));
114
115     width  = st->codec->width;
116     height = st->codec->height;
117
118     ptr = picture->data[0];
119     for (i = 0; i < height; i++) {
120         avio_write(pb, ptr, width);
121         ptr += picture->linesize[0];
122     }
123
124     if (st->codec->pix_fmt != AV_PIX_FMT_GRAY8) {
125         // Adjust for smaller Cb and Cr planes
126         av_pix_fmt_get_chroma_sub_sample(st->codec->pix_fmt, &h_chroma_shift,
127                                          &v_chroma_shift);
128         // Shift right, rounding up
129         width  = -(-width  >> h_chroma_shift);
130         height = -(-height >> v_chroma_shift);
131
132         ptr1 = picture->data[1];
133         ptr2 = picture->data[2];
134         for (i = 0; i < height; i++) {     /* Cb */
135             avio_write(pb, ptr1, width);
136             ptr1 += picture->linesize[1];
137         }
138         for (i = 0; i < height; i++) {     /* Cr */
139             avio_write(pb, ptr2, width);
140             ptr2 += picture->linesize[2];
141         }
142     }
143     return 0;
144 }
145
146 static int yuv4_write_header(AVFormatContext *s)
147 {
148     int *first_pkt = s->priv_data;
149
150     if (s->nb_streams != 1)
151         return AVERROR(EIO);
152
153     if (s->streams[0]->codec->codec_id != AV_CODEC_ID_RAWVIDEO) {
154         av_log(s, AV_LOG_ERROR, "ERROR: Only rawvideo supported.\n");
155         return AVERROR_INVALIDDATA;
156     }
157
158     if (s->streams[0]->codec->pix_fmt == AV_PIX_FMT_YUV411P) {
159         av_log(s, AV_LOG_ERROR, "Warning: generating rarely used 4:1:1 YUV "
160                "stream, some mjpegtools might not work.\n");
161     } else if ((s->streams[0]->codec->pix_fmt != AV_PIX_FMT_YUV420P) &&
162                (s->streams[0]->codec->pix_fmt != AV_PIX_FMT_YUV422P) &&
163                (s->streams[0]->codec->pix_fmt != AV_PIX_FMT_GRAY8)   &&
164                (s->streams[0]->codec->pix_fmt != AV_PIX_FMT_YUV444P)) {
165         av_log(s, AV_LOG_ERROR, "ERROR: yuv4mpeg only handles yuv444p, "
166                "yuv422p, yuv420p, yuv411p and gray pixel formats. "
167                "Use -pix_fmt to select one.\n");
168         return AVERROR(EIO);
169     }
170
171     *first_pkt = 1;
172     return 0;
173 }
174
175 AVOutputFormat ff_yuv4mpegpipe_muxer = {
176     .name              = "yuv4mpegpipe",
177     .long_name         = NULL_IF_CONFIG_SMALL("YUV4MPEG pipe"),
178     .mime_type         = "",
179     .extensions        = "y4m",
180     .priv_data_size    = sizeof(int),
181     .audio_codec       = AV_CODEC_ID_NONE,
182     .video_codec       = AV_CODEC_ID_RAWVIDEO,
183     .write_header      = yuv4_write_header,
184     .write_packet      = yuv4_write_packet,
185     .flags             = AVFMT_RAWPICTURE,
186 };