[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.2, Sat Feb 15 15:22:18 2003 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    #include "../image/reduced.h"
39    
40    /* hard coded motion search parameters for motion_est and smp_motion_est */
41    
42    // very large value
43    #define MV_MAX_ERROR    (4096 * 256)
44    
45    /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */
46    #define MV16_INTER_BIAS 512
47    
48    /* vector map (vlc delta size) smoother parameters ! float !*/
49    #define NEIGH_TEND_16X16        10.5
50    #define NEIGH_TEND_8X8          40.0
51    #define NEIGH_8X8_BIAS          30
52    
53    /* Parameters which control inter/inter4v decision */
54    #define IMV16X16                        2
55    
56    static const int lambda_vec16[32] =
57            {     0    ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5),
58            (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5),
59            (int)(1.49591*NEIGH_TEND_16X16 + 0.5), (int)(1.68601*NEIGH_TEND_16X16 + 0.5),
60            (int)(1.89187*NEIGH_TEND_16X16 + 0.5), (int)(2.11542*NEIGH_TEND_16X16 + 0.5),
61            (int)(2.35878*NEIGH_TEND_16X16 + 0.5), (int)(2.62429*NEIGH_TEND_16X16 + 0.5),
62            (int)(2.91455*NEIGH_TEND_16X16 + 0.5), (int)(3.23253*NEIGH_TEND_16X16 + 0.5),
63            (int)(3.58158*NEIGH_TEND_16X16 + 0.5), (int)(3.96555*NEIGH_TEND_16X16 + 0.5),
64            (int)(4.38887*NEIGH_TEND_16X16 + 0.5), (int)(4.85673*NEIGH_TEND_16X16 + 0.5),
65            (int)(5.37519*NEIGH_TEND_16X16 + 0.5), (int)(5.95144*NEIGH_TEND_16X16 + 0.5),
66            (int)(6.59408*NEIGH_TEND_16X16 + 0.5), (int)(7.31349*NEIGH_TEND_16X16 + 0.5),
67            (int)(8.12242*NEIGH_TEND_16X16 + 0.5), (int)(9.03669*NEIGH_TEND_16X16 + 0.5),
68            (int)(10.0763*NEIGH_TEND_16X16 + 0.5), (int)(11.2669*NEIGH_TEND_16X16 + 0.5),
69            (int)(12.6426*NEIGH_TEND_16X16 + 0.5), (int)(14.2493*NEIGH_TEND_16X16 + 0.5),
70            (int)(16.1512*NEIGH_TEND_16X16 + 0.5), (int)(18.442*NEIGH_TEND_16X16 + 0.5),
71            (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5),
72            (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5)    };
73    
74    static const int lambda_vec8[32] =
75            {     0    ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5),
76            (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5),
77            (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5),
78            (int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5),
79            (int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5),
80            (int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5),
81            (int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5),
82            (int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5),
83            (int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5),
84            (int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5),
85            (int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5),
86            (int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5),
87            (int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5),
88            (int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5),
89            (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5),
90            (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5)        };
91    
92    // mv.length table
93    static const uint32_t mvtab[33] = {
94            1, 2, 3, 4, 6, 7, 7, 7,
95            9, 9, 9, 10, 10, 10, 10, 10,
96            10, 10, 10, 10, 10, 10, 10, 10,
97            10, 11, 11, 11, 11, 11, 11, 12, 12
98    };
99    
100    static const int DQtab[4] = {
101            -1, -2, 1, 2
102    };
103    
104    #define RRV_MV_SCALEDOWN(a)     ( (a)>=0 ? (a+1)/2 : (a-1)/2 )
105    
106    static const VECTOR zeroMV = {0,0};
107    
108    typedef struct
109    {
110    // general fields
111            int max_dx, min_dx, max_dy, min_dy;
112            uint32_t rounding;
113            VECTOR predMV;
114            VECTOR * currentMV;
115            VECTOR * currentQMV;
116            int32_t * iMinSAD;
117            const uint8_t * Ref;
118            const uint8_t * RefH;
119            const uint8_t * RefV;
120            const uint8_t * RefHV;
121            const uint8_t * RefCU;
122            const uint8_t * RefCV;
123            const uint8_t * CurU;
124            const uint8_t * CurV;
125            uint8_t * RefQ;
126            const uint8_t * Cur;
127            uint32_t lambda16;
128            uint32_t lambda8;
129            uint32_t iEdgedWidth;
130            uint32_t iFcode;
131            int * temp;
132            int qpel, qpel_precision;
133            int chroma;
134            int rrv;
135    //fields for interpolate and direct modes
136            const uint8_t * bRef;
137            const uint8_t * bRefH;
138            const uint8_t * bRefV;
139            const uint8_t * bRefHV;
140            const uint8_t * b_RefCU;
141            const uint8_t * b_RefCV;
142    
143            VECTOR bpredMV;
144            uint32_t bFcode;
145    // fields for direct mode
146            VECTOR directmvF[4];
147            VECTOR directmvB[4];
148            const VECTOR * referencemv;
149    
150    } SearchData;
151    
152    
153    typedef void(CheckFunc)(const int x, const int y,
154                                                    const int Direction, int * const dir,
155                                                    const SearchData * const Data);
156    CheckFunc *CheckCandidate;
157    
158    /*
159     * Calculate the min/max range
160     * relative to the _MACROBLOCK_ position
161     */
162    static void __inline
163    get_range(int32_t * const min_dx,
164                      int32_t * const max_dx,
165                      int32_t * const min_dy,
166                      int32_t * const max_dy,
167                      const uint32_t x,
168                      const uint32_t y,
169                      uint32_t block_sz, /* block dimension, 8 or 16 */
170                      const uint32_t width,
171                      const uint32_t height,
172                      const uint32_t fcode,
173                      const int qpel, /* 1 if the resulting range should be in qpel precision; otherwise 0 */
174                      const int rrv)
175    {
176            int k, m = qpel ? 4 : 2;
177            const int search_range = 32 << (fcode - 1);
178            int high = search_range - 1;
179            int low = -search_range;
180    
181            if (rrv) {
182                    high = RRV_MV_SCALEUP(high);
183                    low = RRV_MV_SCALEUP(low);
184                    block_sz *= 2;
185            }
186    
187            k = m * (int)(width - x * block_sz);
188            *max_dx = MIN(high, k);
189            k = m * (int)(height -  y * block_sz);
190            *max_dy = MIN(high, k);
191    
192            k = -m * (int)((x+1) * block_sz);
193            *min_dx = MAX(low, k);
194            k = -m * (int)((y+1) * block_sz);
195            *min_dy = MAX(low, k);
196    }
197    
198    typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection);
199    
200    static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch;
201    
202    static void Search8(const SearchData * const OldData,
203                            const int x, const int y,
204                            const uint32_t MotionFlags,
205                            const MBParam * const pParam,
206                            MACROBLOCK * const pMB,
207                            const MACROBLOCK * const pMBs,
208                            const int block,
209                            SearchData * const Data);
210    
211    bool
212    MotionEstimation(MBParam * const pParam,
213                                     FRAMEINFO * const current,
214                                     FRAMEINFO * const reference,
215                                     const IMAGE * const pRefH,
216                                     const IMAGE * const pRefV,
217                                     const IMAGE * const pRefHV,
218                                     const uint32_t iLimit);
219    
220    static void
221    SearchP(const IMAGE * const pRef,
222                    const uint8_t * const pRefH,
223                    const uint8_t * const pRefV,
224                    const uint8_t * const pRefHV,
225                    const IMAGE * const pCur,
226                    const int x,
227                    const int y,
228                    const uint32_t MotionFlags,
229                    const uint32_t GlobalFlags,
230                    const uint32_t iQuant,
231                    SearchData * const Data,
232                    const MBParam * const pParam,
233                    const MACROBLOCK * const pMBs,
234                    const MACROBLOCK * const prevMBs,
235                    int inter4v,
236                    MACROBLOCK * const pMB);
237    
238    
239    static WARPPOINTS
240    GlobalMotionEst(const MACROBLOCK * const pMBs,
241                                    const MBParam * const pParam,
242                                    const FRAMEINFO * const current,
243                                    const FRAMEINFO * const reference,
244                                    const IMAGE * const pRefH,
245                                    const IMAGE * const pRefV,
246                                    const IMAGE * const pRefHV      );
247    
248    #define iDiamondSize 2
249    
250    static __inline uint32_t
251    MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t GlobalFlags)
252    {
253            uint32_t Flags = MotionFlags;
254    
255            if (!(GlobalFlags & XVID_MODEDECISION_BITS))
256                    Flags &= ~(QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS+EXTSEARCH_BITS);
257    
258            if (Flags & EXTSEARCH_BITS)
259                    Flags |= HALFPELREFINE16_BITS;
260    
261            if (Flags & EXTSEARCH_BITS && MotionFlags & PMV_EXTSEARCH8)
262                    Flags |= HALFPELREFINE8_BITS;
263    
264            if (Flags & HALFPELREFINE16_BITS)
265                    Flags |= QUARTERPELREFINE16_BITS;
266    
267            if (Flags & HALFPELREFINE8_BITS) {
268                    Flags |= QUARTERPELREFINE8_BITS;
269                    Flags &= ~PMV_HALFPELREFINE8;
270            }
271    
272            if (Flags & QUARTERPELREFINE8_BITS)
273                    Flags &= ~PMV_QUARTERPELREFINE8;
274    
275            if (!(GlobalFlags & XVID_QUARTERPEL))
276                    Flags &= ~(PMV_QUARTERPELREFINE16+PMV_QUARTERPELREFINE8+QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS);
277    
278            if (!(GlobalFlags & XVID_HALFPEL))
279                    Flags &= ~(PMV_EXTSEARCH16+PMV_HALFPELREFINE16+PMV_HALFPELREFINE8+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS);
280    
281            if (GlobalFlags & (XVID_GREYSCALE + XVID_REDUCED))
282                    Flags &= ~(PMV_CHROMA16 + PMV_CHROMA8);
283    
284            return Flags;
285    }
286    
287    /* BITS mode decision and search */
288    
289    #include "../bitstream/zigzag.h"
290    #include "../quant/quant_mpeg4.h"
291    #include "../quant/quant_h263.h"
292    #include "../bitstream/vlc_codes.h"
293    
294    static int
295    CountMBBitsInter(SearchData * const Data,
296                                    const MACROBLOCK * const pMBs, const int x, const int y,
297                                    const MBParam * const pParam,
298                                    const uint32_t MotionFlags);
299    
300    static int
301    CountMBBitsInter4v(const SearchData * const Data,
302                                            MACROBLOCK * const pMB, const MACROBLOCK * const pMBs,
303                                            const int x, const int y,
304                                            const MBParam * const pParam, const uint32_t MotionFlags,
305                                            const VECTOR * const backup);
306    
307    static int
308    CountMBBitsIntra(const SearchData * const Data);
309    
310    int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
311    int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
312    
313    #endif                                                  /* _MOTION_EST_H_ */
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.2
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4