/src/libhevc/encoder/hme_interface.c

Source
/******************************************************************************
 *
 * 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 Includes                                                             */
/*****************************************************************************/
/* System include files */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <assert.h>
#include <stdarg.h>
#include <math.h>
#include <limits.h>

/* User include files */
#include "ihevc_typedefs.h"
#include "itt_video_api.h"
#include "ihevce_api.h"

#include "rc_cntrl_param.h"
#include "rc_frame_info_collector.h"
#include "rc_look_ahead_params.h"

#include "ihevc_defs.h"
#include "ihevc_structs.h"
#include "ihevc_platform_macros.h"
#include "ihevc_deblk.h"
#include "ihevc_itrans_recon.h"
#include "ihevc_chroma_itrans_recon.h"
#include "ihevc_chroma_intra_pred.h"
#include "ihevc_intra_pred.h"
#include "ihevc_inter_pred.h"
#include "ihevc_mem_fns.h"
#include "ihevc_padding.h"
#include "ihevc_weighted_pred.h"
#include "ihevc_sao.h"
#include "ihevc_resi_trans.h"
#include "ihevc_quant_iquant_ssd.h"
#include "ihevc_cabac_tables.h"

#include "ihevce_defs.h"
#include "ihevce_lap_enc_structs.h"
#include "ihevce_multi_thrd_structs.h"
#include "ihevce_multi_thrd_funcs.h"
#include "ihevce_me_common_defs.h"
#include "ihevce_had_satd.h"
#include "ihevce_error_codes.h"
#include "ihevce_bitstream.h"
#include "ihevce_cabac.h"
#include "ihevce_rdoq_macros.h"
#include "ihevce_function_selector.h"
#include "ihevce_enc_structs.h"
#include "ihevce_entropy_structs.h"
#include "ihevce_cmn_utils_instr_set_router.h"
#include "ihevce_enc_loop_structs.h"
#include "ihevce_bs_compute_ctb.h"
#include "ihevce_global_tables.h"
#include "ihevce_dep_mngr_interface.h"
#include "hme_datatype.h"
#include "hme_interface.h"
#include "hme_common_defs.h"
#include "hme_defs.h"
#include "ihevce_me_instr_set_router.h"
#include "hme_globals.h"
#include "hme_utils.h"
#include "hme_coarse.h"
#include "hme_refine.h"
#include "hme_err_compute.h"
#include "hme_common_utils.h"
#include "hme_search_algo.h"
#include "ihevce_profile.h"

/*****************************************************************************/
/* Function Definitions                                                      */
/*****************************************************************************/

