/* stats.c, 編碼統(tǒng)計*/
#include <stdio.h>#include <math.h>#include "global.h"/* 函數(shù)聲明 */static void calcSNR1(unsigned char *org, unsigned char *rec,  int lx, int w, int h, double *pv, double *pe);void calcSNR(org,rec)unsigned char *org[3];unsigned char *rec[3];{  int w,h,offs;  double v,e;  w = horizontal_size;  h = (pict_struct==FRAME_PICTURE) ? vertical_size : (vertical_size>>1);  offs = (pict_struct==BOTTOM_FIELD) ? width : 0;  calcSNR1(org[0]+offs,rec[0]+offs,width2,w,h,&v,&e);  fprintf(statfile,"Y: variance=%4.4g, MSE=%3.3g (%3.3g dB), SNR=%3.3g dB\n",    v, e, 10.0*log10(255.0*255.0/e), 10.0*log10(v/e));
  if (chroma_format!=CHROMA444)  {    w >>= 1;    offs >>= 1;  }  if (chroma_format==CHROMA420)    h >>= 1;
  calcSNR1(org[1]+offs,rec[1]+offs,chrom_width2,w,h,&v,&e);
  fprintf(statfile,"U: variance=%4.4g, MSE=%3.3g (%3.3g dB), SNR=%3.3g dB\n",    v, e, 10.0*log10(255.0*255.0/e), 10.0*log10(v/e));  calcSNR1(org[2]+offs,rec[2]+offs,chrom_width2,w,h,&v,&e);  fprintf(statfile,"V: variance=%4.4g, MSE=%3.3g (%3.3g dB), SNR=%3.3g dB\n",    v, e, 10.0*log10(255.0*255.0/e), 10.0*log10(v/e));
}static void calcSNR1(org,rec,lx,w,h,pv,pe)unsigned char *org;unsigned char *rec;int lx,w,h;double *pv,*pe;{  int i, j;  double v1, s1, s2, e2;  s1 = s2 = e2 = 0.0;  for (j=0; j<h; j++)  {    for (i=0; i<w; i++)    {      v1 = org[i];      s1+= v1;      s2+= v1*v1;      v1-= rec[i];      e2+= v1*v1;    }    org += lx;    rec += lx;  }  s1 /= w*h;  s2 /= w*h;  e2 /= w*h;  /* 防止0作分母的現(xiàn)象出現(xiàn)*/  if(e2==0.0)    e2 = 0.00001;  *pv = s2 - s1*s1;   *pe = e2;         }void stats(){  int i, j, k, nmb, mb_type;  int n_skipped, n_intra, n_ncoded, n_blocks, n_interp, n_forward, n_backward;  struct mbinfo *mbi;  nmb = mb_width*mb_height2;  n_skipped=n_intra=n_ncoded=n_blocks=n_interp=n_forward=n_backward=0;  for (k=0; k<nmb; k++)  {    mbi = mbinfo+k;    if (mbi->skipped)      n_skipped++;    else if (mbi->mb_type & MB_INTRA)      n_intra++;    else if (!(mbi->mb_type & MB_PATTERN))      n_ncoded++;    for (i=0; i<block_count; i++)      if (mbi->cbp & (1<<i))        n_blocks++;    if (mbi->mb_type & MB_FORWARD)    {      if (mbi->mb_type & MB_BACKWARD)        n_interp++;      else        n_forward++;    }    else if (mbi->mb_type & MB_BACKWARD)      n_backward++;  }  fprintf(statfile,"\npicture statistics:\n");  fprintf(statfile," # of intra coded macroblocks:  %4d (%.1f%%)\n",    n_intra,100.0*(double)n_intra/nmb);  fprintf(statfile," # of coded blocks:             %4d (%.1f%%)\n",    n_blocks,100.0*(double)n_blocks/(block_count*nmb));  fprintf(statfile," # of not coded macroblocks:    %4d (%.1f%%)\n",    n_ncoded,100.0*(double)n_ncoded/nmb);  fprintf(statfile," # of skipped macroblocks:      %4d (%.1f%%)\n",    n_skipped,100.0*(double)n_skipped/nmb);  fprintf(statfile," # of forw. pred. macroblocks:  %4d (%.1f%%)\n",    n_forward,100.0*(double)n_forward/nmb);  fprintf(statfile," # of backw. pred. macroblocks: %4d (%.1f%%)\n",    n_backward,100.0*(double)n_backward/nmb);  fprintf(statfile," # of interpolated macroblocks: %4d (%.1f%%)\n",    n_interp,100.0*(double)n_interp/nmb);  fprintf(statfile,"\nmacroblock_type map:\n");
  k = 0;  for (j=0; j<mb_height2; j++)  {    for (i=0; i<mb_width; i++)    {      mbi = mbinfo + k;      mb_type = mbi->mb_type;      if (mbi->skipped)        putc('S',statfile);      else if (mb_type & MB_INTRA)        putc('I',statfile);      else switch (mb_type & (MB_FORWARD|MB_BACKWARD))      {      case MB_FORWARD:        putc(mbi->motion_type==MC_FIELD ? 'f' :             mbi->motion_type==MC_DMV   ? 'p' :                                          'F',statfile); break;      case MB_BACKWARD:        putc(mbi->motion_type==MC_FIELD ? 'b' :                                          'B',statfile); break;      case MB_FORWARD|MB_BACKWARD:        putc(mbi->motion_type==MC_FIELD ? 'd' :                                          'D',statfile); break;      default:        putc('0',statfile); break;      }      if (mb_type & MB_QUANT)        putc('Q',statfile);      else if (mb_type & (MB_PATTERN|MB_INTRA))        putc(' ',statfile);      else        putc('N',statfile);      putc(' ',statfile);      k++;    }    putc('\n',statfile);  }  fprintf(statfile,"\nmquant map:\n");  k=0;  for (j=0; j<mb_height2; j++)  {    for (i=0; i<mb_width; i++)    {      if (i==0 || mbinfo[k].mquant!=mbinfo[k-1].mquant)        fprintf(statfile,"%3d",mbinfo[k].mquant);      else        fprintf(statfile,"   ");      k++;    }    putc('\n',statfile);  }#if 0  fprintf(statfile,"\ncbp map:\n");  k=0;  for (j=0; j<mb_height2; j++)  {    for (i=0; i<mb_width; i++)    {      fprintf(statfile,"%02x ",mbinfo[k].cbp);      k++;    }    putc('\n',statfile);  }  if (pict_struct==FRAME_PICTURE && !frame_pred_dct)  {    fprintf(statfile,"\ndct_type map:\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & (MB_PATTERN|MB_INTRA))          fprintf(statfile,"%d  ",mbinfo[k].dct_type);        else          fprintf(statfile,"   ");          k++;      }      putc('\n',statfile);    }  }  if (pict_type!=I_TYPE)  {    fprintf(statfile,"\nforward motion vectors (first vector, horizontal):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_FORWARD)          fprintf(statfile,"%4d",mbinfo[k].MV[0][0][0]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nforward motion vectors (first vector, vertical):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_FORWARD)          fprintf(statfile,"%4d",mbinfo[k].MV[0][0][1]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nforward motion vectors (second vector, horizontal):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_FORWARD            && ((pict_struct==FRAME_PICTURE && mbinfo[k].motion_type==MC_FIELD) ||                (pict_struct!=FRAME_PICTURE && mbinfo[k].motion_type==MC_16X8)))          fprintf(statfile,"%4d",mbinfo[k].MV[1][0][0]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nforward motion vectors (second vector, vertical):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_FORWARD            && ((pict_struct==FRAME_PICTURE && mbinfo[k].motion_type==MC_FIELD) ||                (pict_struct!=FRAME_PICTURE && mbinfo[k].motion_type==MC_16X8)))          fprintf(statfile,"%4d",mbinfo[k].MV[1][0][1]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }  }      if (pict_type==B_TYPE)  {    fprintf(statfile,"\nbackward motion vectors (first vector, horizontal):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_BACKWARD)          fprintf(statfile,"%4d",mbinfo[k].MV[0][1][0]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nbackward motion vectors (first vector, vertical):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_BACKWARD)          fprintf(statfile,"%4d",mbinfo[k].MV[0][1][1]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nbackward motion vectors (second vector, horizontal):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_BACKWARD            && ((pict_struct==FRAME_PICTURE && mbinfo[k].motion_type==MC_FIELD) ||                (pict_struct!=FRAME_PICTURE && mbinfo[k].motion_type==MC_16X8)))          fprintf(statfile,"%4d",mbinfo[k].MV[1][1][0]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }    fprintf(statfile,"\nbackward motion vectors (second vector, vertical):\n");    k=0;    for (j=0; j<mb_height2; j++)    {      for (i=0; i<mb_width; i++)      {        if (mbinfo[k].mb_type & MB_BACKWARD            && ((pict_struct==FRAME_PICTURE && mbinfo[k].motion_type==MC_FIELD) ||                (pict_struct!=FRAME_PICTURE && mbinfo[k].motion_type==MC_16X8)))          fprintf(statfile,"%4d",mbinfo[k].MV[1][1][1]);        else          fprintf(statfile,"   .");          k++;      }      putc('\n',statfile);    }  }#endif    #if 0  /* useful for debugging */  fprintf(statfile,"\nmacroblock info dump:\n");  k=0;  for (j=0; j<mb_height2; j++)  {    for (i=0; i<mb_width; i++)    {      fprintf(statfile,"%d: %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d\n",      k,      mbinfo[k].mb_type,      mbinfo[k].motion_type,      mbinfo[k].dct_type,      mbinfo[k].mquant,      mbinfo[k].cbp,      mbinfo[k].skipped,      mbinfo[k].MV[0][0][0],      mbinfo[k].MV[0][0][1],      mbinfo[k].MV[0][1][0],      mbinfo[k].MV[0][1][1],      mbinfo[k].MV[1][0][0],      mbinfo[k].MV[1][0][1],      mbinfo[k].MV[1][1][0],      mbinfo[k].MV[1][1][1],      mbinfo[k].mv_field_sel[0][0],      mbinfo[k].mv_field_sel[0][1],      mbinfo[k].mv_field_sel[1][0],      mbinfo[k].mv_field_sel[1][1]);      k++;    }  }#endif}