| Hamsalekha S | 8d3d303 | 2015-03-13 21:24:58 +0530 | [diff] [blame] | 1 | /****************************************************************************** |
| 2 | * |
| 3 | * Copyright (C) 2015 The Android Open Source Project |
| 4 | * |
| 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at: |
| 8 | * |
| 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | * |
| 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
| 16 | * |
| 17 | ***************************************************************************** |
| 18 | * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore |
| 19 | */ |
| 20 | |
| 21 | /*****************************************************************************/ |
| 22 | /* File Name : irc_rd_model.h */ |
| 23 | /* */ |
| 24 | /* Description : Implements all the Functions to Model the */ |
| 25 | /* Rate Distortion Behaviour of the Codec over the Last */ |
| 26 | /* Few Frames. */ |
| 27 | /* */ |
| 28 | /* List of Functions : irc_update_frame_rd_model */ |
| 29 | /* estimate_mpeg2_qp_for_resbits */ |
| 30 | /* update_mb_rd_model */ |
| 31 | /* find_model_coeffs */ |
| 32 | /* refine_set_of_points */ |
| 33 | /* init_mb_rd_model */ |
| 34 | /* irc_add_frame_to_rd_model */ |
| 35 | /* irc_find_qp_for_target_bits */ |
| 36 | /* */ |
| 37 | /* */ |
| 38 | /* Issues / Problems : None */ |
| 39 | /* */ |
| 40 | /* Revision History : */ |
| 41 | /* DD MM YYYY Author(s) Changes (Describe the changes made) */ |
| 42 | /* 21 06 2006 Sarat Initial Version */ |
| 43 | /*****************************************************************************/ |
| 44 | |
| 45 | #ifndef RC_RD_MODEL |
| 46 | #define RC_RD_MODEL |
| 47 | |
| 48 | #define MAX_FRAMES_MODELLED 16 |
| 49 | |
| 50 | typedef float model_coeff; |
| 51 | typedef struct rc_rd_model_t *rc_rd_model_handle; |
| 52 | |
| 53 | WORD32 irc_rd_model_num_fill_use_free_memtab(rc_rd_model_handle *pps_rc_rd_model, |
| 54 | itt_memtab_t *ps_memtab, |
| 55 | ITT_FUNC_TYPE_E e_func_type); |
| 56 | /* Interface Functions */ |
| 57 | /* Initialise the rate distortion model */ |
| 58 | void irc_init_frm_rc_rd_model(rc_rd_model_handle ps_rd_model, |
| 59 | UWORD8 u1_max_frames_modelled); |
| 60 | |
| 61 | /* Reset the rate distortion model */ |
| 62 | void irc_reset_frm_rc_rd_model(rc_rd_model_handle ps_rd_model); |
| 63 | |
| 64 | /* Returns the Qp to be used for the given bits and SAD */ |
| 65 | UWORD8 irc_find_qp_for_target_bits(rc_rd_model_handle ps_rd_model, |
| 66 | UWORD32 u4_target_res_bits, |
| 67 | UWORD32 u4_estimated_sad, |
| 68 | UWORD8 u1_max_qp, |
| 69 | UWORD8 u1_min_qp); |
| 70 | |
| 71 | /* Updates the frame level statistics after encoding a frame */ |
| 72 | void irc_add_frame_to_rd_model(rc_rd_model_handle ps_rd_model, |
| 73 | UWORD32 i4_res_bits, |
| 74 | UWORD8 u1_avg_mp2qp, |
| 75 | UWORD32 i4_sad_h264, |
| 76 | UWORD8 u1_num_skips); |
| 77 | |
| 78 | UWORD32 irc_estimate_bits_for_qp(rc_rd_model_handle ps_rd_model, |
| 79 | UWORD32 u4_estimated_sad, |
| 80 | UWORD8 u1_avg_qp); |
| 81 | |
| 82 | /* Get the Linear model coefficient */ |
| 83 | model_coeff irc_get_linear_coefficient(rc_rd_model_handle ps_rd_model); |
| 84 | |
| 85 | WORD32 irc_calc_per_frm_bits(rc_rd_model_handle ps_rd_model, |
| 86 | UWORD16 *pu2_num_pics_of_a_pic_type, |
| 87 | UWORD8 *pu1_update_pic_type_model, |
| 88 | UWORD8 u1_num_pic_types, |
| 89 | UWORD32 *pu4_num_skip_of_a_pic_type, |
| 90 | UWORD8 u1_base_pic_type, |
| 91 | float *pfl_gamma, |
| 92 | float *pfl_eta, |
| 93 | UWORD8 u1_curr_pic_type, |
| 94 | UWORD32 u4_bits_for_sub_gop, |
| 95 | UWORD32 u4_curr_estimated_sad, |
| 96 | UWORD8 *pu1_curr_pic_type_qp); |
| 97 | #endif |
| 98 | |