From 44e9cccffa6eef7fee8662f9e9bd88e3b04b0724 Mon Sep 17 00:00:00 2001 From: Paul B Mahol Date: Sat, 23 Sep 2023 16:38:35 +0200 Subject: avformat: add QOA demuxer --- libavformat/Makefile | 1 + libavformat/allformats.c | 1 + libavformat/qoadec.c | 110 +++++++++++++++++++++++++++++++++++++++++++++++ libavformat/version.h | 2 +- 4 files changed, 113 insertions(+), 1 deletion(-) create mode 100644 libavformat/qoadec.c (limited to 'libavformat') diff --git a/libavformat/Makefile b/libavformat/Makefile index 329055ccfd..2db83aff81 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -484,6 +484,7 @@ OBJS-$(CONFIG_PP_BNK_DEMUXER) += pp_bnk.o OBJS-$(CONFIG_PVA_DEMUXER) += pva.o OBJS-$(CONFIG_PVF_DEMUXER) += pvfdec.o pcm.o OBJS-$(CONFIG_QCP_DEMUXER) += qcp.o +OBJS-$(CONFIG_QOA_DEMUXER) += qoadec.o OBJS-$(CONFIG_R3D_DEMUXER) += r3d.o OBJS-$(CONFIG_RAWVIDEO_DEMUXER) += rawvideodec.o OBJS-$(CONFIG_RAWVIDEO_MUXER) += rawenc.o diff --git a/libavformat/allformats.c b/libavformat/allformats.c index d4b505a5a3..c8bb4e3866 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -383,6 +383,7 @@ extern const FFOutputFormat ff_psp_muxer; extern const AVInputFormat ff_pva_demuxer; extern const AVInputFormat ff_pvf_demuxer; extern const AVInputFormat ff_qcp_demuxer; +extern const AVInputFormat ff_qoa_demuxer; extern const AVInputFormat ff_r3d_demuxer; extern const AVInputFormat ff_rawvideo_demuxer; extern const FFOutputFormat ff_rawvideo_muxer; diff --git a/libavformat/qoadec.c b/libavformat/qoadec.c new file mode 100644 index 0000000000..acf7c02741 --- /dev/null +++ b/libavformat/qoadec.c @@ -0,0 +1,110 @@ +/* + * QOA demuxer + * + * 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 "avformat.h" +#include "avio_internal.h" +#include "internal.h" +#include "libavutil/intreadwrite.h" + +static int qoa_probe(const AVProbeData *p) +{ + if ((p->buf_size < 16) || + (AV_RB32(p->buf) != MKBETAG('q','o','a','f')) || + (AV_RB32(p->buf + 4) == 0) || + (p->buf[8] == 0) || + (AV_RB24(p->buf + 9) == 0) || + (AV_RB16(p->buf + 12) == 0) || + (AV_RB16(p->buf + 14) == 0)) + return 0; + return AVPROBE_SCORE_MAX; +} + +static int qoa_read_header(AVFormatContext *s) +{ + AVIOContext *pb = s->pb; + AVStream *st; + + st = avformat_new_stream(s, NULL); + if (!st) + return AVERROR(ENOMEM); + + avio_skip(pb, 4); + st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; + st->codecpar->codec_id = AV_CODEC_ID_QOA; + st->duration = avio_rb32(pb); + st->start_time = 0; + + ffio_ensure_seekback(pb, 4); + st->codecpar->ch_layout.nb_channels = avio_r8(pb); + if (st->codecpar->ch_layout.nb_channels == 0) + return AVERROR_INVALIDDATA; + + st->codecpar->sample_rate = avio_rb24(pb); + if (st->codecpar->sample_rate == 0) + return AVERROR_INVALIDDATA; + + avio_seek(pb, -4, SEEK_CUR); + + avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); + + return 0; +} + +static int qoa_read_packet(AVFormatContext *s, AVPacket *pkt) +{ + AVIOContext *pb = s->pb; + uint16_t size, duration; + uint8_t hdr[8]; + int64_t pos; + int ret; + + if (avio_feof(pb)) + return AVERROR_EOF; + + pos = avio_tell(pb); + ret = avio_read(pb, hdr, sizeof(hdr)); + if (ret != sizeof(hdr)) + return AVERROR_EOF; + + duration = AV_RB16(hdr + 4); + size = AV_RB16(hdr + 6); + if ((ret = av_new_packet(pkt, size)) < 0) + return ret; + + memcpy(pkt->data, hdr, sizeof(hdr)); + ret = avio_read(pb, pkt->data + sizeof(hdr), size - sizeof(hdr)); + if (ret != size - sizeof(hdr)) + return AVERROR(EIO); + pkt->stream_index = 0; + pkt->pos = pos; + pkt->duration = duration; + + return 0; +} + +const AVInputFormat ff_qoa_demuxer = { + .name = "qoa", + .long_name = NULL_IF_CONFIG_SMALL("QOA"), + .read_probe = qoa_probe, + .read_header = qoa_read_header, + .read_packet = qoa_read_packet, + .flags = AVFMT_GENERIC_INDEX, + .extensions = "qoa", +}; diff --git a/libavformat/version.h b/libavformat/version.h index 2a28a3bf40..6a80f3ac4e 100644 --- a/libavformat/version.h +++ b/libavformat/version.h @@ -31,7 +31,7 @@ #include "version_major.h" -#define LIBAVFORMAT_VERSION_MINOR 17 +#define LIBAVFORMAT_VERSION_MINOR 18 #define LIBAVFORMAT_VERSION_MICRO 100 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \ -- cgit v1.2.3