diff options
author | Paul B Mahol <onemda@gmail.com> | 2012-10-02 13:43:19 +0000 |
---|---|---|
committer | Justin Ruggles <justin.ruggles@gmail.com> | 2012-12-07 16:15:02 -0500 |
commit | 57231e4d5b467833fb289439cd35a92513bb55c1 (patch) | |
tree | 20890a188ba6c166762ce2dfca5eb96f6deb0fe9 /libavformat | |
parent | 096abfa15052977eed93f0b5e01afd2d47c53c1f (diff) | |
download | ffmpeg-57231e4d5b467833fb289439cd35a92513bb55c1.tar.gz |
tak: demuxer, parser, and decoder
Signed-off-by: Paul B Mahol <onemda@gmail.com>
Signed-off-by: Justin Ruggles <justin.ruggles@gmail.com>
Diffstat (limited to 'libavformat')
-rw-r--r-- | libavformat/Makefile | 1 | ||||
-rw-r--r-- | libavformat/allformats.c | 1 | ||||
-rw-r--r-- | libavformat/takdec.c | 185 | ||||
-rw-r--r-- | libavformat/version.h | 2 |
4 files changed, 188 insertions, 1 deletions
diff --git a/libavformat/Makefile b/libavformat/Makefile index 346045ad50..fa9366d08d 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -304,6 +304,7 @@ OBJS-$(CONFIG_SRT_MUXER) += rawenc.o OBJS-$(CONFIG_STR_DEMUXER) += psxstr.o OBJS-$(CONFIG_SWF_DEMUXER) += swfdec.o swf.o OBJS-$(CONFIG_SWF_MUXER) += swfenc.o swf.o +OBJS-$(CONFIG_TAK_DEMUXER) += takdec.o apetag.o img2.o rawdec.o OBJS-$(CONFIG_THP_DEMUXER) += thp.o OBJS-$(CONFIG_TIERTEXSEQ_DEMUXER) += tiertexseq.o OBJS-$(CONFIG_TMV_DEMUXER) += tmv.o diff --git a/libavformat/allformats.c b/libavformat/allformats.c index c7c4be9223..042d3b9aa0 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -214,6 +214,7 @@ void av_register_all(void) REGISTER_MUXDEMUX (SRT, srt); REGISTER_DEMUXER (STR, str); REGISTER_MUXDEMUX (SWF, swf); + REGISTER_DEMUXER (TAK, tak); REGISTER_MUXER (TG2, tg2); REGISTER_MUXER (TGP, tgp); REGISTER_DEMUXER (THP, thp); diff --git a/libavformat/takdec.c b/libavformat/takdec.c new file mode 100644 index 0000000000..13bc49b6b7 --- /dev/null +++ b/libavformat/takdec.c @@ -0,0 +1,185 @@ +/* + * Raw TAK demuxer + * Copyright (c) 2012 Paul B Mahol + * + * This file is part of Libav. + * + * Libav 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. + * + * Libav 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 Libav; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "libavcodec/tak.h" +#include "avformat.h" +#include "internal.h" +#include "rawdec.h" +#include "apetag.h" + +typedef struct TAKDemuxContext { + int mlast_frame; + int64_t data_end; +} TAKDemuxContext; + +static int tak_probe(AVProbeData *p) +{ + if (!memcmp(p->buf, "tBaK", 4)) + return AVPROBE_SCORE_MAX / 2; + return 0; +} + +static int tak_read_header(AVFormatContext *s) +{ + TAKDemuxContext *tc = s->priv_data; + AVIOContext *pb = s->pb; + GetBitContext gb; + AVStream *st; + uint8_t *buffer = NULL; + int ret; + + st = avformat_new_stream(s, 0); + if (!st) + return AVERROR(ENOMEM); + + st->codec->codec_type = AVMEDIA_TYPE_AUDIO; + st->codec->codec_id = AV_CODEC_ID_TAK; + st->need_parsing = AVSTREAM_PARSE_FULL; + + tc->mlast_frame = 0; + if (avio_rl32(pb) != MKTAG('t', 'B', 'a', 'K')) { + avio_seek(pb, -4, SEEK_CUR); + return 0; + } + + while (!pb->eof_reached) { + enum TAKMetaDataType type; + int size; + + type = avio_r8(pb) & 0x7f; + size = avio_rl24(pb); + + switch (type) { + case TAK_METADATA_STREAMINFO: + case TAK_METADATA_LAST_FRAME: + case TAK_METADATA_ENCODER: + buffer = av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE); + if (!buffer) + return AVERROR(ENOMEM); + + if (avio_read(pb, buffer, size) != size) { + av_freep(&buffer); + return AVERROR(EIO); + } + + init_get_bits(&gb, buffer, size * 8); + break; + case TAK_METADATA_MD5: { + uint8_t md5[16]; + int i; + + if (size != 19) + return AVERROR_INVALIDDATA; + avio_read(pb, md5, 16); + avio_skip(pb, 3); + av_log(s, AV_LOG_VERBOSE, "MD5="); + for (i = 0; i < 16; i++) + av_log(s, AV_LOG_VERBOSE, "%02x", md5[i]); + av_log(s, AV_LOG_VERBOSE, "\n"); + break; + } + case TAK_METADATA_END: { + int64_t curpos = avio_tell(pb); + + if (pb->seekable) { + ff_ape_parse_tag(s); + avio_seek(pb, curpos, SEEK_SET); + } + + tc->data_end += curpos; + return 0; + } + default: + ret = avio_skip(pb, size); + if (ret < 0) + return ret; + } + + if (type == TAK_METADATA_STREAMINFO) { + TAKStreamInfo ti; + + avpriv_tak_parse_streaminfo(&gb, &ti); + if (ti.samples > 0) + st->duration = ti.samples; + st->codec->bits_per_coded_sample = ti.bps; + if (ti.ch_layout) + st->codec->channel_layout = ti.ch_layout; + st->codec->sample_rate = ti.sample_rate; + st->codec->channels = ti.channels; + st->start_time = 0; + avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate); + st->codec->extradata = buffer; + st->codec->extradata_size = size; + buffer = NULL; + } else if (type == TAK_METADATA_LAST_FRAME) { + if (size != 11) + return AVERROR_INVALIDDATA; + tc->mlast_frame = 1; + tc->data_end = get_bits64(&gb, TAK_LAST_FRAME_POS_BITS) + + get_bits(&gb, TAK_LAST_FRAME_SIZE_BITS); + av_freep(&buffer); + } else if (type == TAK_METADATA_ENCODER) { + av_log(s, AV_LOG_VERBOSE, "encoder version: %0X\n", + get_bits_long(&gb, TAK_ENCODER_VERSION_BITS)); + av_freep(&buffer); + } + } + + return AVERROR_EOF; +} + +static int raw_read_packet(AVFormatContext *s, AVPacket *pkt) +{ + TAKDemuxContext *tc = s->priv_data; + int ret; + + if (tc->mlast_frame) { + AVIOContext *pb = s->pb; + int64_t size, left; + + left = tc->data_end - avio_tell(s->pb); + size = FFMIN(left, 1024); + if (size <= 0) + return AVERROR_EOF; + + ret = av_get_packet(pb, pkt, size); + if (ret < 0) + return ret; + + pkt->stream_index = 0; + } else { + ret = ff_raw_read_partial_packet(s, pkt); + } + + return ret; +} + +AVInputFormat ff_tak_demuxer = { + .name = "tak", + .long_name = NULL_IF_CONFIG_SMALL("raw TAK"), + .priv_data_size = sizeof(TAKDemuxContext), + .read_probe = tak_probe, + .read_header = tak_read_header, + .read_packet = raw_read_packet, + .flags = AVFMT_GENERIC_INDEX, + .extensions = "tak", + .raw_codec_id = AV_CODEC_ID_TAK, +}; diff --git a/libavformat/version.h b/libavformat/version.h index 034fa0adce..349ba806ff 100644 --- a/libavformat/version.h +++ b/libavformat/version.h @@ -30,7 +30,7 @@ #include "libavutil/avutil.h" #define LIBAVFORMAT_VERSION_MAJOR 54 -#define LIBAVFORMAT_VERSION_MINOR 19 +#define LIBAVFORMAT_VERSION_MINOR 20 #define LIBAVFORMAT_VERSION_MICRO 0 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \ |