xref: /aosp_15_r20/external/libavc/encoder/svc/isvce_ilp_mv.c (revision 495ae853bb871d1e5a258cb02c2cc13cde8ddb9a)

/******************************************************************************
 *
 * Copyright (C) 2022 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
*  isvce_ilp_mv.c
*
* @brief
*  Contains functions used for deriving inter_layer MV's
*
*******************************************************************************
*/
#include <stdint.h>
#include <math.h>
#include <stdbool.h>

#include "ih264_typedefs.h"
#include "ih264_debug.h"
#include "isvc_macros.h"
#include "isvc_defs.h"
#include "isvce_defs.h"
#include "isvce_structs.h"
#include "isvce_ilp_mv_private_defs.h"
#include "isvce_ilp_mv.h"
#include "isvce_ilp_mv_utils.h"

/**
*******************************************************************************
*
* @brief
*  Returns size of buffers for storing ILP MV ctxt
*
* @param[in] u1_num_spatial_layers
*  Num Spatial Layers
*
* @param[in] d_spatial_res_ratio
*  Resolution Ratio b/w spatial layers
*
* @param[in] u4_wd
*  Input Width
*
* @param[in] u4_ht
*  Input Height
*
* @returns  Size of buffers
*
*******************************************************************************
*/
UWORD32 isvce_get_ilp_mv_ctxt_size(UWORD8 u1_num_spatial_layers, DOUBLE d_spatial_res_ratio,
                                   UWORD32 u4_wd, UWORD32 u4_ht)
{
    UWORD32 u4_size = 0;

    if(u1_num_spatial_layers > 1)
    {
        WORD32 i;

        u4_size += MAX_PROCESS_CTXT * sizeof(svc_ilp_mv_ctxt_t);
        u4_size += MAX_PROCESS_CTXT * sizeof(ilp_mv_state_t);

        u4_size += u1_num_spatial_layers * sizeof(ilp_mv_layer_state_t);

        for(i = u1_num_spatial_layers - 1; i >= 1; i--)
        {
            WORD32 i4_layer_luma_wd =
                (WORD32) ((DOUBLE) u4_wd /
                          pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) +
                0.99;
            WORD32 i4_layer_luma_ht =
                ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) + 0.99;
            WORD32 i4_layer_luma_mbs = (i4_layer_luma_wd / MB_SIZE) * (i4_layer_luma_ht / MB_SIZE);

            u4_size += i4_layer_luma_mbs * sizeof(ilp_mv_mb_state_t);
        }
    }

    return u4_size;
}

static FORCEINLINE void isvce_ref_layer_pu_and_mb_pos_init(layer_resampler_props_t *ps_layer_props,
                                                           ilp_mv_mb_state_t *ps_mb_state,
                                                           coordinates_t *ps_mb_pos,
                                                           UWORD32 u4_ref_wd, UWORD32 u4_ref_ht,
                                                           UWORD8 u1_field_pic_flag,
                                                           UWORD8 u1_field_mb_flag)
{
    UWORD32 i, j;

    coordinates_t(*aps_pu_positions)[MAX_PU_IN_MB_ROW] = ps_mb_state->as_pu_positions;
    coordinates_t(*aps_mb_positions)[MAX_PU_IN_MB_ROW] = ps_mb_state->as_mb_positions;

    for(i = 0; i < MAX_PU_IN_MB_COL; i++)
    {
        UWORD32 u4_y_ref16;

        UWORD32 u4_yc = ps_mb_pos->i4_ordinate * ps_layer_props->u4_mb_ht +
                        (4 * i + 1) * (1 + u1_field_mb_flag - u1_field_pic_flag);

        u4_y_ref16 =
            (u4_yc * ps_layer_props->u4_scale_y + (1 << (ps_layer_props->u4_shift_y - 1))) >>
            ps_layer_props->u4_shift_y;
        u4_y_ref16 = MIN(u4_y_ref16, u4_ref_ht - 1);

        for(j = 0; j < MAX_PU_IN_MB_ROW; j++)
        {
            UWORD32 u4_x_ref16;

            UWORD32 u4_xc = ps_mb_pos->i4_abscissa * ps_layer_props->u4_mb_wd + 4 * j + 1;

            u4_x_ref16 =
                (u4_xc * ps_layer_props->u4_scale_x + (1 << (ps_layer_props->u4_shift_x - 1))) >>
                ps_layer_props->u4_shift_x;
            u4_x_ref16 = MIN(u4_x_ref16, u4_ref_wd - 1);

            aps_pu_positions[i][j].i4_abscissa = u4_x_ref16;
            aps_pu_positions[i][j].i4_ordinate = u4_y_ref16;

            aps_mb_positions[i][j].i4_abscissa = (u4_x_ref16 / MB_SIZE);
            aps_mb_positions[i][j].i4_ordinate = (u4_y_ref16 / MB_SIZE);
        }
    }
}

