1 /*
2 * Copyright (c) 2022, Intel Corporation
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included
12 * in all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
21 */
22 //!
23 //! \file     decode_hevc_packet_real_tile.cpp
24 //! \brief    Defines the interface for hevc real tile decode packet
25 //!
26 #include "decode_hevc_packet_real_tile.h"
27 #include "decode_status_report_defs.h"
28 #include "decode_predication_packet.h"
29 #include "decode_marker_packet.h"
30 
31 namespace decode {
32 
~HevcDecodeRealTilePkt()33 HevcDecodeRealTilePkt::~HevcDecodeRealTilePkt()
34 {
35 }
36 
Init()37 MOS_STATUS HevcDecodeRealTilePkt::Init()
38 {
39     DECODE_FUNC_CALL();
40     DECODE_CHK_STATUS(HevcDecodePkt::Init());
41 
42     DECODE_CHK_STATUS(m_statusReport->RegistObserver(this));
43 
44     DecodeSubPacket* subPacket = m_hevcPipeline->GetSubPacket(DecodePacketId(m_hevcPipeline, hevcPictureSubPacketId));
45     m_picturePkt = dynamic_cast<HevcDecodePicPkt*>(subPacket);
46     DECODE_CHK_NULL(m_picturePkt);
47 
48     subPacket = m_hevcPipeline->GetSubPacket(DecodePacketId(m_hevcPipeline, hevcSliceSubPacketId));
49     m_slicePkt = dynamic_cast<HevcDecodeSlcPkt*>(subPacket);
50     DECODE_CHK_NULL(m_slicePkt);
51 
52     return MOS_STATUS_SUCCESS;
53 }
54 
Destroy()55 MOS_STATUS HevcDecodeRealTilePkt::Destroy()
56 {
57     m_statusReport->UnregistObserver(this);
58     return MOS_STATUS_SUCCESS;
59 }
60 
InitSliceLevelCmdBuffer(MHW_BATCH_BUFFER & batchBuffer,uint8_t * batchBufBase,uint32_t tileColNum)61 MOS_STATUS HevcDecodeRealTilePkt::InitSliceLevelCmdBuffer(MHW_BATCH_BUFFER &batchBuffer, uint8_t *batchBufBase, uint32_t tileColNum)
62 {
63     DECODE_FUNC_CALL();
64 
65     DECODE_ASSERT(batchBuffer.count >= tileColNum);
66     if (tileColNum > m_sliceLevelCmdBuffer.size())
67     {
68         m_sliceLevelCmdBuffer.resize(tileColNum);
69     }
70 
71     uint32_t cmdBufferOffset = 0;
72 
73     for (uint32_t i = 0; i < tileColNum; i++)
74     {
75         auto &cmdBuffer = m_sliceLevelCmdBuffer[i];
76         MOS_ZeroMemory(&cmdBuffer, sizeof(MOS_COMMAND_BUFFER));
77         cmdBuffer.pCmdBase   = (uint32_t*)(batchBufBase + cmdBufferOffset);
78         cmdBuffer.pCmdPtr    = cmdBuffer.pCmdBase;
79         cmdBuffer.iRemaining = batchBuffer.iSize;
80 
81         cmdBufferOffset     += batchBuffer.iSize;
82     }
83 
84     return MOS_STATUS_SUCCESS;
85 }
86 
IsPrologRequired()87 bool HevcDecodeRealTilePkt::IsPrologRequired()
88 {
89     return m_hevcPipeline->IsFirstPass();
90 }
91 
IsLastTileCol()92 bool HevcDecodeRealTilePkt::IsLastTileCol()
93 {
94     if (!m_hevcPipeline->IsLastPass())
95     {
96         return false;
97     }
98 
99     uint32_t curTileColumn = m_hevcPipeline->GetCurrentPass() * m_hevcPipeline->GetPipeNum() +
100                              m_hevcPipeline->GetCurrentPipe();
101     return (curTileColumn >= m_hevcPicParams->num_tile_columns_minus1);
102 }
103 
CalculateCommandSize(uint32_t & commandBufferSize,uint32_t & requestedPatchListSize)104 MOS_STATUS HevcDecodeRealTilePkt::CalculateCommandSize(uint32_t &commandBufferSize, uint32_t &requestedPatchListSize)
105 {
106     DECODE_CHK_STATUS(CalculateCommandBufferSize(commandBufferSize));
107     DECODE_CHK_STATUS(CalculatePatchListSize(requestedPatchListSize));
108     return MOS_STATUS_SUCCESS;
109 }
110 
CalculateCommandBufferSize(uint32_t & commandBufferSize)111 MOS_STATUS HevcDecodeRealTilePkt::CalculateCommandBufferSize(uint32_t &commandBufferSize)
112 {
113     // Just need consider picture level cmds for first level command buffer size,
114     // since the slice level cmds are located 2nd lvl batch buffer which is
115     // allocated by pipeline.
116     DECODE_CHK_STATUS(m_picturePkt->CalculateCommandSize(m_pictureStatesSize, m_picturePatchListSize));
117     commandBufferSize = m_pictureStatesSize + COMMAND_BUFFER_RESERVED_SPACE;
118     return MOS_STATUS_SUCCESS;
119 }
120 
CalculatePatchListSize(uint32_t & requestedPatchListSize)121 MOS_STATUS HevcDecodeRealTilePkt::CalculatePatchListSize(uint32_t &requestedPatchListSize)
122 {
123     if (!m_osInterface->bUsesPatchList)
124     {
125         requestedPatchListSize = 0;
126         return MOS_STATUS_SUCCESS;
127     }
128 
129     DECODE_CHK_STATUS(m_slicePkt->CalculateCommandSize(m_sliceStatesSize, m_slicePatchListSize));
130 
131     uint32_t sliceCmdNum   = m_hevcBasicFeature->m_numSlices + 1 +
132                              m_hevcBasicFeature->m_hevcPicParams->num_tile_rows_minus1;
133     requestedPatchListSize = m_picturePatchListSize + (m_slicePatchListSize * sliceCmdNum);
134     return MOS_STATUS_SUCCESS;
135 }
136 
137 }
138