[svn] / branches / dev-api-4 / xvidcore / src / motion / motion_est.h Repository:
ViewVC logotype

Diff of /branches/dev-api-4/xvidcore/src/motion/motion_est.h

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

revision 886, Fri Feb 21 14:49:29 2003 UTC revision 1071, Thu Jun 19 09:59:37 2003 UTC
# Line 1  Line 1 
1  /**************************************************************************  /*****************************************************************************
2   *   *
3   *  XVID MPEG-4 VIDEO CODEC   *  XVID MPEG-4 VIDEO CODEC
4   *  -  Motion estimation header  -   *  - Motion Estimation related header -
5   *   *
6   *  This program is an implementation of a part of one or more MPEG-4   *  Copyright(C) 2002 Christoph Lampert <gruel@web.de>
7   *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending   *               2002 Michael Militzer <michael@xvid.org>
8   *  to use this software module in hardware or software products are   *               2002-2003 Radoslaw Czyz <xvid@syskin.cjb.net>
  *  advised that its use may infringe existing patents or copyrights, and  
  *  any such use would be at such party's own risk.  The original  
  *  developer of this software module and his/her company, and subsequent  
  *  editors and their companies, will have no liability for use of this  
  *  software or modifications or derivatives thereof.  
9   *   *
10   *  This program is free software; you can redistribute it and/or modify   *  This program is free software; you can redistribute it and/or modify
11   *  it under the terms of the GNU General Public License as published by   *  it under the terms of the GNU General Public License as published by
# Line 26  Line 21 
21   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
22   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
23   *   *
24   *  $Id: motion_est.h,v 1.3 2003-02-21 14:40:11 syskin Exp $   * $Id: motion_est.h,v 1.3.2.10 2003-06-19 09:59:37 syskin Exp $
25   *   *
26   ***************************************************************************/   ****************************************************************************/
27    
28  #ifndef _MOTION_EST_H_  #ifndef _MOTION_EST_H_
29  #define _MOTION_EST_H_  #define _MOTION_EST_H_
# Line 39  Line 34 
34    
35  /* hard coded motion search parameters for motion_est and smp_motion_est */  /* hard coded motion search parameters for motion_est and smp_motion_est */
36    
37  // very large value  /* very large value */
38  #define MV_MAX_ERROR    (4096 * 256)  #define MV_MAX_ERROR    (4096 * 256)
39    
40  /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */  /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */
# Line 50  Line 45 
45  #define NEIGH_TEND_8X8          40.0  #define NEIGH_TEND_8X8          40.0
46  #define NEIGH_8X8_BIAS          30  #define NEIGH_8X8_BIAS          30
47    
48    #define BITS_MULT                       16
49    
50  /* Parameters which control inter/inter4v decision */  /* Parameters which control inter/inter4v decision */
51  #define IMV16X16                        2  #define IMV16X16                        2
52    
# Line 89  Line 86 
86          (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5),          (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5),
87          (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5)        };          (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5)        };
88    
89  // mv.length table  /* mv.length table */
90  static const uint32_t mvtab[33] = {  static const int mvtab[64] = {
91          1, 2, 3, 4, 6, 7, 7, 7,          1, 2, 3, 4, 6, 7, 7, 7,
92          9, 9, 9, 10, 10, 10, 10, 10,          9, 9, 9, 10, 10, 10, 10, 10,
93          10, 10, 10, 10, 10, 10, 10, 10,          10, 10, 10, 10, 10, 10, 10, 10,
94          10, 11, 11, 11, 11, 11, 11, 12, 12                  10, 11, 11, 11, 11, 11, 11, 12,
95  };                  12, 12, 12, 12, 12, 12, 12, 12,
96                    12, 12, 12, 12, 12, 12, 12, 12,
97                    12, 12, 12, 12, 12, 12, 12, 12, 12 };
98    
99  static const int DQtab[4] = {  static const int DQtab[4] = {
100          -1, -2, 1, 2          -1, -2, 1, 2
# Line 105  Line 104 
104    
105  typedef struct  typedef struct
106  {  {
107  // general fields          /* general fields */
108          int max_dx, min_dx, max_dy, min_dy;          int max_dx, min_dx, max_dy, min_dy;
109          uint32_t rounding;          uint32_t rounding;
110          VECTOR predMV;          VECTOR predMV;
111          VECTOR * currentMV;          VECTOR * currentMV;
112          VECTOR * currentQMV;          VECTOR * currentQMV;
113          int32_t * iMinSAD;          int32_t * iMinSAD;
114          const uint8_t * Ref;          const uint8_t * RefP[6]; /* N, V, H, HV, cU, cV */
         const uint8_t * RefH;  
         const uint8_t * RefV;  
         const uint8_t * RefHV;  
         const uint8_t * RefCU;  
         const uint8_t * RefCV;  
115          const uint8_t * CurU;          const uint8_t * CurU;
116          const uint8_t * CurV;          const uint8_t * CurV;
117          uint8_t * RefQ;          uint8_t * RefQ;
# Line 130  Line 124 
124          int qpel, qpel_precision;          int qpel, qpel_precision;
125          int chroma;          int chroma;
126          int rrv;          int rrv;
 //fields for interpolate and direct modes  
         const uint8_t * bRef;  
         const uint8_t * bRefH;  
         const uint8_t * bRefV;  
         const uint8_t * bRefHV;  
         const uint8_t * b_RefCU;  
         const uint8_t * b_RefCV;  
127    
128            /* fields for interpolate and direct modes */
129            const uint8_t * b_RefP[6]; /* N, V, H, HV, cU, cV */
130          VECTOR bpredMV;          VECTOR bpredMV;
131          uint32_t bFcode;          uint32_t bFcode;
132  // fields for direct mode  
133            /* fields for direct mode */
134          VECTOR directmvF[4];          VECTOR directmvF[4];
135          VECTOR directmvB[4];          VECTOR directmvB[4];
136          const VECTOR * referencemv;          const VECTOR * referencemv;
137    
138            /* BITS/R-D stuff */
139            int16_t * dctSpace;
140            uint32_t iQuant;
141            uint32_t quant_type;
142    
143  } SearchData;  } SearchData;
144    
145    
# Line 224  Line 219 
219                  const int x,                  const int x,
220                  const int y,                  const int y,
221                  const uint32_t MotionFlags,                  const uint32_t MotionFlags,
222                  const uint32_t GlobalFlags,                  const uint32_t VopFlags,
223                  const uint32_t iQuant,                  const uint32_t VolFlags,
224                  SearchData * const Data,                  SearchData * const Data,
225                  const MBParam * const pParam,                  const MBParam * const pParam,
226                  const MACROBLOCK * const pMBs,                  const MACROBLOCK * const pMBs,
227                  const MACROBLOCK * const prevMBs,                  const MACROBLOCK * const prevMBs,
                 int inter4v,  
228                  MACROBLOCK * const pMB);                  MACROBLOCK * const pMB);
229    
   