static void isvce_ilp_mv_layer_state_init(ilp_mv_layer_state_t *ps_layer_state,
                                          DOUBLE d_spatial_res_ratio, UWORD32 u4_wd, UWORD32 u4_ht)
{
    UWORD32 i, j;

    const UWORD8 u1_ref_layer_field_pic_flag = 0;
    const UWORD8 u1_field_pic_flag = 0;
    const UWORD8 u1_field_mb_flag = 0;

    ilp_mv_mb_state_t *ps_mb_states;
    layer_resampler_props_t *ps_layer_props;

    UWORD32 u4_wd_in_mbs;
    UWORD32 u4_ht_in_mbs;

    UWORD32 u4_ref_wd = (u4_wd / d_spatial_res_ratio);
    UWORD32 u4_ref_ht = (u4_ht / d_spatial_res_ratio) * (1 + u1_ref_layer_field_pic_flag);
    UWORD32 u4_scaled_wd = u4_wd;
    UWORD32 u4_scaled_ht = u4_ht * (1 + u1_field_pic_flag);

    ps_mb_states = ps_layer_state->ps_mb_states;
    ps_layer_props = ps_layer_state->ps_props;

    u4_wd_in_mbs = u4_scaled_wd / ps_layer_props->u4_mb_wd;
    u4_ht_in_mbs = u4_scaled_ht / ps_layer_props->u4_mb_ht;

    ps_layer_state->s_mv_scale.i4_abscissa = ((u4_scaled_wd << 16) + (u4_ref_wd >> 1)) / u4_ref_wd;
    ps_layer_state->s_mv_scale.i4_ordinate = ((u4_scaled_ht << 16) + (u4_ref_ht >> 1)) / u4_ref_ht;

    for(i = 0; i < u4_ht_in_mbs; i++)
    {
        for(j = 0; j < u4_wd_in_mbs; j++)
        {
            coordinates_t s_mb_pos = {j, i};

            isvce_ref_layer_pu_and_mb_pos_init(ps_layer_props, &ps_mb_states[j + i * u4_wd_in_mbs],
                                               &s_mb_pos, u4_ref_wd, u4_ref_ht, u1_field_pic_flag,
                                               u1_field_mb_flag);
        }
    }
}

