/*
***********************************************************************
* COPYRIGHT AND WARRANTY INFORMATION
*
* Copyright 2001, International Telecommunications Union, Geneva
*
* DISCLAIMER OF WARRANTY
*
* These software programs are available to the user without any
* license fee or royalty on an "as is" basis. The ITU disclaims
* any and all warranties, whether express, implied, or
* statutory, including any implied warranties of merchantability
* or of fitness for a particular purpose.  In no event shall the
* contributor or the ITU be liable for any incidental, punitive, or
* consequential damages of any kind whatsoever arising from the
* use of these programs.
*
* This disclaimer of warranty extends to the user of these programs
* and user's customers, employees, agents, transferees, successors,
* and assigns.
*
* The ITU does not represent or warrant that the programs furnished
* hereunder are free of infringement of any third-party patents.
* Commercial implementations of ITU-T Recommendations, including
* shareware, may be subject to royalty fees to patent holders.
* Information regarding the ITU-T patent policy is available from
* the ITU Web site at http://www.itu.int.
*
* THIS IS NOT A GRANT OF PATENT RIGHTS - SEE THE ITU-T PATENT POLICY.
************************************************************************
*/

/*!
 ***************************************************************************
 * \file rdopt.c
 *
 * \brief
 *    Rate-Distortion optimized mode decision
 *
 * \author
 *    Heiko Schwarz <hschwarz@hhi.de>
 *
 * \date
 *    12. April 2001
 **************************************************************************
 */

#include <stdlib.h>
#include <math.h>
#include <memory.h>
#include <assert.h>
#include "rdopt_coding_state.h"
#include "elements.h"
#include "refbuf.h"
#include "intrarefresh.h"
#include "abt.h"

extern const byte PRED_IPRED[10][10][9];

extern       int  QP2QUANT  [40];

//==== MODULE PARAMETERS ====
int   best_mode;
int   rec_mbY[16][16], rec_mbU[8][8], rec_mbV[8][8], rec_mbY8x8[16][16];    // reconstruction values
int   mpr8x8[16][16];
int   ****cofAC=NULL, ****cofAC8x8=NULL;        // [8x8block][4x4block][level/run][scan_pos]
int   ***cofDC=NULL;                       // [yuv][level/run][scan_pos]
int   **cofAC4x4=NULL, ****cofAC4x4intern=NULL; // [level/run][scan_pos]
int   cbp, cbp8x8, cnt_nonz_8x8;
int   cbp_blk, cbp_blk8x8;
int   frefframe[4], brefframe[4], b8mode[4], b8pdir[4];
int   best8x8mode [4];                // [block]
int   best8x8pdir [MAXMODE][4];       // [mode][block]
int   best8x8ref  [MAXMODE][4];       // [mode][block]
int   b8_ipredmode[16], b8_intra_pred_modes[16];
CSptr cs_mb=NULL, cs_b8=NULL, cs_cm=NULL, cs_imb=NULL, cs_ib8=NULL, cs_ib4=NULL, cs_pc=NULL;
int   best_abt_mode[MAXMODE][4];      // [mode][block]
int   ABT_MODE[4];


/*!
 ************************************************************************
 * \brief
 *    delete structure for RD-optimized mode decision
 ************************************************************************
 */
void clear_rdopt ()
{
  free_mem_DCcoeff (cofDC);
  free_mem_ACcoeff (cofAC);
  free_mem_ACcoeff (cofAC8x8);
  free_mem_ACcoeff (cofAC4x4intern);

  // structure for saving the coding state
  delete_coding_state (cs_mb);
  delete_coding_state (cs_b8);
  delete_coding_state (cs_cm);
  delete_coding_state (cs_imb);
  delete_coding_state (cs_ib8);
  delete_coding_state (cs_ib4);
  delete_coding_state (cs_pc);
}


/*!
 ************************************************************************
 * \brief
 *    create structure for RD-optimized mode decision
 ************************************************************************
 */
void
init_rdopt ()
{
  get_mem_DCcoeff (&cofDC);
  get_mem_ACcoeff (&cofAC);
  get_mem_ACcoeff (&cofAC8x8);
  get_mem_ACcoeff (&cofAC4x4intern);
  cofAC4x4 = cofAC4x4intern[0][0];

  // structure for saving the coding state
  cs_mb  = create_coding_state ();
  cs_b8  = create_coding_state ();
  cs_cm  = create_coding_state ();
  cs_imb = create_coding_state ();
  cs_ib8 = create_coding_state ();
  cs_ib4 = create_coding_state ();
  cs_pc  = create_coding_state ();
}



/*! 
 *************************************************************************************
 * \brief
 *    Updates the pixel map that shows, which reference frames are reliable for
 *    each MB-area of the picture.
 *
 * \note
 *    The new values of the pixel_map are taken from the temporary buffer refresh_map
 *
 *************************************************************************************
 */
void UpdatePixelMap()
{
  int mx,my,y,x,i,j;
  if (img->type==INTRA_IMG)
  {
    for (y=0; y<img->height; y++)
    for (x=0; x<img->height; x++)
    {
      pixel_map[y][x]=1;
    }
  }
  else
  {
    for (my=0; my<img->height/8; my++)
    for (mx=0; mx<img->width/8;  mx++)
    {
      j = my*8 + 8;
      i = mx*8 + 8;
      if (refresh_map[my][mx])
      {
        for (y=my*8; y<j; y++)
        for (x=mx*8; x<i; x++)  pixel_map[y][x] = 1;
      }
      else
      {
        for (y=my*8; y<j; y++)
        for (x=mx*8; x<i; x++)  pixel_map[y][x] = min(pixel_map[y][x]+1, input->no_multpred+1);
      }
    }
  }
}

/*! 
 *************************************************************************************
 * \brief
 *    Checks if a given reference frame is reliable for the current 
 *    macroblock, given the motion vectors that the motion search has 
 *    returned.
 *
 * \param ref_frame
 *    The number of the reference frame that we want to check
 *
 * \return
 *    If the return value is 1, the reference frame is reliable. If it 
 *    is 0, then it is not reliable.
 *
 * \note
 *    A specific area in each reference frame is assumed to be unreliable
 *    if the same area has been intra-refreshed in a subsequent frame.
 *    The information about intra-refreshed areas is kept in the pixel_map.
 *
 *************************************************************************************
 */
int CheckReliabilityOfRef (int block, int ref, int mode)
{
  int y,x, block_y, block_x, dy, dx, y_pos, x_pos, yy, xx, pres_x, pres_y;
  int maxold_x  = img->width-1;
  int maxold_y  = img->height-1;
  int ref_frame = ref+1;

  int by0 = (mode>=4?2*(block/2):mode==2?2*block:0);
  int by1 = by0 + (mode>=4||mode==2?2:4);
  int bx0 = (mode>=4?2*(block%2):mode==3?2*block:0);
  int bx1 = bx0 + (mode>=4||mode==3?2:4);

  for (block_y=by0; block_y<by1; block_y++)
    for (block_x=bx0; block_x<bx1; block_x++)
    {
      y_pos  = img->all_mv[block_x][block_y][ref][mode][1];
      y_pos += (img->block_y+block_y) * BLOCK_SIZE * 4;
      x_pos  = img->all_mv[block_x][block_y][ref][mode][0];
      x_pos += (img->block_x+block_x) * BLOCK_SIZE * 4;
      
      /* Here we specify which pixels of the reference frame influence
         the reference values and check their reliability. This is
         based on the function Get_Reference_Pixel */
      
      dy = y_pos & 3;
      dx = x_pos & 3;

      y_pos = (y_pos-dy)/4;
      x_pos = (x_pos-dx)/4;

      if (dy==0 && dx==0) //full-pel
      {
        for (y=0 ; y < BLOCK_SIZE ; y++)
          for (x=0 ; x < BLOCK_SIZE ; x++)
            if (pixel_map[max(0,min(maxold_y,y_pos+y))][max(0,min(maxold_x,x_pos+x))] < ref_frame)
              return 0;
      }
      else if (dx == 3 && dy == 3)  /* funny position */
      {
        for (y=0 ; y < BLOCK_SIZE ; y++)
          for (x=0 ; x < BLOCK_SIZE ; x++)
            if (pixel_map[max(0,min(maxold_y,y_pos+y))  ][max(0,min(maxold_x,x_pos+x))  ] < ref_frame ||
                pixel_map[max(0,min(maxold_y,y_pos+y))  ][max(0,min(maxold_x,x_pos+x+1))] < ref_frame ||
                pixel_map[max(0,min(maxold_y,y_pos+y+1))][max(0,min(maxold_x,x_pos+x+1))] < ref_frame ||
                pixel_map[max(0,min(maxold_y,y_pos+y+1))][max(0,min(maxold_x,x_pos+x))  ] < ref_frame)
              return 0;
      }
      else  /* other positions */
      {
        if (dy == 0) 
        {
          for (y=0 ; y < BLOCK_SIZE ; y++)
            for (x=0 ; x < BLOCK_SIZE ; x++)
            {
              pres_y = max(0,min(maxold_y,y_pos+y));
              for(xx=-2;xx<4;xx++) {
                pres_x = max(0,min(maxold_x,x_pos+x+xx));
                if (pixel_map[pres_y][pres_x] < ref_frame)
                  return 0;
              }
            }
        }

        else if (dx == 0) 
        {
          for (y=0 ; y < BLOCK_SIZE ; y++)
            for (x=0 ; x < BLOCK_SIZE ; x++)
            {
              pres_x = max(0,min(maxold_x,x_pos+x));
              for(yy=-2;yy<4;yy++) {
                pres_y = max(0,min(maxold_y,y_pos+yy+y));
                if (pixel_map[pres_y][pres_x] < ref_frame)
                  return 0;
              }
            }
        }
        else if (dx == 2) 
        {
          for (y=0 ; y < BLOCK_SIZE ; y++)
            for (x=0 ; x < BLOCK_SIZE ; x++)
            {
              for(yy=-2;yy<4;yy++) {
                pres_y = max(0,min(maxold_y,y_pos+yy+y));
                for(xx=-2;xx<4;xx++) {
                  pres_x = max(0,min(maxold_x,x_pos+xx+x));
                  if (pixel_map[pres_y][pres_x] < ref_frame)
                    return 0;
                }
              }
            }
        }
        else if (dy == 2) 
        {
          for (y=0 ; y < BLOCK_SIZE ; y++)
            for (x=0 ; x < BLOCK_SIZE ; x++)
            {
              for(xx=-2;xx<4;xx++) {
                pres_x = max(0,min(maxold_x,x_pos+xx+x));
                for(yy=-2;yy<4;yy++) {
                  pres_y = max(0,min(maxold_y,y_pos+yy+y));
                  if (pixel_map[pres_y][pres_x] < ref_frame)
                    return 0;
                }
              }
            }
        }
        else 
        {
          for (y=0 ; y < BLOCK_SIZE ; y++)
            for (x=0 ; x < BLOCK_SIZE ; x++)
            {
              pres_y = dy == 1 ? y_pos+y : y_pos+y+1;
              pres_y = max(0,min(maxold_y,pres_y));

              for(xx=-2;xx<4;xx++) {
                pres_x = max(0,min(maxold_x,x_pos+xx+x));
                if (pixel_map[pres_y][pres_x] < ref_frame)
                  return 0;
              }

              pres_x = dx == 1 ? x_pos+x : x_pos+x+1;
              pres_x = max(0,min(maxold_x,pres_x));

              for(yy=-2;yy<4;yy++) {
                pres_y = max(0,min(maxold_y,y_pos+yy+y));
                if (pixel_map[pres_y][pres_x] < ref_frame)
                  return 0;
              }
            }
        }

      }
    }

  return 1;
}



