c++H264编解码源代码

/*! *************************************************************************** * \file rdopt.c * * \brief * Rate-Distortion optimized mode decision * * \author * - Heiko Schwarz <[email protected]> * - Valeri George <[email protected]> * - Lowell Winger <[email protected]> * - Alexis Michael Tourapis <[email protected]> * \date * 12. April 2001 ************************************************************************** */ #include <stdlib.h> #include <math.h> #include <assert.h> #include <limits.h> #include "global.h" #include "rdopt_coding_state.h" #include "memalloc.h" #include "mb_access.h" #include "elements.h" #include "intrarefresh.h" #include "image.h" #include "transform8x8.h" #include "cabac.h" // head file for rate control #include "vlc.h" #include "fast_me.h" #include "ratectl.h" // head file for rate control #define KS_MV //Rate control int QP,QP2; int DELTA_QP,DELTA_QP2; int diffy[16][16]; static int pred[16][16]; extern int QP2QUANT [40]; //==== MODULE PARAMETERS ==== int best_mode; imgpel rec_mbY[16][16], rec_mbU[16][16], rec_mbV[16][16], rec_mbY8x8[16][16]; // reconstruction values imgpel mpr8x8[16][16]; int ****cofAC=NULL, ****cofAC8x8=NULL; // [8x8block][4x4block][level/run][scan_pos] int ***cofDC=NULL; // [yuv][level/run][scan_pos] int **cofAC4x4=NULL, ****cofAC4x4intern=NULL; // [level/run][scan_pos] int cbp, cbp8x8, cnt_nonz_8x8; int64 cbp_blk; int cbp_blk8x8; short frefframe[4][4], brefframe[4][4], b8mode[4], b8pdir[4]; short best8x8mode [4]; // [block] short best8x8pdir [MAXMODE][4]; // [mode][block] short best8x8fwref [MAXMODE][4]; // [mode][block] short best8x8bwref [MAXMODE][4]; // [mode][block] CSptr cs_mb=NULL, cs_b8=NULL, cs_cm=NULL, cs_imb=NULL, cs_ib8=NULL, cs_ib4=NULL, cs_pc=NULL; int best_c_imode; int best_i16offset; imgpel temp_imgY[16][16]; // to temp store the Y data for 8x8 transform imgpel temp_imgU[16][16]; imgpel temp_imgV[16][16]; int best_transform_size[4]; //mixed transform sizes definitions int luma_transform_size_8x8_flag; int NoMbPartLessThan8x8Flag; short part8x8pdir[4]; short part8x8fwref[4]; short part8x8bwref[4]; short mv8x8[2][4][4][2]; //[LIST][block_x][block_y][MVx/MVy] short ref8x8[2][4][4]; //[LIST][block_x][block_y] short all_mv8x8[2][2][4][4][2]; //[8x8_data/temp_data][LIST][block_x][block_y][MVx/MVy] short pred_mv8x8[2][2][4][4][2]; //ADD-VG-28062004 int ****cofAC_8x8ts = NULL; // [8x8block][4x4block][level/run][scan_pos] imgpel rec_mbY_8x8ts[16][16]; imgpel mpr_8x8ts[16][16]; int64 cbp_blk8_8x8ts; int cbp8_8x8ts; int cost8_8x8ts; int cnt_nonz8_8x8ts; int part8x8mode[4]; //ADD-VG-28062004-END void StoreMV8x8(int dir); void RestoreMV8x8(int dir); // end of mixed transform sizes definitions // Residue Color Transform int cofAC4x4_chroma[2][2][18]; int rec_resG_8x8[16][16], resTrans_R_8x8[16][16], resTrans_B_8x8[16][16]; int rec_resG_8x8ts[16][16], resTrans_R_8x8ts[16][16], resTrans_B_8x8ts[16][16]; int mprRGB_8x8[3][16][16], mprRGB_8x8ts[3][16][16]; int b4_ipredmode[16], b4_intra_pred_modes[16]; /*! ************************************************************************ * \brief * delete structure for RD-optimized mode decision ************************************************************************ */ void clear_rdopt () { free_mem_DCcoeff (cofDC); free_mem_ACcoeff (cofAC); free_mem_ACcoeff (cofAC8x8); free_mem_ACcoeff (cofAC4x4intern); if (input->AllowTransform8x8) { free_mem_ACcoeff (cofAC_8x8ts); } // structure for saving the coding state delete_coding_state (cs_mb); delete_coding_state (cs_b8); delete_coding_state (cs_cm); delete_coding_state (cs_imb); delete_coding_state (cs_ib8); delete_coding_state (cs_ib4); delete_coding_state (cs_pc); } /*! ************************************************************************ * \brief * create structure for RD-optimized mode decision ************************************************************************ */ void init_rdopt () { rdopt = NULL; get_mem_DCcoeff (&cofDC); get_mem_ACcoeff (&cofAC); get_mem_ACcoeff (&cofAC8x8); get_mem_ACcoeff (&cofAC4x4intern); cofAC4x4 = cofAC4x4intern[0][0]; if (input->AllowTransform8x8) { get_mem_ACcoeff (&cofAC_8x8ts); } // structure for saving the coding state cs_mb = create_coding_state (); cs_b8 = create_coding_state (); cs_cm = create_coding_state (); cs_imb = create_coding_state (); cs_ib8 = create_coding_state (); cs_ib4 = create_coding_state (); cs_pc = create_coding_state (); } /*! ************************************************************************************* * \brief * Updates the pixel map that shows, which reference frames are reliable for * each MB-area of the picture. * * \note * The new values of the pixel_map are taken from the temporary buffer refresh_map * ************************************************************************************* */ void UpdatePixelMap() { int mx,my,y,x,i,j; if (img->type==I_SLICE) { for (y=0; y<img->height; y++) for (x=0; x<img->width; x++) { pixel_map[y][x]=1; } } else { for (my=0; my<img->height/8; my++) for (mx=0; mx<img->width/8; mx++) { j = my*8 + 8; i = mx*8 + 8; if (refresh_map[my][mx]) { for (y=my*8; y<j; y++) for (x=mx*8; x<i; x++) pixel_map[y][x] = 1; } else { for (y=my*8; y<j; y++) for (x=mx*8; x<i; x++) pixel_map[y][x] = min(pixel_map[y][x]+1, input->num_ref_frames+1); } } } } /*! ************************************************************************************* * \brief * Checks if a given reference frame is reliable for the current * macroblock, given the motion vectors that the motion search has * returned. * * \return * If the return value is 1, the reference frame is reliable. If it * is 0, then it is not reliable. * * \note * A specific area in each reference frame is assumed to be unreliable * if the same area has been intra-refreshed in a subsequent frame. * The information about intra-refreshed areas is kept in the pixel_map. * ************************************************************************************* */ int CheckReliabilityOfRef (int block, int list_idx, int ref, int mode) { int y,x, block_y, block_x, dy, dx, y_pos, x_pos, yy, xx, pres_x, pres_y; int maxold_x = img->width-1; int maxold_y = img->height-1; int ref_frame = ref+1; int by0 = (mode>=4?2*(block/2):mode==2?2*block:0); int by1 = by0 + (mode>=4||mode==2?2:4); int bx0 = (mode>=4?2*(block%2):mode==3?2*block:0); int bx1 = bx0 + (mode>=4||mode==3?2:4); for (block_y=by0; block_y<by1; block_y++) for (block_x=bx0; block_x<bx1; block_x++) { y_pos = img->all_mv[block_x][block_y][list_idx][ref][mode][1]; y_pos += (img->block_y+block_y) * BLOCK_SIZE * 4; x_pos = img->all_mv[block_x][block_y][list_idx][ref][mode][0]; x_pos += (img->block_x+block_x) * BLOCK_SIZE * 4; /* Here we specify which pixels of the reference frame influence the reference values and check their reliability. This is based on the function Get_Reference_Pixel */ dy = y_pos & 3; dx = x_pos & 3; y_pos = (y_pos-dy)/4; x_pos = (x_pos-dx)/4; if (dy==0 && dx==0) //full-pel { for (y=0 ; y < BLOCK_SIZE ; y++) for (x=0 ; x < BLOCK_SI