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

   1 /*
   2  * Creative Voice File demuxer.
   3  * Copyright (c) 2006  Aurelien Jacobs <aurel@gnuage.org>
   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 St, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22 #include "voc.h"
  23
  24
  25
  26
  27 static int voc_probe(AVProbeData *p)
  28 {
  29     int version, check;
  30
  31     if (memcmp(p->buf, voc_magic, sizeof(voc_magic) - 1))
  32         return 0;
  33     version = p->buf[22] | (p->buf[23] << 8);
  34     check = p->buf[24] | (p->buf[25] << 8);
  35     if (~version + 0x1234 != check)
  36         return 10;
  37
  38     return AVPROBE_SCORE_MAX;
  39 }
  40
  41 static int voc_read_header(AVFormatContext *s, AVFormatParameters *ap)
  42 {
  43     voc_dec_context_t *voc = s->priv_data;
  44     ByteIOContext *pb = &s->pb;
  45     int header_size;
  46     AVStream *st;
  47
  48     url_fskip(pb, 20);
  49     header_size = get_le16(pb) - 22;
  50     if (header_size != 4) {
  51         av_log(s, AV_LOG_ERROR, "unknown header size: %d\n", header_size);
  52         return AVERROR_NOTSUPP;
  53     }
  54     url_fskip(pb, header_size);
  55     st = av_new_stream(s, 0);
  56     if (!st)
  57         return AVERROR_NOMEM;
  58     st->codec->codec_type = CODEC_TYPE_AUDIO;
  59
  60     voc->remaining_size = 0;
  61     return 0;
  62 }
  63
  64 int
  65 voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
  66 {
  67     voc_dec_context_t *voc = s->priv_data;
  68     AVCodecContext *dec = st->codec;
  69     ByteIOContext *pb = &s->pb;
  70     voc_type_t type;
  71     int size;
  72     int sample_rate = 0;
  73     int channels = 1;
  74
  75     while (!voc->remaining_size) {
  76         type = get_byte(pb);
  77         if (type == VOC_TYPE_EOF)
  78             return AVERROR_IO;
  79         voc->remaining_size = get_le24(pb);
  80         max_size -= 4;
  81
  82         switch (type) {
  83         case VOC_TYPE_VOICE_DATA:
  84             dec->sample_rate = 1000000 / (256 - get_byte(pb));
  85             if (sample_rate)
  86                 dec->sample_rate = sample_rate;
  87             dec->channels = channels;
  88             dec->codec_id = codec_get_id(voc_codec_tags, get_byte(pb));
  89             dec->bits_per_sample = av_get_bits_per_sample(dec->codec_id);
  90             voc->remaining_size -= 2;
  91             max_size -= 2;
  92             channels = 1;
  93             break;
  94
  95         case VOC_TYPE_VOICE_DATA_CONT:
  96             break;
  97
  98         case VOC_TYPE_EXTENDED:
  99             sample_rate = get_le16(pb);
 100             get_byte(pb);
 101             channels = get_byte(pb) + 1;
 102             sample_rate = 256000000 / (channels * (65536 - sample_rate));
 103             voc->remaining_size = 0;
 104             max_size -= 4;
 105             break;
 106
 107         case VOC_TYPE_NEW_VOICE_DATA:
 108             dec->sample_rate = get_le32(pb);
 109             dec->bits_per_sample = get_byte(pb);
 110             dec->channels = get_byte(pb);
 111             dec->codec_id = codec_get_id(voc_codec_tags, get_le16(pb));
 112             url_fskip(pb, 4);
 113             voc->remaining_size -= 12;
 114             max_size -= 12;
 115             break;
 116
 117         default:
 118             url_fskip(pb, voc->remaining_size);
 119             max_size -= voc->remaining_size;
 120             voc->remaining_size = 0;
 121             break;
 122         }
 123     }
 124
 125     dec->bit_rate = dec->sample_rate * dec->bits_per_sample;
 126
 127     if (max_size <= 0)
 128         max_size = 2048;
 129     size = FFMIN(voc->remaining_size, max_size);
 130     voc->remaining_size -= size;
 131     return av_get_packet(pb, pkt, size);
 132 }
 133
 134 static int voc_read_packet(AVFormatContext *s, AVPacket *pkt)
 135 {
 136     return voc_get_packet(s, pkt, s->streams[0], 0);
 137 }
 138
 139 static int voc_read_close(AVFormatContext *s)
 140 {
 141     return 0;
 142 }
 143
 144 AVInputFormat voc_demuxer = {
 145     "voc",
 146     "Creative Voice File format",
 147     sizeof(voc_dec_context_t),
 148     voc_probe,
 149     voc_read_header,
 150     voc_read_packet,
 151     voc_read_close,
 152     .codec_tag=(const AVCodecTag*[]){voc_codec_tags, 0},
 153 };