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

Diff of /xvidcore/src/utils/mbtransquant.c

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

revision 1.7, Wed Jun 12 20:38:41 2002 UTC revision 1.16, Wed Oct 9 22:51:42 2002 UTC
# Line 1  Line 1 
1   /******************************************************************************  /*****************************************************************************
2    *                                                                            *   *
3    *  This file is part of XviD, a free MPEG-4 video encoder/decoder            *   *  XVID MPEG-4 VIDEO CODEC
4    *                                                                            *   *  - MacroBlock transfer and quantization -
5    *  XviD is an implementation of a part of one or more MPEG-4 Video tools     *   *
6    *  as specified in ISO/IEC 14496-2 standard.  Those intending to use this    *   *  Copyright(C) 2002-2001 Michael Militzer <isibaar@xvid.org>
7    *  software module in hardware or software products are advised that its     *   *               2002-2001 Peter Ross <pross@xvid.org>
8    *  use may infringe existing patents or copyrights, and any such use         *   *
9    *  would be at such party's own risk.  The original developer of this        *   *  This program is an implementation of a part of one or more MPEG-4
10    *  software module and his/her company, and subsequent editors and their     *   *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
11    *  companies, will have no liability for use of this software or             *   *  to use this software module in hardware or software products are
12    *  modifications or derivatives thereof.                                     *   *  advised that its use may infringe existing patents or copyrights, and
13    *                                                                            *   *  any such use would be at such party's own risk.  The original
14    *  XviD is free software; you can redistribute it and/or modify it           *   *  developer of this software module and his/her company, and subsequent
15    *  under the terms of the GNU General Public License as published by         *   *  editors and their companies, will have no liability for use of this
16    *  the Free Software Foundation; either version 2 of the License, or         *   *  software or modifications or derivatives thereof.
17    *  (at your option) any later version.                                       *   *
18    *                                                                            *   *  This program is free software; you can redistribute it and/or modify
19    *  XviD is distributed in the hope that it will be useful, but               *   *  it under the terms of the GNU General Public License as published by
20    *  WITHOUT ANY WARRANTY; without even the implied warranty of                *   *  the Free Software Foundation; either version 2 of the License, or
21    *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the             *   *  (at your option) any later version.
22    *  GNU General Public License for more details.                              *   *
23    *                                                                            *   *  This program is distributed in the hope that it will be useful,
24    *  You should have received a copy of the GNU General Public License         *   *  but WITHOUT ANY WARRANTY; without even the implied warranty of
25    *  along with this program; if not, write to the Free Software               *   *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
26    *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA  *   *  GNU General Public License for more details.
27    *                                                                            *   *
28    ******************************************************************************/   *  You should have received a copy of the GNU General Public License
29     *  along with this program; if not, write to the Free Software
30   /******************************************************************************   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
31    *                                                                            *   *
32    *  mbtransquant.c                                                            *   * $Id$
33    *                                                                            *   *
34    *  Copyright (C) 2001 - Peter Ross <pross@cs.rmit.edu.au>                    *   ****************************************************************************/
   *  Copyright (C) 2001 - Michael Militzer <isibaar@xvid.org>                  *  
   *                                                                            *  
   *  For more information visit the XviD homepage: http://www.xvid.org         *  
   *                                                                            *  
   ******************************************************************************/  
   
  /******************************************************************************  
   *                                                                            *  
   *  Revision history:                                                         *  
   *                                                                            *  
   *  29.03.2002 interlacing speedup - used transfer strides instead of  
   *             manual field-to-frame conversion  
   *  26.03.2002 interlacing support - moved transfers outside loops  
   *  22.12.2001 get_dc_scaler() moved to common.h  
   *  19.11.2001 introduced coefficient thresholding (Isibaar)                  *  
   *  17.11.2001 initial version                                                *  
   *                                                                            *  
   ******************************************************************************/  
