AVDataProcess/yuv420p_split.cpp

49 lines
1.2 KiB
C++

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef int BOOL;
#define TRUE 1
#define FALSE 0
//分离YUV420P像素数据中的Y、U、V分量
BOOL yuv420p_split(const char *file, int width, int height)
{
if (file == NULL) {
return FALSE;
}
FILE *fp = fopen(file, "rb+");
FILE *fp0 = fopen("./out/output_420p_yuv.y", "wb+");
FILE *fp1 = fopen("./out/output_420p_y.y", "wb+");
FILE *fp2 = fopen("./out/output_420p_u.y", "wb+");
FILE *fp3 = fopen("./out/output_420p_v.y", "wb+");
unsigned char *data = (unsigned char *)malloc(width*height*3/2);
memset(data, 0, width*height*3/2);
fread(data, 1, width*height*3/2, fp);
fwrite(data, 1, width*height*3/2, fp0);
fwrite(data, 1, width*height, fp1);
fwrite(data+width*height, 1, width*height/4, fp2);
fwrite(data+width*height*5/4, 1, width*height/4, fp3);
free(data);
fclose(fp);
fclose(fp0);
fclose(fp1);
fclose(fp2);
fclose(fp3);
return TRUE;
}
int main(int argc, char *argv[])
{
if (argc == 4) {
yuv420p_split(argv[1], atoi(argv[2]), atoi(argv[3]));
}
return 0;
}