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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 嵌入式/单片机编程 > 其他嵌入式/单片机内容 > 查看源码
readpic.c
资源名称：mpeg2.rar　[点击查看]
上传用户：ma_junhua
上传日期：2008-04-11
资源大小：2752k
文件大小：13k
源码类别：

其他嵌入式/单片机内容

开发平台：

C/C++

readpic.c：源码内容
  1. /* readpic.c, read source pictures                                          */
  3. /* Copyright (C) 1996, MPEG Software Simulation Group. All Rights Reserved. */
  5. /*
  6.  * Disclaimer of Warranty
  7.  *
  8.  * These software programs are available to the user without any license fee or
  9.  * royalty on an "as is" basis.  The MPEG Software Simulation Group disclaims
  10.  * any and all warranties, whether express, implied, or statuary, including any
  11.  * implied warranties or merchantability or of fitness for a particular
  12.  * purpose.  In no event shall the copyright-holder be liable for any
  13.  * incidental, punitive, or consequential damages of any kind whatsoever
  14.  * arising from the use of these programs.
  15.  *
  16.  * This disclaimer of warranty extends to the user of these programs and user's
  17.  * customers, employees, agents, transferees, successors, and assigns.
  18.  *
  19.  * The MPEG Software Simulation Group does not represent or warrant that the
  20.  * programs furnished hereunder are free of infringement of any third-party
  21.  * patents.
  22.  *
  23.  * Commercial implementations of MPEG-1 and MPEG-2 video, including shareware,
  24.  * are subject to royalty fees to patent holders.  Many of these patents are
  25.  * general enough such that they are unavoidable regardless of implementation
  26.  * design.
  27.  *
  28.  */
  30. #include <stdio.h>
  31. #include <stdlib.h>
  32. #include "config.h"
  33. #include "global.h"
  35. /* private prototypes */
  36. static void read_y_u_v _ANSI_ARGS_((char *fname, unsigned char *frame[]));
  37. static void read_yuv _ANSI_ARGS_((char *fname, unsigned char *frame[]));
  38. static void read_ppm _ANSI_ARGS_((char *fname, unsigned char *frame[]));
  39. static void border_extend _ANSI_ARGS_((unsigned char *frame, int w1, int h1,
  40.   int w2, int h2));
  41. static void conv444to422 _ANSI_ARGS_((unsigned char *src, unsigned char *dst));
  42. static void conv422to420 _ANSI_ARGS_((unsigned char *src, unsigned char *dst));
  44. void readframe(fname,frame)
  45. char *fname;
  46. unsigned char *frame[];
  47. {
  48.   switch (inputtype)
  49.   {
  50.   case T_Y_U_V:
  51.     read_y_u_v(fname,frame);
  52.     break;
  53.   case T_YUV:
  54.     read_yuv(fname,frame);
  55.     break;
  56.   case T_PPM:
  57.     read_ppm(fname,frame);
  58.     break;
  59.   default:
  60.     break;
  61.   }
  62. }
  64. static void read_y_u_v(fname,frame)
  65. char *fname;
  66. unsigned char *frame[];
  67. {
  68.   int i;
  69.   int chrom_hsize, chrom_vsize;
  70.   char name[128];
  71.   FILE *fd;
  73.   chrom_hsize = (chroma_format==CHROMA444) ? horizontal_size
  74.                                            : horizontal_size>>1;
  75.   chrom_vsize = (chroma_format!=CHROMA420) ? vertical_size
  76.                                            : vertical_size>>1;
  78.   sprintf(name,"%s.Y",fname);
  79.   if (!(fd = fopen(name,"rb")))
  80.   {
  81.     sprintf(errortext,"Couldn't open %sn",name);
  82.     error(errortext);
  83.   }
  84.   for (i=0; i<vertical_size; i++)
  85.     fread(frame[0]+i*width,1,horizontal_size,fd);
  86.   fclose(fd);
  87.   border_extend(frame[0],horizontal_size,vertical_size,width,height);
  89.   sprintf(name,"%s.U",fname);
  90.   if (!(fd = fopen(name,"rb")))
  91.   {
  92.     sprintf(errortext,"Couldn't open %sn",name);
  93.     error(errortext);
  94.   }
  95.   for (i=0; i<chrom_vsize; i++)
  96.     fread(frame[1]+i*chrom_width,1,chrom_hsize,fd);
  97.   fclose(fd);
  98.   border_extend(frame[1],chrom_hsize,chrom_vsize,chrom_width,chrom_height);
  100.   sprintf(name,"%s.V",fname);
  101.   if (!(fd = fopen(name,"rb")))
  102.   {
  103.     sprintf(errortext,"Couldn't open %sn",name);
  104.     error(errortext);
  105.   }
  106.   for (i=0; i<chrom_vsize; i++)
  107.     fread(frame[2]+i*chrom_width,1,chrom_hsize,fd);
  108.   fclose(fd);
  109.   border_extend(frame[2],chrom_hsize,chrom_vsize,chrom_width,chrom_height);
  110. }
  112. static void read_yuv(fname,frame)
  113. char *fname;
  114. unsigned char *frame[];
  115. {
  116.   int i;
  117.   int chrom_hsize, chrom_vsize;
  118.   char name[128];
  119.   FILE *fd;
  121.   chrom_hsize = (chroma_format==CHROMA444) ? horizontal_size
  122.                                            : horizontal_size>>1;
  123.   chrom_vsize = (chroma_format!=CHROMA420) ? vertical_size
  124.                                            : vertical_size>>1;
  126.   sprintf(name,"%s.yuv",fname);
  127.   if (!(fd = fopen(name,"rb")))
  128.   {
  129.     sprintf(errortext,"Couldn't open %sn",name);
  130.     error(errortext);
  131.   }
  133.   /* Y */
  134.   for (i=0; i<vertical_size; i++)
  135.     fread(frame[0]+i*width,1,horizontal_size,fd);
  136.   border_extend(frame[0],horizontal_size,vertical_size,width,height);
  138.   /* Cb */
  139.   for (i=0; i<chrom_vsize; i++)
  140.     fread(frame[1]+i*chrom_width,1,chrom_hsize,fd);
  141.   border_extend(frame[1],chrom_hsize,chrom_vsize,chrom_width,chrom_height);
  143.   /* Cr */
  144.   for (i=0; i<chrom_vsize; i++)
  145.     fread(frame[2]+i*chrom_width,1,chrom_hsize,fd);
  146.   border_extend(frame[2],chrom_hsize,chrom_vsize,chrom_width,chrom_height);
  148.   fclose(fd);
  149. }
  151. static void read_ppm(fname,frame)
  152. char *fname;
  153. unsigned char *frame[];
  154. {
  155.   int i, j;
  156.   int r, g, b;
  157.   double y, u, v;
  158.   double cr, cg, cb, cu, cv;
  159.   char name[128];
  160.   FILE *fd;
  161.   unsigned char *yp, *up, *vp;
  162.   static unsigned char *u444, *v444, *u422, *v422;
  163.   static double coef[7][3] = {
  164.     {0.2125,0.7154,0.0721}, /* ITU-R Rec. 709 (1990) */
  165.     {0.299, 0.587, 0.114},  /* unspecified */
  166.     {0.299, 0.587, 0.114},  /* reserved */
  167.     {0.30,  0.59,  0.11},   /* FCC */
  168.     {0.299, 0.587, 0.114},  /* ITU-R Rec. 624-4 System B, G */
  169.     {0.299, 0.587, 0.114},  /* SMPTE 170M */
  170.     {0.212, 0.701, 0.087}}; /* SMPTE 240M (1987) */
  172.   i = matrix_coefficients;
  173.   if (i>8)
  174.     i = 3;
  176.   cr = coef[i-1][0];
  177.   cg = coef[i-1][1];
  178.   cb = coef[i-1][2];
  179.   cu = 0.5/(1.0-cb);
  180.   cv = 0.5/(1.0-cr);
  182.   if (chroma_format==CHROMA444)
  183.   {
  184.     u444 = frame[1];
  185.     v444 = frame[2];
  186.   }
  187.   else
  188.   {
  189.     if (!u444)
  190.     {
  191.       if (!(u444 = (unsigned char *)malloc(width*height)))
  192.         error("malloc failed");
  193.       if (!(v444 = (unsigned char *)malloc(width*height)))
  194.         error("malloc failed");
  195.       if (chroma_format==CHROMA420)
  196.       {
  197.         if (!(u422 = (unsigned char *)malloc((width>>1)*height)))
  198.           error("malloc failed");
  199.         if (!(v422 = (unsigned char *)malloc((width>>1)*height)))
  200.           error("malloc failed");
  201.       }
  202.     }
  203.   }
  205.   sprintf(name,"%s.ppm",fname);
  207.   if (!(fd = fopen(name,"rb")))
  208.   {
  209.     sprintf(errortext,"Couldn't open %sn",name);
  210.     error(errortext);
  211.   }
  213.   /* skip header */
  214.   getc(fd); getc(fd); /* magic number (P6) */
  215.   pbm_getint(fd); pbm_getint(fd); pbm_getint(fd); /* width height maxcolors */
  217.   for (i=0; i<vertical_size; i++)
  218.   {
  219.     yp = frame[0] + i*width;
  220.     up = u444 + i*width;
  221.     vp = v444 + i*width;
  223.     for (j=0; j<horizontal_size; j++)
  224.     {
  225.       r=getc(fd); g=getc(fd); b=getc(fd);
  226.       /* convert to YUV */
  227.       y = cr*r + cg*g + cb*b;
  228.       u = cu*(b-y);
  229.       v = cv*(r-y);
  230.       yp[j] = (219.0/256.0)*y + 16.5;  /* nominal range: 16..235 */
  231.       up[j] = (224.0/256.0)*u + 128.5; /* 16..240 */
  232.       vp[j] = (224.0/256.0)*v + 128.5; /* 16..240 */
  233.     }
  234.   }
  235.   fclose(fd);
  237.   border_extend(frame[0],horizontal_size,vertical_size,width,height);
  238.   border_extend(u444,horizontal_size,vertical_size,width,height);
  239.   border_extend(v444,horizontal_size,vertical_size,width,height);
  241.   if (chroma_format==CHROMA422)
  242.   {
  243.     conv444to422(u444,frame[1]);
  244.     conv444to422(v444,frame[2]);
  245.   }
  247.   if (chroma_format==CHROMA420)
  248.   {
  249.     conv444to422(u444,u422);
  250.     conv444to422(v444,v422);
  251.     conv422to420(u422,frame[1]);
  252.     conv422to420(v422,frame[2]);
  253.   }
  254. }
  256. static void border_extend(frame,w1,h1,w2,h2)
  257. unsigned char *frame;
  258. int w1,h1,w2,h2;
  259. {
  260.   int i, j;
  261.   unsigned char *fp;
  263.   /* horizontal pixel replication (right border) */
  265.   for (j=0; j<h1; j++)
  266.   {
  267.     fp = frame + j*w2;
  268.     for (i=w1; i<w2; i++)
  269.       fp[i] = fp[i-1];
  270.   }
  272.   /* vertical pixel replication (bottom border) */
  274.   for (j=h1; j<h2; j++)
  275.   {
  276.     fp = frame + j*w2;
  277.     for (i=0; i<w2; i++)
  278.       fp[i] = fp[i-w2];
  279.   }
  280. }
  282. /* horizontal filter and 2:1 subsampling */
  283. static void conv444to422(src,dst)
  284. unsigned char *src, *dst;
  285. {
  286.   int i, j, im5, im4, im3, im2, im1, ip1, ip2, ip3, ip4, ip5, ip6;
  288.   if (mpeg1)
  289.   {
  290.     for (j=0; j<height; j++)
  291.     {
  292.       for (i=0; i<width; i+=2)
  293.       {
  294.         im5 = (i<5) ? 0 : i-5;
  295.         im4 = (i<4) ? 0 : i-4;
  296.         im3 = (i<3) ? 0 : i-3;
  297.         im2 = (i<2) ? 0 : i-2;
  298.         im1 = (i<1) ? 0 : i-1;
  299.         ip1 = (i<width-1) ? i+1 : width-1;
  300.         ip2 = (i<width-2) ? i+2 : width-1;
  301.         ip3 = (i<width-3) ? i+3 : width-1;
  302.         ip4 = (i<width-4) ? i+4 : width-1;
  303.         ip5 = (i<width-5) ? i+5 : width-1;
  304.         ip6 = (i<width-5) ? i+6 : width-1;
  306.         /* FIR filter with 0.5 sample interval phase shift */
  307.         dst[i>>1] = clp[(int)(228*(src[i]+src[ip1])
  308.                          +70*(src[im1]+src[ip2])
  309.                          -37*(src[im2]+src[ip3])
  310.                          -21*(src[im3]+src[ip4])
  311.                          +11*(src[im4]+src[ip5])
  312.                          + 5*(src[im5]+src[ip6])+256)>>9];
  313.       }
  314.       src+= width;
  315.       dst+= width>>1;
  316.     }
  317.   }
  318.   else
  319.   {
  320.     /* MPEG-2 */
  321.     for (j=0; j<height; j++)
  322.     {
  323.       for (i=0; i<width; i+=2)
  324.       {
  325.         im5 = (i<5) ? 0 : i-5;
  326.         im3 = (i<3) ? 0 : i-3;
  327.         im1 = (i<1) ? 0 : i-1;
  328.         ip1 = (i<width-1) ? i+1 : width-1;
  329.         ip3 = (i<width-3) ? i+3 : width-1;
  330.         ip5 = (i<width-5) ? i+5 : width-1;
  332.         /* FIR filter coefficients (*512): 22 0 -52 0 159 256 159 0 -52 0 22 */
  333.         dst[i>>1] = clp[(int)(  22*(src[im5]+src[ip5])-52*(src[im3]+src[ip3])
  334.                          +159*(src[im1]+src[ip1])+256*src[i]+256)>>9];
  335.       }
  336.       src+= width;
  337.       dst+= width>>1;
  338.     }
  339.   }
  340. }
  342. /* vertical filter and 2:1 subsampling */
  343. static void conv422to420(src,dst)
  344. unsigned char *src, *dst;
  345. {
  346.   int w, i, j, jm6, jm5, jm4, jm3, jm2, jm1;
  347.   int jp1, jp2, jp3, jp4, jp5, jp6;
  349.   w = width>>1;
  351.   if (prog_frame)
  352.   {
  353.     /* intra frame */
  354.     for (i=0; i<w; i++)
  355.     {
  356.       for (j=0; j<height; j+=2)
  357.       {
  358.         jm5 = (j<5) ? 0 : j-5;
  359.         jm4 = (j<4) ? 0 : j-4;
  360.         jm3 = (j<3) ? 0 : j-3;
  361.         jm2 = (j<2) ? 0 : j-2;
  362.         jm1 = (j<1) ? 0 : j-1;
  363.         jp1 = (j<height-1) ? j+1 : height-1;
  364.         jp2 = (j<height-2) ? j+2 : height-1;
  365.         jp3 = (j<height-3) ? j+3 : height-1;
  366.         jp4 = (j<height-4) ? j+4 : height-1;
  367.         jp5 = (j<height-5) ? j+5 : height-1;
  368.         jp6 = (j<height-5) ? j+6 : height-1;
  370.         /* FIR filter with 0.5 sample interval phase shift */
  371.         dst[w*(j>>1)] = clp[(int)(228*(src[w*j]+src[w*jp1])
  372.                              +70*(src[w*jm1]+src[w*jp2])
  373.                              -37*(src[w*jm2]+src[w*jp3])
  374.                              -21*(src[w*jm3]+src[w*jp4])
  375.                              +11*(src[w*jm4]+src[w*jp5])
  376.                              + 5*(src[w*jm5]+src[w*jp6])+256)>>9];
  377.       }
  378.       src++;
  379.       dst++;
  380.     }
  381.   }
  382.   else
  383.   {
  384.     /* intra field */
  385.     for (i=0; i<w; i++)
  386.     {
  387.       for (j=0; j<height; j+=4)
  388.       {
  389.         /* top field */
  390.         jm5 = (j<10) ? 0 : j-10;
  391.         jm4 = (j<8) ? 0 : j-8;
  392.         jm3 = (j<6) ? 0 : j-6;
  393.         jm2 = (j<4) ? 0 : j-4;
  394.         jm1 = (j<2) ? 0 : j-2;
  395.         jp1 = (j<height-2) ? j+2 : height-2;
  396.         jp2 = (j<height-4) ? j+4 : height-2;
  397.         jp3 = (j<height-6) ? j+6 : height-2;
  398.         jp4 = (j<height-8) ? j+8 : height-2;
  399.         jp5 = (j<height-10) ? j+10 : height-2;
  400.         jp6 = (j<height-12) ? j+12 : height-2;
  402.         /* FIR filter with 0.25 sample interval phase shift */
  403.         dst[w*(j>>1)] = clp[(int)(8*src[w*jm5]
  404.                             +5*src[w*jm4]
  405.                            -30*src[w*jm3]
  406.                            -18*src[w*jm2]
  407.                           +113*src[w*jm1]
  408.                           +242*src[w*j]
  409.                           +192*src[w*jp1]
  410.                            +35*src[w*jp2]
  411.                            -38*src[w*jp3]
  412.                            -10*src[w*jp4]
  413.                            +11*src[w*jp5]
  414.                             +2*src[w*jp6]+256)>>9];
  416.         /* bottom field */
  417.         jm6 = (j<9) ? 1 : j-9;
  418.         jm5 = (j<7) ? 1 : j-7;
  419.         jm4 = (j<5) ? 1 : j-5;
  420.         jm3 = (j<3) ? 1 : j-3;
  421.         jm2 = (j<1) ? 1 : j-1;
  422.         jm1 = (j<height-1) ? j+1 : height-1;
  423.         jp1 = (j<height-3) ? j+3 : height-1;
  424.         jp2 = (j<height-5) ? j+5 : height-1;
  425.         jp3 = (j<height-7) ? j+7 : height-1;
  426.         jp4 = (j<height-9) ? j+9 : height-1;
  427.         jp5 = (j<height-11) ? j+11 : height-1;
  428.         jp6 = (j<height-13) ? j+13 : height-1;
  430.         /* FIR filter with 0.25 sample interval phase shift */
  431.         dst[w*((j>>1)+1)] = clp[(int)(8*src[w*jp6]
  432.                                 +5*src[w*jp5]
  433.                                -30*src[w*jp4]
  434.                                -18*src[w*jp3]
  435.                               +113*src[w*jp2]
  436.                               +242*src[w*jp1]
  437.                               +192*src[w*jm1]
  438.                                +35*src[w*jm2]
  439.                                -38*src[w*jm3]
  440.                                -10*src[w*jm4]
  441.                                +11*src[w*jm5]
  442.                                 +2*src[w*jm6]+256)>>9];
  443.       }
  444.       src++;
  445.       dst++;
  446.     }
  447.   }
  448. }
  450. /* pbm_getc() and pbm_getint() are essentially taken from
  451.  * PBMPLUS (C) Jef Poskanzer
  452.  * but without error/EOF checking
  453.  */
  454. char pbm_getc(file)
  455. FILE* file;
  456. {
  457.   char ch;
  459.   ch = getc(file);
  461.   if (ch=='#')
  462.   {
  463.     do
  464.     {
  465.       ch = getc(file);
  466.     }
  467.     while (ch!='n' && ch!='r');
  468.   }
  470.   return ch;
  471. }
  473. int pbm_getint(file)
  474. FILE* file;
  475. {
  476.   char ch;
  477.   int i;
  479.   do
  480.   {
  481.     ch = pbm_getc(file);
  482.   }
  483.   while (ch==' ' || ch=='t' || ch=='n' || ch=='r');
  485.   i = 0;
  486.   do
  487.   {
  488.     i = i*10 + ch-'0';
  489.     ch = pbm_getc(file);
  490.   }
  491.   while (ch>='0' && ch<='9');
  493.   return i;
  494. }