xref: /aosp_15_r20/external/libhevc/encoder/hme_globals.h (revision c83a76b084498d55f252f48b2e3786804cdf24b7)

/******************************************************************************
 *
 * 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 hme_globals.h
*
* \brief
*    Contains all the global declarations used by HME
*
* \date
*    18/09/2012
*
* \author
*    Ittiam
*
******************************************************************************
*/

#ifndef _HME_GLOBALS_H_

/*****************************************************************************/
/* File Includes                                                             */
/*****************************************************************************/

/*****************************************************************************/
/* Globals                                                                   */
/*****************************************************************************/

/**
******************************************************************************
 *  @brief Converts an encode order to raster order x coord. Meant for 16x16
 *         CU within 64x64 or within 32x32
******************************************************************************
 */
extern U08 gau1_encode_to_raster_x[16];

/**
******************************************************************************
 *  @brief Converts an encode order to raster order y coord. Meant for 16x16
 *         CU within 64x64 or within 32x32
******************************************************************************
 */
extern U08 gau1_encode_to_raster_y[16];

/**
******************************************************************************
 *  @brief Given a CU id within the bigger CU (0..3), and the partition type
 *         currently within the small CU, we can figure out candidate
 *         partition types for bigger CU. E.g. IF CU id is 0, and is AMP of
 *         nLx2N, candidate partitions for bigger CU are nLx2N and 2Nx2N
******************************************************************************
 */
extern PART_TYPE_T ge_part_type_to_merge_part[4][MAX_PART_TYPES][3];

/**
******************************************************************************
 *  @brief A given partition type has 1,2 or 4 partitions, each corresponding
 *  to a unique partition id PART_ID_T enum type. So, this global converts
 *  partition type to a bitmask of corresponding partition ids.
******************************************************************************
 */
extern S32 gai4_part_type_to_part_mask[MAX_PART_TYPES];

/**
******************************************************************************
 *  @brief Reads out the index of function pointer to a sad_compute function
 *         of blk given a blk size enumeration
******************************************************************************
 */
extern U08 gau1_blk_size_to_fp[NUM_BLK_SIZES];

/**
******************************************************************************
 *  @brief Reads out the width of blk given a blk size enumeration
******************************************************************************
 */
extern U08 gau1_blk_size_to_wd[NUM_BLK_SIZES];

extern U08 gau1_blk_size_to_wd_shift[NUM_BLK_SIZES];

/**
******************************************************************************
 *  @brief Reads out the height of blk given a blk size enumeration
******************************************************************************
 */
extern U08 gau1_blk_size_to_ht[NUM_BLK_SIZES];

/**
******************************************************************************
 *  @brief Given a minimum pt enum in a 3x3 grid, reads out the list of active
 *  search pts in next iteration as a bit-mask, eliminating need to search
 *  pts that have already been searched in this iteration.
******************************************************************************
 */
extern S32 gai4_opt_grid_mask[NUM_GRID_PTS];

/**
******************************************************************************
 *  @brief Given a minimum pt enum in a 3x3 grid, reads out the x offset of
 *  the min pt relative to center assuming step size of 1
******************************************************************************
 */
extern S08 gai1_grid_id_to_x[NUM_GRID_PTS];

/**
******************************************************************************
 *  @brief Given a minimum pt enum in a 3x3 grid, reads out the y offset of
 *  the min pt relative to center assuming step size of 1
******************************************************************************
*/
extern S08 gai1_grid_id_to_y[NUM_GRID_PTS];

