fix(ActiveDecoder): drop logic effective hevc only

Signed-off-by: pingkai <pingkai010@gmail.com>
alibaba · Feb 14, 2020 · 5372c05 · 5372c05
1 parent 756b399
commit 5372c05
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/framework/codec/ActiveDecoder.cpp b/framework/codec/ActiveDecoder.cpp
@@ -137,8 +137,15 @@ bool ActiveDecoder::needDrop(IAFPacket *packet)
     if (packet == nullptr) {
         return false;
     }
-    if (packet->getInfo().flags & AF_PKT_FLAG_CORRUPT)
+
+    if (packet->getInfo().flags & AF_PKT_FLAG_CORRUPT) {
         return true;
+    }
+
+    // TODO: make sure HEVC not need drop also
+    if (packet->getInfo().codec_id != AF_CODEC_ID_HEVC) {
+        return false;
+    }
 
     if (bNeedKeyFrame) { // need a key frame, when first start or after seek
         if ((packet->getInfo().flags & AF_PKT_FLAG_KEY) == 0) { // drop the frame that not a key frame
@@ -346,7 +353,6 @@ int ActiveDecoder::extract_decoder()
 {
     int count = 0;
     int size;
-
     {
         std::unique_lock <std::mutex> uMutex(mMutex);
         size = mOutputQueue.size();