git.sesse.net Git - ffmpeg/blob - libavformat/yuv4mpegenc.c

   1 /*
   2  * YUV4MPEG muxer
   3  * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
   4  *
   5  * This file is part of FFmpeg.
   6  *
   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.
  11  *
  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.
  16  *
  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
  20  */
  21
  22 #include "libavutil/pixdesc.h"
  23 #include "avformat.h"
  24 #include "internal.h"
  25 #include "yuv4mpeg.h"
  26
  27 static int yuv4_write_header(AVFormatContext *s)
  28 {
  29     AVStream *st;
  30     AVIOContext *pb = s->pb;
  31     int width, height;
  32     int raten, rated, aspectn, aspectd, ret;
  33     char inter;
  34     const char *colorspace = "";
  35     const char *colorrange = "";
  36     int field_order;
  37
  38     st     = s->streams[0];
  39     width  = st->codecpar->width;
  40     height = st->codecpar->height;
  41     field_order = st->codecpar->field_order;
  42
  43     // TODO: should be avg_frame_rate
  44     av_reduce(&raten, &rated, st->time_base.den,
  45               st->time_base.num, (1UL << 31) - 1);
  46
  47     aspectn = st->sample_aspect_ratio.num;
  48     aspectd = st->sample_aspect_ratio.den;
  49
  50     if (aspectn == 0 && aspectd == 1)
  51         aspectd = 0;  // 0:0 means unknown
  52
  53     switch(st->codecpar->color_range) {
  54     case AVCOL_RANGE_MPEG:
  55         colorrange = " XCOLORRANGE=LIMITED";
  56         break;
  57     case AVCOL_RANGE_JPEG:
  58         colorrange = " XCOLORRANGE=FULL";
  59         break;
  60     default:
  61         break;
  62     }
  63
  64     switch (field_order) {
  65     case AV_FIELD_TB:
  66     case AV_FIELD_TT: inter = 't'; break;
  67     case AV_FIELD_BT:
  68     case AV_FIELD_BB: inter = 'b'; break;
  69     default:          inter = 'p'; break;
  70     }
  71
  72     switch (st->codecpar->format) {
  73     case AV_PIX_FMT_GRAY8:
  74         colorspace = " Cmono";
  75         break;
  76     case AV_PIX_FMT_GRAY9:
  77         colorspace = " Cmono9";
  78         break;
  79     case AV_PIX_FMT_GRAY10:
  80         colorspace = " Cmono10";
  81         break;
  82     case AV_PIX_FMT_GRAY12:
  83         colorspace = " Cmono12";
  84         break;
  85     case AV_PIX_FMT_GRAY16:
  86         colorspace = " Cmono16";
  87         break;
  88     case AV_PIX_FMT_YUV411P:
  89         colorspace = " C411 XYSCSS=411";
  90         break;
  91     case AV_PIX_FMT_YUVJ420P:
  92         colorspace = " C420jpeg XYSCSS=420JPEG";
  93         colorrange = " XCOLORRANGE=FULL";
  94         break;
  95     case AV_PIX_FMT_YUVJ422P:
  96         colorspace = " C422 XYSCSS=422";
  97         colorrange = " XCOLORRANGE=FULL";
  98         break;
  99     case AV_PIX_FMT_YUVJ444P:
 100         colorspace = " C444 XYSCSS=444";
 101         colorrange = " XCOLORRANGE=FULL";
 102         break;
 103     case AV_PIX_FMT_YUV420P:
 104         switch (st->codecpar->chroma_location) {
 105         case AVCHROMA_LOC_TOPLEFT: colorspace = " C420paldv XYSCSS=420PALDV"; break;
 106         case AVCHROMA_LOC_LEFT:    colorspace = " C420mpeg2 XYSCSS=420MPEG2"; break;
 107         default:                   colorspace = " C420jpeg XYSCSS=420JPEG";   break;
 108         }
 109         break;
 110     case AV_PIX_FMT_YUV422P:
 111         colorspace = " C422 XYSCSS=422";
 112         break;
 113     case AV_PIX_FMT_YUV444P:
 114         colorspace = " C444 XYSCSS=444";
 115         break;
 116     case AV_PIX_FMT_YUV420P9:
 117         colorspace = " C420p9 XYSCSS=420P9";
 118         break;
 119     case AV_PIX_FMT_YUV422P9:
 120         colorspace = " C422p9 XYSCSS=422P9";
 121         break;
 122     case AV_PIX_FMT_YUV444P9:
 123         colorspace = " C444p9 XYSCSS=444P9";
 124         break;
 125     case AV_PIX_FMT_YUV420P10:
 126         colorspace = " C420p10 XYSCSS=420P10";
 127         break;
 128     case AV_PIX_FMT_YUV422P10:
 129         colorspace = " C422p10 XYSCSS=422P10";
 130         break;
 131     case AV_PIX_FMT_YUV444P10:
 132         colorspace = " C444p10 XYSCSS=444P10";
 133         break;
 134     case AV_PIX_FMT_YUV420P12:
 135         colorspace = " C420p12 XYSCSS=420P12";
 136         break;
 137     case AV_PIX_FMT_YUV422P12:
 138         colorspace = " C422p12 XYSCSS=422P12";
 139         break;
 140     case AV_PIX_FMT_YUV444P12:
 141         colorspace = " C444p12 XYSCSS=444P12";
 142         break;
 143     case AV_PIX_FMT_YUV420P14:
 144         colorspace = " C420p14 XYSCSS=420P14";
 145         break;
 146     case AV_PIX_FMT_YUV422P14:
 147         colorspace = " C422p14 XYSCSS=422P14";
 148         break;
 149     case AV_PIX_FMT_YUV444P14:
 150         colorspace = " C444p14 XYSCSS=444P14";
 151         break;
 152     case AV_PIX_FMT_YUV420P16:
 153         colorspace = " C420p16 XYSCSS=420P16";
 154         break;
 155     case AV_PIX_FMT_YUV422P16:
 156         colorspace = " C422p16 XYSCSS=422P16";
 157         break;
 158     case AV_PIX_FMT_YUV444P16:
 159         colorspace = " C444p16 XYSCSS=444P16";
 160         break;
 161     }
 162
 163     ret = avio_printf(pb, Y4M_MAGIC " W%d H%d F%d:%d I%c A%d:%d%s%s\n",
 164                       width, height, raten, rated, inter,
 165                       aspectn, aspectd, colorspace, colorrange);
 166     if (ret < 0) {
 167         av_log(s, AV_LOG_ERROR,
 168                "Error. YUV4MPEG stream header write failed.\n");
 169         return ret;
 170     }
 171
 172     return 0;
 173 }
 174
 175
 176 static int yuv4_write_packet(AVFormatContext *s, AVPacket *pkt)
 177 {
 178     AVStream *st = s->streams[pkt->stream_index];
 179     AVIOContext *pb = s->pb;
 180     const AVFrame *frame = (const AVFrame *)pkt->data;
 181     int width, height, h_chroma_shift, v_chroma_shift;
 182     int i;
 183     const uint8_t *ptr, *ptr1, *ptr2;
 184
 185     /* construct frame header */
 186
 187     avio_printf(s->pb, Y4M_FRAME_MAGIC "\n");
 188
 189     width  = st->codecpar->width;
 190     height = st->codecpar->height;
 191
 192     ptr = frame->data[0];
 193
 194     switch (st->codecpar->format) {
 195     case AV_PIX_FMT_GRAY8:
 196     case AV_PIX_FMT_YUV411P:
 197     case AV_PIX_FMT_YUV420P:
 198     case AV_PIX_FMT_YUV422P:
 199     case AV_PIX_FMT_YUV444P:
 200     // TODO: remove YUVJ pixel formats when they are completely removed from the codebase.
 201     case AV_PIX_FMT_YUVJ420P:
 202     case AV_PIX_FMT_YUVJ422P:
 203     case AV_PIX_FMT_YUVJ444P:
 204         break;
 205     case AV_PIX_FMT_GRAY9:
 206     case AV_PIX_FMT_GRAY10:
 207     case AV_PIX_FMT_GRAY12:
 208     case AV_PIX_FMT_GRAY16:
 209     case AV_PIX_FMT_YUV420P9:
 210     case AV_PIX_FMT_YUV422P9:
 211     case AV_PIX_FMT_YUV444P9:
 212     case AV_PIX_FMT_YUV420P10:
 213     case AV_PIX_FMT_YUV422P10:
 214     case AV_PIX_FMT_YUV444P10:
 215     case AV_PIX_FMT_YUV420P12:
 216     case AV_PIX_FMT_YUV422P12:
 217     case AV_PIX_FMT_YUV444P12:
 218     case AV_PIX_FMT_YUV420P14:
 219     case AV_PIX_FMT_YUV422P14:
 220     case AV_PIX_FMT_YUV444P14:
 221     case AV_PIX_FMT_YUV420P16:
 222     case AV_PIX_FMT_YUV422P16:
 223     case AV_PIX_FMT_YUV444P16:
 224         width *= 2;
 225         break;
 226     default:
 227         av_log(s, AV_LOG_ERROR, "The pixel format '%s' is not supported.\n",
 228                av_get_pix_fmt_name(st->codecpar->format));
 229         return AVERROR(EINVAL);
 230     }
 231
 232     for (i = 0; i < height; i++) {
 233         avio_write(pb, ptr, width);
 234         ptr += frame->linesize[0];
 235     }
 236
 237     if (st->codecpar->format != AV_PIX_FMT_GRAY8 && st->codecpar->format != AV_PIX_FMT_GRAY9 &&
 238         st->codecpar->format != AV_PIX_FMT_GRAY10 && st->codecpar->format != AV_PIX_FMT_GRAY12 &&
 239         st->codecpar->format != AV_PIX_FMT_GRAY16) {
 240         // Adjust for smaller Cb and Cr planes
 241         av_pix_fmt_get_chroma_sub_sample(st->codecpar->format, &h_chroma_shift,
 242                                          &v_chroma_shift);
 243         // Shift right, rounding up
 244         width  = AV_CEIL_RSHIFT(width,  h_chroma_shift);
 245         height = AV_CEIL_RSHIFT(height, v_chroma_shift);
 246
 247         ptr1 = frame->data[1];
 248         ptr2 = frame->data[2];
 249         for (i = 0; i < height; i++) {     /* Cb */
 250             avio_write(pb, ptr1, width);
 251             ptr1 += frame->linesize[1];
 252         }
 253         for (i = 0; i < height; i++) {     /* Cr */
 254             avio_write(pb, ptr2, width);
 255             ptr2 += frame->linesize[2];
 256         }
 257     }
 258
 259     return 0;
 260 }
 261
 262 static int yuv4_init(AVFormatContext *s)
 263 {
 264     if (s->nb_streams != 1)
 265         return AVERROR(EIO);
 266
 267     if (s->streams[0]->codecpar->codec_id != AV_CODEC_ID_WRAPPED_AVFRAME) {
 268         av_log(s, AV_LOG_ERROR, "ERROR: Codec not supported.\n");
 269         return AVERROR_INVALIDDATA;
 270     }
 271
 272     switch (s->streams[0]->codecpar->format) {
 273     case AV_PIX_FMT_YUV411P:
 274         av_log(s, AV_LOG_WARNING, "Warning: generating rarely used 4:1:1 YUV "
 275                "stream, some mjpegtools might not work.\n");
 276         break;
 277     case AV_PIX_FMT_GRAY8:
 278     case AV_PIX_FMT_YUV420P:
 279     case AV_PIX_FMT_YUV422P:
 280     case AV_PIX_FMT_YUV444P:
 281     // TODO: remove YUVJ pixel formats when they are completely removed from the codebase.
 282     case AV_PIX_FMT_YUVJ420P:
 283     case AV_PIX_FMT_YUVJ422P:
 284     case AV_PIX_FMT_YUVJ444P:
 285         break;
 286     case AV_PIX_FMT_GRAY9:
 287     case AV_PIX_FMT_GRAY10:
 288     case AV_PIX_FMT_GRAY12:
 289     case AV_PIX_FMT_GRAY16:
 290     case AV_PIX_FMT_YUV420P9:
 291     case AV_PIX_FMT_YUV422P9:
 292     case AV_PIX_FMT_YUV444P9:
 293     case AV_PIX_FMT_YUV420P10:
 294     case AV_PIX_FMT_YUV422P10:
 295     case AV_PIX_FMT_YUV444P10:
 296     case AV_PIX_FMT_YUV420P12:
 297     case AV_PIX_FMT_YUV422P12:
 298     case AV_PIX_FMT_YUV444P12:
 299     case AV_PIX_FMT_YUV420P14:
 300     case AV_PIX_FMT_YUV422P14:
 301     case AV_PIX_FMT_YUV444P14:
 302     case AV_PIX_FMT_YUV420P16:
 303     case AV_PIX_FMT_YUV422P16:
 304     case AV_PIX_FMT_YUV444P16:
 305         if (s->strict_std_compliance >= FF_COMPLIANCE_NORMAL) {
 306             av_log(s, AV_LOG_ERROR, "'%s' is not an official yuv4mpegpipe pixel format. "
 307                    "Use '-strict -1' to encode to this pixel format.\n",
 308                    av_get_pix_fmt_name(s->streams[0]->codecpar->format));
 309             return AVERROR(EINVAL);
 310         }
 311         av_log(s, AV_LOG_WARNING, "Warning: generating non standard YUV stream. "
 312                "Mjpegtools will not work.\n");
 313         break;
 314     default:
 315         av_log(s, AV_LOG_ERROR, "ERROR: yuv4mpeg can only handle "
 316                "yuv444p, yuv422p, yuv420p, yuv411p and gray8 pixel formats. "
 317                "And using 'strict -1' also yuv444p9, yuv422p9, yuv420p9, "
 318                "yuv444p10, yuv422p10, yuv420p10, "
 319                "yuv444p12, yuv422p12, yuv420p12, "
 320                "yuv444p14, yuv422p14, yuv420p14, "
 321                "yuv444p16, yuv422p16, yuv420p16, "
 322                "gray9, gray10, gray12 "
 323                "and gray16 pixel formats. "
 324                "Use -pix_fmt to select one.\n");
 325         return AVERROR(EIO);
 326     }
 327
 328     return 0;
 329 }
 330
 331 const AVOutputFormat ff_yuv4mpegpipe_muxer = {
 332     .name              = "yuv4mpegpipe",
 333     .long_name         = NULL_IF_CONFIG_SMALL("YUV4MPEG pipe"),
 334     .extensions        = "y4m",
 335     .audio_codec       = AV_CODEC_ID_NONE,
 336     .video_codec       = AV_CODEC_ID_WRAPPED_AVFRAME,
 337     .init              = yuv4_init,
 338     .write_header      = yuv4_write_header,
 339     .write_packet      = yuv4_write_packet,
 340 };