wvenc.c 2.72 KB
Newer Older
Paul B Mahol's avatar
Paul B Mahol committed
1 2
/*
 * WavPack muxer
3
 * Copyright (c) 2013 Konstantin Shishkov <kostya.shishkov@gmail.com>
Paul B Mahol's avatar
Paul B Mahol committed
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
 * 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
 */

23 24
#include "libavutil/attributes.h"

25
#include "apetag.h"
26 27
#include "avformat.h"
#include "wv.h"
Paul B Mahol's avatar
Paul B Mahol committed
28

29
typedef struct WvMuxContext {
30
    int64_t samples;
31
} WvMuxContext;
Paul B Mahol's avatar
Paul B Mahol committed
32

33
static av_cold int wv_write_header(AVFormatContext *ctx)
Paul B Mahol's avatar
Paul B Mahol committed
34
{
35
    if (ctx->nb_streams > 1 ||
36
        ctx->streams[0]->codecpar->codec_id != AV_CODEC_ID_WAVPACK) {
37
        av_log(ctx, AV_LOG_ERROR, "This muxer only supports a single WavPack stream.\n");
Paul B Mahol's avatar
Paul B Mahol committed
38 39
        return AVERROR(EINVAL);
    }
40

Paul B Mahol's avatar
Paul B Mahol committed
41 42 43
    return 0;
}

44
static int wv_write_packet(AVFormatContext *ctx, AVPacket *pkt)
Paul B Mahol's avatar
Paul B Mahol committed
45
{
46 47 48 49 50 51 52 53 54 55
    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;
Paul B Mahol's avatar
Paul B Mahol committed
56

57
    avio_write(ctx->pb, pkt->data, pkt->size);
Paul B Mahol's avatar
Paul B Mahol committed
58 59 60 61

    return 0;
}

62
static av_cold int wv_write_trailer(AVFormatContext *ctx)
Paul B Mahol's avatar
Paul B Mahol committed
63
{
64
    WvMuxContext *s = ctx->priv_data;
65

66
    /* update total number of samples in the first block */
67
    if ((ctx->pb->seekable & AVIO_SEEKABLE_NORMAL) && s->samples &&
68 69
        s->samples < UINT32_MAX) {
        int64_t pos = avio_tell(ctx->pb);
70
        avio_seek(ctx->pb, 12, SEEK_SET);
71 72
        avio_wl32(ctx->pb, s->samples);
        avio_seek(ctx->pb, pos, SEEK_SET);
Paul B Mahol's avatar
Paul B Mahol committed
73 74
    }

75
    ff_ape_write_tag(ctx);
Paul B Mahol's avatar
Paul B Mahol committed
76 77 78 79 80
    return 0;
}

AVOutputFormat ff_wv_muxer = {
    .name              = "wv",
81 82
    .long_name         = NULL_IF_CONFIG_SMALL("raw WavPack"),
    .mime_type         = "audio/x-wavpack",
Paul B Mahol's avatar
Paul B Mahol committed
83
    .extensions        = "wv",
84
    .priv_data_size    = sizeof(WvMuxContext),
85 86
    .audio_codec       = AV_CODEC_ID_WAVPACK,
    .video_codec       = AV_CODEC_ID_NONE,
87 88 89
    .write_header      = wv_write_header,
    .write_packet      = wv_write_packet,
    .write_trailer     = wv_write_trailer,
90
    .flags             = AVFMT_NOTIMESTAMPS,
Paul B Mahol's avatar
Paul B Mahol committed
91
};