From bab3498c6ab7947e82bb206933711870a30016f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tim-Philipp=20M=C3=BCller?= <tim@centricular.com>
Date: Thu, 16 May 2024 19:20:21 +0100
Subject: [PATCH] rtp: opus: add multichannel pay/depay test

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/-/merge_requests/1571>
---
 .../tests/audiotestsrc-6ch-48kHz-000.opus     |   3 +
 .../tests/audiotestsrc-6ch-48kHz-001.opus     | Bin 0 -> 129 bytes
 .../tests/audiotestsrc-6ch-48kHz-002.opus     | Bin 0 -> 133 bytes
 net/rtp/src/opus/tests/tests.rs               | 105 +++++++++++++++++-
 4 files changed, 107 insertions(+), 1 deletion(-)
 create mode 100644 net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-000.opus
 create mode 100644 net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-001.opus
 create mode 100644 net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-002.opus

diff --git a/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-000.opus b/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-000.opus
new file mode 100644
index 00000000..df6cf4eb
--- /dev/null
+++ b/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-000.opus
@@ -0,0 +1,3 @@
+|(�
�%k�a�`�B""T��j�
+�jj9��f�k#i��G:����
*|(�
�%k�a�`�B""T��j�
+�jj9��f�k#i��G:����
*x�q�HB���gtQ��_~"ر�#��kMax�q�HB���gtQ��_~"ر�#��kMa
\ No newline at end of file
diff --git a/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-001.opus b/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-001.opus
new file mode 100644
index 0000000000000000000000000000000000000000..f058f88eb2caab8d7bbffb933b147fc921bc28df
GIT binary patch
literal 129
zcmb<X=3q_{myOin)4P3YQGo1a-fJhsPOb}#&gB26*>b4kcEr{Oh8kiNRY=(<K438t
d7UL1-wmEFqrgP%Xjm~1Bj+-yj87go~0RZO#Gxq=h

literal 0
HcmV?d00001

diff --git a/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-002.opus b/net/rtp/src/opus/tests/audiotestsrc-6ch-48kHz-002.opus
new file mode 100644
index 0000000000000000000000000000000000000000..3042b7bdaf69e0a6c396ea5b123484f24fb99132
GIT binary patch
literal 133
zcmb<X=3wT}?nv@Gt9n<X|HS!Ad|ARzG%{HBJH7Zj`PE&9(-B*H_-lw!R3T%&??L4n
gt+&iu{$!X;vwt*M$$G2I6sdU|PQFa%sK6%)0QN#Y`2YX_

literal 0
HcmV?d00001

diff --git a/net/rtp/src/opus/tests/tests.rs b/net/rtp/src/opus/tests/tests.rs
index d472c982..06ede4be 100644
--- a/net/rtp/src/opus/tests/tests.rs
+++ b/net/rtp/src/opus/tests/tests.rs
@@ -8,7 +8,9 @@
 //
 // SPDX-License-Identifier: MPL-2.0
 
-use crate::tests::{run_test_pipeline, ExpectedBuffer, ExpectedPacket, Source};
+use crate::tests::{
+    run_test_pipeline, run_test_pipeline_full, ExpectedBuffer, ExpectedPacket, Source,
+};
 use gst::prelude::*;
 use gst_check::Harness;
 
@@ -430,3 +432,104 @@ fn test_opus_payloader_get_caps() {
 
     // Not testing MULTIOPUS, OPUS preference order, doesn't seem very interesting in practice.
 }
+
+// test_opus_pay_depay_multichannel
+//
+// Check basic payloading/depayloading with multichannel audio (MULTIOPUS extension)
+//
+#[test]
+fn test_opus_pay_depay_multichannel() {
+    // gst-launch-1.0 audiotestsrc wave=ticks ! audio/x-raw,channels=6 ! opusenc ! multifilesink
+    const OPUS_BUFFERS: &[&[u8]] = &[
+        include_bytes!("audiotestsrc-6ch-48kHz-000.opus").as_slice(),
+        include_bytes!("audiotestsrc-6ch-48kHz-001.opus").as_slice(),
+        include_bytes!("audiotestsrc-6ch-48kHz-002.opus").as_slice(),
+    ];
+
+    init();
+
+    let input_caps = gst::Caps::builder("audio/x-opus")
+        .field("rate", 48000i32)
+        .field("channels", 6i32)
+        .field("channel-mapping-family", 1i32)
+        .field("stream-count", 4i32)
+        .field("coupled-count", 2i32)
+        .field("channel-mapping", gst::Array::new([0i32, 4, 1, 2, 3, 5]))
+        .build();
+
+    let input_buffers = vec![
+        make_buffer(
+            OPUS_BUFFERS[0],
+            gst::ClockTime::ZERO,
+            gst::ClockTime::from_mseconds(20), // Note: no ClippingMeta for lead-in here unlike opusenc
+            gst::BufferFlags::DISCONT,
+        ),
+        make_buffer(
+            OPUS_BUFFERS[1],
+            gst::ClockTime::from_mseconds(20),
+            gst::ClockTime::from_mseconds(20),
+            gst::BufferFlags::empty(),
+        ),
+        make_buffer(
+            OPUS_BUFFERS[2],
+            gst::ClockTime::from_mseconds(40),
+            gst::ClockTime::from_mseconds(20),
+            gst::BufferFlags::empty(),
+        ),
+    ];
+
+    // TODO: check durations?
+    let expected_pay = vec![
+        vec![ExpectedPacket::builder()
+            .pts(gst::ClockTime::ZERO)
+            .flags(gst::BufferFlags::DISCONT | gst::BufferFlags::MARKER)
+            .pt(96)
+            .rtp_time(0)
+            .marker_bit(true)
+            .build()],
+        vec![ExpectedPacket::builder()
+            .pts(gst::ClockTime::from_mseconds(20))
+            .flags(gst::BufferFlags::empty())
+            .pt(96)
+            .rtp_time(960)
+            .marker_bit(false)
+            .build()],
+        vec![ExpectedPacket::builder()
+            .pts(gst::ClockTime::from_mseconds(40))
+            .flags(gst::BufferFlags::empty())
+            .pt(96)
+            .rtp_time(960 + 960)
+            .marker_bit(false)
+            .build()],
+    ];
+
+    // TODO: check durations?
+    let expected_depay = vec![
+        vec![ExpectedBuffer::builder()
+            .pts(gst::ClockTime::ZERO)
+            .size(143)
+            .flags(gst::BufferFlags::DISCONT | gst::BufferFlags::RESYNC)
+            .build()],
+        vec![ExpectedBuffer::builder()
+            .pts(gst::ClockTime::from_mseconds(20))
+            .size(129)
+            .flags(gst::BufferFlags::empty())
+            .build()],
+        vec![ExpectedBuffer::builder()
+            .pts(gst::ClockTime::from_mseconds(40))
+            .size(133)
+            .flags(gst::BufferFlags::empty())
+            .build()],
+    ];
+
+    let expected_depay_caps = input_caps.clone();
+
+    run_test_pipeline_full(
+        Source::Buffers(input_caps, input_buffers),
+        "rtpopuspay2",
+        "rtpopusdepay2",
+        expected_pay,
+        expected_depay,
+        Some(expected_depay_caps),
+    );
+}