yading@11: /*
yading@11:  * Creative Voice File muxer.
yading@11:  * Copyright (c) 2006  Aurelien Jacobs <aurel@gnuage.org>
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 "voc.h"
yading@11: #include "internal.h"
yading@11: 
yading@11: 
yading@11: typedef struct voc_enc_context {
yading@11:     int param_written;
yading@11: } VocEncContext;
yading@11: 
yading@11: static int voc_write_header(AVFormatContext *s)
yading@11: {
yading@11:     AVIOContext *pb = s->pb;
yading@11:     AVCodecContext *enc = s->streams[0]->codec;
yading@11:     const int header_size = 26;
yading@11:     const int version = 0x0114;
yading@11: 
yading@11:     if (s->nb_streams != 1
yading@11:         || s->streams[0]->codec->codec_type != AVMEDIA_TYPE_AUDIO)
yading@11:         return AVERROR_PATCHWELCOME;
yading@11: 
yading@11:     if (!enc->codec_tag && enc->codec_id != AV_CODEC_ID_PCM_U8) {
yading@11:         av_log(s, AV_LOG_ERROR, "unsupported codec\n");
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11: 
yading@11:     avio_write(pb, ff_voc_magic, sizeof(ff_voc_magic) - 1);
yading@11:     avio_wl16(pb, header_size);
yading@11:     avio_wl16(pb, version);
yading@11:     avio_wl16(pb, ~version + 0x1234);
yading@11: 
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int voc_write_packet(AVFormatContext *s, AVPacket *pkt)
yading@11: {
yading@11:     VocEncContext *voc = s->priv_data;
yading@11:     AVCodecContext *enc = s->streams[0]->codec;
yading@11:     AVIOContext *pb = s->pb;
yading@11: 
yading@11:     if (!voc->param_written) {
yading@11:         if (enc->codec_tag > 3) {
yading@11:             avio_w8(pb, VOC_TYPE_NEW_VOICE_DATA);
yading@11:             avio_wl24(pb, pkt->size + 12);
yading@11:             avio_wl32(pb, enc->sample_rate);
yading@11:             avio_w8(pb, enc->bits_per_coded_sample);
yading@11:             avio_w8(pb, enc->channels);
yading@11:             avio_wl16(pb, enc->codec_tag);
yading@11:             avio_wl32(pb, 0);
yading@11:         } else {
yading@11:             if (s->streams[0]->codec->channels > 1) {
yading@11:                 avio_w8(pb, VOC_TYPE_EXTENDED);
yading@11:                 avio_wl24(pb, 4);
yading@11:                 avio_wl16(pb, 65536-(256000000 + enc->sample_rate*enc->channels/2)/(enc->sample_rate*enc->channels));
yading@11:                 avio_w8(pb, enc->codec_tag);
yading@11:                 avio_w8(pb, enc->channels - 1);
yading@11:             }
yading@11:             avio_w8(pb, VOC_TYPE_VOICE_DATA);
yading@11:             avio_wl24(pb, pkt->size + 2);
yading@11:             avio_w8(pb, 256 - (1000000 + enc->sample_rate/2) / enc->sample_rate);
yading@11:             avio_w8(pb, enc->codec_tag);
yading@11:         }
yading@11:         voc->param_written = 1;
yading@11:     } else {
yading@11:         avio_w8(pb, VOC_TYPE_VOICE_DATA_CONT);
yading@11:         avio_wl24(pb, pkt->size);
yading@11:     }
yading@11: 
yading@11:     avio_write(pb, pkt->data, pkt->size);
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int voc_write_trailer(AVFormatContext *s)
yading@11: {
yading@11:     avio_w8(s->pb, 0);
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: AVOutputFormat ff_voc_muxer = {
yading@11:     .name              = "voc",
yading@11:     .long_name         = NULL_IF_CONFIG_SMALL("Creative Voice"),
yading@11:     .mime_type         = "audio/x-voc",
yading@11:     .extensions        = "voc",
yading@11:     .priv_data_size    = sizeof(VocEncContext),
yading@11:     .audio_codec       = AV_CODEC_ID_PCM_S16LE,
yading@11:     .video_codec       = AV_CODEC_ID_NONE,
yading@11:     .write_header      = voc_write_header,
yading@11:     .write_packet      = voc_write_packet,
yading@11:     .write_trailer     = voc_write_trailer,
yading@11:     .codec_tag         = (const AVCodecTag* const []){ ff_voc_codec_tags, 0 },
yading@11: };