www.pudn.com > jm74.zip > block.c
/*!
***********************************************************************
* \file
* block.c
*
* \brief
* Block functions
*
* \author
* Main contributors (see contributors.h for copyright, address and affiliation details)
* - Inge Lille-Langøy
* - Rickard Sjoberg
***********************************************************************
*/
#include "contributors.h"
#include
#include "global.h"
#include "block.h"
#include "image.h"
#include "mb_access.h"
#define Q_BITS 15
static const int quant_coef[6][4][4] = {
{{13107, 8066,13107, 8066},{ 8066, 5243, 8066, 5243},{13107, 8066,13107, 8066},{ 8066, 5243, 8066, 5243}},
{{11916, 7490,11916, 7490},{ 7490, 4660, 7490, 4660},{11916, 7490,11916, 7490},{ 7490, 4660, 7490, 4660}},
{{10082, 6554,10082, 6554},{ 6554, 4194, 6554, 4194},{10082, 6554,10082, 6554},{ 6554, 4194, 6554, 4194}},
{{ 9362, 5825, 9362, 5825},{ 5825, 3647, 5825, 3647},{ 9362, 5825, 9362, 5825},{ 5825, 3647, 5825, 3647}},
{{ 8192, 5243, 8192, 5243},{ 5243, 3355, 5243, 3355},{ 8192, 5243, 8192, 5243},{ 5243, 3355, 5243, 3355}},
{{ 7282, 4559, 7282, 4559},{ 4559, 2893, 4559, 2893},{ 7282, 4559, 7282, 4559},{ 4559, 2893, 4559, 2893}}
};
static const int A[4][4] = {
{ 16, 20, 16, 20},
{ 20, 25, 20, 25},
{ 16, 20, 16, 20},
{ 20, 25, 20, 25}
};
// Notation for comments regarding prediction and predictors.
// The pels of the 4x4 block are labelled a..p. The predictor pels above
// are labelled A..H, from the left I..L, and from above left X, as follows:
//
// X A B C D E F G H
// I a b c d
// J e f g h
// K i j k l
// L m n o p
//
// Predictor array index definitions
#define P_X (PredPel[0])
#define P_A (PredPel[1])
#define P_B (PredPel[2])
#define P_C (PredPel[3])
#define P_D (PredPel[4])
#define P_E (PredPel[5])
#define P_F (PredPel[6])
#define P_G (PredPel[7])
#define P_H (PredPel[8])
#define P_I (PredPel[9])
#define P_J (PredPel[10])
#define P_K (PredPel[11])
#define P_L (PredPel[12])
/*!
***********************************************************************
* \brief
* makes and returns 4x4 blocks with all 5 intra prediction modes
*
* \return
* DECODING_OK decoding of intraprediction mode was sucessfull \n
* SEARCH_SYNC search next sync element as errors while decoding occured
***********************************************************************
*/
int intrapred(
struct img_par *img, //!< image parameters
int ioff, //!< pixel offset X within MB
int joff, //!< pixel offset Y within MB
int img_block_x, //!< location of block X, multiples of 4
int img_block_y) //!< location of block Y, multiples of 4
{
int i,j;
int s0;
int img_y,img_x;
int PredPel[13]; // array of predictor pels
byte **imgY = dec_picture->imgY;
PixelPos pix_a[4];
PixelPos pix_b, pix_c, pix_d;
int block_available_up;
int block_available_left;
int block_available_up_left;
int block_available_up_right;
int mb_nr=img->current_mb_nr;
byte predmode = img->ipredmode[img_block_x][img_block_y];
img_x=img_block_x*4;
img_y=img_block_y*4;
for (i=0;i<4;i++)
{
getNeighbour(mb_nr, ioff -1 , joff +i , 1, &pix_a[i]);
}
getNeighbour(mb_nr, ioff , joff -1 , 1, &pix_b);
getNeighbour(mb_nr, ioff +4 , joff -1 , 1, &pix_c);
getNeighbour(mb_nr, ioff -1 , joff -1 , 1, &pix_d);
pix_c.available = pix_c.available && !(((ioff==4)||(ioff==12)) && ((joff==4)||(joff==12)));
if (img->constrained_intra_pred_flag)
{
for (i=0, block_available_left=1; i<4;i++)
block_available_left &= pix_a[i].available ? img->intra_block[pix_a[i].mb_addr]: 0;
block_available_up = pix_b.available ? img->intra_block [pix_b.mb_addr] : 0;
block_available_up_right = pix_c.available ? img->intra_block [pix_c.mb_addr] : 0;
block_available_up_left = pix_d.available ? img->intra_block [pix_d.mb_addr] : 0;
}
else
{
block_available_left = pix_a[0].available;
block_available_up = pix_b.available;
block_available_up_right = pix_c.available;
block_available_up_left = pix_d.available;
}
// form predictor pels
if (block_available_up)
{
P_A = imgY[pix_b.pos_y][pix_b.pos_x+0];
P_B = imgY[pix_b.pos_y][pix_b.pos_x+1];
P_C = imgY[pix_b.pos_y][pix_b.pos_x+2];
P_D = imgY[pix_b.pos_y][pix_b.pos_x+3];
}
else
{
P_A = P_B = P_C = P_D = 128;
}
if (block_available_up_right)
{
P_E = imgY[pix_c.pos_y][pix_c.pos_x+0];
P_F = imgY[pix_c.pos_y][pix_c.pos_x+1];
P_G = imgY[pix_c.pos_y][pix_c.pos_x+2];
P_H = imgY[pix_c.pos_y][pix_c.pos_x+3];
}
else
{
P_E = P_F = P_G = P_H = P_D;
}
if (block_available_left)
{
P_I = imgY[pix_a[0].pos_y][pix_a[0].pos_x];
P_J = imgY[pix_a[1].pos_y][pix_a[1].pos_x];
P_K = imgY[pix_a[2].pos_y][pix_a[2].pos_x];
P_L = imgY[pix_a[3].pos_y][pix_a[3].pos_x];
}
else
{
P_I = P_J = P_K = P_L = 128;
}
if (block_available_up_left)
{
P_X = imgY[pix_d.pos_y][pix_d.pos_x];
}
else
{
P_X = 128;
}
switch (predmode)
{
case DC_PRED: /* DC prediction */
s0 = 0;
if (block_available_up && block_available_left)
{
// no edge
s0 = (P_A + P_B + P_C + P_D + P_I + P_J + P_K + P_L + 4)/(2*BLOCK_SIZE);
}
else if (!block_available_up && block_available_left)
{
// upper edge
s0 = (P_I + P_J + P_K + P_L + 2)/BLOCK_SIZE;
}
else if (block_available_up && !block_available_left)
{
// left edge
s0 = (P_A + P_B + P_C + P_D + 2)/BLOCK_SIZE;
}
else //if (!block_available_up && !block_available_left)
{
// top left corner, nothing to predict from
s0 = 128;
}
for (j=0; j < BLOCK_SIZE; j++)
{
for (i=0; i < BLOCK_SIZE; i++)
{
// store DC prediction
img->mpr[i+ioff][j+joff] = s0;
}
}
break;
case VERT_PRED: /* vertical prediction from block above */
if (!block_available_up)
printf ("warning: Intra_4x4_Vertical prediction mode not allowed at mb %d\n",img->current_mb_nr);
for(j=0;jmpr[i+ioff][j+joff]=imgY[pix_b.pos_y][pix_b.pos_x+i];/* store predicted 4x4 block */
break;
case HOR_PRED: /* horizontal prediction from left block */
if (!block_available_left)
printf ("warning: Intra_4x4_Horizontal prediction mode not allowed at mb %d\n",img->current_mb_nr);
for(j=0;jmpr[i+ioff][j+joff]=imgY[pix_a[j].pos_y][pix_a[j].pos_x]; /* store predicted 4x4 block */
break;
case DIAG_DOWN_RIGHT_PRED:
if ((!block_available_up)||(!block_available_left)||(!block_available_up_left))
printf ("warning: Intra_4x4_Diagonal_Down_Right prediction mode not allowed at mb %d\n",img->current_mb_nr);
img->mpr[0+ioff][3+joff] = (P_L + 2*P_K + P_J + 2) / 4;
img->mpr[0+ioff][2+joff] =
img->mpr[1+ioff][3+joff] = (P_K + 2*P_J + P_I + 2) / 4;
img->mpr[0+ioff][1+joff] =
img->mpr[1+ioff][2+joff] =
img->mpr[2+ioff][3+joff] = (P_J + 2*P_I + P_X + 2) / 4;
img->mpr[0+ioff][0+joff] =
img->mpr[1+ioff][1+joff] =
img->mpr[2+ioff][2+joff] =
img->mpr[3+ioff][3+joff] = (P_I + 2*P_X + P_A + 2) / 4;
img->mpr[1+ioff][0+joff] =
img->mpr[2+ioff][1+joff] =
img->mpr[3+ioff][2+joff] = (P_X + 2*P_A + P_B + 2) / 4;
img->mpr[2+ioff][0+joff] =
img->mpr[3+ioff][1+joff] = (P_A + 2*P_B + P_C + 2) / 4;
img->mpr[3+ioff][0+joff] = (P_B + 2*P_C + P_D + 2) / 4;
break;
case DIAG_DOWN_LEFT_PRED:
if (!block_available_up)
printf ("warning: Intra_4x4_Diagonal_Down_Left prediction mode not allowed at mb %d\n",img->current_mb_nr);
img->mpr[0+ioff][0+joff] = (P_A + P_C + 2*(P_B) + 2) / 4;
img->mpr[1+ioff][0+joff] =
img->mpr[0+ioff][1+joff] = (P_B + P_D + 2*(P_C) + 2) / 4;
img->mpr[2+ioff][0+joff] =
img->mpr[1+ioff][1+joff] =
img->mpr[0+ioff][2+joff] = (P_C + P_E + 2*(P_D) + 2) / 4;
img->mpr[3+ioff][0+joff] =
img->mpr[2+ioff][1+joff] =
img->mpr[1+ioff][2+joff] =
img->mpr[0+ioff][3+joff] = (P_D + P_F + 2*(P_E) + 2) / 4;
img->mpr[3+ioff][1+joff] =
img->mpr[2+ioff][2+joff] =
img->mpr[1+ioff][3+joff] = (P_E + P_G + 2*(P_F) + 2) / 4;
img->mpr[3+ioff][2+joff] =
img->mpr[2+ioff][3+joff] = (P_F + P_H + 2*(P_G) + 2) / 4;
img->mpr[3+ioff][3+joff] = (P_G + 3*(P_H) + 2) / 4;
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_4x4_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] = (P_X + P_A + 1) / 2;
img->mpr[1+ioff][0+joff] =
img->mpr[2+ioff][2+joff] = (P_A + P_B + 1) / 2;
img->mpr[2+ioff][0+joff] =
img->mpr[3+ioff][2+joff] = (P_B + P_C + 1) / 2;
img->mpr[3+ioff][0+joff] = (P_C + P_D + 1) / 2;
img->mpr[0+ioff][1+joff] =
img->mpr[1+ioff][3+joff] = (P_I + 2*P_X + P_A + 2) / 4;
img->mpr[1+ioff][1+joff] =
img->mpr[2+ioff][3+joff] = (P_X + 2*P_A + P_B + 2) / 4;
img->mpr[2+ioff][1+joff] =
img->mpr[3+ioff][3+joff] = (P_A + 2*P_B + P_C + 2) / 4;
img->mpr[3+ioff][1+joff] = (P_B + 2*P_C + P_D + 2) / 4;
img->mpr[0+ioff][2+joff] = (P_X + 2*P_I + P_J + 2) / 4;
img->mpr[0+ioff][3+joff] = (P_I + 2*P_J + P_K + 2) / 4;
break;
case VERT_LEFT_PRED:/* diagonal prediction -22.5 deg to horizontal plane */
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) / 2;
img->mpr[1+ioff][0+joff] =
img->mpr[0+ioff][2+joff] = (P_B + P_C + 1) / 2;
img->mpr[2+ioff][0+joff] =
img->mpr[1+ioff][2+joff] = (P_C + P_D + 1) / 2;
img->mpr[3+ioff][0+joff] =
img->mpr[2+ioff][2+joff] = (P_D + P_E + 1) / 2;
img->mpr[3+ioff][2+joff] = (P_E + P_F + 1) / 2;
img->mpr[0+ioff][1+joff] = (P_A + 2*P_B + P_C + 2) / 4;
img->mpr[1+ioff][1+joff] =
img->mpr[0+ioff][3+joff] = (P_B + 2*P_C + P_D + 2) / 4;
img->mpr[2+ioff][1+joff] =
img->mpr[1+ioff][3+joff] = (P_C + 2*P_D + P_E + 2) / 4;
img->mpr[3+ioff][1+joff] =
img->mpr[2+ioff][3+joff] = (P_D + 2*P_E + P_F + 2) / 4;
img->mpr[3+ioff][3+joff] = (P_E + 2*P_F + P_G + 2) / 4;
break;
case HOR_UP_PRED:/* diagonal prediction -22.5 deg to horizontal plane */
if (!block_available_left)
printf ("warning: Intra_4x4_Horizontal_Up prediction mode not allowed at mb %d\n",img->current_mb_nr);
img->mpr[0+ioff][0+joff] = (P_I + P_J + 1) / 2;
img->mpr[1+ioff][0+joff] = (P_I + 2*P_J + P_K + 2) / 4;
img->mpr[2+ioff][0+joff] =
img->mpr[0+ioff][1+joff] = (P_J + P_K + 1) / 2;
img->mpr[3+ioff][0+joff] =
img->mpr[1+ioff][1+joff] = (P_J + 2*P_K + P_L + 2) / 4;
img->mpr[2+ioff][1+joff] =
img->mpr[0+ioff][2+joff] = (P_K + P_L + 1) / 2;
img->mpr[3+ioff][1+joff] =
img->mpr[1+ioff][2+joff] = (P_K + 2*P_L + P_L + 2) / 4;
img->mpr[3+ioff][2+joff] =
img->mpr[1+ioff][3+joff] =
img->mpr[0+ioff][3+joff] =
img->mpr[2+ioff][2+joff] =
img->mpr[2+ioff][3+joff] =
img->mpr[3+ioff][3+joff] = P_L;
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_4x4_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] = (P_X + P_I + 1) / 2;
img->mpr[1+ioff][0+joff] =
img->mpr[3+ioff][1+joff] = (P_I + 2*P_X + P_A + 2) / 4;
img->mpr[2+ioff][0+joff] = (P_X + 2*P_A + P_B + 2) / 4;
img->mpr[3+ioff][0+joff] = (P_A + 2*P_B + P_C + 2) / 4;
img->mpr[0+ioff][1+joff] =
img->mpr[2+ioff][2+joff] = (P_I + P_J + 1) / 2;
img->mpr[1+ioff][1+joff] =
img->mpr[3+ioff][2+joff] = (P_X + 2*P_I + P_J + 2) / 4;
img->mpr[0+ioff][2+joff] =
img->mpr[2+ioff][3+joff] = (P_J + P_K + 1) / 2;
img->mpr[1+ioff][2+joff] =
img->mpr[3+ioff][3+joff] = (P_I + 2*P_J + P_K + 2) / 4;
img->mpr[0+ioff][3+joff] = (P_K + P_L + 1) / 2;
img->mpr[1+ioff][3+joff] = (P_J + 2*P_K + P_L + 2) / 4;
break;
default:
printf("Error: illegal intra_4x4 prediction mode: %d\n",predmode);
return SEARCH_SYNC;
break;
}
return DECODING_OK;
}
/*!
***********************************************************************
* \return
* best SAD
***********************************************************************
*/
int intrapred_luma_16x16(struct img_par *img, //!< image parameters
int predmode) //!< prediction mode
{
int s0=0,s1,s2;
int i,j;
int ih,iv;
int ib,ic,iaa;
byte **imgY=dec_picture->imgY;
int mb_nr=img->current_mb_nr;
PixelPos up; //!< pixel position p(0,-1)
PixelPos left[17]; //!< pixel positions p(-1, -1..15)
int up_avail, left_avail, left_up_avail;
s1=s2=0;
for (i=0;i<17;i++)
{
getNeighbour(mb_nr, -1 , i-1 , 1, &left[i]);
}
getNeighbour(mb_nr, 0 , -1 , 1, &up);
if (!img->constrained_intra_pred_flag)
{
up_avail = up.available;
left_avail = left[1].available;
left_up_avail = left[0].available;
}
else
{
up_avail = up.available ? img->intra_block[up.mb_addr] : 0;
for (i=1, left_avail=1; i<17;i++)
left_avail &= left[i].available ? img->intra_block[left[i].mb_addr]: 0;
left_up_avail = left[0].available ? img->intra_block[left[0].mb_addr]: 0;
}
switch (predmode)
{
case VERT_PRED_16: // vertical prediction from block above
if (!up_avail)
error ("invalid 16x16 intra pred Mode VERT_PRED_16",500);
for(j=0;jmpr[i][j]=imgY[up.pos_y][up.pos_x+i];// store predicted 16x16 block
break;
case HOR_PRED_16: // horisontal prediction from left block
if (!left_avail)
error ("invalid 16x16 intra pred Mode VERT_PRED_16",500);
for(j=0;jmpr[i][j]=imgY[left[j+1].pos_y][left[j+1].pos_x]; // store predicted 16x16 block
break;
case DC_PRED_16: // DC prediction
s1=s2=0;
for (i=0; i < MB_BLOCK_SIZE; i++)
{
if (up_avail)
s1 += imgY[up.pos_y][up.pos_x+i]; // sum hor pix
if (left_avail)
s2 += imgY[left[i+1].pos_y][left[i+1].pos_x]; // sum vert pix
}
if (up_avail && left_avail)
s0=(s1+s2+16)>>5; // no edge
if (!up_avail && left_avail)
s0=(s2+8)>>4; // upper edge
if (up_avail && !left_avail)
s0=(s1+8)>>4; // left edge
if (!up_avail && !left_avail)
s0=128; // top left corner, nothing to predict from
for(i=0;impr[i][j]=s0;
}
break;
case PLANE_16:// 16 bit integer plan pred
if (!up_avail || !left_up_avail || !left_avail)
error ("invalid 16x16 intra pred Mode PLANE_16",500);
ih=0;
iv=0;
for (i=1;i<9;i++)
{
if (i<8)
ih += i*(imgY[up.pos_y][up.pos_x+7+i] - imgY[up.pos_y][up.pos_x+7-i]);
else
ih += i*(imgY[up.pos_y][up.pos_x+7+i] - imgY[left[0].pos_y][left[0].pos_x]);
iv += i*(imgY[left[8+i].pos_y][left[8+i].pos_x] - imgY[left[8-i].pos_y][left[8-i].pos_x]);
}
ib=(5*ih+32)>>6;
ic=(5*iv+32)>>6;
iaa=16*(imgY[up.pos_y][up.pos_x+15]+imgY[left[16].pos_y][left[16].pos_x]);
for (j=0;j< MB_BLOCK_SIZE;j++)
{
for (i=0;i< MB_BLOCK_SIZE;i++)
{
img->mpr[i][j]=max(0,min((iaa+(i-7)*ib +(j-7)*ic + 16)>>5,255));
}
}// store plane prediction
break;
default:
{ // indication of fault in bitstream,exit
printf("illegal 16x16 intra prediction mode input: %d\n",predmode);
return SEARCH_SYNC;
}
}
return DECODING_OK;
}
void intrapred_chroma(struct img_par *img, int uv)
{
int i,j, ii, jj, ioff, joff;
byte ***imgUV = dec_picture->imgUV;
int js0=0;
int js1=0;
int js2=0;
int js3=0;
int js[2][2];
int pred;
int ih, iv, ib, ic, iaa;
int mb_nr=img->current_mb_nr;
Macroblock *currMB = &img->mb_data[img->current_mb_nr];
PixelPos up; //!< pixel position p(0,-1)
PixelPos left[9]; //!< pixel positions p(-1, -1..8)
int up_avail, left_avail, left_up_avail;
for (i=0;i<9;i++)
{
getNeighbour(mb_nr, -1 , i-1 , 0, &left[i]);
}
getNeighbour(mb_nr, 0 , -1 , 0, &up);
if (!img->constrained_intra_pred_flag)
{
up_avail = up.available;
left_avail = left[1].available;
left_up_avail = left[0].available;
}
else
{
up_avail = up.available ? img->intra_block[up.mb_addr] : 0;
for (i=1, left_avail=1; i<9;i++)
left_avail &= left[i].available ? img->intra_block[left[i].mb_addr]: 0;
left_up_avail = left[0].available ? img->intra_block[left[0].mb_addr]: 0;
}
if (currMB->c_ipred_mode == DC_PRED_8)
{
for(i=0;i<4;i++)
{
if(up_avail)
{
js0=js0+imgUV[uv][up.pos_y][up.pos_x+i];
js1=js1+imgUV[uv][up.pos_y][up.pos_x+i+4];
}
if(left_avail)
{
js2=js2+imgUV[uv][left[1+i].pos_y][left[1+i].pos_x];
js3=js3+imgUV[uv][left[1+i+4].pos_y][left[1+i+4].pos_x];
}
}
if(up_avail && left_avail)
{
js[0][0]=(js0+js2+4)/8;
js[1][0]=(js1+2)/4;
js[0][1]=(js3+2)/4;
js[1][1]=(js1+js3+4)/8;
}
if(up_avail && !left_avail)
{
js[0][0]=(js0+2)/4;
js[1][0]=(js1+2)/4;
js[0][1]=(js0+2)/4;
js[1][1]=(js1+2)/4;
}
if(left_avail && !up_avail)
{
js[0][0]=(js2+2)/4;
js[1][0]=(js2+2)/4;
js[0][1]=(js3+2)/4;
js[1][1]=(js3+2)/4;
}
if(!up_avail && !left_avail)
{
js[0][0]=128;
js[1][0]=128;
js[0][1]=128;
js[1][1]=128;
}
}
for (j=0;j<2;j++)
{
joff=j*4;
for(i=0;i<2;i++)
{
ioff=i*4;
switch (currMB->c_ipred_mode)
{
case DC_PRED_8:
for (ii=0; ii<4; ii++)
for (jj=0; jj<4; jj++)
{
img->mpr[ii+ioff][jj+joff]=js[i][j];
}
break;
case HOR_PRED_8:
if (!left_avail)
error("unexpected HOR_PRED_8 chroma intra prediction mode",-1);
for (jj=0; jj<4; jj++)
{
pred = imgUV[uv][left[1+jj+joff].pos_y][left[1+jj+joff].pos_x];
for (ii=0; ii<4; ii++)
img->mpr[ii+ioff][jj+joff]=pred;
}
break;
case VERT_PRED_8:
if (!up_avail)
error("unexpected VERT_PRED_8 chroma intra prediction mode",-1);
for (ii=0; ii<4; ii++)
{
pred = imgUV[uv][up.pos_y][up.pos_x+ii+ioff];
for (jj=0; jj<4; jj++)
img->mpr[ii+ioff][jj+joff]=pred;
}
break;
case PLANE_8:
if (!left_up_avail || !left_avail || !up_avail)
error("unexpected PLANE_8 chroma intra prediction mode",-1);
ih=iv=0;
for (ii=1;ii<5;ii++)
{
if (ii<4)
ih += ii*(imgUV[uv][up.pos_y][up.pos_x+3+ii] - imgUV[uv][up.pos_y][up.pos_x+3-ii]);
else
ih += ii*(imgUV[uv][up.pos_y][up.pos_x+3+ii] - imgUV[uv][left[0].pos_y][left[0].pos_x]);
iv += ii*(imgUV[uv][left[4+ii].pos_y][left[4+ii].pos_x] - imgUV[uv][left[4-ii].pos_y][left[4-ii].pos_x]);
}
ib=(17*ih+16)>>5;
ic=(17*iv+16)>>5;
iaa=16*(imgUV[uv][up.pos_y][up.pos_x+7]+imgUV[uv][left[8].pos_y][left[8].pos_x]);
for (ii=0; ii<4; ii++)
for (jj=0; jj<4; jj++)
img->mpr[ii+ioff][jj+joff]=max(0,min(255,(iaa+(ii+ioff-3)*ib +(jj+joff-3)*ic + 16)>>5));
break;
default:
error("illegal chroma intra prediction mode", 600);
break;
}
}
}
}
/*!
***********************************************************************
* \brief
* Inverse 4x4 transformation, transforms cof to m7
***********************************************************************
*/
void itrans(struct img_par *img, //!< image parameters
int ioff, //!< index to 4x4 block
int joff, //!<
int i0, //!<
int j0) //!<
{
int i,j,i1,j1;
int m5[4];
int m6[4];
// horizontal
for (j=0;jcof[i0][j0][i][j];
}
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (i=0;i<2;i++)
{
i1=3-i;
img->m7[i][j]=m6[i]+m6[i1];
img->m7[i1][j]=m6[i]-m6[i1];
}
}
// vertical
for (i=0;im7[i][j];
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (j=0;j<2;j++)
{
j1=3-j;
img->m7[i][j] =max(0,min(255,(m6[j]+m6[j1]+(img->mpr[i+ioff][j+joff] <>DQ_BITS));
img->m7[i][j1]=max(0,min(255,(m6[j]-m6[j1]+(img->mpr[i+ioff][j1+joff]<>DQ_BITS));
}
}
}
/*!
***********************************************************************
* \brief
* invers transform
***********************************************************************
*/
void itrans_2(
struct img_par *img) //!< image parameters
{
int i,j,i1,j1;
int M5[4];
int M6[4];
int qp_per = (img->qp-MIN_QP)/6;
int qp_rem = (img->qp-MIN_QP)%6;
// horizontal
for (j=0;j<4;j++)
{
for (i=0;i<4;i++)
M5[i]=img->cof[i][j][0][0];
M6[0]=M5[0]+M5[2];
M6[1]=M5[0]-M5[2];
M6[2]=M5[1]-M5[3];
M6[3]=M5[1]+M5[3];
for (i=0;i<2;i++)
{
i1=3-i;
img->cof[i ][j][0][0]= M6[i]+M6[i1];
img->cof[i1][j][0][0]=M6[i]-M6[i1];
}
}
// vertical
for (i=0;i<4;i++)
{
for (j=0;j<4;j++)
M5[j]=img->cof[i][j][0][0];
M6[0]=M5[0]+M5[2];
M6[1]=M5[0]-M5[2];
M6[2]=M5[1]-M5[3];
M6[3]=M5[1]+M5[3];
for (j=0;j<2;j++)
{
j1=3-j;
img->cof[i][j][0][0] = (((M6[j]+M6[j1])*dequant_coef[qp_rem][0][0]<>2;
img->cof[i][j1][0][0]= (((M6[j]-M6[j1])*dequant_coef[qp_rem][0][0]<>2;
}
}
}
void itrans_sp(struct img_par *img, //!< image parameters
int ioff, //!< index to 4x4 block
int joff, //!<
int i0, //!<
int j0) //!<
{
int i,j,i1,j1;
int m5[4];
int m6[4];
int predicted_block[BLOCK_SIZE][BLOCK_SIZE],ilev;
int qp_per = (img->qp-MIN_QP)/6;
int qp_rem = (img->qp-MIN_QP)%6;
int q_bits = Q_BITS+qp_per;
int qp_per_sp = (img->qpsp-MIN_QP)/6;
int qp_rem_sp = (img->qpsp-MIN_QP)%6;
int q_bits_sp = Q_BITS+qp_per_sp;
int qp_const2=(1<sp_switch || img->type == SI_SLICE)
{
qp_per = (img->qpsp-MIN_QP)/6;
qp_rem = (img->qpsp-MIN_QP)%6;
q_bits = Q_BITS+qp_per;
}
for (j=0; j< BLOCK_SIZE; j++)
for (i=0; i< BLOCK_SIZE; i++)
predicted_block[i][j]=img->mpr[i+ioff][j+joff];
for (j=0; j < BLOCK_SIZE; j++)
{
for (i=0; i < 2; i++)
{
i1=3-i;
m5[i]=predicted_block[i][j]+predicted_block[i1][j];
m5[i1]=predicted_block[i][j]-predicted_block[i1][j];
}
predicted_block[0][j]=(m5[0]+m5[1]);
predicted_block[2][j]=(m5[0]-m5[1]);
predicted_block[1][j]=m5[3]*2+m5[2];
predicted_block[3][j]=m5[3]-m5[2]*2;
}
// Vertival transform
for (i=0; i < BLOCK_SIZE; i++)
{
for (j=0; j < 2; j++)
{
j1=3-j;
m5[j]=predicted_block[i][j]+predicted_block[i][j1];
m5[j1]=predicted_block[i][j]-predicted_block[i][j1];
}
predicted_block[i][0]=(m5[0]+m5[1]);
predicted_block[i][2]=(m5[0]-m5[1]);
predicted_block[i][1]=m5[3]*2+m5[2];
predicted_block[i][3]=m5[3]-m5[2]*2;
}
for (j=0;jcof[i0][j0][i][j]=(img->cof[i0][j0][i][j] >> qp_per) / dequant_coef[qp_rem][i][j];
ilev=((img->cof[i0][j0][i][j]*dequant_coef[qp_rem][i][j]*A[i][j]<< qp_per) >>6)+predicted_block[i][j] ;
img->cof[i0][j0][i][j]=sign((abs(ilev) * quant_coef[qp_rem_sp][i][j] + qp_const2) >> q_bits_sp, ilev) * dequant_coef[qp_rem_sp][i][j] << qp_per_sp;
}
// horizontal
for (j=0;jcof[i0][j0][i][j];
}
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (i=0;i<2;i++)
{
i1=3-i;
img->m7[i][j]=m6[i]+m6[i1];
img->m7[i1][j]=m6[i]-m6[i1];
}
}
// vertical
for (i=0;im7[i][j];
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (j=0;j<2;j++)
{
j1=3-j;
img->m7[i][j] =max(0,min(255,(m6[j]+m6[j1]+DQ_ROUND)>>DQ_BITS));
img->m7[i][j1]=max(0,min(255,(m6[j]-m6[j1]+DQ_ROUND)>>DQ_BITS));
}
}
}
/*!
***********************************************************************
* \brief
* The routine performs transform,quantization,inverse transform, adds the diff.
* to the prediction and writes the result to the decoded luma frame. Includes the
* RD constrained quantization also.
*
* \par Input:
* block_x,block_y: Block position inside a macro block (0,4,8,12).
*
* \par Output:
* nonzero: 0 if no levels are nonzero. 1 if there are nonzero levels. \n
* coeff_cost: Counter for nonzero coefficients, used to discard expencive levels.
************************************************************************
*/
void copyblock_sp(struct img_par *img,int block_x,int block_y)
{
int sign(int a,int b);
int i,j,i1,j1,m5[4],m6[4];
int predicted_block[BLOCK_SIZE][BLOCK_SIZE];
int qp_per = (img->qpsp-MIN_QP)/6;
int qp_rem = (img->qpsp-MIN_QP)%6;
int q_bits = Q_BITS+qp_per;
int qp_const2=(1<mpr[i+block_x][j+block_y];
for (j=0; j < BLOCK_SIZE; j++)
{
for (i=0; i < 2; i++)
{
i1=3-i;
m5[i]=predicted_block[i][j]+predicted_block[i1][j];
m5[i1]=predicted_block[i][j]-predicted_block[i1][j];
}
predicted_block[0][j]=(m5[0]+m5[1]);
predicted_block[2][j]=(m5[0]-m5[1]);
predicted_block[1][j]=m5[3]*2+m5[2];
predicted_block[3][j]=m5[3]-m5[2]*2;
}
// Vertival transform
for (i=0; i < BLOCK_SIZE; i++)
{
for (j=0; j < 2; j++)
{
j1=3-j;
m5[j]=predicted_block[i][j]+predicted_block[i][j1];
m5[j1]=predicted_block[i][j]-predicted_block[i][j1];
}
predicted_block[i][0]=(m5[0]+m5[1]);
predicted_block[i][2]=(m5[0]-m5[1]);
predicted_block[i][1]=m5[3]*2+m5[2];
predicted_block[i][3]=m5[3]-m5[2]*2;
}
// Quant
for (j=0;j < BLOCK_SIZE; j++)
for (i=0; i < BLOCK_SIZE; i++)
img->m7[i][j]=sign((abs(predicted_block[i][j])* quant_coef[qp_rem][i][j]+qp_const2)>> q_bits,predicted_block[i][j])*dequant_coef[qp_rem][i][j]<m7[i][j];
}
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (i=0;i<2;i++)
{
i1=3-i;
img->m7[i][j]=m6[i]+m6[i1];
img->m7[i1][j]=m6[i]-m6[i1];
}
}
// vertical
for (i=0;im7[i][j];
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (j=0;j<2;j++)
{
j1=3-j;
img->m7[i][j] =max(0,min(255,(m6[j]+m6[j1]+DQ_ROUND)>>DQ_BITS));
img->m7[i][j1]=max(0,min(255,(m6[j]-m6[j1]+DQ_ROUND)>>DQ_BITS));
}
}
// Decoded block moved to frame memory
for (j=0; j < BLOCK_SIZE; j++)
for (i=0; i < BLOCK_SIZE; i++)
dec_picture->imgY[img->pix_y+block_y+j][img->pix_x+block_x+i]=img->m7[i][j];
}
void itrans_sp_chroma(struct img_par *img,int ll)
{
int i,j,i1,j2,ilev,n2,n1,j1,mb_y;
int m5[BLOCK_SIZE];
int predicted_chroma_block[MB_BLOCK_SIZE/2][MB_BLOCK_SIZE/2],mp1[BLOCK_SIZE];
int qp_per,qp_rem,q_bits;
int qp_per_sp,qp_rem_sp,q_bits_sp,qp_const2;
qp_per = ((img->qp<0?img->qp:QP_SCALE_CR[img->qp])-MIN_QP)/6;
qp_rem = ((img->qp<0?img->qp:QP_SCALE_CR[img->qp])-MIN_QP)%6;
q_bits = Q_BITS+qp_per;
qp_per_sp = ((img->qpsp<0?img->qpsp:QP_SCALE_CR[img->qpsp])-MIN_QP)/6;
qp_rem_sp = ((img->qpsp<0?img->qpsp:QP_SCALE_CR[img->qpsp])-MIN_QP)%6;
q_bits_sp = Q_BITS+qp_per_sp;
qp_const2=(1<sp_switch || img->type == SI_SLICE)
{
qp_per = ((img->qpsp < 0 ? img->qpsp : QP_SCALE_CR[img->qpsp]) - MIN_QP) / 6;
qp_rem = ((img->qpsp < 0 ? img->qpsp : QP_SCALE_CR[img->qpsp]) - MIN_QP) % 6;
q_bits = Q_BITS + qp_per;
}
for (j=0; j < MB_BLOCK_SIZE/2; j++)
for (i=0; i < MB_BLOCK_SIZE/2; i++)
{
predicted_chroma_block[i][j]=img->mpr[i][j];
img->mpr[i][j]=0;
}
for (n2=0; n2 <= BLOCK_SIZE; n2 += BLOCK_SIZE)
{
for (n1=0; n1 <= BLOCK_SIZE; n1 += BLOCK_SIZE)
{
// Horizontal transform.
for (j=0; j < BLOCK_SIZE; j++)
{
mb_y=n2+j;
for (i=0; i < 2; i++)
{
i1=3-i;
m5[i]=predicted_chroma_block[i+n1][mb_y]+predicted_chroma_block[i1+n1][mb_y];
m5[i1]=predicted_chroma_block[i+n1][mb_y]-predicted_chroma_block[i1+n1][mb_y];
}
predicted_chroma_block[n1][mb_y] =(m5[0]+m5[1]);
predicted_chroma_block[n1+2][mb_y]=(m5[0]-m5[1]);
predicted_chroma_block[n1+1][mb_y]=m5[3]*2+m5[2];
predicted_chroma_block[n1+3][mb_y]=m5[3]-m5[2]*2;
}
// Vertical transform.
for (i=0; i < BLOCK_SIZE; i++)
{
j1=n1+i;
for (j=0; j < 2; j++)
{
j2=3-j;
m5[j]=predicted_chroma_block[j1][n2+j]+predicted_chroma_block[j1][n2+j2];
m5[j2]=predicted_chroma_block[j1][n2+j]-predicted_chroma_block[j1][n2+j2];
}
predicted_chroma_block[j1][n2+0]=(m5[0]+m5[1]);
predicted_chroma_block[j1][n2+2]=(m5[0]-m5[1]);
predicted_chroma_block[j1][n2+1]=m5[3]*2+m5[2];
predicted_chroma_block[j1][n2+3]=m5[3]-m5[2]*2;
}
}
}
// 2X2 transform of DC coeffs.
mp1[0]=(predicted_chroma_block[0][0]+predicted_chroma_block[4][0]+predicted_chroma_block[0][4]+predicted_chroma_block[4][4]);
mp1[1]=(predicted_chroma_block[0][0]-predicted_chroma_block[4][0]+predicted_chroma_block[0][4]-predicted_chroma_block[4][4]);
mp1[2]=(predicted_chroma_block[0][0]+predicted_chroma_block[4][0]-predicted_chroma_block[0][4]-predicted_chroma_block[4][4]);
mp1[3]=(predicted_chroma_block[0][0]-predicted_chroma_block[4][0]-predicted_chroma_block[0][4]+predicted_chroma_block[4][4]);
for (n1=0; n1 < 2; n1 ++)
for (n2=0; n2 < 2; n2 ++)
{
ilev=((img->cof[n1+ll][4+n2][0][0]*dequant_coef[qp_rem][0][0]*A[0][0]<< qp_per) >>5)+mp1[n1+n2*2] ;
mp1[n1+n2*2]=sign((abs(ilev)* quant_coef[qp_rem_sp][0][0]+ 2 * qp_const2)>> (q_bits_sp+1),ilev)*dequant_coef[qp_rem_sp][0][0]<cof[n1+ll][4+n2][i][j] = (img->cof[n1+ll][4+n2][i][j] >> qp_per) / dequant_coef[qp_rem][i][j];
ilev=((img->cof[n1+ll][4+n2][i][j]*dequant_coef[qp_rem][i][j]*A[i][j]<< qp_per) >>6)+predicted_chroma_block[n1*BLOCK_SIZE+i][n2*BLOCK_SIZE+j] ;
img->cof[n1+ll][4+n2][i][j] = sign((abs(ilev) * quant_coef[qp_rem_sp][i][j] + qp_const2)>> q_bits_sp,ilev)*dequant_coef[qp_rem_sp][i][j]<cof[0+ll][4][0][0]=(mp1[0]+mp1[1]+mp1[2]+mp1[3])>>1;
img->cof[1+ll][4][0][0]=(mp1[0]-mp1[1]+mp1[2]-mp1[3])>>1;
img->cof[0+ll][5][0][0]=(mp1[0]+mp1[1]-mp1[2]-mp1[3])>>1;
img->cof[1+ll][5][0][0]=(mp1[0]-mp1[1]-mp1[2]+mp1[3])>>1;
}
int sign(int a , int b)
{
int x;
x=abs(a);
if (b>0)
return(x);
else return(-x);
}