From e446cf466fcc8ad4e01fac1c223ffa951852cc83 Mon Sep 17 00:00:00 2001
From: winlin <winlin@vip.126.com>
Date: Wed, 13 May 2020 15:15:57 +0800
Subject: [PATCH] RTC: Extract video to bridger

---
 trunk/src/app/srs_app_rtc_source.cpp | 38 +++++++++++++++++++++++++++-
 trunk/src/app/srs_app_rtc_source.hpp |  3 +++
 2 files changed, 40 insertions(+), 1 deletion(-)

diff --git a/trunk/src/app/srs_app_rtc_source.cpp b/trunk/src/app/srs_app_rtc_source.cpp
index f2a96e94a..a6614e319 100644
--- a/trunk/src/app/srs_app_rtc_source.cpp
+++ b/trunk/src/app/srs_app_rtc_source.cpp
@@ -476,6 +476,7 @@ SrsRtcFromRtmpBridger::SrsRtcFromRtmpBridger(SrsRtcSource* source)
     rtc = new SrsRtc();
     codec = new SrsAudioRecode(kChannel, kSamplerate);
     discard_aac = false;
+    discard_bframe = false;
 }
 
 SrsRtcFromRtmpBridger::~SrsRtcFromRtmpBridger()
@@ -506,7 +507,8 @@ srs_error_t SrsRtcFromRtmpBridger::initialize(SrsRequest* r)
 
     // TODO: FIXME: Support reload and log it.
     discard_aac = _srs_config->get_rtc_aac_discard(req->vhost);
-    srs_trace("RTC bridge from RTMP, discard_aac=%d", discard_aac);
+    discard_bframe = _srs_config->get_rtc_bframe_discard(req->vhost);
+    srs_trace("RTC bridge from RTMP, discard_aac=%d, discard_bframe=%d", discard_aac, discard_bframe);
 
     return err;
 }
@@ -676,3 +678,37 @@ srs_error_t SrsRtcFromRtmpBridger::on_video(SrsSharedPtrMessage* msg)
     return source_->on_video_imp(msg);
 }
 
+srs_error_t SrsRtcFromRtmpBridger::filter(SrsSharedPtrMessage* shared_frame, SrsFormat* format)
+{
+    srs_error_t err = srs_success;
+
+    // If IDR, we will insert SPS/PPS before IDR frame.
+    if (format->video && format->video->has_idr) {
+        shared_frame->set_has_idr(true);
+    }
+
+    // Update samples to shared frame.
+    for (int i = 0; i < format->video->nb_samples; ++i) {
+        SrsSample* sample = &format->video->samples[i];
+
+        // Because RTC does not support B-frame, so we will drop them.
+        // TODO: Drop B-frame in better way, which not cause picture corruption.
+        if (discard_bframe) {
+            if ((err = sample->parse_bframe()) != srs_success) {
+                return srs_error_wrap(err, "parse bframe");
+            }
+            if (sample->bframe) {
+                continue;
+            }
+        }
+    }
+
+    if (format->video->nb_samples <= 0) {
+        return err;
+    }
+
+    shared_frame->set_samples(format->video->samples, format->video->nb_samples);
+
+    return err;
+}
+
diff --git a/trunk/src/app/srs_app_rtc_source.hpp b/trunk/src/app/srs_app_rtc_source.hpp
index 1015c8a15..7a61ae469 100644
--- a/trunk/src/app/srs_app_rtc_source.hpp
+++ b/trunk/src/app/srs_app_rtc_source.hpp
@@ -179,6 +179,7 @@ private:
 private:
     bool discard_aac;
     SrsAudioRecode* codec;
+    bool discard_bframe;
     // rtc handler
     SrsRtc* rtc;
 public:
@@ -195,6 +196,8 @@ private:
     srs_error_t transcode(SrsSharedPtrMessage* shared_audio, char* adts_audio, int nn_adts_audio);
 public:
     virtual srs_error_t on_video(SrsSharedPtrMessage* video);
+private:
+    srs_error_t filter(SrsSharedPtrMessage* shared_video, SrsFormat* format);
 };
 
 #endif