3 * Copyright (c) 2013 Konstantin Shishkov <kostya.shishkov@gmail.com>
4 * Copyright (c) 2012 Paul B Mahol
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #include "libavutil/attributes.h"
29 typedef struct WvMuxContext
{
33 static av_cold
int wv_write_header(AVFormatContext
*ctx
)
35 if (ctx
->nb_streams
> 1 ||
36 ctx
->streams
[0]->codec
->codec_id
!= AV_CODEC_ID_WAVPACK
) {
37 av_log(ctx
, AV_LOG_ERROR
, "This muxer only supports a single WavPack stream.\n");
38 return AVERROR(EINVAL
);
44 static int wv_write_packet(AVFormatContext
*ctx
, AVPacket
*pkt
)
46 WvMuxContext
*s
= ctx
->priv_data
;
50 if (pkt
->size
< WV_HEADER_SIZE
||
51 (ret
= ff_wv_parse_header(&header
, pkt
->data
)) < 0) {
52 av_log(ctx
, AV_LOG_ERROR
, "Invalid WavPack packet.\n");
53 return AVERROR(EINVAL
);
55 s
->samples
+= header
.samples
;
57 avio_write(ctx
->pb
, pkt
->data
, pkt
->size
);
62 static av_cold
int wv_write_trailer(AVFormatContext
*ctx
)
64 WvMuxContext
*s
= ctx
->priv_data
;
66 /* update total number of samples in the first block */
67 if (ctx
->pb
->seekable
&& s
->samples
&&
68 s
->samples
< UINT32_MAX
) {
69 int64_t pos
= avio_tell(ctx
->pb
);
70 avio_seek(ctx
->pb
, 12, SEEK_SET
);
71 avio_wl32(ctx
->pb
, s
->samples
);
72 avio_seek(ctx
->pb
, pos
, SEEK_SET
);
75 ff_ape_write_tag(ctx
);
79 AVOutputFormat ff_wv_muxer
= {
81 .long_name
= NULL_IF_CONFIG_SMALL("raw WavPack"),
82 .mime_type
= "audio/x-wavpack",
84 .priv_data_size
= sizeof(WvMuxContext
),
85 .audio_codec
= AV_CODEC_ID_WAVPACK
,
86 .video_codec
= AV_CODEC_ID_NONE
,
87 .write_header
= wv_write_header
,
88 .write_packet
= wv_write_packet
,
89 .write_trailer
= wv_write_trailer
,
90 .flags
= AVFMT_NOTIMESTAMPS
,