/*! 
 *************************************************************************************
 * \brief
 *    R-D Cost for an 4x4 Intra block
 *************************************************************************************
 */
double
RDCost_for_4x4IntraBlocks (int*    nonzero,
                           int     b8,
                           int     b4,
                           int    ipmode,
                           double  lambda,
                           double  min_rdcost)
{
  double  rdcost;
  int     dummy, x, y, rate;
  int     distortion  = 0;
  int     block_x     = 8*(b8%2)+4*(b4%2);
  int     block_y     = 8*(b8/2)+4*(b4/2);
  int     pic_pix_x   = img->pix_x+block_x;
  int     pic_pix_y   = img->pix_y+block_y;
  int     pic_block_x = pic_pix_x/4;
  int     pic_block_y = pic_pix_y/4;
  int     even_block  = b4%2;

  Slice          *currSlice    =  img->currentSlice;
  Macroblock     *currMB       = &img->mb_data[img->current_mb_nr];
  SyntaxElement  *currSE       = &img->MB_SyntaxElements[currMB->currSEnr];
  const int      *partMap      = assignSE2partition[input->partition_mode];
  DataPartition  *dataPart;


  //===== perform DCT, Q, IQ, IDCT, Reconstruction =====
  *nonzero = dct_luma (block_x, block_y, &dummy, 1);

  //===== get distortion (SSD) of 4x4 block =====
  for (y=pic_pix_y; y<pic_pix_y+4; y++)
  for (x=pic_pix_x; x<pic_pix_x+4; x++)  distortion += img->quad [imgY_org[y][x] - imgY[y][x]];


  //===== RATE for INTRA PREDICTION MODE  (SYMBOL MODE MUST BE SET TO UVLC) =====
  if (even_block)
  {
    // EVEN BLOCK: the values the of syntax element can be set correctly
    currSE->value1 = PRED_IPRED[img->ipredmode[pic_block_x  ][pic_block_y  ]+1]
                               [img->ipredmode[pic_block_x-1][pic_block_y+1]+1]
                               [img->ipredmode[pic_block_x  ][pic_block_y+1]];
    currSE->value2 = PRED_IPRED[img->ipredmode[pic_block_x+1][pic_block_y  ]+1]
                               [img->ipredmode[pic_block_x  ][pic_block_y+1]+1][ipmode];
  }
  else
  {
    // ODD BLOCK:  the second value of the syntax element cannot be set, since
    //             it will be known only after the next step. For estimating
    //             the rate the corresponding prediction mode is set to zero
    //             (minimum rate).
    currSE->value1 = PRED_IPRED[img->ipredmode[pic_block_x+1][pic_block_y  ]+1]
                               [img->ipredmode[pic_block_x  ][pic_block_y+1]+1][ipmode];
    currSE->value2 = 0; // value with maximum probability
  }
  //--- set position and type ---
  currSE->context = 4*b8 + 2*(b4/2);
  currSE->type    = SE_INTRAPREDMODE;
  //--- set function pointer ----
  if (input->symbol_mode == UVLC)    currSE->mapping = intrapred_linfo;
  else                               currSE->writing = writeIntraPredMode2Buffer_CABAC;
  //--- choose data partition ---
  if (img->type!=B_IMG)   dataPart = &(currSlice->partArr[partMap[SE_INTRAPREDMODE]]);
  else                    dataPart = &(currSlice->partArr[partMap[SE_BFRAME]]);
  //--- encode and update rate ---
  dataPart->writeSyntaxElement (currSE, dataPart);
  rate = currSE->len;
  currSE++;
  currMB->currSEnr++;

  //===== RATE for LUMINANCE COEFFICIENTS =====
  if (input->symbol_mode == UVLC 
	  //&&(input->abt != INTER_INTRA_ABT && input->abt != INTER_ABT)
	  )
  {
    rate  += writeCoeff4x4_CAVLC (LUMA, b8, b4, 0);
  }
  else
  {
    rate  += writeLumaCoeff4x4 (b8, b4, 1);
  }
  rdcost = (double)distortion + lambda*(double)rate;

  return rdcost;
}

/*! 
 *************************************************************************************
 * \brief
 *    Mode Decision for an 4x4 Intra block
 *************************************************************************************
 */
int
Mode_Decision_for_4x4IntraBlocks (int  b8,  int  b4,  double  lambda,  int*  min_cost)
{
  int     ipmode, best_ipmode = 0, i, j, k, x, y, cost, dummy;
  int     c_nz, nonzero = 0, rec4x4[4][4], diff[16];
  double  rdcost;
  int     block_x     = 8*(b8%2)+4*(b4%2);
  int     block_y     = 8*(b8/2)+4*(b4/2);
  int     pic_pix_x   = img->pix_x+block_x;
  int     pic_pix_y   = img->pix_y+block_y;
  int     pic_block_x = pic_pix_x/4;
  int     pic_block_y = pic_pix_y/4;
  double  min_rdcost  = 1e30;


  *min_cost = (1<<20);


  //===== INTRA PREDICTION FOR 4x4 BLOCK =====
  intrapred_luma (pic_pix_x, pic_pix_y);


  //===== LOOP OVER ALL 4x4 INTRA PREDICTION MODES =====
  for (ipmode=0; ipmode<NO_INTRA_PMODE; ipmode++)
  {
    if ((ipmode==DC_PRED || ipmode==HOR_PRED  || img->ipredmode[pic_block_x+1][pic_block_y] >= 0) &&
        (ipmode==DC_PRED || ipmode==VERT_PRED || img->ipredmode[pic_block_x][pic_block_y+1] >= 0)  )
    {
      if (!input->rdopt)
      {
        for (k=j=0; j<4; j++)
          for (i=0; i<4; i++, k++)
          {
            diff[k] = imgY_org[pic_pix_y+j][pic_pix_x+i] - img->mprr[ipmode][j][i];
          }
        cost  = (int)floor(2 * lambda * PRED_IPRED[img->ipredmode[pic_block_x+1][pic_block_y  ]+1]
                                                  [img->ipredmode[pic_block_x  ][pic_block_y+1]+1][ipmode] +0.4999);
        cost += SATD (diff, input->hadamard);
        if (cost < *min_cost)
        {
          best_ipmode = ipmode;
          *min_cost   = cost;
        }
      }
      else
      {
        // get prediction and prediction error
        for (j=0; j<4; j++)
        for (i=0; i<4; i++)
        {
          img->mpr[block_x+i][block_y+j]  = img->mprr[ipmode][j][i];
          img->m7[i][j]                   = imgY_org[pic_pix_y+j][pic_pix_x+i] - img->mprr[ipmode][j][i];
        }

        //===== store the coding state =====
        store_coding_state (cs_cm);
        // get and check rate-distortion cost
        if ((rdcost = RDCost_for_4x4IntraBlocks (&c_nz, b8, b4, ipmode, lambda, min_rdcost)) < min_rdcost)
        {
          //--- set coefficients ---
          for (j=0; j<2; j++)
          for (i=0; i<18;i++)  cofAC4x4[j][i]=img->cofAC[b8][b4][j][i];

          //--- set reconstruction ---
          for (y=0; y<4; y++)
          for (x=0; x<4; x++)  rec4x4[y][x] = imgY[pic_pix_y+y][pic_pix_x+x];

          //--- flag if dct-coefficients must be coded ---
          nonzero = c_nz;

          //--- set best mode update minimum cost ---
          min_rdcost  = rdcost;
          best_ipmode = ipmode;
        }
        reset_coding_state (cs_cm);
      }
    }
  }

  //===== set intra mode prediction =====
  img->ipredmode[pic_block_x+1][pic_block_y+1] = best_ipmode;
  img->mb_data[img->current_mb_nr].intra_pred_modes[4*b8+b4]=PRED_IPRED[img->ipredmode[pic_block_x+1][pic_block_y  ]+1]
                                                                       [img->ipredmode[pic_block_x  ][pic_block_y+1]+1]
                                                                       [best_ipmode];
  if (!input->rdopt)
  {
    // get prediction and prediction error
    for (j=0; j<4; j++)
    for (i=0; i<4; i++)
    {
      img->mpr[block_x+i][block_y+j]  = img->mprr[best_ipmode][j][i];
      img->m7[i][j]                   = imgY_org[pic_pix_y+j][pic_pix_x+i] - img->mprr[best_ipmode][j][i];
    }
    nonzero = dct_luma (block_x, block_y, &dummy, 1);
  }
  else
  {
    //===== restore coefficients =====
    for (j=0; j<2; j++)
    for (i=0; i<18;i++)  img->cofAC[b8][b4][j][i]=cofAC4x4[j][i];
  
    //===== restore reconstruction and prediction (needed if single coeffs are removed) =====
    for (y=0; y<4; y++)
    for (x=0; x<4; x++)
    {
      imgY[pic_pix_y+y][pic_pix_x+x] = rec4x4[y][x];
      img->mpr[block_x+x][block_y+y] = img->mprr[best_ipmode][y][x];
    }
  }

  return nonzero;
}


/*! 
 *************************************************************************************
 * \brief
 *    Mode Decision for an 8x8 Intra block
 *************************************************************************************
 */
