[FFmpeg-devel] [PATCH] avformat: add CRI USM demuxer

[Paul B Mahol <onemda@gmail.com>]

[-- Attachment #1: Type: text/plain, Size: 10 bytes --]

Attached.

[-- Attachment #2: 0001-avformat-add-CRI-USM-demuxer.patch --]
[-- Type: text/x-patch, Size: 11757 bytes --]

From fc0f592a04ffa99b94b1402905d0bcfb2b15270c Mon Sep 17 00:00:00 2001
From: Paul B Mahol <onemda@gmail.com>
Date: Tue, 5 Sep 2023 16:53:32 +0200
Subject: [PATCH] avformat: add CRI USM demuxer

Signed-off-by: Paul B Mahol <onemda@gmail.com>
---
 libavformat/Makefile     |   1 +
 libavformat/allformats.c |   1 +
 libavformat/usmdec.c     | 327 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 329 insertions(+)
 create mode 100644 libavformat/usmdec.c

diff --git a/libavformat/Makefile b/libavformat/Makefile
index cc1b12360a..329055ccfd 100644
--- a/libavformat/Makefile
+++ b/libavformat/Makefile
@@ -588,6 +588,7 @@ OBJS-$(CONFIG_TTY_DEMUXER)               += tty.o sauce.o
 OBJS-$(CONFIG_TY_DEMUXER)                += ty.o
 OBJS-$(CONFIG_TXD_DEMUXER)               += txd.o
 OBJS-$(CONFIG_UNCODEDFRAMECRC_MUXER)     += uncodedframecrcenc.o framehash.o
+OBJS-$(CONFIG_USM_DEMUXER)               += usmdec.o
 OBJS-$(CONFIG_V210_DEMUXER)              += rawvideodec.o
 OBJS-$(CONFIG_V210X_DEMUXER)             += rawvideodec.o
 OBJS-$(CONFIG_VAG_DEMUXER)               += vag.o
diff --git a/libavformat/allformats.c b/libavformat/allformats.c
index f4210e4932..d4b505a5a3 100644
--- a/libavformat/allformats.c
+++ b/libavformat/allformats.c
@@ -471,6 +471,7 @@ extern const AVInputFormat  ff_txd_demuxer;
 extern const AVInputFormat  ff_tty_demuxer;
 extern const AVInputFormat  ff_ty_demuxer;
 extern const FFOutputFormat ff_uncodedframecrc_muxer;
+extern const AVInputFormat  ff_usm_demuxer;
 extern const AVInputFormat  ff_v210_demuxer;
 extern const AVInputFormat  ff_v210x_demuxer;
 extern const AVInputFormat  ff_vag_demuxer;
diff --git a/libavformat/usmdec.c b/libavformat/usmdec.c
new file mode 100644
index 0000000000..63b580b9ea
--- /dev/null
+++ b/libavformat/usmdec.c
@@ -0,0 +1,327 @@
+/*
+ * USM demuxer
+ * Copyright (c) 2023 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/intreadwrite.h"
+#include "libavcodec/bytestream.h"
+
+#include "avformat.h"
+#include "internal.h"
+
+#define VIDEOI 0
+#define AUDIOI 1
+
+typedef struct USMChannel {
+    int index;
+    int used;
+} USMChannel;
+
+typedef struct USMDemuxContext {
+    USMChannel ch[2][256];
+    int nb_channels[2];
+    uint8_t *header;
+    unsigned header_size;
+} USMDemuxContext;
+
+static int usm_probe(const AVProbeData *p)
+{
+    if (AV_RL32(p->buf) != MKTAG('C','R','I','D'))
+        return 0;
+
+    if (AV_RL32(p->buf + 4) == 0)
+        return 0;
+
+    return AVPROBE_SCORE_MAX / 3;
+}
+
+static int usm_read_header(AVFormatContext *s)
+{
+    s->ctx_flags |= AVFMTCTX_NOHEADER;
+    return 0;
+}
+
+static int parse_utf(AVFormatContext *s, AVIOContext *pb,
+                     AVStream *st, AVCodecParameters *par, int is_audio)
+{
+    USMDemuxContext *usm = s->priv_data;
+    GetByteContext gb, ugb, sgb;
+    uint32_t chunk_type, chunk_size, offset;
+    uint32_t unique_offset, string_offset;
+    uint32_t byte_offset, payload_name_offset;
+    int nb_items, unique_size, nb_dictionaries;
+    AVRational fps = { 0 };
+    int type;
+
+    chunk_type = avio_rb32(pb);
+    chunk_size = avio_rb32(pb);
+
+    if (chunk_type != MKBETAG('@','U','T','F'))
+        return AVERROR_INVALIDDATA;
+
+    av_fast_malloc(&usm->header, &usm->header_size,
+                   chunk_size + AV_INPUT_BUFFER_PADDING_SIZE);
+    if (!usm->header)
+        return AVERROR(ENOMEM);
+
+    if (avio_read(pb, usm->header, chunk_size) != chunk_size)
+        return AVERROR_INVALIDDATA;
+
+    bytestream2_init(&gb, usm->header, chunk_size);
+    ugb = gb;
+    sgb = gb;
+    unique_offset = bytestream2_get_be32(&gb);
+    string_offset = bytestream2_get_be32(&gb);
+    byte_offset = bytestream2_get_be32(&gb);
+    payload_name_offset = bytestream2_get_be32(&gb);
+    nb_items = bytestream2_get_be16(&gb);
+    unique_size = bytestream2_get_be16(&gb);
+    nb_dictionaries = bytestream2_get_be32(&gb);
+    if (nb_dictionaries == 0)
+        return AVERROR_INVALIDDATA;
+
+    bytestream2_skip(&ugb, unique_offset);
+    bytestream2_skip(&sgb, string_offset);
+
+    for (int i = 0; i < nb_items; i++) {
+        GetByteContext *xgb;
+        uint8_t key[256];
+        int64_t value;
+        int n = 0;
+
+        type = bytestream2_get_byte(&gb);
+        offset = bytestream2_get_be32(&gb);
+
+        bytestream2_seek(&sgb, string_offset + offset, SEEK_SET);
+        while (bytestream2_get_bytes_left(&sgb) > 0) {
+            key[n] = bytestream2_get_byte(&sgb);
+            if (!key[n])
+                break;
+            if (n >= sizeof(key) - 1)
+                break;
+            n++;
+        }
+        key[n] = '\0';
+
+        if ((type >> 5) == 1)
+            xgb = &gb;
+        else
+            xgb = &ugb;
+
+        switch (type & 0x1F) {
+        case 0x10:
+        case 0x11:
+            value = bytestream2_get_byte(xgb);
+            break;
+        case 0x12:
+        case 0x13:
+            value = bytestream2_get_be16(xgb);
+            break;
+        case 0x14:
+        case 0x15:
+            value = bytestream2_get_be32(xgb);
+            break;
+        case 0x16:
+        case 0x17:
+            value = bytestream2_get_be64(xgb);
+            break;
+        case 0x18:
+            value = av_int2float(bytestream2_get_be32(xgb));
+            break;
+        case 0x19:
+            value = av_int2double(bytestream2_get_be64(xgb));
+            break;
+        case 0x1A:
+            break;
+        }
+
+        if (is_audio) {
+            if (!strcmp(key, "sampling_rate")) {
+                par->sample_rate = value;
+                avpriv_set_pts_info(st, 64, 1, value);
+            } else if (!strcmp(key, "num_channels")) {
+                par->ch_layout.nb_channels = value;
+            } else if (!strcmp(key, "total_samples")) {
+                st->duration = value;
+            } else if (!strcmp(key, "audio_codec")) {
+                switch (value) {
+                case 2:
+                    par->codec_id = AV_CODEC_ID_ADPCM_ADX;
+                    break;
+                case 4:
+                    par->codec_id = AV_CODEC_ID_HCA;
+                    break;
+                default:
+                    av_log(s, AV_LOG_ERROR, "unsupported audio: %d\n", (int)value);
+                    break;
+                }
+            }
+        } else {
+            if (!strcmp(key, "width")) {
+                par->width = value;
+            } else if (!strcmp(key, "height")) {
+                par->height = value;
+            } else if (!strcmp(key, "total_frames")) {
+                st->nb_frames = value;
+            } else if (!strcmp(key, "framerate_n")) {
+                fps.num = value;
+            } else if (!strcmp(key, "framerate_d")) {
+                fps.den = value;
+            } else if (!strcmp(key, "mpeg_codec")) {
+                switch (value) {
+                case 1:
+                    par->codec_id = AV_CODEC_ID_MPEG1VIDEO;
+                    break;
+                case 5:
+                    par->codec_id = AV_CODEC_ID_H264;
+                    break;
+                case 9:
+                    par->codec_id = AV_CODEC_ID_VP9;
+                    break;
+                default:
+                    av_log(s, AV_LOG_ERROR, "unsupported video: %d\n", (int)value);
+                    break;
+                }
+            }
+        }
+    }
+
+    if (!is_audio && fps.num && fps.den)
+        avpriv_set_pts_info(st, 64, fps.den, fps.num);
+
+    return 0;
+}
+
+static int parse_chunk(AVFormatContext *s, AVIOContext *pb,
+                       uint32_t chunk_type, uint32_t chunk_size,
+                       AVPacket *pkt)
+{
+    USMDemuxContext *usm = s->priv_data;
+    int64_t chunk_start;
+    int stream_index, payload_type;
+    int payload_offset, frame_time;
+    int frame_rate, padding_size, ret;
+    int is_audio = chunk_type == MKBETAG('@','S','F','A');
+
+    chunk_start = avio_tell(pb);
+    avio_skip(pb, 1);
+    payload_offset = avio_r8(pb);
+    padding_size = avio_rb16(pb);
+    stream_index = avio_r8(pb);
+    avio_skip(pb, 2);
+    payload_type = avio_r8(pb);
+    frame_time = avio_rb32(pb);
+    frame_rate = avio_rb32(pb);
+    avio_skip(pb, 8);
+
+    if (payload_type == 1) {
+        if (usm->ch[is_audio][stream_index].used == 0) {
+            AVStream *st = avformat_new_stream(s, NULL);
+            AVCodecParameters *par;
+            if (!st)
+                return AVERROR(ENOMEM);
+
+            par = st->codecpar;
+            par->codec_type = is_audio ? AVMEDIA_TYPE_AUDIO : AVMEDIA_TYPE_VIDEO;
+
+            usm->ch[is_audio][stream_index].index = st->index;
+            usm->ch[is_audio][stream_index].used = 1;
+            usm->nb_channels[is_audio]++;
+
+            ret = parse_utf(s, pb, st, par, is_audio);
+            if (ret < 0)
+                return ret;
+
+            if (!st->time_base.num || !st->time_base.den)
+                avpriv_set_pts_info(st, 64, 100, frame_rate);
+            ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL_RAW;
+        }
+    } else if (payload_type == 0) {
+        if (usm->ch[is_audio][stream_index].used == 1) {
+            uint32_t pkt_size = chunk_size - (avio_tell(pb) - chunk_start);
+
+            ret = av_get_packet(pb, pkt, pkt_size);
+            if (ret < 0)
+                return ret;
+
+            pkt->stream_index = usm->ch[is_audio][stream_index].index;
+
+            return 1;
+        }
+    }
+
+    avio_skip(pb, padding_size);
+    avio_skip(pb, chunk_size - (avio_tell(pb) - chunk_start));
+
+    return 0;
+}
+
+static int usm_read_packet(AVFormatContext *s, AVPacket *pkt)
+{
+    AVIOContext *pb = s->pb;
+    int64_t ret = AVERROR_EOF;
+
+    while (!avio_feof(pb)) {
+        uint32_t chunk_type, chunk_size;
+        int got_packet = 0;
+
+        chunk_type = avio_rb32(pb);
+        chunk_size = avio_rb32(pb);
+
+        switch (chunk_type) {
+        case MKBETAG('C','R','I','D'):
+        default:
+            ret = avio_skip(pb, chunk_size);
+            break;
+        case MKBETAG('@','S','F','A'):
+        case MKBETAG('@','S','F','V'):
+            ret = parse_chunk(s, pb, chunk_type, chunk_size, pkt);
+            got_packet = ret == 1;
+            break;
+        }
+
+        if (got_packet)
+            ret = 0;
+
+        if (got_packet || ret < 0)
+            break;
+    }
+
+    return ret;
+}
+
+static int usm_read_close(AVFormatContext *s)
+{
+    USMDemuxContext *usm = s->priv_data;
+    av_freep(&usm->header);
+    usm->header_size = 0;
+    return 0;
+}
+
+const AVInputFormat ff_usm_demuxer = {
+    .name           = "usm",
+    .long_name      = NULL_IF_CONFIG_SMALL("CRI USM"),
+    .priv_data_size = sizeof(USMDemuxContext),
+    .read_probe     = usm_probe,
+    .read_header    = usm_read_header,
+    .read_packet    = usm_read_packet,
+    .read_close     = usm_read_close,
+    .extensions     = "usm",
+    .flags          = AVFMT_GENERIC_INDEX,
+};
-- 
2.39.1


[-- Attachment #3: Type: text/plain, Size: 251 bytes --]

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".