1*c83a76b0SSuyog Pawar /****************************************************************************** 2*c83a76b0SSuyog Pawar * 3*c83a76b0SSuyog Pawar * Copyright (C) 2018 The Android Open Source Project 4*c83a76b0SSuyog Pawar * 5*c83a76b0SSuyog Pawar * Licensed under the Apache License, Version 2.0 (the "License"); 6*c83a76b0SSuyog Pawar * you may not use this file except in compliance with the License. 7*c83a76b0SSuyog Pawar * You may obtain a copy of the License at: 8*c83a76b0SSuyog Pawar * 9*c83a76b0SSuyog Pawar * http://www.apache.org/licenses/LICENSE-2.0 10*c83a76b0SSuyog Pawar * 11*c83a76b0SSuyog Pawar * Unless required by applicable law or agreed to in writing, software 12*c83a76b0SSuyog Pawar * distributed under the License is distributed on an "AS IS" BASIS, 13*c83a76b0SSuyog Pawar * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14*c83a76b0SSuyog Pawar * See the License for the specific language governing permissions and 15*c83a76b0SSuyog Pawar * limitations under the License. 16*c83a76b0SSuyog Pawar * 17*c83a76b0SSuyog Pawar ***************************************************************************** 18*c83a76b0SSuyog Pawar * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore 19*c83a76b0SSuyog Pawar */ 20*c83a76b0SSuyog Pawar /*! 21*c83a76b0SSuyog Pawar ****************************************************************************** 22*c83a76b0SSuyog Pawar * \file rc_rd_model.h 23*c83a76b0SSuyog Pawar * 24*c83a76b0SSuyog Pawar * \brief 25*c83a76b0SSuyog Pawar * This file contains all the necessary declarations for 26*c83a76b0SSuyog Pawar * Rate Distortion related functions 27*c83a76b0SSuyog Pawar * 28*c83a76b0SSuyog Pawar * \date 29*c83a76b0SSuyog Pawar * 30*c83a76b0SSuyog Pawar * \author 31*c83a76b0SSuyog Pawar * ittiam 32*c83a76b0SSuyog Pawar * 33*c83a76b0SSuyog Pawar ****************************************************************************** 34*c83a76b0SSuyog Pawar */ 35*c83a76b0SSuyog Pawar #ifndef RC_RD_MODEL 36*c83a76b0SSuyog Pawar #define RC_RD_MODEL 37*c83a76b0SSuyog Pawar 38*c83a76b0SSuyog Pawar /*****************************************************************************/ 39*c83a76b0SSuyog Pawar /* Constant Macros */ 40*c83a76b0SSuyog Pawar /*****************************************************************************/ 41*c83a76b0SSuyog Pawar #define RC_FIXED_POINT 1 42*c83a76b0SSuyog Pawar #define MAX_FRAMES_MODELLED 16 43*c83a76b0SSuyog Pawar 44*c83a76b0SSuyog Pawar #if !RC_FIXED_POINT 45*c83a76b0SSuyog Pawar typedef float model_coeff; 46*c83a76b0SSuyog Pawar #else 47*c83a76b0SSuyog Pawar typedef number_t model_coeff; 48*c83a76b0SSuyog Pawar #endif 49*c83a76b0SSuyog Pawar typedef struct rc_rd_model_t *rc_rd_model_handle; 50*c83a76b0SSuyog Pawar 51*c83a76b0SSuyog Pawar /*****************************************************************************/ 52*c83a76b0SSuyog Pawar /* Function Declarations */ 53*c83a76b0SSuyog Pawar /*****************************************************************************/ 54*c83a76b0SSuyog Pawar WORD32 rc_rd_model_num_fill_use_free_memtab( 55*c83a76b0SSuyog Pawar rc_rd_model_handle *pps_rc_rd_model, itt_memtab_t *ps_memtab, ITT_FUNC_TYPE_E e_func_type); 56*c83a76b0SSuyog Pawar /* Interface Functions */ 57*c83a76b0SSuyog Pawar /* Initialise the rate distortion model */ 58*c83a76b0SSuyog Pawar void init_frm_rc_rd_model(rc_rd_model_handle ps_rd_model, UWORD8 u1_max_frames_modelled); 59*c83a76b0SSuyog Pawar 60*c83a76b0SSuyog Pawar /* Reset the rate distortion model */ 61*c83a76b0SSuyog Pawar void reset_frm_rc_rd_model(rc_rd_model_handle ps_rd_model); 62*c83a76b0SSuyog Pawar 63*c83a76b0SSuyog Pawar /* Returns the Qp to be used for the given bits and SAD */ 64*c83a76b0SSuyog Pawar WORD32 find_qp_for_target_bits( 65*c83a76b0SSuyog Pawar rc_rd_model_handle ps_rd_model, 66*c83a76b0SSuyog Pawar UWORD32 u4_target_res_bits, 67*c83a76b0SSuyog Pawar UWORD32 u4_estimated_sad, 68*c83a76b0SSuyog Pawar WORD32 i4_max_qp_q6, 69*c83a76b0SSuyog Pawar WORD32 i4_min_qp_q6); 70*c83a76b0SSuyog Pawar /* Updates the frame level statistics after encoding a frame */ 71*c83a76b0SSuyog Pawar void add_frame_to_rd_model( 72*c83a76b0SSuyog Pawar rc_rd_model_handle ps_rd_model, 73*c83a76b0SSuyog Pawar UWORD32 i4_res_bits, 74*c83a76b0SSuyog Pawar WORD32 i4_avg_mp2qp_q6, 75*c83a76b0SSuyog Pawar LWORD64 i8_sad_h264, 76*c83a76b0SSuyog Pawar UWORD8 u1_num_skips); 77*c83a76b0SSuyog Pawar 78*c83a76b0SSuyog Pawar UWORD32 estimate_bits_for_qp( 79*c83a76b0SSuyog Pawar rc_rd_model_handle ps_rd_model, UWORD32 u4_estimated_sad, WORD32 i4_avg_qp_q6); 80*c83a76b0SSuyog Pawar /* Get the Linear model coefficient */ 81*c83a76b0SSuyog Pawar model_coeff get_linear_coefficient(rc_rd_model_handle ps_rd_model); 82*c83a76b0SSuyog Pawar 83*c83a76b0SSuyog Pawar void set_linear_coefficient(rc_rd_model_handle ps_rd_model, number_t model_coeff_a_lin_wo_int); 84*c83a76b0SSuyog Pawar 85*c83a76b0SSuyog Pawar WORD32 is_model_valid(rc_rd_model_handle ps_rd_model); 86*c83a76b0SSuyog Pawar 87*c83a76b0SSuyog Pawar #endif 88