[cvs] / xvidcore / src / motion / motion_est.h Repository:
ViewVC logotype

Diff of /xvidcore/src/motion/motion_est.h

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1, Mon Sep 23 20:36:02 2002 UTC revision 1.1.2.10, Wed Dec 11 10:32:29 2002 UTC
# Line 0  Line 1 
1    /**************************************************************************
2     *
3     *  XVID MPEG-4 VIDEO CODEC
4     *  -  Motion estimation header  -
5     *
6     *  This program is an implementation of a part of one or more MPEG-4
7     *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
8     *  to use this software module in hardware or software products are
9     *  advised that its use may infringe existing patents or copyrights, and
10     *  any such use would be at such party's own risk.  The original
11     *  developer of this software module and his/her company, and subsequent
12     *  editors and their companies, will have no liability for use of this
13     *  software or modifications or derivatives thereof.
14     *
15     *  This program is free software; you can redistribute it and/or modify
16     *  it under the terms of the GNU General Public License as published by
17     *  the Free Software Foundation; either version 2 of the License, or
18     *  (at your option) any later version.
19     *
20     *  This program is distributed in the hope that it will be useful,
21     *  but WITHOUT ANY WARRANTY; without even the implied warranty of
22     *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23     *  GNU General Public License for more details.
24     *
25     *  You should have received a copy of the GNU General Public License
26     *  along with this program; if not, write to the Free Software
27     *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
28     *
29     *  $Id$
30     *
31     ***************************************************************************/
32    
33    #ifndef _MOTION_EST_H_
34    #define _MOTION_EST_H_
35    
36    #include "../portab.h"
37    #include "../global.h"
38    
39    /* hard coded motion search parameters for motion_est and smp_motion_est */
40    
41    // very large value
42    #define MV_MAX_ERROR    (4096 * 256)
43    
44    /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */
45    #define MV16_INTER_BIAS 512
46    
47    /* vector map (vlc delta size) smoother parameters ! float !*/
48    #define NEIGH_TEND_16X16        10.5
49    #define NEIGH_TEND_8X8          4.0
50    #define NEIGH_8X8_BIAS          30
51    
52    /* Parameters which control inter/inter4v decision */
53    #define IMV16X16                        2
54    
55    static const int lambda_vec16[32] =
56            {     0    ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5),
57            (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5),
58            (int)(1.49591*NEIGH_TEND_16X16 + 0.5), (int)(1.68601*NEIGH_TEND_16X16 + 0.5),
59            (int)(1.89187*NEIGH_TEND_16X16 + 0.5), (int)(2.11542*NEIGH_TEND_16X16 + 0.5),
60            (int)(2.35878*NEIGH_TEND_16X16 + 0.5), (int)(2.62429*NEIGH_TEND_16X16 + 0.5),
61            (int)(2.91455*NEIGH_TEND_16X16 + 0.5), (int)(3.23253*NEIGH_TEND_16X16 + 0.5),
62            (int)(3.58158*NEIGH_TEND_16X16 + 0.5), (int)(3.96555*NEIGH_TEND_16X16 + 0.5),
63            (int)(4.38887*NEIGH_TEND_16X16 + 0.5), (int)(4.85673*NEIGH_TEND_16X16 + 0.5),
64            (int)(5.37519*NEIGH_TEND_16X16 + 0.5), (int)(5.95144*NEIGH_TEND_16X16 + 0.5),
65            (int)(6.59408*NEIGH_TEND_16X16 + 0.5), (int)(7.31349*NEIGH_TEND_16X16 + 0.5),
66            (int)(8.12242*NEIGH_TEND_16X16 + 0.5), (int)(9.03669*NEIGH_TEND_16X16 + 0.5),
67            (int)(10.0763*NEIGH_TEND_16X16 + 0.5), (int)(11.2669*NEIGH_TEND_16X16 + 0.5),
68            (int)(12.6426*NEIGH_TEND_16X16 + 0.5), (int)(14.2493*NEIGH_TEND_16X16 + 0.5),
69            (int)(16.1512*NEIGH_TEND_16X16 + 0.5), (int)(18.442*NEIGH_TEND_16X16 + 0.5),
70            (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5),
71            (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5)    };
72    
73    static const int lambda_vec8[32] =
74            {     0    ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5),
75            (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5),
76            (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5),
77            (int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5),
78            (int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5),
79            (int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5),
80            (int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5),
81            (int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5),
82            (int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5),
83            (int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5),
84            (int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5),
85            (int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5),
86            (int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5),
87            (int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5),
88            (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5),
89            (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5)        };
90    
91    // mv.length table
92    static const uint32_t mvtab[33] = {
93            1, 2, 3, 4, 6, 7, 7, 7,
94            9, 9, 9, 10, 10, 10, 10, 10,
95            10, 10, 10, 10, 10, 10, 10, 10,
96            10, 11, 11, 11, 11, 11, 11, 12, 12
97    };
98    
99    static const int DQtab[4] = {
100            -1, -2, 1, 2
101    };
102    
103    #define RRV_MV_SCALEDOWN(a)     ( (a)>=0 ? (a+1)/2 : (a-1)/2 )
104    #define RRV_MV_SCALEUP(a)       ( (a)>0 ? 2*(a)-1 : (a)<0 ? 2*(a)+1 : (a) )
105    
106    typedef struct
107            {
108    // general fields
109                    int max_dx, min_dx, max_dy, min_dy;
110                    uint32_t rounding;
111                    VECTOR predMV;
112                    VECTOR *currentMV;
113                    VECTOR *currentQMV;
114                    int32_t *iMinSAD;
115                    const uint8_t * Ref;
116                    const uint8_t * RefH;
117                    const uint8_t * RefV;
118                    const uint8_t * RefHV;
119                    const uint8_t * RefCU;
120                    const uint8_t * RefCV;
121                    const uint8_t * CurU;
122                    const uint8_t * CurV;
123                    uint8_t * RefQ;
124                    const uint8_t * Cur;
125                    uint32_t lambda16;
126                    uint32_t lambda8;
127                    uint32_t iEdgedWidth;
128                    uint32_t iFcode;
129                    int * temp;
130                    int qpel, qpel_precision;
131                    int chroma;
132                    int rrv;
133    //fields for interpolate and direct mode
134                    const uint8_t *bRef;
135                    const uint8_t *bRefH;
136                    const uint8_t *bRefV;
137                    const uint8_t *bRefHV;
138                    VECTOR bpredMV;
139                    uint32_t bFcode;
140    // fields for direct mode
141                    VECTOR directmvF[4];
142                    VECTOR directmvB[4];
143                    const VECTOR * referencemv;
144            }
145            SearchData;
146    
147    
148    typedef void(CheckFunc)(const int x, const int y,
149                                                    const int Direction, int * const dir,
150                                                    const SearchData * const Data);
151    
152    static CheckFunc CheckCandidate16, CheckCandidate16no4v, CheckCandidateInt,
153                            CheckCandidateDirect, CheckCandidateDirectno4v,
154                            CheckCandidate8;
155    CheckFunc *CheckCandidate;
156    
157    /*
158     * Calculate the min/max range (in halfpixels)
159     * relative to the _MACROBLOCK_ position
160     */
161    static void __inline
162    get_range(int32_t * const min_dx,
163                      int32_t * const max_dx,
164                      int32_t * const min_dy,
165                      int32_t * const max_dy,
166                      const uint32_t x,
167                      const uint32_t y,
168                      const uint32_t block_sz,      /* block dimension, 8 or 16 */
169                      const uint32_t width,
170                      const uint32_t height,
171                      const uint32_t fcode)
172    {
173            int k;
174            const int search_range = 32 << (fcode - 1);
175            const int high = search_range - 1;
176            const int low = -search_range;
177    
178            k = 2 * (int)(width - x*block_sz);
179            *max_dx = MIN(high, k);
180            k = 2 * (int)(height -  y*block_sz);
181            *max_dy = MIN(high, k);
182    
183            k = -2 * (int)((x+1) * block_sz);
184            *min_dx = MAX(low, k);
185            k = -2 * (int)((y+1) * block_sz);
186            *min_dy = MAX(low, k);
187    }
188    
189    static void __inline
190    get_range_rrv(int32_t * const min_dx,
191                      int32_t * const max_dx,
192                      int32_t * const min_dy,
193                      int32_t * const max_dy,
194                      const uint32_t x,
195                      const uint32_t y,
196                      const uint32_t block_sz,      /* block dimension, 8 or 16 */
197                      const uint32_t width,
198                      const uint32_t height,
199                      const uint32_t fcode)
200    {
201            int k;
202            const int search_range = 32 << (fcode - 1);
203            const int high = RRV_MV_SCALEUP(search_range - 1); //halfzero -> halfpel, because we do our search in halfpel
204            const int low = RRV_MV_SCALEUP(-search_range);
205    
206            k = 2 * (int)(width - x*block_sz);
207            *max_dx = MIN(high, k);
208            k = 2 * (int)(height -  y*block_sz);
209            *max_dy = MIN(high, k);
210    
211            k = -2 * (int)((x+1) * block_sz);
212            *min_dx = MAX(low, k);
213            k = -2 * (int)((y+1) * block_sz);
214            *min_dy = MAX(low, k);
215    }
216    
217    static void __inline
218    get_range_qpel(int32_t * const min_dx,
219                      int32_t * const max_dx,
220                      int32_t * const min_dy,
221                      int32_t * const max_dy,
222                      const uint32_t x,
223                      const uint32_t y,
224                      const uint32_t block_sz,      /* block dimension, 8 or 16 */
225                      const uint32_t width,
226                      const uint32_t height,
227                      const uint32_t fcode)
228    {
229            int k;
230            const int search_range = 32 << (fcode - 1);
231            const int high = search_range - 1;
232            const int low = -search_range;
233    
234            k = 4 * (int)(width - x*block_sz);
235            *max_dx = MIN(high, k);
236            k = 4 * (int)(height -  y*block_sz);
237            *max_dy = MIN(high, k);
238    
239            k = -4 * (int)((x+1) * block_sz);
240            *min_dx = MAX(low, k);
241            k = -4 * (int)((y+1) * block_sz);
242            *min_dy = MAX(low, k);
243    }
244    
245    
246    typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection);
247    
248    static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch;
249    
250    static void Search8(const SearchData * const OldData,
251                            const int x, const int y,
252                            const uint32_t MotionFlags,
253                            const MBParam * const pParam,
254                            MACROBLOCK * const pMB,
255                            const MACROBLOCK * const pMBs,
256                            const int block,
257                            SearchData * const Data);
258    
259    bool
260    MotionEstimation(MBParam * const pParam,
261                                     FRAMEINFO * const current,
262                                     FRAMEINFO * const reference,
263                                     const IMAGE * const pRefH,
264                                     const IMAGE * const pRefV,
265                                     const IMAGE * const pRefHV,
266                                     const uint32_t iLimit);
267    
268    static void
269    SearchP(const IMAGE * const pRef,
270                    const uint8_t * const pRefH,
271                    const uint8_t * const pRefV,
272                    const uint8_t * const pRefHV,
273                    const IMAGE * const pCur,
274                    const int x,
275                    const int y,
276                    const uint32_t MotionFlags,
277                    const uint32_t iQuant,
278                    SearchData * const Data,
279                    const MBParam * const pParam,
280                    const MACROBLOCK * const pMBs,
281                    const MACROBLOCK * const prevMBs,
282                    int inter4v,
283                    MACROBLOCK * const pMB);
284    
285    #endif                                                  /* _MOTION_EST_H_ */
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.1.2.10
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4