YUY2到YUV420的转换

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <LIMITS.H>

//src_fileName out_fileName src_width src_height frameNo
int main(int argc, char* argv[])
{
char src_fileName[255];
char out_fileName[255];

int src_width ,src_height;
src_width = src_height = 0;

int frameNum = INT_MAX;

int src_size, out_size, tem_size;
src_size = out_size = tem_size = 0;

unsigned char * src_buf, *out_buf, *tem_buf;

FILE* in_file, *out_file;

unsigned char *Y, *U, *V;
unsigned char * Y2, *U2, *V2;

unsigned char * p=NULL;

if(argc>5)
{
   strcpy(src_fileName, argv[1]);
   strcpy(out_fileName, argv[2]);
   src_width = atoi(argv[3]);
   src_height = atoi(argv[4]);
   frameNum = atoi(argv[5]);
}
else
{
   strcpy(src_fileName, argv[1]);
   strcpy(out_fileName, argv[2]);
   src_width = atoi(argv[3]);
   src_height = atoi(argv[4]);
}

src_size = (src_width * src_height) << 1 ;//对于YUY2 4:2:2
src_buf = (unsigned char *)malloc(src_size*sizeof(char));
memset(src_buf, 0, src_size);

tem_size = (src_width * src_height) << 1; ///对于YUV 4:2:2
tem_buf = (unsigned char *)malloc(tem_size*sizeof(char));
memset(tem_buf, 0, tem_size);

out_size = src_width*src_height * 1.5;//对于YUV 4:2:0
out_buf = (unsigned char *)malloc(out_size*sizeof(char));
memset(out_buf, 0, out_size);

in_file = fopen(src_fileName, "rb");
if (!in_file)
    {
        printf("cannot open input file.");
        return 0;
    }

out_file = fopen(out_fileName, "wb");
    if (!out_file)
    {
        printf("cannot write 264 file./n");
        return 0;
    }

while(frameNum>0 && !feof(in_file))
{
   //读出一帧数据
   if (fread(src_buf, src_size, 1, in_file) <= 0)
    printf("cannot read from input file.");
   p = src_buf;
  
   Y = tem_buf;
   U = Y + src_width*src_height;
   V = U + (src_width*src_height>>1);
  
   Y2 = out_buf;
   U2 = Y2 + src_width*src_height;
   V2 = U2 + (src_width*src_height>>2);
  
   /*由打包YUYV变成平板YUV*/
   for(int k=0; k<src_height; ++k)
   {
    for(int j=0; j<(src_width>>1); ++j)
    {    
     Y[j*2] = p[4*j];    
     U[j] = p[4*j+1];
     Y[j*2+1] = p[4*j+2];
     V[j] = p[4*j+3];
    }
    p = p + src_width*2;
   
    Y = Y + src_width;
    U = U + (src_width>>1);
    V = V + (src_width>>1);
   }
  
   //复位
   Y = tem_buf;
   U = Y + src_width*src_height;
   V = U + (src_width*src_height>>1);
  
   for(int l=0; l<src_height/2; ++l)
   {
    memcpy(U2, U, src_width>>1);
    memcpy(V2, V, src_width>>1);
   
    U2 = U2 + (src_width>>1);
    V2 = V2 + (src_width>>1);
   
    U = U + (src_width);
    V = V + (src_width);
   }
  
   memcpy(Y2, Y, src_width*src_height);
  
   fwrite(out_buf, sizeof(char), out_size, out_file);
   printf(".");

   frameNum--;
}

fflush(out_file);

free(src_buf); src_buf=NULL;
free(tem_buf); tem_buf=NULL;
free(out_buf); out_buf=NULL;

fclose(in_file);
fclose(out_file);

return 0;
}