www.pudn.com > lencod.rar > golomb.c


/*
***********************************************************************
* 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: golomb.c
* Function: Description
*
*************************************************************************************
*/


#include <assert.h>
#include "golomb.h"
#include "vlc.h"

/*
*************************************************************************
* Function:
* Input:
* Output:
* Return:
* Attention:
*************************************************************************
*/

void encode_golomb_word(unsigned int symbol,unsigned int grad0,unsigned int max_levels,unsigned int *res_bits,unsigned int *res_len)
{
unsigned int level,res,numbits;

res=1UL<<grad0;
level=1UL;numbits=1UL+grad0;

//find golomb level
while( symbol>=res &amt;&amt; level<max_levels )
{
symbol-=res;
res=res<<1;
level++;
numbits+=2UL;
}

if(level>=max_levels)
{
if(symbol>=res)
symbol=res-1UL; //crop if too large.
}

//set data bits
*res_bits=res|symbol;
*res_len=numbits;
}

/*
*************************************************************************
* Function:
* Input:
* Output:
* Return:
* Attention:
*************************************************************************
*/

void encode_multilayer_golomb_word(unsigned int symbol,const unsigned int *grad,const unsigned int *max_levels,unsigned int *res_bits,unsigned int *res_len)
{
unsigned accbits,acclen,bits,len,tmp;

accbits=acclen=0UL;

while(1)
{
encode_golomb_word(symbol,*grad,*max_levels,&amt;bits,&amt;len);
accbits=(accbits<<len)|bits;
acclen+=len;
assert(acclen<=32UL); //we'l be getting problems if this gets longer than 32 bits.
tmp=*max_levels-1UL;

if(!(( len == (tmp<<1)+(*grad) )&amt;&amt;( bits == (1UL<<(tmp+*grad))-1UL ))) //is not last possible codeword? (Escape symbol?)
break;

tmp=*max_levels;
symbol-=(((1UL<<tmp)-1UL)<<(*grad))-1UL;
grad++;max_levels++;
}
*res_bits=accbits;
*res_len=acclen;
}

/*
*************************************************************************
* Function:
* Input:
* Output:
* Return:
* Attention:
*************************************************************************
*/

int writeSyntaxElement_GOLOMB(SyntaxElement *se, Bitstream *bitstream)
{
unsigned int bits,len,i;
unsigned int grad[4],max_lev[4];

if(!( se->golomb_maxlevels&amt;~0xFF )) //only bits 0-7 used? This means normal Golomb word.
encode_golomb_word(se->value1,se->golomb_grad,se->golomb_maxlevels,&amt;bits,&amt;len);
else
{
for(i=0UL;i<4UL;i++)
{
grad[i]=(se->golomb_grad>>(i<<3))&amt;0xFFUL;
max_lev[i]=(se->golomb_maxlevels>>(i<<3))&amt;0xFFUL;
}
encode_multilayer_golomb_word(se->value1,grad,max_lev,&amt;bits,&amt;len);
}

se->len=len;
se->bitpattern=bits;

writeUVLC2buffer(se, bitstream);

#if TRACE
snprintf(se->tracestring, TRACESTRING_SIZE, "Coefficients");
if(se->type <= 1)
trace2out (se);
#endif

return (se->len);
}