transform8x8.c

包含FRExt部分的JM源码

    img->mpr[0+ioff][2+joff] =
    img->mpr[1+ioff][4+joff] =
    img->mpr[2+ioff][6+joff] = (P_R + P_Z + 2*P_Q + 2) >> 2;
    img->mpr[0+ioff][3+joff] =
    img->mpr[1+ioff][5+joff] =
    img->mpr[2+ioff][7+joff] = (P_S + P_Q + 2*P_R + 2) >> 2;
    img->mpr[0+ioff][4+joff] =
    img->mpr[1+ioff][6+joff] = (P_T + P_R + 2*P_S + 2) >> 2;
    img->mpr[0+ioff][5+joff] =
    img->mpr[1+ioff][7+joff] = (P_U + P_S + 2*P_T + 2) >> 2;
    img->mpr[0+ioff][6+joff] = (P_V + P_T + 2*P_U + 2) >> 2;
    img->mpr[0+ioff][7+joff] = (P_W + P_U + 2*P_V + 2) >> 2;
    break;

  case  HOR_DOWN_PRED:/* diagonal prediction -22.5 deg to horizontal plane */
    if ((!block_available_up)||(!block_available_left)||(!block_available_up_left))
      printf ("warning: Intra_8x8_Horizontal_Down prediction mode not allowed at mb %d\n",img->current_mb_nr);
    
    img->mpr[0+ioff][0+joff] = 
    img->mpr[2+ioff][1+joff] = 
    img->mpr[4+ioff][2+joff] = 
    img->mpr[6+ioff][3+joff] = (P_Q + P_Z + 1) >> 1;
    img->mpr[0+ioff][1+joff] = 
    img->mpr[2+ioff][2+joff] = 
    img->mpr[4+ioff][3+joff] = 
    img->mpr[6+ioff][4+joff] = (P_R + P_Q + 1) >> 1;
    img->mpr[0+ioff][2+joff] = 
    img->mpr[2+ioff][3+joff] = 
    img->mpr[4+ioff][4+joff] = 
    img->mpr[6+ioff][5+joff] = (P_S + P_R + 1) >> 1;
    img->mpr[0+ioff][3+joff] = 
    img->mpr[2+ioff][4+joff] = 
    img->mpr[4+ioff][5+joff] = 
    img->mpr[6+ioff][6+joff] = (P_T + P_S + 1) >> 1;
    img->mpr[0+ioff][4+joff] = 
    img->mpr[2+ioff][5+joff] = 
    img->mpr[4+ioff][6+joff] = 
    img->mpr[6+ioff][7+joff] = (P_U + P_T + 1) >> 1;
    img->mpr[0+ioff][5+joff] = 
    img->mpr[2+ioff][6+joff] = 
    img->mpr[4+ioff][7+joff] = (P_V + P_U + 1) >> 1;
    img->mpr[0+ioff][6+joff] = 
    img->mpr[2+ioff][7+joff] = (P_W + P_V + 1) >> 1;
    img->mpr[0+ioff][7+joff] = (P_X + P_W + 1) >> 1;
    img->mpr[1+ioff][0+joff] =
    img->mpr[3+ioff][1+joff] =
    img->mpr[5+ioff][2+joff] =
    img->mpr[7+ioff][3+joff] = (P_Q + P_A + 2*P_Z + 2) >> 2;
    img->mpr[1+ioff][1+joff] =
    img->mpr[3+ioff][2+joff] =
    img->mpr[5+ioff][3+joff] =
    img->mpr[7+ioff][4+joff] = (P_Z + P_R + 2*P_Q + 2) >> 2;
    img->mpr[1+ioff][2+joff] =
    img->mpr[3+ioff][3+joff] =
    img->mpr[5+ioff][4+joff] =
    img->mpr[7+ioff][5+joff] = (P_Q + P_S + 2*P_R + 2) >> 2;
    img->mpr[1+ioff][3+joff] =
    img->mpr[3+ioff][4+joff] =
    img->mpr[5+ioff][5+joff] =
    img->mpr[7+ioff][6+joff] = (P_R + P_T + 2*P_S + 2) >> 2;
    img->mpr[1+ioff][4+joff] =
    img->mpr[3+ioff][5+joff] =
    img->mpr[5+ioff][6+joff] =
    img->mpr[7+ioff][7+joff] = (P_S + P_U + 2*P_T + 2) >> 2;
    img->mpr[1+ioff][5+joff] =
    img->mpr[3+ioff][6+joff] =
    img->mpr[5+ioff][7+joff] = (P_T + P_V + 2*P_U + 2) >> 2;
    img->mpr[1+ioff][6+joff] =
    img->mpr[3+ioff][7+joff] = (P_U + P_W + 2*P_V + 2) >> 2;
    img->mpr[1+ioff][7+joff] = (P_V + P_X + 2*P_W + 2) >> 2;
    img->mpr[2+ioff][0+joff] = 
    img->mpr[4+ioff][1+joff] = 
    img->mpr[6+ioff][2+joff] = (P_Z + P_B + 2*P_A + 2) >> 2;
    img->mpr[3+ioff][0+joff] = 
    img->mpr[5+ioff][1+joff] = 
    img->mpr[7+ioff][2+joff] = (P_A + P_C + 2*P_B + 2) >> 2;
    img->mpr[4+ioff][0+joff] = 
    img->mpr[6+ioff][1+joff] = (P_B + P_D + 2*P_C + 2) >> 2;
    img->mpr[5+ioff][0+joff] = 
    img->mpr[7+ioff][1+joff] = (P_C + P_E + 2*P_D + 2) >> 2;
    img->mpr[6+ioff][0+joff] = (P_D + P_F + 2*P_E + 2) >> 2;
    img->mpr[7+ioff][0+joff] = (P_E + P_G + 2*P_F + 2) >> 2;
    break;

  case  HOR_UP_PRED:/* diagonal prediction -22.5 deg to horizontal plane */
    if (!block_available_left)
      printf ("warning: Intra_8x8_Horizontal_Up prediction mode not allowed at mb %d\n",img->current_mb_nr);

    img->mpr[0+ioff][0+joff] = (P_Q + P_R + 1) >> 1;
    img->mpr[0+ioff][1+joff] =
    img->mpr[2+ioff][0+joff] = (P_R + P_S + 1) >> 1;
    img->mpr[0+ioff][2+joff] =
    img->mpr[2+ioff][1+joff] =
    img->mpr[4+ioff][0+joff] = (P_S + P_T + 1) >> 1;
    img->mpr[0+ioff][3+joff] =
    img->mpr[2+ioff][2+joff] =
    img->mpr[4+ioff][1+joff] =
    img->mpr[6+ioff][0+joff] = (P_T + P_U + 1) >> 1;
    img->mpr[0+ioff][4+joff] =
    img->mpr[2+ioff][3+joff] =
    img->mpr[4+ioff][2+joff] =
    img->mpr[6+ioff][1+joff] = (P_U + P_V + 1) >> 1;
    img->mpr[0+ioff][5+joff] =
    img->mpr[2+ioff][4+joff] =
    img->mpr[4+ioff][3+joff] =
    img->mpr[6+ioff][2+joff] = (P_V + P_W + 1) >> 1;
    img->mpr[0+ioff][6+joff] =
    img->mpr[2+ioff][5+joff] =
    img->mpr[4+ioff][4+joff] =
    img->mpr[6+ioff][3+joff] = (P_W + P_X + 1) >> 1;
    img->mpr[6+ioff][4+joff] =
      img->mpr[7+ioff][4+joff] =
      img->mpr[4+ioff][5+joff] =
      img->mpr[5+ioff][5+joff] =
      img->mpr[6+ioff][5+joff] =
      img->mpr[7+ioff][5+joff] =
      img->mpr[2+ioff][6+joff] =
      img->mpr[3+ioff][6+joff] =
      img->mpr[4+ioff][6+joff] =
      img->mpr[5+ioff][6+joff] =
      img->mpr[6+ioff][6+joff] =
      img->mpr[7+ioff][6+joff] =
      img->mpr[0+ioff][7+joff] =
      img->mpr[1+ioff][7+joff] =
      img->mpr[2+ioff][7+joff] =
      img->mpr[3+ioff][7+joff] =
      img->mpr[4+ioff][7+joff] =
      img->mpr[5+ioff][7+joff] =
      img->mpr[6+ioff][7+joff] =
      img->mpr[7+ioff][7+joff] = P_X;
    img->mpr[1+ioff][6+joff] =
      img->mpr[3+ioff][5+joff] =
      img->mpr[5+ioff][4+joff] =
      img->mpr[7+ioff][3+joff] = (P_W + 3*P_X + 2) >> 2;
    img->mpr[1+ioff][5+joff] =
      img->mpr[3+ioff][4+joff] =
      img->mpr[5+ioff][3+joff] =
      img->mpr[7+ioff][2+joff] = (P_X + P_V + 2*P_W + 2) >> 2;
    img->mpr[1+ioff][4+joff] =
      img->mpr[3+ioff][3+joff] =
      img->mpr[5+ioff][2+joff] =
      img->mpr[7+ioff][1+joff] = (P_W + P_U + 2*P_V + 2) >> 2;
    img->mpr[1+ioff][3+joff] =
      img->mpr[3+ioff][2+joff] =
      img->mpr[5+ioff][1+joff] =
      img->mpr[7+ioff][0+joff] = (P_V + P_T + 2*P_U + 2) >> 2;
    img->mpr[1+ioff][2+joff] =
      img->mpr[3+ioff][1+joff] =
      img->mpr[5+ioff][0+joff] = (P_U + P_S + 2*P_T + 2) >> 2;
    img->mpr[1+ioff][1+joff] =
      img->mpr[3+ioff][0+joff] = (P_T + P_R + 2*P_S + 2) >> 2;
    img->mpr[1+ioff][0+joff] = (P_S + P_Q + 2*P_R + 2) >> 2;
    break;
    
  default:
    printf("Error: illegal intra_4x4 prediction mode: %d\n",predmode);
    return SEARCH_SYNC;
    break;
  }
  return DECODING_OK;
}