230  static WARPPOINTS  static WARPPOINTS
231  GlobalMotionEst(const MACROBLOCK * const pMBs,  GlobalMotionEst(const MACROBLOCK * const pMBs,
232                                  const MBParam * const pParam,                                  const MBParam * const pParam,
# Line 246  Line 239 
239  #define iDiamondSize 2  #define iDiamondSize 2
240    
241  static __inline uint32_t  static __inline uint32_t
242  MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t GlobalFlags)  MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t VopFlags, const uint32_t VolFlags)
243  {  {
244          uint32_t Flags = MotionFlags;          uint32_t Flags = MotionFlags;
245    
246          if (!(GlobalFlags & XVID_MODEDECISION_BITS))          if (!(VopFlags & XVID_VOP_MODEDECISION_BITS))
247                  Flags &= ~(QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS+EXTSEARCH_BITS);                  Flags &= ~(XVID_ME_QUARTERPELREFINE16_BITS+XVID_ME_QUARTERPELREFINE8_BITS+XVID_ME_HALFPELREFINE16_BITS+XVID_ME_HALFPELREFINE8_BITS+XVID_ME_EXTSEARCH_BITS);
248    
249          if (Flags & EXTSEARCH_BITS)          if (Flags & XVID_ME_EXTSEARCH_BITS)
250                  Flags |= HALFPELREFINE16_BITS;                  Flags |= XVID_ME_HALFPELREFINE16_BITS;
251    
252          if (Flags & EXTSEARCH_BITS && MotionFlags & PMV_EXTSEARCH8)          if (Flags & XVID_ME_EXTSEARCH_BITS && MotionFlags & XVID_ME_EXTSEARCH8)
253                  Flags |= HALFPELREFINE8_BITS;                  Flags |= XVID_ME_HALFPELREFINE8_BITS;
254    
255          if (Flags & HALFPELREFINE16_BITS)          if (Flags & XVID_ME_HALFPELREFINE16_BITS)
256                  Flags |= QUARTERPELREFINE16_BITS;                  Flags |= XVID_ME_QUARTERPELREFINE16_BITS;
257    
258          if (Flags & HALFPELREFINE8_BITS) {          if (Flags & XVID_ME_HALFPELREFINE8_BITS) {
259                  Flags |= QUARTERPELREFINE8_BITS;                  Flags |= XVID_ME_QUARTERPELREFINE8_BITS;
260                  Flags &= ~PMV_HALFPELREFINE8;                  Flags &= ~XVID_ME_HALFPELREFINE8;
261          }          }
262    
263          if (Flags & QUARTERPELREFINE8_BITS)          if (Flags & XVID_ME_QUARTERPELREFINE8_BITS)
264                  Flags &= ~PMV_QUARTERPELREFINE8;                  Flags &= ~XVID_ME_QUARTERPELREFINE8;
265    
266          if (!(GlobalFlags & XVID_QUARTERPEL))          if (!(VolFlags & XVID_VOL_QUARTERPEL))
267                  Flags &= ~(PMV_QUARTERPELREFINE16+PMV_QUARTERPELREFINE8+QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS);                  Flags &= ~(XVID_ME_QUARTERPELREFINE16+XVID_ME_QUARTERPELREFINE8+XVID_ME_QUARTERPELREFINE16_BITS+XVID_ME_QUARTERPELREFINE8_BITS);
268    
269          if (!(GlobalFlags & XVID_HALFPEL))          if (!(VopFlags & XVID_VOP_HALFPEL))
270                  Flags &= ~(PMV_EXTSEARCH16+PMV_HALFPELREFINE16+PMV_HALFPELREFINE8+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS);                  Flags &= ~(XVID_ME_EXTSEARCH16+XVID_ME_HALFPELREFINE16+XVID_ME_HALFPELREFINE8+XVID_ME_HALFPELREFINE16_BITS+XVID_ME_HALFPELREFINE8_BITS);
271    
272          if (GlobalFlags & (XVID_GREYSCALE + XVID_REDUCED))          if ((VopFlags & XVID_VOP_GREYSCALE) || (VopFlags & XVID_VOP_REDUCED))
273                  Flags &= ~(PMV_CHROMA16 + PMV_CHROMA8);                  Flags &= ~(XVID_ME_CHROMA16 + XVID_ME_CHROMA8);
274    
275          return Flags;          return Flags;
276  }  }
# Line 288  Line 281 
281  #include "../quant/quant_mpeg4.h"  #include "../quant/quant_mpeg4.h"
282  #include "../quant/quant_h263.h"  #include "../quant/quant_h263.h"
283  #include "../bitstream/vlc_codes.h"  #include "../bitstream/vlc_codes.h"
284    #include "../dct/fdct.h"
285    
286  static int  static int
287  CountMBBitsInter(SearchData * const Data,  CountMBBitsInter(SearchData * const Data,
# Line 308  Line 302 
302  int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);  int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
303  int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);  int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
304    
305    #define LAMBDA          ( (int)(1.0*BITS_MULT) )
306    
307    static __inline unsigned int
308    Block_CalcBits( int16_t * const coeff,
309                                    int16_t * const data,
310                                    const uint32_t quant, const int quant_type,
311                                    uint32_t * cbp,
312                                    const int block)
313    {
314            int sum;
315            int bits;
316            const int lambda = LAMBDA*quant*quant;
317            int distortion = 0;
318            int i;
319    
320            fdct(data);
321    
322            if (quant_type == 0) sum = quant_inter(coeff, data, quant);
323            else sum = quant4_inter(coeff, data, quant);
324    
325            if (sum > 0) {
326                    *cbp |= 1 << (5 - block);
327                    bits = BITS_MULT * CodeCoeffInter_CalcBits(coeff, scan_tables[0]);
328            } else bits = 0;
329    
330            if (quant_type == 0) dequant_inter(coeff, coeff, quant);
331            else dequant4_inter(coeff, coeff, quant);
332    
333            for (i = 0; i < 64; i++) {
334                    distortion += (data[i] - coeff[i])*(data[i] - coeff[i]);
335            }
336            bits += (BITS_MULT*BITS_MULT*distortion)/lambda;
337    
338    
339            return bits;
340    }
341    
342    static __inline unsigned int
343    Block_CalcBitsIntra(int16_t * const coeff,
344                                            int16_t * const data,
345                                            const uint32_t quant, const int quant_type,
346                                            uint32_t * cbp,
347                                            const int block,
348                                            int * dcpred)
349    {
350            int bits, i;
351            const int lambda = LAMBDA*quant*quant;
352            int distortion = 0;
353            uint32_t iDcScaler = get_dc_scaler(quant, block < 4);
354            int b_dc;
355    
356            fdct(data);
357            data[0] -= 1024;
358    
359            if (quant_type == 0) quant_intra_c(coeff, data, quant, iDcScaler); // MUST BE _C
360            else quant4_intra_c(coeff, data, quant, iDcScaler); // MUST BE _C
361    
362            b_dc = coeff[0];
363            if (block < 4) {
364                    coeff[0] -= *dcpred;
365                    *dcpred = b_dc;
366            }
367    
368            bits = BITS_MULT*CodeCoeffIntra_CalcBits(coeff, scan_tables[0]);
369            if (bits != 0) *cbp |= 1 << (5 - block);
370    
371            if (block < 4) bits += BITS_MULT*dcy_tab[coeff[0] + 255].len;
372            else bits += BITS_MULT*dcc_tab[coeff[0] + 255].len;
373    
374            coeff[0] = b_dc;
375            if (quant_type == 0) dequant_intra(coeff, coeff, quant, iDcScaler);
376            else dequant4_intra(coeff, coeff, quant, iDcScaler);
377    
378            for (i = 0; i < 64; i++) {
379                    distortion += (data[i] - coeff[i])*(data[i] - coeff[i]);
380            }
381    
382            bits += (BITS_MULT*BITS_MULT*distortion)/lambda;
383    
384            return bits;
385    }
386    
387    
388  #endif                                                  /* _MOTION_EST_H_ */  #endif                                                  /* _MOTION_EST_H_ */
Legend:
Removed from v.886  
changed lines
  Added in v.1071
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4