/**
*******************************************************************************
*
* @brief
*  Function to initialize svc ilp buffers
*
* @param[in] ps_codec
*  Pointer to codec context
*
* @param[in] ps_mem_rec
*  Pointer to memory allocated for input buffers
*
*******************************************************************************
*/
void isvce_ilp_mv_ctxt_init(isvce_codec_t *ps_codec, iv_mem_rec_t *ps_mem_rec)
{
    WORD32 i, j;

    const WORD32 i4_num_proc_ctxts = sizeof(ps_codec->as_process) / sizeof(ps_codec->as_process[0]);
    UWORD8 u1_num_spatial_layers = ps_codec->s_cfg.s_svc_params.u1_num_spatial_layers;

    if(u1_num_spatial_layers > 1)
    {
        ilp_mv_layer_state_t *ps_layer_states = NULL;
        ilp_mv_mb_state_t *aps_luma_mb_states[MAX_NUM_SPATIAL_LAYERS];

        DOUBLE d_spatial_res_ratio = ps_codec->s_cfg.s_svc_params.d_spatial_res_ratio;
        UWORD32 u4_wd = ps_codec->s_cfg.u4_wd;
        UWORD32 u4_ht = ps_codec->s_cfg.u4_ht;
        UWORD8 *pu1_buf = ps_mem_rec->pv_base;
        WORD64 i8_alloc_mem_size =
            isvce_get_ilp_mv_ctxt_size(u1_num_spatial_layers, d_spatial_res_ratio, u4_wd, u4_ht);

        for(i = 0; i < i4_num_proc_ctxts; i++)
        {
            ilp_mv_state_t *ps_ilp_mv_state;
            svc_ilp_mv_ctxt_t *ps_ilp_mv_ctxt;

            isvce_process_ctxt_t *ps_proc = ps_codec->as_process + i;

            ps_ilp_mv_ctxt = ps_proc->ps_svc_ilp_mv_ctxt = (svc_ilp_mv_ctxt_t *) pu1_buf;
            pu1_buf += sizeof(svc_ilp_mv_ctxt_t);
            i8_alloc_mem_size -= sizeof(svc_ilp_mv_ctxt_t);

            ps_ilp_mv_ctxt->s_ilp_mv_constants.pv_state = pu1_buf;
            ps_ilp_mv_state = (ilp_mv_state_t *) pu1_buf;
            pu1_buf += sizeof(ilp_mv_state_t);
            i8_alloc_mem_size -= sizeof(ilp_mv_state_t);

            if(0 == i)
            {
                ps_ilp_mv_state->ps_layer_state = (ilp_mv_layer_state_t *) pu1_buf;
                ps_layer_states = ps_ilp_mv_state->ps_layer_state;
                pu1_buf += u1_num_spatial_layers * sizeof(ps_ilp_mv_state->ps_layer_state[0]);
                i8_alloc_mem_size -=
                    u1_num_spatial_layers * sizeof(ps_ilp_mv_state->ps_layer_state[0]);
            }
            else
            {
                ps_ilp_mv_state->ps_layer_state = ps_layer_states;
            }

            ASSERT(i8_alloc_mem_size >= 0);

            if(0 == i)
            {
                for(j = u1_num_spatial_layers - 1; j >= 1; j--)
                {
                    ilp_mv_layer_state_t *ps_layer = &ps_ilp_mv_state->ps_layer_state[j];

                    WORD32 i4_layer_luma_wd =
                        ((DOUBLE) u4_wd / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
                        0.99;
                    WORD32 i4_layer_luma_ht =
                        ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
                        0.99;
                    WORD32 i4_layer_luma_mbs =
                        (i4_layer_luma_wd / MB_SIZE) * (i4_layer_luma_ht / MB_SIZE);

                    ps_layer->ps_mb_states = (ilp_mv_mb_state_t *) pu1_buf;
                    aps_luma_mb_states[j] = ps_layer->ps_mb_states;
                    pu1_buf += i4_layer_luma_mbs * sizeof(ps_layer->ps_mb_states[0]);
                    i8_alloc_mem_size -= u1_num_spatial_layers * sizeof(ps_layer->ps_mb_states[0]);

                    ASSERT(i8_alloc_mem_size >= 0);
                    /* Asserts below verify that
                     * 'ps_codec->s_svc_ilp_data.aps_layer_resampler_props' is initialised
                     */
                    ASSERT(ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j].u4_mb_wd ==
                           MB_SIZE);

                    ps_layer->ps_props = &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j];

                    isvce_ilp_mv_layer_state_init(ps_layer, d_spatial_res_ratio, i4_layer_luma_wd,
                                                  i4_layer_luma_ht);
                }
            }
            else
            {
                for(j = u1_num_spatial_layers - 1; j >= 1; j--)
                {
                    ilp_mv_layer_state_t *ps_layer = &ps_ilp_mv_state->ps_layer_state[j];

                    ps_layer->ps_mb_states = aps_luma_mb_states[j];

                    ps_layer->ps_props = &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j];
                }
            }
        }
    }
    else
    {
        for(i = 0; i < i4_num_proc_ctxts; i++)
        {
            ps_codec->as_process[i].ps_svc_ilp_mv_ctxt = NULL;
        }
    }
}