/*! 
 *************************************************************************************
 * \brief
 *    Prefiltering for Intra8x8 prediction
 *************************************************************************************
 */
void LowPassForIntra8x8Pred(int *PredPel, int block_up_left, int block_up, int block_left)
{
  int i;
  int LoopArray[25];
 

  for(i = 0; i < 25; i++)
     LoopArray[i] = PredPel[i] ;

 	if(block_up)
	{
		if(block_up_left) 
		{
			LoopArray[1] = ((&P_Z)[0] + ((&P_Z)[1]<<1) + (&P_Z)[2] + 2)>>2;
		}
		else
			LoopArray[1] = ((&P_Z)[1] + ((&P_Z)[1]<<1) + (&P_Z)[2] + 2)>>2; 


		for(i = 2; i <16; i++)
		{
			LoopArray[i] = ((&P_Z)[i-1] + ((&P_Z)[i]<<1) + (&P_Z)[i+1] + 2)>>2;
		}
		LoopArray[16] = (P_P + (P_P<<1) + P_O + 2)>>2;
	}

	if(block_up_left) 
	{
		
		if(block_up && block_left)
		{
				LoopArray[0] = (P_Q + (P_Z<<1) + P_A +2)>>2;
		}
		else
		{
			if(block_up)
				LoopArray[0] = (P_Z + (P_Z<<1) + P_A +2)>>2;
			else
				if(block_left)
					LoopArray[0] = (P_Z + (P_Z<<1) + P_Q +2)>>2;
		}

	}

	if(block_left)
	{
		if(block_up_left)
			LoopArray[17] = (P_Z + (P_Q<<1) + P_R + 2)>>2; 
		else
			LoopArray[17] = (P_Q + (P_Q<<1) + P_R + 2)>>2;

		for(i = 18; i <24; i++)
		{
			LoopArray[i] = ((&P_Z)[i-1] + ((&P_Z)[i]<<1) + (&P_Z)[i+1] + 2)>>2;
		}
		LoopArray[24] = (P_W + (P_X<<1) + P_X + 2)>>2;
	}

  for(i = 0; i < 25; i++)
    PredPel[i] = LoopArray[i];
}



