yading@11: /*
yading@11:  * Image format
yading@11:  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
yading@11:  * Copyright (c) 2004 Michael Niedermayer
yading@11:  *
yading@11:  * This file is part of FFmpeg.
yading@11:  *
yading@11:  * FFmpeg is free software; you can redistribute it and/or
yading@11:  * modify it under the terms of the GNU Lesser General Public
yading@11:  * License as published by the Free Software Foundation; either
yading@11:  * version 2.1 of the License, or (at your option) any later version.
yading@11:  *
yading@11:  * FFmpeg is distributed in the hope that it will be useful,
yading@11:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@11:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@11:  * Lesser General Public License for more details.
yading@11:  *
yading@11:  * You should have received a copy of the GNU Lesser General Public
yading@11:  * License along with FFmpeg; if not, write to the Free Software
yading@11:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@11:  */
yading@11: 
yading@11: #include "libavutil/intreadwrite.h"
yading@11: #include "libavutil/avstring.h"
yading@11: #include "libavutil/log.h"
yading@11: #include "libavutil/opt.h"
yading@11: #include "libavutil/pixdesc.h"
yading@11: #include "avformat.h"
yading@11: #include "avio_internal.h"
yading@11: #include "internal.h"
yading@11: #include "libavutil/opt.h"
yading@11: 
yading@11: typedef struct {
yading@11:     const AVClass *class;  /**< Class for private options. */
yading@11:     int img_number;
yading@11:     int is_pipe;
yading@11:     int split_planes;       /**< use independent file for each Y, U, V plane */
yading@11:     char path[1024];
yading@11:     int update;
yading@11: } VideoMuxData;
yading@11: 
yading@11: static int write_header(AVFormatContext *s)
yading@11: {
yading@11:     VideoMuxData *img = s->priv_data;
yading@11:     AVStream *st = s->streams[0];
yading@11:     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(st->codec->pix_fmt);
yading@11:     const char *str;
yading@11: 
yading@11:     av_strlcpy(img->path, s->filename, sizeof(img->path));
yading@11: 
yading@11:     /* find format */
yading@11:     if (s->oformat->flags & AVFMT_NOFILE)
yading@11:         img->is_pipe = 0;
yading@11:     else
yading@11:         img->is_pipe = 1;
yading@11: 
yading@11:     str = strrchr(img->path, '.');
yading@11:     img->split_planes =     str
yading@11:                          && !av_strcasecmp(str + 1, "y")
yading@11:                          && s->nb_streams == 1
yading@11:                          && st->codec->codec_id == AV_CODEC_ID_RAWVIDEO
yading@11:                          && desc
yading@11:                          &&(desc->flags & PIX_FMT_PLANAR)
yading@11:                          && desc->nb_components >= 3;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int write_packet(AVFormatContext *s, AVPacket *pkt)
yading@11: {
yading@11:     VideoMuxData *img = s->priv_data;
yading@11:     AVIOContext *pb[4];
yading@11:     char filename[1024];
yading@11:     AVCodecContext *codec = s->streams[pkt->stream_index]->codec;
yading@11:     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(codec->pix_fmt);
yading@11:     int i;
yading@11: 
yading@11:     if (!img->is_pipe) {
yading@11:         if (img->update) {
yading@11:             av_strlcpy(filename, img->path, sizeof(filename));
yading@11:         } else if (av_get_frame_filename(filename, sizeof(filename), img->path, img->img_number) < 0 &&
yading@11:                    img->img_number > 1) {
yading@11:             av_log(s, AV_LOG_ERROR,
yading@11:                    "Could not get frame filename number %d from pattern '%s' (either set updatefirst or use a pattern like %%03d within the filename pattern)\n",
yading@11:                    img->img_number, img->path);
yading@11:             return AVERROR(EINVAL);
yading@11:         }
yading@11:         for (i = 0; i < 4; i++) {
yading@11:             if (avio_open2(&pb[i], filename, AVIO_FLAG_WRITE,
yading@11:                            &s->interrupt_callback, NULL) < 0) {
yading@11:                 av_log(s, AV_LOG_ERROR, "Could not open file : %s\n", filename);
yading@11:                 return AVERROR(EIO);
yading@11:             }
yading@11: 
yading@11:             if (!img->split_planes || i+1 >= desc->nb_components)
yading@11:                 break;
yading@11:             filename[strlen(filename) - 1] = ((int[]){'U','V','A','x'})[i];
yading@11:         }
yading@11:     } else {
yading@11:         pb[0] = s->pb;
yading@11:     }
yading@11: 
yading@11:     if (img->split_planes) {
yading@11:         int ysize = codec->width * codec->height;
yading@11:         int usize = ((-codec->width)>>desc->log2_chroma_w) * ((-codec->height)>>desc->log2_chroma_h);
yading@11:         if (desc->comp[0].depth_minus1 >= 8) {
yading@11:             ysize *= 2;
yading@11:             usize *= 2;
yading@11:         }
yading@11:         avio_write(pb[0], pkt->data                , ysize);
yading@11:         avio_write(pb[1], pkt->data + ysize        , usize);
yading@11:         avio_write(pb[2], pkt->data + ysize + usize, usize);
yading@11:         avio_close(pb[1]);
yading@11:         avio_close(pb[2]);
yading@11:         if (desc->nb_components > 3) {
yading@11:             avio_write(pb[3], pkt->data + ysize + 2*usize, ysize);
yading@11:             avio_close(pb[3]);
yading@11:         }
yading@11:     } else {
yading@11:         avio_write(pb[0], pkt->data, pkt->size);
yading@11:     }
yading@11:     avio_flush(pb[0]);
yading@11:     if (!img->is_pipe) {
yading@11:         avio_close(pb[0]);
yading@11:     }
yading@11: 
yading@11:     img->img_number++;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: #define OFFSET(x) offsetof(VideoMuxData, x)
yading@11: #define ENC AV_OPT_FLAG_ENCODING_PARAM
yading@11: static const AVOption muxoptions[] = {
yading@11:     { "updatefirst",  "continuously overwrite one file", OFFSET(update),  AV_OPT_TYPE_INT, { .i64 = 0 }, 0,       1, ENC },
yading@11:     { "update",       "continuously overwrite one file", OFFSET(update),  AV_OPT_TYPE_INT, { .i64 = 0 }, 0,       1, ENC },
yading@11:     { "start_number", "set first number in the sequence", OFFSET(img_number), AV_OPT_TYPE_INT,  { .i64 = 1 }, 1, INT_MAX, ENC },
yading@11:     { NULL },
yading@11: };
yading@11: 
yading@11: #if CONFIG_IMAGE2_MUXER
yading@11: static const AVClass img2mux_class = {
yading@11:     .class_name = "image2 muxer",
yading@11:     .item_name  = av_default_item_name,
yading@11:     .option     = muxoptions,
yading@11:     .version    = LIBAVUTIL_VERSION_INT,
yading@11: };
yading@11: 
yading@11: AVOutputFormat ff_image2_muxer = {
yading@11:     .name           = "image2",
yading@11:     .long_name      = NULL_IF_CONFIG_SMALL("image2 sequence"),
yading@11:     .extensions     = "bmp,dpx,jls,jpeg,jpg,ljpg,pam,pbm,pcx,pgm,pgmyuv,png,"
yading@11:                       "ppm,sgi,tga,tif,tiff,jp2,j2c,xwd,sun,ras,rs,im1,im8,im24,"
yading@11:                       "sunras,xbm,xface",
yading@11:     .priv_data_size = sizeof(VideoMuxData),
yading@11:     .video_codec    = AV_CODEC_ID_MJPEG,
yading@11:     .write_header   = write_header,
yading@11:     .write_packet   = write_packet,
yading@11:     .flags          = AVFMT_NOTIMESTAMPS | AVFMT_NODIMENSIONS | AVFMT_NOFILE,
yading@11:     .priv_class     = &img2mux_class,
yading@11: };
yading@11: #endif
yading@11: #if CONFIG_IMAGE2PIPE_MUXER
yading@11: AVOutputFormat ff_image2pipe_muxer = {
yading@11:     .name           = "image2pipe",
yading@11:     .long_name      = NULL_IF_CONFIG_SMALL("piped image2 sequence"),
yading@11:     .priv_data_size = sizeof(VideoMuxData),
yading@11:     .video_codec    = AV_CODEC_ID_MJPEG,
yading@11:     .write_header   = write_header,
yading@11:     .write_packet   = write_packet,
yading@11:     .flags          = AVFMT_NOTIMESTAMPS | AVFMT_NODIMENSIONS
yading@11: };
yading@11: #endif