static void isvce_get_ilp_mvs_for_me(svc_ilp_mv_ctxt_t *ps_ilp_mv_ctxt)
{
    svc_layer_data_t *ps_ref_layer_data;
    ilp_mv_layer_state_t *ps_layer_state;
    ilp_mv_mb_state_t *ps_mb_state;
    isvce_mb_info_t *ps_ref_mb_info;
    coordinates_t s_frame_dims;
    coordinates_t s_frame_dims_in_mbs;
    coordinates_t s_ref_frame_dims;
    coordinates_t s_ref_frame_dims_in_mbs;

    bool b_is_mv_non_identical;
    WORD32 i, j, k;

    ilp_mv_constants_t *ps_ilp_mv_constants = &ps_ilp_mv_ctxt->s_ilp_mv_constants;
    ilp_mv_variables_t *ps_ilp_mv_variables = &ps_ilp_mv_ctxt->s_ilp_mv_variables;
    ilp_mv_outputs_t *ps_ilp_mv_outputs = &ps_ilp_mv_ctxt->s_ilp_mv_outputs;
    ilp_mv_state_t *ps_ilp_mv_state = (ilp_mv_state_t *) ps_ilp_mv_constants->pv_state;
    svc_ilp_data_t *ps_svc_ilp_data = ps_ilp_mv_variables->ps_svc_ilp_data;
    svc_au_data_t *ps_svc_au_data = ps_svc_ilp_data->ps_svc_au_data;
    coordinates_t *ps_mb_pos = &ps_ilp_mv_variables->s_mb_pos;
    const isvce_enc_pu_mv_t s_default_mv = {{0, 0}, -1};

    UWORD8 u1_spatial_layer_id = ps_ilp_mv_variables->u1_spatial_layer_id;
    WORD32 i4_num_ilp_mvs = 0;

    s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
    s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
    s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
    s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
    s_ref_frame_dims.i4_abscissa =
        ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_width;
    s_ref_frame_dims.i4_ordinate =
        ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_height;
    s_ref_frame_dims_in_mbs.i4_abscissa = s_ref_frame_dims.i4_abscissa / MB_SIZE;
    s_ref_frame_dims_in_mbs.i4_ordinate = s_ref_frame_dims.i4_ordinate / MB_SIZE;

    ps_ref_layer_data = &ps_svc_au_data->ps_svc_layer_data[u1_spatial_layer_id - 1];
    ps_layer_state = &ps_ilp_mv_state->ps_layer_state[u1_spatial_layer_id];
    ps_mb_state =
        &ps_layer_state->ps_mb_states[ps_mb_pos->i4_abscissa +
                                      ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa];

    for(i = 0; i < MAX_PU_IN_MB_COL; i++)
    {
        for(j = 0; j < MAX_PU_IN_MB_ROW; j++)
        {
            b_is_mv_non_identical = true;

            ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0] = s_default_mv;
            ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1] = s_default_mv;

            ps_ref_mb_info =
                &ps_ref_layer_data->ps_mb_info[ps_mb_state->as_mb_positions[i][j].i4_abscissa +
                                               ps_mb_state->as_mb_positions[i][j].i4_ordinate *
                                                   s_ref_frame_dims_in_mbs.i4_abscissa];

            if((ps_ref_mb_info->u2_mb_type == P16x16) || (ps_ref_mb_info->u2_mb_type == B16x16))
            {
                ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs] =
                    ps_ref_mb_info->u2_mb_type;

                ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] =
                    ps_ref_mb_info->as_pu->u1_pred_mode;

                if(ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] != L0)
                {
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1] =
                        ps_ref_mb_info->as_pu->as_me_info[L1];

                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvx =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvx *
                             ps_layer_state->s_mv_scale.i4_abscissa +
                         32768) >>
                        16;
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvy =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvy *
                             ps_layer_state->s_mv_scale.i4_ordinate +
                         32768) >>
                        16;
                }

                if(ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] != L1)
                {
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0] =
                        ps_ref_mb_info->as_pu->as_me_info[L0];

                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvx =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvx *
                             ps_layer_state->s_mv_scale.i4_abscissa +
                         32768) >>
                        16;
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvy =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvy *
                             ps_layer_state->s_mv_scale.i4_ordinate +
                         32768) >>
                        16;
                }

                if(i4_num_ilp_mvs == 0)
                {
                    i4_num_ilp_mvs++;
                }
                else
                {
                    for(k = i4_num_ilp_mvs - 1; k >= 0; k--)
                    {
                        if((ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[k] ==
                            ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs]) &&
                           (ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[k] ==
                            ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs]) &&
                           isvce_check_identical_mv(
                               ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[k],
                               ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs],
                               ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[k]))
                        {
                            b_is_mv_non_identical = false;
                        }
                    }

                    if(b_is_mv_non_identical)
                    {
                        i4_num_ilp_mvs++;
                    }
                }
            }
            else
            {
                ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs] = INVALID_MB_TYPE;
            }
        }
    }

    ps_ilp_mv_outputs->s_ilp_me_cands.u4_num_ilp_mvs = i4_num_ilp_mvs;

    for(i = 0; i < MAX_ILP_MV_IN_NBR_RGN; i++)
    {
        b_is_mv_non_identical = true;

        ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0] = s_default_mv;
        ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1] = s_default_mv;

        if(ps_mb_pos->i4_abscissa + gai1_nbr_ilp_mv_map[i][0] >= 0 &&
           ps_mb_pos->i4_abscissa + gai1_nbr_ilp_mv_map[i][0] < s_frame_dims_in_mbs.i4_abscissa &&
           ps_mb_pos->i4_ordinate + gai1_nbr_ilp_mv_map[i][1] >= 0 &&
           ps_mb_pos->i4_ordinate + gai1_nbr_ilp_mv_map[i][1] < s_frame_dims_in_mbs.i4_ordinate)
        {
            ps_mb_state =
                &ps_layer_state->ps_mb_states[(ps_mb_pos->i4_abscissa + gai1_nbr_ilp_mv_map[i][0]) +
                                              (ps_mb_pos->i4_ordinate + gai1_nbr_ilp_mv_map[i][1]) *
                                                  s_frame_dims_in_mbs.i4_abscissa];

            ps_ref_mb_info =
                &ps_ref_layer_data->ps_mb_info[(ps_mb_state
                                                    ->as_mb_positions[gai1_nbr_ilp_mv_map[i][2]]
                                                                     [gai1_nbr_ilp_mv_map[i][3]]
                                                    .i4_abscissa) +
                                               ps_mb_state
                                                       ->as_mb_positions[gai1_nbr_ilp_mv_map[i][2]]
                                                                        [gai1_nbr_ilp_mv_map[i][3]]
                                                       .i4_ordinate *
                                                   s_ref_frame_dims_in_mbs.i4_abscissa];

            if((ps_ref_mb_info->u2_mb_type == P16x16) || (ps_ref_mb_info->u2_mb_type == B16x16))
            {
                ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs] =
                    ps_ref_mb_info->u2_mb_type;

                ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] =
                    ps_ref_mb_info->as_pu->u1_pred_mode;

                if(ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] != L0)
                {
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1] =
                        ps_ref_mb_info->as_pu->as_me_info[L1];

                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvx =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvx *
                             ps_layer_state->s_mv_scale.i4_abscissa +
                         32768) >>
                        16;
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvy =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L1].s_mv.i2_mvy *
                             ps_layer_state->s_mv_scale.i4_ordinate +
                         32768) >>
                        16;
                }

                if(ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs] != L1)
                {
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0] =
                        ps_ref_mb_info->as_pu->as_me_info[L0];

                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvx =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvx *
                             ps_layer_state->s_mv_scale.i4_abscissa +
                         32768) >>
                        16;
                    ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvy =
                        (ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs][L0].s_mv.i2_mvy *
                             ps_layer_state->s_mv_scale.i4_ordinate +
                         32768) >>
                        16;
                }

                if(i4_num_ilp_mvs == 0)
                {
                    i4_num_ilp_mvs++;
                }
                else
                {
                    for(k = i4_num_ilp_mvs - 1; k >= 0; k--)
                    {
                        if((ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[k] ==
                            ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs]) &&
                           (ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[k] ==
                            ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[i4_num_ilp_mvs]) &&
                           isvce_check_identical_mv(
                               ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[k],
                               ps_ilp_mv_outputs->s_ilp_me_cands.as_mv[i4_num_ilp_mvs],
                               ps_ilp_mv_outputs->s_ilp_me_cands.ae_pred_mode[k]))
                            b_is_mv_non_identical = false;
                    }

                    if(b_is_mv_non_identical)
                    {
                        i4_num_ilp_mvs++;
                    }
                }
            }
            else
            {
                ps_ilp_mv_outputs->s_ilp_me_cands.e_mb_type[i4_num_ilp_mvs] = INVALID_MB_TYPE;
            }
        }
    }

    ps_ilp_mv_outputs->s_ilp_me_cands.u4_num_ilp_mvs_incl_nbrs = i4_num_ilp_mvs;
}