int Mode_Decision_for_8x8IntraBlocks(int b8,double lambda,int *cost,int b_useABT,int only_this_ABT_mode)
{
 int nonzero,best_nonzero;
 int s,i,j,b4,need_restore;
 int b8_x,b8_y;
 int pic_block4_x,pic_block4_y;
 int px,py,bs_x,bs_y;
 int cost4x4;
 int ABTmode,ABTmode_max;
 int loc_cost0,loc_cost,best_ABT_cost;
 Macroblock *currMB;
 int best_ABTmode;
 int best_ipredmode[2][2],best_intra_pred_modes[4],best_mpr[8][8],best_imgY[8][8];
 int cbp_blk_part,cbp_blk_mask,best_cbp;
 int best_cofAC[4][2][65];

  currMB=img->mb_data+img->current_mb_nr;

  loc_cost0=(int)floor(6.0 * lambda + 0.4999);


  if(b_useABT)
  {
    //using ABT. loop modes, set best
    cbp_blk_mask=0x0033;
    if(b8&1)cbp_blk_mask<<=2;
    if(b8&2)cbp_blk_mask<<=8;
    ABTmode_max=4;
    if(only_this_ABT_mode>=0)
      ABTmode_max=only_this_ABT_mode+1;
    else
      only_this_ABT_mode=0;	//used for start of loop.
    b8_x=(b8&1)<<3;
    b8_y=(b8&2)<<2;
    pic_block4_x=(img->mb_x<<2)+((b8&1)<<1);
    pic_block4_y=(img->mb_y<<2)+(b8&2);
    best_ABT_cost=0x7FFFFFFFL;	//max int value
    best_ABTmode=-1;
    cbp_blk_part=0;
    best_nonzero=0;
    need_restore=0;
    for( ABTmode=only_this_ABT_mode ; ABTmode<ABTmode_max ; ABTmode++ )
    {
      currMB->abt_mode[b8]=ABTmode;
      nonzero=0;
      loc_cost=loc_cost0;
      bs_x=ABT_TRSIZE[ABTmode][0];
      bs_y=ABT_TRSIZE[ABTmode][1];
      for(py=0;py<8;py+=bs_y)
        for(px=0;px<8;px+=bs_x)
        {
          if(Mode_Decision_for_ABT_IntraBlocks(b8,((py>>1)&2)|((px>>2)&1),lambda,&cost4x4,bs_x,bs_y))
            nonzero=1;
          loc_cost+=cost4x4;
        }
      if(loc_cost<best_ABT_cost)
      {
        //is better than last (or ist first try).
        best_ABT_cost=loc_cost;
        best_nonzero=nonzero;
        best_ABTmode=ABTmode;
        //if not last mode, save (when checking only one mode, no save/restore is needed at all)
        if(ABTmode+1<ABTmode_max)
        {
          //save state of 8*8 block
          for(j=0;j<2;j++)
            for(i=0;i<2;i++)
              best_ipredmode[i][j]=img->ipredmode[ 1 + (img->mb_x<<2) + (b8_x>>2)+i ][ 1 + (img->mb_y<<2) + (b8_y>>2)+j ];
          for(s=0;s<4;s++)
            best_intra_pred_modes[s]=currMB->intra_pred_modes[(b8<<2)+s];
          for(j=0;j<8;j++)
            for(i=0;i<8;i++)
            {
              best_mpr[i][j]=img->mpr[b8_x+i][b8_y+j];
              best_imgY[j][i]=imgY[(img->mb_y<<4)+b8_y+j][(img->mb_x<<4)+b8_x+i];
            }
          for(s=0;s<4;s++)
            for(i=0;i<2;i++)
              for(j=0;j<65;j++)
                best_cofAC[s][i][j]=img->cofAC[b8][s][i][j];
          cbp_blk_part = currMB->cbp_blk & cbp_blk_mask ;
          best_cbp=currMB->cbp&(1<<b8);
        }
        need_restore=0;
      }else
        need_restore=1;
    }
    assert( best_ABTmode >= 0 );	//must have found one.
    loc_cost=best_ABT_cost;
    nonzero=best_nonzero;
    if(need_restore)
    {
      //restore best that was found. (only if not already valid)
      for(j=0;j<2;j++)
        for(i=0;i<2;i++)
          img->ipredmode[ 1 + (img->mb_x<<2) + (b8_x>>2)+i ][ 1 + (img->mb_y<<2) + (b8_y>>2)+j ]=best_ipredmode[i][j];
      for(s=0;s<4;s++)
        currMB->intra_pred_modes[(b8<<2)+s]=best_intra_pred_modes[s];
      for(j=0;j<8;j++)
        for(i=0;i<8;i++)
        {
          img->mpr[b8_x+i][b8_y+j]=best_mpr[i][j];
          imgY[(img->mb_y<<4)+b8_y+j][(img->mb_x<<4)+b8_x+i]=best_imgY[j][i];
        }
      for(s=0;s<4;s++)
        for(i=0;i<2;i++)
          for(j=0;j<65;j++)
            img->cofAC[b8][s][i][j]=best_cofAC[s][i][j];
      currMB->cbp_blk=( currMB->cbp_blk & ~cbp_blk_mask )|cbp_blk_part;
      currMB->cbp = ((currMB->cbp)&(~(1<<b8))) | best_cbp;
    }
    currMB->abt_mode[b8]=currMB->abt_pred_mode[b8]=best_ABTmode;
  }else{
    //not using ABT. code blocks.
    loc_cost=loc_cost0;
    nonzero=0;
    for (b4=0; b4<4; b4++)
    {
      if (Mode_Decision_for_4x4IntraBlocks (b8,b4,lambda,&cost4x4))
        nonzero=1;
      loc_cost += cost4x4;
    }
  }//end if(useABT)

  *cost=loc_cost;
  return nonzero;
}
/*
{
  int  nonzero=0, b4;
  int  cost4x4;
  
  *cost = (int)floor(6.0 * lambda + 0.4999);

  for (b4=0; b4<4; b4++)
  {
    if (Mode_Decision_for_4x4IntraBlocks (b8, b4, lambda, &cost4x4))
    {
      nonzero        = 1;
    }
    *cost += cost4x4;
  }

  return nonzero;
}
*/

/*! 
 *************************************************************************************
 * \brief
 *    4x4 Intra mode decision for an macroblock
 *************************************************************************************
 */
int
Mode_Decision_for_Intra4x4Macroblock (double lambda,  int* cost)
{
 int cbp,b8,cost8x8;
 int i,j,s;
 int useABT,modeABT,loc_cost;
 int best_modeABT,best_cost,best_cbp;
 int best_ipredmode[4][4],best_intra_pred_modes[16],best_mpr[16][16],best_imgY[16][16];
 int best_cofAC[4][4][2][65];
 int need_restore;
 Macroblock *currMB;

  currMB = img->mb_data+img->current_mb_nr ;

  //abt on all or none. no half intra MBs.
  assert( currMB->useABT[0]==currMB->useABT[1] && currMB->useABT[1]==currMB->useABT[2] && currMB->useABT[2]==currMB->useABT[3] );
  useABT=currMB->useABT[0];

  need_restore=0;
  best_cbp=0;
  best_cost=0x7FFFFFFFL;	//max int
  best_modeABT=-1;
  for( modeABT=(useABT?0:3); modeABT<4 ; modeABT++ )	//for non-ABT, this just uses mode 3.
  {
    loc_cost=0;cbp=0;
    for (b8=0; b8<4; b8++)
    {
      if(Mode_Decision_for_8x8IntraBlocks(b8, lambda, &cost8x8,useABT,modeABT))//in case of non-ABT, last param is ignored.
        cbp |= (1<<b8);
      loc_cost+=cost8x8;
    }
    if(loc_cost<best_cost)//choose if better
    {
      best_cost=loc_cost;
      best_modeABT=modeABT;
      best_cbp=cbp;
      //if not last mode, save (when checking only one mode, no save/restore is needed at all)
      if(modeABT+1<4)
      {
        //save state of 16*16 block
        for(j=0;j<4;j++)
          for(i=0;i<4;i++)
            best_ipredmode[i][j]=img->ipredmode[ 1 + (img->mb_x<<2) + i ][ 1 + (img->mb_y<<2) + j ];
        for(s=0;s<16;s++)
          best_intra_pred_modes[s]=currMB->intra_pred_modes[s];
        for(j=0;j<16;j++)
          for(i=0;i<16;i++)
          {
            best_mpr[i][j]=img->mpr[i][j];
            best_imgY[j][i]=imgY[(img->mb_y<<4)+j][(img->mb_x<<4)+i];
          }
        for(b8=0;b8<4;b8++)
          for(s=0;s<4;s++)
            for(i=0;i<2;i++)
              for(j=0;j<65;j++)
                best_cofAC[b8][s][i][j]=img->cofAC[b8][s][i][j];
      }
      need_restore=0;
    }else
      need_restore=1;
  }
  assert(best_modeABT>=0);	//must have found one.
  if(need_restore)
  {
    //restore best that was found. (only if not already valid)
    for(j=0;j<4;j++)
      for(i=0;i<4;i++)
        img->ipredmode[ 1 + (img->mb_x<<2) + i ][ 1 + (img->mb_y<<2) + j ]=best_ipredmode[i][j];
    for(s=0;s<16;s++)
      currMB->intra_pred_modes[s]=best_intra_pred_modes[s];
    for(j=0;j<16;j++)
      for(i=0;i<16;i++)
      {
        img->mpr[i][j]=best_mpr[i][j];
        imgY[(img->mb_y<<4)+j][(img->mb_x<<4)+i]=best_imgY[j][i];
      }
    for(b8=0;b8<4;b8++)
      for(s=0;s<4;s++)
        for(i=0;i<2;i++)
          for(j=0;j<65;j++)
            img->cofAC[b8][s][i][j]=best_cofAC[b8][s][i][j];
  }
  if(useABT)
    for(b8=0;b8<4;b8++)
      currMB->abt_mode[b8]=currMB->abt_pred_mode[b8]=best_modeABT;

  *cost=best_cost;
  return best_cbp;
}
/*
{
  int  cbp=0, b8, cost8x8;

  for (*cost=0, b8=0; b8<4; b8++)
  {
    if (Mode_Decision_for_8x8IntraBlocks (b8, lambda, &cost8x8))
    {
      cbp |= (1<<b8);
    }
    *cost += cost8x8;
  }

  return cbp;
}
*/




/*! 
 *************************************************************************************
 * \brief
 *    R-D Cost for an 8x8 Partition
 *************************************************************************************
 */
