图像处理之基础---2个YUV视频 拼接技术

时间:2023-03-10 07:01:14
图像处理之基础---2个YUV视频 拼接技术
  1. /*************************************************
  2. * 主要功能:两路 YUV4:2:0拼接一路左右半宽格式YUV视频
  3. 参考资料:http://www.pudn.com/downloads271/sourcecode/multimedia/vfw/detail1237363.html
  4. U_size=V_size=Y_size/2
  5. *************************************************/
  6. #include<stdio.h>
  7. #include<stdlib.h>
  8. #include<string.h>
  9. #define PREWEIGHT 1920
  10. #define PREHEIGHT 1080
  11. #define RESWEIGHT 3840
  12. #define RESHEIGHT 1080
  13. #define PREYSIZE ((PREWEIGHT)*(PREHEIGHT))
  14. #define PREUSIZE ((PREWEIGHT/2)*(PREHEIGHT/2))
  15. #define PREVSIZE ((PREWEIGHT/2)*(PREHEIGHT/2))
  16. #define RESYSIZE ((RESWEIGHT)*(RESHEIGHT))
  17. #define RESUSIZE ((RESWEIGHT/2)*(RESHEIGHT/2))
  18. #define RESVSIZE ((RESWEIGHT/2)*(RESHEIGHT/2))
  19. #define PRESIZE ((PREYSIZE)+(PREUSIZE)+(PREVSIZE))
  20. #define RESSIZE ((RESYSIZE)+(RESUSIZE)+(RESVSIZE))
  21. int GetFrameNum(const char *File)
  22. {
  23. FILE *fp;
  24. int size=0;
  25. if (!(fp=fopen(File,"rb")))
  26. {
  27. printf("Open %s error !",File);
  28. exit(1);
  29. }
  30. else
  31. {
  32. fseek(fp,0,SEEK_END);/*将文件指针移到YUV文件的末尾*/
  33. size=ftell(fp);/*计算文件的总大小*/
  34. }
  35. return (size/PRESIZE);
  36. }
  37. void ReadYUV(char *ResBuf,char *PreBuf,int resstart,int prestart,int resoffset,int preoffset,int size,int height)
  38. {
  39. int k;
  40. for (k=0;k<height;k++)
  41. {
  42. memmove(ResBuf+resstart+k*(resoffset),PreBuf+prestart+k*(preoffset),size);//注意这里用memmov不用strncpy
  43. }
  44. }
  45. int main(int argc,char *argv[])
  46. {
  47. const char *FileName[]={"e:\BMX_L_1920x1080_240frms.yuv","e:\BMX_R_1920x1080_240frms.yuv"};/*两路YUV文件名*/
  48. FILE *FileResult;/*输出文件名*/
  49. FILE** fp_combine=(FILE**)malloc(sizeof(FILE *)*3);/*申请文件指针*/
  50. int *FileFrameNum=(int *)malloc(sizeof(int)*3);/*每个YUV的帧数*/
  51. char *PreBuf=(char *)malloc(sizeof(char)*(PRESIZE+1));/*处理前每一帧图像的大小*/
  52. char *ResBuf=(char*)malloc(sizeof(char)*(RESSIZE+1));/*处理后每一帧图像的大小*/
  53. int Y_start_section=0;/*预处理图片Y分量存入目标区域的起始区域*/
  54. int U_start_section = 0;/*预处理图片U分量存入目标区域的起始区域*/
  55. int V_start_section = 0;/*预处理图片V分量存入目标区域的起始区域*/
  56. int File_offset = 0;/*预处理文件偏移值*/
  57. int i_combine=0,j_combine=0,k_combine=0;/*控制循环*/
  58. /*判断申请内存是否成功*/
  59. if (!((fp_combine)&&(FileFrameNum)&&(PreBuf)&&(ResBuf)))
  60. {
  61. printf("Allocate memeroy Faile !");
  62. exit(1);
  63. }
  64. /*初始化申请空间*/
  65. memset(fp_combine,0,sizeof(FILE *)*2);
  66. memset(FileFrameNum,0,sizeof(int)*2);
  67. memset(PreBuf,0,sizeof(char)*PRESIZE);
  68. memset(ResBuf,0,sizeof(char)*RESSIZE);
  69. if (!(FileResult=fopen("hua_result.YUV","wb")))/*创建输出文件*/
  70. {
  71. printf("Creat File faile !");
  72. exit(1);
  73. }
  74. for (i_combine=0;i_combine<2;i_combine++)
  75. {
  76. if(!(fp_combine[i_combine]=fopen(FileName[i_combine],"rb")))/*打开输入文件*/
  77. {
  78. printf("Open File %s Faile !",FileName[i_combine]);
  79. exit(1);
  80. }
  81. else
  82. {
  83. FileFrameNum[i_combine]=GetFrameNum(FileName[i_combine]);/*存储每一个视频的帧数*/
  84. }
  85. }
  86. i_combine=0;
  87. k_combine=FileFrameNum[i_combine];
  88. while (i_combine<k_combine)
  89. {
  90. File_offset = i_combine*PRESIZE;
  91. j_combine=0;
  92. while (j_combine<2)
  93. {
  94. fseek(fp_combine[j_combine],File_offset,SEEK_SET);/*移动文件指针至需要处理的数据的位置*/
  95. fread(PreBuf,1,PRESIZE,fp_combine[j_combine]);/*读取一幅图像*/
  96. if (j_combine==0)
  97. {
  98. /*把读取预处理图片Y/U/V分量的起始位置放置目标对应位置*/
  99. Y_start_section=0;
  100. U_start_section=RESYSIZE;
  101. V_start_section=RESYSIZE+RESUSIZE;
  102. }
  103. else
  104. {
  105. /*把读取预处理图片Y/U/V分量的起始位置放置目标对应位置*/
  106. Y_start_section=PREWEIGHT;
  107. U_start_section=RESYSIZE+PREWEIGHT/2;
  108. V_start_section=RESYSIZE+RESUSIZE+PREWEIGHT/2;
  109. }
  110. /*分别读Y、U、V*/
  111. ReadYUV(ResBuf,PreBuf,Y_start_section,0,                 RESWEIGHT,PREWEIGHT,PREWEIGHT,PREHEIGHT);
  112. ReadYUV(ResBuf,PreBuf,U_start_section,PREYSIZE,          RESWEIGHT/2,PREWEIGHT/2,PREWEIGHT/2,PREHEIGHT/2);
  113. ReadYUV(ResBuf,PreBuf,V_start_section,PREYSIZE+PREUSIZE, RESWEIGHT/2,PREWEIGHT/2,PREWEIGHT/2,PREHEIGHT/2);
  114. j_combine++;
  115. }
  116. fwrite(ResBuf,1,RESSIZE,FileResult);
  117. fflush(FileResult);
  118. i_combine++;
  119. }
  120. fclose(fp_combine[0]);
  121. fclose(fp_combine[1]);
  122. fclose(FileResult);
  123. return 0;
  124. }

http://blog.****.net/huahuahailang/article/details/9040847