#include "prj.h"
#include "stdlib.h"

/* make a dct to a block */
int cod_intra(short qn,short f,short q,short p,short table[N][N])
{
  unsigned short i,j;
  unsigned short dc;

  for(j=0;j<N;j++)
    for(i=0;i<N;i++)
      table[j][i]=picel[f][q*N+j][p*N+i];
  
  dct(table,dct_value[f][q*LY+p]);
  dc=dct_value[f][q*LY+p][0];
  quant(qn,dct_value[f][q*LY+p]);
  dct_value[f][q*LY+p][0]=dc/16;
  
  return bitlen[f][q*LY+p]=8+tcoeff(dct_value[f][q*LY+p]+1,N*N-1);
}

decod_intra(short qn,short f,short q,short p,short table[N][N])
{
  unsigned short i,j;
  short dc;

  dc=dct_value[f][q*LY+p][0];
  iquant(qn,dct_value[f][q*LY+p]);

  dct_value[f][q*LY+p][0]=16*dc+1;
  idct(table,dct_value[f][q*LY+p]);
  for(j=0;j<N;j++)
    for(i=0;i<N;i++)
      picel[f][q*N+j][p*N+i]=(table[j][i]>255)? 255:table[j][i];
    
}

int cod_inter(short qn,short f,short q,short p,short table[N][N])
{
  unsigned short i,j;
 
  residue(f,q*N,p*N,mv_y[f][q/2][p/2],mv_x[f][q/2][p/2],table);
  dct(table,dct_value[f][q*LY+p]);
  quant(qn,dct_value[f][q*LY+p]);
  return bitlen[f][q*LY+p]=tcoeff(dct_value[f][q*LY+p],N*N);
}

int fake_cod_inter(short qn,short f,short q,short p,short fake_x,short fake_y,short table[N][N])
{
  unsigned short i,j;
  
  residue(f,q*N,p*N,fake_y,fake_x,table);
  dct(table,dct_value[f][q*LY+p]);
  quant(qn,dct_value[f][q*LY+p]);
  return bitlen[f][q*LY+p]=tcoeff(dct_value[f][q*LY+p],N*N);
}

int fake_decod_inter(short qn,short f,short q,short p,short fake_x,short fake_y,short table[N][N])
{
  unsigned short i,j;

  iquant(qn,dct_value[f][q*LY+p]);
  idct(table,dct_value[f][q*LY+p]);
  iresidue(f,q*N,p*N,fake_y,fake_x,table);
}



decod_inter(short qn,short f, short q,short p,short table[N][N])
{
  unsigned short i,j;

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

/*   
main()
{
  unsigned short i,j;
  short table[N][N];
  short mx,my,p,q;
  short stat[N][N];

  p=2; q=3;
  mx=-3; my=-4;
  mv_y[1][q/2][p/2]=my;
  mv_x[1][q/2][p/2]=mx;

  set_eigen();
  set_zigzag();

  for(i=0;i<N;i++)
    for(j=0;j<N;j++)
      picel[0][q*N+j][p*N+i]=20+rand()%160;
    
  for(i=0;i<N;i++)
    for(j=0;j<N;j++)
      picel[1][q*N+j-mx][p*N+i-my]=picel[0][q*N+j][p*N+i]+rand()%10-5;

  printf("\n");
  for(i=0;i<N;i++){
    for(j=0;j<N;j++)
      printf("%3d ",stat[j][i]=(picel[0][q*N+j][p*N+i]<100)? 0:1);
    printf("\n");
  }
  
  cod_intra(2,0,q,p,table);
  for(i=0;i<N;i++)
    for(j=0;j<N;j++)
      picel[0][q*N+j][p*N+i]=0;

  decod_intra(2,0,q,p,table);

  printf("\n");
  for(i=0;i<N;i++){
    for(j=0;j<N;j++)
      printf("%3d ",stat[j][i]=(picel[0][q*N+j][p*N+i]<100)? stat[j][i]:(stat[j][i]+1)%2);
    printf("\n");
  }

  printf("\n");
  for(i=0;i<N;i++){
    for(j=0;j<N;j++)
      printf("%3d ",stat[j][i]=(picel[1][q*N+j-mx][p*N+i-my]<100)? 0:1);
    printf("\n");
  }

  cod_inter(2,1,q,p,table);
  decod_inter(2,1,q,p,table);
 
  printf("\n");
  for(i=0;i<N;i++){
    for(j=0;j<N;j++)
      printf("%3d ",stat[j][i]=(picel[1][q*N+j-mx][p*N+i-my]<100)? stat[j][i]:(stat[j][i]+1)%2);
    printf("\n");
  }

}  
  

*/