text_dct.c
上传用户:enenge
上传日期:2007-01-08
资源大小:96k
文件大小:7k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /**************************************************************************
  2.  *                                                                        *
  3.  * This code is developed by Adam Li.  This software is an                *
  4.  * implementation of a part of one or more MPEG-4 Video tools as          *
  5.  * specified in ISO/IEC 14496-2 standard.  Those intending to use this    *
  6.  * software module in hardware or software products are advised that its  *
  7.  * use may infringe existing patents or copyrights, and any such use      *
  8.  * would be at such party's own risk.  The original developer of this     *
  9.  * software module and his/her company, and subsequent editors and their  *
  10.  * companies (including Project Mayo), will have no liability for use of  *
  11.  * this software or modifications or derivatives thereof.                 *
  12.  *                                                                        *
  13.  * Project Mayo gives users of the Codec a license to this software       *
  14.  * module or modifications thereof for use in hardware or software        *
  15.  * products claiming conformance to the MPEG-4 Video Standard as          *
  16.  * described in the Open DivX license.                                    *
  17.  *                                                                        *
  18.  * The complete Open DivX license can be found at                         *
  19.  * http://www.projectmayo.com/opendivx/license.php .                      *
  20.  *                                                                        *
  21.  **************************************************************************/
  22. /**************************************************************************
  23.  *
  24.  *  text_dct.c
  25.  *
  26.  *  Copyright (C) 2001  Project Mayo
  27.  *
  28.  *  Adam Li
  29.  *
  30.  *  DivX Advance Research Center <darc@projectmayo.com>
  31.  *
  32.  **************************************************************************/
  33. /* This file contains some functions for fDCT/iDCT transformation.        */
  34. /* Some codes of this project come from SSG MPEG-2 implementation.        */
  35. /* Please see seperate acknowledgement file for a list of contributors.   */
  36. #include <math.h>
  37. /* The first part of it is for the forward DCT */
  38. #ifndef PI
  39. # ifdef M_PI
  40. #  define PI M_PI
  41. # else
  42. #  define PI 3.14159265358979323846
  43. # endif
  44. #endif
  45. /* private data */
  46. static double c[8][8]; /* transform coefficients */
  47. void fdct_enc(block)
  48. short *block;
  49. {
  50.   int i, j, k;
  51.   double s;
  52.   double tmp[64];
  53.   for (i=0; i<8; i++)
  54.     for (j=0; j<8; j++)
  55.     {
  56.       s = 0.0;
  57.       for (k=0; k<8; k++)
  58.         s += c[j][k] * block[8*i+k];
  59.       tmp[8*i+j] = s;
  60.     }
  61.   for (j=0; j<8; j++)
  62.     for (i=0; i<8; i++)
  63.     {
  64.       s = 0.0;
  65.       for (k=0; k<8; k++)
  66.         s += c[i][k] * tmp[8*k+j];
  67.       block[8*i+j] = (int)floor(s+0.499999);
  68.       /*
  69.        * reason for adding 0.499999 instead of 0.5:
  70.        * s is quite often x.5 (at least for i and/or j = 0 or 4)
  71.        * and setting the rounding threshold exactly to 0.5 leads to an
  72.        * extremely high arithmetic implementation dependency of the result;
  73.        * s being between x.5 and x.500001 (which is now incorrectly rounded
  74.        * downwards instead of upwards) is assumed to occur less often
  75.        * (if at all)
  76.        */
  77.     }
  78. }
  79. void init_fdct_enc()
  80. {
  81.   int i, j;
  82.   double s;
  83.   for (i=0; i<8; i++)
  84.   {
  85.     s = (i==0) ? sqrt(0.125) : 0.5;
  86.     for (j=0; j<8; j++)
  87.       c[i][j] = s * cos((PI/8.0)*i*(j+0.5));
  88.   }
  89. }
  90. /* the second part of it is for the inverse DCT */
  91. #define W1 2841 /* 2048*sqrt(2)*cos(1*pi/16) */
  92. #define W2 2676 /* 2048*sqrt(2)*cos(2*pi/16) */
  93. #define W3 2408 /* 2048*sqrt(2)*cos(3*pi/16) */
  94. #define W5 1609 /* 2048*sqrt(2)*cos(5*pi/16) */
  95. #define W6 1108 /* 2048*sqrt(2)*cos(6*pi/16) */
  96. #define W7 565  /* 2048*sqrt(2)*cos(7*pi/16) */
  97. /* private data */
  98. static short iclip[1024]; /* clipping table */
  99. static short *iclp;
  100. /* private prototypes */
  101. static void idctrow_enc (short *blk);
  102. static void idctcol_enc (short *blk);
  103. /* two dimensional inverse discrete cosine transform */
  104. void idct_enc(block)
  105. short *block;
  106. {
  107.   int i;
  108.   for (i=0; i<8; i++)
  109.     idctrow_enc(block+8*i);
  110.   for (i=0; i<8; i++)
  111.     idctcol_enc(block+i);
  112. }
  113. void init_idct_enc()
  114. {
  115.   int i;
  116.   iclp = iclip+512;
  117.   for (i= -512; i<512; i++)
  118.     iclp[i] = (i<-256) ? -256 : ((i>255) ? 255 : i);
  119. }
  120. /* row (horizontal) IDCT
  121.  *
  122.  *           7                       pi         1
  123.  * dst[k] = sum c[l] * src[l] * cos( -- * ( k + - ) * l )
  124.  *          l=0                      8          2
  125.  *
  126.  * where: c[0]    = 128
  127.  *        c[1..7] = 128*sqrt(2)
  128.  */
  129. static void idctrow_enc(blk)
  130. short *blk;
  131. {
  132.   int x0, x1, x2, x3, x4, x5, x6, x7, x8;
  133.   /* shortcut */
  134.   if (!((x1 = blk[4]<<11) | (x2 = blk[6]) | (x3 = blk[2]) |
  135.         (x4 = blk[1]) | (x5 = blk[7]) | (x6 = blk[5]) | (x7 = blk[3])))
  136.   {
  137.     blk[0]=blk[1]=blk[2]=blk[3]=blk[4]=blk[5]=blk[6]=blk[7]=blk[0]<<3;
  138.     return;
  139.   }
  140.   x0 = (blk[0]<<11) + 128; /* for proper rounding in the fourth stage */
  141.   /* first stage */
  142.   x8 = W7*(x4+x5);
  143.   x4 = x8 + (W1-W7)*x4;
  144.   x5 = x8 - (W1+W7)*x5;
  145.   x8 = W3*(x6+x7);
  146.   x6 = x8 - (W3-W5)*x6;
  147.   x7 = x8 - (W3+W5)*x7;
  148.   
  149.   /* second stage */
  150.   x8 = x0 + x1;
  151.   x0 -= x1;
  152.   x1 = W6*(x3+x2);
  153.   x2 = x1 - (W2+W6)*x2;
  154.   x3 = x1 + (W2-W6)*x3;
  155.   x1 = x4 + x6;
  156.   x4 -= x6;
  157.   x6 = x5 + x7;
  158.   x5 -= x7;
  159.   
  160.   /* third stage */
  161.   x7 = x8 + x3;
  162.   x8 -= x3;
  163.   x3 = x0 + x2;
  164.   x0 -= x2;
  165.   x2 = (181*(x4+x5)+128)>>8;
  166.   x4 = (181*(x4-x5)+128)>>8;
  167.   
  168.   /* fourth stage */
  169.   blk[0] = (x7+x1)>>8;
  170.   blk[1] = (x3+x2)>>8;
  171.   blk[2] = (x0+x4)>>8;
  172.   blk[3] = (x8+x6)>>8;
  173.   blk[4] = (x8-x6)>>8;
  174.   blk[5] = (x0-x4)>>8;
  175.   blk[6] = (x3-x2)>>8;
  176.   blk[7] = (x7-x1)>>8;
  177. }
  178. /* column (vertical) IDCT
  179.  *
  180.  *             7                         pi         1
  181.  * dst[8*k] = sum c[l] * src[8*l] * cos( -- * ( k + - ) * l )
  182.  *            l=0                        8          2
  183.  *
  184.  * where: c[0]    = 1/1024
  185.  *        c[1..7] = (1/1024)*sqrt(2)
  186.  */
  187. static void idctcol_enc(blk)
  188. short *blk;
  189. {
  190.   int x0, x1, x2, x3, x4, x5, x6, x7, x8;
  191.   /* shortcut */
  192.   if (!((x1 = (blk[8*4]<<8)) | (x2 = blk[8*6]) | (x3 = blk[8*2]) |
  193.         (x4 = blk[8*1]) | (x5 = blk[8*7]) | (x6 = blk[8*5]) | (x7 = blk[8*3])))
  194.   {
  195.     blk[8*0]=blk[8*1]=blk[8*2]=blk[8*3]=blk[8*4]=blk[8*5]=blk[8*6]=blk[8*7]=
  196.       iclp[(blk[8*0]+32)>>6];
  197.     return;
  198.   }
  199.   x0 = (blk[8*0]<<8) + 8192;
  200.   /* first stage */
  201.   x8 = W7*(x4+x5) + 4;
  202.   x4 = (x8+(W1-W7)*x4)>>3;
  203.   x5 = (x8-(W1+W7)*x5)>>3;
  204.   x8 = W3*(x6+x7) + 4;
  205.   x6 = (x8-(W3-W5)*x6)>>3;
  206.   x7 = (x8-(W3+W5)*x7)>>3;
  207.   
  208.   /* second stage */
  209.   x8 = x0 + x1;
  210.   x0 -= x1;
  211.   x1 = W6*(x3+x2) + 4;
  212.   x2 = (x1-(W2+W6)*x2)>>3;
  213.   x3 = (x1+(W2-W6)*x3)>>3;
  214.   x1 = x4 + x6;
  215.   x4 -= x6;
  216.   x6 = x5 + x7;
  217.   x5 -= x7;
  218.   
  219.   /* third stage */
  220.   x7 = x8 + x3;
  221.   x8 -= x3;
  222.   x3 = x0 + x2;
  223.   x0 -= x2;
  224.   x2 = (181*(x4+x5)+128)>>8;
  225.   x4 = (181*(x4-x5)+128)>>8;
  226.   
  227.   /* fourth stage */
  228.   blk[8*0] = iclp[(x7+x1)>>14];
  229.   blk[8*1] = iclp[(x3+x2)>>14];
  230.   blk[8*2] = iclp[(x0+x4)>>14];
  231.   blk[8*3] = iclp[(x8+x6)>>14];
  232.   blk[8*4] = iclp[(x8-x6)>>14];
  233.   blk[8*5] = iclp[(x0-x4)>>14];
  234.   blk[8*6] = iclp[(x3-x2)>>14];
  235.   blk[8*7] = iclp[(x7-x1)>>14];
  236. }