[Encode] Enable AVC reference surface dump

Enable AVC reference surface dump for APO code path.

Author: walter-bai

media_softlet/agnostic/common/codec/hal/enc/avc/packet/encode_avc_vdenc_packet.cpp

@@ -192,6 +192,10 @@ namespace encode {
 
         ENCODE_CHK_STATUS_RETURN(Mos_Solo_PostProcessEncode(m_osInterface, &m_basicFeature->m_resBitstreamBuffer, &m_basicFeature->m_reconSurface));
 
+#if USE_CODECHAL_DEBUG_TOOL
+        ENCODE_CHK_STATUS_RETURN(DumpReferences());
+#endif
+
         return MOS_STATUS_SUCCESS;
     }
 
@@ -2677,6 +2681,18 @@ namespace encode {
         return MOS_STATUS_SUCCESS;
     }
 
+    MOS_STATUS AvcVdencPkt::DumpReferences()
+    {
+        ENCODE_FUNC_CALL();
+
+        auto *debugInterface = m_pipeline->GetDebugInterface();
+        ENCODE_CHK_NULL_RETURN(debugInterface);
+
+        ENCODE_CHK_STATUS_RETURN(m_basicFeature->m_ref->DumpReferences(*debugInterface));
+
+        return MOS_STATUS_SUCCESS;
+    }
+
 #endif
 
 }

media_softlet/agnostic/common/codec/hal/enc/avc/packet/encode_avc_vdenc_packet.h

@@ -449,6 +449,8 @@ class AvcVdencPkt : public CmdPacket,
 
     virtual MOS_STATUS DumpEncodeImgStats(
         PMOS_COMMAND_BUFFER cmdbuffer);
+
+    MOS_STATUS DumpReferences();
 #endif
 
     AvcVdencPipeline         *m_pipeline        = nullptr;

media_softlet/agnostic/common/codec/hal/enc/avc/pipeline/encode_avc_reference_frames.cpp

@@ -411,6 +411,64 @@ int32_t AvcReferenceFrames::GetBiWeight(
     return biWeight;
 }
 
+#if USE_CODECHAL_DEBUG_TOOL
+MOS_STATUS AvcReferenceFrames::DumpReferences(CodechalDebugInterface &debugInterface)
+{
+    ENCODE_FUNC_CALL();
+
+    if (m_pictureCodingType == I_TYPE)
+    {
+        return MOS_STATUS_SUCCESS;
+    }
+
+    for (uint8_t refIdx = 0; refIdx <= m_basicFeature->m_sliceParams->num_ref_idx_l0_active_minus1; refIdx++)
+    {
+        auto refPic = m_basicFeature->m_sliceParams->RefPicList[LIST_0][refIdx];
+        if (!CodecHal_PictureIsInvalid(refPic) && m_picIdx[refPic.FrameIdx].bValid)
+        {
+            auto refPicIdx = m_picIdx[refPic.FrameIdx].ucPicIdx;
+            ENCODE_CHK_STATUS_RETURN(debugInterface.DumpYUVSurface(
+                &m_refList[refPicIdx]->sRefReconBuffer,
+                CodechalDbgAttr::attrReferenceSurfaces,
+                ("RefSurfL0" + std::to_string((int)refIdx)).c_str()));
+
+            uint8_t scaledIdx        = m_refList[refPicIdx]->ucScalingIdx;
+            auto    vdenc4xDsSurface = m_basicFeature->m_trackedBuf->GetSurface(BufferType::ds4xSurface, scaledIdx);
+            ENCODE_CHK_STATUS_RETURN(debugInterface.DumpYUVSurface(
+                vdenc4xDsSurface,
+                CodechalDbgAttr::attrReferenceSurfaces,
+                ("4xDsRefSurfL0" + std::to_string((int)refIdx)).c_str()));
+        }
+    }
+
+    if (m_pictureCodingType == B_TYPE)
+    {
+        for (uint8_t refIdx = 0; refIdx <= m_basicFeature->m_sliceParams->num_ref_idx_l1_active_minus1; refIdx++)
+        {
+            auto refPic = m_basicFeature->m_sliceParams->RefPicList[LIST_1][refIdx];
+
+            if (!CodecHal_PictureIsInvalid(refPic) && m_picIdx[refPic.FrameIdx].bValid)
+            {
+                auto refPicIdx = m_picIdx[refPic.FrameIdx].ucPicIdx;
+                ENCODE_CHK_STATUS_RETURN(debugInterface.DumpYUVSurface(
+                    &m_refList[refPicIdx]->sRefReconBuffer,
+                    CodechalDbgAttr::attrReferenceSurfaces,
+                    ("RefSurfL1" + std::to_string((int)refIdx)).c_str()));
+
+                uint8_t scaledIdx        = m_refList[refPicIdx]->ucScalingIdx;
+                auto    vdenc4xDsSurface = m_basicFeature->m_trackedBuf->GetSurface(BufferType::ds4xSurface, scaledIdx);
+                ENCODE_CHK_STATUS_RETURN(debugInterface.DumpYUVSurface(
+                    vdenc4xDsSurface,
+                    CodechalDbgAttr::attrReferenceSurfaces,
+                    ("4xDsRefSurfL1" + std::to_string((int)refIdx)).c_str()));
+            }
+        }
+    }
+
+    return MOS_STATUS_SUCCESS;
+}
+#endif
+
 MOS_STATUS AvcReferenceFrames::SetFrameStoreIds(uint8_t frameIdx)
 {
     uint8_t invalidFrame = (m_basicFeature->m_mode == CODECHAL_DECODE_MODE_AVCVLD) ? 0x7f : 0x1f;

media_softlet/agnostic/common/codec/hal/enc/avc/pipeline/encode_avc_reference_frames.h

@@ -31,6 +31,7 @@
 #include "encode_allocator.h"
 #include "mhw_vdbox_vdenc_itf.h"
 #include "mhw_vdbox_mfx_itf.h"
+#include "codechal_debug.h"
 
 namespace encode
 {
@@ -127,6 +128,10 @@ class AvcReferenceFrames : public mhw::vdbox::vdenc::Itf::ParSetting, public mhw
 
     MHW_SETPAR_DECL_HDR(MFX_PIPE_BUF_ADDR_STATE);
 
+#if USE_CODECHAL_DEBUG_TOOL
+    MOS_STATUS DumpReferences(CodechalDebugInterface &debugInterface);
+#endif
+
 protected:
 
     //!