/******************************************************************************
*
* Copyright (C) 2015 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
*/
/** Includes */
#include <stdio.h>
#include <string.h>
#include "irc_datatypes.h"
#include "irc_mem_req_and_acq.h"
#include "irc_common.h"
#include "irc_cntrl_param.h"
#include "irc_fixed_point_error_bits.h"
#include "irc_rd_model.h"
#include "irc_est_sad.h"
#include "irc_picture_type.h"
#include "irc_bit_allocation.h"
#include "irc_trace_support.h"
/** Macros **/
#define MIN(x,y) ((x) < (y))? (x) : (y)
/* State structure for bit allocation */
typedef struct
{
/* using var_q number as it can cross 31 bits for large intra frameinterval */
number_t vq_rem_bits_in_period;
/* Storing inputs */
WORD32 i4_tot_frms_in_gop;
WORD32 i4_num_intra_frm_interval;
WORD32 i4_bits_per_frm;
} rem_bit_in_prd_t;
typedef struct bit_allocation_t
{
rem_bit_in_prd_t s_rbip;
/* A universal constant giving the relative complexity between pictures */
WORD32 i2_K[MAX_PIC_TYPE];
/* To get a estimate of the header bits consumed */
WORD32 i4_prev_frm_header_bits[MAX_PIC_TYPE];
WORD32 i4_bits_per_frm;
WORD32 i4_num_gops_in_period;
/* Num gops as set by rate control module */
WORD32 i4_actual_num_gops_in_period;
number_t vq_saved_bits;
WORD32 i4_max_bits_per_frm[MAX_NUM_DRAIN_RATES];
WORD32 i4_min_bits_per_frm;
/* Error bits module */
error_bits_handle ps_error_bits;
/* Storing frame rate */
WORD32 i4_frame_rate;
WORD32 i4_bit_rate;
WORD32 ai4_peak_bit_rate[MAX_NUM_DRAIN_RATES];
} bit_allocation_t;
static WORD32 get_number_of_frms_in_a_gop(pic_handling_handle ps_pic_handling)
{
WORD32 i4_tot_frms_in_gop = 0, i;
WORD32 ai4_frms_in_gop[MAX_PIC_TYPE];
/* Query the pic_handling struct for the rem frames in the period */
irc_pic_type_get_frms_in_gop(ps_pic_handling, ai4_frms_in_gop);
/* Get the total frms in the gop */
i4_tot_frms_in_gop = 0;
for(i = 0; i < MAX_PIC_TYPE; i++)
{
i4_tot_frms_in_gop += ai4_frms_in_gop[i];
}
return (i4_tot_frms_in_gop);
}
static void init_rbip(rem_bit_in_prd_t *ps_rbip,
pic_handling_handle ps_pic_handling,
WORD32 i4_bits_per_frm,
WORD32 i4_num_intra_frm_interval)
{
WORD32 i4_tot_frms_in_gop = get_number_of_frms_in_a_gop(ps_pic_handling);
/* rem_bits_in_period = bits_per_frm * tot_frms_in_gop * num_intra_frm_interval */
{
number_t vq_bits_per_frm, vq_tot_frms_in_gop, vq_num_intra_frm_interval;
number_t *pvq_rem_bits_in_period = &ps_rbip->vq_rem_bits_in_period;
SET_VAR_Q(vq_bits_per_frm, i4_bits_per_frm, 0);
SET_VAR_Q(vq_tot_frms_in_gop, i4_tot_frms_in_gop, 0);
SET_VAR_Q(vq_num_intra_frm_interval, i4_num_intra_frm_interval, 0);
/* rem_bits_in_period = bits_per_frm * tot_frms_in_gop */
mult32_var_q(vq_bits_per_frm, vq_tot_frms_in_gop,
pvq_rem_bits_in_period);
/* rem_bits_in_period *= num_intra_frm_interval */
mult32_var_q(vq_num_intra_frm_interval, pvq_rem_bits_in_period[0],
pvq_rem_bits_in_period);
}
/*
* Store the total number of frames in GOP value which is
* used from module A
*/
ps_rbip->i4_tot_frms_in_gop = i4_tot_frms_in_gop;
ps_rbip->i4_num_intra_frm_interval = i4_num_intra_frm_interval;
ps_rbip->i4_bits_per_frm = i4_bits_per_frm;
}
static void check_update_rbip(rem_bit_in_prd_t *ps_rbip,
pic_handling_handle ps_pic_handling)
{
/*
* NOTE: Intra frame interval changes after the first I frame that is
* encoded in a GOP
*/
WORD32 i4_new_tot_frms_in_gop = get_number_of_frms_in_a_gop(
ps_pic_handling);
if(i4_new_tot_frms_in_gop != ps_rbip->i4_tot_frms_in_gop)
{
WORD32 i4_rem_frames_in_period =
ps_rbip->i4_num_intra_frm_interval
* (i4_new_tot_frms_in_gop
- ps_rbip->i4_tot_frms_in_gop);
number_t vq_rem_frms_in_period, s_bits_per_frm, vq_delta_bits_in_period;
SET_VAR_Q(vq_rem_frms_in_period, i4_rem_frames_in_period, 0);
SET_VAR_Q(s_bits_per_frm, ps_rbip->i4_bits_per_frm, 0);
/* delta_bits_in_period = bits_per_frm * rem_frms_in_period */
mult32_var_q(s_bits_per_frm, vq_rem_frms_in_period,
&vq_delta_bits_in_period);
/* rem_bits_in_period += delta_bits_in_period */
add32_var_q(vq_delta_bits_in_period, ps_rbip->vq_rem_bits_in_period,
&ps_rbip->vq_rem_bits_in_period);
}
/* Updated the new values */
ps_rbip->i4_tot_frms_in_gop = i4_new_tot_frms_in_gop;
}
static void irc_ba_update_rbip(rem_bit_in_prd_t *ps_rbip,
pic_handling_handle ps_pic_handling,
WORD32 i4_num_of_bits)
{
number_t vq_num_bits;
check_update_rbip(ps_rbip, ps_pic_handling);
/* rem_bits_in_period += num_of_bits */
SET_VAR_Q(vq_num_bits, i4_num_of_bits, 0);
add32_var_q(vq_num_bits, ps_rbip->vq_rem_bits_in_period,
&ps_rbip->vq_rem_bits_in_period);
}
static void irc_ba_change_rbip(rem_bit_in_prd_t *ps_rbip,
pic_handling_handle ps_pic_handling,
WORD32 i4_new_bits_per_frm,
WORD32 i4_new_num_intra_frm_interval)
{
WORD32 ai4_rem_frms_in_period[MAX_PIC_TYPE], i4_rem_frms_in_gop, i;
irc_pic_type_get_rem_frms_in_gop(ps_pic_handling, ai4_rem_frms_in_period);
i4_rem_frms_in_gop = 0;
for(i = 0; i < MAX_PIC_TYPE; i++)
i4_rem_frms_in_gop += ai4_rem_frms_in_period[i];
if(i4_new_bits_per_frm != ps_rbip->i4_bits_per_frm)
{
WORD32 i4_rem_frms_in_period = (ps_rbip->i4_num_intra_frm_interval - 1)
* ps_rbip->i4_tot_frms_in_gop + i4_rem_frms_in_gop;
number_t vq_rem_frms_in_period, vq_delta_bits_per_frm,
vq_delta_bits_in_period;
/* delta_bits_per_frm = new_bits_per_frm - old_bits_per_frm */
SET_VAR_Q(vq_delta_bits_per_frm,
(i4_new_bits_per_frm - ps_rbip->i4_bits_per_frm), 0);
SET_VAR_Q(vq_rem_frms_in_period, i4_rem_frms_in_period, 0);
/* delta_bits_in_period = delta_bits_per_frm * rem_frms_in_period */
mult32_var_q(vq_delta_bits_per_frm, vq_rem_frms_in_period,
&vq_delta_bits_in_period);
/* ps_rbip->rem_bits_in_period += delta_bits_in_period */
add32_var_q(vq_delta_bits_in_period, ps_rbip->vq_rem_bits_in_period,
&ps_rbip->vq_rem_bits_in_period);
}
if(i4_new_num_intra_frm_interval != ps_rbip->i4_num_intra_frm_interval)
{
WORD32 i4_rem_frms_in_period = ps_rbip->i4_tot_frms_in_gop
* (i4_new_num_intra_frm_interval
- ps_rbip->i4_num_intra_frm_interval);
number_t vq_rem_frms_in_period, vq_new_bits_per_frm,
vq_delta_bits_in_period;
/* new_bits_per_frm = new_new_bits_per_frm - old_new_bits_per_frm */
SET_VAR_Q(vq_new_bits_per_frm, i4_new_bits_per_frm, 0);
SET_VAR_Q(vq_rem_frms_in_period, i4_rem_frms_in_period, 0);
/* delta_bits_in_period = new_bits_per_frm * rem_frms_in_period */
mult32_var_q(vq_new_bits_per_frm, vq_rem_frms_in_period,
&vq_delta_bits_in_period);
/* ps_rbip->rem_bits_in_period += delta_bits_in_period */
add32_var_q(vq_delta_bits_in_period, ps_rbip->vq_rem_bits_in_period,
&ps_rbip->vq_rem_bits_in_period);
}
/* Update the new value */
ps_rbip->i4_num_intra_frm_interval = i4_new_num_intra_frm_interval;
ps_rbip->i4_bits_per_frm = i4_new_bits_per_frm;
}
WORD32 irc_ba_num_fill_use_free_memtab(bit_allocation_t **pps_bit_allocation,
itt_memtab_t *ps_memtab,
ITT_FUNC_TYPE_E e_func_type)
{
WORD32 i4_mem_tab_idx = 0;
bit_allocation_t s_bit_allocation_temp;
/*
* Hack for all alloc, during which we don't have any state memory.
* Dereferencing can cause issues
*/
if(e_func_type == GET_NUM_MEMTAB || e_func_type == FILL_MEMTAB)
(*pps_bit_allocation) = &s_bit_allocation_temp;
/*for src rate control state structure*/
if(e_func_type != GET_NUM_MEMTAB)
{
fill_memtab(&ps_memtab[i4_mem_tab_idx], sizeof(bit_allocation_t),
ALIGN_128_BYTE, PERSISTENT, DDR);
use_or_fill_base(&ps_memtab[0], (void**)pps_bit_allocation,
e_func_type);
}
i4_mem_tab_idx++;
i4_mem_tab_idx += irc_error_bits_num_fill_use_free_memtab(
&pps_bit_allocation[0]->ps_error_bits,
&ps_memtab[i4_mem_tab_idx], e_func_type);
return (i4_mem_tab_idx);
}
/*******************************************************************************
Function Name : irc_ba_init_bit_allocation
Description : Initialize the bit_allocation structure.
******************************************************************************/
void irc_ba_init_bit_allocation(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling,
WORD32 i4_num_intra_frm_interval,
WORD32 i4_bit_rate,
WORD32 i4_frm_rate,
WORD32 *i4_peak_bit_rate,
WORD32 i4_min_bitrate)
{
WORD32 i;
WORD32 i4_bits_per_frm, i4_max_bits_per_frm[MAX_NUM_DRAIN_RATES];
/* Calculate the bits per frame */
X_PROD_Y_DIV_Z(i4_bit_rate, 1000, i4_frm_rate, i4_bits_per_frm);
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
X_PROD_Y_DIV_Z(i4_peak_bit_rate[i], 1000, i4_frm_rate,
i4_max_bits_per_frm[i]);
}
/* Initialize the bits_per_frame */
ps_bit_allocation->i4_bits_per_frm = i4_bits_per_frm;
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
ps_bit_allocation->i4_max_bits_per_frm[i] = i4_max_bits_per_frm[i];
}
X_PROD_Y_DIV_Z(i4_min_bitrate, 1000, i4_frm_rate,
ps_bit_allocation->i4_min_bits_per_frm);
/*
* Initialize the rem_bits in period
* The first gop in case of an OPEN GOP may have fewer B_PICs,
* That condition is not taken care of
*/
init_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling, i4_bits_per_frm,
i4_num_intra_frm_interval);
/* Initialize the num_gops_in_period */
ps_bit_allocation->i4_num_gops_in_period = i4_num_intra_frm_interval;
ps_bit_allocation->i4_actual_num_gops_in_period = i4_num_intra_frm_interval;
/* Relative complexity between I and P frames */
ps_bit_allocation->i2_K[I_PIC] = (1 << K_Q);
ps_bit_allocation->i2_K[P_PIC] = I_TO_P_RATIO;
ps_bit_allocation->i2_K[B_PIC] = (P_TO_B_RATIO * I_TO_P_RATIO) >> K_Q;
/* Initialize the saved bits to 0*/
SET_VAR_Q(ps_bit_allocation->vq_saved_bits, 0, 0);
/* Update the error bits module with average bits */
irc_init_error_bits(ps_bit_allocation->ps_error_bits, i4_frm_rate,
i4_bit_rate);
/* Store the input for implementing change in values */
ps_bit_allocation->i4_frame_rate = i4_frm_rate;
ps_bit_allocation->i4_bit_rate = i4_bit_rate;
memset(ps_bit_allocation->i4_prev_frm_header_bits, 0, sizeof(ps_bit_allocation->i4_prev_frm_header_bits));
for(i=0;i<MAX_NUM_DRAIN_RATES;i++)
ps_bit_allocation->ai4_peak_bit_rate[i] = i4_peak_bit_rate[i];
}
/*******************************************************************************
Function Name : get_cur_frm_est_bits
Description : Based on remaining bits in period and rd_model
the number of bits required for the current frame is estimated.
******************************************************************************/
WORD32 irc_ba_get_cur_frm_est_texture_bits(bit_allocation_t *ps_bit_allocation,
rc_rd_model_handle *pps_rd_model,
est_sad_handle ps_est_sad,
pic_handling_handle ps_pic_handling,
picture_type_e e_pic_type)
{
WORD32 i, j;
WORD32 i4_est_texture_bits_for_frm;
number_t vq_rem_texture_bits;
number_t vq_complexity_estimate[MAX_PIC_TYPE];
WORD32 i4_rem_frms_in_period[MAX_PIC_TYPE], i4_frms_in_period[MAX_PIC_TYPE];
number_t vq_max_consumable_bits;
number_t vq_rem_frms_in_period[MAX_PIC_TYPE], vq_est_texture_bits_for_frm;
number_t vq_prev_hdr_bits[MAX_PIC_TYPE];
WORD32 complexity_est = 0;
/* Get the rem_frms_in_gop & the frms_in_gop from the pic_type state struct */
irc_pic_type_get_rem_frms_in_gop(ps_pic_handling, i4_rem_frms_in_period);
irc_pic_type_get_frms_in_gop(ps_pic_handling, i4_frms_in_period);
/* Depending on the number of gops in a period, find the num_frms_in_prd */
for(j = 0; j < MAX_PIC_TYPE; j++)
{
i4_rem_frms_in_period[j] += (i4_frms_in_period[j]
* (ps_bit_allocation->i4_num_gops_in_period - 1));
i4_frms_in_period[j] *= ps_bit_allocation->i4_num_gops_in_period;
}
/* Remove the header bits from the remaining bits to find how many bits you
can transfer.*/
irc_ba_update_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling, 0);
for(i = 0; i < MAX_PIC_TYPE; i++)
{
SET_VAR_Q(vq_rem_frms_in_period[i], i4_rem_frms_in_period[i], 0);
SET_VAR_Q(vq_prev_hdr_bits[i],
ps_bit_allocation->i4_prev_frm_header_bits[i], 0);
}
{
/*
*rem_texture_bits = rem_bits_in_period -
*(rem_frms_in_period[I_PIC] * prev_frm_header_bits[I_PIC]) -
*(rem_frms_in_period[P_PIC] * prev_frm_header_bits[P_PIC]) -
*(rem_frms_in_period[B_PIC] * prev_frm_header_bits[B_PIC]);
*/
number_t vq_rem_hdr_bits;
vq_rem_texture_bits = ps_bit_allocation->s_rbip.vq_rem_bits_in_period;
mult32_var_q(vq_prev_hdr_bits[I_PIC], vq_rem_frms_in_period[I_PIC],
&vq_rem_hdr_bits);
sub32_var_q(vq_rem_texture_bits, vq_rem_hdr_bits, &vq_rem_texture_bits);
mult32_var_q(vq_prev_hdr_bits[P_PIC], vq_rem_frms_in_period[P_PIC],
&vq_rem_hdr_bits);
sub32_var_q(vq_rem_texture_bits, vq_rem_hdr_bits, &vq_rem_texture_bits);
mult32_var_q(vq_prev_hdr_bits[B_PIC], vq_rem_frms_in_period[B_PIC],
&vq_rem_hdr_bits);
sub32_var_q(vq_rem_texture_bits, vq_rem_hdr_bits, &vq_rem_texture_bits);
}
{
/* max_consumable_bits =
*(frms_in_period[I_PIC] * max_bits_per_frm[0] ) +
*(frms_in_period[P_PIC] + frms_in_period[B_PIC] ) * max_bits_per_frm[1];
*/
number_t vq_max_bits, vq_max_bits_per_frm[2];
SET_VAR_Q(vq_max_bits_per_frm[0],
ps_bit_allocation->i4_max_bits_per_frm[0], 0);
SET_VAR_Q(vq_max_bits_per_frm[1],
ps_bit_allocation->i4_max_bits_per_frm[1], 0);
mult32_var_q(vq_rem_frms_in_period[I_PIC], vq_max_bits_per_frm[0],
&vq_max_bits);
vq_max_consumable_bits = vq_max_bits;
mult32_var_q(vq_rem_frms_in_period[P_PIC], vq_max_bits_per_frm[1],
&vq_max_bits);
add32_var_q(vq_max_bits, vq_max_consumable_bits,
&vq_max_consumable_bits);
mult32_var_q(vq_rem_frms_in_period[B_PIC], vq_max_bits_per_frm[1],
&vq_max_bits);
add32_var_q(vq_max_bits, vq_max_consumable_bits,
&vq_max_consumable_bits);
}
/* rem_texture_bits = MIN(rem_texture_bits, max_consumable_bits) */
MIN_VARQ(vq_max_consumable_bits, vq_rem_texture_bits, vq_rem_texture_bits);
/* The bits are then allocated based on the relative complexity of the
current frame with respect to that of the rest of the frames in period */
for(i = 0; i < MAX_PIC_TYPE; i++)
{
number_t vq_lin_mod_coeff, vq_est_sad, vq_K;
/* Getting the linear model coefficient */
vq_lin_mod_coeff = irc_get_linear_coefficient(pps_rd_model[i]);
/* Getting the estimated SAD */
SET_VAR_Q(vq_est_sad, irc_get_est_sad(ps_est_sad,i), 0);
/* Making K factor a var Q format */
SET_VAR_Q(vq_K, ps_bit_allocation->i2_K[i], K_Q);
/* Complexity_estimate = [ (lin_mod_coeff * estimated_sad) / K factor ] */
mult32_var_q(vq_lin_mod_coeff, vq_est_sad, &vq_lin_mod_coeff);
div32_var_q(vq_lin_mod_coeff, vq_K, &vq_complexity_estimate[i]);
}
/*
* For simple cases, one of the complexities go to zero and in those cases
* distribute the bits evenly among frames based on I_TO_P_RATIO
*/
/* Also check the B-pictures complexity only in case they are present*/
if(i4_frms_in_period[B_PIC] == 0)
{
complexity_est = (vq_complexity_estimate[I_PIC]
&& vq_complexity_estimate[P_PIC]);
}
else
{
complexity_est = (vq_complexity_estimate[I_PIC]
&& vq_complexity_estimate[P_PIC]
&& vq_complexity_estimate[B_PIC]);
}
if(complexity_est)
{
/*
* Estimated texture bits =
* (remaining bits) * (cur frm complexity)
* ---------------------------------------
* (num_i_frm*i_frm_complexity) + (num_p_frm*pfrm_complexity)
* + (b_frm * b_frm_cm)
*/
mult32_var_q(vq_rem_texture_bits, vq_complexity_estimate[e_pic_type],
&vq_rem_texture_bits);
for(i = 0; i < MAX_PIC_TYPE; i++)
{
mult32_var_q(vq_rem_frms_in_period[i], vq_complexity_estimate[i],
&vq_rem_frms_in_period[i]);
}
add32_var_q(vq_rem_frms_in_period[I_PIC], vq_rem_frms_in_period[P_PIC],
&vq_rem_frms_in_period[I_PIC]);
add32_var_q(vq_rem_frms_in_period[I_PIC], vq_rem_frms_in_period[B_PIC],
&vq_rem_frms_in_period[I_PIC]);
div32_var_q(vq_rem_texture_bits, vq_rem_frms_in_period[I_PIC],
&vq_est_texture_bits_for_frm);
number_t_to_word32(vq_est_texture_bits_for_frm,
&i4_est_texture_bits_for_frm);
}
else
{
number_t vq_i_to_p_bit_ratio, vq_rem_frms;
SET_VAR_Q(vq_i_to_p_bit_ratio, I_TO_P_BIT_RATIO, 0);
/* rem_frms = ((I_TO_P_BIT_RATIO * rem_frms_in_period[I_PIC]) +
* rem_frms_in_period[P_PIC] + rem_frms_in_period[B_PIC]);
*/
mult32_var_q(vq_rem_frms_in_period[I_PIC], vq_i_to_p_bit_ratio,
&vq_rem_frms);
add32_var_q(vq_rem_frms_in_period[P_PIC], vq_rem_frms, &vq_rem_frms);
add32_var_q(vq_rem_frms_in_period[B_PIC], vq_rem_frms, &vq_rem_frms);
/* est_texture_bits_for_frm = rem_texture_bits / rem_frms */
div32_var_q(vq_rem_texture_bits, vq_rem_frms,
&vq_est_texture_bits_for_frm);
number_t_to_word32(vq_est_texture_bits_for_frm,
&i4_est_texture_bits_for_frm);
i4_est_texture_bits_for_frm =
(I_PIC == e_pic_type) ?
(i4_est_texture_bits_for_frm
* I_TO_P_BIT_RATIO) :
i4_est_texture_bits_for_frm;
}
/*
* If the remaining bits in the period becomes negative then the estimated
* texture bits would also become negative. This would send a feedback to
* the model which may go for a toss. Thus sending the minimum possible
* value = 0
*/
if(i4_est_texture_bits_for_frm < 0)
{
i4_est_texture_bits_for_frm = 0;
}
return (i4_est_texture_bits_for_frm);
}
/******************************************************************************
Function Name : irc_ba_get_cur_frm_est_header_bits
Description : Based on remaining bits in period and rd_model
the number of bits required for the current frame is estimated.
******************************************************************************/
WORD32 irc_ba_get_cur_frm_est_header_bits(bit_allocation_t *ps_bit_allocation,
picture_type_e e_pic_type)
{
return (ps_bit_allocation->i4_prev_frm_header_bits[e_pic_type]);
}
WORD32 irc_ba_get_rem_bits_in_period(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling)
{
WORD32 i4_rem_bits_in_gop = 0;
irc_ba_update_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling, 0);
number_t_to_word32(ps_bit_allocation->s_rbip.vq_rem_bits_in_period,
&i4_rem_bits_in_gop);
return (i4_rem_bits_in_gop);
}
/*******************************************************************************
Function Name : irc_ba_update_cur_frm_consumed_bits
Description : Based on remaining bits in period and rd_model
the number of bits required for the current frame is estimated.
******************************************************************************/
void irc_ba_update_cur_frm_consumed_bits(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling,
WORD32 i4_total_frame_bits,
WORD32 i4_model_updation_hdr_bits,
picture_type_e e_pic_type,
UWORD8 u1_is_scd,
WORD32 i4_last_frm_in_gop)
{
WORD32 i4_error_bits = irc_get_error_bits(ps_bit_allocation->ps_error_bits);
/* Update the remaining bits in period */
irc_ba_update_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
(-i4_total_frame_bits + i4_error_bits));
/*
* Update the header bits so that it can be used as an estimate to the next
* frame
*/
if(u1_is_scd)
{
/*
* In case of SCD, even though the frame type is P, it is equivalent to
* a I frame and so the corresponding header bits is updated
*/
ps_bit_allocation->i4_prev_frm_header_bits[I_PIC] =
i4_model_updation_hdr_bits;
#define MAX_NUM_GOPS_IN_PERIOD (3)
if(ps_bit_allocation->i4_num_gops_in_period < MAX_NUM_GOPS_IN_PERIOD)
{
/*
* Whenever there is a scene change increase the number of gops by
* 2 so that the number of bits allocated is not very constrained
*/
ps_bit_allocation->i4_num_gops_in_period += 2;
/* Add the extra bits in GOP to remaining bits in period */
irc_ba_change_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
ps_bit_allocation->i4_bits_per_frm,
ps_bit_allocation->i4_num_gops_in_period);
}
}
else
{
ps_bit_allocation->i4_prev_frm_header_bits[e_pic_type] =
i4_model_updation_hdr_bits;
}
if(i4_last_frm_in_gop)
{
WORD32 i4_num_bits_in_a_gop = get_number_of_frms_in_a_gop(
ps_pic_handling) * ps_bit_allocation->i4_bits_per_frm;
/*
* If the number of gops in period has been increased due to scene
* change, slowly bring in down across the gops
*/
if(ps_bit_allocation->i4_num_gops_in_period
> ps_bit_allocation->i4_actual_num_gops_in_period)
{
ps_bit_allocation->i4_num_gops_in_period--;
irc_ba_change_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
ps_bit_allocation->i4_bits_per_frm,
ps_bit_allocation->i4_num_gops_in_period);
}
/*
* If rem_bits_in_period < 0 decrease the number of bits allocated for
* the next period else increase it
*/
irc_ba_update_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
i4_num_bits_in_a_gop);
}
/* Update the lower modules */
irc_update_error_bits(ps_bit_allocation->ps_error_bits);
}
void irc_ba_change_remaining_bits_in_period(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling,
WORD32 i4_bit_rate,
WORD32 i4_frame_rate,
WORD32 *i4_peak_bit_rate)
{
WORD32 i4_new_avg_bits_per_frm;
WORD32 i4_new_peak_bits_per_frm[MAX_NUM_DRAIN_RATES];
WORD32 i4_rem_frms_in_period[MAX_PIC_TYPE];
int i;
/* Calculate the new per frame bits */
X_PROD_Y_DIV_Z(i4_bit_rate, 1000, i4_frame_rate, i4_new_avg_bits_per_frm);
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
X_PROD_Y_DIV_Z(i4_peak_bit_rate[i], 1000, i4_frame_rate,
i4_new_peak_bits_per_frm[i]);
}
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
ps_bit_allocation->i4_max_bits_per_frm[i] = i4_new_peak_bits_per_frm[i];
}
/*
* Get the rem_frms_in_prd & the frms_in_prd from the pic_type state
* struct
*/
irc_pic_type_get_rem_frms_in_gop(ps_pic_handling, i4_rem_frms_in_period);
/*
* If the difference > 0(/ <0), the remaining bits in period needs to be
* increased(/decreased) based on the remaining number of frames
*/
irc_ba_change_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
i4_new_avg_bits_per_frm,
ps_bit_allocation->i4_num_gops_in_period);
/* Update the new average bits per frame */
ps_bit_allocation->i4_bits_per_frm = i4_new_avg_bits_per_frm;
/* change the lower modules state */
irc_change_bitrate_in_error_bits(ps_bit_allocation->ps_error_bits,
i4_bit_rate);
irc_change_frm_rate_in_error_bits(ps_bit_allocation->ps_error_bits,
i4_frame_rate);
/* Store the modified frame_rate */
ps_bit_allocation->i4_frame_rate = i4_frame_rate;
ps_bit_allocation->i4_bit_rate = i4_bit_rate;
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
ps_bit_allocation->ai4_peak_bit_rate[i] = i4_peak_bit_rate[i];
}
void irc_ba_change_ba_peak_bit_rate(bit_allocation_t *ps_bit_allocation,
WORD32 *ai4_peak_bit_rate)
{
WORD32 i;
/* Calculate the bits per frame */
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
X_PROD_Y_DIV_Z(ai4_peak_bit_rate[i], 1000,
ps_bit_allocation->i4_frame_rate,
ps_bit_allocation->i4_max_bits_per_frm[i]);
ps_bit_allocation->ai4_peak_bit_rate[i] = ai4_peak_bit_rate[i];
}
}
/******************************************************************************
* @brief Modifies the remaining bit in period for the gop which has fif.
* since fif would cause a new gop to be created, we need to add the number
* of encoded frames in the fif GOP worth of bits to remaining bits in
* period
******************************************************************************/
void irc_ba_change_rem_bits_in_prd_at_force_I_frame(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling)
{
WORD32 i4_frms_in_period;
i4_frms_in_period = irc_pic_type_get_frms_in_gop_force_I_frm(
ps_pic_handling);
irc_ba_update_rbip(&ps_bit_allocation->s_rbip, ps_pic_handling,
ps_bit_allocation->i4_bits_per_frm * i4_frms_in_period);
}
void irc_ba_check_and_update_bit_allocation(bit_allocation_t *ps_bit_allocation,
pic_handling_handle ps_pic_handling,
WORD32 i4_cur_buf_size,
WORD32 i4_max_buf_size,
WORD32 i4_max_bits_inflow_per_frm,
WORD32 i4_tot_frame_bits)
{
number_t vq_max_drain_bits, vq_extra_bits, vq_less_bits,
vq_allocated_saved_bits, vq_min_bits_for_period;
WORD32 i4_num_frms_in_period = get_number_of_frms_in_a_gop(ps_pic_handling);
number_t vq_rem_bits_in_period, vq_num_frms_in_period, vq_zero;
WORD32 b_rem_bits_gt_max_drain, b_rem_bits_lt_min_bits,
b_saved_bits_gt_zero;
rem_bit_in_prd_t *ps_rbip = &ps_bit_allocation->s_rbip;
UNUSED(i4_cur_buf_size);
UNUSED(i4_max_buf_size);
UNUSED(i4_tot_frame_bits);
/*
* If the remaining bits is greater than what can be drained in that period
* Clip the remaining bits in period to the maximum it can drain in that
* period with the error of current buffer size.Accumulate the saved bits
* if any. else if the remaining bits is lesser than the minimum bit rate
* promised in that period Add the excess bits to remaining bits in period
* and reduce it from the saved bits Else Provide the extra bits from the
* "saved bits pool".
*/
/*
* max_drain_bits = num_gops_in_period * num_frms_in_period *
* * max_bits_inflow_per_frm
*/
SET_VAR_Q(vq_num_frms_in_period,
(ps_bit_allocation->i4_num_gops_in_period * i4_num_frms_in_period),
0);
SET_VAR_Q(vq_max_drain_bits, i4_max_bits_inflow_per_frm, 0);
SET_VAR_Q(vq_zero, 0, 0);
mult32_var_q(vq_max_drain_bits, vq_num_frms_in_period, &vq_max_drain_bits);
/*
* min_bits_for_period = num_gops_in_period * num_frms_in_period *
* min_bits_per_frm
*/
SET_VAR_Q(vq_min_bits_for_period, ps_bit_allocation->i4_min_bits_per_frm,
0);
mult32_var_q(vq_min_bits_for_period, vq_num_frms_in_period,
&vq_min_bits_for_period);
vq_rem_bits_in_period = ps_rbip->vq_rem_bits_in_period;
/* Evaluate rem_bits_in_period > max_drain_bits */
VQ_A_GT_VQ_B(ps_rbip->vq_rem_bits_in_period, vq_max_drain_bits,
b_rem_bits_gt_max_drain);
/* Evaluate rem_bits_in_period < min_bits_for_period */
VQ_A_LT_VQ_B(ps_rbip->vq_rem_bits_in_period, vq_min_bits_for_period,
b_rem_bits_lt_min_bits);
/* Evaluate saved_bits > 0 */
VQ_A_LT_VQ_B(ps_bit_allocation->vq_saved_bits, vq_zero,
b_saved_bits_gt_zero);
/* (i4_rem_bits_in_period > i4_max_drain_bits) */
if(b_rem_bits_gt_max_drain)
{
/* extra_bits = rem_bits_in_period - max_drain_bits */
sub32_var_q(ps_rbip->vq_rem_bits_in_period, vq_max_drain_bits,
&vq_extra_bits);
/* saved_bits += extra_bits */
add32_var_q(ps_bit_allocation->vq_saved_bits, vq_extra_bits,
&ps_bit_allocation->vq_saved_bits);
/* rem_bits_in_period = vq_max_drain_bits */
ps_rbip->vq_rem_bits_in_period = vq_max_drain_bits;
}
else if(b_rem_bits_lt_min_bits)
{
/* extra_bits(-ve) = rem_bits_in_period - i4_min_bits_for_period */
sub32_var_q(ps_rbip->vq_rem_bits_in_period, vq_min_bits_for_period,
&vq_extra_bits);
/* saved_bits += extra_bits(-ve) */
add32_var_q(ps_bit_allocation->vq_saved_bits, vq_extra_bits,
&ps_bit_allocation->vq_saved_bits);
/* rem_bits_in_period = min_bits_for_period */
ps_rbip->vq_rem_bits_in_period = vq_min_bits_for_period;
}
else if(b_saved_bits_gt_zero)
{
/* less_bits = max_drain_bits - _rem_bits_in_period */
sub32_var_q(vq_max_drain_bits, vq_rem_bits_in_period, &vq_less_bits);
/* allocated_saved_bits = MIN (less_bits, saved_bits) */
MIN_VARQ(ps_bit_allocation->vq_saved_bits, vq_less_bits,
vq_allocated_saved_bits);
/* rem_bits_in_period += allocted_save_bits */
add32_var_q(ps_rbip->vq_rem_bits_in_period, vq_allocated_saved_bits,
&ps_rbip->vq_rem_bits_in_period);
/* saved_bits -= allocted_save_bits */
sub32_var_q(ps_bit_allocation->vq_saved_bits, vq_allocated_saved_bits,
&ps_bit_allocation->vq_saved_bits);
}
return;
}
WORD32 irc_ba_get_frame_rate(bit_allocation_t *ps_bit_allocation)
{
return (ps_bit_allocation->i4_frame_rate);
}
WORD32 irc_ba_get_bit_rate(bit_allocation_t *ps_bit_allocation)
{
return (ps_bit_allocation->i4_bit_rate);
}
void irc_ba_get_peak_bit_rate(bit_allocation_t *ps_bit_allocation,
WORD32 *pi4_peak_bit_rate)
{
WORD32 i;
for(i = 0; i < MAX_NUM_DRAIN_RATES; i++)
{
pi4_peak_bit_rate[i] = ps_bit_allocation->ai4_peak_bit_rate[i];
}
}