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

Annotation of /xvidcore/src/encoder.c

Parent Directory Parent Directory | Revision Log Revision Log


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

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4