计算两个YUV视频序列的PSNR

/** 
* Calculate PSNR between 2 YUV420P file 
* @param url1     Location of first Input YUV file. 
* @param url2     Location of another Input YUV file. 
* @param w        Width of Input YUV file. 
* @param h        Height of Input YUV file. 
* @param num      Number of frames to process. 
*/  
#include<iostream>
using namespace std;
int simplest_yuv420_psnr(char *url1,char *url2,int w,int h,int num){  
    FILE *fp1=fopen(url1,"rb+");  
    FILE *fp2=fopen(url2,"rb+");  
    FILE *fp=fopen("Video_PSNR.txt","w");         //创建存取PSNR值信息的txt文件


    unsigned char *pic1=(unsigned char *)malloc(w*h);  
    unsigned char *pic2=(unsigned char *)malloc(w*h);  
    
    float Sum_PSNR=0;    //定义所有帧的PSNR和
    float average_PSNR;  //定义平均PSNR
    for(int i=0;i<num;i++){  
        fread(pic1,1,w*h,fp1);  
        fread(pic2,1,w*h,fp2);  
        double mse_sum=0,mse=0,psnr=0;  
        for(int j=0;j<w*h;j++){  
            mse_sum+=pow((double)(pic1[j]-pic2[j]),2);  
        }  
        mse=mse_sum/(w*h);  
        psnr=10*log10(255.0*255.0/mse);  
        printf("the %d frame PSNR is %5.3f\n",i,psnr);          
        fprintf(fp,"the %d frame PSNR is %5.3f\n",i,psnr);      //存取每一帧的PSNR值到txt
        Sum_PSNR+=psnr;
        fseek(fp1,w*h/2,SEEK_CUR);  
        fseek(fp2,w*h/2,SEEK_CUR);   

    }  
    average_PSNR=Sum_PSNR/num;
    printf("the average PSNR is %5.3f\n",average_PSNR);
    fprintf(fp,"the average PSNR is %5.3f\n",average_PSNR);     //存取所有帧的平均PSNR值到txt
    free(pic1);  
    free(pic2);  
    fclose(fp1);  
    fclose(fp2);  
    return 0;  
}  

int main()
{   
//参数分别为(原始视频序列,压缩后视频序列,宽,高,帧数);   
    simplest_yuv420_psnr("BasketballDrill_832x480_50.yuv","BQMall_832x480_60.yuv",832,480,5);  
//注意视频序列所在路径，默认的是跟当前工程在一个目录下,当然也可以自己指定
system("pause");
    }