double
RDCost_for_8x8blocks (int*    cnt_nonz,   // --> number of nonzero coefficients
                      int*    cbp_blk,    // --> cbp blk
                      double  lambda,     // <-- lagrange multiplier
                      int     block,      // <-- 8x8 block number
                      int     mode,       // <-- partitioning mode
                      int     pdir,       // <-- prediction direction
                      int     ref)        // <-- reference frame
{
  int  i, j, k;
  int  rate=0, distortion=0;
  int  dummy, mrate;
  int  fw_mode, bw_mode;
  int  cbp     = 0;
  int  pax     = 8*(block%2);
  int  pay     = 8*(block/2);
  int  i0      = pax/4;
  int  j0      = pay/4;
  int  bframe  = (img->type==B_IMG);
  int  intra   = (mode==IBLOCK);
  int  direct  = (bframe && mode==0);
  int  b8value = B8Mode2Value (mode, pdir);

  Macroblock    *currMB    = &img->mb_data[img->current_mb_nr];
  SyntaxElement *currSE    = &img->MB_SyntaxElements[currMB->currSEnr];
  Slice         *currSlice = img->currentSlice;
  DataPartition *dataPart;
  const int     *partMap   = assignSE2partition[input->partition_mode];

  EncodingEnvironmentPtr eep_dp;


  //=====
  //=====  GET COEFFICIENTS, RECONSTRUCTIONS, CBP
  //=====
  if (!intra)
  {
    if (direct)
    {
      *cnt_nonz = LumaResidualCoding8x8 (&cbp, cbp_blk, block, 0, 0, max(0,refFrArr[img->block_y+j0][img->block_x+i0]),currMB->abt_mode[block]);
    }
    else
    {
      fw_mode   = (pdir==0||pdir==2 ? mode : 0);
      bw_mode   = (pdir==1||pdir==2 ? mode : 0);
      *cnt_nonz = LumaResidualCoding8x8 (&cbp, cbp_blk, block, fw_mode, bw_mode, ref, currMB->abt_mode[block]);
    }
  }

  //===== get residue =====
  if (input->rdopt==2 && !bframe)
  {
    // We need the reconstructed prediction residue for the simulated decoders.
    compute_residue_b8block (block, -1);
  }

  //=====
  //=====   GET DISTORTION
  //=====
  if (input->rdopt==2 && !bframe)
  {
    for (k=0; k<input->NoOfDecoders ;k++)
    {
      decode_one_b8block (k, P8x8, block, mode, ref);
      for (j=img->pix_y+pay; j<img->pix_y+pay+8; j++)
      for (i=img->pix_x+pax; i<img->pix_x+pax+8; i++)
      {
        distortion += img->quad[imgY_org[j][i] - decs->decY[k][j][i]];
      }
    }
    distortion /= input->NoOfDecoders;
  }
  else
  {
    for (j=img->pix_y+pay; j<img->pix_y+pay+8; j++)
    for (i=img->pix_x+pax; i<img->pix_x+pax+8; i++)
    {
      distortion += img->quad [imgY_org[j][i] - imgY[j][i]];
    }
  }

  //=====
  //=====   GET RATE
  //=====
  //----- block 8x8 mode -----
  if (input->symbol_mode == UVLC)
  {
    n_linfo2 (b8value, dummy, &mrate, &dummy);
    rate += mrate;
  }
  else
  {
    dataPart        = &(currSlice->partArr[partMap[SE_BFRAME]]);
    currSE->value1  = b8value;
    currSE->writing = writeB8_typeInfo2Buffer_CABAC;
    currSE->type    = SE_MBTYPE;
    dataPart->writeSyntaxElement (currSE, dataPart);
    rate += currSE->len;
    currSE++;
    currMB->currSEnr++;
  }

  //----- intra 4x4 prediction modes -----
  if (intra)
  {
    currMB->b8mode[block] = IBLOCK; // added for new routine
    rate += writeIntra4x4Modes (block);
  }

  //----- motion information -----
  if (!intra && !direct)
  {
    if ((input->no_multpred>1 || input->add_ref_frame>0) && (pdir==0 || pdir==2))
      rate  += writeReferenceFrame (mode, i0, j0, ref);

    if (pdir==0 || pdir==2)
      rate  += writeMotionVector8x8 (i0, j0, i0+2, j0+2, ref, mode);
    if (pdir==1 || pdir==2)
      rate  += writeMotionVector8x8 (i0, j0, i0+2, j0+2,  -1, mode);
  }

  //----- coded block pattern (for CABAC only) -----
  if (input->symbol_mode == CABAC)
  {
    dataPart = &(currSlice->partArr[partMap[SE_CBP_INTER]]);
    eep_dp   = &(dataPart->ee_cabac);
    mrate    = arienco_bits_written (eep_dp);
    writeCBP_BIT_CABAC (block, ((*cnt_nonz>0)?1:0), cbp8x8, currMB, (intra?0:1), eep_dp);
    mrate    = arienco_bits_written (eep_dp) - mrate;
    rate    += mrate;
  }

  //----- luminance coefficients -----
  if (*cnt_nonz)
  {
    if (currMB->useABT[block] && !intra) { currMB->cbp = cbp; } // needed for writeLumaCoeffABT_B8()
    rate += writeLumaCoeff8x8 (block, intra, currMB->abt_mode[block]);
  }

  return (double)distortion + lambda * (double)rate;
}


/*! 
 *************************************************************************************
 * \brief
 *    Gets mode offset for intra16x16 mode
 *************************************************************************************
 */
int
I16Offset (int cbp, int i16mode)
{
  return (cbp&15?13:1) + i16mode + ((cbp&0x30)>>2);
}


/*! 
 *************************************************************************************
 * \brief
 *    Sets modes and reference frames for an macroblock
 *************************************************************************************
 */
void
SetModesAndRefframeForBlocks (int mode)
{
  int i,j,k,l;
  Macroblock *currMB = &img->mb_data[img->current_mb_nr];
  int bframe         = (img->type==B_IMG);
  int spframe        = (img->types==SP_IMG);

  //--- macroblock type ---
  currMB->mb_type = mode;

  //--- block 8x8 mode and prediction direction ---
  switch (mode)
  {
    case 0:
      for(i=0;i<4;i++)
      {
        currMB->b8mode[i] = 0;
        currMB->b8pdir[i] = (bframe?2:0);
        if (bframe && input->abt)              // might need to revisit dependency on input->abt
        {
          getDirectModeABT(i);
          currMB->useABT  [i] = (currMB->abt_mode[i]!=ABT_OFF);
        }
        else if (spframe && input->abt)        // might need to revisit dependency on input->abt
        {
          currMB->useABT  [i] = !NO_ABT;
          currMB->abt_mode[i] = B8x8;
        }
        else
        {
          currMB->useABT  [i] = NO_ABT;
          currMB->abt_mode[i] = ABT_OFF;
        }
      }
      break;
    case 1:
    case 2:
    case 3:
      for(i=0;i<4;i++)
      {
        currMB->b8mode[i] = mode;
        currMB->b8pdir[i] = best8x8pdir[mode][i];
        currMB->useABT  [i] = (best_abt_mode[mode][i]!=ABT_OFF);
        currMB->abt_mode[i] = best_abt_mode[mode][i];
      }
      break;
    case P8x8:
      for(i=0;i<4;i++)
      {
        currMB->b8mode[i]   = best8x8mode[i];
        currMB->b8pdir[i]   = best8x8pdir[mode][i];
        currMB->useABT  [i] = (best_abt_mode[mode][i]!=ABT_OFF);
        currMB->abt_mode[i] = best_abt_mode[mode][i];
      }
      break;
    case I4MB:
      for(i=0;i<4;i++)
      {
        currMB->b8mode[i] = IBLOCK;
        currMB->b8pdir[i] = -1;
        currMB->useABT  [i] = (best_abt_mode[mode][i]!=ABT_OFF);
        currMB->abt_mode[i] = best_abt_mode[mode][i];
     }
      break;
    case I16MB:
      for(i=0;i<4;i++)
      {
        currMB->b8mode[i] = 0;
        currMB->b8pdir[i] = -1;
        currMB->useABT  [i] = NO_ABT;    // no ABT with I16MB
        currMB->abt_mode[i] = ABT_OFF;   // no ABT with I16MB
      }
      break;
    default:
      printf ("Unsupported mode in SetModesAndRefframeForBlocks!\n");
      exit (1);
  }

#define IS_FW ((best8x8pdir[mode][k]==0 || best8x8pdir[mode][k]==2) && (mode!=P8x8 || best8x8mode[k]!=0 || !bframe))
#define IS_BW ((best8x8pdir[mode][k]==1 || best8x8pdir[mode][k]==2) && (mode!=P8x8 || best8x8mode[k]!=0))
  //--- reference frame arrays ---
  if (mode==0 || mode==I4MB || mode==I16MB)
  {
    if (bframe)
    { 
      for (j=0;j<4;j++)
      for (i=0;i<4;i++)
      {
        if(!mode)
        {						
          fw_refFrArr[img->block_y+j][img->block_x+i] = refFrArr[img->block_y+j][img->block_x+i];
          bw_refFrArr[img->block_y+j][img->block_x+i] = min(refFrArr[img->block_y+j][img->block_x+i],0);
        }
        else
        {
          fw_refFrArr[img->block_y+j][img->block_x+i] = -1;
          bw_refFrArr[img->block_y+j][img->block_x+i] = -1;          
        }
      }
    }
    else
    {
      for (j=0;j<4;j++)
      for (i=0;i<4;i++)
      {
        refFrArr [img->block_y+j][img->block_x+i] = (mode==0?0:-1);
      }
    }
  }
  else
  {
    if (bframe)
    {
      for (j=0;j<4;j++)
      for (i=0;i<4;i++)
      {
        k = 2*(j/2)+(i/2);
        l = 2*(j%2)+(i%2);
        if(mode == P8x8 && best8x8mode[k]==0)
        {						
          fw_refFrArr[img->block_y+j][img->block_x+i] = refFrArr[img->block_y+j][img->block_x+i];
          bw_refFrArr[img->block_y+j][img->block_x+i] = min(refFrArr[img->block_y+j][img->block_x+i],0);
        }
        else
        {
          fw_refFrArr[img->block_y+j][img->block_x+i] = (IS_FW ? best8x8ref[mode][k] : -1);
          bw_refFrArr[img->block_y+j][img->block_x+i] = (IS_BW ?                   0 : -1);
        }
      }
    }
    else
    {
      for (j=0;j<4;j++)
      for (i=0;i<4;i++)
      {
        k = 2*(j/2)+(i/2);
        l = 2*(j%2)+(i%2);
        refFrArr [img->block_y+j][img->block_x+i] = (IS_FW ? best8x8ref[mode][k] : -1);
      }
    }
  }
#undef IS_FW
#undef IS_BW
}


/*! 
 *************************************************************************************
 * \brief
 *    Intra 16x16 mode decision
 *************************************************************************************
 */
void
Intra16x16_Mode_Decision (Macroblock* currMB, int* i16mode)
{
  intrapred_luma_2 ();   /* make intra pred for all 4 new modes */
  find_sad2 (i16mode);   /* get best new intra mode */
  currMB->cbp = dct_luma2 (*i16mode);
}



/*! 
 *************************************************************************************
 * \brief
 *    Sets Coefficients and reconstruction for an 8x8 block
 *************************************************************************************
 */
void SetCoeffAndReconstruction8x8 (Macroblock* currMB)
{
  int ****ip4, k, j, i;

  ip4=img->cofAC; img->cofAC=cofAC8x8; cofAC8x8=ip4;
  if (cnt_nonz_8x8<=5 && img->types!=SP_IMG)
  {
    currMB->cbp     = 0;
    currMB->cbp_blk = 0;
    for (j=0; j<16; j++)
    for (i=0; i<16; i++)  imgY[img->pix_y+j][img->pix_x+i] = mpr8x8[j][i];
  }
  else
  {
    currMB->cbp     = cbp8x8;
    currMB->cbp_blk = cbp_blk8x8;
    for (j=0; j<16; j++)
    for (i=0; i<16; i++)  imgY[img->pix_y+j][img->pix_x+i] = rec_mbY8x8[j][i];
  }

  //===== restore intra prediction modes for 8x8+ macroblock mode =====
  for (k=0, j=img->block_y+1; j<img->block_y+5; j++)
  for (     i=img->block_x+1; i<img->block_x+5; i++, k++)
  {
    img   ->ipredmode    [i][j] = b8_ipredmode       [k];
    currMB->intra_pred_modes[k] = b8_intra_pred_modes[k];
  }
}


/*! 
 *************************************************************************************
 * \brief
 *    Sets motion vectors for an macroblock
 *************************************************************************************
 */
