0,0 → 1,91 |
/* |
* WavPack muxer |
* Copyright (c) 2013 Konstantin Shishkov <kostya.shishkov@gmail.com> |
* Copyright (c) 2012 Paul B Mahol |
* |
* This file is part of FFmpeg. |
* |
* FFmpeg is free software; you can redistribute it and/or |
* modify it under the terms of the GNU Lesser General Public |
* License as published by the Free Software Foundation; either |
* version 2.1 of the License, or (at your option) any later version. |
* |
* FFmpeg is distributed in the hope that it will be useful, |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
* Lesser General Public License for more details. |
* |
* You should have received a copy of the GNU Lesser General Public |
* License along with FFmpeg; if not, write to the Free Software |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
*/ |
|
#include "libavutil/attributes.h" |
|
#include "apetag.h" |
#include "avformat.h" |
#include "wv.h" |
|
typedef struct WvMuxContext { |
int64_t samples; |
} WvMuxContext; |
|
static av_cold int wv_write_header(AVFormatContext *ctx) |
{ |
if (ctx->nb_streams > 1 || |
ctx->streams[0]->codec->codec_id != AV_CODEC_ID_WAVPACK) { |
av_log(ctx, AV_LOG_ERROR, "This muxer only supports a single WavPack stream.\n"); |
return AVERROR(EINVAL); |
} |
|
return 0; |
} |
|
static int wv_write_packet(AVFormatContext *ctx, AVPacket *pkt) |
{ |
WvMuxContext *s = ctx->priv_data; |
WvHeader header; |
int ret; |
|
if (pkt->size < WV_HEADER_SIZE || |
(ret = ff_wv_parse_header(&header, pkt->data)) < 0) { |
av_log(ctx, AV_LOG_ERROR, "Invalid WavPack packet.\n"); |
return AVERROR(EINVAL); |
} |
s->samples += header.samples; |
|
avio_write(ctx->pb, pkt->data, pkt->size); |
|
return 0; |
} |
|
static av_cold int wv_write_trailer(AVFormatContext *ctx) |
{ |
WvMuxContext *s = ctx->priv_data; |
|
/* update total number of samples in the first block */ |
if (ctx->pb->seekable && s->samples && |
s->samples < UINT32_MAX) { |
int64_t pos = avio_tell(ctx->pb); |
avio_seek(ctx->pb, 12, SEEK_SET); |
avio_wl32(ctx->pb, s->samples); |
avio_seek(ctx->pb, pos, SEEK_SET); |
} |
|
ff_ape_write_tag(ctx); |
return 0; |
} |
|
AVOutputFormat ff_wv_muxer = { |
.name = "wv", |
.long_name = NULL_IF_CONFIG_SMALL("raw WavPack"), |
.mime_type = "audio/x-wavpack", |
.extensions = "wv", |
.priv_data_size = sizeof(WvMuxContext), |
.audio_codec = AV_CODEC_ID_WAVPACK, |
.video_codec = AV_CODEC_ID_NONE, |
.write_header = wv_write_header, |
.write_packet = wv_write_packet, |
.write_trailer = wv_write_trailer, |
.flags = AVFMT_NOTIMESTAMPS, |
}; |