/****************************************************************************** * * 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 #include #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;iai4_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]; } }