1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
/*
* Copyright (c) 2017-2021, Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
//!
//! \file media_ddi_decode_av1.h
//! \brief Defines DdiDecodeAV1 class for AV1 decode
//!
#ifndef __MEDIA_DDI_DECODER_AV1_H__
#define __MEDIA_DDI_DECODER_AV1_H__
#include <va/va.h>
#include "media_ddi_decode_base.h"
#include "codec_def_decode_av1.h"
#define DECODE_ID_AV1 "VIDEO_DEC_AV1"
#define MAX_ANCHOR_FRAME_NUM_AV1 128
//!
//! \class DdiDecodeAV1
//! \brief Ddi decode AV1
//!
typedef struct _DDI_CODEC_BUFFER_PARAM_AV1
{
// one picture buffer
VADecPictureParameterBufferAV1 PicParamAV1;
VASliceParameterBufferAV1 *pVASliceParameterBufferAV1;
} DDI_CODEC_BUFFER_PARAM_AV1;
class DdiDecodeAV1 : public DdiMediaDecode
{
public:
//!
//! \brief Constructor
//!
DdiDecodeAV1(DDI_DECODE_CONFIG_ATTR *ddiDecodeAttr) : DdiMediaDecode(ddiDecodeAttr)
{
MOS_ZeroMemory(&outputSurface, sizeof(outputSurface));
};
//!
//! \brief Destructor
//!
virtual ~DdiDecodeAV1() {};
// inherited virtual functions
virtual void DestroyContext(
VADriverContextP ctx) override;
virtual VAStatus RenderPicture(
VADriverContextP ctx,
VAContextID context,
VABufferID *buffers,
int32_t numBuffers) override;
virtual VAStatus InitDecodeParams(
VADriverContextP ctx,
VAContextID context) override;
virtual VAStatus SetDecodeParams() override;
virtual MOS_FORMAT GetFormat() override;
/*virtual VAStatus EndPicture(
VADriverContextP ctx,
VAContextID context) override;*/
virtual void ContextInit(
int32_t picWidth,
int32_t picHeight) override;
virtual VAStatus CodecHalInit(
DDI_MEDIA_CONTEXT *mediaCtx,
void *ptr) override;
virtual VAStatus AllocSliceControlBuffer(
DDI_MEDIA_BUFFER *buf) override;
virtual uint8_t* GetPicParamBuf(
DDI_CODEC_COM_BUFFER_MGR *bufMgr) override;
private:
//!
//! \brief ParseSliceParam for AV1
//! \details parse the sliceParam info required by AV1 decoding for
//! each slice
//!
//! \param [in] *mediaCtx
//! DDI_MEDIA_CONTEXT
//! \param [in] *slcParam
//! VASliceParameterBufferAV1
//! \param [in] numTile
//!
//! \return VA_STATUS_SUCCESS is returned if it is parsed successfully.
//! else fail reason
VAStatus ParseTileParams(
DDI_MEDIA_CONTEXT *mediaCtx,
VASliceParameterBufferAV1 *slcParam,
uint32_t numTiles);
//! \brief ParsePicParam for AV1
//! \details parse the PicParam info required by AV1 decoding
//!
//! \param [in] *mediaCtx
//! DDI_MEDIA_CONTEXT
//! \param [in] *qMatrix
//! VAIQMatrixBufferH264
//!
//! \return VAStatus
//! VA_STATUS_SUCCESS if success, else fail reason
//! \return VA_STATUS_SUCCESS is returned if it is parsed successfully.
//! else fail reason
VAStatus ParsePicParams(
DDI_MEDIA_CONTEXT *mediaCtx,
VADecPictureParameterBufferAV1 *picParam);
VAStatus ParseAv1SegFilterLevel(
DDI_MEDIA_CONTEXT *mediaCtx,
VADecPictureParameterBufferAV1 *picParam);
VAStatus Av1LoopFilterFrameInit(
DDI_MEDIA_CONTEXT *mediaCtx,
VADecPictureParameterBufferAV1 *picParam,
int defaultFiltLvl,
int defaultFiltLvlR,
int plane);
uint32_t Av1GetQindex(
CodecAv1SegmentsParams *segInfo,
uint32_t segment_id,
uint8_t base_qindex);
int Av1Clamp(int value, int low, int high);
//! \brief Init Resource buffer for AV1
//! \details Initialize and allocate the Resource buffer for AV1
//!
//! \return VA_STATUS_SUCCESS is returned if it is parsed successfully.
//! else fail reason
VAStatus InitResourceBuffer();
//! \brief Free Resource buffer for AV1
//!
void FreeResourceBuffer();
//! \brief film grain output surface
PDDI_MEDIA_SURFACE filmGrainOutSurface = nullptr;
//! \brief film grain output surface structure
MOS_SURFACE outputSurface;
MOS_SURFACE anchorFrameList[MAX_ANCHOR_FRAME_NUM_AV1];
VASurfaceID anchorFrameListVA[MAX_ANCHOR_FRAME_NUM_AV1] = {0};
};
#endif /* _MEDIA_DDI_DECODE_AV1_H */
|
