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

   1 /*
   2  * RSO muxer
   3  * Copyright (c) 2001 Fabrice Bellard (original AU code)
   4  * Copyright (c) 2010 Rafael Carre
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include "avformat.h"
  24 #include "internal.h"
  25 #include "rawenc.h"
  26 #include "riff.h"
  27 #include "rso.h"
  28
  29 static int rso_write_header(AVFormatContext *s)
  30 {
  31     AVIOContext  *pb  = s->pb;
  32     AVCodecParameters *par = s->streams[0]->codecpar;
  33
  34     if (!par->codec_tag)
  35         return AVERROR_INVALIDDATA;
  36
  37     if (par->channels != 1) {
  38         av_log(s, AV_LOG_ERROR, "RSO only supports mono\n");
  39         return AVERROR_INVALIDDATA;
  40     }
  41
  42     if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
  43         av_log(s, AV_LOG_ERROR, "muxer does not support non seekable output\n");
  44         return AVERROR_INVALIDDATA;
  45     }
  46
  47     /* XXX: find legal sample rates (if any) */
  48     if (par->sample_rate >= 1u<<16) {
  49         av_log(s, AV_LOG_ERROR, "Sample rate must be < 65536\n");
  50         return AVERROR_INVALIDDATA;
  51     }
  52
  53     if (par->codec_id == AV_CODEC_ID_ADPCM_IMA_WAV) {
  54         avpriv_report_missing_feature(s, "ADPCM in RSO");
  55         return AVERROR_PATCHWELCOME;
  56     }
  57
  58     /* format header */
  59     avio_wb16(pb, par->codec_tag);   /* codec ID */
  60     avio_wb16(pb, 0);                /* data size, will be written at EOF */
  61     avio_wb16(pb, par->sample_rate);
  62     avio_wb16(pb, 0x0000);           /* play mode ? (0x0000 = don't loop) */
  63
  64     return 0;
  65 }
  66
  67 static int rso_write_trailer(AVFormatContext *s)
  68 {
  69     AVIOContext *pb = s->pb;
  70     int64_t file_size;
  71     uint16_t coded_file_size;
  72
  73     file_size = avio_tell(pb);
  74
  75     if (file_size < 0)
  76         return file_size;
  77
  78     if (file_size > 0xffff + RSO_HEADER_SIZE) {
  79         av_log(s, AV_LOG_WARNING,
  80                "Output file is too big (%"PRId64" bytes >= 64kB)\n", file_size);
  81         coded_file_size = 0xffff;
  82     } else {
  83         coded_file_size = file_size - RSO_HEADER_SIZE;
  84     }
  85
  86     /* update file size */
  87     avio_seek(pb, 2, SEEK_SET);
  88     avio_wb16(pb, coded_file_size);
  89     avio_seek(pb, file_size, SEEK_SET);
  90
  91     return 0;
  92 }
  93
  94 AVOutputFormat ff_rso_muxer = {
  95     .name           =   "rso",
  96     .long_name      =   NULL_IF_CONFIG_SMALL("Lego Mindstorms RSO"),
  97     .extensions     =   "rso",
  98     .audio_codec    =   AV_CODEC_ID_PCM_U8,
  99     .video_codec    =   AV_CODEC_ID_NONE,
 100     .write_header   =   rso_write_header,
 101     .write_packet   =   ff_raw_write_packet,
 102     .write_trailer  =   rso_write_trailer,
 103     .codec_tag      =   ff_rso_codec_tags_list,
 104     .flags          =   AVFMT_NOTIMESTAMPS,
 105 };