[Decode] Complete cur frame in child thread to unlock res (#6509)

Currently, main thread is responsiable to complete finished frames to unlock resource, but all available surfaces maybe locked by multi-frame submission that causes main thread/ application unable to find a free surface to continue even previous frames decoding are done, so application has no chance to call async decode api to complete them. Co-authored-by: Xu, Zhengguo <[email protected]>
intel · Feb 27, 2024 · c40736a · c40736a
1 parent 4b592c2
commit c40736a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/_studio/mfx_lib/decode/vp9/src/mfx_vp9_dec_decode_hw.cpp b/_studio/mfx_lib/decode/vp9/src/mfx_vp9_dec_decode_hw.cpp
@@ -276,6 +276,18 @@ class FrameStorage
         }
     }
 
+    void CompleteCurFrame(UMC::FrameMemID frameId)
+    {
+        auto find_it = std::find_if(m_submittedFrames.begin(), m_submittedFrames.end(),
+            [frameId](const UMC_VP9_DECODER::VP9DecoderFrame & item) { return item.currFrame == frameId; });
+
+        if (find_it != m_submittedFrames.end() && find_it->isDecoded)
+        {
+            UnLockResources(*find_it);
+            m_submittedFrames.erase(find_it);
+        }
+    }
+
     void CompleteFrames()
     {
         for (auto it = m_submittedFrames.begin(); it != m_submittedFrames.end(); )
@@ -923,6 +935,7 @@ mfxStatus MFX_CDECL VP9DECODERoutine(void *p_state, void * /* pp_param */, mfxU3
             decoder.m_surface_source->SetFreeSurfaceAllowedFlag(false);
         }
         decoder.m_framesStorage->DecodeFrame(data.currFrameId);
+        decoder.m_framesStorage->CompleteCurFrame(data.currFrameId);
 
         return MFX_ERR_NONE;
     }
@@ -960,6 +973,7 @@ mfxStatus MFX_CDECL VP9DECODERoutine(void *p_state, void * /* pp_param */, mfxU3
     if (data.currFrameId != -1)
         decoder.m_surface_source->DecreaseReference(data.currFrameId);
     decoder.m_framesStorage->DecodeFrame(data.currFrameId);
+    decoder.m_framesStorage->CompleteCurFrame(data.currFrameId);
 
     return MFX_TASK_DONE;
 }