184 lines
7.2 KiB
C
184 lines
7.2 KiB
C
|
/******************************************************************************
|
||
|
|
*
|
||
|
|
* 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 mb_model_based.c
|
||
|
|
*
|
||
|
|
* \brief
|
||
|
|
* This file contain mb level API functions
|
||
|
|
*
|
||
|
|
* \date
|
||
|
|
*
|
||
|
|
* \author
|
||
|
|
* ittiam
|
||
|
|
*
|
||
|
|
******************************************************************************
|
||
|
|
*/
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* File Includes */
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* User include files */
|
||
|
|
#include "ittiam_datatypes.h"
|
||
|
|
#include "rc_common.h"
|
||
|
|
#include "rc_cntrl_param.h"
|
||
|
|
#include "var_q_operator.h"
|
||
|
|
#include "mem_req_and_acq.h"
|
||
|
|
#include "mb_model_based.h"
|
||
|
|
|
||
|
|
typedef struct mb_rate_control_t
|
||
|
|
{
|
||
|
|
/* Frame Qp */
|
||
|
|
UWORD8 u1_frm_qp;
|
||
|
|
/* Estimated average activity for the current frame (updated with the previous
|
||
|
|
frame activity since it is independent of picture type whether it is I or P) */
|
||
|
|
WORD32 i4_avg_activity;
|
||
|
|
} mb_rate_control_t;
|
||
|
|
|
||
|
|
WORD32 mbrc_num_fill_use_free_memtab(
|
||
|
|
mb_rate_control_t **pps_mb_rate_control, itt_memtab_t *ps_memtab, ITT_FUNC_TYPE_E e_func_type)
|
||
|
|
{
|
||
|
|
WORD32 i4_mem_tab_idx = 0;
|
||
|
|
static mb_rate_control_t s_mb_rate_control_temp;
|
||
|
|
|
||
|
|
/* Hack for al alloc, during which we dont have any state memory.
|
||
|
|
Dereferencing can cause issues */
|
||
|
|
if(e_func_type == GET_NUM_MEMTAB || e_func_type == FILL_MEMTAB)
|
||
|
|
(*pps_mb_rate_control) = &s_mb_rate_control_temp;
|
||
|
|
|
||
|
|
/*for src rate control state structure*/
|
||
|
|
if(e_func_type != GET_NUM_MEMTAB)
|
||
|
|
{
|
||
|
|
fill_memtab(
|
||
|
|
&ps_memtab[i4_mem_tab_idx],
|
||
|
|
sizeof(mb_rate_control_t),
|
||
|
|
MEM_TAB_ALIGNMENT,
|
||
|
|
PERSISTENT,
|
||
|
|
DDR);
|
||
|
|
use_or_fill_base(&ps_memtab[0], (void **)pps_mb_rate_control, e_func_type);
|
||
|
|
}
|
||
|
|
i4_mem_tab_idx++;
|
||
|
|
|
||
|
|
return (i4_mem_tab_idx);
|
||
|
|
}
|
||
|
|
|
||
|
|
/********************************************************************************
|
||
|
|
MB LEVEL API FUNCTIONS
|
||
|
|
********************************************************************************/
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : init_mb_level_rc
|
||
|
|
Description : Initialise the mb model and the average activity to default values
|
||
|
|
Arguments :
|
||
|
|
Return Values : void
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
void init_mb_level_rc(mb_rate_control_t *ps_mb_rate_control)
|
||
|
|
{
|
||
|
|
/* Set values to default */
|
||
|
|
ps_mb_rate_control->i4_avg_activity = 0;
|
||
|
|
}
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : mb_init_frame_level
|
||
|
|
Description : Initialise the mb state with frame level decisions
|
||
|
|
Arguments : u1_frame_qp - Frame level qp
|
||
|
|
Return Values : void
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
void mb_init_frame_level(mb_rate_control_t *ps_mb_rate_control, UWORD8 u1_frame_qp)
|
||
|
|
{
|
||
|
|
/* Update frame level QP */
|
||
|
|
ps_mb_rate_control->u1_frm_qp = u1_frame_qp;
|
||
|
|
}
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : reset_mb_activity
|
||
|
|
Description : Reset the mb activity - Whenever there is SCD
|
||
|
|
the mb activity is reset
|
||
|
|
Arguments :
|
||
|
|
Return Values : void
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
void reset_mb_activity(mb_rate_control_t *ps_mb_rate_control)
|
||
|
|
{
|
||
|
|
ps_mb_rate_control->i4_avg_activity = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : get_mb_qp
|
||
|
|
Description : Calculates the mb level qp
|
||
|
|
Arguments : i4_cur_mb_activity - current frame mb activity
|
||
|
|
pi4_mb_qp - Array of 2 values for before and after mb activity
|
||
|
|
modulation
|
||
|
|
Return Values : void
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
void get_mb_qp(mb_rate_control_t *ps_mb_rate_control, WORD32 i4_cur_mb_activity, WORD32 *pi4_mb_qp)
|
||
|
|
{
|
||
|
|
WORD32 i4_qp;
|
||
|
|
/* Initialise the mb level qp with the frame level qp */
|
||
|
|
i4_qp = ps_mb_rate_control->u1_frm_qp;
|
||
|
|
|
||
|
|
/* Store the model based QP - This is used for updating the rate control model */
|
||
|
|
pi4_mb_qp[0] = i4_qp;
|
||
|
|
|
||
|
|
/* Modulate the Qp based on the activity */
|
||
|
|
if((ps_mb_rate_control->i4_avg_activity) && (i4_qp < 100))
|
||
|
|
{
|
||
|
|
i4_qp = ((((2 * i4_cur_mb_activity)) + ps_mb_rate_control->i4_avg_activity) * i4_qp +
|
||
|
|
((i4_cur_mb_activity + 2 * ps_mb_rate_control->i4_avg_activity) >> 1)) /
|
||
|
|
(i4_cur_mb_activity + 2 * ps_mb_rate_control->i4_avg_activity);
|
||
|
|
|
||
|
|
if(i4_qp > ((3 * ps_mb_rate_control->u1_frm_qp) >> 1))
|
||
|
|
i4_qp = ((3 * ps_mb_rate_control->u1_frm_qp) >> 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Store the qp modulated by mb activity - This is used for encoding the MB */
|
||
|
|
pi4_mb_qp[1] = i4_qp;
|
||
|
|
}
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : get_frm_level_qp
|
||
|
|
Description : Returns the stored frame level QP
|
||
|
|
Arguments :
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
UWORD8 get_frm_level_qp(mb_rate_control_t *ps_mb_rate_control)
|
||
|
|
{
|
||
|
|
return (ps_mb_rate_control->u1_frm_qp);
|
||
|
|
}
|
||
|
|
/******************************************************************************
|
||
|
|
Function Name : mb_update_frame_level
|
||
|
|
Description : Update the frame level info collected
|
||
|
|
Arguments : i4_avg_activity - Average activity fot frame
|
||
|
|
Return Values :
|
||
|
|
Revision History:
|
||
|
|
13 03 2008 KJN Creation
|
||
|
|
*********************************************************************************/
|
||
|
|
void mb_update_frame_level(mb_rate_control_t *ps_mb_rate_control, WORD32 i4_avg_activity)
|
||
|
|
{
|
||
|
|
/*****************************************************************************
|
||
|
|
Update the Average Activity
|
||
|
|
*****************************************************************************/
|
||
|
|
ps_mb_rate_control->i4_avg_activity = i4_avg_activity;
|
||
|
|
}
|