[cvs] / xvidcore / src / encoder.c Repository:
ViewVC logotype

Annotation of /xvidcore/src/encoder.c

Parent Directory Parent Directory | Revision Log Revision Log

Revision 1.6 - (view) (download)

1 : Isibaar 1.1 #include <stdlib.h>
2 :     #include <stdio.h>
3 :     #include <math.h>
4 :    
5 :     #include "encoder.h"
6 :     #include "prediction/mbprediction.h"
7 :     #include "global.h"
8 :     #include "utils/timer.h"
9 :     #include "image/image.h"
10 :     #include "bitstream/cbp.h"
11 :     #include "utils/mbfunctions.h"
12 :     #include "bitstream/bitstream.h"
13 :     #include "bitstream/mbcoding.h"
14 :     #include "utils/ratecontrol.h"
15 :     #include "utils/emms.h"
16 :     #include "bitstream/mbcoding.h"
17 :     #include "quant/adapt_quant.h"
18 : Isibaar 1.2 #include "quant/quant_matrix.h"
19 : Isibaar 1.1
20 :     #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT
21 :    
22 :    
23 :     static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits);
24 :     static int FrameCodeP(Encoder * pEnc, Bitstream * bs, uint32_t *pBits, bool force_inter, bool vol_header);
25 :    
26 :     static int DQtab[4] =
27 :     {
28 :     -1, -2, 1, 2
29 :     };
30 :    
31 :     static int iDQtab[5] =
32 :     {
33 :     1, 0, NO_CHANGE, 2, 3
34 :     };
35 :    
36 :    
37 :     int encoder_create(XVID_ENC_PARAM * pParam)
38 :     {
39 : edgomez 1.3 Encoder *pEnc;
40 : Isibaar 1.1 uint32_t i;
41 :    
42 : edgomez 1.3 pParam->handle = NULL;
43 : Isibaar 1.1
44 : edgomez 1.3 ENC_CHECK(pParam);
45 : Isibaar 1.1
46 : edgomez 1.3 ENC_CHECK(pParam->width > 0 && pParam->width <= 1920);
47 :     ENC_CHECK(pParam->height > 0 && pParam->height <= 1280);
48 :     ENC_CHECK(!(pParam->width % 2));
49 :     ENC_CHECK(!(pParam->height % 2));
50 : Isibaar 1.1
51 :     if (pParam->fincr <= 0 || pParam->fbase <= 0)
52 :     {
53 :     pParam->fincr = 1;
54 :     pParam->fbase = 25;
55 :     }
56 :    
57 :     // simplify the "fincr/fbase" fraction
58 :     // (neccessary, since windows supplies us with huge numbers)
59 :    
60 :     i = pParam->fincr;
61 :     while (i > 1)
62 :     {
63 :     if (pParam->fincr % i == 0 && pParam->fbase % i == 0)
64 :     {
65 :     pParam->fincr /= i;
66 :     pParam->fbase /= i;
67 :     i = pParam->fincr;
68 :     continue;
69 :     }
70 :     i--;
71 :     }
72 :    
73 :     if (pParam->fbase > 65535)
74 :     {
75 :     float div = (float)pParam->fbase / 65535;
76 :     pParam->fbase = (int)(pParam->fbase / div);
77 :     pParam->fincr = (int)(pParam->fincr / div);
78 :     }
79 :    
80 :     if (pParam->bitrate <= 0)
81 :     pParam->bitrate = 900000;
82 :    
83 : edgomez 1.3 if (pParam->rc_buffersize <= 0)
84 : Isibaar 1.1 pParam->rc_buffersize = pParam->bitrate * pParam->fbase;
85 :    
86 : edgomez 1.3 if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))
87 : Isibaar 1.1 pParam->min_quantizer = 1;
88 :    
89 : edgomez 1.3 if ((pParam->max_quantizer <= 0) || (pParam->max_quantizer > 31))
90 : Isibaar 1.1 pParam->max_quantizer = 31;
91 :    
92 : edgomez 1.3 if (pParam->max_key_interval == 0) /* 1 keyframe each 10 seconds */
93 : Isibaar 1.1 pParam->max_key_interval = 10 * pParam->fincr / pParam->fbase;
94 :    
95 : edgomez 1.3 if (pParam->max_quantizer < pParam->min_quantizer)
96 : Isibaar 1.1 pParam->max_quantizer = pParam->min_quantizer;
97 :    
98 : edgomez 1.3 if ((pEnc = (Encoder *) malloc(sizeof(Encoder))) == NULL)
99 : Isibaar 1.1 return XVID_ERR_MEMORY;
100 :    
101 :     /* Fill members of Encoder structure */
102 :    
103 : edgomez 1.3 pEnc->mbParam.width = pParam->width;
104 :     pEnc->mbParam.height = pParam->height;
105 : Isibaar 1.1
106 :     pEnc->mbParam.mb_width = (pEnc->mbParam.width + 15) / 16;
107 :     pEnc->mbParam.mb_height = (pEnc->mbParam.height + 15) / 16;
108 :    
109 :     pEnc->mbParam.edged_width = 16 * pEnc->mbParam.mb_width + 2 * EDGE_SIZE;
110 :     pEnc->mbParam.edged_height = 16 * pEnc->mbParam.mb_height + 2 * EDGE_SIZE;
111 :    
112 : edgomez 1.3 pEnc->sStat.fMvPrevSigma = -1;
113 : Isibaar 1.1
114 :     /* Fill rate control parameters */
115 :    
116 : edgomez 1.3 pEnc->mbParam.quant = 4;
117 : Isibaar 1.1
118 :     pEnc->bitrate = pParam->bitrate;
119 :    
120 : edgomez 1.3 pEnc->iFrameNum = 0;
121 :     pEnc->iMaxKeyInterval = pParam->max_key_interval;
122 : Isibaar 1.1
123 : edgomez 1.3 if (image_create(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)
124 :     {
125 : Isibaar 1.1 free(pEnc);
126 :     return XVID_ERR_MEMORY;
127 : edgomez 1.3 }
128 : Isibaar 1.1
129 :     if (image_create(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)
130 : edgomez 1.3 {
131 : Isibaar 1.1 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
132 :     free(pEnc);
133 :     return XVID_ERR_MEMORY;
134 : edgomez 1.3 }
135 : Isibaar 1.1
136 : edgomez 1.3 if (image_create(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)
137 :     {
138 : Isibaar 1.1 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
139 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
140 :     free(pEnc);
141 :     return XVID_ERR_MEMORY;
142 : edgomez 1.3 }
143 : Isibaar 1.1
144 : edgomez 1.3 if (image_create(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)
145 :     {
146 : Isibaar 1.1 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
147 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
148 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
149 :     free(pEnc);
150 :     return XVID_ERR_MEMORY;
151 : edgomez 1.3 }
152 : Isibaar 1.1
153 : edgomez 1.3 if (image_create(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)
154 :     {
155 : Isibaar 1.1 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
156 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
157 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
158 :     image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
159 :     free(pEnc);
160 :     return XVID_ERR_MEMORY;
161 : edgomez 1.3 }
162 : Isibaar 1.1
163 :     pEnc->pMBs = malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height);
164 :     if (pEnc->pMBs == NULL)
165 :     {
166 :     image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
167 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
168 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
169 :     image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
170 :     image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
171 :     free(pEnc);
172 :     return XVID_ERR_MEMORY;
173 :     }
174 :    
175 :     // init macroblock array
176 :     for (i = 0; i < pEnc->mbParam.mb_width * pEnc->mbParam.mb_height; i++)
177 :     {
178 :     pEnc->pMBs[i].dquant = NO_CHANGE;
179 :     }
180 :    
181 : edgomez 1.3 pParam->handle = (void *)pEnc;
182 : Isibaar 1.1
183 :     if (pParam->bitrate)
184 :     {
185 :     RateControlInit(pParam->bitrate, pParam->rc_buffersize, pParam->fbase, pParam->width,
186 : edgomez 1.3 pParam->height, pParam->max_quantizer, pParam->min_quantizer);
187 : Isibaar 1.1 }
188 :    
189 :     create_vlc_tables();
190 : Isibaar 1.6 init_timer();
191 : Isibaar 1.1
192 :     return XVID_ERR_OK;
193 :     }
194 :    
195 :    
196 :     int encoder_destroy(Encoder * pEnc)
197 :     {
198 : edgomez 1.3 ENC_CHECK(pEnc);
199 :     ENC_CHECK(pEnc->sCurrent.y);
200 :     ENC_CHECK(pEnc->sReference.y);
201 : Isibaar 1.1
202 :     free(pEnc->pMBs);
203 : edgomez 1.3 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
204 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
205 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
206 :     image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
207 :     image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
208 :     free(pEnc);
209 : Isibaar 1.1
210 :     destroy_vlc_tables();
211 :    
212 : edgomez 1.3 return XVID_ERR_OK;
213 : Isibaar 1.1 }
214 :    
215 :     int encoder_encode(Encoder * pEnc, XVID_ENC_FRAME * pFrame, XVID_ENC_STATS * pResult)
216 :     {
217 : edgomez 1.3 uint16_t x, y;
218 :     Bitstream bs;
219 :     uint32_t bits;
220 : Isibaar 1.2 uint16_t write_vol_header = 0;
221 : Isibaar 1.1
222 :     start_global_timer();
223 :    
224 : edgomez 1.3 ENC_CHECK(pEnc);
225 :     ENC_CHECK(pFrame);
226 :     ENC_CHECK(pFrame->bitstream);
227 :     ENC_CHECK(pFrame->image);
228 : Isibaar 1.1
229 :     pEnc->mbParam.global_flags = pFrame->general;
230 :     pEnc->mbParam.motion_flags = pFrame->motion;
231 :    
232 :     start_timer();
233 :     if (image_input(&pEnc->sCurrent, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width,
234 : edgomez 1.3 pFrame->image, pFrame->colorspace))
235 : Isibaar 1.1 {
236 :     return XVID_ERR_FORMAT;
237 :     }
238 :     stop_conv_timer();
239 :    
240 : edgomez 1.3 BitstreamInit(&bs, pFrame->bitstream, 0);
241 : Isibaar 1.1
242 :     if (pFrame->quant == 0)
243 :     {
244 :     pEnc->mbParam.quant = RateControlGetQ(0);
245 :     }
246 :     else
247 :     {
248 :     pEnc->mbParam.quant = pFrame->quant;
249 :     }
250 :    
251 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)
252 :     {
253 :     int * temp_dquants = (int *) malloc(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * sizeof(int));
254 :    
255 :     pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y, pEnc->mbParam.width,
256 : edgomez 1.3 temp_dquants, pFrame->quant, pFrame->quant,
257 :     2*pFrame->quant, pEnc->mbParam.mb_width, pEnc->mbParam.mb_height);
258 : Isibaar 1.1
259 :     for (y = 0; y < pEnc->mbParam.mb_height; y++)
260 :     for (x = 0; x < pEnc->mbParam.mb_width; x++)
261 :     {
262 :     MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
263 :     pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];
264 :     }
265 :     free(temp_dquants);
266 :     }
267 :    
268 : Isibaar 1.4 if(pEnc->mbParam.global_flags & XVID_H263QUANT) {
269 :     if(pEnc->mbParam.quant_type != H263_QUANT)
270 :     write_vol_header = 1;
271 :     pEnc->mbParam.quant_type = H263_QUANT;
272 : Isibaar 1.1 }
273 : Isibaar 1.4 else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {
274 :     int ret1, ret2;
275 : Isibaar 1.1
276 : Isibaar 1.4 if(pEnc->mbParam.quant_type != MPEG4_QUANT)
277 :     write_vol_header = 1;
278 :    
279 :     pEnc->mbParam.quant_type = MPEG4_QUANT;
280 :    
281 :     if ((pEnc->mbParam.global_flags & XVID_CUSTOM_QMATRIX) > 0) {
282 :     if(pFrame->quant_intra_matrix != NULL)
283 :     ret1 = set_intra_matrix(pFrame->quant_intra_matrix);
284 :     if(pFrame->quant_inter_matrix != NULL)
285 :     ret2 = set_inter_matrix(pFrame->quant_inter_matrix);
286 :     }
287 :     else {
288 :     ret1 = set_intra_matrix(get_default_intra_matrix());
289 :     ret2 = set_inter_matrix(get_default_inter_matrix());
290 :     }
291 : Isibaar 1.2 if(write_vol_header == 0)
292 :     write_vol_header = ret1 | ret2;
293 :     }
294 : Isibaar 1.1
295 :     if (pFrame->intra < 0)
296 : edgomez 1.3 {
297 : Isibaar 1.1 if ((pEnc->iFrameNum == 0) || ((pEnc->iMaxKeyInterval > 0)
298 : edgomez 1.3 && (pEnc->iFrameNum >= pEnc->iMaxKeyInterval)))
299 : Isibaar 1.1
300 :     pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
301 :     else
302 : Isibaar 1.2 pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 0, write_vol_header);
303 : edgomez 1.3 }
304 :     else
305 :     {
306 : Isibaar 1.1 if (pFrame->intra == 1)
307 : edgomez 1.3 pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
308 : Isibaar 1.1 else
309 : Isibaar 1.2 pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 1, write_vol_header);
310 : edgomez 1.3 }
311 : Isibaar 1.1
312 :     BitstreamPutBits(&bs, 0xFFFF, 16);
313 : edgomez 1.3 BitstreamPutBits(&bs, 0xFFFF, 16);
314 :     BitstreamPad(&bs);
315 :     pFrame->length = BitstreamLength(&bs);
316 : h 1.5
317 : Isibaar 1.1 if (pResult)
318 : edgomez 1.3 {
319 : Isibaar 1.1 pResult->quant = pEnc->mbParam.quant;
320 :     pResult->hlength = pFrame->length - (pEnc->sStat.iTextBits / 8);
321 :     pResult->kblks = pEnc->sStat.kblks;
322 :     pResult->mblks = pEnc->sStat.mblks;
323 :     pResult->ublks = pEnc->sStat.ublks;
324 : edgomez 1.3 }
325 : Isibaar 1.1
326 : h 1.5 if (pFrame->quant == 0)
327 : Isibaar 1.1 {
328 :     RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);
329 :     }
330 :    
331 :     pEnc->iFrameNum++;
332 : edgomez 1.3 image_swap(&pEnc->sCurrent, &pEnc->sReference);
333 : Isibaar 1.1
334 :     stop_global_timer();
335 :     write_timer();
336 :    
337 :     return XVID_ERR_OK;
338 :     }
339 :    
340 :    
341 :     static __inline void CodeIntraMB(Encoder *pEnc, MACROBLOCK *pMB) {
342 :    
343 :     pMB->mode = MODE_INTRA;
344 :    
345 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0) {
346 :     if(pMB->dquant != NO_CHANGE)
347 :     {
348 :     pMB->mode = MODE_INTRA_Q;
349 :     pEnc->mbParam.quant += DQtab[pMB->dquant];
350 :    
351 :     if (pEnc->mbParam.quant > 31) pEnc->mbParam.quant = 31;
352 :     if (pEnc->mbParam.quant < 1) pEnc->mbParam.quant = 1;
353 :     }
354 :     }
355 :    
356 :     pMB->quant = pEnc->mbParam.quant;
357 :     }
358 :    
359 :    
360 :     static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
361 :     {
362 : edgomez 1.3 int16_t dct_codes[6][64];
363 :     int16_t qcoeff[6][64];
364 :     uint16_t x, y;
365 :    
366 :     pEnc->iFrameNum = 0;
367 :     pEnc->mbParam.rounding_type = 1;
368 :     pEnc->mbParam.coding_type = I_VOP;
369 : Isibaar 1.1
370 :     BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);
371 :     BitstreamWriteVopHeader(bs, I_VOP, pEnc->mbParam.rounding_type,
372 : edgomez 1.3 pEnc->mbParam.quant,
373 :     pEnc->mbParam.fixed_code);
374 : Isibaar 1.1
375 : edgomez 1.3 *pBits = BitstreamPos(bs);
376 : Isibaar 1.1
377 :     pEnc->sStat.iTextBits = 0;
378 :     pEnc->sStat.kblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;
379 :     pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
380 :    
381 : edgomez 1.3 for (y = 0; y < pEnc->mbParam.mb_height; y++)
382 : Isibaar 1.1 for (x = 0; x < pEnc->mbParam.mb_width; x++)
383 :     {
384 : edgomez 1.3 MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
385 : Isibaar 1.1
386 :     CodeIntraMB(pEnc, pMB);
387 :    
388 : edgomez 1.3 MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, &pEnc->sCurrent);
389 : Isibaar 1.1
390 :     start_timer();
391 :     MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
392 :     stop_prediction_timer();
393 :    
394 :     start_timer();
395 :     MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);
396 :     stop_coding_timer();
397 :     }
398 :    
399 :     emms();
400 :    
401 : edgomez 1.3 *pBits = BitstreamPos(bs) - *pBits;
402 :     pEnc->sStat.fMvPrevSigma = -1;
403 :     pEnc->sStat.iMvSum = 0;
404 :     pEnc->sStat.iMvCount = 0;
405 :     pEnc->mbParam.fixed_code = 2;
406 : Isibaar 1.1
407 : edgomez 1.3 return 1; // intra
408 : Isibaar 1.1 }
409 :    
410 :    
411 :     #define INTRA_THRESHOLD 0.5
412 :    
413 :     static int FrameCodeP(Encoder * pEnc, Bitstream * bs, uint32_t *pBits, bool force_inter, bool vol_header)
414 :     {
415 : edgomez 1.3 float fSigma;
416 :     int16_t dct_codes[6][64];
417 :     int16_t qcoeff[6][64];
418 : Isibaar 1.1 int iLimit;
419 : edgomez 1.3 uint32_t x, y;
420 :     int iSearchRange;
421 : Isibaar 1.1 bool bIntra;
422 :    
423 : edgomez 1.3 IMAGE *pCurrent = &pEnc->sCurrent;
424 :     IMAGE *pRef = &pEnc->sReference;
425 : Isibaar 1.1
426 :     image_setedges(pRef,pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);
427 :    
428 : edgomez 1.3 pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;
429 : Isibaar 1.1
430 :     if (!force_inter)
431 :     iLimit = (int)(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * INTRA_THRESHOLD);
432 : edgomez 1.3 else
433 : Isibaar 1.1 iLimit = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height + 1;
434 :    
435 :     if ((pEnc->mbParam.global_flags & XVID_HALFPEL) > 0) {
436 :     start_timer();
437 :     image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
438 : edgomez 1.3 pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
439 :     pEnc->mbParam.rounding_type);
440 : Isibaar 1.1 stop_inter_timer();
441 :     }
442 :    
443 :     start_timer();
444 :     bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,
445 : edgomez 1.3 &pEnc->vInterH, &pEnc->vInterV,
446 :     &pEnc->vInterHV, &pEnc->sCurrent, iLimit);
447 : Isibaar 1.1 stop_motion_timer();
448 :    
449 :     if (bIntra == 1)
450 :     return FrameCodeI(pEnc, bs, pBits);
451 :    
452 : edgomez 1.3 pEnc->mbParam.coding_type = P_VOP;
453 : Isibaar 1.1
454 :     if(vol_header)
455 :     BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);
456 :    
457 : edgomez 1.3 BitstreamWriteVopHeader(bs, P_VOP, pEnc->mbParam.rounding_type,
458 :     pEnc->mbParam.quant,
459 :     pEnc->mbParam.fixed_code);
460 :    
461 :     *pBits = BitstreamPos(bs);
462 :    
463 :     pEnc->sStat.iTextBits = 0;
464 :     pEnc->sStat.iMvSum = 0;
465 :     pEnc->sStat.iMvCount = 0;
466 : Isibaar 1.1 pEnc->sStat.kblks = pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
467 :    
468 : edgomez 1.3 for(y = 0; y < pEnc->mbParam.mb_height; y++)
469 : Isibaar 1.1 {
470 :     for(x = 0; x < pEnc->mbParam.mb_width; x++)
471 :     {
472 :     MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
473 :    
474 : edgomez 1.3 bIntra = (pMB->mode == MODE_INTRA) || (pMB->mode == MODE_INTRA_Q);
475 : Isibaar 1.1
476 :     if (!bIntra)
477 : edgomez 1.3 {
478 : Isibaar 1.1 start_timer();
479 :     MBMotionCompensation(pMB, x, y, &pEnc->sReference,
480 : edgomez 1.3 &pEnc->vInterH, &pEnc->vInterV,
481 :     &pEnc->vInterHV, &pEnc->sCurrent, dct_codes,
482 :     pEnc->mbParam.width,
483 :     pEnc->mbParam.height,
484 :     pEnc->mbParam.edged_width,
485 :     pEnc->mbParam.rounding_type);
486 : Isibaar 1.1 stop_comp_timer();
487 :    
488 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0) {
489 :     if(pMB->dquant != NO_CHANGE) {
490 :     pMB->mode = MODE_INTER_Q;
491 :     pEnc->mbParam.quant += DQtab[pMB->dquant];
492 :     if (pEnc->mbParam.quant > 31) pEnc->mbParam.quant = 31;
493 :     else if(pEnc->mbParam.quant < 1) pEnc->mbParam.quant = 1;
494 :     }
495 :     }
496 :     pMB->quant = pEnc->mbParam.quant;
497 :    
498 :     pMB->cbp = MBTransQuantInter(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);
499 : edgomez 1.3 }
500 : Isibaar 1.1 else
501 :     {
502 :     CodeIntraMB(pEnc, pMB);
503 :     MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);
504 :     }
505 :    
506 : edgomez 1.3 start_timer();
507 : Isibaar 1.1 MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
508 :     stop_prediction_timer();
509 :    
510 :     if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q)
511 :     {
512 :     pEnc->sStat.kblks++;
513 :     }
514 :     else if (pMB->cbp ||
515 : edgomez 1.3 pMB->mvs[0].x || pMB->mvs[0].y ||
516 :     pMB->mvs[1].x || pMB->mvs[1].y ||
517 :     pMB->mvs[2].x || pMB->mvs[2].y ||
518 :     pMB->mvs[3].x || pMB->mvs[3].y)
519 : Isibaar 1.1 {
520 :     pEnc->sStat.mblks++;
521 :     }
522 :     else
523 :     {
524 :     pEnc->sStat.ublks++;
525 :     }
526 :    
527 :     start_timer();
528 :     MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);
529 :     stop_coding_timer();
530 :     }
531 :     }
532 :    
533 :     emms();
534 :    
535 :     if (pEnc->sStat.iMvCount == 0)
536 :     pEnc->sStat.iMvCount = 1;
537 :    
538 : edgomez 1.3 fSigma = (float)sqrt((float) pEnc->sStat.iMvSum / pEnc->sStat.iMvCount);
539 : Isibaar 1.1
540 : edgomez 1.3 iSearchRange = 1 << (3 + pEnc->mbParam.fixed_code);
541 : Isibaar 1.1
542 : edgomez 1.3 if ((fSigma > iSearchRange / 3)
543 :     && (pEnc->mbParam.fixed_code <= 3)) // maximum search range 128
544 :     {
545 : Isibaar 1.1 pEnc->mbParam.fixed_code++;
546 :     iSearchRange *= 2;
547 : edgomez 1.3 }
548 :     else if ((fSigma < iSearchRange / 6)
549 :     && (pEnc->sStat.fMvPrevSigma >= 0)
550 :     && (pEnc->sStat.fMvPrevSigma < iSearchRange / 6)
551 :     && (pEnc->mbParam.fixed_code >= 2)) // minimum search range 16
552 :     {
553 : Isibaar 1.1 pEnc->mbParam.fixed_code--;
554 :     iSearchRange /= 2;
555 : edgomez 1.3 }
556 : Isibaar 1.1
557 : edgomez 1.3 pEnc->sStat.fMvPrevSigma = fSigma;
558 : Isibaar 1.1
559 :     *pBits = BitstreamPos(bs) - *pBits;
560 :    
561 : edgomez 1.3 return 0; // inter
562 : Isibaar 1.1 }
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4