Parent Directory | Revision Log
Revision 1.3.2.16 - (view) (download)
1 : | edgomez | 1.3.2.9 | /***************************************************************************** |
2 : | edgomez | 1.2 | * |
3 : | * XVID MPEG-4 VIDEO CODEC | ||
4 : | edgomez | 1.3.2.9 | * - Motion Estimation related header - |
5 : | edgomez | 1.2 | * |
6 : | edgomez | 1.3.2.9 | * Copyright(C) 2002 Christoph Lampert <gruel@web.de> |
7 : | * 2002 Michael Militzer <michael@xvid.org> | ||
8 : | * 2002-2003 Radoslaw Czyz <xvid@syskin.cjb.net> | ||
9 : | edgomez | 1.2 | * |
10 : | edgomez | 1.3.2.9 | * This program is free software ; you can redistribute it and/or modify |
11 : | edgomez | 1.2 | * it under the terms of the GNU General Public License as published by |
12 : | edgomez | 1.3.2.9 | * the Free Software Foundation ; either version 2 of the License, or |
13 : | edgomez | 1.2 | * (at your option) any later version. |
14 : | * | ||
15 : | * This program is distributed in the hope that it will be useful, | ||
16 : | edgomez | 1.3.2.9 | * but WITHOUT ANY WARRANTY ; without even the implied warranty of |
17 : | edgomez | 1.2 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
18 : | * GNU General Public License for more details. | ||
19 : | * | ||
20 : | * You should have received a copy of the GNU General Public License | ||
21 : | edgomez | 1.3.2.9 | * along with this program ; if not, write to the Free Software |
22 : | edgomez | 1.2 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 : | * | ||
24 : | syskin | 1.3.2.16 | * $Id: motion_est.h,v 1.3.2.15 2003/08/03 10:10:54 syskin Exp $ |
25 : | edgomez | 1.2 | * |
26 : | edgomez | 1.3.2.9 | ****************************************************************************/ |
27 : | edgomez | 1.2 | |
28 : | #ifndef _MOTION_EST_H_ | ||
29 : | #define _MOTION_EST_H_ | ||
30 : | |||
31 : | #include "../portab.h" | ||
32 : | #include "../global.h" | ||
33 : | #include "../image/reduced.h" | ||
34 : | |||
35 : | /* hard coded motion search parameters for motion_est and smp_motion_est */ | ||
36 : | |||
37 : | edgomez | 1.3.2.8 | /* very large value */ |
38 : | edgomez | 1.2 | #define MV_MAX_ERROR (4096 * 256) |
39 : | |||
40 : | /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ | ||
41 : | #define MV16_INTER_BIAS 512 | ||
42 : | |||
43 : | /* vector map (vlc delta size) smoother parameters ! float !*/ | ||
44 : | #define NEIGH_TEND_16X16 10.5 | ||
45 : | #define NEIGH_TEND_8X8 40.0 | ||
46 : | #define NEIGH_8X8_BIAS 30 | ||
47 : | |||
48 : | edgomez | 1.3.2.7 | #define BITS_MULT 16 |
49 : | |||
50 : | edgomez | 1.2 | /* Parameters which control inter/inter4v decision */ |
51 : | #define IMV16X16 2 | ||
52 : | |||
53 : | static const int lambda_vec16[32] = | ||
54 : | { 0 ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5), | ||
55 : | (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5), | ||
56 : | (int)(1.49591*NEIGH_TEND_16X16 + 0.5), (int)(1.68601*NEIGH_TEND_16X16 + 0.5), | ||
57 : | (int)(1.89187*NEIGH_TEND_16X16 + 0.5), (int)(2.11542*NEIGH_TEND_16X16 + 0.5), | ||
58 : | (int)(2.35878*NEIGH_TEND_16X16 + 0.5), (int)(2.62429*NEIGH_TEND_16X16 + 0.5), | ||
59 : | (int)(2.91455*NEIGH_TEND_16X16 + 0.5), (int)(3.23253*NEIGH_TEND_16X16 + 0.5), | ||
60 : | (int)(3.58158*NEIGH_TEND_16X16 + 0.5), (int)(3.96555*NEIGH_TEND_16X16 + 0.5), | ||
61 : | (int)(4.38887*NEIGH_TEND_16X16 + 0.5), (int)(4.85673*NEIGH_TEND_16X16 + 0.5), | ||
62 : | (int)(5.37519*NEIGH_TEND_16X16 + 0.5), (int)(5.95144*NEIGH_TEND_16X16 + 0.5), | ||
63 : | (int)(6.59408*NEIGH_TEND_16X16 + 0.5), (int)(7.31349*NEIGH_TEND_16X16 + 0.5), | ||
64 : | (int)(8.12242*NEIGH_TEND_16X16 + 0.5), (int)(9.03669*NEIGH_TEND_16X16 + 0.5), | ||
65 : | (int)(10.0763*NEIGH_TEND_16X16 + 0.5), (int)(11.2669*NEIGH_TEND_16X16 + 0.5), | ||
66 : | (int)(12.6426*NEIGH_TEND_16X16 + 0.5), (int)(14.2493*NEIGH_TEND_16X16 + 0.5), | ||
67 : | (int)(16.1512*NEIGH_TEND_16X16 + 0.5), (int)(18.442*NEIGH_TEND_16X16 + 0.5), | ||
68 : | (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5), | ||
69 : | (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5) }; | ||
70 : | |||
71 : | static const int lambda_vec8[32] = | ||
72 : | { 0 ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5), | ||
73 : | (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5), | ||
74 : | (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5), | ||
75 : | (int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5), | ||
76 : | (int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5), | ||
77 : | (int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5), | ||
78 : | (int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5), | ||
79 : | (int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5), | ||
80 : | (int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5), | ||
81 : | (int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5), | ||
82 : | (int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5), | ||
83 : | (int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5), | ||
84 : | (int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5), | ||
85 : | (int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5), | ||
86 : | (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) }; | ||
88 : | |||
89 : | edgomez | 1.3.2.8 | /* mv.length table */ |
90 : | edgomez | 1.3.2.4 | static const int mvtab[64] = { |
91 : | 1, 2, 3, 4, 6, 7, 7, 7, | ||
92 : | 9, 9, 9, 10, 10, 10, 10, 10, | ||
93 : | 10, 10, 10, 10, 10, 10, 10, 10, | ||
94 : | 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 : | edgomez | 1.2 | |
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 : | |||
105 : | typedef struct | ||
106 : | { | ||
107 : | edgomez | 1.3.2.8 | /* general fields */ |
108 : | edgomez | 1.2 | int max_dx, min_dx, max_dy, min_dy; |
109 : | uint32_t rounding; | ||
110 : | VECTOR predMV; | ||
111 : | VECTOR * currentMV; | ||
112 : | VECTOR * currentQMV; | ||
113 : | int32_t * iMinSAD; | ||
114 : | edgomez | 1.3.2.8 | const uint8_t * RefP[6]; /* N, V, H, HV, cU, cV */ |
115 : | edgomez | 1.2 | const uint8_t * CurU; |
116 : | const uint8_t * CurV; | ||
117 : | uint8_t * RefQ; | ||
118 : | const uint8_t * Cur; | ||
119 : | uint32_t lambda16; | ||
120 : | uint32_t lambda8; | ||
121 : | uint32_t iEdgedWidth; | ||
122 : | uint32_t iFcode; | ||
123 : | int * temp; | ||
124 : | int qpel, qpel_precision; | ||
125 : | int chroma; | ||
126 : | int rrv; | ||
127 : | edgomez | 1.3.2.8 | |
128 : | /* fields for interpolate and direct modes */ | ||
129 : | const uint8_t * b_RefP[6]; /* N, V, H, HV, cU, cV */ | ||
130 : | edgomez | 1.2 | VECTOR bpredMV; |
131 : | uint32_t bFcode; | ||
132 : | edgomez | 1.3.2.8 | |
133 : | /* fields for direct mode */ | ||
134 : | edgomez | 1.2 | VECTOR directmvF[4]; |
135 : | VECTOR directmvB[4]; | ||
136 : | const VECTOR * referencemv; | ||
137 : | edgomez | 1.3.2.8 | |
138 : | /* BITS/R-D stuff */ | ||
139 : | edgomez | 1.3.2.3 | int16_t * dctSpace; |
140 : | edgomez | 1.3.2.6 | uint32_t iQuant; |
141 : | uint32_t quant_type; | ||
142 : | syskin | 1.3.2.13 | int * cbp; |
143 : | edgomez | 1.2 | |
144 : | } SearchData; | ||
145 : | |||
146 : | |||
147 : | typedef void(CheckFunc)(const int x, const int y, | ||
148 : | const int Direction, int * const dir, | ||
149 : | const SearchData * const Data); | ||
150 : | CheckFunc *CheckCandidate; | ||
151 : | |||
152 : | /* | ||
153 : | * Calculate the min/max range | ||
154 : | * relative to the _MACROBLOCK_ position | ||
155 : | */ | ||
156 : | static void __inline | ||
157 : | get_range(int32_t * const min_dx, | ||
158 : | int32_t * const max_dx, | ||
159 : | int32_t * const min_dy, | ||
160 : | int32_t * const max_dy, | ||
161 : | const uint32_t x, | ||
162 : | const uint32_t y, | ||
163 : | syskin | 1.3.2.16 | uint32_t block_sz, /* block dimension, 3(8) or 4(16) */ |
164 : | edgomez | 1.2 | const uint32_t width, |
165 : | const uint32_t height, | ||
166 : | const uint32_t fcode, | ||
167 : | syskin | 1.3.2.16 | const int precision, /* 2 for qpel, 1 for halfpel */ |
168 : | edgomez | 1.2 | const int rrv) |
169 : | { | ||
170 : | syskin | 1.3.2.16 | int k; |
171 : | const int search_range = 16 << fcode; | ||
172 : | edgomez | 1.2 | int high = search_range - 1; |
173 : | int low = -search_range; | ||
174 : | |||
175 : | if (rrv) { | ||
176 : | high = RRV_MV_SCALEUP(high); | ||
177 : | low = RRV_MV_SCALEUP(low); | ||
178 : | syskin | 1.3.2.16 | block_sz++; |
179 : | edgomez | 1.2 | } |
180 : | |||
181 : | syskin | 1.3.2.16 | k = (int)(width - (x<<block_sz))<<precision; |
182 : | *max_dx = min(high, k); | ||
183 : | k = (int)(height - (y<<block_sz))<<precision; | ||
184 : | *max_dy = min(high, k); | ||
185 : | edgomez | 1.2 | |
186 : | syskin | 1.3.2.16 | k = (-(int)((x+1)<<block_sz))<<precision; |
187 : | *min_dx = max(low, k); | ||
188 : | k = (-(int)((y+1)<<block_sz))<<precision; | ||
189 : | *min_dy = max(low, k); | ||
190 : | edgomez | 1.2 | } |
191 : | |||
192 : | typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection); | ||
193 : | |||
194 : | static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; | ||
195 : | |||
196 : | static void Search8(const SearchData * const OldData, | ||
197 : | const int x, const int y, | ||
198 : | const uint32_t MotionFlags, | ||
199 : | const MBParam * const pParam, | ||
200 : | MACROBLOCK * const pMB, | ||
201 : | const MACROBLOCK * const pMBs, | ||
202 : | const int block, | ||
203 : | SearchData * const Data); | ||
204 : | |||
205 : | bool | ||
206 : | MotionEstimation(MBParam * const pParam, | ||
207 : | syskin | 1.3 | FRAMEINFO * const current, |
208 : | FRAMEINFO * const reference, | ||
209 : | const IMAGE * const pRefH, | ||
210 : | const IMAGE * const pRefV, | ||
211 : | const IMAGE * const pRefHV, | ||
212 : | chl | 1.3.2.12 | const IMAGE * const pGMC, |
213 : | syskin | 1.3 | const uint32_t iLimit); |
214 : | edgomez | 1.2 | |
215 : | static void | ||
216 : | SearchP(const IMAGE * const pRef, | ||
217 : | const uint8_t * const pRefH, | ||
218 : | const uint8_t * const pRefV, | ||
219 : | const uint8_t * const pRefHV, | ||
220 : | const IMAGE * const pCur, | ||
221 : | const int x, | ||
222 : | const int y, | ||
223 : | const uint32_t MotionFlags, | ||
224 : | edgomez | 1.3.2.6 | const uint32_t VopFlags, |
225 : | const uint32_t VolFlags, | ||
226 : | edgomez | 1.2 | SearchData * const Data, |
227 : | const MBParam * const pParam, | ||
228 : | const MACROBLOCK * const pMBs, | ||
229 : | const MACROBLOCK * const prevMBs, | ||
230 : | MACROBLOCK * const pMB); | ||
231 : | |||
232 : | chl | 1.3.2.12 | static __inline void |
233 : | GMEanalyzeMB (const uint8_t * const pCur, | ||
234 : | const uint8_t * const pRef, | ||
235 : | const uint8_t * const pRefH, | ||
236 : | const uint8_t * const pRefV, | ||
237 : | const uint8_t * const pRefHV, | ||
238 : | const int x, | ||
239 : | const int y, | ||
240 : | const MBParam * const pParam, | ||
241 : | MACROBLOCK * const pMBs, | ||
242 : | SearchData * const Data); | ||
243 : | |||
244 : | void | ||
245 : | GMEanalysis(const MBParam * const pParam, | ||
246 : | const FRAMEINFO * const current, | ||
247 : | const FRAMEINFO * const reference, | ||
248 : | const IMAGE * const pRefH, | ||
249 : | const IMAGE * const pRefV, | ||
250 : | const IMAGE * const pRefHV); | ||
251 : | |||
252 : | |||
253 : | |||
254 : | WARPPOINTS | ||
255 : | GlobalMotionEst(MACROBLOCK * const pMBs, | ||
256 : | edgomez | 1.2 | const MBParam * const pParam, |
257 : | const FRAMEINFO * const current, | ||
258 : | const FRAMEINFO * const reference, | ||
259 : | const IMAGE * const pRefH, | ||
260 : | const IMAGE * const pRefV, | ||
261 : | const IMAGE * const pRefHV ); | ||
262 : | chl | 1.3.2.12 | |
263 : | |||
264 : | int | ||
265 : | GlobalMotionEstRefine(WARPPOINTS *const startwp, | ||
266 : | MACROBLOCK * const pMBs, | ||
267 : | const MBParam * const pParam, | ||
268 : | const FRAMEINFO * const current, | ||
269 : | const FRAMEINFO * const reference, | ||
270 : | const IMAGE * const pCurr, | ||
271 : | const IMAGE * const pRef, | ||
272 : | const IMAGE * const pRefH, | ||
273 : | const IMAGE * const pRefV, | ||
274 : | const IMAGE * const pRefHV); | ||
275 : | |||
276 : | |||
277 : | |||
278 : | |||
279 : | int | ||
280 : | globalSAD(const WARPPOINTS *const wp, | ||
281 : | const MBParam * const pParam, | ||
282 : | const MACROBLOCK * const pMBs, | ||
283 : | const FRAMEINFO * const current, | ||
284 : | const IMAGE * const pRef, | ||
285 : | const IMAGE * const pCurr, | ||
286 : | uint8_t *const GMCblock); | ||
287 : | edgomez | 1.2 | |
288 : | #define iDiamondSize 2 | ||
289 : | |||
290 : | static __inline uint32_t | ||
291 : | suxen_drol | 1.3.2.1 | MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t VopFlags, const uint32_t VolFlags) |
292 : | edgomez | 1.2 | { |
293 : | uint32_t Flags = MotionFlags; | ||
294 : | |||
295 : | edgomez | 1.3.2.14 | if (!(VopFlags & XVID_VOP_MODEDECISION_RD)) |
296 : | Flags &= ~(XVID_ME_QUARTERPELREFINE16_RD+XVID_ME_QUARTERPELREFINE8_RD+XVID_ME_HALFPELREFINE16_RD+XVID_ME_HALFPELREFINE8_RD+XVID_ME_EXTSEARCH_RD); | ||
297 : | edgomez | 1.2 | |
298 : | edgomez | 1.3.2.14 | if (Flags & XVID_ME_EXTSEARCH_RD) |
299 : | Flags |= XVID_ME_HALFPELREFINE16_RD; | ||
300 : | edgomez | 1.2 | |
301 : | edgomez | 1.3.2.14 | if (Flags & XVID_ME_EXTSEARCH_RD && MotionFlags & XVID_ME_EXTSEARCH8) |
302 : | Flags |= XVID_ME_HALFPELREFINE8_RD; | ||
303 : | edgomez | 1.2 | |
304 : | edgomez | 1.3.2.14 | if (Flags & XVID_ME_HALFPELREFINE16_RD) |
305 : | Flags |= XVID_ME_QUARTERPELREFINE16_RD; | ||
306 : | edgomez | 1.2 | |
307 : | edgomez | 1.3.2.14 | if (Flags & XVID_ME_HALFPELREFINE8_RD) { |
308 : | Flags |= XVID_ME_QUARTERPELREFINE8_RD; | ||
309 : | edgomez | 1.3.2.2 | Flags &= ~XVID_ME_HALFPELREFINE8; |
310 : | edgomez | 1.2 | } |
311 : | |||
312 : | edgomez | 1.3.2.14 | if (Flags & XVID_ME_QUARTERPELREFINE8_RD) |
313 : | edgomez | 1.3.2.2 | Flags &= ~XVID_ME_QUARTERPELREFINE8; |
314 : | edgomez | 1.2 | |
315 : | edgomez | 1.3.2.2 | if (!(VolFlags & XVID_VOL_QUARTERPEL)) |
316 : | edgomez | 1.3.2.14 | Flags &= ~(XVID_ME_QUARTERPELREFINE16+XVID_ME_QUARTERPELREFINE8+XVID_ME_QUARTERPELREFINE16_RD+XVID_ME_QUARTERPELREFINE8_RD); |
317 : | edgomez | 1.2 | |
318 : | edgomez | 1.3.2.2 | if (!(VopFlags & XVID_VOP_HALFPEL)) |
319 : | edgomez | 1.3.2.14 | Flags &= ~(XVID_ME_EXTSEARCH16+XVID_ME_HALFPELREFINE16+XVID_ME_HALFPELREFINE8+XVID_ME_HALFPELREFINE16_RD+XVID_ME_HALFPELREFINE8_RD); |
320 : | edgomez | 1.2 | |
321 : | edgomez | 1.3.2.2 | if ((VopFlags & XVID_VOP_GREYSCALE) || (VopFlags & XVID_VOP_REDUCED)) |
322 : | edgomez | 1.3.2.14 | Flags &= ~(XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP); |
323 : | edgomez | 1.2 | |
324 : | return Flags; | ||
325 : | } | ||
326 : | |||
327 : | edgomez | 1.3.2.14 | /* RD mode decision and search */ |
328 : | edgomez | 1.2 | |
329 : | #include "../bitstream/zigzag.h" | ||
330 : | #include "../quant/quant_mpeg4.h" | ||
331 : | #include "../quant/quant_h263.h" | ||
332 : | #include "../bitstream/vlc_codes.h" | ||
333 : | edgomez | 1.3.2.6 | #include "../dct/fdct.h" |
334 : | edgomez | 1.2 | |
335 : | static int | ||
336 : | syskin | 1.3.2.15 | findRDinter(SearchData * const Data, |
337 : | const MACROBLOCK * const pMBs, const int x, const int y, | ||
338 : | const MBParam * const pParam, | ||
339 : | const uint32_t MotionFlags); | ||
340 : | edgomez | 1.2 | |
341 : | static int | ||
342 : | syskin | 1.3.2.15 | findRDinter4v(const SearchData * const Data, |
343 : | MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, | ||
344 : | const int x, const int y, | ||
345 : | const MBParam * const pParam, const uint32_t MotionFlags, | ||
346 : | const VECTOR * const backup); | ||
347 : | edgomez | 1.2 | |
348 : | static int | ||
349 : | syskin | 1.3.2.15 | findRDintra(const SearchData * const Data); |
350 : | edgomez | 1.2 | |
351 : | syskin | 1.3.2.13 | static int |
352 : | syskin | 1.3.2.15 | findRDgmc(const SearchData * const Data, const IMAGE * const vGMC, const int x, const int y); |
353 : | syskin | 1.3.2.13 | |
354 : | edgomez | 1.2 | int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
355 : | int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); | ||
356 : | edgomez | 1.3.2.7 | |
357 : | syskin | 1.3.2.11 | /* one over lambda for R-D mode decision and motion search */ |
358 : | #define LAMBDA ( (int)(BITS_MULT/1.0) ) | ||
359 : | edgomez | 1.3.2.7 | |
360 : | edgomez | 1.3.2.6 | static __inline unsigned int |
361 : | edgomez | 1.3.2.7 | Block_CalcBits( int16_t * const coeff, |
362 : | int16_t * const data, | ||
363 : | syskin | 1.3.2.11 | int16_t * const dqcoeff, |
364 : | edgomez | 1.3.2.7 | const uint32_t quant, const int quant_type, |
365 : | uint32_t * cbp, | ||
366 : | const int block) | ||
367 : | edgomez | 1.3.2.6 | { |
368 : | int sum; | ||
369 : | edgomez | 1.3.2.7 | int bits; |
370 : | int distortion = 0; | ||
371 : | int i; | ||
372 : | edgomez | 1.3.2.6 | |
373 : | fdct(data); | ||
374 : | |||
375 : | syskin | 1.3.2.13 | if (quant_type) sum = quant_inter(coeff, data, quant); |
376 : | edgomez | 1.3.2.6 | else sum = quant4_inter(coeff, data, quant); |
377 : | |||
378 : | if (sum > 0) { | ||
379 : | *cbp |= 1 << (5 - block); | ||
380 : | edgomez | 1.3.2.7 | bits = BITS_MULT * CodeCoeffInter_CalcBits(coeff, scan_tables[0]); |
381 : | |||
382 : | syskin | 1.3.2.13 | if (quant_type) dequant_inter(dqcoeff, coeff, quant); |
383 : | else dequant4_inter(dqcoeff, coeff, quant); | ||
384 : | edgomez | 1.3.2.7 | |
385 : | syskin | 1.3.2.13 | for (i = 0; i < 64; i++) |
386 : | distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); | ||
387 : | |||
388 : | } else { | ||
389 : | bits = 0; | ||
390 : | for (i = 0; i < 64; i++) | ||
391 : | distortion += data[i]*data[i]; | ||
392 : | edgomez | 1.3.2.7 | } |
393 : | |||
394 : | syskin | 1.3.2.15 | return bits + (LAMBDA*distortion)/(quant*quant); |
395 : | edgomez | 1.3.2.7 | } |
396 : | |||
397 : | static __inline unsigned int | ||
398 : | Block_CalcBitsIntra(int16_t * const coeff, | ||
399 : | int16_t * const data, | ||
400 : | syskin | 1.3.2.11 | int16_t * const dqcoeff, |
401 : | edgomez | 1.3.2.7 | const uint32_t quant, const int quant_type, |
402 : | uint32_t * cbp, | ||
403 : | const int block, | ||
404 : | int * dcpred) | ||
405 : | { | ||
406 : | int bits, i; | ||
407 : | int distortion = 0; | ||
408 : | syskin | 1.3.2.10 | uint32_t iDcScaler = get_dc_scaler(quant, block < 4); |
409 : | edgomez | 1.3.2.7 | int b_dc; |
410 : | |||
411 : | fdct(data); | ||
412 : | data[0] -= 1024; | ||
413 : | |||
414 : | syskin | 1.3.2.13 | if (quant_type) quant_intra(coeff, data, quant, iDcScaler); |
415 : | syskin | 1.3.2.11 | else quant4_intra(coeff, data, quant, iDcScaler); |
416 : | edgomez | 1.3.2.7 | |
417 : | b_dc = coeff[0]; | ||
418 : | if (block < 4) { | ||
419 : | coeff[0] -= *dcpred; | ||
420 : | *dcpred = b_dc; | ||
421 : | } | ||
422 : | |||
423 : | bits = BITS_MULT*CodeCoeffIntra_CalcBits(coeff, scan_tables[0]); | ||
424 : | if (bits != 0) *cbp |= 1 << (5 - block); | ||
425 : | |||
426 : | syskin | 1.3.2.10 | if (block < 4) bits += BITS_MULT*dcy_tab[coeff[0] + 255].len; |
427 : | else bits += BITS_MULT*dcc_tab[coeff[0] + 255].len; | ||
428 : | |||
429 : | edgomez | 1.3.2.7 | coeff[0] = b_dc; |
430 : | syskin | 1.3.2.13 | if (quant_type) dequant_intra(dqcoeff, coeff, quant, iDcScaler); |
431 : | syskin | 1.3.2.11 | else dequant4_intra(dqcoeff, coeff, quant, iDcScaler); |
432 : | edgomez | 1.3.2.7 | |
433 : | syskin | 1.3.2.15 | for (i = 0; i < 64; i++) |
434 : | syskin | 1.3.2.11 | distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); |
435 : | edgomez | 1.3.2.7 | |
436 : | syskin | 1.3.2.15 | return bits + (LAMBDA*distortion)/(quant*quant); |
437 : | edgomez | 1.3.2.6 | } |
438 : | edgomez | 1.2 | |
439 : | #endif /* _MOTION_EST_H_ */ |
No admin address has been configured | ViewVC Help |
Powered by ViewVC 1.0.4 |