/**
******************************************************************************
 *  @brief Lookup of the blk size enum, given a specific partition and cu size
******************************************************************************
*/
extern BLK_SIZE_T ge_part_id_to_blk_size[NUM_CU_SIZES][TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief For a given partition split, find number of partitions
******************************************************************************
*/
extern U08 gau1_num_parts_in_part_type[MAX_PART_TYPES];

/**
******************************************************************************
 *  @brief For a given partition split, returns the enumerations of specific
 *  partitions in raster order. E.g. for PART_2NxN, part id 0 is
 *  PART_ID_2NxN_T and part id 1 is PART_ID_2NxN_B
******************************************************************************
*/
extern PART_ID_T ge_part_type_to_part_id[MAX_PART_TYPES][MAX_NUM_PARTS];

/**
******************************************************************************
 *  @brief For a given partition id, returs the rectangular position and size
 *  of partition within cu relative ot cu start.
******************************************************************************
*/
extern part_attr_t gas_part_attr_in_cu[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief Gives the CU type enumeration given a  blk size.
******************************************************************************
*/
extern CU_SIZE_T ge_blk_size_to_cu_size[NUM_BLK_SIZES];

/**
******************************************************************************

 *  @brief Given a minimum pt enum in a diamond grid, reads out the list
 *  of active search pts in next iteration as a bit-mask, eliminating need
 *  to search pts that have already been searched in this iteration.
******************************************************************************
 */
extern S32 gai4_opt_grid_mask_diamond[5];

/**
******************************************************************************
 *  @brief Given a minimum pt enum in a 9 point grid, reads out the list
 *  of active search pts in next iteration as a bit-mask, eliminating need
 *  to search pts that have already been searched in this iteration.
******************************************************************************
 */

extern S32 gai4_opt_grid_mask_conventional[9];

/**
******************************************************************************
 *  @brief Given a raster coord x, y, this aray returns the CU id in encoding
 *         order. Indexed as [y][x]
******************************************************************************
 */
extern U08 gau1_cu_id_raster_to_enc[4][4];
/**
******************************************************************************
 *  @brief Given a CU size, this array returns blk size enum
******************************************************************************
 */
extern BLK_SIZE_T ge_cu_size_to_blk_size[NUM_CU_SIZES];

/**
******************************************************************************
 *  @brief Given a part type, returns whether the part type is vertically
 *          oriented.
******************************************************************************
 */
extern U08 gau1_is_vert_part[MAX_PART_TYPES];

/**
******************************************************************************
 *  @brief Given a partition, returns the number of best results to consider
 *          for full pell refinement.
******************************************************************************
 */
extern U08 gau1_num_best_results_PQ[TOT_NUM_PARTS];
extern U08 gau1_num_best_results_HQ[TOT_NUM_PARTS];
extern U08 gau1_num_best_results_MS[TOT_NUM_PARTS];
extern U08 gau1_num_best_results_HS[TOT_NUM_PARTS];
extern U08 gau1_num_best_results_XS[TOT_NUM_PARTS];
extern U08 gau1_num_best_results_XS25[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief gau1_cu_tr_valid[y][x] returns the validity of a top rt candt for
 *         CU with raster id x, y within CTB. Valid for 16x16 CUs and above
******************************************************************************
 */
extern U08 gau1_cu_tr_valid[4][4];
/**
******************************************************************************
 *  @brief gau1_cu_tr_valid[y][x] returns the validity of a bot lt candt for
 *         CU with raster id x, y within CTB. Valid for 16x16 CUs and above
******************************************************************************
 */
extern U08 gau1_cu_bl_valid[4][4];

/**
******************************************************************************
 *  @brief Returns the validity of top rt candt for a given part id, will not
 *          be valid if tr of a part pts to a non causal neighbour like 16x8B
******************************************************************************
 */
extern U08 gau1_partid_tr_valid[TOT_NUM_PARTS];
/**
******************************************************************************
 *  @brief Returns the validity of bottom left cant for given part id, will
 *  not be valid, if bl of a part pts to a non causal neighbour like 8x16R
******************************************************************************
 */
extern U08 gau1_partid_bl_valid[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief The number of partition id in the CU, e.g. PART_ID_16x8_B is 2nd
******************************************************************************
 */
extern U08 gau1_part_id_to_part_num[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief Returns partition type for a given partition id, e.g.
 *  PART_ID_16x8_B returns PRT_TYPE_16x8
******************************************************************************
 */
extern PART_TYPE_T ge_part_id_to_part_type[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief given raster id x, y of 8x8 blk in 64x64 CTB, return the enc order
******************************************************************************
 */
extern U08 gau1_8x8_cu_id_raster_to_enc[8][8];

/**
******************************************************************************
 *  @brief Return the bits for a given partition id which gets added to the
 *  cost. Although the bits are for a given partition type, we add off the
 *  bits per partition while computing mv cost. For example, if the bits for
 *  2NxN part type is 3, we add 1.5 bits for 2NxN_T and 1.5 for 2NxN_B.
 *  Hence this is stored in Q1 format
******************************************************************************
*/
extern U08 gau1_bits_for_part_id_q1[TOT_NUM_PARTS];

/**
******************************************************************************
 *  @brief Returns 1 if there are qpel points to the top and bottom of the
 *         current point
******************************************************************************
*/
extern S32 gai4_2pt_qpel_interpol_possible_vert[4][4];

/**
******************************************************************************
 *  @brief Returns 1 if there are qpel points to the left and right of the
 *         current point
******************************************************************************
*/
extern S32 gai4_2pt_qpel_interpol_possible_horz[4][4];

/**
******************************************************************************
 *  @brief For a given partition split, num bits to encode the partition type
 *         and split cu,tu bits; assuming one bin equal to one bit for now
******************************************************************************
*/
extern U08 gau1_num_bits_for_part_type[MAX_PART_TYPES];

/**
******************************************************************************
 * @brief Used exclusively in the Intrinsics version of the function
 *       'hme_combine_4x4_sads_and_compute_cost_high_speed' instead
 *       of calling get_range()
******************************************************************************
 */
extern S16 gi2_mvy_range[MAX_MVY_SUPPORTED_IN_COARSE_LAYER + 1][8];

/**
******************************************************************************
 * @brief Used exclusively in the Intrinsics version of the function
 *       'hme_combine_4x4_sads_and_compute_cost_high_speed' instead
 *       of calling get_range()
******************************************************************************
 */
extern S16 gi2_mvx_range[MAX_MVX_SUPPORTED_IN_COARSE_LAYER * 2 + 1][8];

extern S32 gai4_select_qpel_function_vert[4][16];

extern S32 gai4_select_qpel_function_horz[4][16];

extern S32 gai4_partition_area[TOT_NUM_PARTS];

extern const U08 gau1_search_cand_priority_in_l1_and_l2_me[2][NUM_SEARCH_CAND_TYPES];

/* 12 cases are -  */
/* case  0 - P picture, num_refs=1, 4x4 in L1ME = 0 */
/* case  1 - P picture, num_refs=1, 4x4 in L1ME = 1 */
/* case  2 - P picture, num_refs=2, 4x4 in L1ME = 0 */
/* case  3 - P picture, num_refs=2, 4x4 in L1ME = 1 */
/* case  4 - P picture, num_refs=3, 4x4 in L1ME = 0 */
/* case  5 - P picture, num_refs=3, 4x4 in L1ME = 1 */
/* case  6 - P picture, num_refs=3, 4x4 in L1ME = 0 */
/* case  7 - P picture, num_refs=3, 4x4 in L1ME = 1 */
/* case  8 - B picture, num_refs=1, 4x4 in L1ME = 0 */
/* case  9 - B picture, num_refs=1, 4x4 in L1ME = 1 */
/* case 10 - B picture, num_refs=2, 4x4 in L1ME = 0 */
/* case 11 - B picture, num_refs=2, 4x4 in L1ME = 1 */
extern const U08 gau1_search_cand_priority_in_l0_me[12][NUM_SEARCH_CAND_TYPES];

extern const SEARCH_CANDIDATE_TYPE_T
    gae_search_cand_priority_to_search_cand_type_map_in_l0_me[12][NUM_SEARCH_CAND_TYPES];

extern const U08 gau1_max_num_search_cands_in_l0_me[12];

extern const SEARCH_CAND_LOCATIONS_T gae_search_cand_type_to_location_map[NUM_SEARCH_CAND_TYPES];

extern const S08 gai1_search_cand_type_to_result_id_map[NUM_SEARCH_CAND_TYPES];

extern const U08 gau1_search_cand_type_to_spatiality_map[NUM_SEARCH_CAND_TYPES];

extern const S32 gai4_search_cand_location_to_x_offset_map[NUM_SEARCH_CAND_LOCATIONS];

extern const S32 gai4_search_cand_location_to_y_offset_map[NUM_SEARCH_CAND_LOCATIONS];

/**
******************************************************************************
 * @brief Used exclusively in the Intrinsics version of the function
 *       'hme_combine_4x4_sads_and_compute_cost_high_quality' instead
 *       of calling get_range()
******************************************************************************
 */
extern S16 gi2_mvx_range_high_quality[MAX_MVX_SUPPORTED_IN_COARSE_LAYER * 2 + 1][8];

extern const S16 gai2_mvx_range_mapping[8193][8];

extern const S16 gai2_mvy_range_mapping[4097][8];

extern const S16 gai2_set_best_cost_max[8][8];

extern const S08 gai1_mv_adjust[8][2];

extern const S08 gai1_mv_offsets_from_center_in_rect_grid[NUM_POINTS_IN_RECTANGULAR_GRID][2];

#endif /* #ifndef _HME_GLOBALS_H_*/