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
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
|
/******************************************************************************
*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*****************************************************************************
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
*/
/*!
******************************************************************************
* \file ihevce_me_pass.h
*
* \brief
* Interfaces to create, control and run the ME module
*
* \date
* 22/10/2012
*
* \author
* Ittiam
*
******************************************************************************
*/
#ifndef _IHEVCE_ME_PASS_H_
#define _IHEVCE_ME_PASS_H_
/*****************************************************************************/
/* Constant Macros */
/*****************************************************************************/
/*****************************************************************************/
/* Function Macros */
/*****************************************************************************/
/*****************************************************************************/
/* Typedefs */
/*****************************************************************************/
/*****************************************************************************/
/* Enums */
/*****************************************************************************/
/*****************************************************************************/
/* Structure */
/*****************************************************************************/
/*****************************************************************************/
/* Extern Variable Declarations */
/*****************************************************************************/
/*****************************************************************************/
/* Extern Function Declarations */
/*****************************************************************************/
/*!
******************************************************************************
* \if Function name : ihevce_me_get_num_mem_recs \endif
*
* \brief
* Number of memory records are returned for ME module
*
*
* \return
* Number of memory records
*
* \author
* Ittiam
*
*****************************************************************************
*/
WORD32 ihevce_me_get_num_mem_recs(WORD32 i4_num_me_frm_pllel);
/*!
******************************************************************************
* \if Function name : ihevce_me_get_mem_recs \endif
*
* \brief
* Memory requirements are returned for ME.
*
* \param[in,out] ps_mem_tab : pointer to memory descriptors table
* \param[in] ps_init_prms : Create time static parameters
*
* \return
* Number of records
*
* \author
* Ittiam
*
*****************************************************************************
*/
WORD32 ihevce_me_get_mem_recs(
iv_mem_rec_t *ps_mem_tab,
ihevce_static_cfg_params_t *ps_init_prms,
WORD32 i4_num_proc_thrds,
WORD32 i4_mem_space,
WORD32 i4_resolution_id,
WORD32 i4_num_me_frm_pllel);
/*!
******************************************************************************
* \if Function name : ihevce_me_init \endif
*
* \brief
* Intialization for ME context state structure .
*
* \param[in] ps_mem_tab : pointer to memory descriptors table
* \param[in] ps_init_prms : Create time static parameters
* \param[in] pv_osal_handle : Osal handle
*
* \return
* Handle to the ME context
*
* \author
* Ittiam
*
*****************************************************************************
*/
void *ihevce_me_init(
iv_mem_rec_t *ps_mem_tab,
ihevce_static_cfg_params_t *ps_init_prms,
WORD32 i4_num_proc_thrds,
void *pv_osal_handle,
rc_quant_t *ps_rc_quant_ctxt,
void *pv_tile_params_base,
WORD32 i4_resolution_id,
WORD32 i4_num_me_frm_pllel,
UWORD8 u1_is_popcnt_available);
/**
*******************************************************************************
* \if Function name : ihevce_me_set_resolution \endif
*
* \brief
* Sets the resolution for ME state
*
* \par Description:
* ME requires information of resolution to prime up its layer descriptors
* and contexts. This API is called whenever a control call from application
* causes a change of resolution. Has to be called once initially before
* processing any frame. Again this is just a glue function and calls the
* actual ME API for the same.
*
* \param[in,out] pv_me_ctxt: Handle to the ME context
* \param[in] n_enc_layers: Number of layers getting encoded
* \param[in] p_wd : Pointer containing widths of each layer getting encoded.
* \param[in] p_ht : Pointer containing heights of each layer getting encoded.
*
* \returns
* none
*
* \author
* Ittiam
*
*******************************************************************************
*/
void ihevce_me_set_resolution(void *pv_me_ctxt, WORD32 n_enc_layers, WORD32 *p_wd, WORD32 *p_ht);
/*!
******************************************************************************
* \if Function name : ihevce_me_frame_init \endif
*
* \brief
* Frame level ME initialisation function
*
* \par Description:
* The following pre-conditions exist for this function: a. We have the input
* pic ready for encode, b. We have the reference list with POC, L0/L1 IDs
* and ref ptrs ready for this picture and c. ihevce_me_set_resolution has
* been called atleast once. Once these are supplied, the following are
* done here: a. Input pyramid creation, b. Updation of ME's internal DPB
* based on available ref list information
*
* \param[in] pv_ctxt : pointer to ME module
* \param[in] ps_frm_ctb_prms : CTB characteristics parameters
* \param[in] ps_frm_lamda : Frame level Lambda params
* \param[in] num_ref_l0 : Number of reference pics in L0 list
* \param[in] num_ref_l1 : Number of reference pics in L1 list
* \param[in] num_ref_l0_active : Active reference pics in L0 dir for current frame (shall be <= num_ref_l0)
* \param[in] num_ref_l1_active : Active reference pics in L1 dir for current frame (shall be <= num_ref_l1)
* \param[in] pps_rec_list_l0 : List of recon pics in L0 list
* \param[in] pps_rec_list_l1 : List of recon pics in L1 list
* \param[in] ps_enc_lap_inp : pointer to input yuv buffer (frame buffer)
* \param[in] i4_frm_qp : current picture QP
*
* \return
* None
*
* \author
* Ittiam
*
*****************************************************************************
*/
void ihevce_me_frame_init(
void *pv_me_ctxt,
me_enc_rdopt_ctxt_t *ps_cur_out_me_prms,
ihevce_static_cfg_params_t *ps_stat_prms,
frm_ctb_ctxt_t *ps_frm_ctb_prms,
frm_lambda_ctxt_t *ps_frm_lamda,
WORD32 num_ref_l0,
WORD32 num_ref_l1,
WORD32 num_ref_l0_active,
WORD32 num_ref_l1_active,
recon_pic_buf_t **pps_rec_list_l0,
recon_pic_buf_t **pps_rec_list_l1,
recon_pic_buf_t *(*aps_ref_list)[HEVCE_MAX_REF_PICS * 2],
func_selector_t *ps_func_selector,
ihevce_lap_enc_buf_t *ps_enc_lap_inp,
void *pv_coarse_layer,
WORD32 i4_me_frm_id,
WORD32 i4_thrd_id,
WORD32 i4_frm_qp,
WORD32 i4_temporal_layer_id,
WORD8 i1_cu_qp_delta_enabled_flag,
void *pv_dep_mngr_encloop_dep_me);
/*!
******************************************************************************
* \if Function name : ihevce_me_process \endif
*
* \brief
* Frame level ME function
*
* \par Description:
* Processing of all layers starting from coarse and going
* to the refinement layers, all layers
* that are encoded go CTB by CTB. Outputs of this function are populated
* ctb_analyse_t structures, one per CTB.
*
* \param[in] pv_ctxt : pointer to ME module
* \param[in] ps_enc_lap_inp : pointer to input yuv buffer (frame buffer)
* \param[in,out] ps_ctb_out : pointer to CTB analyse output structure (frame buffer)
* \param[out] ps_cu_out : pointer to CU analyse output structure (frame buffer)
* \param[in] pd_intra_costs : pointerto intra cost buffer
* \param[in] ps_multi_thrd_ctxt : pointer to multi thread ctxt
* \param[in] thrd_id : Thread id of the current thrd in which function is executed
*
* \return
* None
*
* \author
* Ittiam
*
*****************************************************************************
*/
void ihevce_me_process(
void *pv_me_ctxt,
ihevce_lap_enc_buf_t *ps_enc_lap_inp,
ctb_analyse_t *ps_ctb_out,
me_enc_rdopt_ctxt_t *ps_cur_out_me_prms,
double *pd_intra_costs,
ipe_l0_ctb_analyse_for_me_t *ps_ipe_analyse_ctb,
pre_enc_L0_ipe_encloop_ctxt_t *ps_l0_ipe_input,
void *pv_coarse_layer,
multi_thrd_ctxt_t *ps_multi_thrd_ctxt,
WORD32 i4_frame_parallelism_level,
WORD32 thrd_id,
WORD32 i4_me_frm_id);
/*!
******************************************************************************
* \if Function name : ihevce_me_frame_dpb_update \endif
*
* \brief
* Frame level ME initialisation function
*
* \par Description:
* Updation of ME's internal DPB
* based on available ref list information
*
* \param[in] pv_ctxt : pointer to ME module
* \param[in] num_ref_l0 : Number of reference pics in L0 list
* \param[in] num_ref_l1 : Number of reference pics in L1 list
* \param[in] pps_rec_list_l0 : List of recon pics in L0 list
* \param[in] pps_rec_list_l1 : List of recon pics in L1 list
*
* \return
* None
*
* \author
* Ittiam
*
*****************************************************************************
*/
void ihevce_me_frame_dpb_update(
void *pv_me_ctxt,
WORD32 num_ref_l0,
WORD32 num_ref_l1,
recon_pic_buf_t **pps_rec_list_l0,
recon_pic_buf_t **pps_rec_list_l1,
WORD32 i4_thrd_id);
void ihevce_derive_me_init_prms(
ihevce_static_cfg_params_t *ps_init_prms,
hme_init_prms_t *ps_hme_init_prms,
S32 i4_num_proc_thrds,
WORD32 i4_resolution_id);
void ihevc_me_update_ref_desc(
hme_ref_desc_t *ps_ref_desc,
recon_pic_buf_t *ps_recon_pic,
WORD32 ref_id_l0,
WORD32 ref_id_l1,
WORD32 ref_id_lc,
WORD32 is_fwd);
WORD32 ihevce_me_find_poc_in_list(
recon_pic_buf_t **pps_rec_list, WORD32 poc, WORD32 i4_idr_gop_num, WORD32 num_ref);
void ihevce_me_create_ref_map(
recon_pic_buf_t **pps_rec_list_l0,
recon_pic_buf_t **pps_rec_list_l1,
WORD32 num_ref_l0_active,
WORD32 num_ref_l1_active,
WORD32 num_ref,
hme_ref_map_t *ps_ref_map);
/*!
******************************************************************************
* \if Function name : ihevce_l0_me_frame_end \endif
*
* \brief
* End of frame update function performs
* - Dynamic Search Range collation
*
* \param[in] pv_ctxt : pointer to ME module
*
* \return
* None
*
* \author
* Ittiam
*
*****************************************************************************
*/
void ihevce_l0_me_frame_end(
void *pv_me_ctxt, WORD32 i4_idx_dvsr_p, WORD32 i4_display_num, WORD32 i4_me_frm_id);
#endif /* _IHEVCE_ME_PASS_H_ */
|
