开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图形图像处理 > 查看源码
Store.cpp
资源名称：263.rar　[点击查看]
上传用户：szklck
上传日期：2007-01-22
资源大小：925k
文件大小：10k
源码类别：

图形图像处理

开发平台：

Visual C++

Store.cpp：源码内容
  1. // Store.cpp: implementation of the CStore class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "视频编解码器.h"
  7. #include "Store.h"
  9. #ifdef _DEBUG
  10. #undef THIS_FILE
  11. static char THIS_FILE[]=__FILE__;
  12. #define new DEBUG_NEW
  13. #endif
  15. //////////////////////////////////////////////////////////////////////
  16. // Construction/Destruction
  17. //////////////////////////////////////////////////////////////////////
  18. #define OBFRSIZE 4096
  19. static unsigned char obfr[OBFRSIZE];
  20. static unsigned char *optr;
  21. static int outfile;
  22. extern unsigned char *clp;
  23. extern int outtype,horizontal_size,coded_picture_width,coded_picture_height;
  24. extern int quiet,vertical_size;
  25. CStore::CStore()
  26. {
  28. }
  30. CStore::~CStore()
  31. {
  33. }
  34. // store a picture as either one frame or two fields
  35. void CStore::storeframe(unsigned char *src[], int frame)
  36. {
  37.     char outname[8]="D:\测试";
  38.     //outname=
  39. store_one(outname,src,0,coded_picture_width,vertical_size);
  40. }
  42. void CStore::store_one(char *outname, unsigned char *src[], 
  43.        int offset, int incr, int height)
  44. {
  45.  switch (outtype)
  46.   {
  47.   case T_BMP:
  48.     store_bmp(outname,src,offset,incr,height);
  49. break;
  50.   case T_YUV:
  51.    // store_yuv(outname,src,offset,incr,height);
  52.     break;
  53.   case T_YUV_CONC:
  54.    // store_yuv_append(outname,src,offset,incr,height);
  55.     break;
  56.   case T_SIF:
  57.    // store_sif(outname,src,offset,incr,height);
  58.     break;
  59.   case T_TGA:
  60.    // store_ppm_tga(outname,src,offset,incr,height,1);
  61.     break;
  62.   case T_PPM:
  63.   //  store_ppm_tga(outname,src,offset,incr,height,0);
  64.     break;
  65. #ifdef DISPLAY
  66.   case T_X11:
  67.     dither(src);
  68.     break;
  69. #endif
  70.   default:
  71.     break;
  72.   }
  73. }
  75. void CStore::store_bmp(char *outname, unsigned char *src[],  
  76.                int offset, int incr, int height)
  77. {
  78.   
  80. }
  82. /*
  83. //没有文件头的Y U V格式文件
  84. void CStore::store_yuv(char *outname, unsigned char *src[], 
  85.        int offset, int incr, int height)
  86. {
  87.   int hsize;
  88.   char tmpname[32];
  90.   hsize = horizontal_size;
  92.   sprintf(tmpname,"%s.Y",outname);
  93.   store_yuv1(tmpname,src[0],offset,incr,hsize,height,0);
  95.   offset>>=1; incr>>=1; hsize>>=1;
  96.   height>>=1;
  98.   sprintf(tmpname,"%s.U",outname);
  99.   store_yuv1(tmpname,src[1],offset,incr,hsize,height,0);
  101.   sprintf(tmpname,"%s.V",outname);
  102.   store_yuv1(tmpname,src[2],offset,incr,hsize,height,0);
  104. }
  105. //有文件头的YUV格式的文件 
  106. void CStore::store_yuv_append(char *outname, unsigned char *src[], int offset, int incr, int height)
  107. {
  108.   int hsize;
  109.   hsize = horizontal_size;
  111.   store_yuv1(outname,src[0],offset,incr,hsize,height,1);
  113.   offset>>=1; incr>>=1; hsize>>=1;
  114.   height>>=1;
  116.   store_yuv1(outname,src[1],offset,incr,hsize,height,1);
  117.   store_yuv1(outname,src[2],offset,incr,hsize,height,1);
  119. }
  121. void CStore::store_sif(char *outname, unsigned char *src[], int offset, int incr, int height)
  122. {
  123.   int i,j;
  124.   unsigned char *py, *pu, *pv;
  125.   static unsigned char *u422, *v422;
  127.   if (!u422) 
  128.   {
  129.     if (!(u422 = (unsigned char *)malloc((coded_picture_width>>1)
  130.  *coded_picture_height)))
  131. {
  132. AfxMessageBox("malloc failed!");
  133. return;
  134. }
  135.     if (!(v422 = (unsigned char *)malloc((coded_picture_width>>1)
  136.  *coded_picture_height)))
  137. {
  138. AfxMessageBox("malloc failed!");
  139. return;
  140. }
  141.   }
  142.   
  143.   conv420to422(src[1],u422);
  144.   conv420to422(src[2],v422);
  146.   strcat(outname,".SIF");
  148.   if (!quiet)
  149.     fprintf(stderr,"saving %sn",outname);
  151.   if ((outfile = open(outname,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
  152.   {
  153.       AfxMessageBox("Couldn't create outname!");
  154.       
  155.   }
  157.   optr = obfr;
  159.   for (i=0; i<height; i++)
  160.   {
  161.     py = src[0] + offset + incr*i;
  162.     pu = u422 + (offset>>1) + (incr>>1)*i;
  163.     pv = v422 + (offset>>1) + (incr>>1)*i;
  165.     for (j=0; j<horizontal_size; j+=2)
  166.     {
  167.       putbyte(*pu++);
  168.       putbyte(*py++);
  169.       putbyte(*pv++);
  170.       putbyte(*py++);
  171.     }
  172.   }
  174.   if (optr!=obfr)
  175.     write(outfile,obfr,optr-obfr);
  177.   close(outfile);
  179. }
  181. void CStore::store_ppm_tga(char *outname, unsigned char *src[], int offset, int incr, int height, int tgaflag)
  182. {
  183.   int i, j;
  184.   int y, u, v, r, g, b;
  185.   int crv, cbu, cgu, cgv;
  186.   unsigned char *py, *pu, *pv;
  187.   static unsigned char tga24[14] = {0,0,2,0,0,0,0, 0,0,0,0,0,24,32};
  188.   char header[32];
  189.   static unsigned char *u422, *v422, *u444, *v444;
  191.   if (!u444) {
  192.     if (!(u422 = (unsigned char *)malloc((coded_picture_width>>1)
  193.  *coded_picture_height)))
  194.       error("malloc failed");
  195.     if (!(v422 = (unsigned char *)malloc((coded_picture_width>>1)
  196.  *coded_picture_height)))
  197.       error("malloc failed");
  199.     if (!(u444 = (unsigned char *)malloc(coded_picture_width
  200.  *coded_picture_height)))
  201.       error("malloc failed");
  203.     if (!(v444 = (unsigned char *)malloc(coded_picture_width
  204.  *coded_picture_height)))
  205.       error("malloc failed");
  206.   }
  208.   conv420to422(src[1],u422);
  209.   conv420to422(src[2],v422);
  210.   conv422to444(u422,u444);
  211.   conv422to444(v422,v444);
  213.   strcat(outname,tgaflag ? ".tga" : ".ppm");
  215.   if (!quiet)
  216.     fprintf(stderr,"saving %sn",outname);
  218.   if ((outfile = open(outname,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
  219.     {
  220.       sprintf(errortext,"Couldn't create %sn",outname);
  221.       error(errortext);
  222.     }
  224.   optr = obfr;
  226.   if (tgaflag) {
  227.     //* TGA header 
  228.     for (i=0; i<12; i++)
  229.       putbyte(tga24[i]);
  231.     putword(horizontal_size); putword(height);
  232.     putbyte(tga24[12]); putbyte(tga24[13]);
  233.   }
  234.   else {
  235.     // PPM header 
  236.     sprintf(header,"P6n%d %dn255n",horizontal_size,height);
  238.     for (i=0; header[i]!=0; i++)
  239.       putbyte(header[i]);
  240.   }
  242.   //* matrix coefficients 
  243.   crv = convmat[matrix_coefficients][0];
  244.   cbu = convmat[matrix_coefficients][1];
  245.   cgu = convmat[matrix_coefficients][2];
  246.   cgv = convmat[matrix_coefficients][3];
  247.   
  248.   for (i=0; i<height; i++)
  249.   {
  250.     py = src[0] + offset + incr*i;
  251.     pu = u444 + offset + incr*i;
  252.     pv = v444 + offset + incr*i;
  254.     for (j=0; j<horizontal_size; j++)
  255. {
  256.       u = *pu++ - 128;
  257.       v = *pv++ - 128;
  258.       y = 76309 * (*py++ - 16); //* (255/219)*65536 
  259.       r = clp[(y + crv*v + 32768)>>16];
  260.       g = clp[(y - cgu*u - cgv*v + 32768)>>16];
  261.       b = clp[(y + cbu*u + 32786)>>16];
  263.       if (tgaflag) 
  264.   {
  265.      putbyte(b);
  266. putbyte(g); 
  267. putbyte(r);
  268.       }
  269.       else 
  270.   {
  271.     putbyte(r);
  272. putbyte(g);
  273. putbyte(b);
  274.       }
  275.     }
  276.   }
  278.   if (optr!=obfr)
  279.     write(outfile,obfr,optr-obfr);
  281.   close(outfile);
  283. }
  285. void CStore::store_yuv1(char *name, unsigned char *src, int offset, int incr, int width, int height, int append)
  286. {
  287.  int i, j;
  288.   unsigned char *p;
  290.   if (append)
  291.   {
  292.     if ((outfile = open(name,O_APPEND|O_WRONLY|O_BINARY,0666))==-1) 
  293. {
  294.       sprintf(errortext,"Couldn't append to %sn",name);
  295.       error(errortext);
  296.     }
  297.   }
  298.   else 
  299.   {
  300.     if ((outfile = open(name,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
  301. {
  302.       AfxMessageBox("Couldn't create file name!");
  303.       error(errortext);
  304.     }
  305.   }
  306.   if (!quiet)
  307.     AfxMessageBox("saving file!");
  309.   optr=obfr;
  311.   for (i=0; i<height; i++) 
  312.   {
  313.     p = src + offset + incr*i;
  314.     for (j=0; j<width; j++)
  315.       putbyte(*p++);
  316.   }
  318.   if (optr!=obfr)
  319.     write(outfile,obfr,optr-obfr);
  321.   close(outfile);
  323. }
  324. */
  325. void CStore::putbyte(int c)
  326. {
  327.   *optr++ = c;
  329.   if (optr == obfr+OBFRSIZE)
  330.   {
  331. //    write(outfile,obfr,OBFRSIZE);
  332.     optr = obfr;
  333.   }
  335. }
  337. void CStore::putword(int w)
  338. {
  339.   putbyte(w); 
  340.   putbyte(w>>8);
  342. }
  343. //  将420格式的文件转化为444的文件
  344. //  1.垂直插值   422
  345. //  2.水平插值
  346. // 水平方向的插值 1:2 interpolation filter 
  347. void CStore::conv422to444(unsigned char *src, unsigned char *dst)
  348. {
  349.  int i, i2, w, j, im3, im2, im1, ip1, ip2, ip3;
  351.   w = coded_picture_width>>1;
  353.   for (j=0; j<coded_picture_height; j++)
  354.   {
  355.     for (i=0; i<w; i++)
  356. {
  357.       i2 = i<<1;
  358.       im3 = (i<3) ? 0 : i-3;  //左右边界
  359.       im2 = (i<2) ? 0 : i-2;
  360.       im1 = (i<1) ? 0 : i-1;
  361.       ip1 = (i<w-1) ? i+1 : w-1;
  362.       ip2 = (i<w-2) ? i+2 : w-1;
  363.       ip3 = (i<w-3) ? i+3 : w-1;
  365.       // FIR filter coefficients (*256): 5 -21 70 228 -37 11 
  366.       dst[i2] =   clp[(int)(  5*src[im3]
  367.       -21*src[im2]
  368.       +70*src[im1]
  369.       +228*src[i]
  370.       -37*src[ip1]
  371.       +11*src[ip2]+128)>>8];
  373.       dst[i2+1] = clp[(int)(  5*src[ip3]
  374.       -21*src[ip2]
  375.       +70*src[ip1]
  376.       +228*src[i]
  377.       -37*src[im1]
  378.       +11*src[im2]+128)>>8];
  379.     }
  380.     src+= w;
  381.     dst+= coded_picture_width;
  382.   }
  383. }
  384. // 垂直方向的插值 1:2 interpolation filter 
  385. void CStore::conv420to422(unsigned char *src, unsigned char *dst)
  386. {
  387.   int w, h, i, j, j2;
  388.   int jm3, jm2, jm1, jp1, jp2, jp3;
  390.   w = coded_picture_width>>1;
  391.   h = coded_picture_height>>1;
  393.   /* intra frame */
  394.   for (i=0; i<w; i++) 
  395.   {
  396.     for (j=0; j<h; j++) 
  397. {
  398.       j2 = j<<1;//
  399.       jm3 = (j<3) ? 0 : j-3;//边界点
  400.       jm2 = (j<2) ? 0 : j-2;
  401.       jm1 = (j<1) ? 0 : j-1;
  402.       jp1 = (j<h-1) ? j+1 : h-1;//边界点
  403.       jp2 = (j<h-2) ? j+2 : h-1;
  404.       jp3 = (j<h-3) ? j+3 : h-1;
  405.        
  406.       // 进行平滑滤波
  407.   //FIR filter coefficients (*256): 5 -21 70 228 -37 11 
  408.       // New FIR filter coefficients (*256): 3 -16 67 227 -32 7 
  409.       dst[w*j2] =clp[(int)(  3*src[w*jm3]
  410.   -16*src[w*jm2]
  411.   +67*src[w*jm1]
  412.   +227*src[w*j]
  413.   -32*src[w*jp1]
  414.   +7*src[w*jp2]+128)>>8];
  416.       dst[w*(j2+1)] = clp[(int)(  3*src[w*jp3]
  417.   -16*src[w*jp2]
  418.   +67*src[w*jp1]
  419.   +227*src[w*j]
  420.   -32*src[w*jm1]
  421.   +7*src[w*jm2]+128)>>8];
  422.     }
  423.     src++;
  424.     dst++;
  425.   }
  426. }