Add bitstream dump option

Signed-off-by: Yu, Dingfeng <[email protected]>

Author: Yu, Dingfeng

src/player/app/linux/config.xml

@@ -28,6 +28,7 @@
     <ProjectionType>ERP</ProjectionType>
     <frameRate>30</frameRate>
     <frameNum>100</frameNum>
+    <enableDump>false</enableDump>
     <predict enable="0">
      <!-- <plugin>libViewportPredict_LR.so</plugin>
      <path>../plugins/ViewportPredict_Plugin/predict_LR/</path> -->

src/player/player_lib/MediaSource/WebRTCMediaSource.cpp

@@ -243,7 +243,10 @@ WebRTCMediaSource::WebRTCMediaSource()
       m_pitch(0),
       m_rtcp_feedback(nullptr),
       m_parserRWPKHandle(nullptr),
-      m_ready(false) {
+      m_ready(false),
+      m_enableBsDump(false),
+      m_bsDumpfp(nullptr) {
+
   LOG(INFO) << __FUNCTION__ << std::endl;
 
   s_CurObj = this;
@@ -260,6 +263,11 @@ WebRTCMediaSource::~WebRTCMediaSource() {
     I360SCVP_unInit(m_parserRWPKHandle);
     m_parserRWPKHandle = nullptr;
   }
+
+  if (m_bsDumpfp) {
+    fclose(m_bsDumpfp);
+    m_bsDumpfp = nullptr;
+  }
 }
 
 void WebRTCMediaSource::parseOptions() {
@@ -284,6 +292,11 @@ void WebRTCMediaSource::parseOptions() {
     exit(-1);
   }
   m_serverAddress = std::string(server_url);
+
+  if (!strcmp(info->FirstChildElement("enableDump")->GetText(), "true")) {
+    m_enableBsDump = true;
+  }
+
 }
 
 void WebRTCMediaSource::setMediaInfo() {
@@ -336,10 +349,31 @@ void WebRTCMediaSource::setMediaInfo() {
   }
 }
 
+void WebRTCMediaSource::initDump() {
+  if (m_enableBsDump) {
+    char dumpFileName[128];
+    snprintf(dumpFileName, 128, "/tmp/webrtcsource-%p.%s", this, "hevc");
+    m_bsDumpfp = fopen(dumpFileName, "wb");
+    if (m_bsDumpfp) {
+      LOG(INFO) << "Enable bitstream dump " << dumpFileName << std::endl;
+    } else {
+      LOG(ERROR) << "Can not open dump file " << dumpFileName << std::endl;
+    }
+  }
+}
+
+void WebRTCMediaSource::dump(const uint8_t* buf, int len, FILE* fp) {
+  if (fp) {
+    fwrite(buf, 1, len, fp);
+    fflush(fp);
+  }
+}
+
 RenderStatus WebRTCMediaSource::Initialize(struct RenderConfig renderConfig,
                                            RenderSourceFactory* rsFactory) {
   m_rsFactory = rsFactory;
   setMediaInfo();
+  initDump();
 
   m_DecoderManager = std::make_shared<DecoderManager>();
   RenderStatus ret = m_DecoderManager->Initialize(rsFactory);
@@ -466,6 +500,9 @@ bool WebRTCMediaSource::OnVideoPacket(
   std::shared_ptr<SimpleBuffer> bitstream_buf =
       std::make_shared<SimpleBuffer>();
   bitstream_buf->insert(frame->buffer, frame->length);
+  if (m_enableBsDump) {
+    dump(frame->buffer, frame->length, m_bsDumpfp);
+  }
 
   std::shared_ptr<SimpleBuffer> sei_buf = std::make_shared<SimpleBuffer>();
   filter_RWPK_SEI(bitstream_buf, sei_buf);

src/player/player_lib/MediaSource/WebRTCMediaSource.h

@@ -161,6 +161,8 @@ class WebRTCMediaSource : public MediaSource, public WebRTCVideoPacketListener {
  private:
   void parseOptions();
   void setMediaInfo();
+  void initDump();
+  void dump(const uint8_t* buf, int len, FILE* fp);
 
   int32_t m_source_width;
   int32_t m_source_height;
@@ -187,6 +189,8 @@ class WebRTCMediaSource : public MediaSource, public WebRTCVideoPacketListener {
   void* m_parserRWPKHandle;
 
   bool m_ready;
+  bool m_enableBsDump;
+  FILE* m_bsDumpfp;
 };
 
 class SimpleBuffer {