H.264/AVC Reference Software Encoder: me

00001 
00002 /*!
00003  ************************************************************************
00004  *
00005  * \file me_umhex.h
00006  *
00007  * \brief
00008  *   Macro definitions and global variables for UMHEX fast
00009  *   integer pel motion estimation and fractional pel motion estimation
00010  *
00011  * \author
00012  *   Main contributors: (see contributors.h for copyright, address and affiliation details)
00013  *    - Zhibo Chen         <chenzhibo@tsinghua.org.cn>
00014  *    - JianFeng Xu        <fenax@video.mdc.tsinghua.edu.cn>
00015  *    - Wenfang Fu         <fwf@video.mdc.tsinghua.edu.cn>
00016  *    - Xiaozhong Xu       <xxz@video.mdc.tsinghua.edu.cn>
00017  *
00018  * \date
00019  *   2006.1
00020  ************************************************************************
00021  */
00022 
00023 #ifndef _ME_UMHEX_H_
00024 #define _ME_UMHEX_H_
00025 
00026 #include "mbuffer.h"
00027 
00028 #define EARLY_TERMINATION                                                             \
00029   if ((min_mcost-pred_SAD)<pred_SAD*betaFourth_2)                                     \
00030   goto fourth_2_step;                                                                 \
00031   else if((min_mcost-pred_SAD)<pred_SAD*betaFourth_1)                                 \
00032   goto fourth_1_step;
00033 
00034 #define SEARCH_ONE_PIXEL                                                              \
00035   if(iabs(cand_x - center_x) <=search_range && iabs(cand_y - center_y)<= search_range)\
00036   {                                                                                   \
00037     if(!McostState[cand_y-center_y+search_range][cand_x-center_x+search_range])       \
00038     {                                                                                 \
00039       mcost = MV_COST (lambda_factor, mvshift, cand_x, cand_y, pred_x, pred_y);       \
00040       if(mcost<min_mcost)                                                             \
00041       {                                                                               \
00042         mcost += computeUniPred[dist_method](orig_pic,                                \
00043         blocksize_y,blocksize_x, min_mcost - mcost,                                   \
00044         (cand_x << 2) + IMG_PAD_SIZE_TIMES4, (cand_y << 2) + IMG_PAD_SIZE_TIMES4);    \
00045         McostState[cand_y-center_y+search_range][cand_x-center_x+search_range] = 1;   \
00046         if (mcost < min_mcost)                                                        \
00047         {                                                                             \
00048           best_x = cand_x;                                                            \
00049           best_y = cand_y;                                                            \
00050           min_mcost = mcost;                                                          \
00051         }                                                                             \
00052       }                                                                               \
00053     }                                                                                 \
00054    }
00055 
00056 #define SEARCH_ONE_PIXEL_BIPRED                                                       \
00057 if(iabs(cand_x - center2_x) <=search_range && iabs(cand_y - center2_y)<= search_range)\
00058 {                                                                                     \
00059   if(!McostState[cand_y-center2_y+search_range][cand_x-center2_x+search_range])       \
00060   {                                                                                   \
00061     mcost  = MV_COST (lambda_factor, mvshift, center1_x, center1_y, pred_x1, pred_y1);\
00062     mcost += MV_COST (lambda_factor, mvshift, cand_x, cand_y, pred_x2, pred_y2);      \
00063   if(mcost<min_mcost)                                                                 \
00064   {                                                                                   \
00065       mcost  += computeBiPred(cur_pic, blocksize_y, blocksize_x,                      \
00066       min_mcost - mcost,                                                              \
00067       (center1_x << 2) + IMG_PAD_SIZE_TIMES4,                                         \
00068       (center1_y << 2) + IMG_PAD_SIZE_TIMES4,                                         \
00069       (cand_x << 2) + IMG_PAD_SIZE_TIMES4,                                            \
00070       (cand_y << 2) + IMG_PAD_SIZE_TIMES4);                                           \
00071       McostState[cand_y-center2_y+search_range][cand_x-center2_x+search_range] = 1;   \
00072       if (mcost < min_mcost)                                                          \
00073       {                                                                               \
00074         best_x = cand_x;                                                              \
00075         best_y = cand_y;                                                              \
00076         min_mcost = mcost;                                                            \
00077       }                                                                               \
00078     }                                                                                   \
00079   }                                                                                   \
00080 }
00081 
00082 byte **McostState;                          //!< state for integer pel search
00083 byte **SearchState;                         //!< state for fractional pel search
00084 
00085 int ****fastme_ref_cost;                    //!< store SAD information needed for forward ref-frame prediction
00086 int ***fastme_l0_cost;                      //!< store SAD information needed for forward median and uplayer prediction
00087 int ***fastme_l1_cost;                      //!< store SAD information needed for backward median and uplayer prediction
00088 int ***fastme_l0_cost_bipred;               //!< store SAD information for bipred mode
00089 int ***fastme_l1_cost_bipred;               //!< store SAD information for bipred mode
00090 int bipred_flag;                            //!< flag for bipred
00091 int **fastme_best_cost;                     //!< for multi ref early termination threshold
00092 int pred_SAD;                               //!<  SAD prediction in use.
00093 int pred_MV_ref[2], pred_MV_uplayer[2];     //!< pred motion vector by space or temporal correlation,Median is provided
00094 
00095 int UMHEX_blocktype;                        //!< blocktype for UMHEX SetMotionVectorPredictor
00096 int predict_point[5][2];
00097 int SAD_a,SAD_b,SAD_c,SAD_d;
00098 int Threshold_DSR_MB[8];                    //!<  Threshold for usage of DSR. DSR refer to JVT-Q088
00099 //for early termination
00100 float  Bsize[8];
00101 float AlphaFourth_1[8];
00102 float AlphaFourth_2[8];
00103 byte *flag_intra;
00104 int  flag_intra_SAD;
00105 
00106 void UMHEX_DefineThreshold(void);
00107 void UMHEX_DefineThresholdMB(void);
00108 int  UMHEX_get_mem(void);
00109 void UMHEX_free_mem(void);
00110 
00111 void UMHEX_decide_intrabk_SAD(void);
00112 void UMHEX_skip_intrabk_SAD(int best_mode, int ref_max);
00113 void UMHEX_setup(short ref, int list, int block_y, int block_x, int blocktype, short   ******all_mv);
00114 
00115 int                                     //  ==> minimum motion cost after search
00116 UMHEXIntegerPelBlockMotionSearch  (Macroblock *currMB,     // <--  current Macroblock
00117                                   imgpel   *orig_pic,      // <--  not used
00118                                   short     ref,           // <--  reference frame (0... or -1 (backward))
00119                                   int       list,          // <--  reference picture list
00120                                   char   ***refPic,        // <--  reference array
00121                                   short ****tmp_mv,        // <--  mv array
00122                                   int       pic_pix_x,     // <--  absolute x-coordinate of regarded AxB block
00123                                   int       pic_pix_y,     // <--  absolute y-coordinate of regarded AxB block
00124                                   int       blocktype,     // <--  block type (1-16x16 ... 7-4x4)
00125                                   MotionVector *pred,      // <--  motion vector predictor (x) in sub-pel units
00126                                   MotionVector   *mv,      //  --> motion vector (x) - in pel units
00127                                   int       search_range,  // <--  1-d search range in pel units
00128                                   int       min_mcost,     // <--  minimum motion cost (cost for center or huge value)
00129                                   int       lambda_factor, // <--  lagrangian parameter for determining motion cost
00130                                   int       apply_weights
00131                                   );
00132 
00133 int                                                   //  ==> minimum motion cost after search
00134 UMHEXSubPelBlockMotionSearch (
00135                              imgpel*   orig_pic,      // <--  original pixel values for the AxB block
00136                              short       ref,         // <--  reference frame (0... or -1 (backward))
00137                              int       list,          // <--  reference picture list
00138                              int       list_offset,   // <--  MBAFF list offset
00139                              int       pic_pix_x,     // <--  absolute x-coordinate of regarded AxB block
00140                              int       pic_pix_y,     // <--  absolute y-coordinate of regarded AxB block
00141                              int       blocktype,     // <--  block type (1-16x16 ... 7-4x4)
00142                              MotionVector *pred,      // <--  motion vector predictor (x) in sub-pel units
00143                              MotionVector  *mv,       // <--> in: search center (x) / out: motion vector (x) - in pel units
00144                              int       search_pos2,   // <--  search positions for    half-pel search  (default: 9)
00145                              int       search_pos4,   // <--  search positions for quarter-pel search  (default: 9)
00146                              int       min_mcost,     // <--  minimum motion cost (cost for center or huge value)
00147                              int       lambda_factor, // <--  lagrangian parameter for determining motion cost
00148                              int       apply_weights
00149                              );
00150 
00151 extern int UMHEXSubPelBlockME ( //  ==> minimum motion cost after search
00152                              imgpel*   orig_pic,      // <--  original pixel values for the AxB block
00153                              short       ref,         // <--  reference frame (0... or -1 (backward))
00154                              int       list,          // <--  reference picture list
00155                              int       list_offset,   // <--  MBAFF list offset
00156                              int       pic_pix_x,     // <--  absolute x-coordinate of regarded AxB block
00157                              int       pic_pix_y,     // <--  absolute y-coordinate of regarded AxB block
00158                              int       blocktype,     // <--  block type (1-16x16 ... 7-4x4)
00159                              MotionVector *pred,      // <--  motion vector predictor (x) in sub-pel units
00160                              MotionVector  *mv,       // <--> in: search center (x) / out: motion vector (x) - in pel units
00161                              int       search_pos2,   // <--  search positions for    half-pel search  (default: 9)
00162                              int       search_pos4,   // <--  search positions for quarter-pel search  (default: 9)
00163                              int       min_mcost,     // <--  minimum motion cost (cost for center or huge value)
00164                              int*      lambda, // <--  lagrangian parameter for determining motion cost
00165                              int       apply_weights
00166                              );
00167 
00168 
00169 extern int                                         //  ==> minimum motion cost after search
00170 SubPelBlockMotionSearch (imgpel*   orig_pic,       // <--  original pixel values for the AxB block
00171                          short     ref,            // <--  reference frame (0... or -1 (backward))
00172                          int       list,
00173                          int       list_offset,   // <--  MBAFF list offset
00174                          int       pic_pix_x,      // <--  absolute x-coordinate of regarded AxB block
00175                          int       pic_pix_y,      // <--  absolute y-coordinate of regarded AxB block
00176                          int       blocktype,      // <--  block type (1-16x16 ... 7-4x4)
00177                          MotionVector *pred,     // <--  motion vector predictor in sub-pel units
00178                          MotionVector *mv,          // <--> in: search center  / out: motion vector - in pel units
00179                          int       search_pos2,    // <--  search positions for    half-pel search  (default: 9)
00180                          int       search_pos4,    // <--  search positions for quarter-pel search  (default: 9)
00181                          int       min_mcost,      // <--  minimum motion cost (cost for center or huge value)
00182                          int       *lambda_factor, // <--  lagrangian parameter for determining motion cost
00183                          int       apply_weights);
00184 
00185 int                                                //  ==> minimum motion cost after search
00186 UMHEXBipredIntegerPelBlockMotionSearch (Macroblock *currMB, // <--  current Macroblock
00187                                        imgpel*   orig_pic,      // <--  original pixel values for the AxB block
00188                                        short     ref,           // <--  reference frame (0... or -1 (backward))
00189                                        int       list,          // <--  Current reference list
00190                                        char   ***refPic,        // <--  reference array
00191                                        short ****tmp_mv,        // <--  mv array
00192                                        int       pic_pix_x,     // <--  absolute x-coordinate of regarded AxB block
00193                                        int       pic_pix_y,     // <--  absolute y-coordinate of regarded AxB block
00194                                        int       blocktype,     // <--  block type (1-16x16 ... 7-4x4)
00195                                        MotionVector *pred1,   // <--  motion vector predictor (x|y) in sub-pel units
00196                                        MotionVector *pred2,   // <--  motion vector predictor (x|y) in sub-pel units
00197                                        MotionVector   *mv1,         // <--> in: search center (x) / out: motion vector (x) - in pel units
00198                                        MotionVector   *mv2,       // <--> in: search center (x) / out: motion vector (x) - in pel units
00199                                        int       search_range,  // <--  1-d search range in pel units
00200                                        int       min_mcost,     // <--  minimum motion cost (cost for center or huge value)
00201                                        int       iteration_no,  // <--  bi pred iteration number
00202                                        int       lambda_factor, // <--  lagrangian parameter for determining motion cost
00203                                        int       apply_weights
00204                                        );
00205 
00206 void UMHEXSetMotionVectorPredictor (Macroblock *currMB, short pmv[2], char **refPic, short ***tmp_mv,
00207                                     short  ref_frame, int list, int mb_x, int mb_y, int bl_x, int bl_y, int *search_range);
00208 
00209 #endif