03678c827d3d6b0004541e461755ccacc32fa65d
[x262.git] / encoder / me.h
1 /*****************************************************************************
2  * me.h: h264 encoder library (Motion Estimation)
3  *****************************************************************************
4  * Copyright (C) 2003 Laurent Aimar
5  * $Id: me.h,v 1.1 2004/06/03 19:27:08 fenrir Exp $
6  *
7  * Authors: 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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 #ifndef _ME_H
25 #define _ME_H 1
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     int mvp[2];
43
44     /* output */
45     int cost_mv;        /* lambda * nbits for the chosen mv */
46     int cost;           /* satd + lambda * nbits */
47     int mv[2];
48 } x264_me_t;
49
50 void x264_me_search_ref( x264_t *h, x264_me_t *m, int (*mvc)[2], int i_mvc, int *p_fullpel_thresh );
51 static inline void x264_me_search( x264_t *h, x264_me_t *m, int (*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 int x264_me_refine_bidir( x264_t *h, x264_me_t *m0, x264_me_t *m1, int i_weight );
56
57 #endif