void SetMotionVectorsMB (Macroblock* currMB, int bframe)
{
  int i, j, k, l, mode8, pdir8, ref, by, bx, bxr, dref;

  for (j=0; j<4; j++)
  for (i=0; i<4; i++)
  {
    mode8 = currMB->b8mode[k=2*(j/2)+(i/2)];
    pdir8 = currMB->b8pdir[k];
    l     = 2*(j%2)+(i%2);
    by    = img->block_y+j;
    bxr   = img->block_x+i;
    bx    = img->block_x+i+4;
    ref   = (bframe?fw_refFrArr:refFrArr)[by][bxr];

    if (!bframe)
    {
      if (mode8!=IBLOCK && ref != -1)
      {
        tmp_mv   [0][by][bx] = img->all_mv [i][j][ ref][mode8][0];
        tmp_mv   [1][by][bx] = img->all_mv [i][j][ ref][mode8][1];
      }
      else
      {
        tmp_mv   [0][by][bx] = 0;
        tmp_mv   [1][by][bx] = 0;
      }
    }
    else
    {
      if (pdir8==-1) // intra
      {
        tmp_fwMV [0][by][bx] = 0;
        tmp_fwMV [1][by][bx] = 0;
        tmp_bwMV [0][by][bx] = 0;
        tmp_bwMV [1][by][bx] = 0;
        dfMV     [0][by][bx] = 0;
        dfMV     [1][by][bx] = 0;
        dbMV     [0][by][bx] = 0;
        dbMV     [1][by][bx] = 0;
      }
      else if (pdir8==0) // forward
      {
        tmp_fwMV [0][by][bx] = img->all_mv [i][j][ ref][mode8][0];
        tmp_fwMV [1][by][bx] = img->all_mv [i][j][ ref][mode8][1];
        tmp_bwMV [0][by][bx] = 0;
        tmp_bwMV [1][by][bx] = 0;
        dfMV     [0][by][bx] = 0;
        dfMV     [1][by][bx] = 0;
        dbMV     [0][by][bx] = 0;
        dbMV     [1][by][bx] = 0;
      }
      else if (pdir8==1) // backward
      {
        tmp_fwMV [0][by][bx] = 0;
        tmp_fwMV [1][by][bx] = 0;
        tmp_bwMV [0][by][bx] = img->all_bmv[i][j][   0][mode8][0];
        tmp_bwMV [1][by][bx] = img->all_bmv[i][j][   0][mode8][1];
        dfMV     [0][by][bx] = 0;
        dfMV     [1][by][bx] = 0;
        dbMV     [0][by][bx] = 0;
        dbMV     [1][by][bx] = 0;
      }
      else if (mode8!=0) // bidirect
      {
        tmp_fwMV [0][by][bx] = img->all_mv [i][j][ ref][mode8][0];
        tmp_fwMV [1][by][bx] = img->all_mv [i][j][ ref][mode8][1];
        tmp_bwMV [0][by][bx] = img->all_bmv[i][j][   0][mode8][0];
        tmp_bwMV [1][by][bx] = img->all_bmv[i][j][   0][mode8][1];
        dfMV     [0][by][bx] = 0;
        dfMV     [1][by][bx] = 0;
        dbMV     [0][by][bx] = 0;
        dbMV     [1][by][bx] = 0;
      }
      else // direct
      {
        dref = max(0,refFrArr[by][bxr]);
        tmp_fwMV [0][by][bx] = dfMV     [0][by][bx] = img->all_mv [i][j][dref][    0][0];
        tmp_fwMV [1][by][bx] = dfMV     [1][by][bx] = img->all_mv [i][j][dref][    0][1];
        tmp_bwMV [0][by][bx] = dbMV     [0][by][bx] = img->all_bmv[i][j][   0][    0][0];
        tmp_bwMV [1][by][bx] = dbMV     [1][by][bx] = img->all_bmv[i][j][   0][    0][1];
      }
    }
  }
}



/*! 
 *************************************************************************************
 * \brief
 *    R-D Cost for a macroblock
 *************************************************************************************
 */
int
RDCost_for_macroblocks (double   lambda,      // <-- lagrange multiplier
                        int      mode,        // <-- modus (0-COPY/DIRECT, 1-16x16, 2-16x8, 3-8x16, 4-8x8(+), 5-Intra4x4, 6-Intra16x16)
                        double*  min_rdcost)  // <-> minimum rate-distortion cost
{
  int         i, j, k; //, k, ****ip4;
  int         i16mode, rate=0, distortion=0;
  double      rdcost;
  Macroblock  *currMB   = &img->mb_data[img->current_mb_nr];
  int         bframe    = (img->type==B_IMG);
  int         tmp_cc;
  int         use_of_cc =  (img->type!=INTRA_IMG &&  input->symbol_mode!=CABAC);
  int         cc_rate, dummy;


  //=====
  //=====  SET REFERENCE FRAMES AND BLOCK MODES
  //=====
  SetModesAndRefframeForBlocks (mode);

  //=====
  //=====  GET COEFFICIENTS, RECONSTRUCTIONS, CBP
  //=====
  if (mode<P8x8)
  {
    LumaResidualCoding ();
  }
  else if (mode==P8x8)
  {
    SetCoeffAndReconstruction8x8 (currMB);
  }
  else if (mode==I4MB)
  {
    for(i=0;i<4;i++)
      currMB->useABT[i]=(input->abt==INTER_INTRA_ABT);
    currMB->cbp = Mode_Decision_for_Intra4x4Macroblock (lambda, &dummy);
  }
  else if (mode==I16MB)
  {
    Intra16x16_Mode_Decision  (currMB, &i16mode);
  }

  if (input->rdopt==2 && !bframe)
  {
    // We need the reconstructed prediction residue for the simulated decoders.
    compute_residue_mb (mode==I16MB?i16mode:-1);
  }

  ChromaResidualCoding (&dummy);
  if (mode==I16MB)     img->i16offset = I16Offset  (currMB->cbp, i16mode);


  //=====
  //=====   GET DISTORTION
  //=====
  // LUMA
  if (input->rdopt==2 && !bframe)
  {
    for (k=0; k<input->NoOfDecoders ;k++)
    {
      decode_one_mb (k, currMB);
      for (j=img->pix_y; j<img->pix_y+MB_BLOCK_SIZE; j++)
      for (i=img->pix_x; i<img->pix_x+MB_BLOCK_SIZE; i++)
      {
        distortion += img->quad [imgY_org[j][i] - decs->decY[k][j][i]];
      }
    }
    distortion /= input->NoOfDecoders;
  }
  else
  {
    for (j=img->pix_y; j<img->pix_y+16; j++)
    for (i=img->pix_x; i<img->pix_x+16; i++)
    {
      distortion += img->quad [imgY_org[j][i] - imgY[j][i]];
    }
  }

  // CHROMA
  for (j=img->pix_c_y; j<img->pix_c_y+8; j++)
  for (i=img->pix_c_x; i<img->pix_c_x+8; i++)
  {
    distortion += img->quad [imgUV_org[0][j][i] - imgUV[0][j][i]];
    distortion += img->quad [imgUV_org[1][j][i] - imgUV[1][j][i]];
  }


  //=====   S T O R E   C O D I N G   S T A T E   =====
  //---------------------------------------------------
  store_coding_state (cs_cm);
  

  //=====
  //=====   GET RATE
  //=====
  //----- macroblock header -----
  if (use_of_cc)
  {
    if (currMB->mb_type!=0 || (img->type==B_IMG && currMB->cbp!=0))
    {
      // cod counter and macroblock mode are written ==> do not consider code counter
      tmp_cc = img->cod_counter;
      rate   = writeMBHeader ();
      n_linfo2 (tmp_cc, dummy, &cc_rate, &dummy);
      rate  -= cc_rate;
      img->cod_counter = tmp_cc;
    }
    else
    {
      // cod counter is just increased  ==> get additional rate
      n_linfo2 (img->cod_counter+1, dummy, &rate,    &dummy);
      n_linfo2 (img->cod_counter,   dummy, &cc_rate, &dummy);
      rate -= cc_rate;
    }
  }
  else
  {
    rate = writeMBHeader ();
  }
  if (mode)
  {
    //----- motion information -----
    rate  += writeMotionInfo2NAL  ();
  }
  if (mode || (bframe && (currMB->cbp!=0 || input->symbol_mode==CABAC)))
  {
    rate  += writeCBPandLumaCoeff ();
    rate  += writeChromaCoeff     ();
  }


  //=====   R E S T O R E   C O D I N G   S T A T E   =====
  //-------------------------------------------------------
  reset_coding_state (cs_cm);


  rdcost = (double)distortion + lambda * (double)rate;


  if (rdcost >= *min_rdcost)
  {
    return 0;
  }



  //=====   U P D A T E   M I N I M U M   C O S T   =====
  //-----------------------------------------------------
  *min_rdcost = rdcost;
  return 1;
}





/*! 
 *************************************************************************************
 * \brief
 *    Store macroblock parameters
 *************************************************************************************
 */
void
store_macroblock_parameters (int mode)
{
  int  i, j, k, ****i4p, ***i3p;
  Macroblock *currMB  = &img->mb_data[img->current_mb_nr];
  int        bframe   = (img->type==B_IMG);
  int        **frefar = (img->type==B_IMG ? fw_refFrArr : refFrArr);
  int        **brefar = bw_refFrArr;

  //--- store best mode ---
  best_mode = mode;
  for (i=0; i<4; i++)
  {
    b8mode[i]   = currMB->b8mode[i];
    b8pdir[i]   = currMB->b8pdir[i];
    ABT_MODE[i] = currMB->abt_mode[i];
  }

  //--- reconstructed blocks ----
  for (j=0; j<16; j++)
  for (i=0; i<16; i++)
  {
    rec_mbY[j][i] = imgY[img->pix_y+j][img->pix_x+i];
  }
  for (j=0; j<8; j++)
  for (i=0; i<8; i++)
  {
    rec_mbU[j][i] = imgUV[0][img->pix_c_y+j][img->pix_c_x+i];
    rec_mbV[j][i] = imgUV[1][img->pix_c_y+j][img->pix_c_x+i];
  }


  //--- store results of decoders ---
  if (input->rdopt==2 && !bframe)
  {
    for (k=0; k<input->NoOfDecoders; k++)
    {
      for (j=img->pix_y; j<img->pix_y+16; j++)
      for (i=img->pix_x; i<img->pix_x+16; i++)
      {
        // Keep the decoded values of each MB for updating the ref frames
        decs->decY_best[k][j][i] = decs->decY[k][j][i];
      }
    }
  }

  //--- coeff, cbp, kac ---
  if (mode || bframe)
  {
    i4p=cofAC; cofAC=img->cofAC; img->cofAC=i4p;
    i3p=cofDC; cofDC=img->cofDC; img->cofDC=i3p;
    cbp     = currMB->cbp;
    cbp_blk = currMB->cbp_blk;
  }
  else
  {
    cbp = cbp_blk = 0;
  }

  //--- references ---
  frefframe[0] = frefar[img->block_y  ][img->block_x  ];
  frefframe[1] = frefar[img->block_y  ][img->block_x+2];
  frefframe[2] = frefar[img->block_y+2][img->block_x  ];
  frefframe[3] = frefar[img->block_y+2][img->block_x+2];
  if (bframe)
  {
    brefframe[0] = brefar[img->block_y  ][img->block_x  ]; 
    brefframe[1] = brefar[img->block_y  ][img->block_x+2];
    brefframe[2] = brefar[img->block_y+2][img->block_x  ];
    brefframe[3] = brefar[img->block_y+2][img->block_x+2];
  }
}


/*! 
 *************************************************************************************
 * \brief
 *    Set stored macroblock parameters
 *************************************************************************************
 */
