382 lines
11 KiB
C
382 lines
11 KiB
C
/******************************************************************************
|
|
*
|
|
* 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
|