git.sesse.net Git - x264/blob - encoder/me.h

   1 /*****************************************************************************
   2  * me.h: h264 encoder library (Motion Estimation)
   3  *****************************************************************************
   4  * Copyright (C) 2003-2008 x264 project
   5  *
   6  * Authors: Loren Merritt <lorenm@u.washington.edu>
   7  *          Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 #ifndef X264_ME_H
  25 #define X264_ME_H
  26
  27 #define COST_MAX (1<<28)
  28
  29 typedef struct
  30 {
  31     /* input */
  32     int      i_pixel;   /* PIXEL_WxH */
  33     int16_t *p_cost_mv; /* lambda * nbits for each possible mv */
  34     int      i_ref_cost;
  35     int      i_ref;
  36
  37     uint8_t *p_fref[6];
  38     uint8_t *p_fenc[3];
  39     uint16_t *integral;
  40     int      i_stride[2];
  41
  42     DECLARE_ALIGNED_4( int16_t mvp[2] );
  43
  44     /* output */
  45     int cost_mv;        /* lambda * nbits for the chosen mv */
  46     int cost;           /* satd + lambda * nbits */
  47     DECLARE_ALIGNED_4( int16_t mv[2] );
  48 } DECLARE_ALIGNED_16( x264_me_t );
  49
  50 void x264_me_search_ref( x264_t *h, x264_me_t *m, int16_t (*mvc)[2], int i_mvc, int *p_fullpel_thresh );
  51 static inline void x264_me_search( x264_t *h, x264_me_t *m, int16_t (*mvc)[2], int i_mvc )
  52     { x264_me_search_ref( h, m, mvc, i_mvc, NULL ); }
  53
  54 void x264_me_refine_qpel( x264_t *h, x264_me_t *m );
  55 void x264_me_refine_qpel_rd( x264_t *h, x264_me_t *m, int i_lambda2, int i8 );
  56 int x264_me_refine_bidir( x264_t *h, x264_me_t *m0, x264_me_t *m1, int i_weight );
  57 int x264_rd_cost_part( x264_t *h, int i_lambda2, int i8, int i_pixel );
  58
  59 extern uint16_t *x264_cost_mv_fpel[52][4];
  60
  61 #define COPY1_IF_LT(x,y)\
  62 if((y)<(x))\
  63     (x)=(y);
  64
  65 #define COPY2_IF_LT(x,y,a,b)\
  66 if((y)<(x))\
  67 {\
  68     (x)=(y);\
  69     (a)=(b);\
  70 }
  71
  72 #define COPY3_IF_LT(x,y,a,b,c,d)\
  73 if((y)<(x))\
  74 {\
  75     (x)=(y);\
  76     (a)=(b);\
  77     (c)=(d);\
  78 }
  79
  80 #define COPY4_IF_LT(x,y,a,b,c,d,f,e)\
  81 if((y)<(x))\
  82 {\
  83     (x)=(y);\
  84     (a)=(b);\
  85     (c)=(d);\
  86     (f)=(e);\
  87 }
  88
  89 #endif