Dct.cpp
上传用户:szklck
上传日期:2007-01-22
资源大小:925k
文件大小:5k
源码类别:

图形图像处理

开发平台:

Visual C++

  1. #include"stdafx.h"
  2. #include"Global.h"
  3. #include <math.h>
  4. #define PI 3.14159265358979323846
  5. int zigzag[8][8] = {
  6.   {0, 1, 5, 6,14,15,27,28},
  7.   {2, 4, 7,13,16,26,29,42},
  8.   {3, 8,12,17,25,30,41,43},
  9.   {9,11,18,24,31,40,44,53},
  10.   {10,19,23,32,39,45,52,54},
  11.   {20,22,33,38,46,51,55,60},
  12.   {21,34,37,47,50,56,59,61},
  13.   {35,36,48,49,57,58,62,63},
  14. };
  15. /**********************************************************************
  16.  *
  17.  * Name:        Dct
  18.  * Description: Does dct on an 8x8 block, does zigzag-scanning of
  19.  *        coefficients
  20.  *
  21.  * Input:        64 pixels in a 1D array
  22.  * Returns: 64 coefficients in a 1D array
  23.  * Side effects:
  24.  *
  25.  * Date: 930128 Author: Robert.Danielsen@nta.no
  26.  *
  27.  **********************************************************************/
  28. int Dct( int *block, int *coeff)
  29. {
  30.   int        j1, i, j, k;
  31.   float b[8];
  32.   float        b1[8];
  33.   float        d[8][8];
  34.   float f0=(float).7071068;
  35.   float f1=(float).4903926;
  36.   float f2=(float).4619398;
  37.   float f3=(float).4157348;
  38.   float f4=(float).3535534;
  39.   float f5=(float).2777851;
  40.   float f6=(float).1913417;
  41.   float f7=(float).0975452;
  42.   for (i = 0, k = 0; i < 8; i++, k += 8) {
  43.     for (j = 0; j < 8; j++) {
  44.       b[j] = (float)block[k+j];
  45.     }
  46.     /* Horizontal transform */
  47.     for (j = 0; j < 4; j++) {
  48.       j1 = 7 - j;
  49.       b1[j] = b[j] + b[j1];
  50.       b1[j1] = b[j] - b[j1];
  51.     }
  52.     b[0] = b1[0] + b1[3];
  53.     b[1] = b1[1] + b1[2];
  54.     b[2] = b1[1] - b1[2];
  55.     b[3] = b1[0] - b1[3];
  56.     b[4] = b1[4];
  57.     b[5] = (b1[6] - b1[5]) * f0;
  58.     b[6] = (b1[6] + b1[5]) * f0;
  59.     b[7] = b1[7];
  60.     d[i][0] = (b[0] + b[1]) * f4;
  61.     d[i][4] = (b[0] - b[1]) * f4;
  62.     d[i][2] = b[2] * f6 + b[3] * f2;
  63.     d[i][6] = b[3] * f6 - b[2] * f2;
  64.     b1[4] = b[4] + b[5];
  65.     b1[7] = b[7] + b[6];
  66.     b1[5] = b[4] - b[5];
  67.     b1[6] = b[7] - b[6];
  68.     d[i][1] = b1[4] * f7 + b1[7] * f1;
  69.     d[i][5] = b1[5] * f3 + b1[6] * f5;
  70.     d[i][7] = b1[7] * f7 - b1[4] * f1;
  71.     d[i][3] = b1[6] * f3 - b1[5] * f5;
  72.   }
  73.   /* Vertical transform */
  74.   for (i = 0; i < 8; i++) {
  75.     for (j = 0; j < 4; j++) {
  76.       j1 = 7 - j;
  77.       b1[j] = d[j][i] + d[j1][i];
  78.       b1[j1] = d[j][i] - d[j1][i];
  79.     }
  80.     b[0] = b1[0] + b1[3];
  81.     b[1] = b1[1] + b1[2];
  82.     b[2] = b1[1] - b1[2];
  83.     b[3] = b1[0] - b1[3];
  84.     b[4] = b1[4];
  85.     b[5] = (b1[6] - b1[5]) * f0;
  86.     b[6] = (b1[6] + b1[5]) * f0;
  87.     b[7] = b1[7];
  88.     d[0][i] = (b[0] + b[1]) * f4;
  89.     d[4][i] = (b[0] - b[1]) * f4;
  90.     d[2][i] = b[2] * f6 + b[3] * f2;
  91.     d[6][i] = b[3] * f6 - b[2] * f2;
  92.     b1[4] = b[4] + b[5];
  93.     b1[7] = b[7] + b[6];
  94.     b1[5] = b[4] - b[5];
  95.     b1[6] = b[7] - b[6];
  96.     d[1][i] = b1[4] * f7 + b1[7] * f1;
  97.     d[5][i] = b1[5] * f3 + b1[6] * f5;
  98.     d[7][i] = b1[7] * f7 - b1[4] * f1;
  99.     d[3][i] = b1[6] * f3 - b1[5] * f5;
  100.   }
  101.   /* Zigzag - scanning */
  102.   for (i = 0; i < 8; i++) {
  103.     for (j = 0; j < 8; j++) {
  104.       *(coeff + zigzag[i][j]) = (int)(d[i][j]);
  105.     }
  106.   }
  107.   return 0;
  108. }
  109. /**********************************************************************
  110.  *
  111.  * Name:        idct
  112.  * Description: Descans zigzag-scanned coefficients and does
  113.  *        inverse dct on 64 coefficients
  114.  *                      single precision floats
  115.  *
  116.  * Input:        64 coefficients, block for 64 pixels
  117.  * Returns:        0
  118.  * Side effects:
  119.  *
  120.  * Date: 930128 Author: Robert.Danielsen@nta.no
  121.  *
  122.  **********************************************************************/
  123. int idct(int *coeff,int *block)
  124. {
  125.   int                j1, i, j;
  126.   double b[8], b1[8], d[8][8];
  127.   double f0=.7071068;
  128.   double f1=.4903926;
  129.   double f2=.4619398;
  130.   double f3=.4157348;
  131.   double f4=.3535534;
  132.   double f5=.2777851;
  133.   double f6=.1913417;
  134.   double f7=.0975452;
  135.   double e, f, g, h;
  136.   /* Horizontal */
  137.   /* Descan coefficients first */
  138.   for (i = 0; i < 8; i++) {
  139.     for (j = 0; j < 8; j++) {
  140.       b[j] = *( coeff + zigzag[i][j]);
  141.     }
  142.     e = b[1] * f7 - b[7] * f1;
  143.     h = b[7] * f7 + b[1] * f1;
  144.     f = b[5] * f3 - b[3] * f5;
  145.     g = b[3] * f3 + b[5] * f5;
  146.     b1[0] = (b[0] + b[4]) * f4;
  147.     b1[1] = (b[0] - b[4]) * f4;
  148.     b1[2] = b[2] * f6 - b[6] * f2;
  149.     b1[3] = b[6] * f6 + b[2] * f2;
  150.     b[4] = e + f;
  151.     b1[5] = e - f;
  152.     b1[6] = h - g;
  153.     b[7] = h + g;
  154.     
  155.     b[5] = (b1[6] - b1[5]) * f0;
  156.     b[6] = (b1[6] + b1[5]) * f0;
  157.     b[0] = b1[0] + b1[3];
  158.     b[1] = b1[1] + b1[2];
  159.     b[2] = b1[1] - b1[2];
  160.     b[3] = b1[0] - b1[3];
  161.     for (j = 0; j < 4; j++) {
  162.       j1 = 7 - j;
  163.       d[i][j] = b[j] + b[j1];
  164.       d[i][j1] = b[j] - b[j1];
  165.     }
  166.   }
  167.   /* Vertical */
  168.   
  169.   for (i = 0; i < 8; i++) {
  170.     for (j = 0; j < 8; j++) {
  171.       b[j] = d[j][i];
  172.     }
  173.     e = b[1] * f7 - b[7] * f1;
  174.     h = b[7] * f7 + b[1] * f1;
  175.     f = b[5] * f3 - b[3] * f5;
  176.     g = b[3] * f3 + b[5] * f5;
  177.     b1[0] = (b[0] + b[4]) * f4;
  178.     b1[1] = (b[0] - b[4]) * f4;
  179.     b1[2] = b[2] * f6 - b[6] * f2;
  180.     b1[3] = b[6] * f6 + b[2] * f2;
  181.     b[4] = e + f;
  182.     b1[5] = e - f;
  183.     b1[6] = h - g;
  184.     b[7] = h + g;
  185.     b[5] = (b1[6] - b1[5]) * f0;
  186.     b[6] = (b1[6] + b1[5]) * f0;
  187.     b[0] = b1[0] + b1[3];
  188.     b[1] = b1[1] + b1[2];
  189.     b[2] = b1[1] - b1[2];
  190.     b[3] = b1[0] - b1[3];
  191.     for (j = 0; j < 4; j++) {
  192.       j1 = 7 - j;
  193.       d[j][i] = b[j] + b[j1];
  194.       d[j1][i] = b[j] - b[j1];
  195.     }
  196.   }
  197.   for (i = 0; i < 8; i++) {
  198.     for (j = 0; j < 8; j++) {
  199.       *(block + i * 8 + j) = mnint(d[i][j]);
  200.     }
  201.   }
  202.   return 0;
  203. }