transform8x8.c

包含FRExt部分的JM源码

    P_X = imgY[pix_a[7].pos_y][pix_a[7].pos_x];
  }
  else
  {
    P_Q = P_R = P_S = P_T = P_U = P_V = P_W = P_X = img->dc_pred_value;
  }

  if (block_available_up_left)
  {
    P_Z = imgY[pix_d.pos_y][pix_d.pos_x];
  }
  else
  {
    P_Z = img->dc_pred_value;
  }
  
  LowPassForIntra8x8Pred(&(P_Z), block_available_up_left, block_available_up, block_available_left);

//img->mpr[x][y]
  switch(predmode)
  {
  case DC_PRED:
    s0 = 0;
    if (block_available_up && block_available_left)
    {   
      // no edge
      s0 = (P_A + P_B + P_C + P_D + P_E + P_F + P_G + P_H + P_Q + P_R + P_S + P_T + P_U + P_V + P_W + P_X + 8) >> 4;
    }
    else if (!block_available_up && block_available_left)
    {
      // upper edge
      s0 = (P_Q + P_R + P_S + P_T + P_U + P_V + P_W + P_X + 4) >> 3;             
    }
    else if (block_available_up && !block_available_left)
    {
      // left edge
      s0 = (P_A + P_B + P_C + P_D + P_E + P_F + P_G + P_H + 4) >> 3;             
    }
    else //if (!block_available_up && !block_available_left)
    {
      // top left corner, nothing to predict from
      s0 = img->dc_pred_value;                           
    }
    for(i = 0; i < 2*BLOCK_SIZE; i++)
      for(j = 0; j < 2*BLOCK_SIZE; j++)
        img->mpr[i+ioff][j+joff] = s0;
    break;

  case VERT_PRED:
    if (!block_available_up)
      printf ("warning: Intra_8x8_Vertical prediction mode not allowed at mb %d\n",img->current_mb_nr);

    for (i=0; i < 2*BLOCK_SIZE; i++)
    {
      img->mpr[i+ioff][0+joff] = 
      img->mpr[i+ioff][1+joff] = 
      img->mpr[i+ioff][2+joff] = 
      img->mpr[i+ioff][3+joff] = 
      img->mpr[i+ioff][4+joff] = 
      img->mpr[i+ioff][5+joff] = 
      img->mpr[i+ioff][6+joff] = 
      img->mpr[i+ioff][7+joff] = (&P_A)[i];
    }
    break;
  case HOR_PRED:
    if (!block_available_left)
      printf ("warning: Intra_8x8_Horizontal prediction mode not allowed at mb %d\n",img->current_mb_nr);

    for (j=0; j < 2*BLOCK_SIZE; j++)
    {
      img->mpr[0+ioff][j+joff]  = 
      img->mpr[1+ioff][j+joff]  = 
      img->mpr[2+ioff][j+joff]  = 
      img->mpr[3+ioff][j+joff]  = 
      img->mpr[4+ioff][j+joff]  = 
      img->mpr[5+ioff][j+joff]  = 
      img->mpr[6+ioff][j+joff]  = 
      img->mpr[7+ioff][j+joff]  = (&P_Q)[j];
    }
    break;

  case DIAG_DOWN_LEFT_PRED:
    if (!block_available_up)
      printf ("warning: Intra_8x8_Diagonal_Down_Left prediction mode not allowed at mb %d\n",img->current_mb_nr);
    // Mode DIAG_DOWN_LEFT_PRED
    img->mpr[0+ioff][0+joff] = (P_A + P_C + 2*(P_B) + 2) >> 2;
    img->mpr[0+ioff][1+joff] = 
    img->mpr[1+ioff][0+joff] = (P_B + P_D + 2*(P_C) + 2) >> 2;
    img->mpr[0+ioff][2+joff] =
    img->mpr[1+ioff][1+joff] =
    img->mpr[2+ioff][0+joff] = (P_C + P_E + 2*(P_D) + 2) >> 2;
    img->mpr[0+ioff][3+joff] = 
    img->mpr[1+ioff][2+joff] = 
    img->mpr[2+ioff][1+joff] = 
    img->mpr[3+ioff][0+joff] = (P_D + P_F + 2*(P_E) + 2) >> 2;
    img->mpr[0+ioff][4+joff] = 
    img->mpr[1+ioff][3+joff] = 
    img->mpr[2+ioff][2+joff] = 
    img->mpr[3+ioff][1+joff] = 
    img->mpr[4+ioff][0+joff] = (P_E + P_G + 2*(P_F) + 2) >> 2;
    img->mpr[0+ioff][5+joff] = 
    img->mpr[1+ioff][4+joff] = 
    img->mpr[2+ioff][3+joff] = 
    img->mpr[3+ioff][2+joff] = 
    img->mpr[4+ioff][1+joff] = 
    img->mpr[5+ioff][0+joff] = (P_F + P_H + 2*(P_G) + 2) >> 2;
    img->mpr[0+ioff][6+joff] = 
    img->mpr[1+ioff][5+joff] = 
    img->mpr[2+ioff][4+joff] = 
    img->mpr[3+ioff][3+joff] = 
    img->mpr[4+ioff][2+joff] = 
    img->mpr[5+ioff][1+joff] = 
    img->mpr[6+ioff][0+joff] = (P_G + P_I + 2*(P_H) + 2) >> 2;
    img->mpr[0+ioff][7+joff] = 
    img->mpr[1+ioff][6+joff] = 
    img->mpr[2+ioff][5+joff] = 
    img->mpr[3+ioff][4+joff] = 
    img->mpr[4+ioff][3+joff] = 
    img->mpr[5+ioff][2+joff] = 
    img->mpr[6+ioff][1+joff] = 
    img->mpr[7+ioff][0+joff] = (P_H + P_J + 2*(P_I) + 2) >> 2;
    img->mpr[1+ioff][7+joff] = 
    img->mpr[2+ioff][6+joff] = 
    img->mpr[3+ioff][5+joff] = 
    img->mpr[4+ioff][4+joff] = 
    img->mpr[5+ioff][3+joff] = 
    img->mpr[6+ioff][2+joff] = 
    img->mpr[7+ioff][1+joff] = (P_I + P_K + 2*(P_J) + 2) >> 2;
    img->mpr[2+ioff][7+joff] = 
    img->mpr[3+ioff][6+joff] = 
    img->mpr[4+ioff][5+joff] = 
    img->mpr[5+ioff][4+joff] = 
    img->mpr[6+ioff][3+joff] = 
    img->mpr[7+ioff][2+joff] = (P_J + P_L + 2*(P_K) + 2) >> 2;
    img->mpr[3+ioff][7+joff] = 
    img->mpr[4+ioff][6+joff] = 
    img->mpr[5+ioff][5+joff] = 
    img->mpr[6+ioff][4+joff] = 
    img->mpr[7+ioff][3+joff] = (P_K + P_M + 2*(P_L) + 2) >> 2;
    img->mpr[4+ioff][7+joff] = 
    img->mpr[5+ioff][6+joff] = 
    img->mpr[6+ioff][5+joff] = 
    img->mpr[7+ioff][4+joff] = (P_L + P_N + 2*(P_M) + 2) >> 2;
    img->mpr[5+ioff][7+joff] = 
    img->mpr[6+ioff][6+joff] = 
    img->mpr[7+ioff][5+joff] = (P_M + P_O + 2*(P_N) + 2) >> 2;
    img->mpr[6+ioff][7+joff] = 
    img->mpr[7+ioff][6+joff] = (P_N + P_P + 2*(P_O) + 2) >> 2;
    img->mpr[7+ioff][7+joff] = (P_O + 3*(P_P) + 2) >> 2;
    break;

  case VERT_LEFT_PRED:
    if (!block_available_up)
      printf ("warning: Intra_4x4_Vertical_Left prediction mode not allowed at mb %d\n",img->current_mb_nr);

    img->mpr[0+ioff][0+joff] = (P_A + P_B + 1) >> 1;
    img->mpr[1+ioff][0+joff] = 
    img->mpr[0+ioff][2+joff] = (P_B + P_C + 1) >> 1;
    img->mpr[2+ioff][0+joff] = 
    img->mpr[1+ioff][2+joff] = 
    img->mpr[0+ioff][4+joff] = (P_C + P_D + 1) >> 1;
    img->mpr[3+ioff][0+joff] = 
    img->mpr[2+ioff][2+joff] = 
    img->mpr[1+ioff][4+joff] = 
    img->mpr[0+ioff][6+joff] = (P_D + P_E + 1) >> 1;
    img->mpr[4+ioff][0+joff] = 
    img->mpr[3+ioff][2+joff] = 
    img->mpr[2+ioff][4+joff] = 
    img->mpr[1+ioff][6+joff] = (P_E + P_F + 1) >> 1;
    img->mpr[5+ioff][0+joff] = 
    img->mpr[4+ioff][2+joff] = 
    img->mpr[3+ioff][4+joff] = 
    img->mpr[2+ioff][6+joff] = (P_F + P_G + 1) >> 1;
    img->mpr[6+ioff][0+joff] = 
    img->mpr[5+ioff][2+joff] = 
    img->mpr[4+ioff][4+joff] = 
    img->mpr[3+ioff][6+joff] = (P_G + P_H + 1) >> 1;
    img->mpr[7+ioff][0+joff] = 
    img->mpr[6+ioff][2+joff] = 
    img->mpr[5+ioff][4+joff] = 
    img->mpr[4+ioff][6+joff] = (P_H + P_I + 1) >> 1;
    img->mpr[7+ioff][2+joff] = 
    img->mpr[6+ioff][4+joff] = 
    img->mpr[5+ioff][6+joff] = (P_I + P_J + 1) >> 1;
    img->mpr[7+ioff][4+joff] = 
    img->mpr[6+ioff][6+joff] = (P_J + P_K + 1) >> 1;
    img->mpr[7+ioff][6+joff] = (P_K + P_L + 1) >> 1;
    img->mpr[0+ioff][1+joff] = (P_A + P_C + 2*P_B + 2) >> 2;
    img->mpr[1+ioff][1+joff] = 
    img->mpr[0+ioff][3+joff] = (P_B + P_D + 2*P_C + 2) >> 2;
    img->mpr[2+ioff][1+joff] = 
    img->mpr[1+ioff][3+joff] = 
    img->mpr[0+ioff][5+joff] = (P_C + P_E + 2*P_D + 2) >> 2;
    img->mpr[3+ioff][1+joff] = 
    img->mpr[2+ioff][3+joff] = 
    img->mpr[1+ioff][5+joff] = 
    img->mpr[0+ioff][7+joff] = (P_D + P_F + 2*P_E + 2) >> 2;
    img->mpr[4+ioff][1+joff] = 
    img->mpr[3+ioff][3+joff] = 
    img->mpr[2+ioff][5+joff] = 
    img->mpr[1+ioff][7+joff] = (P_E + P_G + 2*P_F + 2) >> 2;
    img->mpr[5+ioff][1+joff] = 
    img->mpr[4+ioff][3+joff] = 
    img->mpr[3+ioff][5+joff] = 
    img->mpr[2+ioff][7+joff] = (P_F + P_H + 2*P_G + 2) >> 2;
    img->mpr[6+ioff][1+joff] = 
    img->mpr[5+ioff][3+joff] = 
    img->mpr[4+ioff][5+joff] = 
    img->mpr[3+ioff][7+joff] = (P_G + P_I + 2*P_H + 2) >> 2;
    img->mpr[7+ioff][1+joff] = 
    img->mpr[6+ioff][3+joff] = 
    img->mpr[5+ioff][5+joff] = 
    img->mpr[4+ioff][7+joff] = (P_H + P_J + 2*P_I + 2) >> 2;
    img->mpr[7+ioff][3+joff] = 
    img->mpr[6+ioff][5+joff] = 
    img->mpr[5+ioff][7+joff] = (P_I + P_K + 2*P_J + 2) >> 2;
    img->mpr[7+ioff][5+joff] = 
    img->mpr[6+ioff][7+joff] = (P_J + P_L + 2*P_K + 2) >> 2;
    img->mpr[7+ioff][7+joff] = (P_K + P_M + 2*P_L + 2) >> 2;
    break;


  case DIAG_DOWN_RIGHT_PRED:
    if ((!block_available_up)||(!block_available_left)||(!block_available_up_left))
      printf ("warning: Intra_8x8_Diagonal_Down_Right prediction mode not allowed at mb %d\n",img->current_mb_nr);

    // Mode DIAG_DOWN_RIGHT_PRED
    img->mpr[0+ioff][7+joff] = (P_X + P_V + 2*(P_W) + 2) >> 2;
    img->mpr[0+ioff][6+joff] = 
    img->mpr[1+ioff][7+joff] = (P_W + P_U + 2*(P_V) + 2) >> 2;
    img->mpr[0+ioff][5+joff] = 
    img->mpr[1+ioff][6+joff] = 
    img->mpr[2+ioff][7+joff] = (P_V + P_T + 2*(P_U) + 2) >> 2;
    img->mpr[0+ioff][4+joff] = 
    img->mpr[1+ioff][5+joff] = 
    img->mpr[2+ioff][6+joff] = 
    img->mpr[3+ioff][7+joff] = (P_U + P_S + 2*(P_T) + 2) >> 2;
    img->mpr[0+ioff][3+joff] = 
    img->mpr[1+ioff][4+joff] = 
    img->mpr[2+ioff][5+joff] = 
    img->mpr[3+ioff][6+joff] = 
    img->mpr[4+ioff][7+joff] = (P_T + P_R + 2*(P_S) + 2) >> 2;
    img->mpr[0+ioff][2+joff] = 
    img->mpr[1+ioff][3+joff] = 
    img->mpr[2+ioff][4+joff] = 
    img->mpr[3+ioff][5+joff] = 
    img->mpr[4+ioff][6+joff] = 
    img->mpr[5+ioff][7+joff] = (P_S + P_Q + 2*(P_R) + 2) >> 2;
    img->mpr[0+ioff][1+joff] = 
    img->mpr[1+ioff][2+joff] = 
    img->mpr[2+ioff][3+joff] = 
    img->mpr[3+ioff][4+joff] = 
    img->mpr[4+ioff][5+joff] = 
    img->mpr[5+ioff][6+joff] = 
    img->mpr[6+ioff][7+joff] = (P_R + P_Z + 2*(P_Q) + 2) >> 2;
    img->mpr[0+ioff][0+joff] = 
    img->mpr[1+ioff][1+joff] = 
    img->mpr[2+ioff][2+joff] = 
    img->mpr[3+ioff][3+joff] = 
    img->mpr[4+ioff][4+joff] = 
    img->mpr[5+ioff][5+joff] = 
    img->mpr[6+ioff][6+joff] = 
    img->mpr[7+ioff][7+joff] = (P_Q + P_A + 2*(P_Z) + 2) >> 2;
    img->mpr[1+ioff][0+joff] = 
    img->mpr[2+ioff][1+joff] = 
    img->mpr[3+ioff][2+joff] = 
    img->mpr[4+ioff][3+joff] = 
    img->mpr[5+ioff][4+joff] = 
    img->mpr[6+ioff][5+joff] = 
    img->mpr[7+ioff][6+joff] = (P_Z + P_B + 2*(P_A) + 2) >> 2;
    img->mpr[2+ioff][0+joff] = 
    img->mpr[3+ioff][1+joff] = 
    img->mpr[4+ioff][2+joff] = 
    img->mpr[5+ioff][3+joff] = 
    img->mpr[6+ioff][4+joff] = 
    img->mpr[7+ioff][5+joff] = (P_A + P_C + 2*(P_B) + 2) >> 2;
    img->mpr[3+ioff][0+joff] = 
    img->mpr[4+ioff][1+joff] = 
    img->mpr[5+ioff][2+joff] = 
    img->mpr[6+ioff][3+joff] = 
    img->mpr[7+ioff][4+joff] = (P_B + P_D + 2*(P_C) + 2) >> 2;
    img->mpr[4+ioff][0+joff] = 
    img->mpr[5+ioff][1+joff] = 
    img->mpr[6+ioff][2+joff] = 
    img->mpr[7+ioff][3+joff] = (P_C + P_E + 2*(P_D) + 2) >> 2;
    img->mpr[5+ioff][0+joff] = 
    img->mpr[6+ioff][1+joff] = 
    img->mpr[7+ioff][2+joff] = (P_D + P_F + 2*(P_E) + 2) >> 2;
    img->mpr[6+ioff][0+joff] = 
    img->mpr[7+ioff][1+joff] = (P_E + P_G + 2*(P_F) + 2) >> 2;
    img->mpr[7+ioff][0+joff] = (P_F + P_H + 2*(P_G) + 2) >> 2;
    break;

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

    img->mpr[0+ioff][0+joff] = 
    img->mpr[1+ioff][2+joff] = 
    img->mpr[2+ioff][4+joff] = 
    img->mpr[3+ioff][6+joff] = (P_Z + P_A + 1) >> 1;
    img->mpr[1+ioff][0+joff] = 
    img->mpr[2+ioff][2+joff] = 
    img->mpr[3+ioff][4+joff] = 
    img->mpr[4+ioff][6+joff] = (P_A + P_B + 1) >> 1;
    img->mpr[2+ioff][0+joff] = 
    img->mpr[3+ioff][2+joff] = 
    img->mpr[4+ioff][4+joff] = 
    img->mpr[5+ioff][6+joff] = (P_B + P_C + 1) >> 1;
    img->mpr[3+ioff][0+joff] = 
    img->mpr[4+ioff][2+joff] = 
    img->mpr[5+ioff][4+joff] = 
    img->mpr[6+ioff][6+joff] = (P_C + P_D + 1) >> 1;
    img->mpr[4+ioff][0+joff] = 
    img->mpr[5+ioff][2+joff] = 
    img->mpr[6+ioff][4+joff] = 
    img->mpr[7+ioff][6+joff] = (P_D + P_E + 1) >> 1;
    img->mpr[5+ioff][0+joff] = 
    img->mpr[6+ioff][2+joff] = 
    img->mpr[7+ioff][4+joff] = (P_E + P_F + 1) >> 1;
    img->mpr[6+ioff][0+joff] = 
    img->mpr[7+ioff][2+joff] = (P_F + P_G + 1) >> 1;
    img->mpr[7+ioff][0+joff] = (P_G + P_H + 1) >> 1;
    img->mpr[0+ioff][1+joff] = 
    img->mpr[1+ioff][3+joff] = 
    img->mpr[2+ioff][5+joff] = 
    img->mpr[3+ioff][7+joff] = (P_Q + P_A + 2*P_Z + 2) >> 2;
    img->mpr[1+ioff][1+joff] = 
    img->mpr[2+ioff][3+joff] = 
    img->mpr[3+ioff][5+joff] = 
    img->mpr[4+ioff][7+joff] = (P_Z + P_B + 2*P_A + 2) >> 2;
    img->mpr[2+ioff][1+joff] = 
    img->mpr[3+ioff][3+joff] = 
    img->mpr[4+ioff][5+joff] = 
    img->mpr[5+ioff][7+joff] = (P_A + P_C + 2*P_B + 2) >> 2;
    img->mpr[3+ioff][1+joff] = 
    img->mpr[4+ioff][3+joff] = 
    img->mpr[5+ioff][5+joff] = 
    img->mpr[6+ioff][7+joff] = (P_B + P_D + 2*P_C + 2) >> 2;
    img->mpr[4+ioff][1+joff] = 
    img->mpr[5+ioff][3+joff] = 
    img->mpr[6+ioff][5+joff] = 
    img->mpr[7+ioff][7+joff] = (P_C + P_E + 2*P_D + 2) >> 2;
    img->mpr[5+ioff][1+joff] = 
    img->mpr[6+ioff][3+joff] = 
    img->mpr[7+ioff][5+joff] = (P_D + P_F + 2*P_E + 2) >> 2;
    img->mpr[6+ioff][1+joff] = 
    img->mpr[7+ioff][3+joff] = (P_E + P_G + 2*P_F + 2) >> 2;
    img->mpr[7+ioff][1+joff] = (P_F + P_H + 2*P_G + 2) >> 2;