[svn] / branches / dev-api-3 / xvidcore / src / encoder.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-3/xvidcore/src/encoder.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 101 - (view) (download)
Original Path: trunk/xvidcore/src/encoder.c

1 : Isibaar 3 #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 4 #include "quant/quant_matrix.h"
19 : Isibaar 41 #include "utils/mem_align.h"
20 : Isibaar 3
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 13 Encoder *pEnc;
41 : Isibaar 3 uint32_t i;
42 :    
43 : edgomez 13 pParam->handle = NULL;
44 : Isibaar 3
45 : edgomez 13 ENC_CHECK(pParam);
46 : Isibaar 3
47 : edgomez 13 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 3
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 13 if (pParam->rc_buffersize <= 0)
85 : h 64 pParam->rc_buffersize = 16;
86 : Isibaar 3
87 : edgomez 13 if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))
88 : Isibaar 3 pParam->min_quantizer = 1;
89 :    
90 : edgomez 13 if ((pParam->max_quantizer <= 0) || (pParam->max_quantizer > 31))
91 : Isibaar 3 pParam->max_quantizer = 31;
92 :    
93 : edgomez 13 if (pParam->max_key_interval == 0) /* 1 keyframe each 10 seconds */
94 : Isibaar 3 pParam->max_key_interval = 10 * pParam->fincr / pParam->fbase;
95 :    
96 : edgomez 13 if (pParam->max_quantizer < pParam->min_quantizer)
97 : Isibaar 3 pParam->max_quantizer = pParam->min_quantizer;
98 :    
99 : canard 66 if ((pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE)) == NULL)
100 : Isibaar 3 return XVID_ERR_MEMORY;
101 :    
102 :     /* Fill members of Encoder structure */
103 :    
104 : edgomez 13 pEnc->mbParam.width = pParam->width;
105 :     pEnc->mbParam.height = pParam->height;
106 : Isibaar 3
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 13 pEnc->sStat.fMvPrevSigma = -1;
114 : Isibaar 3
115 :     /* Fill rate control parameters */
116 :    
117 : edgomez 13 pEnc->mbParam.quant = 4;
118 : Isibaar 3
119 :     pEnc->bitrate = pParam->bitrate;
120 :    
121 : edgomez 13 pEnc->iFrameNum = 0;
122 :     pEnc->iMaxKeyInterval = pParam->max_key_interval;
123 : Isibaar 3
124 : h 86 /* try to allocate memory */
125 : Isibaar 3
126 : h 86 pEnc->sCurrent.y = pEnc->sCurrent.u = pEnc->sCurrent.v = NULL;
127 :     pEnc->sReference.y = pEnc->sReference.u = pEnc->sReference.v = NULL;
128 :     pEnc->vInterH.y = pEnc->vInterH.u = pEnc->vInterH.v = NULL;
129 :     pEnc->vInterV.y = pEnc->vInterV.u = pEnc->vInterV.v = NULL;
130 :     pEnc->vInterVf.y = pEnc->vInterVf.u = pEnc->vInterVf.v = NULL;
131 :     pEnc->vInterHV.y = pEnc->vInterHV.u = pEnc->vInterHV.v = NULL;
132 :     pEnc->vInterHVf.y = pEnc->vInterHVf.u = pEnc->vInterHVf.v = NULL;
133 : Isibaar 3
134 : h 86 pEnc->pMBs = NULL;
135 : Isibaar 3
136 : h 86 if (image_create(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
137 :     image_create(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
138 :     image_create(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
139 :     image_create(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
140 :     image_create(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
141 :     image_create(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
142 :     image_create(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
143 :     (pEnc->pMBs = xvid_malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height, CACHE_LINE)) == NULL)
144 : edgomez 13 {
145 : Isibaar 3 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
146 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
147 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
148 :     image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
149 : h 86 image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
150 : Isibaar 3 image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
151 : h 86 image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
152 :     if (pEnc)
153 :     {
154 :     xvid_free(pEnc);
155 :     }
156 : Isibaar 3 return XVID_ERR_MEMORY;
157 :     }
158 :    
159 :     // init macroblock array
160 :     for (i = 0; i < pEnc->mbParam.mb_width * pEnc->mbParam.mb_height; i++)
161 :     {
162 :     pEnc->pMBs[i].dquant = NO_CHANGE;
163 :     }
164 :    
165 : edgomez 13 pParam->handle = (void *)pEnc;
166 : Isibaar 3
167 :     if (pParam->bitrate)
168 :     {
169 : h 64 RateControlInit(pParam->bitrate, pParam->rc_buffersize, pParam->fbase * 100 / pParam->fincr,
170 :     pParam->max_quantizer, pParam->min_quantizer);
171 : Isibaar 3 }
172 :    
173 : Isibaar 36 init_timer();
174 : Isibaar 3
175 :     return XVID_ERR_OK;
176 :     }
177 :    
178 :    
179 :     int encoder_destroy(Encoder * pEnc)
180 :     {
181 : edgomez 13 ENC_CHECK(pEnc);
182 :     ENC_CHECK(pEnc->sCurrent.y);
183 :     ENC_CHECK(pEnc->sReference.y);
184 : Isibaar 3
185 : Isibaar 41 xvid_free(pEnc->pMBs);
186 : edgomez 13 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
187 :     image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
188 :     image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
189 :     image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
190 : h 87 image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
191 : edgomez 13 image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
192 : h 87 image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
193 : Isibaar 41 xvid_free(pEnc);
194 : Isibaar 3
195 : edgomez 13 return XVID_ERR_OK;
196 : Isibaar 3 }
197 :    
198 :     int encoder_encode(Encoder * pEnc, XVID_ENC_FRAME * pFrame, XVID_ENC_STATS * pResult)
199 :     {
200 : edgomez 13 uint16_t x, y;
201 :     Bitstream bs;
202 :     uint32_t bits;
203 : Isibaar 4 uint16_t write_vol_header = 0;
204 : Isibaar 3
205 :     start_global_timer();
206 :    
207 : edgomez 13 ENC_CHECK(pEnc);
208 :     ENC_CHECK(pFrame);
209 :     ENC_CHECK(pFrame->bitstream);
210 :     ENC_CHECK(pFrame->image);
211 : Isibaar 3
212 :     pEnc->mbParam.global_flags = pFrame->general;
213 :     pEnc->mbParam.motion_flags = pFrame->motion;
214 : h 101 pEnc->mbParam.hint = &pFrame->hint;
215 : Isibaar 3
216 :     start_timer();
217 :     if (image_input(&pEnc->sCurrent, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width,
218 : edgomez 13 pFrame->image, pFrame->colorspace))
219 : Isibaar 3 {
220 :     return XVID_ERR_FORMAT;
221 :     }
222 :     stop_conv_timer();
223 :    
224 : Isibaar 41 EMMS();
225 :    
226 : edgomez 13 BitstreamInit(&bs, pFrame->bitstream, 0);
227 : Isibaar 3
228 :     if (pFrame->quant == 0)
229 :     {
230 :     pEnc->mbParam.quant = RateControlGetQ(0);
231 :     }
232 :     else
233 :     {
234 :     pEnc->mbParam.quant = pFrame->quant;
235 :     }
236 :    
237 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)
238 :     {
239 : canard 66 int * temp_dquants = (int *) xvid_malloc(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * sizeof(int), CACHE_LINE);
240 : Isibaar 3
241 : edgomez 78 pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y,
242 :     pEnc->mbParam.width,
243 :     temp_dquants,
244 : h 84 pEnc->mbParam.quant,
245 :     pEnc->mbParam.quant,
246 :     2*pEnc->mbParam.quant,
247 : edgomez 78 pEnc->mbParam.mb_width,
248 :     pEnc->mbParam.mb_height);
249 : Isibaar 3
250 :     for (y = 0; y < pEnc->mbParam.mb_height; y++)
251 :     for (x = 0; x < pEnc->mbParam.mb_width; x++)
252 :     {
253 :     MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
254 :     pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];
255 :     }
256 : Isibaar 41 xvid_free(temp_dquants);
257 : Isibaar 3 }
258 :    
259 : Isibaar 20 if(pEnc->mbParam.global_flags & XVID_H263QUANT) {
260 :     if(pEnc->mbParam.quant_type != H263_QUANT)
261 :     write_vol_header = 1;
262 :     pEnc->mbParam.quant_type = H263_QUANT;
263 : Isibaar 3 }
264 : Isibaar 20 else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {
265 :     int ret1, ret2;
266 : Isibaar 3
267 : edgomez 78 ret1 = ret2 = 0;
268 :    
269 : Isibaar 20 if(pEnc->mbParam.quant_type != MPEG4_QUANT)
270 :     write_vol_header = 1;
271 :    
272 :     pEnc->mbParam.quant_type = MPEG4_QUANT;
273 :    
274 :     if ((pEnc->mbParam.global_flags & XVID_CUSTOM_QMATRIX) > 0) {
275 :     if(pFrame->quant_intra_matrix != NULL)
276 :     ret1 = set_intra_matrix(pFrame->quant_intra_matrix);
277 :     if(pFrame->quant_inter_matrix != NULL)
278 :     ret2 = set_inter_matrix(pFrame->quant_inter_matrix);
279 :     }
280 :     else {
281 :     ret1 = set_intra_matrix(get_default_intra_matrix());
282 :     ret2 = set_inter_matrix(get_default_inter_matrix());
283 :     }
284 : Isibaar 4 if(write_vol_header == 0)
285 :     write_vol_header = ret1 | ret2;
286 :     }
287 : Isibaar 3
288 :     if (pFrame->intra < 0)
289 : edgomez 13 {
290 : Isibaar 3 if ((pEnc->iFrameNum == 0) || ((pEnc->iMaxKeyInterval > 0)
291 : edgomez 13 && (pEnc->iFrameNum >= pEnc->iMaxKeyInterval)))
292 : Isibaar 3
293 :     pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
294 :     else
295 : Isibaar 4 pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 0, write_vol_header);
296 : edgomez 13 }
297 :     else
298 :     {
299 : Isibaar 3 if (pFrame->intra == 1)
300 : edgomez 13 pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
301 : Isibaar 3 else
302 : Isibaar 4 pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 1, write_vol_header);
303 : edgomez 13 }
304 : Isibaar 3
305 :     BitstreamPutBits(&bs, 0xFFFF, 16);
306 : edgomez 13 BitstreamPutBits(&bs, 0xFFFF, 16);
307 :     BitstreamPad(&bs);
308 :     pFrame->length = BitstreamLength(&bs);
309 : h 29
310 : Isibaar 3 if (pResult)
311 : edgomez 13 {
312 : Isibaar 3 pResult->quant = pEnc->mbParam.quant;
313 :     pResult->hlength = pFrame->length - (pEnc->sStat.iTextBits / 8);
314 :     pResult->kblks = pEnc->sStat.kblks;
315 :     pResult->mblks = pEnc->sStat.mblks;
316 :     pResult->ublks = pEnc->sStat.ublks;
317 : edgomez 13 }
318 : Isibaar 3
319 : Isibaar 41 EMMS();
320 :    
321 : h 29 if (pFrame->quant == 0)
322 : Isibaar 3 {
323 :     RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);
324 :     }
325 :    
326 :     pEnc->iFrameNum++;
327 : edgomez 13 image_swap(&pEnc->sCurrent, &pEnc->sReference);
328 : Isibaar 3
329 :     stop_global_timer();
330 :     write_timer();
331 :    
332 :     return XVID_ERR_OK;
333 :     }
334 :    
335 :    
336 :     static __inline void CodeIntraMB(Encoder *pEnc, MACROBLOCK *pMB) {
337 :    
338 :     pMB->mode = MODE_INTRA;
339 :    
340 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0) {
341 :     if(pMB->dquant != NO_CHANGE)
342 :     {
343 :     pMB->mode = MODE_INTRA_Q;
344 :     pEnc->mbParam.quant += DQtab[pMB->dquant];
345 :    
346 :     if (pEnc->mbParam.quant > 31) pEnc->mbParam.quant = 31;
347 :     if (pEnc->mbParam.quant < 1) pEnc->mbParam.quant = 1;
348 :     }
349 :     }
350 :    
351 :     pMB->quant = pEnc->mbParam.quant;
352 :     }
353 :    
354 :    
355 :     static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
356 :     {
357 : edgomez 78
358 :     DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
359 :     DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
360 :    
361 : edgomez 13 uint16_t x, y;
362 : Isibaar 3
363 : edgomez 13 pEnc->iFrameNum = 0;
364 :     pEnc->mbParam.rounding_type = 1;
365 :     pEnc->mbParam.coding_type = I_VOP;
366 : Isibaar 3
367 : h 69 BitstreamWriteVolHeader(bs, &pEnc->mbParam);
368 :     BitstreamWriteVopHeader(bs, &pEnc->mbParam);
369 : Isibaar 3
370 : edgomez 13 *pBits = BitstreamPos(bs);
371 : Isibaar 3
372 :     pEnc->sStat.iTextBits = 0;
373 :     pEnc->sStat.kblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;
374 :     pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
375 :    
376 : edgomez 13 for (y = 0; y < pEnc->mbParam.mb_height; y++)
377 : Isibaar 3 for (x = 0; x < pEnc->mbParam.mb_width; x++)
378 :     {
379 : edgomez 13 MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
380 : Isibaar 3
381 :     CodeIntraMB(pEnc, pMB);
382 :    
383 : h 69 MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, &pEnc->sCurrent);
384 : Isibaar 3
385 :     start_timer();
386 :     MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
387 :     stop_prediction_timer();
388 :    
389 :     start_timer();
390 :     MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);
391 :     stop_coding_timer();
392 :     }
393 :    
394 :     emms();
395 :    
396 : edgomez 13 *pBits = BitstreamPos(bs) - *pBits;
397 :     pEnc->sStat.fMvPrevSigma = -1;
398 :     pEnc->sStat.iMvSum = 0;
399 :     pEnc->sStat.iMvCount = 0;
400 :     pEnc->mbParam.fixed_code = 2;
401 : Isibaar 3
402 : edgomez 13 return 1; // intra
403 : Isibaar 3 }
404 :    
405 :    
406 : h 101 #define FCODEBITS 3
407 :     #define MODEBITS 5
408 :    
409 :     void HintedMESet(Encoder * pEnc, int * intra)
410 :     {
411 :     HINTINFO * hint;
412 :     Bitstream bs;
413 :     int length, high;
414 :     uint32_t x, y;
415 :    
416 :     hint = pEnc->mbParam.hint;
417 :    
418 :     if (hint->rawhints)
419 :     {
420 :     *intra = hint->mvhint.intra;
421 :     }
422 :     else
423 :     {
424 :     BitstreamInit(&bs, hint->hintstream, hint->hintlength);
425 :     *intra = BitstreamGetBit(&bs);
426 :     }
427 :    
428 :     if (*intra)
429 :     {
430 :     return;
431 :     }
432 :    
433 :     pEnc->mbParam.fixed_code = (hint->rawhints) ? hint->mvhint.fcode : BitstreamGetBits(&bs, FCODEBITS);
434 :    
435 :     length = pEnc->mbParam.fixed_code + 5;
436 :     high = 1 << (length - 1);
437 :    
438 :     for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
439 :     {
440 :     for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
441 :     {
442 :     MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
443 :     MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
444 :     VECTOR pred[4];
445 :     VECTOR tmp;
446 :     int dummy[4];
447 :     int vec;
448 :    
449 :     pMB->mode = (hint->rawhints) ? bhint->mode : BitstreamGetBits(&bs, MODEBITS);
450 :    
451 :     if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
452 :     {
453 :     tmp.x = (hint->rawhints) ? bhint->mvs[0].x : BitstreamGetBits(&bs, length);
454 :     tmp.y = (hint->rawhints) ? bhint->mvs[0].y : BitstreamGetBits(&bs, length);
455 :     tmp.x -= (tmp.x >= high) ? high*2 : 0;
456 :     tmp.y -= (tmp.y >= high) ? high*2 : 0;
457 :    
458 :     get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, 0, pred, dummy);
459 :    
460 :     for (vec=0 ; vec<4 ; ++vec)
461 :     {
462 :     pMB->mvs[vec].x = tmp.x;
463 :     pMB->mvs[vec].y = tmp.y;
464 :     pMB->pmvs[vec].x = pMB->mvs[0].x - pred[0].x;
465 :     pMB->pmvs[vec].y = pMB->mvs[0].y - pred[0].y;
466 :     }
467 :     }
468 :     else if (pMB->mode == MODE_INTER4V)
469 :     {
470 :     for (vec=0 ; vec<4 ; ++vec)
471 :     {
472 :     tmp.x = (hint->rawhints) ? bhint->mvs[vec].x : BitstreamGetBits(&bs, length);
473 :     tmp.y = (hint->rawhints) ? bhint->mvs[vec].y : BitstreamGetBits(&bs, length);
474 :     tmp.x -= (tmp.x >= high) ? high*2 : 0;
475 :     tmp.y -= (tmp.y >= high) ? high*2 : 0;
476 :    
477 :     get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, vec, pred, dummy);
478 :    
479 :     pMB->mvs[vec].x = tmp.x;
480 :     pMB->mvs[vec].y = tmp.y;
481 :     pMB->pmvs[vec].x = pMB->mvs[vec].x - pred[0].x;
482 :     pMB->pmvs[vec].y = pMB->mvs[vec].y - pred[0].y;
483 :     }
484 :     }
485 :     else // intra / intra_q / stuffing / not_coded
486 :     {
487 :     for (vec=0 ; vec<4 ; ++vec)
488 :     {
489 :     pMB->mvs[vec].x = pMB->mvs[vec].y = 0;
490 :     }
491 :     }
492 :     }
493 :     }
494 :     }
495 :    
496 :    
497 :     void HintedMEGet(Encoder * pEnc, int intra)
498 :     {
499 :     HINTINFO * hint;
500 :     Bitstream bs;
501 :     uint32_t x, y;
502 :     int length, high;
503 :    
504 :     hint = pEnc->mbParam.hint;
505 :    
506 :     if (hint->rawhints)
507 :     {
508 :     hint->mvhint.intra = intra;
509 :     }
510 :     else
511 :     {
512 :     BitstreamInit(&bs, hint->hintstream, 0);
513 :     BitstreamPutBit(&bs, intra);
514 :     }
515 :    
516 :     if (intra)
517 :     {
518 :     if (!hint->rawhints)
519 :     {
520 :     BitstreamPad(&bs);
521 :     hint->hintlength = BitstreamLength(&bs);
522 :     }
523 :     return;
524 :     }
525 :    
526 :     length = pEnc->mbParam.fixed_code + 5;
527 :     high = 1 << (length - 1);
528 :    
529 :     if (hint->rawhints)
530 :     {
531 :     hint->mvhint.fcode = pEnc->mbParam.fixed_code;
532 :     }
533 :     else
534 :     {
535 :     BitstreamPutBits(&bs, pEnc->mbParam.fixed_code, FCODEBITS);
536 :     }
537 :    
538 :     for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
539 :     {
540 :     for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
541 :     {
542 :     MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
543 :     MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
544 :     VECTOR tmp;
545 :    
546 :     if (hint->rawhints)
547 :     {
548 :     bhint->mode = pMB->mode;
549 :     }
550 :     else
551 :     {
552 :     BitstreamPutBits(&bs, pMB->mode, MODEBITS);
553 :     }
554 :    
555 :     if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
556 :     {
557 :     tmp.x = pMB->mvs[0].x;
558 :     tmp.y = pMB->mvs[0].y;
559 :     tmp.x += (tmp.x < 0) ? high*2 : 0;
560 :     tmp.y += (tmp.y < 0) ? high*2 : 0;
561 :    
562 :     if (hint->rawhints)
563 :     {
564 :     bhint->mvs[0].x = tmp.x;
565 :     bhint->mvs[0].y = tmp.y;
566 :     }
567 :     else
568 :     {
569 :     BitstreamPutBits(&bs, tmp.x, length);
570 :     BitstreamPutBits(&bs, tmp.y, length);
571 :     }
572 :     }
573 :     else if (pMB->mode == MODE_INTER4V)
574 :     {
575 :     int vec;
576 :    
577 :     for (vec=0 ; vec<4 ; ++vec)
578 :     {
579 :     tmp.x = pMB->mvs[vec].x;
580 :     tmp.y = pMB->mvs[vec].y;
581 :     tmp.x += (tmp.x < 0) ? high*2 : 0;
582 :     tmp.y += (tmp.y < 0) ? high*2 : 0;
583 :    
584 :     if (hint->rawhints)
585 :     {
586 :     bhint->mvs[vec].x = tmp.x;
587 :     bhint->mvs[vec].y = tmp.y;
588 :     }
589 :     else
590 :     {
591 :     BitstreamPutBits(&bs, tmp.x, length);
592 :     BitstreamPutBits(&bs, tmp.y, length);
593 :     }
594 :     }
595 :     }
596 :     }
597 :     }
598 :    
599 :     if (!hint->rawhints)
600 :     {
601 :     BitstreamPad(&bs);
602 :     hint->hintlength = BitstreamLength(&bs);
603 :     }
604 :     }
605 :    
606 :    
607 : Isibaar 3 #define INTRA_THRESHOLD 0.5
608 :    
609 :     static int FrameCodeP(Encoder * pEnc, Bitstream * bs, uint32_t *pBits, bool force_inter, bool vol_header)
610 :     {
611 : edgomez 13 float fSigma;
612 : Isibaar 42
613 : edgomez 78 DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
614 :     DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
615 :    
616 : Isibaar 3 int iLimit;
617 : edgomez 13 uint32_t x, y;
618 :     int iSearchRange;
619 : Isibaar 3 bool bIntra;
620 :    
621 : edgomez 13 IMAGE *pCurrent = &pEnc->sCurrent;
622 :     IMAGE *pRef = &pEnc->sReference;
623 : Isibaar 3
624 : h 69 start_timer();
625 : edgomez 78 image_setedges(pRef,
626 : h 85 pEnc->mbParam.edged_width,
627 :     pEnc->mbParam.edged_height,
628 :     pEnc->mbParam.width,
629 :     pEnc->mbParam.height,
630 :     pEnc->mbParam.global_flags & XVID_INTERLACING);
631 : h 69 stop_edges_timer();
632 : Isibaar 3
633 : edgomez 13 pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;
634 : Isibaar 3
635 :     if (!force_inter)
636 :     iLimit = (int)(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * INTRA_THRESHOLD);
637 : edgomez 13 else
638 : Isibaar 3 iLimit = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height + 1;
639 :    
640 : h 85 if ((pEnc->mbParam.global_flags & XVID_HALFPEL) > 0) {
641 : Isibaar 3 start_timer();
642 : h 85 image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
643 :     pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
644 :     pEnc->mbParam.rounding_type);
645 : Isibaar 3 stop_inter_timer();
646 :     }
647 :    
648 :     start_timer();
649 : h 101 if (pEnc->mbParam.global_flags & XVID_HINTEDME_SET)
650 :     {
651 :     HintedMESet(pEnc, &bIntra);
652 :     }
653 :     else
654 :     {
655 :     bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,
656 :     &pEnc->vInterH, &pEnc->vInterV,
657 :     &pEnc->vInterHV, &pEnc->sCurrent, iLimit);
658 :     }
659 : Isibaar 3 stop_motion_timer();
660 :    
661 :     if (bIntra == 1)
662 : h 101 {
663 :     if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
664 :     {
665 :     HintedMEGet(pEnc, 1);
666 :     }
667 : Isibaar 3 return FrameCodeI(pEnc, bs, pBits);
668 : h 101 }
669 : Isibaar 3
670 : edgomez 13 pEnc->mbParam.coding_type = P_VOP;
671 : Isibaar 3
672 :     if(vol_header)
673 : h 69 BitstreamWriteVolHeader(bs, &pEnc->mbParam);
674 : Isibaar 3
675 : h 69 BitstreamWriteVopHeader(bs, &pEnc->mbParam);
676 : Isibaar 3
677 : edgomez 13 *pBits = BitstreamPos(bs);
678 : Isibaar 3
679 : edgomez 13 pEnc->sStat.iTextBits = 0;
680 :     pEnc->sStat.iMvSum = 0;
681 :     pEnc->sStat.iMvCount = 0;
682 : Isibaar 3 pEnc->sStat.kblks = pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
683 :    
684 : edgomez 13 for(y = 0; y < pEnc->mbParam.mb_height; y++)
685 : Isibaar 3 {
686 :     for(x = 0; x < pEnc->mbParam.mb_width; x++)
687 :     {
688 :     MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
689 :    
690 : edgomez 13 bIntra = (pMB->mode == MODE_INTRA) || (pMB->mode == MODE_INTRA_Q);
691 : Isibaar 3
692 :     if (!bIntra)
693 : edgomez 13 {
694 : Isibaar 3 start_timer();
695 : edgomez 78 MBMotionCompensation(pMB,
696 :     x, y,
697 :     &pEnc->sReference,
698 :     &pEnc->vInterH,
699 : h 85 &pEnc->vInterV,
700 :     &pEnc->vInterHV,
701 : edgomez 78 &pEnc->sCurrent,
702 :     dct_codes,
703 : edgomez 13 pEnc->mbParam.width,
704 :     pEnc->mbParam.height,
705 :     pEnc->mbParam.edged_width,
706 :     pEnc->mbParam.rounding_type);
707 : Isibaar 3 stop_comp_timer();
708 :    
709 :     if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0) {
710 :     if(pMB->dquant != NO_CHANGE) {
711 :     pMB->mode = MODE_INTER_Q;
712 :     pEnc->mbParam.quant += DQtab[pMB->dquant];
713 :     if (pEnc->mbParam.quant > 31) pEnc->mbParam.quant = 31;
714 :     else if(pEnc->mbParam.quant < 1) pEnc->mbParam.quant = 1;
715 :     }
716 :     }
717 :     pMB->quant = pEnc->mbParam.quant;
718 :    
719 : h 69 pMB->field_pred = 0;
720 :    
721 :     pMB->cbp = MBTransQuantInter(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
722 : edgomez 13 }
723 : Isibaar 3 else
724 :     {
725 :     CodeIntraMB(pEnc, pMB);
726 : h 69 MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
727 : Isibaar 3 }
728 :    
729 : edgomez 13 start_timer();
730 : Isibaar 3 MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
731 :     stop_prediction_timer();
732 :    
733 :     if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q)
734 :     {
735 :     pEnc->sStat.kblks++;
736 :     }
737 :     else if (pMB->cbp ||
738 : edgomez 13 pMB->mvs[0].x || pMB->mvs[0].y ||
739 :     pMB->mvs[1].x || pMB->mvs[1].y ||
740 :     pMB->mvs[2].x || pMB->mvs[2].y ||
741 :     pMB->mvs[3].x || pMB->mvs[3].y)
742 : Isibaar 3 {
743 :     pEnc->sStat.mblks++;
744 :     }
745 :     else
746 :     {
747 :     pEnc->sStat.ublks++;
748 :     }
749 :    
750 :     start_timer();
751 :     MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);
752 :     stop_coding_timer();
753 :     }
754 :     }
755 :    
756 :     emms();
757 :    
758 : h 101 if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
759 :     {
760 :     HintedMEGet(pEnc, 0);
761 :     }
762 :    
763 : Isibaar 3 if (pEnc->sStat.iMvCount == 0)
764 :     pEnc->sStat.iMvCount = 1;
765 :    
766 : edgomez 13 fSigma = (float)sqrt((float) pEnc->sStat.iMvSum / pEnc->sStat.iMvCount);
767 : Isibaar 3
768 : edgomez 13 iSearchRange = 1 << (3 + pEnc->mbParam.fixed_code);
769 : Isibaar 3
770 : edgomez 13 if ((fSigma > iSearchRange / 3)
771 :     && (pEnc->mbParam.fixed_code <= 3)) // maximum search range 128
772 :     {
773 : Isibaar 3 pEnc->mbParam.fixed_code++;
774 :     iSearchRange *= 2;
775 : edgomez 13 }
776 :     else if ((fSigma < iSearchRange / 6)
777 :     && (pEnc->sStat.fMvPrevSigma >= 0)
778 :     && (pEnc->sStat.fMvPrevSigma < iSearchRange / 6)
779 :     && (pEnc->mbParam.fixed_code >= 2)) // minimum search range 16
780 :     {
781 : Isibaar 3 pEnc->mbParam.fixed_code--;
782 :     iSearchRange /= 2;
783 : edgomez 13 }
784 : Isibaar 3
785 : edgomez 13 pEnc->sStat.fMvPrevSigma = fSigma;
786 : Isibaar 3
787 :     *pBits = BitstreamPos(bs) - *pBits;
788 :    
789 : edgomez 13 return 0; // inter
790 : Isibaar 3 }

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