1 /*
2 * Copyright (c) 2019-2021, 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_slice_packet_xe_m_base.h
24 //! \brief    Defines the implementation of hevc decode slice packet
25 //!
26 
27 #ifndef __DECODE_HEVC_SLICE_PACKET_XE_M_BASE_H__
28 #define __DECODE_HEVC_SLICE_PACKET_XE_M_BASE_H__
29 
30 #include "media_cmd_packet.h"
31 #include "decode_hevc_pipeline.h"
32 #include "decode_utils.h"
33 #include "decode_hevc_basic_feature.h"
34 #include "codechal_hw.h"
35 
36 namespace decode
37 {
38 
39 class HevcDecodeSlcPktXe_M_Base : public DecodeSubPacket
40 {
41 public:
HevcDecodeSlcPktXe_M_Base(HevcPipeline * pipeline,CodechalHwInterface * hwInterface)42     HevcDecodeSlcPktXe_M_Base(HevcPipeline *pipeline, CodechalHwInterface *hwInterface)
43         : DecodeSubPacket(pipeline, *hwInterface), m_hevcPipeline(pipeline)
44     {
45         m_hwInterface = hwInterface;
46         if (m_hwInterface != nullptr)
47         {
48             m_miInterface  = m_hwInterface->GetMiInterface();
49             m_osInterface  = m_hwInterface->GetOsInterface();
50             m_hcpInterface  = hwInterface->GetHcpInterface();
51         }
52     }
53     virtual ~HevcDecodeSlcPktXe_M_Base();
54 
55     //!
56     //! \brief  Initialize the media packet, allocate required resources
57     //! \return MOS_STATUS
58     //!         MOS_STATUS_SUCCESS if success, else fail reason
59     //!
60     virtual MOS_STATUS Init() override;
61 
62     //!
63     //! \brief  Prepare interal parameters, should be invoked for each frame
64     //! \return MOS_STATUS
65     //!         MOS_STATUS_SUCCESS if success, else fail reason
66     //!
67     virtual MOS_STATUS Prepare() override;
68 
69     //!
70     //! \brief  Execute hevc slice packet
71     //! \return MOS_STATUS
72     //!         MOS_STATUS_SUCCESS if success, else fail reason
73     //!
74     virtual MOS_STATUS Execute(MOS_COMMAND_BUFFER& cmdBuffer, uint32_t sliceIdx, uint32_t subTileIdx) = 0;
75 
76     //!
77     //! \brief  Calculate Command Size
78     //!
79     //! \param  [in, out] commandBufferSize
80     //!         requested size
81     //! \param  [in, out] requestedPatchListSize
82     //!         requested size
83     //! \return MOS_STATUS
84     //!         status
85     //!
86     MOS_STATUS CalculateCommandSize(
87         uint32_t &commandBufferSize,
88         uint32_t &requestedPatchListSize) override;
89 
90 protected:
91     virtual MOS_STATUS ValidateSubTileIdx(const HevcTileCoding::SliceTileInfo &sliceTileInfo,
92                                           uint32_t subTileIdx);
93 
94     virtual MOS_STATUS SetHcpSliceStateParams(MHW_VDBOX_HEVC_SLICE_STATE &sliceStateParams,
95                                               uint32_t sliceIdx, uint32_t subTileIdx);
96 
97     virtual MOS_STATUS SetRefIdxParams(MHW_VDBOX_HEVC_REF_IDX_PARAMS &refIdxParams,
98                                        uint32_t sliceIdx);
99 
100     virtual MOS_STATUS SetWeightOffsetParams(
101         MHW_VDBOX_HEVC_WEIGHTOFFSET_PARAMS &weightOffsetParams, uint32_t sliceIdx);
102     virtual MOS_STATUS AddWeightOffset(MOS_COMMAND_BUFFER &cmdBuffer, uint32_t sliceIdx);
103 
104     virtual MOS_STATUS SetBsdObjParams(MHW_VDBOX_HCP_BSD_PARAMS &bsdObjParams,
105                                        uint32_t sliceIdx, uint32_t subTileIdx);
106     virtual MOS_STATUS AddBsdObj(MOS_COMMAND_BUFFER &cmdBuffer, uint32_t sliceIdx,
107                                  uint32_t subTileIdx);
108     MOS_STATUS AddHcpCpState(MOS_COMMAND_BUFFER &cmdBuffer, uint32_t sliceIdx, uint32_t subTileIdx);
109 
110     //!
111     //! \brief  Calculate slcie level command Buffer Size
112     //!
113     //! \return uint32_t
114     //!         Command buffer size calculated
115     //!
116     virtual MOS_STATUS CalculateSliceStateCommandSize();
117 
118     HevcPipeline *         m_hevcPipeline     = nullptr;
119     MhwVdboxHcpInterface * m_hcpInterface     = nullptr;
120     HevcBasicFeature *     m_hevcBasicFeature = nullptr;
121     DecodeAllocator *      m_allocator        = nullptr;
122     CodechalHwInterface   *m_hwInterface      = nullptr;
123     MhwMiInterface        *m_miInterface      = nullptr;
124 
125     // Parameters passed from application
126     PCODEC_HEVC_PIC_PARAMS       m_hevcPicParams = nullptr;      //!< Pointer to picture parameter
127     PCODEC_HEVC_SLICE_PARAMS     m_hevcSliceParams = nullptr;    //!< Pointer to slice parameter
128     PCODEC_HEVC_EXT_PIC_PARAMS   m_hevcRextPicParams = nullptr;  //!< Extended pic params for Rext
129     PCODEC_HEVC_EXT_SLICE_PARAMS m_hevcRextSliceParams = nullptr;//!< Extended slice params for Rext
130     PCODEC_HEVC_SCC_PIC_PARAMS   m_hevcSccPicParams = nullptr;   //!< Pic params for SCC
131 
132     uint32_t m_sliceStatesSize      = 0;  //!< Slice state command size
133     uint32_t m_slicePatchListSize   = 0;  //!< Slice patch list size
134 MEDIA_CLASS_DEFINE_END(decode__HevcDecodeSlcPktXe_M_Base)
135 };
136 
137 }  // namespace decode
138 #endif
139