yading@11: /*
yading@11:  * WavPack muxer
yading@11:  * Copyright (c) 2012 Paul B Mahol
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 "avformat.h"
yading@11: #include "internal.h"
yading@11: #include "avio_internal.h"
yading@11: #include "apetag.h"
yading@11: 
yading@11: #define WV_EXTRA_SIZE 12
yading@11: #define WV_END_BLOCK  0x1000
yading@11: 
yading@11: typedef struct{
yading@11:     uint32_t duration;
yading@11: } WVMuxContext;
yading@11: 
yading@11: static int write_header(AVFormatContext *s)
yading@11: {
yading@11:     AVCodecContext *codec = s->streams[0]->codec;
yading@11: 
yading@11:     if (s->nb_streams > 1) {
yading@11:         av_log(s, AV_LOG_ERROR, "only one stream is supported\n");
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11:     if (codec->codec_id != AV_CODEC_ID_WAVPACK) {
yading@11:         av_log(s, AV_LOG_ERROR, "unsupported codec\n");
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11:     if (codec->extradata_size > 0) {
yading@11:         avpriv_report_missing_feature(s, "remuxing from matroska container");
yading@11:         return AVERROR_PATCHWELCOME;
yading@11:     }
yading@11:     avpriv_set_pts_info(s->streams[0], 64, 1, codec->sample_rate);
yading@11: 
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int write_packet(AVFormatContext *s, AVPacket *pkt)
yading@11: {
yading@11:     WVMuxContext *wc = s->priv_data;
yading@11:     AVCodecContext *codec = s->streams[0]->codec;
yading@11:     AVIOContext *pb = s->pb;
yading@11:     uint64_t size;
yading@11:     uint32_t flags;
yading@11:     uint32_t left = pkt->size;
yading@11:     uint8_t *ptr = pkt->data;
yading@11:     int off = codec->channels > 2 ? 4 : 0;
yading@11: 
yading@11:     /* FIXME: Simplify decoder/demuxer so bellow code can support midstream
yading@11:      *        change of stream parameters */
yading@11:     wc->duration += pkt->duration;
yading@11:     ffio_wfourcc(pb, "wvpk");
yading@11:     if (off) {
yading@11:         size = AV_RL32(pkt->data);
yading@11:         if (size <= 12)
yading@11:             return AVERROR_INVALIDDATA;
yading@11:         size -= 12;
yading@11:     } else {
yading@11:         size = pkt->size;
yading@11:     }
yading@11: 
yading@11:     if (size + off > left)
yading@11:         return AVERROR_INVALIDDATA;
yading@11: 
yading@11:     avio_wl32(pb, size + 12);
yading@11:     avio_wl16(pb, 0x410);
yading@11:     avio_w8(pb, 0);
yading@11:     avio_w8(pb, 0);
yading@11:     avio_wl32(pb, -1);
yading@11:     avio_wl32(pb, pkt->pts);
yading@11:     ptr += off; left -= off;
yading@11:     flags = AV_RL32(ptr + 4);
yading@11:     avio_write(pb, ptr, size);
yading@11:     ptr += size; left -= size;
yading@11: 
yading@11:     while (!(flags & WV_END_BLOCK) &&
yading@11:             (left >= 4 + WV_EXTRA_SIZE)) {
yading@11:         ffio_wfourcc(pb, "wvpk");
yading@11:         size = AV_RL32(ptr);
yading@11:         ptr += 4; left -= 4;
yading@11:         if (size < 24 || size - 24 > left)
yading@11:             return AVERROR_INVALIDDATA;
yading@11:         avio_wl32(pb, size);
yading@11:         avio_wl16(pb, 0x410);
yading@11:         avio_w8(pb, 0);
yading@11:         avio_w8(pb, 0);
yading@11:         avio_wl32(pb, -1);
yading@11:         avio_wl32(pb, pkt->pts);
yading@11:         flags = AV_RL32(ptr + 4);
yading@11:         avio_write(pb, ptr, WV_EXTRA_SIZE);
yading@11:         ptr += WV_EXTRA_SIZE; left -= WV_EXTRA_SIZE;
yading@11:         avio_write(pb, ptr, size - 24);
yading@11:         ptr += size - 24; left -= size - 24;
yading@11:     }
yading@11: 
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int write_trailer(AVFormatContext *s)
yading@11: {
yading@11:     WVMuxContext *wc = s->priv_data;
yading@11:     AVIOContext *pb = s->pb;
yading@11: 
yading@11:     ff_ape_write(s);
yading@11: 
yading@11:     if (pb->seekable) {
yading@11:         avio_seek(pb, 12, SEEK_SET);
yading@11:         avio_wl32(pb, wc->duration);
yading@11:         avio_flush(pb);
yading@11:     }
yading@11: 
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: AVOutputFormat ff_wv_muxer = {
yading@11:     .name              = "wv",
yading@11:     .long_name         = NULL_IF_CONFIG_SMALL("WavPack"),
yading@11:     .priv_data_size    = sizeof(WVMuxContext),
yading@11:     .extensions        = "wv",
yading@11:     .audio_codec       = AV_CODEC_ID_WAVPACK,
yading@11:     .video_codec       = AV_CODEC_ID_NONE,
yading@11:     .write_header      = write_header,
yading@11:     .write_packet      = write_packet,
yading@11:     .write_trailer     = write_trailer,
yading@11: };