void
set_stored_macroblock_parameters ()
{
  int  i, j, k, ****i4p, ***i3p;
  Macroblock  *currMB  = &img->mb_data[img->current_mb_nr];
  int         mode     = best_mode;
  int         bframe   = (img->type==B_IMG);
  int         **frefar = (img->type==B_IMG ? fw_refFrArr : refFrArr);
  int         **brefar = bw_refFrArr;

  //===== reconstruction values =====
  for (j=0; j<16; j++)
  for (i=0; i<16; i++)
  {
    imgY[img->pix_y+j][img->pix_x+i] = rec_mbY[j][i];
  }
  for (j=0; j<8; j++)
  for (i=0; i<8; i++)
  {
    imgUV[0][img->pix_c_y+j][img->pix_c_x+i] = rec_mbU[j][i];
    imgUV[1][img->pix_c_y+j][img->pix_c_x+i] = rec_mbV[j][i];
  }

  //===== coefficients and cbp =====
  i4p=cofAC; cofAC=img->cofAC; img->cofAC=i4p;
  i3p=cofDC; cofDC=img->cofDC; img->cofDC=i3p;
  currMB->cbp      = cbp;
  currMB->cbp_blk = cbp_blk;

  //==== macroblock type ====
  currMB->mb_type = mode;
  for (i=0; i<4; i++)
  {
    currMB->b8mode[i]   = b8mode[i];
    currMB->b8pdir[i]   = b8pdir[i];
    currMB->useABT[i]   = (ABT_MODE[i]!=ABT_OFF);
    currMB->abt_mode[i] = ABT_MODE[i];
  }
  if (input->rdopt==2 && !bframe)
  {
    //! save the MB Mode of every macroblock
    decs->dec_mb_mode[img->mb_x][img->mb_y] = mode;
  }

  //==== reference frames =====
  for (j=0; j<4; j++)
  for (i=0; i<4; i++)
  {
    frefar[img->block_y+j][img->block_x+i] = frefframe[2*(j/2) + (i/2)];
  }
  if (bframe)
  {
    for (j=0; j<4; j++)
    for (i=0; i<4; i++)
    {
      brefar[img->block_y+j][img->block_x+i] = brefframe[2*(j/2) + (i/2)];
    }
  }

  //==== intra prediction modes ====
  if (mode==P8x8)
  {
    for (k=0, j=img->block_y+1; j<img->block_y+5; j++)
    for (     i=img->block_x+1; i<img->block_x+5; i++, k++)
    {
      img   ->ipredmode    [i][j] = b8_ipredmode       [k];
      currMB->intra_pred_modes[k] = b8_intra_pred_modes[k];
    }
  }
  else if (mode!=I4MB)
  {
    for (k=0, j=img->block_y+1; j<img->block_y+5; j++)
    for (     i=img->block_x+1; i<img->block_x+5; i++, k++)
    {
      img   ->ipredmode    [i][j] = 0;
      currMB->intra_pred_modes[k] = 0;
    }
  }

  //==== motion vectors =====
  SetMotionVectorsMB (currMB, bframe);
}



/*! 
 *************************************************************************************
 * \brief
 *    Set reference frames and motion vectors
 *************************************************************************************
 */
void
SetRefAndMotionVectors (int block, int mode, int ref, int pdir)
{
  int i, j;
  int     bframe  = (img->type==B_IMG);
  int     pmode   = (mode==1||mode==2||mode==3?mode:4);
  int     j0      = ((block/2)<<1);
  int     i0      = ((block%2)<<1);
  int     j1      = j0 + (input->blc_size[pmode][1]>>2);
  int     i1      = i0 + (input->blc_size[pmode][0]>>2);
  int**   frefArr = (bframe ? fw_refFrArr : refFrArr);
  int**   brefArr = bw_refFrArr;
  int***  fmvArr  = (bframe ? tmp_fwMV    : tmp_mv);
  int***  bmvArr  = tmp_bwMV;

  if ((pdir==0 || pdir==2) && (mode!=IBLOCK && mode!=0))
  {
    for (j=j0; j<j1; j++)
    for (i=i0; i<i1; i++)
    {
      fmvArr[0][img->block_y+j][img->block_x+i+4] = img->all_mv[i][j][ref][mode][0];
      fmvArr[1][img->block_y+j][img->block_x+i+4] = img->all_mv[i][j][ref][mode][1];
      frefArr  [img->block_y+j][img->block_x+i  ] = ref;
    }
  }
  else
  {
    if(!mode&&bframe)
    {
      for (j=j0; j<j0+2; j++)
        for (i=i0; i<i0+2; i++)
        {
          ref = refFrArr[img->block_y+j][img->block_x+i];        
          if(ref==-1)
          {
            fmvArr[0][img->block_y+j][img->block_x+i+4] = 0;
            fmvArr[1][img->block_y+j][img->block_x+i+4] = 0;
            frefArr  [img->block_y+j][img->block_x+i  ] = -1;
          }
          else
          {
            fmvArr[0][img->block_y+j][img->block_x+i+4] = img->all_mv[i][j][ref][mode][0];
            fmvArr[1][img->block_y+j][img->block_x+i+4] = img->all_mv[i][j][ref][mode][1];
            frefArr  [img->block_y+j][img->block_x+i  ] = ref;
          }
        }
    }
    else    
      for (j=j0; j<j0+2; j++)
        for (i=i0; i<i0+2; i++)
        {
          fmvArr[0][img->block_y+j][img->block_x+i+4] = 0;
          fmvArr[1][img->block_y+j][img->block_x+i+4] = 0;
          frefArr  [img->block_y+j][img->block_x+i  ] = -1;
        }
  }
  if ((pdir==1 || pdir==2) && (mode!=IBLOCK && mode!=0))
  {
    for (j=j0; j<j0+2; j++)
    for (i=i0; i<i0+2; i++)
    {
      bmvArr[0][img->block_y+j][img->block_x+i+4] = img->all_bmv[i][j][0][mode][0];
      bmvArr[1][img->block_y+j][img->block_x+i+4] = img->all_bmv[i][j][0][mode][1];
      brefArr  [img->block_y+j][img->block_x+i  ] = 0;
    }
  }
  else if (bframe)
  {
    if(!mode)
      for (j=j0; j<j0+2; j++)
      for (i=i0; i<i0+2; i++)
      {
        bmvArr[0][img->block_y+j][img->block_x+i+4] = img->all_bmv[i][j][0][mode][0];
        bmvArr[1][img->block_y+j][img->block_x+i+4] = img->all_bmv[i][j][0][mode][1];
        brefArr  [img->block_y+j][img->block_x+i  ] = min(ref,0); 
      }	  
    else
      for (j=j0; j<j0+2; j++)
      for (i=i0; i<i0+2; i++)
      {
        bmvArr[0][img->block_y+j][img->block_x+i+4] = 0;
        bmvArr[1][img->block_y+j][img->block_x+i+4] = 0;
        brefArr  [img->block_y+j][img->block_x+i  ] = -1;
      }
  }
}




/*! 
 *************************************************************************************
 * \brief
 *    Mode Decision for a macroblock
 *************************************************************************************
 */
