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

   1 /*
   2  * SMJPEG muxer
   3  * Copyright (c) 2012 Paul B Mahol
   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 /**
  23  * @file
  24  * This is a muxer for Loki SDL Motion JPEG files
  25  */
  26
  27 #include "avformat.h"
  28 #include "internal.h"
  29 #include "smjpeg.h"
  30
  31 typedef struct SMJPEGMuxContext {
  32     uint32_t duration;
  33 } SMJPEGMuxContext;
  34
  35 static int smjpeg_write_header(AVFormatContext *s)
  36 {
  37     AVDictionaryEntry *t = NULL;
  38     AVIOContext *pb = s->pb;
  39     int n, tag;
  40
  41     if (s->nb_streams > 2) {
  42         av_log(s, AV_LOG_ERROR, "more than >2 streams are not supported\n");
  43         return AVERROR(EINVAL);
  44     }
  45     avio_write(pb, SMJPEG_MAGIC, 8);
  46     avio_wb32(pb, 0);
  47     avio_wb32(pb, 0);
  48
  49     while ((t = av_dict_get(s->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) {
  50         avio_wl32(pb, SMJPEG_TXT);
  51         avio_wb32(pb, strlen(t->key) + strlen(t->value) + 3);
  52         avio_write(pb, t->key, strlen(t->key));
  53         avio_write(pb, " = ", 3);
  54         avio_write(pb, t->value, strlen(t->value));
  55     }
  56
  57     for (n = 0; n < s->nb_streams; n++) {
  58         AVStream *st = s->streams[n];
  59         AVCodecContext *codec = st->codec;
  60         if (codec->codec_type == AVMEDIA_TYPE_AUDIO) {
  61             tag = ff_codec_get_tag(ff_codec_smjpeg_audio_tags, codec->codec_id);
  62             if (!tag) {
  63                 av_log(s, AV_LOG_ERROR, "unsupported audio codec\n");
  64                 return AVERROR(EINVAL);
  65             }
  66             avio_wl32(pb, SMJPEG_SND);
  67             avio_wb32(pb, 8);
  68             avio_wb16(pb, codec->sample_rate);
  69             avio_w8(pb, av_get_bits_per_sample(codec->codec_id));
  70             avio_w8(pb, codec->channels);
  71             avio_wl32(pb, tag);
  72             avpriv_set_pts_info(st, 32, 1, 1000);
  73         } else if (codec->codec_type == AVMEDIA_TYPE_VIDEO) {
  74             tag = ff_codec_get_tag(ff_codec_smjpeg_video_tags, codec->codec_id);
  75             if (!tag) {
  76                 av_log(s, AV_LOG_ERROR, "unsupported video codec\n");
  77                 return AVERROR(EINVAL);
  78             }
  79             avio_wl32(pb, SMJPEG_VID);
  80             avio_wb32(pb, 12);
  81             avio_wb32(pb, 0);
  82             avio_wb16(pb, codec->width);
  83             avio_wb16(pb, codec->height);
  84             avio_wl32(pb, tag);
  85             avpriv_set_pts_info(st, 32, 1, 1000);
  86         }
  87     }
  88
  89     avio_wl32(pb, SMJPEG_HEND);
  90     avio_flush(pb);
  91
  92     return 0;
  93 }
  94
  95 static int smjpeg_write_packet(AVFormatContext *s, AVPacket *pkt)
  96 {
  97     SMJPEGMuxContext *smc = s->priv_data;
  98     AVIOContext *pb = s->pb;
  99     AVStream *st = s->streams[pkt->stream_index];
 100     AVCodecContext *codec = st->codec;
 101
 102     if (codec->codec_type == AVMEDIA_TYPE_AUDIO)
 103         avio_wl32(pb, SMJPEG_SNDD);
 104     else if (codec->codec_type == AVMEDIA_TYPE_VIDEO)
 105         avio_wl32(pb, SMJPEG_VIDD);
 106     else
 107         return 0;
 108
 109     avio_wb32(pb, pkt->pts);
 110     avio_wb32(pb, pkt->size);
 111     avio_write(pb, pkt->data, pkt->size);
 112
 113     smc->duration = FFMAX(smc->duration, pkt->pts + pkt->duration);
 114     return 0;
 115 }
 116
 117 static int smjpeg_write_trailer(AVFormatContext *s)
 118 {
 119     SMJPEGMuxContext *smc = s->priv_data;
 120     AVIOContext *pb = s->pb;
 121     int64_t currentpos;
 122
 123     if (pb->seekable) {
 124         currentpos = avio_tell(pb);
 125         avio_seek(pb, 12, SEEK_SET);
 126         avio_wb32(pb, smc->duration);
 127         avio_seek(pb, currentpos, SEEK_SET);
 128     }
 129
 130     avio_wl32(pb, SMJPEG_DONE);
 131
 132     return 0;
 133 }
 134
 135 AVOutputFormat ff_smjpeg_muxer = {
 136     .name           = "smjpeg",
 137     .long_name      = NULL_IF_CONFIG_SMALL("Loki SDL MJPEG"),
 138     .priv_data_size = sizeof(SMJPEGMuxContext),
 139     .audio_codec    = AV_CODEC_ID_PCM_S16LE,
 140     .video_codec    = AV_CODEC_ID_MJPEG,
 141     .write_header   = smjpeg_write_header,
 142     .write_packet   = smjpeg_write_packet,
 143     .write_trailer  = smjpeg_write_trailer,
 144     .flags          = AVFMT_GLOBALHEADER | AVFMT_TS_NONSTRICT,
 145     .codec_tag      = (const AVCodecTag *const []){ ff_codec_smjpeg_video_tags, ff_codec_smjpeg_audio_tags, 0 },
 146 };