#include "prj.h"
#include <time.h>

main(int argc,char* argv[1])
{
  FILE *fp;
  register short q,p;
  register short j,i;
  register short frame;
  short table[N][N];
  int dc,count,temp;
  short num,step;
  short qn; /* quant size */
  long total_bit;
  clock_t start,end;
  

  set_eigen();
  set_zigzag();
  set_rand(); 


  /* step 1: find the mv */   
  read_source(argv[1]); 
  num=256;
  tot_count=0;
  printf("\nnum: %3d\n",num);
  start=clock();
  for(frame=1;frame<FRAME;frame++) 
    for(q=0;q<LY;q++)
      for(p=0;p<LX;p++)
	if(mv2(frame,q*N2,p*N2,num)){
	  printf("error in motion estimation\n");
	  printf("frame=%d,q=%d,p=%d",frame,q,p);
	  exit (1);
	}
  end=clock();
  printf("total time: %11d\n",end-start);
  printf("total MAD: %8d\n",tot_count);
  
  /* step 2: quantize frame 0 and recover it,INTRA */
  qn=8;
  for(q=0;q<2*LY;q++)
    for(p=0;p<2*LX;p++){

      for(j=0;j<N;j++)
	for(i=0;i<N;i++)
	  table[j][i]=picel[0][q*N+j][p*N+i];

      dct(table,dct_value[0][q*LY+p]);
      dc=dct_value[0][q*LY+p][0];
      quant(qn,dct_value[0][q*LY+p]);
      dct_value[0][q*LY+p][0]=dc/16;

      bitlen[0][q*LY+p]=8+tcoeff(dct_value[0][q*LY+p]+1,N*N-1);

      dc=dct_value[0][q*LY+p][0];
      iquant(qn,dct_value[0][q*LY+p]);
      dct_value[0][q*LY+p][0]=16*dc+1;
      idct(table,dct_value[0][q*LY+p]);

      for(j=0;j<N;j++)
	for(i=0;i<N;i++)
	  picel[0][q*N+j][p*N+i]=(table[j][i]>255)? 255:table[j][i];
    }



  /* step 3: other frame one by one */ 
  total_bit=0;
  for(frame=1;frame<FRAME;frame++)
    for(q=0;q<2*LY;q++)
      for(p=0;p<2*LX;p++){
	
	residue(frame,q*N,p*N,mv_y[frame][q/2][p/2],mv_x[frame][q/2][p/2],table);
        dct(table,dct_value[frame][q*LY+p]);
        quant(qn,dct_value[frame][q*LY+p]);

        bitlen[frame][q*LY+p]=tcoeff(dct_value[frame][q*LY+p],N*N);
	total_bit+=bitlen[frame][q*LY+p];

        iquant(qn,dct_value[frame][q*LY+p]);
        idct(table,dct_value[frame][q*LY+p]);
	iresidue(frame,q*N,p*N,mv_y[frame][q/2][p/2],mv_x[frame][q/2][p/2],table);
	
      }

  printf("total_bit:%11d\n",total_bit);	


}