void isvce_get_mb_ilp_mv(svc_ilp_mv_ctxt_t *ps_ilp_mv_ctxt)
{
    svc_layer_data_t *ps_ref_layer_data;
    ilp_mv_layer_state_t *ps_layer_state;
    ilp_mv_mb_state_t *ps_mb_state;
    isvce_mb_info_t *ps_ref_mb_info;
    coordinates_t s_frame_dims;
    coordinates_t s_frame_dims_in_mbs;
    coordinates_t s_ref_frame_dims;
    coordinates_t s_ref_frame_dims_in_mbs;

    WORD32 i, j;

    ilp_mv_constants_t *ps_ilp_mv_constants = &ps_ilp_mv_ctxt->s_ilp_mv_constants;
    ilp_mv_variables_t *ps_ilp_mv_variables = &ps_ilp_mv_ctxt->s_ilp_mv_variables;
    ilp_mv_outputs_t *ps_ilp_mv_outputs = &ps_ilp_mv_ctxt->s_ilp_mv_outputs;
    ilp_mv_state_t *ps_ilp_mv_state = (ilp_mv_state_t *) ps_ilp_mv_constants->pv_state;
    svc_ilp_data_t *ps_svc_ilp_data = ps_ilp_mv_variables->ps_svc_ilp_data;
    svc_au_data_t *ps_svc_au_data = ps_svc_ilp_data->ps_svc_au_data;
    coordinates_t *ps_mb_pos = &ps_ilp_mv_variables->s_mb_pos;
    const isvce_enc_pu_mv_t s_default_mv = {{0, 0}, -1};

    UWORD8 u1_spatial_layer_id = ps_ilp_mv_variables->u1_spatial_layer_id;

    s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
    s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
    s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
    s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
    s_ref_frame_dims.i4_abscissa =
        ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_width;
    s_ref_frame_dims.i4_ordinate =
        ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_height;
    s_ref_frame_dims_in_mbs.i4_abscissa = s_ref_frame_dims.i4_abscissa / MB_SIZE;
    s_ref_frame_dims_in_mbs.i4_ordinate = s_ref_frame_dims.i4_ordinate / MB_SIZE;

    ps_ref_layer_data = &ps_svc_au_data->ps_svc_layer_data[u1_spatial_layer_id - 1];
    ps_layer_state = &ps_ilp_mv_state->ps_layer_state[u1_spatial_layer_id];
    ps_mb_state =
        &ps_layer_state->ps_mb_states[ps_mb_pos->i4_abscissa +
                                      ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa];

    ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0] = s_default_mv;
    ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1] = s_default_mv;

    ps_ref_mb_info = &ps_ref_layer_data->ps_mb_info[ps_mb_state->as_mb_positions[0][0].i4_abscissa +
                                                    ps_mb_state->as_mb_positions[0][0].i4_ordinate *
                                                        s_ref_frame_dims_in_mbs.i4_abscissa];

    if((ps_ref_mb_info->u2_mb_type == P16x16) || (ps_ref_mb_info->u2_mb_type == B16x16))
    {
        ps_ilp_mv_outputs->s_ilp_mv.e_mb_type = ps_ref_mb_info->u2_mb_type;

        ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] = ps_ref_mb_info->as_pu->u1_pred_mode;

        if(ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] != L0)
        {
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1] = ps_ref_mb_info->as_pu->as_me_info[L1];
        }

        if(ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] != L1)
        {
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0] = ps_ref_mb_info->as_pu->as_me_info[L0];
        }
    }
    else
    {
        ps_ilp_mv_outputs->s_ilp_mv.e_mb_type = INVALID_MB_TYPE;
    }

    /* Function call to get non 16x16 ilp mvs for me candidates */
    isvce_get_ilp_mvs_for_me(ps_ilp_mv_ctxt);

    /* Encoder supports only 16x16 partition. */
    /* The code below ensures only 16x16 ILP MV's are used */
    for(i = 0; i < MAX_PU_IN_MB_COL; i++)
    {
        for(j = 0; j < MAX_PU_IN_MB_ROW; j++)
        {
            bool b_unsupported_mv;

            ps_ref_mb_info =
                &ps_ref_layer_data->ps_mb_info[ps_mb_state->as_mb_positions[i][j].i4_abscissa +
                                               ps_mb_state->as_mb_positions[i][j].i4_ordinate *
                                                   s_ref_frame_dims_in_mbs.i4_abscissa];

            b_unsupported_mv =
                (ps_ref_mb_info->u2_mb_type != ps_ilp_mv_outputs->s_ilp_mv.e_mb_type) ||
                (ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] !=
                 ps_ref_mb_info->as_pu->u1_pred_mode) ||
                !isvce_check_identical_mv(ps_ilp_mv_outputs->s_ilp_mv.as_mv[0],
                                          ps_ref_mb_info->as_pu->as_me_info,
                                          ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0]);

            if(b_unsupported_mv)
            {
                ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0] = s_default_mv;
                ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1] = s_default_mv;
                ps_ilp_mv_outputs->s_ilp_mv.e_mb_type = INVALID_MB_TYPE;

                return;
            }
        }
    }

    if(ps_ilp_mv_outputs->s_ilp_mv.e_mb_type != INVALID_MB_TYPE)
    {
        if(ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] != L0)
        {
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1].s_mv.i2_mvx =
                (ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1].s_mv.i2_mvx *
                     ps_layer_state->s_mv_scale.i4_abscissa +
                 32768) >>
                16;
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1].s_mv.i2_mvy =
                (ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L1].s_mv.i2_mvy *
                     ps_layer_state->s_mv_scale.i4_ordinate +
                 32768) >>
                16;
        }

        if(ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] != L1)
        {
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0].s_mv.i2_mvx =
                (ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0].s_mv.i2_mvx *
                     ps_layer_state->s_mv_scale.i4_abscissa +
                 32768) >>
                16;
            ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0].s_mv.i2_mvy =
                (ps_ilp_mv_outputs->s_ilp_mv.as_mv[0][L0].s_mv.i2_mvy *
                     ps_layer_state->s_mv_scale.i4_ordinate +
                 32768) >>
                16;
        }
    }
    else
    {
        ps_ilp_mv_outputs->s_ilp_mv.e_mb_type = INVALID_MB_TYPE;
        ps_ilp_mv_outputs->s_ilp_mv.ae_pred_mode[0] = INVALID_PRED_MODE;
    }
}

