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

Diff of /branches/dev-api-4/xvidcore/src/motion/motion_comp.c

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

revision 3, Fri Mar 8 02:46:11 2002 UTC revision 437, Sat Sep 7 09:12:22 2002 UTC
# Line 1  Line 1 
1    /*****************************************************************************
2     *
3     *  XVID MPEG-4 VIDEO CODEC
4     *  - Motion Compensation module -
5     *
6     *  Copyright(C) 2002 Michael Militzer <michael@xvid.org>
7     *  Copyright(C) 2002 Edouard Gomez <ed.gomez@wanadoo.fr>
8     *  Copyright(C) 2002 Christoph Lampert <gruel@web.de>
9     *
10     *  This program is an implementation of a part of one or more MPEG-4
11     *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
12     *  to use this software module in hardware or software products are
13     *  advised that its use may infringe existing patents or copyrights, and
14     *  any such use would be at such party's own risk.  The original
15     *  developer of this software module and his/her company, and subsequent
16     *  editors and their companies, will have no liability for use of this
17     *  software or modifications or derivatives thereof.
18     *
19     *  This program is free software; you can redistribute it and/or modify
20     *  it under the terms of the GNU General Public License as published by
21     *  the Free Software Foundation; either version 2 of the License, or
22     *  (at your option) any later version.
23     *
24     *  This program is distributed in the hope that it will be useful,
25     *  but WITHOUT ANY WARRANTY; without even the implied warranty of
26     *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
27     *  GNU General Public License for more details.
28     *
29     *  You should have received a copy of the GNU General Public License
30     *  along with this program; if not, write to the Free Software
31     *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
32     *
33     *************************************************************************/
34    
35  #include "../encoder.h"  #include "../encoder.h"
36  #include "../utils/mbfunctions.h"  #include "../utils/mbfunctions.h"
37  #include "../image/interpolate8x8.h"  #include "../image/interpolate8x8.h"
38  #include "../utils/timer.h"  #include "../utils/timer.h"
39    #include "motion.h"
40    
41  #define ABS(X) (((X)>0)?(X):-(X))  #define ABS(X) (((X)>0)?(X):-(X))
42  #define SIGN(X) (((X)>0)?1:-1)  #define SIGN(X) (((X)>0)?1:-1)
43    
44  static __inline void compensate8x8_halfpel(  static __inline void
45                                  int16_t * const dct_codes,  compensate8x8_halfpel(int16_t * const dct_codes,
46                                  uint8_t * const cur,                                  uint8_t * const cur,
47                                  const uint8_t * const ref,                                  const uint8_t * const ref,
48                                  const uint8_t * const refh,                                  const uint8_t * const refh,
49                                  const uint8_t * const refv,                                  const uint8_t * const refv,
50                                  const uint8_t * const refhv,                                  const uint8_t * const refhv,
51                                  const uint32_t x, const uint32_t y,                                            const uint32_t x,
52                                  const int32_t dx,  const int dy,                                            const uint32_t y,
53                                              const int32_t dx,
54                                              const int dy,
55                                  const uint32_t stride)                                  const uint32_t stride)
56  {  {
57          int32_t ddx,ddy;          int32_t ddx,ddy;
# Line 25  Line 62 
62                  ddx = dx/2;                  ddx = dx/2;
63                  ddy = dy/2;                  ddy = dy/2;
64                  transfer_8to16sub(dct_codes, cur + y*stride + x,                  transfer_8to16sub(dct_codes, cur + y*stride + x,
65                                  ref + (y+ddy)*stride + x+ddx, stride);                                                    ref + (int) ((y + ddy) * stride + x + ddx), stride);
66                  break;                  break;
67    
68      case 1 :      case 1 :
69                  ddx = dx/2;                  ddx = dx/2;
70                  ddy = (dy-1)/2;                  ddy = (dy-1)/2;
71                  transfer_8to16sub(dct_codes, cur + y*stride + x,                  transfer_8to16sub(dct_codes, cur + y*stride + x,
72                                  refv + (y+ddy)*stride + x+ddx, stride);                                                    refv + (int) ((y + ddy) * stride + x + ddx), stride);
73                  break;                  break;
74    
75      case 2 :      case 2 :
76                  ddx = (dx-1)/2;                  ddx = (dx-1)/2;
77                  ddy = dy/2;                  ddy = dy/2;
78                  transfer_8to16sub(dct_codes, cur + y*stride + x,                  transfer_8to16sub(dct_codes, cur + y*stride + x,
79                                  refh + (y+ddy)*stride + x+ddx, stride);                                                    refh + (int) ((y + ddy) * stride + x + ddx), stride);
80                  break;                  break;
81    
82          default :       // case 3:          default :       // case 3:
83                  ddx = (dx-1)/2;                  ddx = (dx-1)/2;
84                  ddy = (dy-1)/2;                  ddy = (dy-1)/2;
85                  transfer_8to16sub(dct_codes, cur + y*stride + x,                  transfer_8to16sub(dct_codes, cur + y*stride + x,
86                                  refhv + (y+ddy)*stride + x+ddx, stride);                                                    refhv + (int) ((y + ddy) * stride + x + ddx), stride);
87                  break;                  break;
88      }      }
89  }  }
90    
91    
92    
93  void MBMotionCompensation(  void
94                          MACROBLOCK * const mb,  MBMotionCompensation(MACROBLOCK * const mb,
95                      const uint32_t i,                      const uint32_t i,
96                          const uint32_t j,                          const uint32_t j,
97                      const IMAGE * const ref,                      const IMAGE * const ref,
# Line 62  Line 99 
99                      const IMAGE * const refv,                      const IMAGE * const refv,
100                          const IMAGE * const refhv,                          const IMAGE * const refhv,
101                      IMAGE * const cur,                      IMAGE * const cur,
102                      int16_t dct_codes[][64],                                           int16_t * dct_codes,
103                          const uint32_t width,                          const uint32_t width,
104                          const uint32_t height,                          const uint32_t height,
105                          const uint32_t edged_width,                          const uint32_t edged_width,
# Line 72  Line 109 
109                  { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };                  { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };
110    
111    
112          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
         {  
113                  int32_t dx = mb->mvs[0].x;                  int32_t dx = mb->mvs[0].x;
114                  int32_t dy = mb->mvs[0].y;                  int32_t dy = mb->mvs[0].y;
115    
116                  compensate8x8_halfpel(dct_codes[0], cur->y, ref->y, refh->y, refv->y, refhv->y,                  compensate8x8_halfpel(&dct_codes[0 * 64], cur->y, ref->y, refh->y,
117                                                                  16*i,     16*j,     dx, dy, edged_width);                                                            refv->y, refhv->y, 16 * i, 16 * j, dx, dy,
118                  compensate8x8_halfpel(dct_codes[1], cur->y, ref->y, refh->y, refv->y, refhv->y,                                                            edged_width);
119                                                                  16*i + 8, 16*j,     dx, dy, edged_width);                  compensate8x8_halfpel(&dct_codes[1 * 64], cur->y, ref->y, refh->y,
120                  compensate8x8_halfpel(dct_codes[2], cur->y, ref->y, refh->y, refv->y, refhv->y,                                                            refv->y, refhv->y, 16 * i + 8, 16 * j, dx, dy,
121                                                                  16*i,     16*j + 8, dx, dy, edged_width);                                                            edged_width);
122                  compensate8x8_halfpel(dct_codes[3], cur->y, ref->y, refh->y, refv->y, refhv->y,                  compensate8x8_halfpel(&dct_codes[2 * 64], cur->y, ref->y, refh->y,
123                                                                  16*i + 8, 16*j + 8, dx, dy, edged_width);                                                            refv->y, refhv->y, 16 * i, 16 * j + 8, dx, dy,
124                                                              edged_width);
125                    compensate8x8_halfpel(&dct_codes[3 * 64], cur->y, ref->y, refh->y,
126                                                              refv->y, refhv->y, 16 * i + 8, 16 * j + 8, dx,
127                                                              dy, edged_width);
128    
129                  dx = (dx & 3) ? (dx >> 1) | 1 : dx / 2;                  dx = (dx & 3) ? (dx >> 1) | 1 : dx / 2;
130                  dy = (dy & 3) ? (dy >> 1) | 1 : dy / 2;                  dy = (dy & 3) ? (dy >> 1) | 1 : dy / 2;
131    
132                  /* uv-image-based compensation                  /* uv-image-based compensation */
                 compensate8x8_halfpel(dct_codes[4], cur->u, ref->u, refh->u, refv->u, refhv->u,  
                                                                 8*i, 8*j, dx, dy, edged_width/2);  
                 compensate8x8_halfpel(dct_codes[5], cur->v, ref->v, refh->v, refv->v, refhv->v,  
                                                                 8*i, 8*j, dx, dy, edged_width/2);               */  
133    
134                  /* uv-block-based compensation */                  interpolate8x8_switch(refv->u, ref->u, 8 * i, 8 * j, dx, dy,
135                  interpolate8x8_switch(refv->u, ref->u, 8*i, 8*j, dx, dy, edged_width/2, rounding);                                                            edged_width / 2, rounding);
136                  transfer_8to16sub(dct_codes[4],                  transfer_8to16sub(&dct_codes[4 * 64],
137                                  cur->u + 8*j*edged_width/2 + 8*i,                                  cur->u + 8*j*edged_width/2 + 8*i,
138                                  refv->u + 8*j*edged_width/2 + 8*i, edged_width/2);                                                    refv->u + 8 * j * edged_width / 2 + 8 * i,
139                                                      edged_width / 2);
140    
141                  interpolate8x8_switch(refv->v, ref->v, 8*i, 8*j, dx, dy, edged_width/2, rounding);                  interpolate8x8_switch(refv->v, ref->v, 8 * i, 8 * j, dx, dy,
142                  transfer_8to16sub(dct_codes[5],                                                            edged_width / 2, rounding);
143                    transfer_8to16sub(&dct_codes[5 * 64],
144                                  cur->v + 8*j*edged_width/2 + 8*i,                                  cur->v + 8*j*edged_width/2 + 8*i,
145                                  refv->v + 8*j*edged_width/2 + 8*i, edged_width/2);                                                    refv->v + 8 * j * edged_width / 2 + 8 * i,
146                                                      edged_width / 2);
147    
148          }          } else                                          // mode == MODE_INTER4V
         else    // mode == MODE_INTER4V  
149          {          {
150                  int32_t sum, dx, dy;                  int32_t sum, dx, dy;
151    
152                  compensate8x8_halfpel(dct_codes[0], cur->y, ref->y, refh->y, refv->y, refhv->y,                  compensate8x8_halfpel(&dct_codes[0 * 64], cur->y, ref->y, refh->y,
153                                                                  16*i,     16*j,     mb->mvs[0].x, mb->mvs[0].y, edged_width);                                                            refv->y, refhv->y, 16 * i, 16 * j, mb->mvs[0].x,
154                  compensate8x8_halfpel(dct_codes[1], cur->y, ref->y, refh->y, refv->y, refhv->y,                                                            mb->mvs[0].y, edged_width);
155                                                                  16*i + 8, 16*j,     mb->mvs[1].x, mb->mvs[1].y, edged_width);                  compensate8x8_halfpel(&dct_codes[1 * 64], cur->y, ref->y, refh->y,
156                  compensate8x8_halfpel(dct_codes[2], cur->y, ref->y, refh->y, refv->y, refhv->y,                                                            refv->y, refhv->y, 16 * i + 8, 16 * j,
157                                                                  16*i,     16*j + 8, mb->mvs[2].x, mb->mvs[2].y, edged_width);                                                            mb->mvs[1].x, mb->mvs[1].y, edged_width);
158                  compensate8x8_halfpel(dct_codes[3], cur->y, ref->y, refh->y, refv->y, refhv->y,                  compensate8x8_halfpel(&dct_codes[2 * 64], cur->y, ref->y, refh->y,
159                                                                  16*i + 8, 16*j + 8, mb->mvs[3].x, mb->mvs[3].y, edged_width);                                                            refv->y, refhv->y, 16 * i, 16 * j + 8,
160                                                              mb->mvs[2].x, mb->mvs[2].y, edged_width);
161                    compensate8x8_halfpel(&dct_codes[3 * 64], cur->y, ref->y, refh->y,
162                                                              refv->y, refhv->y, 16 * i + 8, 16 * j + 8,
163                                                              mb->mvs[3].x, mb->mvs[3].y, edged_width);
164    
165                  sum = mb->mvs[0].x + mb->mvs[1].x + mb->mvs[2].x + mb->mvs[3].x;                  sum = mb->mvs[0].x + mb->mvs[1].x + mb->mvs[2].x + mb->mvs[3].x;
166                  dx = (sum ? SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0);                  dx = (sum ? SIGN(sum) *
167                              (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0);
168    
169                  sum = mb->mvs[0].y + mb->mvs[1].y + mb->mvs[2].y + mb->mvs[3].y;                  sum = mb->mvs[0].y + mb->mvs[1].y + mb->mvs[2].y + mb->mvs[3].y;
170                  dy = (sum ? SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0);                  dy = (sum ? SIGN(sum) *
171                              (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0);
                 /* uv-image-based compensation  
                 compensate8x8_halfpel(dct_codes[4], cur->u, ref->u, refh->u, refv->u, refhv->u,  
                                                                 8*i, 8*j, dx, dy, edged_width/2);  
                 compensate8x8_halfpel(dct_codes[5], cur->v, ref->v, refh->v, refv->v, refhv->v,  
                                                                 8*i, 8*j, dx, dy, edged_width/2);               */  
172    
173                  /* uv-block-based compensation */                  /* uv-block-based compensation */
174                  interpolate8x8_switch(refv->u, ref->u, 8*i, 8*j, dx, dy, edged_width/2, rounding);                  interpolate8x8_switch(refv->u, ref->u, 8 * i, 8 * j, dx, dy,
175                  transfer_8to16sub(dct_codes[4],                                                            edged_width / 2, rounding);
176                    transfer_8to16sub(&dct_codes[4 * 64],
177                                  cur->u + 8*j*edged_width/2 + 8*i,                                  cur->u + 8*j*edged_width/2 + 8*i,
178                                  refv->u + 8*j*edged_width/2 + 8*i, edged_width/2);                                                    refv->u + 8 * j * edged_width / 2 + 8 * i,
179                                                      edged_width / 2);
180    
181                  interpolate8x8_switch(refv->v, ref->v, 8*i, 8*j, dx, dy, edged_width/2, rounding);                  interpolate8x8_switch(refv->v, ref->v, 8 * i, 8 * j, dx, dy,
182                  transfer_8to16sub(dct_codes[5],                                                            edged_width / 2, rounding);
183                    transfer_8to16sub(&dct_codes[5 * 64],
184                                  cur->v + 8*j*edged_width/2 + 8*i,                                  cur->v + 8*j*edged_width/2 + 8*i,
185                                  refv->v + 8*j*edged_width/2 + 8*i, edged_width/2);                                                    refv->v + 8 * j * edged_width / 2 + 8 * i,
186                                                      edged_width / 2);
187          }          }
188  }  }
Legend:
Removed from v.3  
changed lines
  Added in v.437
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4