void hme_init_globals()
{
    GRID_PT_T id;
    S32 i, j;
    /*************************************************************************/
    /* Initialize the lookup table for x offset, y offset, optimized mask    */
    /* based on grid id. The design is as follows:                           */
    /*                                                                       */
    /*     a  b  c  d                                                        */
    /*    TL  T TR  e                                                        */
    /*     L  C  R  f                                                        */
    /*    BL  B BR                                                           */
    /*                                                                       */
    /*  IF a non corner pt, like T is the new minima, then we need to        */
    /*  evaluate only 3 new pts, in this case, a, b, c. So the optimal       */
    /*  grid mask would reflect this. If a corner pt like TR is the new      */
    /*  minima, then we need to evaluate 5 new pts, in this case, b, c, d,   */
    /*  e and f. So the grid mask will have 5 pts enabled.                   */
    /*************************************************************************/

    id = PT_C;
    gai4_opt_grid_mask[id] = GRID_ALL_PTS_VALID ^ (BIT_EN(PT_C));
    gai1_grid_id_to_x[id] = 0;
    gai1_grid_id_to_y[id] = 0;
    gai4_opt_grid_mask_diamond[id] = GRID_DIAMOND_ENABLE_ALL ^ (BIT_EN(PT_C));
    gai4_opt_grid_mask_conventional[id] = GRID_ALL_PTS_VALID ^ (BIT_EN(PT_C));

    id = PT_L;
    gai4_opt_grid_mask[id] = BIT_EN(PT_TL) | BIT_EN(PT_L) | BIT_EN(PT_BL);
    gai1_grid_id_to_x[id] = -1;
    gai1_grid_id_to_y[id] = 0;
    gai4_opt_grid_mask_diamond[id] = BIT_EN(PT_T) | BIT_EN(PT_L) | BIT_EN(PT_B);
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_T) | BIT_EN(PT_L) | BIT_EN(PT_B);

    id = PT_R;
    gai4_opt_grid_mask[id] = BIT_EN(PT_TR) | BIT_EN(PT_R) | BIT_EN(PT_BR);
    gai1_grid_id_to_x[id] = 1;
    gai1_grid_id_to_y[id] = 0;
    gai4_opt_grid_mask_diamond[id] = BIT_EN(PT_T) | BIT_EN(PT_R) | BIT_EN(PT_B);
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_T) | BIT_EN(PT_R) | BIT_EN(PT_B);

    id = PT_T;
    gai4_opt_grid_mask[id] = BIT_EN(PT_TL) | BIT_EN(PT_T) | BIT_EN(PT_TR);
    gai1_grid_id_to_x[id] = 0;
    gai1_grid_id_to_y[id] = -1;
    gai4_opt_grid_mask_diamond[id] = BIT_EN(PT_R) | BIT_EN(PT_L) | BIT_EN(PT_T);
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_R) | BIT_EN(PT_L) | BIT_EN(PT_T);

    id = PT_B;
    gai4_opt_grid_mask[id] = BIT_EN(PT_BL) | BIT_EN(PT_B) | BIT_EN(PT_BR);
    gai1_grid_id_to_x[id] = 0;
    gai1_grid_id_to_y[id] = 1;
    gai4_opt_grid_mask_diamond[id] = BIT_EN(PT_B) | BIT_EN(PT_L) | BIT_EN(PT_R);
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_B) | BIT_EN(PT_L) | BIT_EN(PT_R);

    id = PT_TL;
    gai4_opt_grid_mask[id] = gai4_opt_grid_mask[PT_L] | gai4_opt_grid_mask[PT_T];
    gai1_grid_id_to_x[id] = -1;
    gai1_grid_id_to_y[id] = -1;
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_T) | BIT_EN(PT_L);

    id = PT_TR;
    gai4_opt_grid_mask[id] = gai4_opt_grid_mask[PT_R] | gai4_opt_grid_mask[PT_T];
    gai1_grid_id_to_x[id] = 1;
    gai1_grid_id_to_y[id] = -1;
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_T) | BIT_EN(PT_R);

    id = PT_BL;
    gai4_opt_grid_mask[id] = gai4_opt_grid_mask[PT_L] | gai4_opt_grid_mask[PT_B];
    gai1_grid_id_to_x[id] = -1;
    gai1_grid_id_to_y[id] = 1;
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_L) | BIT_EN(PT_B);

    id = PT_BR;
    gai4_opt_grid_mask[id] = gai4_opt_grid_mask[PT_R] | gai4_opt_grid_mask[PT_B];
    gai1_grid_id_to_x[id] = 1;
    gai1_grid_id_to_y[id] = 1;
    gai4_opt_grid_mask_conventional[id] = BIT_EN(PT_R) | BIT_EN(PT_B);

    ge_part_id_to_blk_size[CU_8x8][PART_ID_2Nx2N] = BLK_8x8;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxN_T] = BLK_8x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxN_B] = BLK_8x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_Nx2N_L] = BLK_4x8;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_Nx2N_R] = BLK_4x8;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_NxN_TL] = BLK_4x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_NxN_TR] = BLK_4x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_NxN_BL] = BLK_4x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_NxN_BR] = BLK_4x4;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxnU_T] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxnU_B] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxnD_T] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_2NxnD_B] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_nLx2N_L] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_nLx2N_R] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_nRx2N_L] = BLK_INVALID;
    ge_part_id_to_blk_size[CU_8x8][PART_ID_nRx2N_R] = BLK_INVALID;

    ge_part_id_to_blk_size[CU_16x16][PART_ID_2Nx2N] = BLK_16x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxN_T] = BLK_16x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxN_B] = BLK_16x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_Nx2N_L] = BLK_8x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_Nx2N_R] = BLK_8x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_NxN_TL] = BLK_8x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_NxN_TR] = BLK_8x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_NxN_BL] = BLK_8x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_NxN_BR] = BLK_8x8;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxnU_T] = BLK_16x4;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxnU_B] = BLK_16x12;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxnD_T] = BLK_16x12;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_2NxnD_B] = BLK_16x4;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_nLx2N_L] = BLK_4x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_nLx2N_R] = BLK_12x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_nRx2N_L] = BLK_12x16;
    ge_part_id_to_blk_size[CU_16x16][PART_ID_nRx2N_R] = BLK_4x16;

    ge_part_id_to_blk_size[CU_32x32][PART_ID_2Nx2N] = BLK_32x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxN_T] = BLK_32x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxN_B] = BLK_32x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_Nx2N_L] = BLK_16x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_Nx2N_R] = BLK_16x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_NxN_TL] = BLK_16x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_NxN_TR] = BLK_16x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_NxN_BL] = BLK_16x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_NxN_BR] = BLK_16x16;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxnU_T] = BLK_32x8;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxnU_B] = BLK_32x24;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxnD_T] = BLK_32x24;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_2NxnD_B] = BLK_32x8;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_nLx2N_L] = BLK_8x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_nLx2N_R] = BLK_24x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_nRx2N_L] = BLK_24x32;
    ge_part_id_to_blk_size[CU_32x32][PART_ID_nRx2N_R] = BLK_8x32;

    ge_part_id_to_blk_size[CU_64x64][PART_ID_2Nx2N] = BLK_64x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxN_T] = BLK_64x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxN_B] = BLK_64x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_Nx2N_L] = BLK_32x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_Nx2N_R] = BLK_32x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_NxN_TL] = BLK_32x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_NxN_TR] = BLK_32x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_NxN_BL] = BLK_32x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_NxN_BR] = BLK_32x32;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxnU_T] = BLK_64x16;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxnU_B] = BLK_64x48;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxnD_T] = BLK_64x48;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_2NxnD_B] = BLK_64x16;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_nLx2N_L] = BLK_16x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_nLx2N_R] = BLK_48x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_nRx2N_L] = BLK_48x64;
    ge_part_id_to_blk_size[CU_64x64][PART_ID_nRx2N_R] = BLK_16x64;

    gau1_num_parts_in_part_type[PRT_2Nx2N] = 1;
    gau1_num_parts_in_part_type[PRT_2NxN] = 2;
    gau1_num_parts_in_part_type[PRT_Nx2N] = 2;
    gau1_num_parts_in_part_type[PRT_NxN] = 4;
    gau1_num_parts_in_part_type[PRT_2NxnU] = 2;
    gau1_num_parts_in_part_type[PRT_2NxnD] = 2;
    gau1_num_parts_in_part_type[PRT_nLx2N] = 2;
    gau1_num_parts_in_part_type[PRT_nRx2N] = 2;

    for(i = 0; i < MAX_PART_TYPES; i++)
        for(j = 0; j < MAX_NUM_PARTS; j++)
            ge_part_type_to_part_id[i][j] = PART_ID_INVALID;

    /* 2Nx2N only one partition */
    ge_part_type_to_part_id[PRT_2Nx2N][0] = PART_ID_2Nx2N;

    /* 2NxN 2 partitions */
    ge_part_type_to_part_id[PRT_2NxN][0] = PART_ID_2NxN_T;
    ge_part_type_to_part_id[PRT_2NxN][1] = PART_ID_2NxN_B;

    /* Nx2N 2 partitions */
    ge_part_type_to_part_id[PRT_Nx2N][0] = PART_ID_Nx2N_L;
    ge_part_type_to_part_id[PRT_Nx2N][1] = PART_ID_Nx2N_R;

    /* NxN 4 partitions */
    ge_part_type_to_part_id[PRT_NxN][0] = PART_ID_NxN_TL;
    ge_part_type_to_part_id[PRT_NxN][1] = PART_ID_NxN_TR;
    ge_part_type_to_part_id[PRT_NxN][2] = PART_ID_NxN_BL;
    ge_part_type_to_part_id[PRT_NxN][3] = PART_ID_NxN_BR;

    /* AMP 2Nx (N/2 + 3N/2) 2 partitions */
    ge_part_type_to_part_id[PRT_2NxnU][0] = PART_ID_2NxnU_T;
    ge_part_type_to_part_id[PRT_2NxnU][1] = PART_ID_2NxnU_B;

    /* AMP 2Nx (3N/2 + N/2) 2 partitions */
    ge_part_type_to_part_id[PRT_2NxnD][0] = PART_ID_2NxnD_T;
    ge_part_type_to_part_id[PRT_2NxnD][1] = PART_ID_2NxnD_B;

    /* AMP (N/2 + 3N/2) x 2N 2 partitions */
    ge_part_type_to_part_id[PRT_nLx2N][0] = PART_ID_nLx2N_L;
    ge_part_type_to_part_id[PRT_nLx2N][1] = PART_ID_nLx2N_R;

    /* AMP (3N/2 + N/2) x 2N 2 partitions */
    ge_part_type_to_part_id[PRT_nRx2N][0] = PART_ID_nRx2N_L;
    ge_part_type_to_part_id[PRT_nRx2N][1] = PART_ID_nRx2N_R;

    /*************************************************************************/
    /* initialize attributes for each partition id within the cu.            */
    /*************************************************************************/
    {
        part_attr_t *ps_part_attr;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2Nx2N];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxN_T];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxN_B];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 4;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_Nx2N_L];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_Nx2N_R];
        ps_part_attr->u1_x_start = 4;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_NxN_TL];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_NxN_TR];
        ps_part_attr->u1_x_start = 4;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_NxN_BL];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 4;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_NxN_BR];
        ps_part_attr->u1_x_start = 4;
        ps_part_attr->u1_y_start = 4;
        ps_part_attr->u1_x_count = 4;
        ps_part_attr->u1_y_count = 4;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxnU_T];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 2;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxnU_B];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 2;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 6;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxnD_T];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 6;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_2NxnD_B];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 6;
        ps_part_attr->u1_x_count = 8;
        ps_part_attr->u1_y_count = 2;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_nLx2N_L];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 2;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_nLx2N_R];
        ps_part_attr->u1_x_start = 2;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 6;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_nRx2N_L];
        ps_part_attr->u1_x_start = 0;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 6;
        ps_part_attr->u1_y_count = 8;

        ps_part_attr = &gas_part_attr_in_cu[PART_ID_nRx2N_R];
        ps_part_attr->u1_x_start = 6;
        ps_part_attr->u1_y_start = 0;
        ps_part_attr->u1_x_count = 2;
        ps_part_attr->u1_y_count = 8;
    }
    for(i = 0; i < NUM_BLK_SIZES; i++)
        ge_blk_size_to_cu_size[i] = CU_INVALID;

    ge_blk_size_to_cu_size[BLK_8x8] = CU_8x8;
    ge_blk_size_to_cu_size[BLK_16x16] = CU_16x16;
    ge_blk_size_to_cu_size[BLK_32x32] = CU_32x32;
    ge_blk_size_to_cu_size[BLK_64x64] = CU_64x64;

    /* This is the reverse, given cU size, get blk size */
    ge_cu_size_to_blk_size[CU_8x8] = BLK_8x8;
    ge_cu_size_to_blk_size[CU_16x16] = BLK_16x16;
    ge_cu_size_to_blk_size[CU_32x32] = BLK_32x32;
    ge_cu_size_to_blk_size[CU_64x64] = BLK_64x64;

    gau1_is_vert_part[PRT_2Nx2N] = 0;
    gau1_is_vert_part[PRT_2NxN] = 0;
    gau1_is_vert_part[PRT_Nx2N] = 1;
    gau1_is_vert_part[PRT_NxN] = 1;
    gau1_is_vert_part[PRT_2NxnU] = 0;
    gau1_is_vert_part[PRT_2NxnD] = 0;
    gau1_is_vert_part[PRT_nLx2N] = 1;
    gau1_is_vert_part[PRT_nRx2N] = 1;

    /* Initialise the number of best results for the full pell refinement */
    gau1_num_best_results_PQ[PART_ID_2Nx2N] = 2;
    gau1_num_best_results_PQ[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_PQ[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_PQ[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_PQ[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_PQ[PART_ID_NxN_TL] = 1;
    gau1_num_best_results_PQ[PART_ID_NxN_TR] = 1;
    gau1_num_best_results_PQ[PART_ID_NxN_BL] = 1;
    gau1_num_best_results_PQ[PART_ID_NxN_BR] = 1;
    gau1_num_best_results_PQ[PART_ID_2NxnU_T] = 1;
    gau1_num_best_results_PQ[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_PQ[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_PQ[PART_ID_2NxnD_B] = 1;
    gau1_num_best_results_PQ[PART_ID_nLx2N_L] = 1;
    gau1_num_best_results_PQ[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_PQ[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_PQ[PART_ID_nRx2N_R] = 1;

    gau1_num_best_results_HQ[PART_ID_2Nx2N] = 2;
    gau1_num_best_results_HQ[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_HQ[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_HQ[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_HQ[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_HQ[PART_ID_NxN_TL] = 1;
    gau1_num_best_results_HQ[PART_ID_NxN_TR] = 1;
    gau1_num_best_results_HQ[PART_ID_NxN_BL] = 1;
    gau1_num_best_results_HQ[PART_ID_NxN_BR] = 1;
    gau1_num_best_results_HQ[PART_ID_2NxnU_T] = 1;
    gau1_num_best_results_HQ[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_HQ[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_HQ[PART_ID_2NxnD_B] = 1;
    gau1_num_best_results_HQ[PART_ID_nLx2N_L] = 1;
    gau1_num_best_results_HQ[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_HQ[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_HQ[PART_ID_nRx2N_R] = 1;

    gau1_num_best_results_MS[PART_ID_2Nx2N] = 2;
    gau1_num_best_results_MS[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_MS[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_MS[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_MS[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_MS[PART_ID_NxN_TL] = 1;
    gau1_num_best_results_MS[PART_ID_NxN_TR] = 1;
    gau1_num_best_results_MS[PART_ID_NxN_BL] = 1;
    gau1_num_best_results_MS[PART_ID_NxN_BR] = 1;
    gau1_num_best_results_MS[PART_ID_2NxnU_T] = 1;
    gau1_num_best_results_MS[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_MS[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_MS[PART_ID_2NxnD_B] = 1;
    gau1_num_best_results_MS[PART_ID_nLx2N_L] = 1;
    gau1_num_best_results_MS[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_MS[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_MS[PART_ID_nRx2N_R] = 1;

    gau1_num_best_results_HS[PART_ID_2Nx2N] = 2;
    gau1_num_best_results_HS[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_HS[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_HS[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_HS[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_HS[PART_ID_NxN_TL] = 0;
    gau1_num_best_results_HS[PART_ID_NxN_TR] = 0;
    gau1_num_best_results_HS[PART_ID_NxN_BL] = 0;
    gau1_num_best_results_HS[PART_ID_NxN_BR] = 0;
    gau1_num_best_results_HS[PART_ID_2NxnU_T] = 0;
    gau1_num_best_results_HS[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_HS[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_HS[PART_ID_2NxnD_B] = 0;
    gau1_num_best_results_HS[PART_ID_nLx2N_L] = 0;
    gau1_num_best_results_HS[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_HS[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_HS[PART_ID_nRx2N_R] = 0;

    gau1_num_best_results_XS[PART_ID_2Nx2N] = 2;
    gau1_num_best_results_XS[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_XS[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_XS[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_XS[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_XS[PART_ID_NxN_TL] = 0;
    gau1_num_best_results_XS[PART_ID_NxN_TR] = 0;
    gau1_num_best_results_XS[PART_ID_NxN_BL] = 0;
    gau1_num_best_results_XS[PART_ID_NxN_BR] = 0;
    gau1_num_best_results_XS[PART_ID_2NxnU_T] = 0;
    gau1_num_best_results_XS[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_XS[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_XS[PART_ID_2NxnD_B] = 0;
    gau1_num_best_results_XS[PART_ID_nLx2N_L] = 0;
    gau1_num_best_results_XS[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_XS[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_XS[PART_ID_nRx2N_R] = 0;

    gau1_num_best_results_XS25[PART_ID_2Nx2N] = MAX_NUM_CANDS_FOR_FPEL_REFINE_IN_XS25;
    gau1_num_best_results_XS25[PART_ID_2NxN_T] = 0;
    gau1_num_best_results_XS25[PART_ID_2NxN_B] = 0;
    gau1_num_best_results_XS25[PART_ID_Nx2N_L] = 0;
    gau1_num_best_results_XS25[PART_ID_Nx2N_R] = 0;
    gau1_num_best_results_XS25[PART_ID_NxN_TL] = 0;
    gau1_num_best_results_XS25[PART_ID_NxN_TR] = 0;
    gau1_num_best_results_XS25[PART_ID_NxN_BL] = 0;
    gau1_num_best_results_XS25[PART_ID_NxN_BR] = 0;
    gau1_num_best_results_XS25[PART_ID_2NxnU_T] = 0;
    gau1_num_best_results_XS25[PART_ID_2NxnU_B] = 0;
    gau1_num_best_results_XS25[PART_ID_2NxnD_T] = 0;
    gau1_num_best_results_XS25[PART_ID_2NxnD_B] = 0;
    gau1_num_best_results_XS25[PART_ID_nLx2N_L] = 0;
    gau1_num_best_results_XS25[PART_ID_nLx2N_R] = 0;
    gau1_num_best_results_XS25[PART_ID_nRx2N_L] = 0;
    gau1_num_best_results_XS25[PART_ID_nRx2N_R] = 0;

    /* Top right validity for each part id */
    gau1_partid_tr_valid[PART_ID_2Nx2N] = 1;
    gau1_partid_tr_valid[PART_ID_2NxN_T] = 1;
    gau1_partid_tr_valid[PART_ID_2NxN_B] = 0;
    gau1_partid_tr_valid[PART_ID_Nx2N_L] = 1;
    gau1_partid_tr_valid[PART_ID_Nx2N_R] = 1;
    gau1_partid_tr_valid[PART_ID_NxN_TL] = 1;
    gau1_partid_tr_valid[PART_ID_NxN_TR] = 1;
    gau1_partid_tr_valid[PART_ID_NxN_BL] = 1;
    gau1_partid_tr_valid[PART_ID_NxN_BR] = 0;
    gau1_partid_tr_valid[PART_ID_2NxnU_T] = 1;
    gau1_partid_tr_valid[PART_ID_2NxnU_B] = 0;
    gau1_partid_tr_valid[PART_ID_2NxnD_T] = 1;
    gau1_partid_tr_valid[PART_ID_2NxnD_B] = 0;
    gau1_partid_tr_valid[PART_ID_nLx2N_L] = 1;
    gau1_partid_tr_valid[PART_ID_nLx2N_R] = 1;
    gau1_partid_tr_valid[PART_ID_nRx2N_L] = 1;
    gau1_partid_tr_valid[PART_ID_nRx2N_R] = 1;

    /* Bot Left validity for each part id */
    gau1_partid_bl_valid[PART_ID_2Nx2N] = 1;
    gau1_partid_bl_valid[PART_ID_2NxN_T] = 1;
    gau1_partid_bl_valid[PART_ID_2NxN_B] = 1;
    gau1_partid_bl_valid[PART_ID_Nx2N_L] = 1;
    gau1_partid_bl_valid[PART_ID_Nx2N_R] = 0;
    gau1_partid_bl_valid[PART_ID_NxN_TL] = 1;
    gau1_partid_bl_valid[PART_ID_NxN_TR] = 0;
    gau1_partid_bl_valid[PART_ID_NxN_BL] = 1;
    gau1_partid_bl_valid[PART_ID_NxN_BR] = 0;
    gau1_partid_bl_valid[PART_ID_2NxnU_T] = 1;
    gau1_partid_bl_valid[PART_ID_2NxnU_B] = 1;
    gau1_partid_bl_valid[PART_ID_2NxnD_T] = 1;
    gau1_partid_bl_valid[PART_ID_2NxnD_B] = 1;
    gau1_partid_bl_valid[PART_ID_nLx2N_L] = 1;
    gau1_partid_bl_valid[PART_ID_nLx2N_R] = 0;
    gau1_partid_bl_valid[PART_ID_nRx2N_L] = 1;
    gau1_partid_bl_valid[PART_ID_nRx2N_R] = 0;

    /*Part id to part num of this partition id in the CU */
    gau1_part_id_to_part_num[PART_ID_2Nx2N] = 0;
    gau1_part_id_to_part_num[PART_ID_2NxN_T] = 0;
    gau1_part_id_to_part_num[PART_ID_2NxN_B] = 1;
    gau1_part_id_to_part_num[PART_ID_Nx2N_L] = 0;
    gau1_part_id_to_part_num[PART_ID_Nx2N_R] = 1;
    gau1_part_id_to_part_num[PART_ID_NxN_TL] = 0;
    gau1_part_id_to_part_num[PART_ID_NxN_TR] = 1;
    gau1_part_id_to_part_num[PART_ID_NxN_BL] = 2;
    gau1_part_id_to_part_num[PART_ID_NxN_BR] = 3;
    gau1_part_id_to_part_num[PART_ID_2NxnU_T] = 0;
    gau1_part_id_to_part_num[PART_ID_2NxnU_B] = 1;
    gau1_part_id_to_part_num[PART_ID_2NxnD_T] = 0;
    gau1_part_id_to_part_num[PART_ID_2NxnD_B] = 1;
    gau1_part_id_to_part_num[PART_ID_nLx2N_L] = 0;
    gau1_part_id_to_part_num[PART_ID_nLx2N_R] = 1;
    gau1_part_id_to_part_num[PART_ID_nRx2N_L] = 0;
    gau1_part_id_to_part_num[PART_ID_nRx2N_R] = 1;

    /*Which partition type does this partition id belong to */
    ge_part_id_to_part_type[PART_ID_2Nx2N] = PRT_2Nx2N;
    ge_part_id_to_part_type[PART_ID_2NxN_T] = PRT_2NxN;
    ge_part_id_to_part_type[PART_ID_2NxN_B] = PRT_2NxN;
    ge_part_id_to_part_type[PART_ID_Nx2N_L] = PRT_Nx2N;
    ge_part_id_to_part_type[PART_ID_Nx2N_R] = PRT_Nx2N;
    ge_part_id_to_part_type[PART_ID_NxN_TL] = PRT_NxN;
    ge_part_id_to_part_type[PART_ID_NxN_TR] = PRT_NxN;
    ge_part_id_to_part_type[PART_ID_NxN_BL] = PRT_NxN;
    ge_part_id_to_part_type[PART_ID_NxN_BR] = PRT_NxN;
    ge_part_id_to_part_type[PART_ID_2NxnU_T] = PRT_2NxnU;
    ge_part_id_to_part_type[PART_ID_2NxnU_B] = PRT_2NxnU;
    ge_part_id_to_part_type[PART_ID_2NxnD_T] = PRT_2NxnD;
    ge_part_id_to_part_type[PART_ID_2NxnD_B] = PRT_2NxnD;
    ge_part_id_to_part_type[PART_ID_nLx2N_L] = PRT_nLx2N;
    ge_part_id_to_part_type[PART_ID_nLx2N_R] = PRT_nLx2N;
    ge_part_id_to_part_type[PART_ID_nRx2N_L] = PRT_nRx2N;
    ge_part_id_to_part_type[PART_ID_nRx2N_R] = PRT_nRx2N;

    /*************************************************************************/
    /* Set up the bits to be taken up for the part type. This is equally     */
    /* divided up between the various partitions in the part-type.           */
    /* For NxN @ CU 16x16, we assume it as CU 8x8, so consider it as         */
    /* partition 2Nx2N.                                                      */
    /*************************************************************************/
    /* 1 bit for 2Nx2N partition */
    gau1_bits_for_part_id_q1[PART_ID_2Nx2N] = 2;

    /* 3 bits for symmetric part types, so 1.5 bits per partition */
    gau1_bits_for_part_id_q1[PART_ID_2NxN_T] = 3;
    gau1_bits_for_part_id_q1[PART_ID_2NxN_B] = 3;
    gau1_bits_for_part_id_q1[PART_ID_Nx2N_L] = 3;
    gau1_bits_for_part_id_q1[PART_ID_Nx2N_R] = 3;

    /* 1 bit for NxN partitions, assuming these to be 2Nx2N CUs of lower level */
    gau1_bits_for_part_id_q1[PART_ID_NxN_TL] = 2;
    gau1_bits_for_part_id_q1[PART_ID_NxN_TR] = 2;
    gau1_bits_for_part_id_q1[PART_ID_NxN_BL] = 2;
    gau1_bits_for_part_id_q1[PART_ID_NxN_BR] = 2;

    /* 4 bits for AMP so 2 bits per partition */
    gau1_bits_for_part_id_q1[PART_ID_2NxnU_T] = 4;
    gau1_bits_for_part_id_q1[PART_ID_2NxnU_B] = 4;
    gau1_bits_for_part_id_q1[PART_ID_2NxnD_T] = 4;
    gau1_bits_for_part_id_q1[PART_ID_2NxnD_B] = 4;
    gau1_bits_for_part_id_q1[PART_ID_nLx2N_L] = 4;
    gau1_bits_for_part_id_q1[PART_ID_nLx2N_R] = 4;
    gau1_bits_for_part_id_q1[PART_ID_nRx2N_L] = 4;
    gau1_bits_for_part_id_q1[PART_ID_nRx2N_R] = 4;
}

/**
********************************************************************************
*  @fn     hme_enc_num_alloc()
*
*  @brief  returns number of memtabs that is required by hme module
*
*  @return   Number of memtabs required
********************************************************************************
*/
S32 hme_enc_num_alloc(WORD32 i4_num_me_frm_pllel)
{
    if(i4_num_me_frm_pllel > 1)
    {
        return ((S32)MAX_HME_ENC_TOT_MEMTABS);
    }
    else
    {
        return ((S32)MIN_HME_ENC_TOT_MEMTABS);
    }
}

/**
********************************************************************************
*  @fn     hme_coarse_num_alloc()
*
*  @brief  returns number of memtabs that is required by hme module
*
*  @return   Number of memtabs required
********************************************************************************
*/
S32 hme_coarse_num_alloc()
{
    return ((S32)HME_COARSE_TOT_MEMTABS);
}

/**
********************************************************************************
*  @fn     hme_coarse_dep_mngr_num_alloc()
*
*  @brief  returns number of memtabs that is required by Dep Mngr for hme module
*
*  @return   Number of memtabs required
********************************************************************************
*/
WORD32 hme_coarse_dep_mngr_num_alloc()
{
    return ((WORD32)((MAX_NUM_HME_LAYERS - 1) * ihevce_dmgr_get_num_mem_recs()));
}

S32 hme_validate_init_prms(hme_init_prms_t *ps_prms)
{
    S32 n_layers = ps_prms->num_simulcast_layers;

    /* The final layer has got to be a non encode coarse layer */
    if(n_layers > (MAX_NUM_LAYERS - 1))
        return (-1);

    if(n_layers < 1)
        return (-1);

    /* Width of the coarsest encode layer got to be >= 2*min_wd where min_Wd */
    /* represents the min allowed width in any layer. Ditto with ht          */
    if(ps_prms->a_wd[n_layers - 1] < 2 * (MIN_WD_COARSE))
        return (-1);
    if(ps_prms->a_ht[n_layers - 1] < 2 * (MIN_HT_COARSE))
        return (-1);
    if(ps_prms->max_num_ref > MAX_NUM_REF)
        return (-1);
    if(ps_prms->max_num_ref < 0)
        return (-1);

    return (0);
}
void hme_set_layer_res_attrs(
    layer_ctxt_t *ps_layer, S32 wd, S32 ht, S32 disp_wd, S32 disp_ht, U08 u1_enc)
{
    ps_layer->i4_wd = wd;
    ps_layer->i4_ht = ht;
    ps_layer->i4_disp_wd = disp_wd;
    ps_layer->i4_disp_ht = disp_ht;
    if(0 == u1_enc)
    {
        ps_layer->i4_inp_stride = wd + 32 + 4;
        ps_layer->i4_inp_offset = (ps_layer->i4_inp_stride * 16) + 16;
        ps_layer->i4_pad_x_inp = 16;
        ps_layer->i4_pad_y_inp = 16;
        ps_layer->pu1_inp = ps_layer->pu1_inp_base + ps_layer->i4_inp_offset;
    }
}

/**
********************************************************************************
*  @fn     hme_coarse_get_layer1_mv_bank_ref_idx_size()
*
*  @brief  returns the MV bank and ref idx size of Layer 1 (penultimate)
*
*  @return   none
********************************************************************************
*/
void hme_coarse_get_layer1_mv_bank_ref_idx_size(
    S32 n_tot_layers,
    S32 *a_wd,
    S32 *a_ht,
    S32 max_num_ref,
    S32 *pi4_mv_bank_size,
    S32 *pi4_ref_idx_size)
{
    S32 num_blks, num_mvs_per_blk, num_ref;
    S32 num_cols, num_rows, num_mvs_per_row;
    S32 is_explicit_store = 1;
    S32 wd, ht, num_layers_explicit_search;
    S32 num_results, use_4x4;
    wd = a_wd[1];
    ht = a_ht[1];

    /* Assuming abt 4 layers for 1080p, we do explicit search across all ref */
    /* frames in all but final layer In final layer, it could be 1/2 */
    //ps_hme_init_prms->num_layers_explicit_search = 3;
    num_layers_explicit_search = 3;

    if(num_layers_explicit_search <= 0)
        num_layers_explicit_search = n_tot_layers - 1;

    num_layers_explicit_search = MIN(num_layers_explicit_search, n_tot_layers - 1);

    /* Possibly implicit search for lower (finer) layers */
    if(n_tot_layers - 1 > num_layers_explicit_search)
        is_explicit_store = 0;

    /* coarsest layer alwasy uses 4x4 blks to store results */
    if(1 == (n_tot_layers - 1))
    {
        /* we store 4 results in coarsest layer per blk. 8x4L, 8x4R, 4x8T, 4x8B */
        //ps_hme_init_prms->max_num_results_coarse = 4;
        //vijay : with new algo in coarseset layer this has to be revisited
        num_results = 4;
    }
    else
    {
        /* Every refinement layer stores a max of 2 results per partition */
        //ps_hme_init_prms->max_num_results = 2;
        num_results = 2;
    }
    use_4x4 = hme_get_mv_blk_size(1, 1, n_tot_layers, 0);

    num_cols = use_4x4 ? ((wd >> 2) + 2) : ((wd >> 3) + 2);
    num_rows = use_4x4 ? ((ht >> 2) + 2) : ((ht >> 3) + 2);

    if(is_explicit_store)
        num_ref = max_num_ref;
    else
        num_ref = 2;

    num_blks = num_cols * num_rows;
    num_mvs_per_blk = num_ref * num_results;
    num_mvs_per_row = num_mvs_per_blk * num_cols;

    /* stroe the sizes */
    *pi4_mv_bank_size = num_blks * num_mvs_per_blk * sizeof(hme_mv_t);
    *pi4_ref_idx_size = num_blks * num_mvs_per_blk * sizeof(S08);

    return;
}
/**
********************************************************************************
*  @fn     hme_alloc_init_layer_mv_bank()
*
*  @brief  memory alloc and init function for MV bank
*
*  @return   Number of memtabs required
********************************************************************************
*/
S32 hme_alloc_init_layer_mv_bank(
    hme_memtab_t *ps_memtab,
    S32 max_num_results,
    S32 max_num_ref,
    S32 use_4x4,
    S32 mem_avail,
    S32 u1_enc,
    S32 wd,
    S32 ht,
    S32 is_explicit_store,
    hme_mv_t **pps_mv_base,
    S08 **pi1_ref_idx_base,
    S32 *pi4_num_mvs_per_row)
{
    S32 count = 0;
    S32 size;
    S32 num_blks, num_mvs_per_blk;
    S32 num_ref;
    S32 num_cols, num_rows, num_mvs_per_row;

    if(is_explicit_store)
        num_ref = max_num_ref;
    else
        num_ref = 2;

    /* MV Bank allocation takes into consideration following */
    /* number of results per reference x max num refrences is the amount     */
    /* bufffered up per blk. Numbero f blks in pic deps on the blk size,     */
    /* which could be either 4x4 or 8x8.                                     */
    num_cols = use_4x4 ? ((wd >> 2) + 2) : ((wd >> 3) + 2);
    num_rows = use_4x4 ? ((ht >> 2) + 2) : ((ht >> 3) + 2);

    if(u1_enc)
    {
        /* TODO: CTB64x64 is assumed. FIX according to actual CTB */
        WORD32 num_ctb_cols = ((wd + 63) >> 6);
        WORD32 num_ctb_rows = ((ht + 63) >> 6);

        num_cols = (num_ctb_cols << 3) + 2;
        num_rows = (num_ctb_rows << 3) + 2;
    }
    num_blks = num_cols * num_rows;
    num_mvs_per_blk = num_ref * max_num_results;
    num_mvs_per_row = num_mvs_per_blk * num_cols;

    size = num_blks * num_mvs_per_blk * sizeof(hme_mv_t);
    if(mem_avail)
    {
        /* store this for run time verifications */
        *pi4_num_mvs_per_row = num_mvs_per_row;
        ASSERT(ps_memtab[count].size == size);
        *pps_mv_base = (hme_mv_t *)ps_memtab[count].pu1_mem;
    }
    else
    {
        ps_memtab[count].size = size;
        ps_memtab[count].align = 4;
        ps_memtab[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;
    /* Ref idx takes the same route as mvbase */

    size = num_blks * num_mvs_per_blk * sizeof(S08);
    if(mem_avail)
    {
        ASSERT(ps_memtab[count].size == size);
        *pi1_ref_idx_base = (S08 *)ps_memtab[count].pu1_mem;
    }
    else
    {
        ps_memtab[count].size = size;
        ps_memtab[count].align = 4;
        ps_memtab[count].e_mem_attr = HME_PERSISTENT_MEM;
    }
    count++;

    return (count);
}
/**
********************************************************************************
*  @fn     hme_alloc_init_layer()
*
*  @brief  memory alloc and init function
*
*  @return   Number of memtabs required
********************************************************************************
*/
S32 hme_alloc_init_layer(
    hme_memtab_t *ps_memtab,
    S32 max_num_results,
    S32 max_num_ref,
    S32 use_4x4,
    S32 mem_avail,
    S32 u1_enc,
    S32 wd,
    S32 ht,
    S32 disp_wd,
    S32 disp_ht,
    S32 segment_layer,
    S32 is_explicit_store,
    layer_ctxt_t **pps_layer)
{
    S32 count = 0;
    layer_ctxt_t *ps_layer = NULL;
    S32 size;
    S32 num_ref;

    ARG_NOT_USED(segment_layer);

    if(is_explicit_store)
        num_ref = max_num_ref;
    else
        num_ref = 2;

    /* We do not store 4x4 results for encoding layers */
    if(u1_enc)
        use_4x4 = 0;

    size = sizeof(layer_ctxt_t);
    if(mem_avail)
    {
        ASSERT(ps_memtab[count].size == size);
        ps_layer = (layer_ctxt_t *)ps_memtab[count].pu1_mem;
        *pps_layer = ps_layer;
    }
    else
    {
        ps_memtab[count].size = size;
        ps_memtab[count].align = 8;
        ps_memtab[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    /* Input luma buffer allocated only for non encode case */
    if(0 == u1_enc)
    {
        /* Allocate input with padding of 16 pixels */
        size = (wd + 32 + 4) * (ht + 32 + 4);
        if(mem_avail)
        {
            ASSERT(ps_memtab[count].size == size);
            ps_layer->pu1_inp_base = ps_memtab[count].pu1_mem;
        }
        else
        {
            ps_memtab[count].size = size;
            ps_memtab[count].align = 16;
            ps_memtab[count].e_mem_attr = HME_PERSISTENT_MEM;
        }
        count++;
    }

    /* Allocate memory or just the layer mvbank strcture. */
    /* TODO : see if this can be removed by moving it to layer_ctxt */
    size = sizeof(layer_mv_t);

    if(mem_avail)
    {
        ASSERT(ps_memtab[count].size == size);
        ps_layer->ps_layer_mvbank = (layer_mv_t *)ps_memtab[count].pu1_mem;
    }
    else
    {
        ps_memtab[count].size = size;
        ps_memtab[count].align = 8;
        ps_memtab[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    if(mem_avail)
    {
        hme_set_layer_res_attrs(ps_layer, wd, ht, disp_wd, disp_ht, u1_enc);
    }

    return (count);
}

S32 hme_alloc_init_search_nodes(
    search_results_t *ps_search_results,
    hme_memtab_t *ps_memtabs,
    S32 mem_avail,
    S32 max_num_ref,
    S32 max_num_results)
{
    S32 size = max_num_results * sizeof(search_node_t) * max_num_ref * TOT_NUM_PARTS;
    S32 j, k;
    search_node_t *ps_search_node;

    if(mem_avail == 0)
    {
        ps_memtabs->size = size;
        ps_memtabs->align = 4;
        ps_memtabs->e_mem_attr = HME_SCRATCH_OVLY_MEM;
        return (1);
    }

    ps_search_node = (search_node_t *)ps_memtabs->pu1_mem;
    ASSERT(ps_memtabs->size == size);
    /****************************************************************************/
    /* For each CU, we search and store N best results, per partition, per ref  */
    /* So, number of memtabs is  num_refs * num_parts                           */
    /****************************************************************************/
    for(j = 0; j < max_num_ref; j++)
    {
        for(k = 0; k < TOT_NUM_PARTS; k++)
        {
            ps_search_results->aps_part_results[j][k] = ps_search_node;
            ps_search_node += max_num_results;
        }
    }
    return (1);
}

S32 hme_derive_num_layers(S32 n_enc_layers, S32 *p_wd, S32 *p_ht, S32 *p_disp_wd, S32 *p_disp_ht)
{
    S32 i;
    /* We keep downscaling by 2 till we hit one of the conditions:           */
    /* 1. MAX_NUM_LAYERS reached.                                            */
    /* 2. Width or ht goes below min width and ht allowed at coarsest layer  */
    ASSERT(n_enc_layers < MAX_NUM_LAYERS);
    ASSERT(n_enc_layers > 0);
    ASSERT(p_wd[0] <= HME_MAX_WIDTH);
    ASSERT(p_ht[0] <= HME_MAX_HEIGHT);

    p_disp_wd[0] = p_wd[0];
    p_disp_ht[0] = p_ht[0];
    /*************************************************************************/
    /* Verify that for simulcast, lower layer to higher layer ratio is bet   */
    /* 2 (dyadic) and 1.33. Typically it should be 1.5.                      */
    /* TODO : for interlace, we may choose to have additional downscaling for*/
    /* width alone in coarsest layer to next layer.                          */
    /*************************************************************************/
    for(i = 1; i < n_enc_layers; i++)
    {
        S32 wd1, wd2, ht1, ht2;
        wd1 = FLOOR16(p_wd[i - 1] >> 1);
        wd2 = CEIL16((p_wd[i - 1] * 3) >> 2);
        ASSERT(p_wd[i] >= wd1);
        ASSERT(p_wd[i] <= wd2);
        ht1 = FLOOR16(p_ht[i - 1] >> 1);
        ht2 = CEIL16((p_ht[i - 1] * 3) >> 2);
        ASSERT(p_ht[i] >= ht1);
        ASSERT(p_ht[i] <= ht2);
    }
    ASSERT(p_wd[n_enc_layers - 1] >= 2 * MIN_WD_COARSE);
    ASSERT(p_ht[n_enc_layers - 1] >= 2 * MIN_HT_COARSE);

    for(i = n_enc_layers; i < MAX_NUM_LAYERS; i++)
    {
        if((p_wd[i - 1] < 2 * MIN_WD_COARSE) || (p_ht[i - 1] < 2 * MIN_HT_COARSE))
        {
            return (i);
        }
        /* Use CEIL16 to facilitate 16x16 searches in future, or to do       */
        /* segmentation study in future                                      */
        p_wd[i] = CEIL16(p_wd[i - 1] >> 1);
        p_ht[i] = CEIL16(p_ht[i - 1] >> 1);

        p_disp_wd[i] = p_disp_wd[i - 1] >> 1;
        p_disp_ht[i] = p_disp_ht[i - 1] >> 1;
    }
    return (i);
}

/**
********************************************************************************
*  @fn     hme_get_mv_blk_size()
*
*  @brief  returns whether blk uses 4x4 size or something else.
*
*  @param[in] enable_4x4 : input param from application to enable 4x4
*
*  @param[in] layer_id : id of current layer (0 finest)
*
*  @param[in] num_layeers : total num layers
*
*  @param[in] is_enc : Whether encoding enabled for layer
*
*  @return   1 for 4x4 blks, 0 for 8x8
********************************************************************************
*/
S32 hme_get_mv_blk_size(S32 enable_4x4, S32 layer_id, S32 num_layers, S32 is_enc)
{
    S32 use_4x4 = enable_4x4;

    if((layer_id <= 1) && (num_layers >= 4))
        use_4x4 = USE_4x4_IN_L1;
    if(layer_id == num_layers - 1)
        use_4x4 = 1;
    if(is_enc)
        use_4x4 = 0;

    return (use_4x4);
}

/**
********************************************************************************
*  @fn     hme_enc_alloc_init_mem()
*
*  @brief  Requests/ assign memory based on mem avail
*
*  @param[in] ps_memtabs : memtab array
*
*  @param[in] ps_prms : init prms
*
*  @param[in] pv_ctxt : ME ctxt
*
*  @param[in] mem_avail : request/assign flag
*
*  @return   1 for 4x4 blks, 0 for 8x8
********************************************************************************
*/
S32 hme_enc_alloc_init_mem(
    hme_memtab_t *ps_memtabs,
    hme_init_prms_t *ps_prms,
    void *pv_ctxt,
    S32 mem_avail,
    S32 i4_num_me_frm_pllel)
{
    me_master_ctxt_t *ps_master_ctxt = (me_master_ctxt_t *)pv_ctxt;
    me_ctxt_t *ps_ctxt;
    S32 count = 0, size, i, j, use_4x4;
    S32 n_tot_layers, n_enc_layers;
    S32 num_layers_explicit_search;
    S32 a_wd[MAX_NUM_LAYERS], a_ht[MAX_NUM_LAYERS];
    S32 a_disp_wd[MAX_NUM_LAYERS], a_disp_ht[MAX_NUM_LAYERS];
    S32 num_results;
    S32 num_thrds;
    S32 ctb_wd = 1 << ps_prms->log_ctb_size;

    /* MV bank changes */
    hme_mv_t *aps_mv_bank[((DEFAULT_MAX_REFERENCE_PICS << 1) * MAX_NUM_ME_PARALLEL) + 1] = { NULL };
    S32 i4_num_mvs_per_row = 0;
    S08 *api1_ref_idx[((DEFAULT_MAX_REFERENCE_PICS << 1) * MAX_NUM_ME_PARALLEL) + 1] = { NULL };

    n_enc_layers = ps_prms->num_simulcast_layers;

    /* Memtab 0: handle */
    size = sizeof(me_master_ctxt_t);
    if(mem_avail)
    {
        /* store the number of processing threads */
        ps_master_ctxt->i4_num_proc_thrds = ps_prms->i4_num_proc_thrds;
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 8;
        ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    /* Memtab 1: ME threads ctxt */
    size = ps_prms->i4_num_proc_thrds * sizeof(me_ctxt_t);
    if(mem_avail)
    {
        me_ctxt_t *ps_me_tmp_ctxt = (me_ctxt_t *)ps_memtabs[count].pu1_mem;

        /* store the indivisual thread ctxt pointers */
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            ps_master_ctxt->aps_me_ctxt[num_thrds] = ps_me_tmp_ctxt++;
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 8;
        ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    /* Memtab 2: ME frame ctxts */
    size = sizeof(me_frm_ctxt_t) * MAX_NUM_ME_PARALLEL * ps_prms->i4_num_proc_thrds;
    if(mem_avail)
    {
        me_frm_ctxt_t *ps_me_frm_tmp_ctxt = (me_frm_ctxt_t *)ps_memtabs[count].pu1_mem;

        for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
        {
            /* store the indivisual thread ctxt pointers */
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_master_ctxt->aps_me_ctxt[num_thrds]->aps_me_frm_prms[i] = ps_me_frm_tmp_ctxt;

                ps_me_frm_tmp_ctxt++;
            }
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 8;
        ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    memcpy(a_wd, ps_prms->a_wd, sizeof(S32) * ps_prms->num_simulcast_layers);
    memcpy(a_ht, ps_prms->a_ht, sizeof(S32) * ps_prms->num_simulcast_layers);
    /*************************************************************************/
    /* Derive the number of HME layers, including both encoded and non encode*/
    /* This function also derives the width and ht of each layer.            */
    /*************************************************************************/
    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    num_layers_explicit_search = ps_prms->num_layers_explicit_search;
    if(num_layers_explicit_search <= 0)
        num_layers_explicit_search = n_tot_layers - 1;

    num_layers_explicit_search = MIN(num_layers_explicit_search, n_tot_layers - 1);

    if(mem_avail)
    {
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            me_frm_ctxt_t *ps_frm_ctxt;
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
            {
                ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[i];

                memset(ps_frm_ctxt->u1_encode, 0, n_tot_layers);
                memset(ps_frm_ctxt->u1_encode, 1, n_enc_layers);

                /* only one enocde layer is used */
                ps_frm_ctxt->num_layers = 1;

                ps_frm_ctxt->i4_wd = a_wd[0];
                ps_frm_ctxt->i4_ht = a_ht[0];
                /*
            memcpy(ps_ctxt->a_wd, a_wd, sizeof(S32)*n_tot_layers);
            memcpy(ps_ctxt->a_ht, a_ht, sizeof(S32)*n_tot_layers);
*/
                ps_frm_ctxt->num_layers_explicit_search = num_layers_explicit_search;
                ps_frm_ctxt->max_num_results = ps_prms->max_num_results;
                ps_frm_ctxt->max_num_results_coarse = ps_prms->max_num_results_coarse;
                ps_frm_ctxt->max_num_ref = ps_prms->max_num_ref;
            }
        }
    }

    /* Memtabs : Layers MV bank for encode layer */
    /* Each ref_desr in master ctxt will have seperate layer ctxt */

    for(i = 0; i < (ps_prms->max_num_ref * i4_num_me_frm_pllel) + 1; i++)
    {
        for(j = 0; j < 1; j++)
        {
            S32 is_explicit_store = 1;
            S32 wd, ht;
            U08 u1_enc = 1;
            wd = a_wd[j];
            ht = a_ht[j];

            /* Possibly implicit search for lower (finer) layers */
            if(n_tot_layers - j > num_layers_explicit_search)
                is_explicit_store = 0;

            /* Even if explicit search, we store only 2 results (L0 and L1) */
            /* in finest layer */
            if(j == 0)
            {
                is_explicit_store = 0;
            }

            /* coarsest layer alwasy uses 4x4 blks to store results */
            if(j == n_tot_layers - 1)
            {
                num_results = ps_prms->max_num_results_coarse;
            }
            else
            {
                num_results = ps_prms->max_num_results;
                if(j == 0)
                    num_results = 1;
            }
            use_4x4 = hme_get_mv_blk_size(ps_prms->use_4x4, j, n_tot_layers, u1_enc);

            count += hme_alloc_init_layer_mv_bank(
                &ps_memtabs[count],
                num_results,
                ps_prms->max_num_ref,
                use_4x4,
                mem_avail,
                u1_enc,
                wd,
                ht,
                is_explicit_store,
                &aps_mv_bank[i],
                &api1_ref_idx[i],
                &i4_num_mvs_per_row);
        }
    }

    /* Memtabs : Layers * num-ref + 1 */
    for(i = 0; i < (ps_prms->max_num_ref * i4_num_me_frm_pllel) + 1; i++)
    {
        /* layer memory allocated only for enocde layer */
        for(j = 0; j < 1; j++)
        {
            layer_ctxt_t *ps_layer;
            S32 is_explicit_store = 1;
            S32 segment_this_layer = (j == 0) ? 1 : ps_prms->segment_higher_layers;
            S32 wd, ht;
            U08 u1_enc = 1;
            wd = a_wd[j];
            ht = a_ht[j];

            /* Possibly implicit search for lower (finer) layers */
            if(n_tot_layers - j > num_layers_explicit_search)
                is_explicit_store = 0;

            /* Even if explicit search, we store only 2 results (L0 and L1) */
            /* in finest layer */
            if(j == 0)
            {
                is_explicit_store = 0;
            }

            /* coarsest layer alwasy uses 4x4 blks to store results */
            if(j == n_tot_layers - 1)
            {
                num_results = ps_prms->max_num_results_coarse;
            }
            else
            {
                num_results = ps_prms->max_num_results;
                if(j == 0)
                    num_results = 1;
            }
            use_4x4 = hme_get_mv_blk_size(ps_prms->use_4x4, j, n_tot_layers, u1_enc);

            count += hme_alloc_init_layer(
                &ps_memtabs[count],
                num_results,
                ps_prms->max_num_ref,
                use_4x4,
                mem_avail,
                u1_enc,
                wd,
                ht,
                a_disp_wd[j],
                a_disp_ht[j],
                segment_this_layer,
                is_explicit_store,
                &ps_layer);
            if(mem_avail)
            {
                /* same ps_layer memory pointer is stored in all the threads */
                for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
                {
                    ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                    ps_ctxt->as_ref_descr[i].aps_layers[j] = ps_layer;
                }

                /* store the MV bank pointers */
                ps_layer->ps_layer_mvbank->max_num_mvs_per_row = i4_num_mvs_per_row;
                ps_layer->ps_layer_mvbank->ps_mv_base = aps_mv_bank[i];
                ps_layer->ps_layer_mvbank->pi1_ref_idx_base = api1_ref_idx[i];
            }
        }
    }

    /* Memtabs : Buf Mgr for predictor bufs and working mem */
    /* TODO : Parameterise this appropriately */
    size = MAX_WKG_MEM_SIZE_PER_THREAD * ps_prms->i4_num_proc_thrds * i4_num_me_frm_pllel;

    if(mem_avail)
    {
        U08 *pu1_mem = ps_memtabs[count].pu1_mem;

        ASSERT(ps_memtabs[count].size == size);

        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            me_frm_ctxt_t *ps_frm_ctxt;
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
            {
                ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[i];

                hme_init_wkg_mem(&ps_frm_ctxt->s_buf_mgr, pu1_mem, MAX_WKG_MEM_SIZE_PER_THREAD);

                if(i4_num_me_frm_pllel != 1)
                {
                    /* update the memory buffer pointer */
                    pu1_mem += MAX_WKG_MEM_SIZE_PER_THREAD;
                }
            }
            if(i4_num_me_frm_pllel == 1)
            {
                pu1_mem += MAX_WKG_MEM_SIZE_PER_THREAD;
            }
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 4;
        ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
    }
    count++;

    /*************************************************************************/
    /* Memtab : We need 64x64 buffer to store the entire CTB input for bidir */
    /* refinement. This memtab stores 2I - P0, I is input and P0 is L0 pred  */
    /*************************************************************************/
    size = sizeof(S16) * CTB_BLK_SIZE * CTB_BLK_SIZE * ps_prms->i4_num_proc_thrds *
           i4_num_me_frm_pllel;

    if(mem_avail)
    {
        S16 *pi2_mem = (S16 *)ps_memtabs[count].pu1_mem;

        ASSERT(ps_memtabs[count].size == size);

        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            me_frm_ctxt_t *ps_frm_ctxt;
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
            {
                ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[i];

                ps_frm_ctxt->pi2_inp_bck = pi2_mem;
                /** If no me frames running in parallel update the other aps_me_frm_prms indices with same memory **/
                if(i4_num_me_frm_pllel != 1)
                {
                    pi2_mem += (CTB_BLK_SIZE * CTB_BLK_SIZE);
                }
            }
            if(i4_num_me_frm_pllel == 1)
            {
                pi2_mem += (CTB_BLK_SIZE * CTB_BLK_SIZE);
            }
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 16;
        ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
    }

    count++;

    /* Allocate a memtab for each histogram. As many as num ref and number of threads */
    /* Loop across for each ME_FRM in PARALLEL */
    for(j = 0; j < MAX_NUM_ME_PARALLEL; j++)
    {
        for(i = 0; i < ps_prms->max_num_ref; i++)
        {
            size = ps_prms->i4_num_proc_thrds * sizeof(mv_hist_t);
            if(mem_avail)
            {
                mv_hist_t *ps_mv_hist = (mv_hist_t *)ps_memtabs[count].pu1_mem;

                ASSERT(size == ps_memtabs[count].size);

                /* divide the memory accross the threads */
                for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
                {
                    ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

                    ps_ctxt->aps_me_frm_prms[j]->aps_mv_hist[i] = ps_mv_hist;
                    ps_mv_hist++;
                }
            }
            else
            {
                ps_memtabs[count].size = size;
                ps_memtabs[count].align = 8;
                ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
            }
            count++;
        }
        if((i4_num_me_frm_pllel == 1) && (j != (MAX_NUM_ME_PARALLEL - 1)))
        {
            /** If no me frames running in parallel update the other aps_me_frm_prms indices with same memory **/
            /** bring the count back to earlier value if there are no me frames in parallel. don't decrement for last loop **/
            count -= ps_prms->max_num_ref;
        }
    }

    /* Memtabs : Search nodes for 16x16 CUs, 32x32 and 64x64 CUs */
    for(j = 0; j < MAX_NUM_ME_PARALLEL; j++)
    {
        S32 count_cpy = count;
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            if(mem_avail)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
            }

            for(i = 0; i < 21; i++)
            {
                search_results_t *ps_search_results = NULL;
                if(mem_avail)
                {
                    if(i < 16)
                    {
                        ps_search_results =
                            &ps_ctxt->aps_me_frm_prms[j]->as_search_results_16x16[i];
                    }
                    else if(i < 20)
                    {
                        ps_search_results =
                            &ps_ctxt->aps_me_frm_prms[j]->as_search_results_32x32[i - 16];
                        ps_search_results->ps_cu_results =
                            &ps_ctxt->aps_me_frm_prms[j]->as_cu32x32_results[i - 16];
                    }
                    else if(i == 20)
                    {
                        ps_search_results = &ps_ctxt->aps_me_frm_prms[j]->s_search_results_64x64;
                        ps_search_results->ps_cu_results =
                            &ps_ctxt->aps_me_frm_prms[j]->s_cu64x64_results;
                    }
                    else
                    {
                        /* 8x8 search results are not required in LO ME */
                        ASSERT(0);
                    }
                }
                count += hme_alloc_init_search_nodes(
                    ps_search_results, &ps_memtabs[count], mem_avail, 2, ps_prms->max_num_results);
            }
        }

        if((i4_num_me_frm_pllel == 1) && (j != (MAX_NUM_ME_PARALLEL - 1)))
        {
            count = count_cpy;
        }
    }

    /* Weighted inputs, one for each ref + one non weighted */
    for(j = 0; j < MAX_NUM_ME_PARALLEL; j++)
    {
        size = (ps_prms->max_num_ref + 1) * ctb_wd * ctb_wd * ps_prms->i4_num_proc_thrds;
        if(mem_avail)
        {
            U08 *pu1_mem;
            ASSERT(ps_memtabs[count].size == size);
            pu1_mem = ps_memtabs[count].pu1_mem;

            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

                for(i = 0; i < ps_prms->max_num_ref + 1; i++)
                {
                    ps_ctxt->aps_me_frm_prms[j]->s_wt_pred.apu1_wt_inp_buf_array[i] = pu1_mem;
                    pu1_mem += (ctb_wd * ctb_wd);
                }
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 16;
            ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
        }
        if((i4_num_me_frm_pllel != 1) || (j == (MAX_NUM_ME_PARALLEL - 1)))
        {
            count++;
        }
    }

    /* if memory is allocated the intislaise the frm prms ptr to each thrd */
    if(mem_avail)
    {
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            me_frm_ctxt_t *ps_frm_ctxt;
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
            {
                ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[i];

                ps_frm_ctxt->ps_hme_frm_prms = &ps_master_ctxt->as_frm_prms[i];
                ps_frm_ctxt->ps_hme_ref_map = &ps_master_ctxt->as_ref_map[i];
            }
        }
    }

    /* Memory allocation for use in Clustering */
    if(ps_prms->s_me_coding_tools.e_me_quality_presets == ME_PRISTINE_QUALITY)
    {
        for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
        {
            size = 16 * sizeof(cluster_16x16_blk_t) + 4 * sizeof(cluster_32x32_blk_t) +
                   sizeof(cluster_64x64_blk_t) + sizeof(ctb_cluster_info_t);
            size *= ps_prms->i4_num_proc_thrds;

            if(mem_avail)
            {
                U08 *pu1_mem;

                ASSERT(ps_memtabs[count].size == size);
                pu1_mem = ps_memtabs[count].pu1_mem;

                for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
                {
                    ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

                    ps_ctxt->aps_me_frm_prms[i]->ps_blk_16x16 = (cluster_16x16_blk_t *)pu1_mem;
                    pu1_mem += (16 * sizeof(cluster_16x16_blk_t));

                    ps_ctxt->aps_me_frm_prms[i]->ps_blk_32x32 = (cluster_32x32_blk_t *)pu1_mem;
                    pu1_mem += (4 * sizeof(cluster_32x32_blk_t));

                    ps_ctxt->aps_me_frm_prms[i]->ps_blk_64x64 = (cluster_64x64_blk_t *)pu1_mem;
                    pu1_mem += (sizeof(cluster_64x64_blk_t));

                    ps_ctxt->aps_me_frm_prms[i]->ps_ctb_cluster_info =
                        (ctb_cluster_info_t *)pu1_mem;
                    pu1_mem += (sizeof(ctb_cluster_info_t));
                }
            }
            else
            {
                ps_memtabs[count].size = size;
                ps_memtabs[count].align = 16;
                ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
            }

            if((i4_num_me_frm_pllel != 1) || (i == (MAX_NUM_ME_PARALLEL - 1)))
            {
                count++;
            }
        }
    }
    else if(mem_avail)
    {
        for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
        {
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

                ps_ctxt->aps_me_frm_prms[i]->ps_blk_16x16 = NULL;

                ps_ctxt->aps_me_frm_prms[i]->ps_blk_32x32 = NULL;

                ps_ctxt->aps_me_frm_prms[i]->ps_blk_64x64 = NULL;

                ps_ctxt->aps_me_frm_prms[i]->ps_ctb_cluster_info = NULL;
            }
        }
    }

    for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
    {
        size = sizeof(fullpel_refine_ctxt_t);
        size *= ps_prms->i4_num_proc_thrds;

        if(mem_avail)
        {
            U08 *pu1_mem;

            ASSERT(ps_memtabs[count].size == size);
            pu1_mem = ps_memtabs[count].pu1_mem;

            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

                ps_ctxt->aps_me_frm_prms[i]->ps_fullpel_refine_ctxt =
                    (fullpel_refine_ctxt_t *)pu1_mem;
                pu1_mem += (sizeof(fullpel_refine_ctxt_t));
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 16;
            ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
        }

        if((i4_num_me_frm_pllel != 1) || (i == (MAX_NUM_ME_PARALLEL - 1)))
        {
            count++;
        }
    }

    /* Memory for ihevce_me_optimised_function_list_t struct  */
    if(mem_avail)
    {
        ps_master_ctxt->pv_me_optimised_function_list = (void *)ps_memtabs[count++].pu1_mem;
    }
    else
    {
        ps_memtabs[count].size = sizeof(ihevce_me_optimised_function_list_t);
        ps_memtabs[count].align = 16;
        ps_memtabs[count++].e_mem_attr = HME_SCRATCH_OVLY_MEM;
    }

    ASSERT(count < hme_enc_num_alloc(i4_num_me_frm_pllel));
    return (count);
}

/**
********************************************************************************
*  @fn     hme_coarse_alloc_init_mem()
*
*  @brief  Requests/ assign memory based on mem avail
*
*  @param[in] ps_memtabs : memtab array
*
*  @param[in] ps_prms : init prms
*
*  @param[in] pv_ctxt : ME ctxt
*
*  @param[in] mem_avail : request/assign flag
*
*  @return  number of memtabs
********************************************************************************
*/
S32 hme_coarse_alloc_init_mem(
    hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms, void *pv_ctxt, S32 mem_avail)
{
    coarse_me_master_ctxt_t *ps_master_ctxt = (coarse_me_master_ctxt_t *)pv_ctxt;
    coarse_me_ctxt_t *ps_ctxt;
    S32 count = 0, size, i, j, use_4x4, wd;
    S32 n_tot_layers;
    S32 num_layers_explicit_search;
    S32 a_wd[MAX_NUM_LAYERS], a_ht[MAX_NUM_LAYERS];
    S32 a_disp_wd[MAX_NUM_LAYERS], a_disp_ht[MAX_NUM_LAYERS];
    S32 num_results;
    S32 num_thrds;
    //S32 ctb_wd = 1 << ps_prms->log_ctb_size;
    S32 sad_4x4_block_size, sad_4x4_block_stride, search_step, num_rows;
    S32 layer1_blk_width = 8;  // 8x8 search
    S32 blk_shift;

    /* MV bank changes */
    hme_mv_t *aps_mv_bank[MAX_NUM_LAYERS] = { NULL };
    S32 ai4_num_mvs_per_row[MAX_NUM_LAYERS] = { 0 };
    S08 *api1_ref_idx[MAX_NUM_LAYERS] = { NULL };

    /* Memtab 0: handle */
    size = sizeof(coarse_me_master_ctxt_t);
    if(mem_avail)
    {
        /* store the number of processing threads */
        ps_master_ctxt->i4_num_proc_thrds = ps_prms->i4_num_proc_thrds;
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 8;
        ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    /* Memtab 1: ME threads ctxt */
    size = ps_prms->i4_num_proc_thrds * sizeof(coarse_me_ctxt_t);
    if(mem_avail)
    {
        coarse_me_ctxt_t *ps_me_tmp_ctxt = (coarse_me_ctxt_t *)ps_memtabs[count].pu1_mem;

        /* store the indivisual thread ctxt pointers */
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            ps_master_ctxt->aps_me_ctxt[num_thrds] = ps_me_tmp_ctxt++;
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 8;
        ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
    }

    count++;

    memcpy(a_wd, ps_prms->a_wd, sizeof(S32) * ps_prms->num_simulcast_layers);
    memcpy(a_ht, ps_prms->a_ht, sizeof(S32) * ps_prms->num_simulcast_layers);
    /*************************************************************************/
    /* Derive the number of HME layers, including both encoded and non encode*/
    /* This function also derives the width and ht of each layer.            */
    /*************************************************************************/
    n_tot_layers = hme_derive_num_layers(1, a_wd, a_ht, a_disp_wd, a_disp_ht);

    num_layers_explicit_search = ps_prms->num_layers_explicit_search;

    if(num_layers_explicit_search <= 0)
        num_layers_explicit_search = n_tot_layers - 1;

    num_layers_explicit_search = MIN(num_layers_explicit_search, n_tot_layers - 1);

    if(mem_avail)
    {
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
            memset(ps_ctxt->u1_encode, 0, n_tot_layers);

            /* encode layer should be excluded during processing */
            ps_ctxt->num_layers = n_tot_layers;

            memcpy(ps_ctxt->a_wd, a_wd, sizeof(S32) * n_tot_layers);
            memcpy(ps_ctxt->a_ht, a_ht, sizeof(S32) * n_tot_layers);

            ps_ctxt->num_layers_explicit_search = num_layers_explicit_search;
            ps_ctxt->max_num_results = ps_prms->max_num_results;
            ps_ctxt->max_num_results_coarse = ps_prms->max_num_results_coarse;
            ps_ctxt->max_num_ref = ps_prms->max_num_ref;
        }
    }

    /* Memtabs : Layers MV bank for total layers - 2  */
    /* for penultimate layer MV bank will be initialsed at every frame level */
    for(j = 1; j < n_tot_layers; j++)
    {
        S32 is_explicit_store = 1;
        S32 wd, ht;
        U08 u1_enc = 0;
        wd = a_wd[j];
        ht = a_ht[j];

        /* Possibly implicit search for lower (finer) layers */
        if(n_tot_layers - j > num_layers_explicit_search)
            is_explicit_store = 0;

        /* Even if explicit search, we store only 2 results (L0 and L1) */
        /* in finest layer */
        if(j == 0)
        {
            is_explicit_store = 0;
        }

        /* coarsest layer alwasy uses 4x4 blks to store results */
        if(j == n_tot_layers - 1)
        {
            num_results = ps_prms->max_num_results_coarse;
        }
        else
        {
            num_results = ps_prms->max_num_results;
            if(j == 0)
                num_results = 1;
        }
        use_4x4 = hme_get_mv_blk_size(ps_prms->use_4x4, j, n_tot_layers, u1_enc);

        /* for penultimate compute the parameters and store */
        if(j == 1)
        {
            S32 num_blks, num_mvs_per_blk, num_ref;
            S32 num_cols, num_rows, num_mvs_per_row;

            num_cols = use_4x4 ? ((wd >> 2) + 2) : ((wd >> 3) + 2);
            num_rows = use_4x4 ? ((ht >> 2) + 2) : ((ht >> 3) + 2);

            if(is_explicit_store)
                num_ref = ps_prms->max_num_ref;
            else
                num_ref = 2;

            num_blks = num_cols * num_rows;
            num_mvs_per_blk = num_ref * num_results;
            num_mvs_per_row = num_mvs_per_blk * num_cols;

            ai4_num_mvs_per_row[j] = num_mvs_per_row;
            aps_mv_bank[j] = NULL;
            api1_ref_idx[j] = NULL;
        }
        else
        {
            count += hme_alloc_init_layer_mv_bank(
                &ps_memtabs[count],
                num_results,
                ps_prms->max_num_ref,
                use_4x4,
                mem_avail,
                u1_enc,
                wd,
                ht,
                is_explicit_store,
                &aps_mv_bank[j],
                &api1_ref_idx[j],
                &ai4_num_mvs_per_row[j]);
        }
    }

    /* Memtabs : Layers * num-ref + 1 */
    for(i = 0; i < ps_prms->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        /* for all layer except encode layer */
        for(j = 1; j < n_tot_layers; j++)
        {
            layer_ctxt_t *ps_layer;
            S32 is_explicit_store = 1;
            S32 segment_this_layer = (j == 0) ? 1 : ps_prms->segment_higher_layers;
            S32 wd, ht;
            U08 u1_enc = 0;
            wd = a_wd[j];
            ht = a_ht[j];

            /* Possibly implicit search for lower (finer) layers */
            if(n_tot_layers - j > num_layers_explicit_search)
                is_explicit_store = 0;

            /* Even if explicit search, we store only 2 results (L0 and L1) */
            /* in finest layer */
            if(j == 0)
            {
                is_explicit_store = 0;
            }

            /* coarsest layer alwasy uses 4x4 blks to store results */
            if(j == n_tot_layers - 1)
            {
                num_results = ps_prms->max_num_results_coarse;
            }
            else
            {
                num_results = ps_prms->max_num_results;
                if(j == 0)
                    num_results = 1;
            }
            use_4x4 = hme_get_mv_blk_size(ps_prms->use_4x4, j, n_tot_layers, u1_enc);

            count += hme_alloc_init_layer(
                &ps_memtabs[count],
                num_results,
                ps_prms->max_num_ref,
                use_4x4,
                mem_avail,
                u1_enc,
                wd,
                ht,
                a_disp_wd[j],
                a_disp_ht[j],
                segment_this_layer,
                is_explicit_store,
                &ps_layer);
            if(mem_avail)
            {
                /* same ps_layer memory pointer is stored in all the threads */
                for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
                {
                    ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                    ps_ctxt->as_ref_descr[i].aps_layers[j] = ps_layer;
                }

                /* store the MV bank pointers */
                ps_layer->ps_layer_mvbank->max_num_mvs_per_row = ai4_num_mvs_per_row[j];
                ps_layer->ps_layer_mvbank->ps_mv_base = aps_mv_bank[j];
                ps_layer->ps_layer_mvbank->pi1_ref_idx_base = api1_ref_idx[j];
            }
        }
    }

    /* Memtabs : Prev Row search node at coarsest layer */
    wd = a_wd[n_tot_layers - 1];

    /* Allocate a memtab for storing 4x4 SADs for n rows. As many as num ref and number of threads */
    num_rows = ps_prms->i4_num_proc_thrds + 1;
    if(ps_prms->s_me_coding_tools.e_me_quality_presets < ME_MEDIUM_SPEED)
        search_step = HME_COARSE_STEP_SIZE_HIGH_QUALITY;
    else
        search_step = HME_COARSE_STEP_SIZE_HIGH_SPEED;

    /*shift factor*/
    blk_shift = 2; /*4x4*/
    search_step >>= 1;

    sad_4x4_block_size = ((2 * MAX_MVX_SUPPORTED_IN_COARSE_LAYER) >> search_step) *
                         ((2 * MAX_MVY_SUPPORTED_IN_COARSE_LAYER) >> search_step);
    sad_4x4_block_stride = ((wd >> blk_shift) + 1) * sad_4x4_block_size;

    size = num_rows * sad_4x4_block_stride * sizeof(S16);
    for(i = 0; i < ps_prms->max_num_ref; i++)
    {
        if(mem_avail)
        {
            ASSERT(size == ps_memtabs[count].size);

            /* same row memory pointer is stored in all the threads */
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                ps_ctxt->api2_sads_4x4_n_rows[i] = (S16 *)ps_memtabs[count].pu1_mem;
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 4;
            ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
        }
        count++;
    }

    /* Allocate a memtab for storing best search nodes 8x4 for n rows. Row is allocated for worst case (2*min_wd_coarse/4). As many as num ref and number of threads */
    size = num_rows * ((wd >> blk_shift) + 1) * sizeof(search_node_t);
    for(i = 0; i < ps_prms->max_num_ref; i++)
    {
        if(mem_avail)
        {
            ASSERT(size == ps_memtabs[count].size);

            /* same row memory pointer is stored in all the threads */
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                ps_ctxt->aps_best_search_nodes_8x4_n_rows[i] =
                    (search_node_t *)ps_memtabs[count].pu1_mem;
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 4;
            ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
        }
        count++;
    }
    /* Allocate a memtab for storing best search nodes 4x8 for n rows. Row is allocated for worst case (2*min_wd_coarse/4). As many as num ref and number of threads */
    size = num_rows * ((wd >> blk_shift) + 1) * sizeof(search_node_t);
    for(i = 0; i < ps_prms->max_num_ref; i++)
    {
        if(mem_avail)
        {
            ASSERT(size == ps_memtabs[count].size);

            /* same row memory pointer is stored in all the threads */
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                ps_ctxt->aps_best_search_nodes_4x8_n_rows[i] =
                    (search_node_t *)ps_memtabs[count].pu1_mem;
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 4;
            ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
        }
        count++;
    }

    /* Allocate a memtab for each histogram. As many as num ref and number of threads */
    for(i = 0; i < ps_prms->max_num_ref; i++)
    {
        size = ps_prms->i4_num_proc_thrds * sizeof(mv_hist_t);
        if(mem_avail)
        {
            mv_hist_t *ps_mv_hist = (mv_hist_t *)ps_memtabs[count].pu1_mem;

            ASSERT(size == ps_memtabs[count].size);

            /* divide the memory accross the threads */
            for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
            {
                ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
                ps_ctxt->aps_mv_hist[i] = ps_mv_hist;
                ps_mv_hist++;
            }
        }
        else
        {
            ps_memtabs[count].size = size;
            ps_memtabs[count].align = 8;
            ps_memtabs[count].e_mem_attr = HME_PERSISTENT_MEM;
        }
        count++;
    }

    /* Memtabs : Search nodes for 8x8 blks */
    for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
    {
        search_results_t *ps_search_results = NULL;

        if(mem_avail)
        {
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
        }

        if(mem_avail)
        {
            ps_search_results = &ps_ctxt->s_search_results_8x8;
        }
        count += hme_alloc_init_search_nodes(
            ps_search_results,
            &ps_memtabs[count],
            mem_avail,
            ps_prms->max_num_ref,
            ps_prms->max_num_results);
    }

    /* Weighted inputs, one for each ref  */
    size = (ps_prms->max_num_ref + 1) * layer1_blk_width * layer1_blk_width *
           ps_prms->i4_num_proc_thrds;
    if(mem_avail)
    {
        U08 *pu1_mem;
        ASSERT(ps_memtabs[count].size == size);
        pu1_mem = ps_memtabs[count].pu1_mem;

        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            for(i = 0; i < ps_prms->max_num_ref + 1; i++)
            {
                ps_ctxt->s_wt_pred.apu1_wt_inp_buf_array[i] = pu1_mem;
                pu1_mem += (layer1_blk_width * layer1_blk_width);
            }
        }
    }
    else
    {
        ps_memtabs[count].size = size;
        ps_memtabs[count].align = 16;
        ps_memtabs[count].e_mem_attr = HME_SCRATCH_OVLY_MEM;
    }
    count++;

    /* if memory is allocated the intislaise the frm prms ptr to each thrd */
    if(mem_avail)
    {
        for(num_thrds = 0; num_thrds < ps_prms->i4_num_proc_thrds; num_thrds++)
        {
            ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

            ps_ctxt->ps_hme_frm_prms = &ps_master_ctxt->s_frm_prms;
            ps_ctxt->ps_hme_ref_map = &ps_master_ctxt->s_ref_map;
        }
    }

    /* Memory for ihevce_me_optimised_function_list_t struct  */
    if(mem_avail)
    {
        ps_master_ctxt->pv_me_optimised_function_list = (void *)ps_memtabs[count++].pu1_mem;
    }
    else
    {
        ps_memtabs[count].size = sizeof(ihevce_me_optimised_function_list_t);
        ps_memtabs[count].align = 16;
        ps_memtabs[count++].e_mem_attr = HME_SCRATCH_OVLY_MEM;
    }

    //ASSERT(count < hme_enc_num_alloc());
    ASSERT(count < hme_coarse_num_alloc());
    return (count);
}

/*!
******************************************************************************
* \if Function name : ihevce_coarse_me_get_lyr_prms_dep_mngr \endif
*
* \brief Returns to the caller key attributes relevant for dependency manager,
*        ie, the number of vertical units in each layer
*
* \par Description:
*    This function requires the precondition that the width and ht of encode
*    layer is known.
*    The number of layers, number of vertical units in each layer, and for
*    each vertial unit in each layer, its dependency on previous layer's units
*    From ME's perspective, a vertical unit is one which is smallest min size
*    vertically (and spans the entire row horizontally). This is CTB for encode
*    layer, and 8x8 / 4x4 for non encode layers.
*
* \param[in] num_layers : Number of ME Layers
* \param[in] pai4_ht    : Array storing ht at each layer
* \param[in] pai4_wd    : Array storing wd at each layer
* \param[out] pi4_num_vert_units_in_lyr : Array of size N (num layers), each
*                     entry has num vertical units in that particular layer
*
* \return
*    None
*
* \author
*  Ittiam
*
*****************************************************************************
*/
void ihevce_coarse_me_get_lyr_prms_dep_mngr(
    WORD32 num_layers, WORD32 *pai4_ht, WORD32 *pai4_wd, WORD32 *pai4_num_vert_units_in_lyr)
{
    /* Height of current and next layers */
    WORD32 ht_c, ht_n;
    /* Blk ht at a given layer and next layer*/
    WORD32 unit_ht_c, unit_ht_n, blk_ht_c, blk_ht_n;
    /* Number of vertical units in current and next layer */
    WORD32 num_vert_c, num_vert_n;

    WORD32 ctb_size = 64, num_enc_layers = 1, use_4x4 = 1, i;
    UWORD8 au1_encode[MAX_NUM_LAYERS];

    memset(au1_encode, 0, num_layers);
    memset(au1_encode, 1, num_enc_layers);

    ht_n = pai4_ht[num_layers - 2];
    ht_c = pai4_ht[num_layers - 1];

    /* compute blk ht and unit ht for c and n */
    if(au1_encode[num_layers - 1])
    {
        blk_ht_c = 16;
        unit_ht_c = ctb_size;
    }
    else
    {
        blk_ht_c = hme_get_blk_size(use_4x4, num_layers - 1, num_layers, 0);
        unit_ht_c = blk_ht_c;
    }

    num_vert_c = (ht_c + unit_ht_c - 1) / unit_ht_c;
    /* For new design in Coarsest HME layer we need */
    /* one additional row extra at the end of frame */
    /* hence num_vert_c is incremented by 1         */
    num_vert_c++;

    /*************************************************************************/
    /* Run through each layer, set the number of vertical units              */
    /*************************************************************************/
    for(i = num_layers - 1; i > 0; i--)
    {
        pai4_num_vert_units_in_lyr[i] = num_vert_c;

        /* "n" is computed for first time */
        ht_n = pai4_ht[i - 1];
        blk_ht_n = hme_get_blk_size(use_4x4, i - 1, num_layers, 0);
        unit_ht_n = blk_ht_n;
        if(au1_encode[i - 1])
            unit_ht_n = ctb_size;

        num_vert_n = (ht_n + unit_ht_n - 1) / unit_ht_n;

        /* Compute the blk size and vert unit size in each layer             */
        /* "c" denotes curr layer, and "n" denotes the layer to which result */
        /* is projected to                                                   */
        ht_c = ht_n;
        blk_ht_c = blk_ht_n;
        unit_ht_c = unit_ht_n;
        num_vert_c = num_vert_n;
    }

    /* LAYER 0 OR ENCODE LAYER UPDATE : NO OUTPUT DEPS */
    /* set the numebr of vertical units */
    pai4_num_vert_units_in_lyr[0] = num_vert_c;
}

/**
********************************************************************************
*  @fn     hme_coarse_dep_mngr_alloc_mem()
*
*  @brief  Requests memory for HME Dep Mngr
*
* \param[in,out]  ps_mem_tab : pointer to memory descriptors table
* \param[in] ps_init_prms : Create time static parameters
* \param[in] i4_mem_space : memspace in whihc memory request should be done
*
*  @return  number of memtabs
********************************************************************************
*/
WORD32 hme_coarse_dep_mngr_alloc_mem(
    iv_mem_rec_t *ps_mem_tab,
    ihevce_static_cfg_params_t *ps_init_prms,
    WORD32 i4_mem_space,
    WORD32 i4_num_proc_thrds,
    WORD32 i4_resolution_id)
{
    WORD32 ai4_num_vert_units_in_lyr[MAX_NUM_HME_LAYERS];
    WORD32 a_wd[MAX_NUM_HME_LAYERS], a_ht[MAX_NUM_HME_LAYERS];
    WORD32 a_disp_wd[MAX_NUM_HME_LAYERS], a_disp_ht[MAX_NUM_HME_LAYERS];
    WORD32 n_enc_layers = 1, n_tot_layers, n_dep_tabs = 0, i;
    WORD32 min_cu_size;

    /* get the min cu size from config params */
    min_cu_size = ps_init_prms->s_config_prms.i4_min_log2_cu_size;

    min_cu_size = 1 << min_cu_size;

    /* Get the width and heights of different decomp layers */
    *a_wd = ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width +
            SET_CTB_ALIGN(
                ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width, min_cu_size);

    *a_ht =
        ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height +
        SET_CTB_ALIGN(
            ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height, min_cu_size);

    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    ASSERT(n_tot_layers >= 3);

    /* --- Get the number of vartical units in each layer for dep. mngr -- */
    ihevce_coarse_me_get_lyr_prms_dep_mngr(
        n_tot_layers, &a_ht[0], &a_wd[0], &ai4_num_vert_units_in_lyr[0]);

    /* Fill memtabs for HME layers,except for L0 layer */
    for(i = 1; i < n_tot_layers; i++)
    {
        n_dep_tabs += ihevce_dmgr_get_mem_recs(
            &ps_mem_tab[n_dep_tabs],
            DEP_MNGR_ROW_ROW_SYNC,
            ai4_num_vert_units_in_lyr[i],
            1, /* Number of Col Tiles :  Not supported in PreEnc */
            i4_num_proc_thrds,
            i4_mem_space);
    }

    ASSERT(n_dep_tabs <= hme_coarse_dep_mngr_num_alloc());

    return (n_dep_tabs);
}

/**
********************************************************************************
*  @fn     hme_coarse_dep_mngr_init()
*
*  @brief  Assign memory for HME Dep Mngr
*
* \param[in,out]  ps_mem_tab : pointer to memory descriptors table
* \param[in] ps_init_prms : Create time static parameters
*  @param[in] pv_ctxt : ME ctxt
* \param[in] pv_osal_handle : Osal handle
*
*  @return  number of memtabs
********************************************************************************
*/
WORD32 hme_coarse_dep_mngr_init(
    iv_mem_rec_t *ps_mem_tab,
    ihevce_static_cfg_params_t *ps_init_prms,
    void *pv_ctxt,
    void *pv_osal_handle,
    WORD32 i4_num_proc_thrds,
    WORD32 i4_resolution_id)
{
    WORD32 ai4_num_vert_units_in_lyr[MAX_NUM_HME_LAYERS];
    WORD32 a_wd[MAX_NUM_HME_LAYERS], a_ht[MAX_NUM_HME_LAYERS];
    WORD32 a_disp_wd[MAX_NUM_HME_LAYERS], a_disp_ht[MAX_NUM_HME_LAYERS];
    WORD32 n_enc_layers = 1, n_tot_layers, n_dep_tabs = 0, i;
    WORD32 min_cu_size;

    coarse_me_master_ctxt_t *ps_me_ctxt = (coarse_me_master_ctxt_t *)pv_ctxt;

    /* get the min cu size from config params */
    min_cu_size = ps_init_prms->s_config_prms.i4_min_log2_cu_size;

    min_cu_size = 1 << min_cu_size;

    /* Get the width and heights of different decomp layers */
    *a_wd = ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width +
            SET_CTB_ALIGN(
                ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width, min_cu_size);
    *a_ht =
        ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height +
        SET_CTB_ALIGN(
            ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height, min_cu_size);

    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    ASSERT(n_tot_layers >= 3);

    /* --- Get the number of vartical units in each layer for dep. mngr -- */
    ihevce_coarse_me_get_lyr_prms_dep_mngr(
        n_tot_layers, &a_ht[0], &a_wd[0], &ai4_num_vert_units_in_lyr[0]);

    /* --- HME sync Dep Mngr Mem init --    */
    for(i = 1; i < n_tot_layers; i++)
    {
        WORD32 num_blks_in_row, num_blks_in_pic, blk_size_shift;

        if(i == (n_tot_layers - 1)) /* coarsest layer */
            blk_size_shift = 2;
        else
            blk_size_shift = 3; /* refine layers */

        GET_NUM_BLKS_IN_PIC(a_wd[i], a_ht[i], blk_size_shift, num_blks_in_row, num_blks_in_pic);

        /* Coarsest layer : 1 block extra, since the last block */
        if(i == (n_tot_layers - 1)) /*  in a row needs East block */
            num_blks_in_row += 1;

        /* Note : i-1, only for HME layers, L0 is separate */
        ps_me_ctxt->apv_dep_mngr_hme_sync[i - 1] = ihevce_dmgr_init(
            &ps_mem_tab[n_dep_tabs],
            pv_osal_handle,
            DEP_MNGR_ROW_ROW_SYNC,
            ai4_num_vert_units_in_lyr[i],
            num_blks_in_row,
            1, /* Number of Col Tiles : Not supported in PreEnc */
            i4_num_proc_thrds,
            1 /*Sem disabled*/
        );

        n_dep_tabs += ihevce_dmgr_get_num_mem_recs();
    }

    return n_dep_tabs;
}

/**
********************************************************************************
*  @fn     hme_coarse_dep_mngr_reg_sem()
*
*  @brief  Assign semaphores for HME Dep Mngr
*
* \param[in] pv_me_ctxt : pointer to Coarse ME ctxt
* \param[in] ppv_sem_hdls : Arry of semaphore handles
* \param[in] i4_num_proc_thrds : Number of processing threads
*
*  @return  number of memtabs
********************************************************************************
*/
void hme_coarse_dep_mngr_reg_sem(void *pv_ctxt, void **ppv_sem_hdls, WORD32 i4_num_proc_thrds)
{
    WORD32 i;
    coarse_me_master_ctxt_t *ps_me_ctxt = (coarse_me_master_ctxt_t *)pv_ctxt;
    coarse_me_ctxt_t *ps_ctxt = ps_me_ctxt->aps_me_ctxt[0];

    /* --- HME sync Dep Mngr semaphore init --    */
    for(i = 1; i < ps_ctxt->num_layers; i++)
    {
        ihevce_dmgr_reg_sem_hdls(
            ps_me_ctxt->apv_dep_mngr_hme_sync[i - 1], ppv_sem_hdls, i4_num_proc_thrds);
    }

    return;
}

/**
********************************************************************************
*  @fn     hme_coarse_dep_mngr_delete()
*
*    Destroy Coarse ME Dep Mngr module
*   Note : Only Destroys the resources allocated in the module like
*   semaphore,etc. Memory free is done Separately using memtabs
*
* \param[in] pv_me_ctxt : pointer to Coarse ME ctxt
* \param[in] ps_init_prms : Create time static parameters
*
*  @return  none
********************************************************************************
*/
void hme_coarse_dep_mngr_delete(
    void *pv_me_ctxt, ihevce_static_cfg_params_t *ps_init_prms, WORD32 i4_resolution_id)
{
    WORD32 a_wd[MAX_NUM_HME_LAYERS], a_ht[MAX_NUM_HME_LAYERS];
    WORD32 a_disp_wd[MAX_NUM_HME_LAYERS], a_disp_ht[MAX_NUM_HME_LAYERS];
    WORD32 n_enc_layers = 1, n_tot_layers, i;
    WORD32 min_cu_size;

    coarse_me_master_ctxt_t *ps_me_ctxt = (coarse_me_master_ctxt_t *)pv_me_ctxt;

    /* get the min cu size from config params */
    min_cu_size = ps_init_prms->s_config_prms.i4_min_log2_cu_size;

    min_cu_size = 1 << min_cu_size;

    /* Get the width and heights of different decomp layers */
    *a_wd = ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width +
            SET_CTB_ALIGN(
                ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_width, min_cu_size);
    *a_ht =
        ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height +
        SET_CTB_ALIGN(
            ps_init_prms->s_tgt_lyr_prms.as_tgt_params[i4_resolution_id].i4_height, min_cu_size);
    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    ASSERT(n_tot_layers >= 3);

    /* --- HME sync Dep Mngr Delete --    */
    for(i = 1; i < n_tot_layers; i++)
    {
        /* Note : i-1, only for HME layers, L0 is separate */
        ihevce_dmgr_del(ps_me_ctxt->apv_dep_mngr_hme_sync[i - 1]);
    }
}

/**
*******************************************************************************
*  @fn     S32 hme_enc_alloc(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms)
*
*  @brief  Fills up memtabs with memory information details required by HME
*
*  @param[out] ps_memtabs : Pointre to an array of memtabs where module fills
*              up its requirements of memory
*
*  @param[in] ps_prms : Input parameters to module crucial in calculating reqd
*                       amt of memory
*
*  @return   Number of memtabs required
*******************************************************************************
*/
S32 hme_enc_alloc(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms, WORD32 i4_num_me_frm_pllel)
{
    S32 num, tot, i;

    /* Validation of init params */
    if(-1 == hme_validate_init_prms(ps_prms))
        return (-1);

    num = hme_enc_alloc_init_mem(ps_memtabs, ps_prms, NULL, 0, i4_num_me_frm_pllel);
    tot = hme_enc_num_alloc(i4_num_me_frm_pllel);
    for(i = num; i < tot; i++)
    {
        ps_memtabs[i].size = 4;
        ps_memtabs[i].align = 4;
        ps_memtabs[i].e_mem_attr = HME_PERSISTENT_MEM;
    }
    return (tot);
}

/**
*******************************************************************************
*  @fn     S32 hme_coarse_alloc(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms)
*
*  @brief  Fills up memtabs with memory information details required by Coarse HME
*
*  @param[out] ps_memtabs : Pointre to an array of memtabs where module fills
*              up its requirements of memory
*
*  @param[in] ps_prms : Input parameters to module crucial in calculating reqd
*                       amt of memory
*
*  @return   Number of memtabs required
*******************************************************************************
*/
S32 hme_coarse_alloc(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms)
{
    S32 num, tot, i;

    /* Validation of init params */
    if(-1 == hme_validate_init_prms(ps_prms))
        return (-1);

    num = hme_coarse_alloc_init_mem(ps_memtabs, ps_prms, NULL, 0);
    tot = hme_coarse_num_alloc();
    for(i = num; i < tot; i++)
    {
        ps_memtabs[i].size = 4;
        ps_memtabs[i].align = 4;
        ps_memtabs[i].e_mem_attr = HME_PERSISTENT_MEM;
    }
    return (tot);
}

/**
*******************************************************************************
*  @fn hme_coarse_dep_mngr_alloc
*
*  @brief  Fills up memtabs with memory information details required by Coarse HME
*
* \param[in,out]  ps_mem_tab : pointer to memory descriptors table
* \param[in] ps_init_prms : Create time static parameters
* \param[in] i4_mem_space : memspace in whihc memory request should be done
*
*  @return   Number of memtabs required
*******************************************************************************
*/
WORD32 hme_coarse_dep_mngr_alloc(
    iv_mem_rec_t *ps_mem_tab,
    ihevce_static_cfg_params_t *ps_init_prms,
    WORD32 i4_mem_space,
    WORD32 i4_num_proc_thrds,
    WORD32 i4_resolution_id)
{
    S32 num, tot, i;

    num = hme_coarse_dep_mngr_alloc_mem(
        ps_mem_tab, ps_init_prms, i4_mem_space, i4_num_proc_thrds, i4_resolution_id);
    tot = hme_coarse_dep_mngr_num_alloc();
    for(i = num; i < tot; i++)
    {
        ps_mem_tab[i].i4_mem_size = 4;
        ps_mem_tab[i].i4_mem_alignment = 4;
        ps_mem_tab[i].e_mem_type = (IV_MEM_TYPE_T)i4_mem_space;
    }
    return (tot);
}

/**
********************************************************************************
*  @fn     hme_coarse_init_ctxt()
*
*  @brief  initialise context memory
*
*  @param[in] ps_prms : init prms
*
*  @param[in] pv_ctxt : ME ctxt
*
*  @return  number of memtabs
********************************************************************************
*/
void hme_coarse_init_ctxt(coarse_me_master_ctxt_t *ps_master_ctxt, hme_init_prms_t *ps_prms)
{
    S32 i, j, num_thrds;
    coarse_me_ctxt_t *ps_ctxt;
    S32 num_rows_coarse;

    /* initialise the parameters inot context of all threads */
    for(num_thrds = 0; num_thrds < ps_master_ctxt->i4_num_proc_thrds; num_thrds++)
    {
        ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];

        /* Copy the init prms to context */
        ps_ctxt->s_init_prms = *ps_prms;

        /* Initialize some other variables in ctxt */
        ps_ctxt->i4_prev_poc = -1;

        ps_ctxt->num_b_frms = ps_prms->num_b_frms;

        ps_ctxt->apu1_ref_bits_tlu_lc[0] = &ps_ctxt->au1_ref_bits_tlu_lc[0][0];
        ps_ctxt->apu1_ref_bits_tlu_lc[1] = &ps_ctxt->au1_ref_bits_tlu_lc[1][0];

        /* Initialize num rows lookuptable */
        ps_ctxt->i4_num_row_bufs = ps_prms->i4_num_proc_thrds + 1;
        num_rows_coarse = ps_ctxt->i4_num_row_bufs;
        for(i = 0; i < ((HEVCE_MAX_HEIGHT >> 1) >> 2); i++)
        {
            ps_ctxt->ai4_row_index[i] = (i % num_rows_coarse);
        }
    }

    /* since same layer desc pointer is stored in all the threads ctxt */
    /* layer init is done only using 0th thread ctxt                   */
    ps_ctxt = ps_master_ctxt->aps_me_ctxt[0];

    /* Initialize all layers descriptors to have -1 = poc meaning unfilled */
    for(i = 0; i < ps_ctxt->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        for(j = 1; j < ps_ctxt->num_layers; j++)
        {
            layer_ctxt_t *ps_layer;
            ps_layer = ps_ctxt->as_ref_descr[i].aps_layers[j];
            ps_layer->i4_poc = -1;
            ps_layer->ppu1_list_inp = &ps_ctxt->apu1_list_inp[j][0];
            memset(
                ps_layer->s_global_mv, 0, sizeof(hme_mv_t) * ps_ctxt->max_num_ref * NUM_GMV_LOBES);
        }
    }
}

/**
********************************************************************************
*  @fn     hme_enc_init_ctxt()
*
*  @brief  initialise context memory
*
*  @param[in] ps_prms : init prms
*
*  @param[in] pv_ctxt : ME ctxt
*
*  @return  number of memtabs
********************************************************************************
*/
void hme_enc_init_ctxt(
    me_master_ctxt_t *ps_master_ctxt, hme_init_prms_t *ps_prms, rc_quant_t *ps_rc_quant_ctxt)
{
    S32 i, j, num_thrds;
    me_ctxt_t *ps_ctxt;
    me_frm_ctxt_t *ps_frm_ctxt;

    /* initialise the parameters in context of all threads */
    for(num_thrds = 0; num_thrds < ps_master_ctxt->i4_num_proc_thrds; num_thrds++)
    {
        ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
        /* Store Tile params base into ME context */
        ps_ctxt->pv_tile_params_base = ps_master_ctxt->pv_tile_params_base;

        for(i = 0; i < MAX_NUM_ME_PARALLEL; i++)
        {
            ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[i];

            /* Copy the init prms to context */
            ps_ctxt->s_init_prms = *ps_prms;

            /* Initialize some other variables in ctxt */
            ps_frm_ctxt->i4_prev_poc = INVALID_POC;

            ps_frm_ctxt->log_ctb_size = ps_prms->log_ctb_size;

            ps_frm_ctxt->num_b_frms = ps_prms->num_b_frms;

            ps_frm_ctxt->i4_is_prev_frame_reference = 0;

            ps_frm_ctxt->ps_rc_quant_ctxt = ps_rc_quant_ctxt;

            /* Initialize mv grids for L0 and L1 used in final refinement layer */
            {
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid[0]);
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid[1]);
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid_fpel[0]);
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid_fpel[1]);
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid_qpel[0]);
                hme_init_mv_grid(&ps_frm_ctxt->as_mv_grid_qpel[1]);
            }

            ps_frm_ctxt->apu1_ref_bits_tlu_lc[0] = &ps_frm_ctxt->au1_ref_bits_tlu_lc[0][0];
            ps_frm_ctxt->apu1_ref_bits_tlu_lc[1] = &ps_frm_ctxt->au1_ref_bits_tlu_lc[1][0];
        }
    }

    /* since same layer desc pointer is stored in all the threads ctxt */
    /* layer init is done only using 0th thread ctxt                   */
    ps_ctxt = ps_master_ctxt->aps_me_ctxt[0];

    ps_frm_ctxt = ps_ctxt->aps_me_frm_prms[0];

    /* Initialize all layers descriptors to have -1 = poc meaning unfilled */
    for(i = 0; i < (ps_frm_ctxt->max_num_ref * ps_master_ctxt->i4_num_me_frm_pllel) + 1; i++)
    {
        /* only enocde layer is processed */
        for(j = 0; j < 1; j++)
        {
            layer_ctxt_t *ps_layer;
            ps_layer = ps_ctxt->as_ref_descr[i].aps_layers[j];
            ps_layer->i4_poc = INVALID_POC;
            ps_layer->i4_is_free = 1;
            ps_layer->ppu1_list_inp = &ps_frm_ctxt->apu1_list_inp[j][0];
            ps_layer->ppu1_list_rec_fxfy = &ps_frm_ctxt->apu1_list_rec_fxfy[j][0];
            ps_layer->ppu1_list_rec_hxfy = &ps_frm_ctxt->apu1_list_rec_hxfy[j][0];
            ps_layer->ppu1_list_rec_fxhy = &ps_frm_ctxt->apu1_list_rec_fxhy[j][0];
            ps_layer->ppu1_list_rec_hxhy = &ps_frm_ctxt->apu1_list_rec_hxhy[j][0];
            ps_layer->ppv_dep_mngr_recon = &ps_frm_ctxt->apv_list_dep_mngr[j][0];

            memset(
                ps_layer->s_global_mv,
                0,
                sizeof(hme_mv_t) * ps_frm_ctxt->max_num_ref * NUM_GMV_LOBES);
        }
    }
}

/**
*******************************************************************************
*  @fn     S32 hme_enc_init(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms,rc_quant_t *ps_rc_quant_ctxt)
*
*  @brief  Initialises the Encode Layer HME ctxt
*
*  @param[out] ps_memtabs : Pointer to an array of memtabs where module fills
*              up its requirements of memory
*
*  @param[in] ps_prms : Input parameters to module crucial in calculating reqd
*                       amt of memory
*
*  @return   Number of memtabs required
*******************************************************************************
*/
S32 hme_enc_init(
    void *pv_ctxt,
    hme_memtab_t *ps_memtabs,
    hme_init_prms_t *ps_prms,
    rc_quant_t *ps_rc_quant_ctxt,
    WORD32 i4_num_me_frm_pllel)
{
    S32 num, tot;
    me_master_ctxt_t *ps_ctxt = (me_master_ctxt_t *)pv_ctxt;

    tot = hme_enc_num_alloc(i4_num_me_frm_pllel);
    /* Validation of init params */
    if(-1 == hme_validate_init_prms(ps_prms))
        return (-1);

    num = hme_enc_alloc_init_mem(ps_memtabs, ps_prms, pv_ctxt, 1, i4_num_me_frm_pllel);
    if(num > tot)
        return (-1);

    /* Initialize all enumerations based globals */
    //hme_init_globals(); /* done as part of coarse me */

    /* Copy the memtabs into the context for returning during free */
    memcpy(ps_ctxt->as_memtabs, ps_memtabs, sizeof(hme_memtab_t) * tot);

    /* initialize the context and related buffers */
    hme_enc_init_ctxt(ps_ctxt, ps_prms, ps_rc_quant_ctxt);
    return (0);
}

/**
*******************************************************************************
*  @fn     S32 hme_coarse_init(hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms)
*
*  @brief  Initialises the Coarse HME ctxt
*
*  @param[out] ps_memtabs : Pointer to an array of memtabs where module fills
*              up its requirements of memory
*
*  @param[in] ps_prms : Input parameters to module crucial in calculating reqd
*                       amt of memory
*
*  @return   Number of memtabs required
*******************************************************************************
*/
S32 hme_coarse_init(void *pv_ctxt, hme_memtab_t *ps_memtabs, hme_init_prms_t *ps_prms)
{
    S32 num, tot;
    coarse_me_master_ctxt_t *ps_ctxt = (coarse_me_master_ctxt_t *)pv_ctxt;

    tot = hme_coarse_num_alloc();
    /* Validation of init params */
    if(-1 == hme_validate_init_prms(ps_prms))
        return (-1);

    num = hme_coarse_alloc_init_mem(ps_memtabs, ps_prms, pv_ctxt, 1);
    if(num > tot)
        return (-1);

    /* Initialize all enumerations based globals */
    hme_init_globals();

    /* Copy the memtabs into the context for returning during free */
    memcpy(ps_ctxt->as_memtabs, ps_memtabs, sizeof(hme_memtab_t) * tot);

    /* initialize the context and related buffers */
    hme_coarse_init_ctxt(ps_ctxt, ps_prms);

    return (0);
}

/**
*******************************************************************************
*  @fn     S32 hme_set_resolution(void *pv_me_ctxt,
*                                   S32 n_enc_layers,
*                                   S32 *p_wd,
*                                   S32 *p_ht
*
*  @brief  Sets up the layers based on resolution information.
*
*  @param[in, out] pv_me_ctxt : ME handle, updated with the resolution info
*
*  @param[in] n_enc_layers : Number of layers encoded
*
*  @param[in] p_wd : Pointer to an array having widths for each encode layer
*
*  @param[in] p_ht : Pointer to an array having heights for each encode layer
*
*  @return   void
*******************************************************************************
*/

void hme_set_resolution(void *pv_me_ctxt, S32 n_enc_layers, S32 *p_wd, S32 *p_ht, S32 me_frm_id)
{
    S32 n_tot_layers, num_layers_explicit_search, i, j;
    me_ctxt_t *ps_thrd_ctxt;
    me_frm_ctxt_t *ps_ctxt;

    S32 a_wd[MAX_NUM_LAYERS], a_ht[MAX_NUM_LAYERS];
    S32 a_disp_wd[MAX_NUM_LAYERS], a_disp_ht[MAX_NUM_LAYERS];
    memcpy(a_wd, p_wd, n_enc_layers * sizeof(S32));
    memcpy(a_ht, p_ht, n_enc_layers * sizeof(S32));

    ps_thrd_ctxt = (me_ctxt_t *)pv_me_ctxt;

    ps_ctxt = ps_thrd_ctxt->aps_me_frm_prms[me_frm_id];

    /*************************************************************************/
    /* Derive the number of HME layers, including both encoded and non encode*/
    /* This function also derives the width and ht of each layer.            */
    /*************************************************************************/
    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    num_layers_explicit_search = ps_thrd_ctxt->s_init_prms.num_layers_explicit_search;
    if(num_layers_explicit_search <= 0)
        num_layers_explicit_search = n_tot_layers - 1;

    num_layers_explicit_search = MIN(num_layers_explicit_search, n_tot_layers - 1);
    ps_ctxt->num_layers_explicit_search = num_layers_explicit_search;
    memset(ps_ctxt->u1_encode, 0, n_tot_layers);
    memset(ps_ctxt->u1_encode, 1, n_enc_layers);

    /* only encode layer should be processed */
    ps_ctxt->num_layers = n_tot_layers;

    ps_ctxt->i4_wd = a_wd[0];
    ps_ctxt->i4_ht = a_ht[0];

    /* Memtabs : Layers * num-ref + 1 */
    for(i = 0; i < ps_ctxt->max_num_ref + 1; i++)
    {
        for(j = 0; j < 1; j++)
        {
            S32 wd, ht;
            layer_ctxt_t *ps_layer;
            U08 u1_enc = ps_ctxt->u1_encode[j];
            wd = a_wd[j];
            ht = a_ht[j];
            ps_layer = ps_thrd_ctxt->as_ref_descr[i].aps_layers[j];
            hme_set_layer_res_attrs(ps_layer, wd, ht, a_disp_wd[j], a_disp_ht[j], u1_enc);
        }
    }
}

/**
*******************************************************************************
*  @fn     S32 hme_coarse_set_resolution(void *pv_me_ctxt,
*                                   S32 n_enc_layers,
*                                   S32 *p_wd,
*                                   S32 *p_ht
*
*  @brief  Sets up the layers based on resolution information.
*
*  @param[in, out] pv_me_ctxt : ME handle, updated with the resolution info
*
*  @param[in] n_enc_layers : Number of layers encoded
*
*  @param[in] p_wd : Pointer to an array having widths for each encode layer
*
*  @param[in] p_ht : Pointer to an array having heights for each encode layer
*
*  @return   void
*******************************************************************************
*/

void hme_coarse_set_resolution(void *pv_me_ctxt, S32 n_enc_layers, S32 *p_wd, S32 *p_ht)
{
    S32 n_tot_layers, num_layers_explicit_search, i, j;
    coarse_me_ctxt_t *ps_ctxt;
    S32 a_wd[MAX_NUM_LAYERS], a_ht[MAX_NUM_LAYERS];
    S32 a_disp_wd[MAX_NUM_LAYERS], a_disp_ht[MAX_NUM_LAYERS];
    memcpy(a_wd, p_wd, n_enc_layers * sizeof(S32));
    memcpy(a_ht, p_ht, n_enc_layers * sizeof(S32));

    ps_ctxt = (coarse_me_ctxt_t *)pv_me_ctxt;
    /*************************************************************************/
    /* Derive the number of HME layers, including both encoded and non encode*/
    /* This function also derives the width and ht of each layer.            */
    /*************************************************************************/
    n_tot_layers = hme_derive_num_layers(n_enc_layers, a_wd, a_ht, a_disp_wd, a_disp_ht);
    num_layers_explicit_search = ps_ctxt->s_init_prms.num_layers_explicit_search;
    if(num_layers_explicit_search <= 0)
        num_layers_explicit_search = n_tot_layers - 1;

    num_layers_explicit_search = MIN(num_layers_explicit_search, n_tot_layers - 1);
    ps_ctxt->num_layers_explicit_search = num_layers_explicit_search;
    memset(ps_ctxt->u1_encode, 0, n_tot_layers);
    memset(ps_ctxt->u1_encode, 1, n_enc_layers);

    /* encode layer should be excluded */
    ps_ctxt->num_layers = n_tot_layers;

    memcpy(ps_ctxt->a_wd, a_wd, sizeof(S32) * n_tot_layers);
    memcpy(ps_ctxt->a_ht, a_ht, sizeof(S32) * n_tot_layers);

    /* Memtabs : Layers * num-ref + 1 */
    for(i = 0; i < ps_ctxt->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        for(j = 1; j < n_tot_layers; j++)
        {
            S32 wd, ht;
            layer_ctxt_t *ps_layer;
            U08 u1_enc = ps_ctxt->u1_encode[j];
            wd = a_wd[j];
            ht = a_ht[j];
            ps_layer = ps_ctxt->as_ref_descr[i].aps_layers[j];
            hme_set_layer_res_attrs(ps_layer, wd, ht, a_disp_wd[j], a_disp_ht[j], u1_enc);
        }
    }
}

S32 hme_find_descr_idx(me_ctxt_t *ps_ctxt, S32 i4_poc, S32 i4_idr_gop_num, S32 i4_num_me_frm_pllel)
{
    S32 i;

    for(i = 0; i < (ps_ctxt->aps_me_frm_prms[0]->max_num_ref * i4_num_me_frm_pllel) + 1; i++)
    {
        if(ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_poc == i4_poc &&
           ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_idr_gop_num == i4_idr_gop_num)
            return i;
    }
    /* Should not come here */
    ASSERT(0);
    return (-1);
}

S32 hme_coarse_find_descr_idx(coarse_me_ctxt_t *ps_ctxt, S32 i4_poc)
{
    S32 i;

    for(i = 0; i < ps_ctxt->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        if(ps_ctxt->as_ref_descr[i].aps_layers[1]->i4_poc == i4_poc)
            return i;
    }
    /* Should not come here */
    ASSERT(0);
    return (-1);
}

S32 hme_find_free_descr_idx(me_ctxt_t *ps_ctxt, S32 i4_num_me_frm_pllel)
{
    S32 i;

    for(i = 0; i < (ps_ctxt->aps_me_frm_prms[0]->max_num_ref * i4_num_me_frm_pllel) + 1; i++)
    {
        if(ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_is_free == 1)
        {
            ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_is_free = 0;
            return i;
        }
    }
    /* Should not come here */
    ASSERT(0);
    return (-1);
}

S32 hme_coarse_find_free_descr_idx(void *pv_ctxt)
{
    S32 i;

    coarse_me_ctxt_t *ps_ctxt = (coarse_me_ctxt_t *)pv_ctxt;

    for(i = 0; i < ps_ctxt->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        if(ps_ctxt->as_ref_descr[i].aps_layers[1]->i4_poc == -1)
            return i;
    }
    /* Should not come here */
    ASSERT(0);
    return (-1);
}

void hme_discard_frm(
    void *pv_me_ctxt, S32 *p_pocs_to_remove, S32 i4_idr_gop_num, S32 i4_num_me_frm_pllel)
{
    me_ctxt_t *ps_ctxt = (me_ctxt_t *)pv_me_ctxt;
    S32 count = 0, idx, i;
    layers_descr_t *ps_descr;

    /* Search for the id of the layer descriptor that has this poc */
    while(p_pocs_to_remove[count] != INVALID_POC)
    {
        ASSERT(count == 0);
        idx = hme_find_descr_idx(
            ps_ctxt, p_pocs_to_remove[count], i4_idr_gop_num, i4_num_me_frm_pllel);
        ps_descr = &ps_ctxt->as_ref_descr[idx];
        /*********************************************************************/
        /* Setting i4_is_free = 1 in all layers invalidates this layer ctxt        */
        /* Now this can be used for a fresh picture.                         */
        /*********************************************************************/
        for(i = 0; i < 1; i++)
        {
            ps_descr->aps_layers[i]->i4_is_free = 1;
        }
        count++;
    }
}

void hme_coarse_discard_frm(void *pv_me_ctxt, S32 *p_pocs_to_remove)
{
    coarse_me_ctxt_t *ps_ctxt = (coarse_me_ctxt_t *)pv_me_ctxt;
    S32 count = 0, idx, i;
    layers_descr_t *ps_descr;

    /* Search for the id of the layer descriptor that has this poc */
    while(p_pocs_to_remove[count] != -1)
    {
        idx = hme_coarse_find_descr_idx(ps_ctxt, p_pocs_to_remove[count]);
        ps_descr = &ps_ctxt->as_ref_descr[idx];
        /*********************************************************************/
        /* Setting poc = -1 in all layers invalidates this layer ctxt        */
        /* Now this can be used for a fresh picture.                         */
        /*********************************************************************/
        for(i = 1; i < ps_ctxt->num_layers; i++)
        {
            ps_descr->aps_layers[i]->i4_poc = -1;
        }
        count++;
    }
}

void hme_update_layer_desc(
    layers_descr_t *ps_layers_desc,
    hme_ref_desc_t *ps_ref_desc,
    S32 start_lyr_id,
    S32 num_layers,
    layers_descr_t *ps_curr_desc)
{
    layer_ctxt_t *ps_layer_ctxt, *ps_curr_layer;
    S32 i;
    for(i = start_lyr_id; i < num_layers; i++)
    {
        ps_layer_ctxt = ps_layers_desc->aps_layers[i];
        ps_curr_layer = ps_curr_desc->aps_layers[i];

        ps_layer_ctxt->i4_poc = ps_ref_desc->i4_poc;
        ps_layer_ctxt->i4_idr_gop_num = ps_ref_desc->i4_GOP_num;

        /* Copy the recon planes for the given reference pic at given layer */
        ps_layer_ctxt->pu1_rec_fxfy = ps_ref_desc->as_ref_info[i].pu1_rec_fxfy;
        ps_layer_ctxt->pu1_rec_hxfy = ps_ref_desc->as_ref_info[i].pu1_rec_hxfy;
        ps_layer_ctxt->pu1_rec_fxhy = ps_ref_desc->as_ref_info[i].pu1_rec_fxhy;
        ps_layer_ctxt->pu1_rec_hxhy = ps_ref_desc->as_ref_info[i].pu1_rec_hxhy;

        /*********************************************************************/
        /* reconstruction strides, offsets and padding info are copied for   */
        /* this reference pic. It is assumed that these will be same across  */
        /* pics, so even the current pic has this info updated, though the   */
        /* current pic still does not have valid recon pointers.             */
        /*********************************************************************/
        ps_layer_ctxt->i4_rec_stride = ps_ref_desc->as_ref_info[i].luma_stride;
        ps_layer_ctxt->i4_rec_offset = ps_ref_desc->as_ref_info[i].luma_offset;
        ps_layer_ctxt->i4_pad_x_rec = ps_ref_desc->as_ref_info[i].u1_pad_x;
        ps_layer_ctxt->i4_pad_y_rec = ps_ref_desc->as_ref_info[i].u1_pad_y;

        ps_curr_layer->i4_rec_stride = ps_ref_desc->as_ref_info[i].luma_stride;
        ps_curr_layer->i4_pad_x_rec = ps_ref_desc->as_ref_info[i].u1_pad_x;
        ps_curr_layer->i4_pad_y_rec = ps_ref_desc->as_ref_info[i].u1_pad_y;
    }
}

void hme_add_inp(void *pv_me_ctxt, hme_inp_desc_t *ps_inp_desc, S32 me_frm_id, S32 i4_thrd_id)
{
    layers_descr_t *ps_desc;
    layer_ctxt_t *ps_layer_ctxt;
    me_master_ctxt_t *ps_master_ctxt = (me_master_ctxt_t *)pv_me_ctxt;
    me_ctxt_t *ps_thrd_ctxt;
    me_frm_ctxt_t *ps_ctxt;

    hme_inp_buf_attr_t *ps_attr;
    S32 i4_poc, idx, i, i4_prev_poc;
    S32 num_thrds, prev_me_frm_id;
    S32 i4_idr_gop_num, i4_is_reference;

    /* since same layer desc pointer is stored in all thread ctxt */
    /* a free idx is obtained using 0th thread ctxt pointer */

    ps_thrd_ctxt = ps_master_ctxt->aps_me_ctxt[i4_thrd_id];

    ps_ctxt = ps_thrd_ctxt->aps_me_frm_prms[me_frm_id];

    /* Deriving the previous poc from previous frames context */
    if(me_frm_id == 0)
        prev_me_frm_id = (MAX_NUM_ME_PARALLEL - 1);
    else
        prev_me_frm_id = me_frm_id - 1;

    i4_prev_poc = ps_thrd_ctxt->aps_me_frm_prms[prev_me_frm_id]->i4_curr_poc;

    /* Obtain an empty layer descriptor */
    idx = hme_find_free_descr_idx(ps_thrd_ctxt, ps_master_ctxt->i4_num_me_frm_pllel);
    ps_desc = &ps_thrd_ctxt->as_ref_descr[idx];

    /* initialise the parameters for all the threads */
    for(num_thrds = 0; num_thrds < ps_master_ctxt->i4_num_proc_thrds; num_thrds++)
    {
        me_frm_ctxt_t *ps_tmp_frm_ctxt;

        ps_thrd_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
        ps_tmp_frm_ctxt = ps_thrd_ctxt->aps_me_frm_prms[me_frm_id];

        ps_tmp_frm_ctxt->ps_curr_descr = &ps_thrd_ctxt->as_ref_descr[idx];

        /* Do the initialization for the first thread alone */
        i4_poc = ps_inp_desc->i4_poc;
        i4_idr_gop_num = ps_inp_desc->i4_idr_gop_num;
        i4_is_reference = ps_inp_desc->i4_is_reference;
        /*Update poc id of previously encoded frm and curr frm */
        ps_tmp_frm_ctxt->i4_prev_poc = i4_prev_poc;
        ps_tmp_frm_ctxt->i4_curr_poc = i4_poc;
    }

    /* since same layer desc pointer is stored in all thread ctxt */
    /* following processing is done using 0th thread ctxt pointer */
    ps_thrd_ctxt = ps_master_ctxt->aps_me_ctxt[0];

    /* only encode layer */
    for(i = 0; i < 1; i++)
    {
        ps_layer_ctxt = ps_desc->aps_layers[i];
        ps_attr = &ps_inp_desc->s_layer_desc[i];

        ps_layer_ctxt->i4_poc = i4_poc;
        ps_layer_ctxt->i4_idr_gop_num = i4_idr_gop_num;
        ps_layer_ctxt->i4_is_reference = i4_is_reference;
        ps_layer_ctxt->i4_non_ref_free = 0;

        /* If this layer is encoded, copy input attributes */
        if(ps_ctxt->u1_encode[i])
        {
            ps_layer_ctxt->pu1_inp = ps_attr->pu1_y;
            ps_layer_ctxt->i4_inp_stride = ps_attr->luma_stride;
            ps_layer_ctxt->i4_pad_x_inp = 0;
            ps_layer_ctxt->i4_pad_y_inp = 0;
        }
        else
        {
            /* If not encoded, then ME owns the buffer.*/
            S32 wd, dst_stride;

            ASSERT(i != 0);

            wd = ps_ctxt->i4_wd;

            /* destination has padding on either side of 16 */
            dst_stride = CEIL16((wd >> 1)) + 32 + 4;
            ps_layer_ctxt->i4_inp_stride = dst_stride;
        }
    }

    return;
}

void hme_coarse_add_inp(void *pv_me_ctxt, hme_inp_desc_t *ps_inp_desc, WORD32 i4_curr_idx)
{
    layers_descr_t *ps_desc;
    layer_ctxt_t *ps_layer_ctxt;
    coarse_me_master_ctxt_t *ps_master_ctxt = (coarse_me_master_ctxt_t *)pv_me_ctxt;
    coarse_me_ctxt_t *ps_ctxt;
    hme_inp_buf_attr_t *ps_attr;
    S32 i4_poc, i;
    S32 num_thrds;

    /* since same layer desc pointer is stored in all thread ctxt */
    /* a free idx is obtained using 0th thread ctxt pointer */
    ps_ctxt = ps_master_ctxt->aps_me_ctxt[0];

    ps_desc = &ps_ctxt->as_ref_descr[i4_curr_idx];

    /* initialise the parameters for all the threads */
    for(num_thrds = 0; num_thrds < ps_master_ctxt->i4_num_proc_thrds; num_thrds++)
    {
        ps_ctxt = ps_master_ctxt->aps_me_ctxt[num_thrds];
        ps_ctxt->ps_curr_descr = &ps_ctxt->as_ref_descr[i4_curr_idx];
        i4_poc = ps_inp_desc->i4_poc;

        /*Update poc id of previously encoded frm and curr frm */
        ps_ctxt->i4_prev_poc = ps_ctxt->i4_curr_poc;
        ps_ctxt->i4_curr_poc = i4_poc;
    }

    /* since same layer desc pointer is stored in all thread ctxt */
    /* following processing is done using 0th thread ctxt pointer */
    ps_ctxt = ps_master_ctxt->aps_me_ctxt[0];

    /* only non encode layer */
    for(i = 1; i < ps_ctxt->num_layers; i++)
    {
        ps_layer_ctxt = ps_desc->aps_layers[i];
        ps_attr = &ps_inp_desc->s_layer_desc[i];

        ps_layer_ctxt->i4_poc = i4_poc;
        /* If this layer is encoded, copy input attributes */
        if(ps_ctxt->u1_encode[i])
        {
            ps_layer_ctxt->pu1_inp = ps_attr->pu1_y;
            ps_layer_ctxt->i4_inp_stride = ps_attr->luma_stride;
            ps_layer_ctxt->i4_pad_x_inp = 0;
            ps_layer_ctxt->i4_pad_y_inp = 0;
        }
        else
        {
            /* If not encoded, then ME owns the buffer.           */
            /* decomp of lower layers happens on a seperate pass  */
            /* Coarse Me should export the pointers to the caller */
            S32 wd, dst_stride;

            ASSERT(i != 0);

            wd = ps_ctxt->a_wd[i - 1];

            /* destination has padding on either side of 16 */
            dst_stride = CEIL16((wd >> 1)) + 32 + 4;
            ps_layer_ctxt->i4_inp_stride = dst_stride;
        }
    }
}

static __inline U08 hme_determine_num_results_per_part(
    U08 u1_layer_id, U08 u1_num_layers, ME_QUALITY_PRESETS_T e_quality_preset)
{
    U08 u1_num_results_per_part = MAX_RESULTS_PER_PART;

    if((u1_layer_id == 0) && !!RESTRICT_NUM_PARTITION_LEVEL_L0ME_RESULTS_TO_1)
    {
        switch(e_quality_preset)
        {
        case ME_XTREME_SPEED_25:
        case ME_XTREME_SPEED:
        case ME_HIGH_SPEED:
        case ME_MEDIUM_SPEED:
        case ME_HIGH_QUALITY:
        case ME_PRISTINE_QUALITY:
        {
            u1_num_results_per_part = 1;

            break;
        }
        default:
        {
            u1_num_results_per_part = MAX_RESULTS_PER_PART;

            break;
        }
        }
    }
    else if((u1_layer_id == 1) && !!RESTRICT_NUM_PARTITION_LEVEL_L1ME_RESULTS_TO_1)
    {
        switch(e_quality_preset)
        {
        case ME_XTREME_SPEED_25:
        case ME_HIGH_QUALITY:
        case ME_PRISTINE_QUALITY:
        {
            u1_num_results_per_part = 1;

            break;
        }
        default:
        {
            u1_num_results_per_part = MAX_RESULTS_PER_PART;

            break;
        }
        }
    }
    else if((u1_layer_id == 2) && (u1_num_layers > 3) && !!RESTRICT_NUM_PARTITION_LEVEL_L2ME_RESULTS_TO_1)
    {
        switch(e_quality_preset)
        {
        case ME_XTREME_SPEED_25:
        case ME_XTREME_SPEED:
        case ME_HIGH_SPEED:
        case ME_MEDIUM_SPEED:
        {
            u1_num_results_per_part = 1;

            break;
        }
        default:
        {
            u1_num_results_per_part = MAX_RESULTS_PER_PART;

            break;
        }
        }
    }

    return u1_num_results_per_part;
}

static __inline void hme_max_search_cands_per_search_cand_loc_populator(
    hme_frm_prms_t *ps_frm_prms,
    U08 *pu1_num_fpel_search_cands,
    U08 u1_layer_id,
    ME_QUALITY_PRESETS_T e_quality_preset)
{
    if(0 == u1_layer_id)
    {
        S32 i;

        for(i = 0; i < NUM_SEARCH_CAND_LOCATIONS; i++)
        {
            switch(e_quality_preset)
            {
#if RESTRICT_NUM_SEARCH_CANDS_PER_SEARCH_CAND_LOC
            case ME_XTREME_SPEED_25:
            case ME_XTREME_SPEED:
            case ME_HIGH_SPEED:
            case ME_MEDIUM_SPEED:
            {
                pu1_num_fpel_search_cands[i] = 1;

                break;
            }
#endif
            default:
            {
                pu1_num_fpel_search_cands[i] =
                    MAX(2,
                        MAX(ps_frm_prms->u1_num_active_ref_l0, ps_frm_prms->u1_num_active_ref_l1) *
                            ((COLOCATED == (SEARCH_CAND_LOCATIONS_T)i) + 1));

                break;
            }
            }
        }
    }
}

static __inline U08
    hme_determine_max_2nx2n_tu_recur_cands(U08 u1_layer_id, ME_QUALITY_PRESETS_T e_quality_preset)
{
    U08 u1_num_cands = 2;

    if((u1_layer_id == 0) && !!RESTRICT_NUM_2NX2N_TU_RECUR_CANDS)
    {
        switch(e_quality_preset)
        {
        case ME_XTREME_SPEED_25:
        case ME_XTREME_SPEED:
        case ME_HIGH_SPEED:
        case ME_MEDIUM_SPEED:
        {
            u1_num_cands = 1;

            break;
        }
        default:
        {
            u1_num_cands = 2;

            break;
        }
        }
    }

    return u1_num_cands;
}

static __inline U08
    hme_determine_max_num_fpel_refine_centers(U08 u1_layer_id, ME_QUALITY_PRESETS_T e_quality_preset)
{
    U08 i;

    U08 u1_num_centers = 0;

    if(0 == u1_layer_id)
    {
        switch(e_quality_preset)
        {
        case ME_XTREME_SPEED_25:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_XS25[i];
            }

            break;
        }
        case ME_XTREME_SPEED:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_XS[i];
            }

            break;
        }
        case ME_HIGH_SPEED:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_HS[i];
            }

            break;
        }
        case ME_MEDIUM_SPEED:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_MS[i];
            }

            break;
        }
        case ME_HIGH_QUALITY:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_HQ[i];
            }

            break;
        }
        case ME_PRISTINE_QUALITY:
        {
            for(i = 0; i < TOT_NUM_PARTS; i++)
            {
                u1_num_centers += gau1_num_best_results_PQ[i];
            }

            break;
        }
        }
    }

    return u1_num_centers;
}

static __inline U08 hme_determine_max_num_subpel_refine_centers(
    U08 u1_layer_id, U08 u1_max_2Nx2N_subpel_cands, U08 u1_max_NxN_subpel_cands)
{
    U08 u1_num_centers = 0;

    if(0 == u1_layer_id)
    {
        u1_num_centers += u1_max_2Nx2N_subpel_cands + 4 * u1_max_NxN_subpel_cands;
    }

    return u1_num_centers;
}

void hme_set_refine_prms(
    void *pv_refine_prms,
    U08 u1_encode,
    S32 num_ref,
    S32 layer_id,
    S32 num_layers,
    S32 num_layers_explicit_search,
    S32 use_4x4,
    hme_frm_prms_t *ps_frm_prms,
    double **ppd_intra_costs,
    me_coding_params_t *ps_me_coding_tools)
{
    refine_prms_t *ps_refine_prms = (refine_prms_t *)pv_refine_prms;

    ps_refine_prms->i4_encode = u1_encode;
    ps_refine_prms->bidir_enabled = ps_frm_prms->bidir_enabled;
    ps_refine_prms->i4_layer_id = layer_id;
    /*************************************************************************/
    /* Refinement layers have two lambdas, one for closed loop, another for  */
    /* open loop. Non encode layers use only open loop lambda.               */
    /*************************************************************************/
    ps_refine_prms->lambda_inp = ps_frm_prms->i4_ol_sad_lambda_qf;
    ps_refine_prms->lambda_recon = ps_frm_prms->i4_cl_sad_lambda_qf;
    ps_refine_prms->lambda_q_shift = ps_frm_prms->lambda_q_shift;
    ps_refine_prms->lambda_inp =
        ((float)ps_refine_prms->lambda_inp) * (100.0f - ME_LAMBDA_DISCOUNT) / 100.0f;
    ps_refine_prms->lambda_recon =
        ((float)ps_refine_prms->lambda_recon) * (100.0f - ME_LAMBDA_DISCOUNT) / 100.0f;

    if((u1_encode) && (NULL != ppd_intra_costs))
    {
        ps_refine_prms->pd_intra_costs = ppd_intra_costs[layer_id];
    }

    /* Explicit or implicit depends on number of layers having eplicit search */
    if((layer_id == 0) || (num_layers - layer_id > num_layers_explicit_search))
    {
        ps_refine_prms->explicit_ref = 0;
        ps_refine_prms->i4_num_ref_fpel = MIN(2, num_ref);
    }
    else
    {
        ps_refine_prms->explicit_ref = 1;
        ps_refine_prms->i4_num_ref_fpel = num_ref;
    }

    ps_refine_prms->e_search_complexity = SEARCH_CX_HIGH;

    ps_refine_prms->i4_num_steps_hpel_refine = ps_me_coding_tools->i4_num_steps_hpel_refine;
    ps_refine_prms->i4_num_steps_qpel_refine = ps_me_coding_tools->i4_num_steps_qpel_refine;

    if(u1_encode)
    {
        ps_refine_prms->i4_num_mvbank_results = 1;
        ps_refine_prms->i4_use_rec_in_fpel = 1;
        ps_refine_prms->i4_num_steps_fpel_refine = 1;

        if(ps_me_coding_tools->e_me_quality_presets == ME_PRISTINE_QUALITY)
        {
            ps_refine_prms->i4_num_fpel_results = 4;
            ps_refine_prms->i4_num_32x32_merge_results = 4;
            ps_refine_prms->i4_num_64x64_merge_results = 4;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 3;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 2;
            ps_refine_prms->u1_max_subpel_candts_NxN = 1;
            ps_refine_prms->u1_subpel_candt_threshold = 1;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_PQ;
            ps_refine_prms->limit_active_partitions = 0;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_HIGH_QUALITY)
        {
            ps_refine_prms->i4_num_fpel_results = 4;
            ps_refine_prms->i4_num_32x32_merge_results = 4;
            ps_refine_prms->i4_num_64x64_merge_results = 4;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 3;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 2;
            ps_refine_prms->u1_max_subpel_candts_NxN = 1;
            ps_refine_prms->u1_subpel_candt_threshold = 2;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_HQ;
            ps_refine_prms->limit_active_partitions = 0;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_MEDIUM_SPEED)
        {
            ps_refine_prms->i4_num_fpel_results = 1;
            ps_refine_prms->i4_num_32x32_merge_results = 2;
            ps_refine_prms->i4_num_64x64_merge_results = 2;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 2;
            ps_refine_prms->u1_max_subpel_candts_NxN = 1;
            ps_refine_prms->u1_subpel_candt_threshold = 3;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_MS;
            ps_refine_prms->limit_active_partitions = 1;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_HIGH_SPEED)
        {
            ps_refine_prms->i4_num_fpel_results = 1;
            ps_refine_prms->i4_num_32x32_merge_results = 2;
            ps_refine_prms->i4_num_64x64_merge_results = 2;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 1;
            ps_refine_prms->u1_max_subpel_candts_NxN = 1;
            ps_refine_prms->i4_use_satd_subpel = 0;
            ps_refine_prms->u1_subpel_candt_threshold = 0;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_HS;
            ps_refine_prms->limit_active_partitions = 1;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_XTREME_SPEED)
        {
            ps_refine_prms->i4_num_fpel_results = 1;
            ps_refine_prms->i4_num_32x32_merge_results = 2;
            ps_refine_prms->i4_num_64x64_merge_results = 2;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 0;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 1;
            ps_refine_prms->u1_max_subpel_candts_NxN = 0;
            ps_refine_prms->u1_subpel_candt_threshold = 0;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_XS;
            ps_refine_prms->limit_active_partitions = 1;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_XTREME_SPEED_25)
        {
            ps_refine_prms->i4_num_fpel_results = 1;
            ps_refine_prms->i4_num_32x32_merge_results = 2;
            ps_refine_prms->i4_num_64x64_merge_results = 2;
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 0;
            ps_refine_prms->u1_max_subpel_candts_2Nx2N = 1;
            ps_refine_prms->u1_max_subpel_candts_NxN = 0;
            ps_refine_prms->u1_subpel_candt_threshold = 0;
            ps_refine_prms->e_search_complexity = SEARCH_CX_LOW;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_XS25;
            ps_refine_prms->limit_active_partitions = 1;
        }
    }
    else
    {
        ps_refine_prms->i4_num_fpel_results = 2;
        ps_refine_prms->i4_use_rec_in_fpel = 0;
        ps_refine_prms->i4_num_steps_fpel_refine = 1;
        ps_refine_prms->i4_num_steps_hpel_refine = 0;
        ps_refine_prms->i4_num_steps_qpel_refine = 0;

        if(ps_me_coding_tools->e_me_quality_presets == ME_HIGH_SPEED)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->e_search_complexity = SEARCH_CX_LOW;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_HS;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_XTREME_SPEED)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 0;
            ps_refine_prms->e_search_complexity = SEARCH_CX_LOW;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_XS;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_XTREME_SPEED_25)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 0;
            ps_refine_prms->e_search_complexity = SEARCH_CX_LOW;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_XS25;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_PRISTINE_QUALITY)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 2;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_PQ;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_HIGH_QUALITY)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 2;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->e_search_complexity = SEARCH_CX_MED;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_HQ;
        }
        else if(ps_me_coding_tools->e_me_quality_presets == ME_MEDIUM_SPEED)
        {
            ps_refine_prms->i4_num_steps_post_refine_fpel = 0;
            ps_refine_prms->i4_use_satd_subpel = 1;
            ps_refine_prms->e_search_complexity = SEARCH_CX_LOW;
            ps_refine_prms->pu1_num_best_results = gau1_num_best_results_MS;
        }

        /* Following fields unused in the non-encode layers */
        /* But setting the same to default values           */
        ps_refine_prms->i4_num_32x32_merge_results = 4;
        ps_refine_prms->i4_num_64x64_merge_results = 4;

        if(!ps_frm_prms->bidir_enabled)
        {
            ps_refine_prms->limit_active_partitions = 0;
        }
        else
        {
            ps_refine_prms->limit_active_partitions = 1;
        }
    }

    ps_refine_prms->i4_enable_4x4_part =
        hme_get_mv_blk_size(use_4x4, layer_id, num_layers, u1_encode);

    if(!ps_me_coding_tools->u1_l0_me_controlled_via_cmd_line)
    {
        ps_refine_prms->i4_num_results_per_part = hme_determine_num_results_per_part(
            layer_id, num_layers, ps_me_coding_tools->e_me_quality_presets);

        hme_max_search_cands_per_search_cand_loc_populator(
            ps_frm_prms,
            ps_refine_prms->au1_num_fpel_search_cands,
            layer_id,
            ps_me_coding_tools->e_me_quality_presets);

        ps_refine_prms->u1_max_2nx2n_tu_recur_cands = hme_determine_max_2nx2n_tu_recur_cands(
            layer_id, ps_me_coding_tools->e_me_quality_presets);

        ps_refine_prms->u1_max_num_fpel_refine_centers = hme_determine_max_num_fpel_refine_centers(
            layer_id, ps_me_coding_tools->e_me_quality_presets);

        ps_refine_prms->u1_max_num_subpel_refine_centers =
            hme_determine_max_num_subpel_refine_centers(
                layer_id,
                ps_refine_prms->u1_max_subpel_candts_2Nx2N,
                ps_refine_prms->u1_max_subpel_candts_NxN);
    }
    else
    {
        if(0 == layer_id)
        {
            ps_refine_prms->i4_num_results_per_part =
                ps_me_coding_tools->u1_num_results_per_part_in_l0me;
        }
        else if(1 == layer_id)
        {
            ps_refine_prms->i4_num_results_per_part =
                ps_me_coding_tools->u1_num_results_per_part_in_l1me;
        }
        else if((2 == layer_id) && (num_layers > 3))
        {
            ps_refine_prms->i4_num_results_per_part =
                ps_me_coding_tools->u1_num_results_per_part_in_l2me;
        }
        else
        {
            ps_refine_prms->i4_num_results_per_part = hme_determine_num_results_per_part(
                layer_id, num_layers, ps_me_coding_tools->e_me_quality_presets);
        }

        memset(
            ps_refine_prms->au1_num_fpel_search_cands,
            ps_me_coding_tools->u1_max_num_coloc_cands,
            sizeof(ps_refine_prms->au1_num_fpel_search_cands));

        ps_refine_prms->u1_max_2nx2n_tu_recur_cands =
            ps_me_coding_tools->u1_max_2nx2n_tu_recur_cands;

        ps_refine_prms->u1_max_num_fpel_refine_centers =
            ps_me_coding_tools->u1_max_num_fpel_refine_centers;

        ps_refine_prms->u1_max_num_subpel_refine_centers =
            ps_me_coding_tools->u1_max_num_subpel_refine_centers;
    }

    if(layer_id != 0)
    {
        ps_refine_prms->i4_num_mvbank_results = ps_refine_prms->i4_num_results_per_part;
    }

    /* 4 * lambda */
    ps_refine_prms->sdi_threshold =
        (ps_refine_prms->lambda_recon + (1 << (ps_frm_prms->lambda_q_shift - 1))) >>
        (ps_frm_prms->lambda_q_shift - 2);

    ps_refine_prms->u1_use_lambda_derived_from_min_8x8_act_in_ctb =
        MODULATE_LAMDA_WHEN_SPATIAL_MOD_ON && ps_frm_prms->u1_is_cu_qp_delta_enabled;
}

void hme_set_ctb_boundary_attrs(ctb_boundary_attrs_t *ps_attrs, S32 num_8x8_horz, S32 num_8x8_vert)
{
    S32 cu_16x16_valid_flag = 0, merge_pattern_x, merge_pattern_y;
    S32 blk, blk_x, blk_y;
    S32 num_16x16_horz, num_16x16_vert;
    blk_ctb_attrs_t *ps_blk_attrs = &ps_attrs->as_blk_attrs[0];

    num_16x16_horz = (num_8x8_horz + 1) >> 1;
    num_16x16_vert = (num_8x8_vert + 1) >> 1;
    ps_attrs->u1_num_blks_in_ctb = (U08)(num_16x16_horz * num_16x16_vert);

    /*************************************************************************/
    /* Run through each blk assuming all 16x16 CUs valid. The order would be */
    /* 0   1   4   5                                                         */
    /* 2   3   6   7                                                         */
    /* 8   9   12  13                                                        */
    /* 10  11  14  15                                                        */
    /* Out of these some may not be valid. For example, if num_16x16_horz is */
    /* 2 and num_16x16_vert is 4, then right 2 columns not valid. In this    */
    /* case, blks 8-11 get encoding number of 4-7. Further, the variable     */
    /* cu_16x16_valid_flag will be 1111 0000 1111 0000. Also, the variable   */
    /* u1_merge_to_32x32_flag will be 1010, and u1_merge_to_64x64_flag 0     */
    /*************************************************************************/
    for(blk = 0; blk < 16; blk++)
    {
        U08 u1_blk_8x8_mask = 0xF;
        blk_x = gau1_encode_to_raster_x[blk];
        blk_y = gau1_encode_to_raster_y[blk];
        if((blk_x >= num_16x16_horz) || (blk_y >= num_16x16_vert))
        {
            continue;
        }

        /* The CU at encode location blk is valid */
        cu_16x16_valid_flag |= (1 << blk);
        ps_blk_attrs->u1_blk_id_in_full_ctb = blk;
        ps_blk_attrs->u1_blk_x = blk_x;
        ps_blk_attrs->u1_blk_y = blk_y;

        /* Disable blks 1 and 3 if the 16x16 blk overshoots on rt border */
        if(((blk_x << 1) + 2) > num_8x8_horz)
            u1_blk_8x8_mask &= 0x5;
        /* Disable blks 2 and 3 if the 16x16 blk overshoots on bot border */
        if(((blk_y << 1) + 2) > num_8x8_vert)
            u1_blk_8x8_mask &= 0x3;
        ps_blk_attrs->u1_blk_8x8_mask = u1_blk_8x8_mask;
        ps_blk_attrs++;
    }

    ps_attrs->cu_16x16_valid_flag = cu_16x16_valid_flag;

    /* 32x32 merge is logical combination of what merge is possible          */
    /* horizontally as well as vertically.                                   */
    if(num_8x8_horz < 4)
        merge_pattern_x = 0x0;
    else if(num_8x8_horz < 8)
        merge_pattern_x = 0x5;
    else
        merge_pattern_x = 0xF;

    if(num_8x8_vert < 4)
        merge_pattern_y = 0x0;
    else if(num_8x8_vert < 8)
        merge_pattern_y = 0x3;
    else
        merge_pattern_y = 0xF;

    ps_attrs->u1_merge_to_32x32_flag = (U08)(merge_pattern_x & merge_pattern_y);

    /* Do not attempt 64x64 merge if any blk invalid */
    if(ps_attrs->u1_merge_to_32x32_flag != 0xF)
        ps_attrs->u1_merge_to_64x64_flag = 0;
    else
        ps_attrs->u1_merge_to_64x64_flag = 1;
}

void hme_set_ctb_attrs(ctb_boundary_attrs_t *ps_attrs, S32 wd, S32 ht)
{
    S32 is_cropped_rt, is_cropped_bot;

    is_cropped_rt = ((wd & 63) != 0) ? 1 : 0;
    is_cropped_bot = ((ht & 63) != 0) ? 1 : 0;

    if(is_cropped_rt)
    {
        hme_set_ctb_boundary_attrs(&ps_attrs[CTB_RT_PIC_BOUNDARY], (wd & 63) >> 3, 8);
    }
    if(is_cropped_bot)
    {
        hme_set_ctb_boundary_attrs(&ps_attrs[CTB_BOT_PIC_BOUNDARY], 8, (ht & 63) >> 3);
    }
    if(is_cropped_rt & is_cropped_bot)
    {
        hme_set_ctb_boundary_attrs(
            &ps_attrs[CTB_BOT_RT_PIC_BOUNDARY], (wd & 63) >> 3, (ht & 63) >> 3);
    }
    hme_set_ctb_boundary_attrs(&ps_attrs[CTB_CENTRE], 8, 8);
}

/**
********************************************************************************
*  @fn     hme_scale_for_ref_idx(S32 curr_poc, S32 poc_from, S32 poc_to)
*
*  @brief  When we have an mv with ref id "poc_to" for which predictor to be
*          computed, and predictor is ref id "poc_from", this funciton returns
*          scale factor in Q8 for such a purpose
*
*  @param[in] curr_poc : input picture poc
*
*  @param[in] poc_from : POC of the pic, pointed to by ref id to be scaled
*
*  @param[in] poc_to : POC of hte pic, pointed to by ref id to be scaled to
*
*  @return Scale factor in Q8 format
********************************************************************************
*/
S16 hme_scale_for_ref_idx(S32 curr_poc, S32 poc_from, S32 poc_to)
{
    S32 td, tx, tb;
    S16 i2_scf;
    /*************************************************************************/
    /* Approximate scale factor: 256 * num / denom                           */
    /* num = curr_poc - poc_to, denom = curr_poc - poc_from                  */
    /* Exact implementation as per standard.                                 */
    /*************************************************************************/

    tb = HME_CLIP((curr_poc - poc_to), -128, 127);
    td = HME_CLIP((curr_poc - poc_from), -128, 127);

    tx = (16384 + (ABS(td) >> 1)) / td;
    //i2_scf = HME_CLIP((((tb*tx)+32)>>6), -128, 127);
    i2_scf = HME_CLIP((((tb * tx) + 32) >> 6), -4096, 4095);

    return (i2_scf);
}

/**
********************************************************************************
*  @fn     hme_process_frm_init
*
*  @brief  HME frame level initialsation processing function
*
*  @param[in] pv_me_ctxt : ME ctxt pointer
*
*  @param[in] ps_ref_map : Reference map prms pointer
*
*  @param[in] ps_frm_prms :Pointer to frame params
*
*  called only for encode layer
*
*  @return Scale factor in Q8 format
********************************************************************************
*/
void hme_process_frm_init(
    void *pv_me_ctxt,
    hme_ref_map_t *ps_ref_map,
    hme_frm_prms_t *ps_frm_prms,
    WORD32 i4_me_frm_id,
    WORD32 i4_num_me_frm_pllel)
{
    me_ctxt_t *ps_thrd_ctxt = (me_ctxt_t *)pv_me_ctxt;
    me_frm_ctxt_t *ps_ctxt = (me_frm_ctxt_t *)ps_thrd_ctxt->aps_me_frm_prms[i4_me_frm_id];

    S32 i, j, desc_idx;
    S16 i2_max_x = 0, i2_max_y = 0;

    /* Set the Qp of current frm passed by caller. Required for intra cost */
    ps_ctxt->frm_qstep = ps_frm_prms->qstep;
    ps_ctxt->qstep_ls8 = ps_frm_prms->qstep_ls8;

    /* Bidir enabled or not */
    ps_ctxt->s_frm_prms = *ps_frm_prms;

    /*************************************************************************/
    /* Set up the ref pic parameters across all layers. For this, we do the  */
    /* following: the application has given us a ref pic list, we go index   */
    /* by index and pick up the picture. A picture can be uniquely be mapped */
    /* to a POC. So we search all layer descriptor array to find the POC     */
    /* Once found, we update all attributes in this descriptor.              */
    /* During this updation process we also create an index of descriptor id */
    /* to ref id mapping. It is important to find the same POC in the layers */
    /* descr strcture since it holds the pyramid inputs for non encode layers*/
    /* Apart from this, e also update array containing the index of the descr*/
    /* During processing for ease of access, each layer has a pointer to aray*/
    /* of pointers containing fxfy, fxhy, hxfy, hxhy and inputs for each ref */
    /* we update this too.                                                   */
    /*************************************************************************/
    ps_ctxt->num_ref_past = 0;
    ps_ctxt->num_ref_future = 0;
    for(i = 0; i < ps_ref_map->i4_num_ref; i++)
    {
        S32 ref_id_lc, idx;
        hme_ref_desc_t *ps_ref_desc;

        ps_ref_desc = &ps_ref_map->as_ref_desc[i];
        ref_id_lc = ps_ref_desc->i1_ref_id_lc;
        /* Obtain the id of descriptor that contains this POC */
        idx = hme_find_descr_idx(
            ps_thrd_ctxt, ps_ref_desc->i4_poc, ps_ref_desc->i4_GOP_num, i4_num_me_frm_pllel);

        /* Update all layers in this descr with the reference attributes */
        hme_update_layer_desc(
            &ps_thrd_ctxt->as_ref_descr[idx],
            ps_ref_desc,
            0,
            1,  //ps_ctxt->num_layers,
            ps_ctxt->ps_curr_descr);

        /* Update the pointer holder for the recon planes */
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppu1_list_inp = &ps_ctxt->apu1_list_inp[0][0];
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppu1_list_rec_fxfy =
            &ps_ctxt->apu1_list_rec_fxfy[0][0];
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppu1_list_rec_hxfy =
            &ps_ctxt->apu1_list_rec_hxfy[0][0];
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppu1_list_rec_fxhy =
            &ps_ctxt->apu1_list_rec_fxhy[0][0];
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppu1_list_rec_hxhy =
            &ps_ctxt->apu1_list_rec_hxhy[0][0];
        ps_ctxt->ps_curr_descr->aps_layers[0]->ppv_dep_mngr_recon =
            &ps_ctxt->apv_list_dep_mngr[0][0];

        /* Update the array having ref id lc to descr id mapping */
        ps_ctxt->a_ref_to_descr_id[ps_ref_desc->i1_ref_id_lc] = idx;

        /* From ref id lc we need to work out the POC, So update this array */
        ps_ctxt->ai4_ref_idx_to_poc_lc[ref_id_lc] = ps_ref_desc->i4_poc;

        /* When computing costs in L0 and L1 directions, we need the */
        /* respective ref id L0 and L1, so update this mapping */
        ps_ctxt->a_ref_idx_lc_to_l0[ref_id_lc] = ps_ref_desc->i1_ref_id_l0;
        ps_ctxt->a_ref_idx_lc_to_l1[ref_id_lc] = ps_ref_desc->i1_ref_id_l1;
        if((ps_ctxt->i4_curr_poc > ps_ref_desc->i4_poc) || ps_ctxt->i4_curr_poc == 0)
        {
            ps_ctxt->au1_is_past[ref_id_lc] = 1;
            ps_ctxt->ai1_past_list[ps_ctxt->num_ref_past] = ref_id_lc;
            ps_ctxt->num_ref_past++;
        }
        else
        {
            ps_ctxt->au1_is_past[ref_id_lc] = 0;
            ps_ctxt->ai1_future_list[ps_ctxt->num_ref_future] = ref_id_lc;
            ps_ctxt->num_ref_future++;
        }

        if(1 == ps_ctxt->i4_wt_pred_enable_flag)
        {
            /* copy the weight and offsets from current ref desc */
            ps_ctxt->s_wt_pred.a_wpred_wt[ref_id_lc] = ps_ref_desc->i2_weight;

            /* inv weight is stored in Q15 format */
            ps_ctxt->s_wt_pred.a_inv_wpred_wt[ref_id_lc] =
                ((1 << 15) + (ps_ref_desc->i2_weight >> 1)) / ps_ref_desc->i2_weight;
            ps_ctxt->s_wt_pred.a_wpred_off[ref_id_lc] = ps_ref_desc->i2_offset;
        }
        else
        {
            /* store default wt and offset*/
            ps_ctxt->s_wt_pred.a_wpred_wt[ref_id_lc] = WGHT_DEFAULT;

            /* inv weight is stored in Q15 format */
            ps_ctxt->s_wt_pred.a_inv_wpred_wt[ref_id_lc] =
                ((1 << 15) + (WGHT_DEFAULT >> 1)) / WGHT_DEFAULT;

            ps_ctxt->s_wt_pred.a_wpred_off[ref_id_lc] = 0;
        }
    }

    ps_ctxt->ai1_future_list[ps_ctxt->num_ref_future] = -1;
    ps_ctxt->ai1_past_list[ps_ctxt->num_ref_past] = -1;

    /*************************************************************************/
    /* Preparation of the TLU for bits for reference indices.                */
    /* Special case is that of numref = 2. (TEV)                             */
    /* Other cases uses UEV                                                  */
    /*************************************************************************/
    for(i = 0; i < MAX_NUM_REF; i++)
    {
        ps_ctxt->au1_ref_bits_tlu_lc[0][i] = 0;
        ps_ctxt->au1_ref_bits_tlu_lc[1][i] = 0;
    }

    if(ps_ref_map->i4_num_ref == 2)
    {
        ps_ctxt->au1_ref_bits_tlu_lc[0][0] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[1][0] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[0][1] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[1][1] = 1;
    }
    else if(ps_ref_map->i4_num_ref > 2)
    {
        for(i = 0; i < ps_ref_map->i4_num_ref; i++)
        {
            S32 l0, l1;
            l0 = ps_ctxt->a_ref_idx_lc_to_l0[i];
            l1 = ps_ctxt->a_ref_idx_lc_to_l1[i];
            ps_ctxt->au1_ref_bits_tlu_lc[0][i] = gau1_ref_bits[l0];
            ps_ctxt->au1_ref_bits_tlu_lc[1][i] = gau1_ref_bits[l1];
        }
    }

    /*************************************************************************/
    /* Preparation of the scaling factors for reference indices. The scale   */
    /* factor depends on distance of the two ref indices from current input  */
    /* in terms of poc delta.                                                */
    /*************************************************************************/
    for(i = 0; i < ps_ref_map->i4_num_ref; i++)
    {
        for(j = 0; j < ps_ref_map->i4_num_ref; j++)
        {
            S16 i2_scf_q8;
            S32 poc_from, poc_to;

            poc_from = ps_ctxt->ai4_ref_idx_to_poc_lc[j];
            poc_to = ps_ctxt->ai4_ref_idx_to_poc_lc[i];

            i2_scf_q8 = hme_scale_for_ref_idx(ps_ctxt->i4_curr_poc, poc_from, poc_to);
            ps_ctxt->ai2_ref_scf[j + i * MAX_NUM_REF] = i2_scf_q8;
        }
    }

    /*************************************************************************/
    /* We store simplified look ups for 4 hpel planes and inp y plane for    */
    /* every layer and for every ref id in the layer. So update these lookups*/
    /*************************************************************************/
    for(i = 0; i < 1; i++)
    {
        U08 **ppu1_rec_fxfy, **ppu1_rec_hxfy, **ppu1_rec_fxhy, **ppu1_rec_hxhy;
        U08 **ppu1_inp;
        void **ppvlist_dep_mngr;
        layer_ctxt_t *ps_layer_ctxt = ps_ctxt->ps_curr_descr->aps_layers[i];

        ppvlist_dep_mngr = &ps_ctxt->apv_list_dep_mngr[i][0];
        ppu1_rec_fxfy = &ps_ctxt->apu1_list_rec_fxfy[i][0];
        ppu1_rec_hxfy = &ps_ctxt->apu1_list_rec_hxfy[i][0];
        ppu1_rec_fxhy = &ps_ctxt->apu1_list_rec_fxhy[i][0];
        ppu1_rec_hxhy = &ps_ctxt->apu1_list_rec_hxhy[i][0];
        ppu1_inp = &ps_ctxt->apu1_list_inp[i][0];
        for(j = 0; j < ps_ref_map->i4_num_ref; j++)
        {
            hme_ref_desc_t *ps_ref_desc;
            hme_ref_buf_info_t *ps_buf_info;
            layer_ctxt_t *ps_layer;
            S32 ref_id_lc;

            ps_ref_desc = &ps_ref_map->as_ref_desc[j];
            ps_buf_info = &ps_ref_desc->as_ref_info[i];
            ref_id_lc = ps_ref_desc->i1_ref_id_lc;

            desc_idx = ps_ctxt->a_ref_to_descr_id[ref_id_lc];
            ps_layer = ps_thrd_ctxt->as_ref_descr[desc_idx].aps_layers[i];

            ppu1_inp[j] = ps_buf_info->pu1_ref_src;
            ppu1_rec_fxfy[j] = ps_buf_info->pu1_rec_fxfy;
            ppu1_rec_hxfy[j] = ps_buf_info->pu1_rec_hxfy;
            ppu1_rec_fxhy[j] = ps_buf_info->pu1_rec_fxhy;
            ppu1_rec_hxhy[j] = ps_buf_info->pu1_rec_hxhy;
            ppvlist_dep_mngr[j] = ps_buf_info->pv_dep_mngr;

            /* Update the curr descriptors reference pointers here */
            ps_layer_ctxt->ppu1_list_inp[j] = ps_buf_info->pu1_ref_src;
            ps_layer_ctxt->ppu1_list_rec_fxfy[j] = ps_buf_info->pu1_rec_fxfy;
            ps_layer_ctxt->ppu1_list_rec_hxfy[j] = ps_buf_info->pu1_rec_hxfy;
            ps_layer_ctxt->ppu1_list_rec_fxhy[j] = ps_buf_info->pu1_rec_fxhy;
            ps_layer_ctxt->ppu1_list_rec_hxhy[j] = ps_buf_info->pu1_rec_hxhy;
        }
    }
    /*************************************************************************/
    /* The mv range for each layer is computed. For dyadic layers it will    */
    /* keep shrinking by 2, for non dyadic it will shrink by ratio of wd and */
    /* ht. In general formula used is scale by ratio of wd for x and ht for y*/
    /*************************************************************************/
    for(i = 0; i < 1; i++)
    {
        layer_ctxt_t *ps_layer_ctxt;
        if(i == 0)
        {
            i2_max_x = ps_frm_prms->i2_mv_range_x;
            i2_max_y = ps_frm_prms->i2_mv_range_y;
        }
        else
        {
            i2_max_x = (S16)FLOOR8(((i2_max_x * ps_ctxt->i4_wd) / ps_ctxt->i4_wd));
            i2_max_y = (S16)FLOOR8(((i2_max_y * ps_ctxt->i4_ht) / ps_ctxt->i4_ht));
        }
        ps_layer_ctxt = ps_ctxt->ps_curr_descr->aps_layers[i];
        ps_layer_ctxt->i2_max_mv_x = i2_max_x;
        ps_layer_ctxt->i2_max_mv_y = i2_max_y;

        /*********************************************************************/
        /* Every layer maintains a reference id lc to POC mapping. This is   */
        /* because the mapping is unique for every frm. Also, in next frm,   */
        /* we require colocated mvs which means scaling according to temporal*/
        /*distance. Hence this mapping needs to be maintained in every       */
        /* layer ctxt                                                        */
        /*********************************************************************/
        memset(ps_layer_ctxt->ai4_ref_id_to_poc_lc, -1, sizeof(S32) * ps_ctxt->max_num_ref);
        if(ps_ref_map->i4_num_ref)
        {
            memcpy(
                ps_layer_ctxt->ai4_ref_id_to_poc_lc,
                ps_ctxt->ai4_ref_idx_to_poc_lc,
                ps_ref_map->i4_num_ref * sizeof(S32));
        }
    }

    return;
}

/**
********************************************************************************
*  @fn     hme_coarse_process_frm_init
*
*  @brief  HME frame level initialsation processing function
*
*  @param[in] pv_me_ctxt : ME ctxt pointer
*
*  @param[in] ps_ref_map : Reference map prms pointer
*
*  @param[in] ps_frm_prms :Pointer to frame params
*
*  @return Scale factor in Q8 format
********************************************************************************
*/
void hme_coarse_process_frm_init(
    void *pv_me_ctxt, hme_ref_map_t *ps_ref_map, hme_frm_prms_t *ps_frm_prms)
{
    coarse_me_ctxt_t *ps_ctxt = (coarse_me_ctxt_t *)pv_me_ctxt;
    S32 i, j, desc_idx;
    S16 i2_max_x = 0, i2_max_y = 0;

    /* Set the Qp of current frm passed by caller. Required for intra cost */
    ps_ctxt->frm_qstep = ps_frm_prms->qstep;

    /* Bidir enabled or not */
    ps_ctxt->s_frm_prms = *ps_frm_prms;

    /*************************************************************************/
    /* Set up the ref pic parameters across all layers. For this, we do the  */
    /* following: the application has given us a ref pic list, we go index   */
    /* by index and pick up the picture. A picture can be uniquely be mapped */
    /* to a POC. So we search all layer descriptor array to find the POC     */
    /* Once found, we update all attributes in this descriptor.              */
    /* During this updation process we also create an index of descriptor id */
    /* to ref id mapping. It is important to find the same POC in the layers */
    /* descr strcture since it holds the pyramid inputs for non encode layers*/
    /* Apart from this, e also update array containing the index of the descr*/
    /* During processing for ease of access, each layer has a pointer to aray*/
    /* of pointers containing fxfy, fxhy, hxfy, hxhy and inputs for each ref */
    /* we update this too.                                                   */
    /*************************************************************************/
    ps_ctxt->num_ref_past = 0;
    ps_ctxt->num_ref_future = 0;
    for(i = 0; i < ps_ref_map->i4_num_ref; i++)
    {
        S32 ref_id_lc, idx;
        hme_ref_desc_t *ps_ref_desc;

        ps_ref_desc = &ps_ref_map->as_ref_desc[i];
        ref_id_lc = ps_ref_desc->i1_ref_id_lc;
        /* Obtain the id of descriptor that contains this POC */
        idx = hme_coarse_find_descr_idx(ps_ctxt, ps_ref_desc->i4_poc);

        /* Update all layers in this descr with the reference attributes */
        hme_update_layer_desc(
            &ps_ctxt->as_ref_descr[idx],
            ps_ref_desc,
            1,
            ps_ctxt->num_layers - 1,
            ps_ctxt->ps_curr_descr);

        /* Update the array having ref id lc to descr id mapping */
        ps_ctxt->a_ref_to_descr_id[ps_ref_desc->i1_ref_id_lc] = idx;

        /* From ref id lc we need to work out the POC, So update this array */
        ps_ctxt->ai4_ref_idx_to_poc_lc[ref_id_lc] = ps_ref_desc->i4_poc;

        /* From ref id lc we need to work out the display num, So update this array */
        ps_ctxt->ai4_ref_idx_to_disp_num[ref_id_lc] = ps_ref_desc->i4_display_num;

        /* When computing costs in L0 and L1 directions, we need the */
        /* respective ref id L0 and L1, so update this mapping */
        ps_ctxt->a_ref_idx_lc_to_l0[ref_id_lc] = ps_ref_desc->i1_ref_id_l0;
        ps_ctxt->a_ref_idx_lc_to_l1[ref_id_lc] = ps_ref_desc->i1_ref_id_l1;
        if((ps_ctxt->i4_curr_poc > ps_ref_desc->i4_poc) || ps_ctxt->i4_curr_poc == 0)
        {
            ps_ctxt->au1_is_past[ref_id_lc] = 1;
            ps_ctxt->ai1_past_list[ps_ctxt->num_ref_past] = ref_id_lc;
            ps_ctxt->num_ref_past++;
        }
        else
        {
            ps_ctxt->au1_is_past[ref_id_lc] = 0;
            ps_ctxt->ai1_future_list[ps_ctxt->num_ref_future] = ref_id_lc;
            ps_ctxt->num_ref_future++;
        }
        if(1 == ps_ctxt->i4_wt_pred_enable_flag)
        {
            /* copy the weight and offsets from current ref desc */
            ps_ctxt->s_wt_pred.a_wpred_wt[ref_id_lc] = ps_ref_desc->i2_weight;

            /* inv weight is stored in Q15 format */
            ps_ctxt->s_wt_pred.a_inv_wpred_wt[ref_id_lc] =
                ((1 << 15) + (ps_ref_desc->i2_weight >> 1)) / ps_ref_desc->i2_weight;

            ps_ctxt->s_wt_pred.a_wpred_off[ref_id_lc] = ps_ref_desc->i2_offset;
        }
        else
        {
            /* store default wt and offset*/
            ps_ctxt->s_wt_pred.a_wpred_wt[ref_id_lc] = WGHT_DEFAULT;

            /* inv weight is stored in Q15 format */
            ps_ctxt->s_wt_pred.a_inv_wpred_wt[ref_id_lc] =
                ((1 << 15) + (WGHT_DEFAULT >> 1)) / WGHT_DEFAULT;

            ps_ctxt->s_wt_pred.a_wpred_off[ref_id_lc] = 0;
        }
    }

    ps_ctxt->ai1_future_list[ps_ctxt->num_ref_future] = -1;
    ps_ctxt->ai1_past_list[ps_ctxt->num_ref_past] = -1;

    /*************************************************************************/
    /* Preparation of the TLU for bits for reference indices.                */
    /* Special case is that of numref = 2. (TEV)                             */
    /* Other cases uses UEV                                                  */
    /*************************************************************************/
    for(i = 0; i < MAX_NUM_REF; i++)
    {
        ps_ctxt->au1_ref_bits_tlu_lc[0][i] = 0;
        ps_ctxt->au1_ref_bits_tlu_lc[1][i] = 0;
    }

    if(ps_ref_map->i4_num_ref == 2)
    {
        ps_ctxt->au1_ref_bits_tlu_lc[0][0] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[1][0] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[0][1] = 1;
        ps_ctxt->au1_ref_bits_tlu_lc[1][1] = 1;
    }
    else if(ps_ref_map->i4_num_ref > 2)
    {
        for(i = 0; i < ps_ref_map->i4_num_ref; i++)
        {
            S32 l0, l1;
            l0 = ps_ctxt->a_ref_idx_lc_to_l0[i];
            l1 = ps_ctxt->a_ref_idx_lc_to_l1[i];
            ps_ctxt->au1_ref_bits_tlu_lc[0][i] = gau1_ref_bits[l0];
            ps_ctxt->au1_ref_bits_tlu_lc[1][i] = gau1_ref_bits[l1];
        }
    }

    /*************************************************************************/
    /* Preparation of the scaling factors for reference indices. The scale   */
    /* factor depends on distance of the two ref indices from current input  */
    /* in terms of poc delta.                                                */
    /*************************************************************************/
    for(i = 0; i < ps_ref_map->i4_num_ref; i++)
    {
        for(j = 0; j < ps_ref_map->i4_num_ref; j++)
        {
            S16 i2_scf_q8;
            S32 poc_from, poc_to;

            poc_from = ps_ctxt->ai4_ref_idx_to_poc_lc[j];
            poc_to = ps_ctxt->ai4_ref_idx_to_poc_lc[i];

            i2_scf_q8 = hme_scale_for_ref_idx(ps_ctxt->i4_curr_poc, poc_from, poc_to);
            ps_ctxt->ai2_ref_scf[j + i * MAX_NUM_REF] = i2_scf_q8;
        }
    }

    /*************************************************************************/
    /* We store simplified look ups for inp y plane for                      */
    /* every layer and for every ref id in the layer.                        */
    /*************************************************************************/
    for(i = 1; i < ps_ctxt->num_layers; i++)
    {
        U08 **ppu1_inp;

        ppu1_inp = &ps_ctxt->apu1_list_inp[i][0];
        for(j = 0; j < ps_ref_map->i4_num_ref; j++)
        {
            hme_ref_desc_t *ps_ref_desc;
            hme_ref_buf_info_t *ps_buf_info;
            layer_ctxt_t *ps_layer;
            S32 ref_id_lc;

            ps_ref_desc = &ps_ref_map->as_ref_desc[j];
            ps_buf_info = &ps_ref_desc->as_ref_info[i];
            ref_id_lc = ps_ref_desc->i1_ref_id_lc;

            desc_idx = ps_ctxt->a_ref_to_descr_id[ref_id_lc];
            ps_layer = ps_ctxt->as_ref_descr[desc_idx].aps_layers[i];

            ppu1_inp[j] = ps_layer->pu1_inp;
        }
    }
    /*************************************************************************/
    /* The mv range for each layer is computed. For dyadic layers it will    */
    /* keep shrinking by 2, for non dyadic it will shrink by ratio of wd and */
    /* ht. In general formula used is scale by ratio of wd for x and ht for y*/
    /*************************************************************************/

    /* set to layer 0 search range params */
    i2_max_x = ps_frm_prms->i2_mv_range_x;
    i2_max_y = ps_frm_prms->i2_mv_range_y;

    for(i = 1; i < ps_ctxt->num_layers; i++)
    {
        layer_ctxt_t *ps_layer_ctxt;

        {
            i2_max_x = (S16)FLOOR8(((i2_max_x * ps_ctxt->a_wd[i]) / ps_ctxt->a_wd[i - 1]));
            i2_max_y = (S16)FLOOR8(((i2_max_y * ps_ctxt->a_ht[i]) / ps_ctxt->a_ht[i - 1]));
        }
        ps_layer_ctxt = ps_ctxt->ps_curr_descr->aps_layers[i];
        ps_layer_ctxt->i2_max_mv_x = i2_max_x;
        ps_layer_ctxt->i2_max_mv_y = i2_max_y;

        /*********************************************************************/
        /* Every layer maintains a reference id lc to POC mapping. This is   */
        /* because the mapping is unique for every frm. Also, in next frm,   */
        /* we require colocated mvs which means scaling according to temporal*/
        /*distance. Hence this mapping needs to be maintained in every       */
        /* layer ctxt                                                        */
        /*********************************************************************/
        memset(ps_layer_ctxt->ai4_ref_id_to_poc_lc, -1, sizeof(S32) * ps_ctxt->max_num_ref);
        if(ps_ref_map->i4_num_ref)
        {
            memcpy(
                ps_layer_ctxt->ai4_ref_id_to_poc_lc,
                ps_ctxt->ai4_ref_idx_to_poc_lc,
                ps_ref_map->i4_num_ref * sizeof(S32));
            memcpy(
                ps_layer_ctxt->ai4_ref_id_to_disp_num,
                ps_ctxt->ai4_ref_idx_to_disp_num,
                ps_ref_map->i4_num_ref * sizeof(S32));
        }
    }

    return;
}

/**
********************************************************************************
*  @fn     hme_process_frm
*
*  @brief  HME frame level processing function
*
*  @param[in] pv_me_ctxt : ME ctxt pointer
*
*  @param[in] ps_ref_map : Reference map prms pointer
*
*  @param[in] ppd_intra_costs : pointer to array of intra cost cost buffers for each layer
*
*  @param[in] ps_frm_prms : pointer to Frame level parameters of HME
*
*  @param[in] pf_ext_update_fxn : function pointer to update CTb results
*
*  @param[in] pf_get_intra_cu_and_cost :function pointer to get intra cu size and cost
*
*  @param[in] ps_multi_thrd_ctxt :function pointer to get intra cu size and cost
*
*  @return Scale factor in Q8 format
********************************************************************************
*/

void hme_process_frm(
    void *pv_me_ctxt,
    pre_enc_L0_ipe_encloop_ctxt_t *ps_l0_ipe_input,
    hme_ref_map_t *ps_ref_map,
    double **ppd_intra_costs,
    hme_frm_prms_t *ps_frm_prms,
    PF_EXT_UPDATE_FXN_T pf_ext_update_fxn,
    void *pv_coarse_layer,
    void *pv_multi_thrd_ctxt,
    S32 i4_frame_parallelism_level,
    S32 thrd_id,
    S32 i4_me_frm_id)
{
    refine_prms_t s_refine_prms;
    me_ctxt_t *ps_thrd_ctxt = (me_ctxt_t *)pv_me_ctxt;
    me_frm_ctxt_t *ps_ctxt = ps_thrd_ctxt->aps_me_frm_prms[i4_me_frm_id];

    S32 lyr_job_type;
    multi_thrd_ctxt_t *ps_multi_thrd_ctxt;
    layer_ctxt_t *ps_coarse_layer = (layer_ctxt_t *)pv_coarse_layer;

    ps_multi_thrd_ctxt = (multi_thrd_ctxt_t *)pv_multi_thrd_ctxt;

    lyr_job_type = ME_JOB_ENC_LYR;
    /*************************************************************************/
    /* Final L0 layer ME call                                                */
    /*************************************************************************/
    {
        /* Set the CTB attributes dependin on corner/rt edge/bot edge/center*/
        hme_set_ctb_attrs(ps_ctxt->as_ctb_bound_attrs, ps_ctxt->i4_wd, ps_ctxt->i4_ht);

        hme_set_refine_prms(
            &s_refine_prms,
            ps_ctxt->u1_encode[0],
            ps_ref_map->i4_num_ref,
            0,
            ps_ctxt->num_layers,
            ps_ctxt->num_layers_explicit_search,
            ps_thrd_ctxt->s_init_prms.use_4x4,
            ps_frm_prms,
            ppd_intra_costs,
            &ps_thrd_ctxt->s_init_prms.s_me_coding_tools);

        hme_refine(
            ps_thrd_ctxt,
            &s_refine_prms,
            pf_ext_update_fxn,
            ps_coarse_layer,
            ps_multi_thrd_ctxt,
            lyr_job_type,
            thrd_id,
            i4_me_frm_id,
            ps_l0_ipe_input);

        /* Set current ref pic status which will used as perv frame ref pic */
        if(i4_frame_parallelism_level)
        {
            ps_ctxt->i4_is_prev_frame_reference = 0;
        }
        else
        {
            ps_ctxt->i4_is_prev_frame_reference =
                ps_multi_thrd_ctxt->aps_cur_inp_me_prms[i4_me_frm_id]
                    ->ps_curr_inp->s_lap_out.i4_is_ref_pic;
        }
    }

    return;
}

/**
********************************************************************************
*  @fn     hme_coarse_process_frm
*
*  @brief  HME frame level processing function (coarse + refine)
*
*  @param[in] pv_me_ctxt : ME ctxt pointer
*
*  @param[in] ps_ref_map : Reference map prms pointer
*
*  @param[in] ps_frm_prms : pointer to Frame level parameters of HME
*
*  @param[in] ps_multi_thrd_ctxt :Multi thread related ctxt
*
*  @return Scale factor in Q8 format
********************************************************************************
*/

void hme_coarse_process_frm(
    void *pv_me_ctxt,
    hme_ref_map_t *ps_ref_map,
    hme_frm_prms_t *ps_frm_prms,
    void *pv_multi_thrd_ctxt,
    WORD32 i4_ping_pong,
    void **ppv_dep_mngr_hme_sync)
{
    S16 i2_max;
    S32 layer_id;
    coarse_prms_t s_coarse_prms;
    refine_prms_t s_refine_prms;
    coarse_me_ctxt_t *ps_ctxt = (coarse_me_ctxt_t *)pv_me_ctxt;
    S32 lyr_job_type;
    multi_thrd_ctxt_t *ps_multi_thrd_ctxt;

    ps_multi_thrd_ctxt = (multi_thrd_ctxt_t *)pv_multi_thrd_ctxt;
    /*************************************************************************/
    /* Fire processing of all layers, starting with coarsest layer.          */
    /*************************************************************************/
    layer_id = ps_ctxt->num_layers - 1;
    i2_max = ps_ctxt->ps_curr_descr->aps_layers[layer_id]->i2_max_mv_x;
    i2_max = MAX(i2_max, ps_ctxt->ps_curr_descr->aps_layers[layer_id]->i2_max_mv_y);
    s_coarse_prms.i4_layer_id = layer_id;
    {
        S32 log_start_step;
        /* Based on Preset, set the starting step size for Refinement */
        if(ME_MEDIUM_SPEED > ps_ctxt->s_init_prms.s_me_coding_tools.e_me_quality_presets)
        {
            log_start_step = 0;
        }
        else
        {
            log_start_step = 1;
        }

        s_coarse_prms.i4_max_iters = i2_max >> log_start_step;
        s_coarse_prms.i4_start_step = 1 << log_start_step;
    }
    s_coarse_prms.i4_num_ref = ps_ref_map->i4_num_ref;
    s_coarse_prms.do_full_search = 1;
    if(s_coarse_prms.do_full_search)
    {
        /* Set to 2 or 4 */
        if(ps_ctxt->s_init_prms.s_me_coding_tools.e_me_quality_presets < ME_MEDIUM_SPEED)
            s_coarse_prms.full_search_step = HME_COARSE_STEP_SIZE_HIGH_QUALITY;
        else if(ps_ctxt->s_init_prms.s_me_coding_tools.e_me_quality_presets >= ME_MEDIUM_SPEED)
            s_coarse_prms.full_search_step = HME_COARSE_STEP_SIZE_HIGH_SPEED;
    }
    s_coarse_prms.num_results = ps_ctxt->max_num_results_coarse;

    /* Coarse layer uses only 1 lambda, i.e. the one for open loop ME */
    s_coarse_prms.lambda = ps_frm_prms->i4_ol_sad_lambda_qf;
    s_coarse_prms.lambda_q_shift = ps_frm_prms->lambda_q_shift;
    s_coarse_prms.lambda = ((float)s_coarse_prms.lambda * (100.0 - ME_LAMBDA_DISCOUNT) / 100.0);

    hme_coarsest(ps_ctxt, &s_coarse_prms, ps_multi_thrd_ctxt, i4_ping_pong, ppv_dep_mngr_hme_sync);

    /* all refinement layer processed in the loop below */
    layer_id--;
    lyr_job_type = ps_multi_thrd_ctxt->i4_me_coarsest_lyr_type + 1;

    /*************************************************************************/
    /* This loop will run for all refine layers (non- encode layers)          */
    /*************************************************************************/
    while(layer_id > 0)
    {
        hme_set_refine_prms(
            &s_refine_prms,
            ps_ctxt->u1_encode[layer_id],
            ps_ref_map->i4_num_ref,
            layer_id,
            ps_ctxt->num_layers,
            ps_ctxt->num_layers_explicit_search,
            ps_ctxt->s_init_prms.use_4x4,
            ps_frm_prms,
            NULL,
            &ps_ctxt->s_init_prms.s_me_coding_tools);

        hme_refine_no_encode(
            ps_ctxt,
            &s_refine_prms,
            ps_multi_thrd_ctxt,
            lyr_job_type,
            i4_ping_pong,
            ppv_dep_mngr_hme_sync);

        layer_id--;
        lyr_job_type++;
    }
}
/**
********************************************************************************
*  @fn     hme_fill_neighbour_mvs
*
*  @brief  HME neighbour MV population function
*
*  @param[in] pps_mv_grid : MV grid array pointer
*
*  @param[in] i4_ctb_x : CTB pos X

*  @param[in] i4_ctb_y : CTB pos Y
*
*  @remarks :  Needs to be populated for proper implementation of cost fxn
*
*  @return Scale factor in Q8 format
********************************************************************************
*/
void hme_fill_neighbour_mvs(
    mv_grid_t **pps_mv_grid, S32 i4_ctb_x, S32 i4_ctb_y, S32 i4_num_ref, void *pv_ctxt)
{
    /* TODO : Needs to be populated for proper implementation of cost fxn */
    ARG_NOT_USED(pps_mv_grid);
    ARG_NOT_USED(i4_ctb_x);
    ARG_NOT_USED(i4_ctb_y);
    ARG_NOT_USED(i4_num_ref);
    ARG_NOT_USED(pv_ctxt);
}

/**
*******************************************************************************
*  @fn     void hme_get_active_pocs_list(void *pv_me_ctxt,
*                                       S32 *p_pocs_buffered_in_me)
*
*  @brief  Returns the list of active POCs in ME ctxt
*
*  @param[in] pv_me_ctxt : handle to ME context
*
*  @param[out] p_pocs_buffered_in_me : pointer to an array which this fxn
*                                      populates with pocs active
*
*  @return   void
*******************************************************************************
*/
WORD32 hme_get_active_pocs_list(void *pv_me_ctxt, S32 i4_num_me_frm_pllel)
{
    me_ctxt_t *ps_ctxt = (me_ctxt_t *)pv_me_ctxt;
    S32 i, count = 0;

    for(i = 0; i < (ps_ctxt->aps_me_frm_prms[0]->max_num_ref * i4_num_me_frm_pllel) + 1; i++)
    {
        S32 poc = ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_poc;
        S32 i4_is_free = ps_ctxt->as_ref_descr[i].aps_layers[0]->i4_is_free;

        if((i4_is_free == 0) && (poc != INVALID_POC))
        {
            count++;
        }
    }
    if(count == (ps_ctxt->aps_me_frm_prms[0]->max_num_ref * i4_num_me_frm_pllel) + 1)
    {
        return 1;
    }
    else
    {
        return 0;
    }
}

/**
*******************************************************************************
*  @fn     void hme_coarse_get_active_pocs_list(void *pv_me_ctxt,
*                                       S32 *p_pocs_buffered_in_me)
*
*  @brief  Returns the list of active POCs in ME ctxt
*
*  @param[in] pv_me_ctxt : handle to ME context
*
*  @param[out] p_pocs_buffered_in_me : pointer to an array which this fxn
*                                      populates with pocs active
*
*  @return   void
*******************************************************************************
*/
void hme_coarse_get_active_pocs_list(void *pv_me_ctxt, S32 *p_pocs_buffered_in_me)
{
    coarse_me_ctxt_t *ps_ctxt = (coarse_me_ctxt_t *)pv_me_ctxt;
    S32 i, count = 0;

    for(i = 0; i < ps_ctxt->max_num_ref + 1 + NUM_BUFS_DECOMP_HME; i++)
    {
        S32 poc = ps_ctxt->as_ref_descr[i].aps_layers[1]->i4_poc;

        if(poc != -1)
        {
            p_pocs_buffered_in_me[count] = poc;
            count++;
        }
    }
    p_pocs_buffered_in_me[count] = -1;
}

S32 hme_get_blk_size(S32 use_4x4, S32 layer_id, S32 n_layers, S32 encode)
{
    /* coarsest layer uses 4x4 blks, lowermost layer/encode layer uses 16x16 */
    if(layer_id == n_layers - 1)
        return 4;
    else if((layer_id == 0) || (encode))
        return 16;

    /* Intermediate non encode layers use 8 */
    return 8;
}

Coverage Report

Created: 2025-09-17 06:42