Add debug environment for analytics (#925)

* Add debug environment variables for analytics

Author: qwu16

source/agent/analytics/videoGstPipeline/GstInternalIn.cpp

@@ -6,6 +6,17 @@
 #include <gst/gst.h>
 #include <stdio.h>
 
+static void dump(void* index, uint8_t* buf, int len)
+{
+    char dumpFileName[128];
+
+    snprintf(dumpFileName, 128, "/tmp/analyticsIn-%p", index);
+    FILE* bsDumpfp = fopen(dumpFileName, "ab");
+    if (bsDumpfp) {
+        fwrite(buf, 1, len, bsDumpfp);
+        fclose(bsDumpfp);
+    }
+}
 
 DEFINE_LOGGER(GstInternalIn, "GstInternalIn");
 GstInternalIn::GstInternalIn(GstAppSrc *data, unsigned int minPort, unsigned int maxPort)
@@ -20,7 +31,12 @@ GstInternalIn::GstInternalIn(GstAppSrc *data, unsigned int minPort, unsigned int
     appsrc = data;
     m_needKeyFrame = true;
     m_start = false;
-
+    m_dumpIn = false;
+    char* pIn = std::getenv("DUMP_ANALYTICS_IN");
+    if(pIn != NULL) {
+        ELOG_INFO("Dump analytics in stream");
+        m_dumpIn = true;
+    }
 }
 
 GstInternalIn::~GstInternalIn()
@@ -49,8 +65,6 @@ void GstInternalIn::onTransportData(char* buf, int len)
 {
     if(!m_start) {
         ELOG_INFO("Not start yet, stop pushing data to appsrc\n");
-        pthread_t tid;
-        tid = pthread_self();
         return;
     }
 
@@ -88,6 +102,10 @@ void GstInternalIn::onTransportData(char* buf, int len)
             memcpy(map.data, frame, headerLength);
             memcpy(map.data + headerLength, frame->payload, payloadLength);
 
+            if(m_dumpIn) {
+                dump(this, map.data, map.size);
+            }
+
             gst_buffer_unmap(buffer, &map);
             g_signal_emit_by_name(appsrc, "push-buffer", buffer, &ret);
 

source/agent/analytics/videoGstPipeline/GstInternalIn.h

@@ -33,6 +33,7 @@ class GstInternalIn : public owt_base::RawTransportListener{
 private:
     bool m_start;
     bool m_needKeyFrame;
+    bool m_dumpIn;
     GstAppSrc *appsrc;
     boost::shared_ptr<owt_base::RawTransportInterface> m_transport;
 };

source/agent/analytics/videoGstPipeline/VideoGstAnalyzer.cc

@@ -67,13 +67,31 @@ inline bool isVp8KeyFrame(uint8_t *data, size_t len)
     return (key_frame == 0);
 }
 
+static void dump(void* index, uint8_t* buf, int len)
+{
+    char dumpFileName[128];
+
+    snprintf(dumpFileName, 128, "/tmp/analyticsOut-%p", index);
+    FILE* bsDumpfp = fopen(dumpFileName, "ab");
+    if (bsDumpfp) {
+        fwrite(buf, 1, len, bsDumpfp);
+        fclose(bsDumpfp);
+    }
+}
+
 VideoGstAnalyzer::VideoGstAnalyzer() {
     ELOG_INFO("Init");
     sourceid = 0;
     sink = NULL;
     encoder_pad = NULL;
     addlistener = false;
     m_frameCount = 0;
+    m_dumpOut = false;
+    char* pOut = std::getenv("DUMP_ANALYTICS_OUT");
+    if(pOut != NULL) {
+        ELOG_INFO("Dump analytics Out stream");
+        m_dumpOut = true;
+    }
 }
 
 VideoGstAnalyzer::~VideoGstAnalyzer() {
@@ -100,7 +118,6 @@ gboolean VideoGstAnalyzer::StreamEventCallBack(GstBus *bus, GstMessage *message,
             g_error_free(err);
             g_free(debug);
             g_main_loop_quit(pStreamObj->loop);
-    
             break;
         }
         case GST_MESSAGE_EOS:
@@ -238,6 +255,9 @@ void VideoGstAnalyzer::new_sample_from_sink (GstElement * source, gpointer data)
     outFrame.payload = map.data;
 
     pStreamObj->m_gstinternalout->onFrame(outFrame);
+    if(pStreamObj->m_dumpOut) {
+        dump(pStreamObj, map.data, map.size);
+    }
 
     gst_buffer_unmap(buffer, &map);
     gst_sample_unref(sample);

source/agent/analytics/videoGstPipeline/VideoGstAnalyzer.h

@@ -86,6 +86,7 @@ class VideoGstAnalyzer{
     int framerate,bitrate;
     int kfi; //keyFrameInterval
     bool addlistener;
+    bool m_dumpOut;
 };
 
 }