xref: /aosp_15_r20/external/libavc/encoder/svc/isvce_me.h (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_me.h
 *
 * @brief
 *  Contains declarations of global variables for H264 encoder
 *
 * @author
 *  ittiam
 *
 * @remarks
 *
 *******************************************************************************
 */

#ifndef _ISVCE_ME_H_
#define _ISVCE_ME_H_

#include "ih264_typedefs.h"

#include "isvce_structs.h"

/*****************************************************************************/
/* Constant Macros                                                           */
/*****************************************************************************/

/**
******************************************************************************
*  @brief     Skip Bias value for P slice
******************************************************************************
*/
#define SKIP_BIAS_P 0

/**
******************************************************************************
*  @brief     Skip Bias value for B slice
******************************************************************************
*/
#define SKIP_BIAS_B 0

/*****************************************************************************/
/* Function Macros                                                           */
/*****************************************************************************/

/**
 ******************************************************************************
 *  @brief      compute median of 3 elements (a, b, c) and store the output
 *  in to result. This is used for mv prediction
 ******************************************************************************
 */

#define MEDIAN(a, b, c, result) \
    if(a > b)                   \
    {                           \
        if(b > c)               \
            result = b;         \
        else                    \
        {                       \
            if(a > c)           \
                result = c;     \
            else                \
                result = a;     \
        }                       \
    }                           \
    else                        \
    {                           \
        if(c > b)               \
            result = b;         \
        else                    \
        {                       \
            if(c > a)           \
                result = c;     \
            else                \
                result = a;     \
        }                       \
    }

/*****************************************************************************/
/* Extern Function Declarations                                              */
/*****************************************************************************/

/**
 *******************************************************************************
 *
 * @brief
 *  This function populates the length of the codewords for motion vectors in
 *the range (-search range, search range) in pixels
 *
 * @param[in] ps_me
 *  Pointer to me ctxt
 *
 * @param[out] pu1_mv_bits
 *  length of the codeword for all mv's
 *
 * @remarks The length of the code words are derived from signed exponential
 * goloumb codes.
 *
 *******************************************************************************
 */
void isvce_init_mv_bits(isvce_me_ctxt_t *ps_me);

/**
 *******************************************************************************
 *
 * @brief The function computes the parameters for a P skip MB
 *
 * @par Description:
 *  The function computes the parameters for a P skip MB
 *
 * @param[in] ps_proc
 *  Process context
 *
 * @param[in] u4_for_me
 *  Flag to indicate the purpose of computing skip
 *
 * @param[out] ps_pred_mv
 *  Flag to indicate the current active refernce list
 *
 * @returns
 *       1) Updates skip MV in proc
 *       2) Returns if the current MB can be coded as skip or not
 *
 * @remarks The code implements the logic as described in sec 8.4.1.1 in H264
 *   specification.
 *
 *******************************************************************************
 */
FT_FIND_SKIP_PARAMS isvce_find_pskip_params;

/**
 *******************************************************************************
 *
 * @brief The function computes the parameters for a P skip MB
 *
 * @par Description:
 *  The function computes the parameters for a P skip MB
 *
 * @param[in] ps_proc
 *  Process context
 *
 * @param[in] u4_for_me
 *  Flag to indicate the purpose of computing skip
 *
 * @param[out] ps_pred_mv
 *  Flag to indicate the current active refernce list
 *
 * @returns
 *       1) Updates skip MV in proc
 *       2) Returns if the current MB can be coded as skip or not
 *
 * @remarks The code implements the logic as described in sec 8.4.1.1 in H264
 *   specification.
 *
 *******************************************************************************
 */
FT_FIND_SKIP_PARAMS isvce_find_pskip_params_me;

/**
 *******************************************************************************
 *
 * @brief The function computes the parameters for a B skip MB
 *
 * @par Description:
 *  The function computes the parameters for a B skip MB
 *
 * @param[in] ps_proc
 *  Process context
 *
 * @param[in] u4_for_me
 *  Flag to indicate the purpose of computing skip
 *
 * @param[out] ps_pred_mv
 *  Flag to indicate the current active refernce list
 *
 * @returns
 *       1) Updates skip MV in proc
 *       2) Returns if the current MB can be coded as skip or not
 *
 * @remarks The code implements the logic as described in sec 8.4.1.1 in H264
 *   specification.
 *
 *******************************************************************************
 */
FT_FIND_SKIP_PARAMS isvce_find_bskip_params;

/**
 *******************************************************************************
 *
 * @brief The function computes the parameters for a B skip MB
 *
 * @par Description:
 *  The function computes the parameters for a B skip MB
 *
 * @param[in] ps_proc
 *  Process context
 *
 * @param[in] u4_for_me
 *  Flag to indicate the purpose of computing skip
 *
 * @param[out] ps_pred_mv
 *  Flag to indicate the current active refernce list
 *
 * @returns
 *       1) Updates skip MV in proc
 *       2) The type of SKIP [L0/L1/BI]
 *
 * @remarks
 *******************************************************************************
 */
FT_FIND_SKIP_PARAMS isvce_find_bskip_params_me;

/**
 *******************************************************************************
 *
 * @brief motion vector predictor
 *
 * @par Description:
 *  The routine calculates the motion vector predictor for a given block,
 *  given the candidate MV predictors.
 *
 * @param[in] ps_left_mb_pu
 *  pointer to left mb motion vector info
 *
 * @param[in] ps_top_row_pu
 *  pointer to top & top right mb motion vector info
 *
 * @param[out] ps_pred_mv
 *  pointer to candidate predictors for the current block
 *
 * @returns  The x & y components of the MV predictor.
 *
 * @remarks The code implements the logic as described in sec 8.4.1.3 in H264
 *   specification.
 *   Assumptions : 1. Assumes Only partition of size 16x16
 *
 *******************************************************************************
 */
void isvce_get_mv_predictor(isvce_enc_pu_mv_t *ps_pred_mv, isvce_enc_pu_mv_t *ps_neig_mv,
                            WORD32 pred_algo);

/**
 *******************************************************************************
 *
 * @brief This fucntion evalues ME for 2 reference lists
 *
 * @par Description:
 *  It evaluates skip, full-pel an half-pel and assigns the correct MV in proc
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns  none
 *
 * @remarks none
 *
 *******************************************************************************
 */
FT_ME_ALGORITHM isvce_compute_me_multi_reflist;

/**
 *******************************************************************************
 *
 * @brief This fucntion evalues ME for single reflist [Pred L0]
 *
 * @par Description:
 *  It evaluates skip, full-pel an half-pel and assigns the correct MV in proc
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns  none
 *
 * @remarks none
 *
 *******************************************************************************
 */
FT_ME_ALGORITHM isvce_compute_me_single_reflist;

/**
 *******************************************************************************
 *
 * @brief This function initializes me ctxt
 *
 * @par Description:
 *  Before dispatching the current job to me thread, the me context associated
 *  with the job is initialized.
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns  none
 *
 * @remarks none
 *
 *******************************************************************************
 */
void isvce_init_me(isvce_process_ctxt_t *ps_proc);

/**
 *******************************************************************************
 *
 * @brief This function performs motion estimation for the current NMB
 *
 * @par Description:
 *  Intializes input and output pointers required by the function
 *isvce_compute_me and calls the function isvce_compute_me in a loop to
 *process NMBs.
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns
 *
 * @remarks none
 *
 *******************************************************************************
 */
void isvce_compute_me_nmb(isvce_process_ctxt_t *ps_proc, UWORD32 u4_nmb_count);

/**
 *******************************************************************************
 *
 * @brief This function performs MV prediction
 *
 * @par Description:
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns  none
 *
 * @remarks none
 *  This function will update the MB availability since intra inter decision
 *  should be done before the call
 *
 *******************************************************************************
 */
void isvce_mv_pred(isvce_process_ctxt_t *ps_proc, WORD32 i4_reflist);

/**
 *******************************************************************************
 *
 * @brief This function approximates Pred. MV
 *
 * @par Description:
 *
 * @param[in] ps_proc
 *  Process context corresponding to the job
 *
 * @returns  none
 *
 * @remarks none
 *  Motion estimation happens at nmb level. For cost calculations, mv is appro
 *  ximated using this function
 *
 *******************************************************************************
 */
void isvce_mv_pred_me(isvce_process_ctxt_t *ps_proc, WORD32 i4_ref_list);

#endif