void
encode_one_macroblock ()
{
  static const int  b8_mode_table[6]  = {0, 4, 5, 6, 7, IBLOCK};         // DO NOT CHANGE ORDER !!!
  static const int  mb_mode_table[7]  = {0, 1, 2, 3, P8x8, I16MB, I4MB}; // DO NOT CHANGE ORDER !!!
  static const int  abt_mode_table[12] = {ABT_OFF, 0, 0, 0, 0, 1, 2, 3, ABT_OFF, ABT_OFF, ABT_OFF, ABT_OFF};  // DO NOT CHANGE ORDER !!!   // mode->abt_mode

  int         valid[MAXMODE];
  int         rerun, block, index, mode, i0, i1, j0, j1, pdir, ref, i, j, k, ctr16x16, dummy;
  double      qp, lambda_mode, lambda_motion, min_rdcost, rdcost = 0, max_rdcost=1e30;
  double      lambda_inter, lambda_intra;
  int         lambda_motion_factor;
  int         fw_mcost, bw_mcost, bid_mcost, mcost, max_mcost=(1<<30);
  int         curr_cbp_blk, cnt_nonz = 0, best_cnt_nonz = 0, best_fw_ref = 0, best_pdir;
  int         cost, min_cost, cost8x8, cost_direct=0, have_direct=0, i16mode;
  int         intra1[4];

  int         intra       = ((img->type==INTER_IMG && img->mb_y==img->mb_y_upd && img->mb_y_upd!=img->mb_y_intra) || img->type==INTRA_IMG);
  int         spframe     = (img->type==INTER_IMG && img->types==SP_IMG);
  int         siframe     = (img->type==INTRA_IMG && img->types==SP_IMG);
  int         bframe      = (img->type==B_IMG);
  int         write_ref   = (input->no_multpred>1 || input->add_ref_frame>0);
  int         runs        = (input->RestrictRef==1 && input->rdopt==2 && img->type==INTER_IMG ? 2 : 1);
  int         max_ref     = img->nb_references;
  // Tian Dong. PLUS1. Add the following line:
  int         adjust_ref;
  int         checkref    = (input->rdopt && input->RestrictRef && img->type==INTER_IMG);
  Macroblock* currMB      = &img->mb_data[img->current_mb_nr];

  int         useABT = NO_ABT;

  intra |= RandomIntra (img->current_mb_nr);    // Forced Pseudo-Random Intra

  adjust_ref  = (img->type==B_IMG? (mref==mref_fld)? 2 : 1 : 0);
  adjust_ref  = min(adjust_ref, max_ref-1);
  
  if(mref==mref_fld)
  {
    max_ref = min (img->number-((mref==mref_fld)&&img->fld_type&&bframe), img->buf_cycle);
    adjust_ref = 0;
  }

  //===== SET VALID MODES =====
  valid[I4MB]   = 1;
  valid[I16MB]  = 1;
  if(input->abt==INTER_INTRA_ABT)
    valid[I16MB]  = 0;	//no intraNew for ABT
  // HS: I'm not sure when the Intra Mode on 8x8 basis should be unvalid
  //     Is it o.k. for data partitioning? (where the syntax elements have to written to?)

  if(input->partition_mode || spframe || siframe)
    valid[IBLOCK] = 0;
  else
    valid[IBLOCK] = 1;

  valid[0]      = (!intra );
  valid[1]      = (!intra && input->InterSearch16x16);
  valid[2]      = (!intra && input->InterSearch16x8);
  valid[3]      = (!intra && input->InterSearch8x16);
  valid[4]      = (!intra && input->InterSearch8x8);
  valid[5]      = (!intra && input->InterSearch8x4);
  valid[6]      = (!intra && input->InterSearch4x8);
  valid[7]      = (!intra && input->InterSearch4x4);
  valid[P8x8]   = (valid[4] || valid[5] || valid[6] || valid[7] || (bframe && valid[0] && valid[IBLOCK]));
  valid[12]    =  (siframe);

  for (i=0; i<MAXMODE; i++){ for (j=0; j<4; j++)  best_abt_mode[i][j] = ABT_OFF; }

  //===== SET LAGRANGE PARAMETERS =====
  if (input->rdopt)
  {
    qp            = (double)img->qp - SHIFT_QP;
    lambda_mode   = 0.85 * pow (2, qp/3.0) * (bframe? 4.0:spframe?max(1.4,min(3.0,(qp / 12.0))):1.0);  // ????? WHY FACTOR 4 FOR SP-FRAME ?????
    lambda_motion = sqrt (lambda_mode);
    lambda_inter  = lambda_mode;
    lambda_intra  = lambda_mode;
  }
  else
  {
    lambda_mode = lambda_motion = QP2QUANT[max(0,img->qp-SHIFT_QP)];
    lambda_inter = lambda_intra = lambda_mode;
  }
  lambda_motion_factor = LAMBDA_FACTOR (lambda_motion);


  for (rerun=0; rerun<runs; rerun++)
  {
    if (runs==2)
    {
      if (rerun==0)   input->rdopt=1;
      else            input->rdopt=2;
    }

    if (!intra)
    {
      //===== set direct motion vectors =====
      if (bframe)
      {
        Get_Direct_Motion_Vectors ();
      }

      if (valid[P8x8])
      {
        cost8x8 = 0;

        //===== store coding state of macroblock =====
        store_coding_state (cs_mb);

        //=====  LOOP OVER 8x8 SUB-PARTITIONS  (Motion Estimation & Mode Decision) =====
        for (cbp8x8=cbp_blk8x8=cnt_nonz_8x8=0, block=0; block<4; block++)
        {
          //--- set coordinates ---
          j0 = ((block/2)<<3);    j1 = (j0>>2);
          i0 = ((block%2)<<3);    i1 = (i0>>2);

          //=====  LOOP OVER POSSIBLE CODING MODES FOR 8x8 SUB-PARTITION  =====
          for (min_cost=(1<<20), min_rdcost=1e30, index=(bframe?0:1); index<6; index++)
          {
            if (valid[mode=b8_mode_table[index]])
            {
              curr_cbp_blk = 0;

              lambda_mode = lambda_inter;
              if (mode==IBLOCK)
              {
                //===== USE ABT =====
                useABT = currMB->useABT[block] = (input->abt==INTER_INTRA_ABT);
                currMB->abt_mode[block] = useABT ? abt_mode_table[mode] : ABT_OFF;
                if (useABT)
                  lambda_mode = lambda_intra;

                //--- Intra4x4 mode decision for 8x8 block ---
                cnt_nonz    = Mode_Decision_for_8x8IntraBlocks (block, lambda_mode, &cost,useABT,-1);
                best_fw_ref = -1;
                best_pdir   = -1;
              } // if (mode==IBLOCK)
              else if (mode==0)
              {
                //===== USE ABT =====
                if (input->abt)
                {
                  getDirectModeABT(block); // set currMB->abt_mode to ABT mode from collocated MB
                  useABT = currMB->useABT[block] = !NO_ABT;
                }
                else
                {
                  currMB->abt_mode[block] = ABT_OFF;
                  useABT = currMB->useABT[block] = NO_ABT;
                }

               //--- Direct Mode ---
                if (!input->rdopt)
                {
                  cost_direct += (cost = Get_Direct_Cost8x8 ( block, lambda_mode, currMB->abt_mode[block] ));
                  have_direct ++;
                }
                best_fw_ref = -1;
                best_pdir   =  2;
              } // if (mode==0)
              else
              {
                //===== USE ABT =====
                useABT = currMB->useABT[block] = (input->abt!=NO_ABT);
                currMB->abt_mode[block] = useABT ? abt_mode_table[mode] : ABT_OFF;

                //--- motion estimation for all reference frames ---
                PartitionMotionSearch (mode, block, lambda_motion, useABT);

                //--- get cost and reference frame for forward prediction ---
                for (fw_mcost=max_mcost, ref=0; ref<max_ref-adjust_ref; ref++) // Tian Dong. PLUS1. -adjust_ref. June 06, 2002
                {
                  if (!checkref || ref==0 || CheckReliabilityOfRef (block, ref, mode))
                  {
                    mcost  = (input->rdopt ? write_ref ? REF_COST (lambda_motion_factor, ref) : 0 : (int)(2*lambda_motion*min(ref,1)));

                    mcost += motion_cost[mode][ref+1][block];
                    if (mcost < fw_mcost)
                    {
                      fw_mcost    = mcost;
                      best_fw_ref = ref;
                    }
                  }
                }

                if (bframe)
                {
                  //--- get cost for backward prediction ---
                  bw_mcost   = motion_cost[mode][0][block];

                  //--- get cost for bidirectional prediction ---
                  bid_mcost  = (input->rdopt ? write_ref ? REF_COST (lambda_motion_factor, best_fw_ref) : 0 : (int)(2*lambda_motion*min(best_fw_ref,1)));
                  bid_mcost += BIDPartitionCost (mode, block, best_fw_ref, lambda_motion_factor, useABT);

                 //--- get prediction direction ----
                  if      (fw_mcost<=bw_mcost && fw_mcost<=bid_mcost)    {best_pdir = 0;  cost = fw_mcost; }
                  else if (bw_mcost<=fw_mcost && bw_mcost<=bid_mcost)    {best_pdir = 1;  cost = bw_mcost; }
                  else                                                   {best_pdir = 2;  cost = bid_mcost;}
                } // if (bframe)
                else
                {
                  best_pdir = 0;
                  cost      = fw_mcost;
                }
              } // if (mode!=IBLOCK && mode!=0)

              //--- store coding state before coding with current mode ---
              store_coding_state (cs_cm);

              if (input->rdopt)
              {
                //--- get and check rate-distortion cost ---
                rdcost = RDCost_for_8x8blocks (&cnt_nonz, &curr_cbp_blk, lambda_mode,
                                               block, mode, best_pdir, best_fw_ref);
              }
              else
              {
                cost += (REF_COST (lambda_motion_factor, B8Mode2Value (mode, best_pdir)) - 1);
              }

              //--- set variables if best mode has changed ---
              if (( input->rdopt && rdcost < min_rdcost) ||
                  (!input->rdopt && cost   < min_cost  )   )
              {
                min_cost                   = cost;
                min_rdcost                 = rdcost;
                best8x8mode        [block] = mode;
                best8x8pdir  [P8x8][block] = best_pdir;
                best8x8ref   [P8x8][block] = best_fw_ref;
                best_abt_mode[P8x8][block] = useABT ? currMB->abt_mode[block] : ABT_OFF;

                //--- store number of nonzero coefficients ---
                best_cnt_nonz  = cnt_nonz;

                if (input->rdopt)
                {
                  //--- store block cbp ---
                  cbp_blk8x8    &= (~(0x33 << (((block>>1)<<3)+((block%2)<<1)))); // delete bits for block
                  cbp_blk8x8    |= curr_cbp_blk;

                  //--- store coefficients ---
                  for (k=0; k< 4; k++)
                  for (j=0; j< 2; j++)
                  for (i=0; i<65; i++)  cofAC8x8[block][k][j][i] = img->cofAC[block][k][j][i]; // 18->65 for ABT

                  //--- store reconstruction and prediction ---
                  for (j=j0; j<j0+8; j++)
                  for (i=i0; i<i0+8; i++)
                  {
                    rec_mbY8x8[j][i] = imgY[img->pix_y+j][img->pix_x+i];
                    mpr8x8    [j][i] = img->mpr[i][j];
                  }
                }

                //--- store coding state ---
                store_coding_state (cs_b8);
              } // if (rdcost <= min_rdcost)

              //--- re-set coding state as it was before coding with current mode was performed ---
              reset_coding_state (cs_cm);
            } // if (valid[mode=b8_mode_table[index]])
          } // for (min_rdcost=1e30, index=(bframe?0:1); index<6; index++)

          cost8x8 += min_cost;

          if (!input->rdopt)
          {
            mode = best8x8mode[block];
            pdir = best8x8pdir[P8x8][block];

            if (mode!=IBLOCK)
            {
              curr_cbp_blk  = 0;
              best_cnt_nonz = LumaResidualCoding8x8 (&dummy, &curr_cbp_blk, block,
                                                     (pdir==0||pdir==2?mode:0),
                                                     (pdir==1||pdir==2?mode:0),
                                                     (mode!=0?best8x8ref[P8x8][block]:max(0,refFrArr[img->block_y+j1][img->block_x+i1])),
                                                     best_abt_mode[P8x8][block]);
              cbp_blk8x8   &= (~(0x33 << (((block>>1)<<3)+((block%2)<<1)))); // delete bits for block
              cbp_blk8x8   |= curr_cbp_blk;
            }

            //--- store coefficients ---
            for (k=0; k< 4; k++)
            for (j=0; j< 2; j++)
            for (i=0; i<65; i++)  cofAC8x8[block][k][j][i] = img->cofAC[block][k][j][i]; // 18->65 for ABT

            //--- store reconstruction and prediction ---
            for (j=j0; j<j0+8; j++)
            for (i=i0; i<i0+8; i++)
            {
              rec_mbY8x8[j][i] = imgY[img->pix_y+j][img->pix_x+i];
              mpr8x8    [j][i] = img->mpr[i][j];
            }
          }

          //----- set cbp and count of nonzero coefficients ---
          if (best_cnt_nonz)
          {
            cbp8x8        |= (1<<block);
            cnt_nonz_8x8  += (best8x8mode[block]==IBLOCK ? MAX_VALUE : best_cnt_nonz);
          }
          else if (best8x8mode[block]==IBLOCK && cnt_nonz_8x8)
          {
            // coefficients cannot be discarded if a following INTRA4x4 blocks needs they for prediction
            cnt_nonz_8x8  += MAX_VALUE;
          }

          if ((mode=best8x8mode[block])!=IBLOCK)
          {
            //===== reset intra prediction modes (needed for prediction, must be stored after 8x8 mode dec.) =====
            j0 = img->block_y+1+2*(block/2);
            i0 = img->block_x+1+2*(block%2);
            for (j=j0; j<j0+2; j++)
            for (i=i0; i<i0+2; i++)    img->ipredmode[i][j]         = 0;
            i0 = 4*block;
            for (i=i0; i<i0+4; i++)    currMB->intra_pred_modes[i]  = 0;
          }

          if (block<3)
          {
            //===== set motion vectors and reference frames (prediction) =====
            SetRefAndMotionVectors (block, mode, best8x8ref[P8x8][block], best8x8pdir[P8x8][block]);

            //===== re-set reconstructed block =====
            j0   = 8*(block/2);
            i0   = 8*(block%2);
            for (j=j0; j<j0+8; j++)
            for (i=i0; i<i0+8; i++)  imgY[img->pix_y+j][img->pix_x+i] = rec_mbY8x8[j][i];
          } // if (block<3)

          //===== set the coding state after current block =====
          reset_coding_state (cs_b8);
        } // for (cbp8x8=cbp_blk8x8=cnt_nonz_8x8=0, block=0; block<4; block++)

        //===== store intra prediction modes for 8x8+ macroblock mode =====
        for (k=0, j=img->block_y+1; j<img->block_y+5; j++)
        for (     i=img->block_x+1; i<img->block_x+5; i++, k++)
        {
          b8_ipredmode       [k] = img   ->ipredmode    [i][j];
          b8_intra_pred_modes[k] = currMB->intra_pred_modes[k];
        }

        //--- re-set coding state (as it was before 8x8 block coding) ---
        reset_coding_state (cs_mb);
      }
      else // if (valid[P8x8])
      {
        cost8x8 = (1<<20);
      }



      //===== MOTION ESTIMATION FOR 16x16, 16x8, 8x16 BLOCKS =====
      for (min_cost=cost8x8, best_mode=P8x8, mode=3; mode>0; mode--)
      {
        if (valid[mode])
        {
          //===== USE ABT =====
          useABT = (input->abt!=NO_ABT);
          for (i=0; i<4; i++)
          {
            currMB->useABT[i]   = useABT;
            currMB->abt_mode[i] = best_abt_mode[mode][i] = useABT ? B8x8 : ABT_OFF;
          }

          for (cost=0, block=0; block<(mode==1?1:2); block++)
          {
            PartitionMotionSearch (mode, block, lambda_motion, useABT);

            //--- set 4x4 block indizes (for getting MV) ---
            j = (block==1 && mode==2 ? 2 : 0);
            i = (block==1 && mode==3 ? 2 : 0);

            //--- get cost and reference frame for forward prediction ---
            for (fw_mcost=max_mcost, ref=0; ref<max_ref-adjust_ref; ref++)   // Tian Dong. PLUS1. -adjust_ref. June 06, 2002
            {
              if (!checkref || ref==0 || CheckReliabilityOfRef (block, ref, mode))
              {
                mcost  = (input->rdopt ? write_ref ? REF_COST (lambda_motion_factor, ref) : 0 : (int)(2*lambda_motion*min(ref,1)));

                mcost += motion_cost[mode][ref+1][block];
                if (mcost < fw_mcost)
                {
                  fw_mcost    = mcost;
                  best_fw_ref = ref;
                }
              }
            }

            if (bframe)
            {
              //--- get cost for backward prediction ---
              bw_mcost   = motion_cost[mode][0][block];

              //--- get cost for bidirectional prediction ---
              bid_mcost  = (input->rdopt ? write_ref ? REF_COST (lambda_motion_factor, best_fw_ref) : 0 : (int)(2*lambda_motion*min(best_fw_ref,1)));
              bid_mcost += BIDPartitionCost (mode, block, best_fw_ref, lambda_motion_factor, useABT);

              //--- get prediction direction ----
              if      (fw_mcost<=bw_mcost && fw_mcost<=bid_mcost)    {best_pdir = 0;  cost += fw_mcost; }
              else if (bw_mcost<=fw_mcost && bw_mcost<=bid_mcost)    {best_pdir = 1;  cost += bw_mcost; }
              else                                                   {best_pdir = 2;  cost += bid_mcost;}
            }
            else // if (bframe)
            {
              best_pdir  = 0;
              cost      += fw_mcost;
            }


            //----- set reference frame and direction parameters -----
            if (mode==3)
            {
              best8x8ref [3][  block] = best8x8ref [3][  block+2] = best_fw_ref;
              best8x8pdir[3][  block] = best8x8pdir[3][  block+2] = best_pdir;
            }
            else if (mode==2)
            {
              best8x8ref [2][2*block] = best8x8ref [2][2*block+1] = best_fw_ref;
              best8x8pdir[2][2*block] = best8x8pdir[2][2*block+1] = best_pdir;
            }
            else
            {
              best8x8ref [1][0] = best8x8ref [1][1] = best8x8ref [1][2] = best8x8ref [1][3] = best_fw_ref;
              best8x8pdir[1][0] = best8x8pdir[1][1] = best8x8pdir[1][2] = best8x8pdir[1][3] = best_pdir;
            }

            //--- set reference frames and motion vectors ---
            if (mode>1 && block==0)   SetRefAndMotionVectors (block, mode, best_fw_ref, best_pdir);

          } // for (block=0; block<(mode==1?1:2); block++)

          if (cost < min_cost)
          {
            best_mode = mode;
            min_cost  = cost;
          }
        } // if (valid[mode])
      } // for (mode=3; mode>0; mode--)

      // Find a motion vector for the Skip mode
      if(img->type == INTER_IMG)
        FindSkipModeMotionVector ();
    }
    else // if (img->type!=INTRA_IMG)
    {
      min_cost = (1<<20);
    }

    if (input->rdopt)
    {
      //===== GET BEST MACROBLOCK MODE =====
      for (ctr16x16=0, min_rdcost=max_rdcost, index=0; index<7; index++)
      {
        mode = mb_mode_table[index];

        if ( (input->abt==INTER_INTRA_ABT) && ((mode==I4MB)||(mode==I16MB)) && (!intra) )
          lambda_mode = lambda_intra;
        else
          lambda_mode = lambda_inter;

        //--- for INTER16x16 check all prediction directions ---
        if (mode==1 && bframe)
        {
          best8x8pdir[1][0] = best8x8pdir[1][1] = best8x8pdir[1][2] = best8x8pdir[1][3] = ctr16x16;
          if (ctr16x16 < 2) index--;
          ctr16x16++;
        }

        if (valid[mode])
        {
          if (RDCost_for_macroblocks (lambda_mode, mode, &min_rdcost))
          {
            store_macroblock_parameters (mode);
          }
        }
      }
    }
    else
    {
      if (valid[0] && bframe) // check DIRECT MODE
      {
        cost  = (have_direct?cost_direct:Get_Direct_CostMB (lambda_mode));
        cost -= (int)floor(16*lambda_motion+0.4999);
        if (cost <= min_cost)
        {
          min_cost  = cost;
          best_mode = 0;
        }
      }
      if (valid[I4MB]) // check INTRA4x4
      {
        //===== USE ABT =====
        useABT = (input->abt==INTER_INTRA_ABT);
        for (i=0; i<4; i++)
        {
          currMB->useABT[i]   = useABT;
          currMB->abt_mode[i] = ABT_OFF; //will be overwritten in case of ABT intrablocks.
        }

        currMB->cbp = Mode_Decision_for_Intra4x4Macroblock (lambda_mode, &cost);
        if (cost <= min_cost)
        {
          min_cost  = cost;
          best_mode = I4MB;
          for (i=0; i<4; i++)
            best_abt_mode[I4MB][i] = useABT ? currMB->abt_mode[i] : ABT_OFF;
        }
      }
      if (valid[I16MB]) // check INTRA16x16
      {
        //===== USE ABT =====
        useABT = NO_ABT;
        for (i=0; i<4; i++)
        {
          currMB->useABT[i]   = useABT;
          currMB->abt_mode[i] = ABT_OFF; // no JM intra16x16 with ABT
        }

        intrapred_luma_2 ();
        cost = find_sad2 (&i16mode);
        if (cost < min_cost)
        {
          best_mode   = I16MB;
          currMB->cbp = dct_luma2 (i16mode);
          for (i=0; i<4; i++) best_abt_mode[I16MB][i] = ABT_OFF;
        }
      }
    }

    if (rerun==0)
    {
      intra1[0] = (currMB->mb_type==I16MB || currMB->b8mode[0]==IBLOCK ? 1 : 0);
      intra1[1] = (currMB->mb_type==I16MB || currMB->b8mode[1]==IBLOCK ? 1 : 0);
      intra1[2] = (currMB->mb_type==I16MB || currMB->b8mode[2]==IBLOCK ? 1 : 0);
      intra1[3] = (currMB->mb_type==I16MB || currMB->b8mode[3]==IBLOCK ? 1 : 0);
    }
  } // for (rerun=0; rerun<runs; rerun++)

  if (input->rdopt)
  {
    set_stored_macroblock_parameters ();
  }
  else
  {
    //===== set parameters for chosen mode =====
    SetModesAndRefframeForBlocks (best_mode);
    if (best_mode==P8x8)
    {
      SetCoeffAndReconstruction8x8 (currMB);
    }
    else
    {
      if (best_mode!=I4MB)
      {
        for (k=0, j=img->block_y+1; j<img->block_y+5; j++)
        for (     i=img->block_x+1; i<img->block_x+5; i++, k++)
        {
          img   ->ipredmode    [i][j] = 0;
          currMB->intra_pred_modes[k] = 0;
        }
        if (best_mode!=I16MB)
        {
          LumaResidualCoding ();
        }
      }
    }
    ChromaResidualCoding (&dummy);
    if (best_mode==I16MB)
    {
      img->i16offset = I16Offset  (currMB->cbp, i16mode);
    }
    SetMotionVectorsMB (currMB, bframe);

    //===== check for SKIP mode =====
    if (img->type==INTER_IMG && best_mode==1 && currMB->cbp==0 &&
        refFrArr [img->block_y][img->block_x  ]==0 &&
        tmp_mv[0][img->block_y][img->block_x+4]==img->all_mv [0][0][0][0][0] &&
        tmp_mv[1][img->block_y][img->block_x+4]==img->all_mv [0][0][0][0][1]               )
    {
      currMB->mb_type=currMB->b8mode[0]=currMB->b8mode[1]=currMB->b8mode[2]=currMB->b8mode[3]=0;
    }
  }
/*
  if (input->rdopt==2 && !bframe)
  {
    for (j=0 ;j<input->NoOfDecoders; j++)  DeblockMb(img, decs->decY_best[j], NULL);
  }
*/

  //===== store abt_mode for direct mode blocks =====
  if (!bframe)
  {
    for (i=0; i<4; i++)
      setDirectModeABT(i);
  }

  //===== init and update number of intra macroblocks =====
  if (img->current_mb_nr==0)                      intras=0;
  if (img->type==INTER_IMG && IS_INTRA(currMB))   intras++;

  //===== Decide if this MB will restrict the reference frames =====
  if (input->RestrictRef==1)
  {
    if (input->rdopt==1)
    {
      refresh_map[2*img->mb_y  ][2*img->mb_x  ] = (intra ? 1 : 0);
      refresh_map[2*img->mb_y  ][2*img->mb_x+1] = (intra ? 1 : 0);
      refresh_map[2*img->mb_y+1][2*img->mb_x  ] = (intra ? 1 : 0);
      refresh_map[2*img->mb_y+1][2*img->mb_x+1] = (intra ? 1 : 0);
    }
    else if (input->rdopt==2)
    {
      refresh_map[2*img->mb_y  ][2*img->mb_x  ] = (intra1[0]==0 && (currMB->mb_type==I16MB || currMB->b8mode[0]==IBLOCK) ? 1 : 0);
      refresh_map[2*img->mb_y  ][2*img->mb_x+1] = (intra1[1]==0 && (currMB->mb_type==I16MB || currMB->b8mode[1]==IBLOCK) ? 1 : 0);
      refresh_map[2*img->mb_y+1][2*img->mb_x  ] = (intra1[2]==0 && (currMB->mb_type==I16MB || currMB->b8mode[2]==IBLOCK) ? 1 : 0);
      refresh_map[2*img->mb_y+1][2*img->mb_x+1] = (intra1[3]==0 && (currMB->mb_type==I16MB || currMB->b8mode[3]==IBLOCK) ? 1 : 0);
    }
  }
  else if (input->RestrictRef==2)
  {
    refresh_map[2*img->mb_y  ][2*img->mb_x  ] = (currMB->mb_type==I16MB || currMB->b8mode[0]==IBLOCK ? 1 : 0);
    refresh_map[2*img->mb_y  ][2*img->mb_x+1] = (currMB->mb_type==I16MB || currMB->b8mode[1]==IBLOCK ? 1 : 0);
    refresh_map[2*img->mb_y+1][2*img->mb_x  ] = (currMB->mb_type==I16MB || currMB->b8mode[2]==IBLOCK ? 1 : 0);
    refresh_map[2*img->mb_y+1][2*img->mb_x+1] = (currMB->mb_type==I16MB || currMB->b8mode[3]==IBLOCK ? 1 : 0);
  }
}


