xref: /aosp_15_r20/external/libhevc/encoder/ihevce_enc_cu_recursion.h (revision c83a76b084498d55f252f48b2e3786804cdf24b7)
1*c83a76b0SSuyog Pawar /******************************************************************************
2*c83a76b0SSuyog Pawar  *
3*c83a76b0SSuyog Pawar  * Copyright (C) 2018 The Android Open Source Project
4*c83a76b0SSuyog Pawar  *
5*c83a76b0SSuyog Pawar  * Licensed under the Apache License, Version 2.0 (the "License");
6*c83a76b0SSuyog Pawar  * you may not use this file except in compliance with the License.
7*c83a76b0SSuyog Pawar  * You may obtain a copy of the License at:
8*c83a76b0SSuyog Pawar  *
9*c83a76b0SSuyog Pawar  * http://www.apache.org/licenses/LICENSE-2.0
10*c83a76b0SSuyog Pawar  *
11*c83a76b0SSuyog Pawar  * Unless required by applicable law or agreed to in writing, software
12*c83a76b0SSuyog Pawar  * distributed under the License is distributed on an "AS IS" BASIS,
13*c83a76b0SSuyog Pawar  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*c83a76b0SSuyog Pawar  * See the License for the specific language governing permissions and
15*c83a76b0SSuyog Pawar  * limitations under the License.
16*c83a76b0SSuyog Pawar  *
17*c83a76b0SSuyog Pawar  *****************************************************************************
18*c83a76b0SSuyog Pawar  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19*c83a76b0SSuyog Pawar */
20*c83a76b0SSuyog Pawar /*!
21*c83a76b0SSuyog Pawar ******************************************************************************
22*c83a76b0SSuyog Pawar * \file ihevce_enc_cu_recursion.h
23*c83a76b0SSuyog Pawar *
24*c83a76b0SSuyog Pawar * \brief
25*c83a76b0SSuyog Pawar *    This file contains interface declarations of encoder normative loop pass
26*c83a76b0SSuyog Pawar *    related functions
27*c83a76b0SSuyog Pawar *
28*c83a76b0SSuyog Pawar * \date
29*c83a76b0SSuyog Pawar *    18/09/2012
30*c83a76b0SSuyog Pawar *
31*c83a76b0SSuyog Pawar * \author
32*c83a76b0SSuyog Pawar *    Ittiam
33*c83a76b0SSuyog Pawar *
34*c83a76b0SSuyog Pawar ******************************************************************************
35*c83a76b0SSuyog Pawar */
36*c83a76b0SSuyog Pawar 
37*c83a76b0SSuyog Pawar #ifndef _IHEVCE_ENC_CU_RECURION_H_
38*c83a76b0SSuyog Pawar #define _IHEVCE_ENC_CU_RECURION_H_
39*c83a76b0SSuyog Pawar 
40*c83a76b0SSuyog Pawar /*****************************************************************************/
41*c83a76b0SSuyog Pawar /* Constant Macros                                                           */
42*c83a76b0SSuyog Pawar /*****************************************************************************/
43*c83a76b0SSuyog Pawar #define ENABLE_TREE_DUMP 0
44*c83a76b0SSuyog Pawar 
45*c83a76b0SSuyog Pawar /*****************************************************************************/
46*c83a76b0SSuyog Pawar /* Extern Function Declarations                                              */
47*c83a76b0SSuyog Pawar /*****************************************************************************/
48*c83a76b0SSuyog Pawar void ihevce_store_cu_final(
49*c83a76b0SSuyog Pawar     ihevce_enc_loop_ctxt_t *ps_ctxt,
50*c83a76b0SSuyog Pawar     cu_enc_loop_out_t *ps_cu_final,
51*c83a76b0SSuyog Pawar     UWORD8 *pu1_ecd_data,
52*c83a76b0SSuyog Pawar     ihevce_enc_cu_node_ctxt_t *ps_enc_out_ctxt,
53*c83a76b0SSuyog Pawar     enc_loop_cu_prms_t *ps_cu_prms);
54*c83a76b0SSuyog Pawar 
55*c83a76b0SSuyog Pawar void ihevce_populate_cu_tree(
56*c83a76b0SSuyog Pawar     ipe_l0_ctb_analyse_for_me_t *ps_cur_ipe_ctb,
57*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_cu_tree,
58*c83a76b0SSuyog Pawar     WORD32 tree_depth,
59*c83a76b0SSuyog Pawar     IHEVCE_QUALITY_CONFIG_T e_quality_preset,
60*c83a76b0SSuyog Pawar     CU_POS_T e_grandparent_blk_pos,
61*c83a76b0SSuyog Pawar     CU_POS_T e_parent_blk_pos,
62*c83a76b0SSuyog Pawar     CU_POS_T e_cur_blk_pos);
63*c83a76b0SSuyog Pawar 
64*c83a76b0SSuyog Pawar void ihevce_update_final_cu_results(
65*c83a76b0SSuyog Pawar     ihevce_enc_loop_ctxt_t *ps_ctxt,
66*c83a76b0SSuyog Pawar     ihevce_enc_cu_node_ctxt_t *ps_enc_out_ctxt,
67*c83a76b0SSuyog Pawar     enc_loop_cu_prms_t *ps_cu_prms,
68*c83a76b0SSuyog Pawar     pu_col_mv_t **pps_row_col_pu,
69*c83a76b0SSuyog Pawar     WORD32 *pi4_col_pu_map_idx,
70*c83a76b0SSuyog Pawar     cu_final_update_prms *ps_cu_update_prms,
71*c83a76b0SSuyog Pawar     WORD32 ctb_ctr,
72*c83a76b0SSuyog Pawar     WORD32 vert_ctb_ctr);
73*c83a76b0SSuyog Pawar 
74*c83a76b0SSuyog Pawar WORD32 ihevce_cu_recurse_decide(
75*c83a76b0SSuyog Pawar     ihevce_enc_loop_ctxt_t *ps_ctxt,
76*c83a76b0SSuyog Pawar     enc_loop_cu_prms_t *ps_cu_prms,
77*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_cu_tree_analyse,
78*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_cu_tree_analyse_parent,
79*c83a76b0SSuyog Pawar     ipe_l0_ctb_analyse_for_me_t *ps_cur_ipe_ctb,
80*c83a76b0SSuyog Pawar     me_ctb_data_t *ps_cu_me_data,
81*c83a76b0SSuyog Pawar     pu_col_mv_t **pps_col_pu,
82*c83a76b0SSuyog Pawar     cu_final_update_prms *ps_cu_update_prms,
83*c83a76b0SSuyog Pawar     UWORD8 *pu1_col_pu_map,
84*c83a76b0SSuyog Pawar     WORD32 *pi4_col_start_pu_idx,
85*c83a76b0SSuyog Pawar     WORD32 i4_tree_depth,
86*c83a76b0SSuyog Pawar     WORD32 i4_ctb_x_off,
87*c83a76b0SSuyog Pawar     WORD32 i4_ctb_y_off,
88*c83a76b0SSuyog Pawar     WORD32 cur_ctb_ht);
89*c83a76b0SSuyog Pawar 
90*c83a76b0SSuyog Pawar void ihevce_store_cu_results(
91*c83a76b0SSuyog Pawar     ihevce_enc_loop_ctxt_t *ps_ctxt,
92*c83a76b0SSuyog Pawar     enc_loop_cu_prms_t *ps_cu_prms,
93*c83a76b0SSuyog Pawar     final_mode_state_t *ps_final_state);
94*c83a76b0SSuyog Pawar 
95*c83a76b0SSuyog Pawar void ihevce_enc_loop_cu_bot_copy(
96*c83a76b0SSuyog Pawar     ihevce_enc_loop_ctxt_t *ps_ctxt,
97*c83a76b0SSuyog Pawar     enc_loop_cu_prms_t *ps_cu_prms,
98*c83a76b0SSuyog Pawar     ihevce_enc_cu_node_ctxt_t *ps_enc_out_ctxt,
99*c83a76b0SSuyog Pawar     WORD32 curr_cu_pos_in_row,
100*c83a76b0SSuyog Pawar     WORD32 curr_cu_pos_in_ctb);
101*c83a76b0SSuyog Pawar 
102*c83a76b0SSuyog Pawar void ihevce_intra_and_inter_cuTree_merger(
103*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_merged_tree,
104*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_intra_tree,
105*c83a76b0SSuyog Pawar     cur_ctb_cu_tree_t *ps_inter_tree,
106*c83a76b0SSuyog Pawar     WORD8 *pi1_8x8CULevel_intraData_availability_indicator);
107*c83a76b0SSuyog Pawar 
108*c83a76b0SSuyog Pawar #endif /* _IHEVCE_ENC_CU_RECURION_H_ */
109