35    
36  #include <string.h>  #include <string.h>
37    
# Line 68  Line 50 
50  #define MIN(X, Y) ((X)<(Y)?(X):(Y))  #define MIN(X, Y) ((X)<(Y)?(X):(Y))
51  #define MAX(X, Y) ((X)>(Y)?(X):(Y))  #define MAX(X, Y) ((X)>(Y)?(X):(Y))
52    
53  #define TOOSMALL_LIMIT 1                /* skip blocks having a coefficient sum below this value */  #define TOOSMALL_LIMIT 3                /* skip blocks having a coefficient sum below this value */
54    
55  /* this isnt pretty, but its better than 20 ifdefs */  /* this isnt pretty, but its better than 20 ifdefs */
56    
# Line 105  Line 87 
87    
88          start_timer();          start_timer();
89          pMB->field_dct = 0;          pMB->field_dct = 0;
90          if ((frame->global_flags & XVID_INTERLACING)) {          if ((frame->global_flags & XVID_INTERLACING) &&
91                    (x_pos>0) && (x_pos<pParam->mb_width-1) &&
92                    (y_pos>0) && (y_pos<pParam->mb_height-1)) {
93                  pMB->field_dct = MBDecideFieldDCT(data);                  pMB->field_dct = MBDecideFieldDCT(data);
94          }          }
95          stop_interlacing_timer();          stop_interlacing_timer();
# Line 183  Line 167 
167    
168          start_timer();          start_timer();
169          pMB->field_dct = 0;          pMB->field_dct = 0;
170          if ((frame->global_flags & XVID_INTERLACING)) {          if ((frame->global_flags & XVID_INTERLACING) &&
171                    (x_pos>0) && (x_pos<pParam->mb_width-1) &&
172                    (y_pos>0) && (y_pos<pParam->mb_height-1)) {
173                  pMB->field_dct = MBDecideFieldDCT(data);                  pMB->field_dct = MBDecideFieldDCT(data);
174          }          }
175          stop_interlacing_timer();          stop_interlacing_timer();
# Line 207  Line 193 
193                          stop_quant_timer();                          stop_quant_timer();
194                  }                  }
195    
196                  if (sum >= TOOSMALL_LIMIT) {    // skip block ?                  if ((sum >= TOOSMALL_LIMIT) || (qcoeff[i*64] != 0) ||
197                            (qcoeff[i*64+1] != 0) || (qcoeff[i*64+8] != 0)) {
198    
199                          if (pParam->m_quant_type == H263_QUANT) {                          if (pParam->m_quant_type == H263_QUANT) {
200                                  start_timer();                                  start_timer();
# Line 251  Line 238 
238    
239  }  }
240    
241    void
242    MBTransQuantIntra2(const MBParam * pParam,
243                                      FRAMEINFO * frame,
244                                      MACROBLOCK * pMB,
245                                      const uint32_t x_pos,
246                                      const uint32_t y_pos,
247                                      int16_t data[6 * 64],
248                                      int16_t qcoeff[6 * 64])
249    {
250            MBTrans(pParam,frame,pMB,x_pos,y_pos,data);
251            MBfDCT(pParam,frame,pMB,data);
252            MBQuantIntra(pParam,frame,pMB,data,qcoeff);
253            MBDeQuantIntra(pParam,frame->quant,data,qcoeff);
254            MBiDCT(data,0x3F);
255            MBTransAdd(pParam,frame,pMB,x_pos,y_pos,data,0x3F);
256    }
257    
258    
259    uint8_t
260    MBTransQuantInter2(const MBParam * pParam,
261                                      FRAMEINFO * frame,
262                                      MACROBLOCK * pMB,
263                                      const uint32_t x_pos,
264                                      const uint32_t y_pos,
265                                      int16_t data[6 * 64],
266                                      int16_t qcoeff[6 * 64])
267    {
268            uint8_t cbp;
269    
270    /* there is no MBTrans for Inter block, that's done in motion compensation already */
271    
272            MBfDCT(pParam,frame,pMB,data);
273            cbp = MBQuantInter(pParam,frame->quant,data,qcoeff);
274            MBDeQuantInter(pParam,frame->quant,data,qcoeff,cbp);
275            MBiDCT(data,cbp);
276            MBTransAdd(pParam,frame,pMB,x_pos,y_pos,data,cbp);
277    
278            return cbp;
279    }
280    
281    uint8_t
282    MBTransQuantInterBVOP(const MBParam * pParam,
283                                      FRAMEINFO * frame,
284                                      MACROBLOCK * pMB,
285                                      int16_t data[6 * 64],
286                                      int16_t qcoeff[6 * 64])
287    {
288            uint8_t cbp;
289    
290    /* there is no MBTrans for Inter block, that's done in motion compensation already */
291    
292            MBfDCT(pParam,frame,pMB,data);
293            cbp = MBQuantInter(pParam,frame->quant,data,qcoeff);
294    
295    /* we don't have to DeQuant, iDCT and Transfer back data for B-frames */
296    
297            return cbp;
298    }
299    
300    
301    void
302    MBfDCT(const MBParam * pParam,
303                                      FRAMEINFO * frame,
304                                      MACROBLOCK * pMB,
305                                      int16_t data[6 * 64])
306    {
307            int i;
308    
309            start_timer();
310            pMB->field_dct = 0;
311            if ((frame->global_flags & XVID_INTERLACING)) {
312                    pMB->field_dct = MBDecideFieldDCT(data);
313            }
314            stop_interlacing_timer();
315    
316            for (i = 0; i < 6; i++) {
317                    start_timer();
318                    fdct(&data[i * 64]);
319                    stop_dct_timer();
320            }
321    }
322    
323    void
324    MBQuantDeQuantIntra(const MBParam * pParam,
325                                            FRAMEINFO * frame,
326                                            MACROBLOCK * pMB,
327                                            int16_t qcoeff[6 * 64],
328                                            int16_t data[6*64])
329    {
330            int i;
331            int iQuant = frame->quant;
332    
333            start_timer();
334            pMB->field_dct = 0;
335            if ((frame->global_flags & XVID_INTERLACING)) {
336                    pMB->field_dct = MBDecideFieldDCT(data);
337            }
338            stop_interlacing_timer();
339    
340            for (i = 0; i < 6; i++) {
341                    uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
342    
343                    if (pParam->m_quant_type == H263_QUANT) {
344                            start_timer();
345                            quant_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);
346                            stop_quant_timer();
347    
348                            start_timer();
349                            dequant_intra(&data[i * 64], &qcoeff[i * 64], iQuant, iDcScaler);
350                            stop_iquant_timer();
351                    } else {
352                            start_timer();
353                            quant4_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);
354                            stop_quant_timer();
355    
356                            start_timer();
357                            dequant4_intra(&data[i * 64], &qcoeff[i * 64], iQuant, iDcScaler);
358                            stop_iquant_timer();
359                    }
360            }
361    }
362    
363    void
364    MBQuantIntra(const MBParam * pParam,
365                             FRAMEINFO * frame,
366                             MACROBLOCK *pMB,
367                             int16_t data[6 * 64],
368                         int16_t qcoeff[6 * 64])
369    {
370            int i;
371            int iQuant = frame->quant;
372    
373            start_timer();
374            pMB->field_dct = 0;
375            if ((frame->global_flags & XVID_INTERLACING)) {
376                    pMB->field_dct = MBDecideFieldDCT(data);
377            }
378            stop_interlacing_timer();
379    
380            for (i = 0; i < 6; i++) {
381                    uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
382    
383                    if (pParam->m_quant_type == H263_QUANT) {
384                            start_timer();
385                            quant_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);
386                            stop_quant_timer();
387                    } else {
388                            start_timer();
389                            quant4_intra(&qcoeff[i * 64], &data[i * 64], iQuant, iDcScaler);
390                            stop_quant_timer();
391                    }
392            }
393    }
394    
395    void
396    MBDeQuantIntra(const MBParam * pParam,
397                               const int iQuant,
398                                      int16_t qcoeff[6 * 64],
399                                      int16_t data[6*64])
400    {
401            int i;
402    
403            for (i = 0; i < 6; i++) {
404                    uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
405    
406                    if (pParam->m_quant_type == H263_QUANT) {
407                            start_timer();
408                            dequant_intra(&data[i * 64], &qcoeff[i * 64], iQuant, iDcScaler);
409                            stop_iquant_timer();
410                    } else {
411                            start_timer();
412                            dequant4_intra(&data[i * 64], &qcoeff[i * 64], iQuant, iDcScaler);
413                            stop_iquant_timer();
414                    }
415            }
416    }
417    
418    uint8_t
419    MBQuantInter(const MBParam * pParam,
420                             const int iQuant,
421                                      int16_t data[6 * 64],
422                                      int16_t qcoeff[6 * 64])
423    {
424    
425            int i;
426            uint8_t cbp = 0;
427            int sum;
428    
429            for (i = 0; i < 6; i++) {
430    
431                    if (pParam->m_quant_type == 0) {
432                            start_timer();
433                            sum = quant_inter(&qcoeff[i * 64], &data[i * 64], iQuant);
434                            stop_quant_timer();
435                    } else {
436                            start_timer();
437                            sum = quant4_inter(&qcoeff[i * 64], &data[i * 64], iQuant);
438                            stop_quant_timer();
439                    }
440    
441                    if (sum >= TOOSMALL_LIMIT) {    // skip block ?
442                            cbp |= 1 << (5 - i);
443                    }
444            }
445            return cbp;
446    }
447    
448    void
449    MBDeQuantInter( const MBParam * pParam,
450                                    const int iQuant,
451                                      int16_t data[6 * 64],
452                                      int16_t qcoeff[6 * 64],
453                                      const uint8_t cbp)
454    {
455            int i;
456    
457            for (i = 0; i < 6; i++) {
458                    if (cbp & (1 << (5 - i)))
459                    {
460                            if (pParam->m_quant_type == H263_QUANT) {
461                                    start_timer();
462                                    dequant_inter(&data[i * 64], &qcoeff[i * 64], iQuant);
463                                    stop_iquant_timer();
464                            } else {
465                                    start_timer();
466                                    dequant4_inter(&data[i * 64], &qcoeff[i * 64], iQuant);
467                                    stop_iquant_timer();
468                            }
469                    }
470            }
471    }
472    
473    void
474    MBiDCT( int16_t data[6 * 64],
475                    const uint8_t cbp)
476    {
477            int i;
478    
479            for (i = 0; i < 6; i++) {
480                    if (cbp & (1 << (5 - i)))
481                    {
482                            start_timer();
483                            idct(&data[i * 64]);
484                            stop_idct_timer();
485    
486                    }
487            }
488    }
489    
490    
491    void
492    MBTrans(const MBParam * pParam,
493                                      FRAMEINFO * frame,
494                                      MACROBLOCK * pMB,
495                                      const uint32_t x_pos,
496                                      const uint32_t y_pos,
497                                      int16_t data[6 * 64])
498    {
499            uint32_t stride = pParam->edged_width;
500            uint32_t stride2 = stride / 2;
501            uint32_t next_block = stride * 8;
502            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
503            IMAGE *pCurrent = &frame->image;
504    
505            pY_Cur = pCurrent->y + (y_pos << 4) * stride + (x_pos << 4);
506            pU_Cur = pCurrent->u + (y_pos << 3) * stride2 + (x_pos << 3);
507            pV_Cur = pCurrent->v + (y_pos << 3) * stride2 + (x_pos << 3);
508    
509            start_timer();
510            transfer_8to16copy(&data[0 * 64], pY_Cur, stride);
511            transfer_8to16copy(&data[1 * 64], pY_Cur + 8, stride);
512            transfer_8to16copy(&data[2 * 64], pY_Cur + next_block, stride);
513            transfer_8to16copy(&data[3 * 64], pY_Cur + next_block + 8, stride);
514            transfer_8to16copy(&data[4 * 64], pU_Cur, stride2);
515            transfer_8to16copy(&data[5 * 64], pV_Cur, stride2);
516            stop_transfer_timer();
517    }
518    
519    void
520    MBTransAdd(const MBParam * pParam,
521                                      FRAMEINFO * frame,
522                                      MACROBLOCK * pMB,
523                                      const uint32_t x_pos,
524                                      const uint32_t y_pos,
525                                      int16_t data[6 * 64],
526                                      const uint8_t cbp)
527    {
528            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
529            uint32_t stride = pParam->edged_width;
530            uint32_t stride2 = stride / 2;
531            uint32_t next_block = stride * 8;
532            IMAGE *pCurrent = &frame->image;
533    
534            pY_Cur = pCurrent->y + (y_pos << 4) * stride + (x_pos << 4);
535            pU_Cur = pCurrent->u + (y_pos << 3) * stride2 + (x_pos << 3);
536            pV_Cur = pCurrent->v + (y_pos << 3) * stride2 + (x_pos << 3);
537    
538            if (pMB->field_dct) {
539                    next_block = stride;
540                    stride *= 2;
541            }
542    
543            start_timer();
544            if (cbp & 32)
545                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
546            if (cbp & 16)
547                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
548            if (cbp & 8)
549                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
550            if (cbp & 4)
551                    transfer_16to8add(pY_Cur + next_block + 8, &data[3 * 64], stride);
552            if (cbp & 2)
553                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
554            if (cbp & 1)
555                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
556            stop_transfer_timer();
557    }
558    
559    
560    
561  /* if sum(diff between field lines) < sum(diff between frame lines), use field dct */  /* if sum(diff between field lines) < sum(diff between frame lines), use field dct */
562    
# Line 296  Line 602 
602                  MBFrameToField(data);                  MBFrameToField(data);
603          }          }
604    
605          return (frame > field);          return (frame > (field + 350));
606  }  }
607    
608    

Legend:
Removed from v.1.7  
changed lines
  Added in v.1.16

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