/*!
 ***********************************************************************
 * \brief
 *    Inverse 8x8 transformation
 ***********************************************************************
 */
void itrans8x8(struct img_par *img, //!< image parameters
              int ioff,            //!< index to 4x4 block
              int joff)            //!<
{
  int i,j;
  int m6[8][8];
  Boolean lossless_qpprime = ((img->qp + img->bitdepth_luma_qp_scale)==0 && img->lossless_qpprime_flag==1); // S.SUN 06072004

  int residue_transform_flag = img->residue_transform_flag;// D.BIRINOV SAMSUNG AIT


  // HB 10062004 start
  for( i=0; i<8 && !lossless_qpprime; i++)  // S.SUN 06072004
  {
    int a[8], b[8];
    a[0] = img->m7[ioff + 0][joff + i] + img->m7[ioff + 4][joff + i];
    a[4] = img->m7[ioff + 0][joff + i] - img->m7[ioff + 4][joff + i];
    a[2] = (img->m7[ioff + 2][joff + i]>>1) - img->m7[ioff + 6][joff + i];
    a[6] = img->m7[ioff + 2][joff + i] + (img->m7[ioff + 6][joff + i]>>1);

    b[0] = a[0] + a[6];
    b[2] = a[4] + a[2];
    b[4] = a[4] - a[2];
    b[6] = a[0] - a[6];

    a[1] = -img->m7[ioff + 3][joff + i] + img->m7[ioff + 5][joff + i] - img->m7[ioff + 7][joff + i] - (img->m7[ioff + 7][joff + i]>>1);
    a[3] = img->m7[ioff + 1][joff + i] + img->m7[ioff + 7][joff + i] - img->m7[ioff + 3][joff + i] - (img->m7[ioff + 3][joff + i]>>1);
    a[5] = -img->m7[ioff + 1][joff + i] + img->m7[ioff + 7][joff + i] + img->m7[ioff + 5][joff + i] + (img->m7[ioff + 5][joff + i]>>1);
    a[7] = img->m7[ioff + 3][joff + i] + img->m7[ioff + 5][joff + i] + img->m7[ioff + 1][joff + i] + (img->m7[ioff + 1][joff + i]>>1);

    b[1] = a[1] + (a[7]>>2);
//    b[7] = -(a[1]>>2 + 0) + a[7];  KS: do we need to add zero?
    b[7] = -(a[1]>>2) + a[7];
    b[3] = a[3] + (a[5]>>2);
    b[5] = (a[3]>>2) - a[5];

    m6[0][i] = b[0] + b[7];
    m6[1][i] = b[2] + b[5];
    m6[2][i] = b[4] + b[3];
    m6[3][i] = b[6] + b[1];
    m6[4][i] = b[6] - b[1];
    m6[5][i] = b[4] - b[3];
    m6[6][i] = b[2] - b[5];
    m6[7][i] = b[0] - b[7];
  }
  for( i=0; i<8 && !lossless_qpprime; i++) // S.SUN 06072004
  {
    int a[8], b[8];
    a[0] = m6[i][0] + m6[i][4];
    a[4] = m6[i][0] - m6[i][4];
    a[2] = (m6[i][2]>>1) - m6[i][6];
    a[6] = m6[i][2] + (m6[i][6]>>1);

    b[0] = a[0] + a[6];
    b[2] = a[4] + a[2];
    b[4] = a[4] - a[2];
    b[6] = a[0] - a[6];

    a[1] = -m6[i][3] + m6[i][5] - m6[i][7] - (m6[i][7]>>1);
    a[3] = m6[i][1] + m6[i][7] - m6[i][3] - (m6[i][3]>>1);
    a[5] = -m6[i][1] + m6[i][7] + m6[i][5] + (m6[i][5]>>1);
    a[7] = m6[i][3] + m6[i][5] + m6[i][1] + (m6[i][1]>>1);

    b[1] = a[1] + (a[7]>>2);
    b[7] = -(a[1]>>2) + a[7];
    b[3] = a[3] + (a[5]>>2);
    b[5] = (a[3]>>2) - a[5];

    img->m7[ioff + i][joff + 0] = b[0] + b[7];
    img->m7[ioff + i][joff + 1] = b[2] + b[5];
    img->m7[ioff + i][joff + 2] = b[4] + b[3];
    img->m7[ioff + i][joff + 3] = b[6] + b[1];
    img->m7[ioff + i][joff + 4] = b[6] - b[1];
    img->m7[ioff + i][joff + 5] = b[4] - b[3];
    img->m7[ioff + i][joff + 6] = b[2] - b[5];
    img->m7[ioff + i][joff + 7] = b[0] - b[7];
  }
  // HB 10062004 end
  for( i=0; i<8; i++)
  {
    for( j=0; j<8; j++)
    {
      // Residue Color Transform
      if(!residue_transform_flag){
        if(lossless_qpprime)  // S.SUN 06072004
          img->m7[i+ioff][j+joff] =min(img->max_imgpel_value,max(0,img->m7[ioff + i][joff + j]+(long)img->mpr[i+ioff][j+joff]));
        else
          img->m7[i+ioff][j+joff] =min(img->max_imgpel_value,max(0,(img->m7[ioff + i][joff + j]+((long)img->mpr[i+ioff][j+joff] << DQ_BITS_8)+DQ_ROUND_8)>>DQ_BITS_8));
      }
	  else{
        if(lossless_qpprime)  // S.SUN 06072004
          img->m7[i+ioff][j+joff] = img->m7[ioff + i][joff + j];
        else
          img->m7[i+ioff][j+joff] =(img->m7[ioff + i][joff + j]+DQ_ROUND_8)>>DQ_BITS_8;
      }
      // D.BIRINOV SAMSUNG AIT
    }
  }

/*
  // Decoded block moved to frame memory
  for (j=0; j < 2*BLOCK_SIZE; j++)
    for (i=0; i < 2*BLOCK_SIZE; i++)
      enc_picture->imgY[img->pix_y+joff+j][img->pix_x+ioff+i]=img->m7[i][j];
*/
        
}