rdopt_coding_state.c

AVS视频编解码器 能实现视频图像的高效率压缩 能在ＶＣ上高速运行

/*
***********************************************************************
* COPYRIGHT AND WARRANTY INFORMATION
*
* Copyright 2003, Advanced Audio Video Coding Standard, Part II
*
* DISCLAIMER OF WARRANTY
*
* These software programs are available to the users without any
* license fee or royalty on an "as is" basis. The AVS 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 contributors or 
* the AVS be liable for any incidental, punitive, or consequential
* damages of any kind whatsoever arising from the use of this program.
*
* This disclaimer of warranty extends to the user of this program
* and user's customers, employees, agents, transferees, successors,
* and assigns.
*
* The AVS does not represent or warrant that the program furnished
* hereunder are free of infringement of any third-party patents.
* Commercial implementations of AVS, including shareware, may be
* subject to royalty fees to patent holders. Information regarding
* the AVS patent policy is available from the AVS Web site at
* http://www.avs.org.cn
*
* THIS IS NOT A GRANT OF PATENT RIGHTS - SEE THE AVS PATENT POLICY.
************************************************************************
*/

/*
*************************************************************************************
* File name: rdopt_coding_state.c
* Function:  
    Storing/restoring coding state for
    Rate-Distortion optimized mode decision
*
*************************************************************************************
*/


#include <stdlib.h>
#include <math.h>
#include <memory.h>

#include "rdopt_coding_state.h"

/*
*************************************************************************
* Function:delete structure for storing coding state
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

void delete_coding_state (CSptr cs)
{
  if (cs != NULL)
  {
    if (cs->bitstream != NULL)
      free (cs->bitstream);

    //=== coding state structure ===
    free (cs);
    cs=NULL;
  }

}

/*
*************************************************************************
* Function:create structure for storing coding state
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

CSptr create_coding_state ()
{
  CSptr cs;

  //=== coding state structure ===
  if ((cs = (CSptr) calloc (1, sizeof(CSobj))) == NULL)
    no_mem_exit("init_coding_state: cs");

  //=== important variables of data partition array ===

  if ((cs->bitstream = (Bitstream*) calloc (1, sizeof(Bitstream))) == NULL)
    no_mem_exit("init_coding_state: cs->bitstream");

  return cs;
}

/*
*************************************************************************
* Function:store coding state (for rd-optimized mode decision)
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

void store_coding_state (CSptr cs)
{
  Bitstream            *bs_src, *bs_dest;
  Macroblock           *currMB  = &(img->mb_data [img->current_mb_nr]);

  if (!input->rdopt)
    return;

  //=== important variables of data partition array ===
	bs_src  =  currBitStream;
	bs_dest = cs->bitstream;
	
	memcpy (bs_dest, bs_src, sizeof(Bitstream));

  //=== syntax element number and bitcounters ===
  cs->currSEnr = currMB->currSEnr;
  memcpy (cs->bitcounter, currMB->bitcounter, MAX_BITCOUNTER_MB*sizeof(int));

  //=== elements of current macroblock ===
  memcpy (cs->mvd, currMB->mvd, 2*2*BLOCK_MULTIPLE*BLOCK_MULTIPLE*sizeof(int));
  cs->cbp_bits = currMB->cbp_bits;

}

/*
*************************************************************************
* Function:restore coding state (for rd-optimized mode decision)
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

void reset_coding_state (CSptr cs)
{
  Bitstream            *bs_src, *bs_dest;
  Macroblock           *currMB  = &(img->mb_data [img->current_mb_nr]);

  if (!input->rdopt)
    return;
	
	bs_dest =   currBitStream;
	bs_src  =   cs->bitstream;
	
	memcpy (bs_dest, bs_src, sizeof(Bitstream));

  //=== syntax element number and bitcounters ===
  currMB->currSEnr = cs->currSEnr;
  memcpy (currMB->bitcounter, cs->bitcounter, MAX_BITCOUNTER_MB*sizeof(int));

  //=== elements of current macroblock ===
  memcpy (currMB->mvd, cs->mvd, 2*2*BLOCK_MULTIPLE*BLOCK_MULTIPLE*sizeof(int));
  currMB->cbp_bits = cs->cbp_bits;

}