void isvce_mvp_idx_eval(isvce_mb_info_t *ps_mb_info, isvce_enc_pu_mv_t *ps_spatial_mvp,
                        isvce_enc_pu_mv_t *ps_ilp_mvp, UWORD8 *pu1_mvd_costs)
{
    if(USE_ILP_MV_AS_MVP && ps_ilp_mvp && !ps_mb_info->u1_is_intra &&
       (ps_mb_info->u2_mb_type != PSKIP) && (ps_mb_info->u2_mb_type != BSKIP) &&
       (ps_mb_info->u2_mb_type != BASE_MODE))
    {
        isvce_enc_pu_mv_t *ps_mv;
        isvce_enc_pu_mv_t *aps_mvps[2];

        WORD32 ai4_mvd_costs[2];
        WORD32 i, j;

        for(i = 0; i < NUM_PRED_DIRS; i++)
        {
            PRED_MODE_T e_pred_mode = (PRED_MODE_T) i;
            PRED_MODE_T e_cmpl_pred_mode = (e_pred_mode == L0) ? L1 : L0;

            if(ps_mb_info->as_pu->u1_pred_mode != e_pred_mode)
            {
                ps_mv = &ps_mb_info->as_pu->as_me_info[e_cmpl_pred_mode];
                aps_mvps[0] = &ps_spatial_mvp[e_cmpl_pred_mode];
                aps_mvps[1] = &ps_ilp_mvp[e_cmpl_pred_mode];

                for(j = 0; j < 2; j++)
                {
                    if((aps_mvps[j]->i1_ref_idx != -1) &&
                       (!j || ((j == 1) && (ps_mv->i1_ref_idx == aps_mvps[j]->i1_ref_idx))))
                    {
                        ai4_mvd_costs[j] =
                            pu1_mvd_costs[ps_mv->s_mv.i2_mvx - aps_mvps[j]->s_mv.i2_mvx] +
                            pu1_mvd_costs[ps_mv->s_mv.i2_mvy - aps_mvps[j]->s_mv.i2_mvy];
                    }
                    else
                    {
                        ai4_mvd_costs[j] = INT32_MAX;
                    }
                }

                ps_mb_info->as_pu->au1_mvp_idx[e_cmpl_pred_mode] =
                    ai4_mvd_costs[0] > ai4_mvd_costs[1];
            }
            else
            {
                ps_mb_info->as_pu->au1_mvp_idx[e_cmpl_pred_mode] = 0;
            }
        }
    }
    else
    {
        ps_mb_info->as_pu->au1_mvp_idx[L0] = 0;
        ps_mb_info->as_pu->au1_mvp_idx[L1] = 0;
    }
}