From f5f658e4746db338f6e10cdd24476b5ab5368c98 Mon Sep 17 00:00:00 2001
From: xia-chu <771730766@qq.com>
Date: Sat, 9 Dec 2023 21:24:34 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E8=AE=A2=E5=A4=9Atrack=E6=97=B6rtmp?=
 =?UTF-8?q?=E5=8D=8F=E8=AE=AE=E5=BC=82=E5=B8=B8=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Rtmp/RtmpMuxer.cpp | 32 ++++++++++++++++++--------------
 src/Rtmp/RtmpMuxer.h   |  7 +++++--
 2 files changed, 23 insertions(+), 16 deletions(-)

diff --git a/src/Rtmp/RtmpMuxer.cpp b/src/Rtmp/RtmpMuxer.cpp
index 52f83811..5a5e8b2d 100644
--- a/src/Rtmp/RtmpMuxer.cpp
+++ b/src/Rtmp/RtmpMuxer.cpp
@@ -23,17 +23,22 @@ RtmpMuxer::RtmpMuxer(const TitleMeta::Ptr &title) {
 }
 
 bool RtmpMuxer::addTrack(const Track::Ptr &track) {
-    auto &encoder = _encoder[track->getTrackType()];
-    if (encoder) {
-        WarnL << "Already add a track kind of: " << track->getTrackTypeStr()
-              << ", ignore track: " << track->getCodecName();
+    if (_track_existed[track->getTrackType()]) {
+        // rtmp不支持多个同类型track
+        WarnL << "Already add a track kind of: " << track->getTrackTypeStr() << ", ignore track: " << track->getCodecName();
         return false;
     }
+
+    auto &encoder = _encoders[track->getIndex()];
+    CHECK(!encoder);
     encoder = Factory::getRtmpEncoderByTrack(track);
     if (!encoder) {
         return false;
     }
 
+    // 标记已经存在该类型track
+    _track_existed[track->getTrackType()] = true;
+
     // 设置rtmp输出环形缓存
     encoder->setRtmpRing(_rtmp_ring);
 
@@ -43,22 +48,22 @@ bool RtmpMuxer::addTrack(const Track::Ptr &track) {
 }
 
 bool RtmpMuxer::inputFrame(const Frame::Ptr &frame) {
-    auto &encoder = _encoder[frame->getTrackType()];
+    auto &encoder = _encoders[frame->getIndex()];
     return encoder ? encoder->inputFrame(frame) : false;
 }
 
 void RtmpMuxer::flush() {
-    for (auto &encoder : _encoder) {
-        if (encoder) {
-            encoder->flush();
+    for (auto &pr : _encoders) {
+        if (pr.second) {
+            pr.second->flush();
         }
     }
 }
 
 void RtmpMuxer::makeConfigPacket() {
-    for (auto &encoder : _encoder) {
-        if (encoder) {
-            encoder->makeConfigPacket();
+    for (auto &pr : _encoders) {
+        if (pr.second) {
+            pr.second->makeConfigPacket();
         }
     }
 }
@@ -73,9 +78,8 @@ RtmpRing::RingType::Ptr RtmpMuxer::getRtmpRing() const {
 
 void RtmpMuxer::resetTracks() {
     _metadata.clear();
-    for (auto &encoder : _encoder) {
-        encoder = nullptr;
-    }
+    _encoders.clear();
+    CLEAR_ARR(_track_existed);
 }
 
 } /* namespace mediakit */
\ No newline at end of file
diff --git a/src/Rtmp/RtmpMuxer.h b/src/Rtmp/RtmpMuxer.h
index f64b875a..d029a632 100644
--- a/src/Rtmp/RtmpMuxer.h
+++ b/src/Rtmp/RtmpMuxer.h
@@ -64,10 +64,13 @@ public:
      * 生成config包
      */
      void makeConfigPacket();
+
 private:
-    RtmpRing::RingType::Ptr _rtmp_ring;
+    bool _track_existed[2] = { false, false };
+
     AMFValue _metadata;
-    RtmpCodec::Ptr _encoder[TrackMax];
+    RtmpRing::RingType::Ptr _rtmp_ring;
+    std::unordered_map<int, RtmpCodec::Ptr> _encoders;
 };