int decode_cabac_mb_chroma_pre_mode( CABACContext *c, unsigned char *state, int AvailMask, CABAC_NEIGH *neigh) {
int intra_chroma_pred_mode_top = neigh -> intra_chroma_pred_mode_top;
int intra_chroma_pred_mode_left = neigh -> intra_chroma_pred_mode_left;
int ctx = 0;
/* No need to test for IS_INTRA4x4 and IS_INTRA16x16, as we set chroma_pred_mode_table to 0 */
if( (AvailMask & 0x01) && intra_chroma_pred_mode_left!= 0){
ctx++;
}
if(((AvailMask >> 1) & 0x01) && intra_chroma_pred_mode_top != 0 ){
ctx++;
}
if( get_cabac( c, &state[64 + ctx] ) == 0 ){
return 0;
}
if( get_cabac( c, &state[64 + 3] ) == 0 ){
return 1;
}
if( get_cabac( c, &state[64 + 3] ) == 0 ){
return 2;
}else {
return 3;
}
}
int decode_cabac_mb_skip( CABACContext *c, unsigned char *state, int AvailMask, unsigned char slice_table[],
RESIDU *CurrResidu, int slice_type, int mb_xy, int mb_stride)
{
const int mba_xy = mb_xy - 1;
const int mbb_xy = mb_xy - mb_stride;
int ctx = 0;
if( (AvailMask & 0x01) && slice_table[mba_xy] == slice_table[mb_xy]
&& CurrResidu [mba_xy] . MbType != P_Skip && CurrResidu [mba_xy] . MbType != B_Skip ){
ctx++;
}
if( ((AvailMask >> 1) & 0x01) && slice_table[mbb_xy] == slice_table[mb_xy]
&& CurrResidu [mbb_xy] . MbType != P_Skip && CurrResidu [mbb_xy] . MbType != B_Skip){
ctx++;
}
if (slice_type == SLICE_TYPE_P){
return get_cabac( c, &state[11 + ctx] );
} else {
return get_cabac( c, &state[24 + ctx] );
}
}
int main(){
CABACContext c;
uint8_t b[9*SIZE];
uint8_t r[9*SIZE];
int i;
uint8_t state[10]= {0};
ff_init_cabac_encoder(&c, b, SIZE);
ff_init_cabac_states(&c, ff_h264_lps_range, ff_h264_mps_state, ff_h264_lps_state, 64);
for(i=0; i<SIZE; i++){
r[i]= random()%7;
}
for(i=0; i<SIZE; i++){
START_TIMER
put_cabac_bypass(&c, r[i]&1);
STOP_TIMER("put_cabac_bypass")
}
for(i=0; i<SIZE; i++){
START_TIMER
put_cabac(&c, state, r[i]&1);
STOP_TIMER("put_cabac")
}
for(i=0; i<SIZE; i++){
START_TIMER
put_cabac_u(&c, state, r[i], 6, 3, i&1);
STOP_TIMER("put_cabac_u")
}
for(i=0; i<SIZE; i++){
START_TIMER
put_cabac_ueg(&c, state, r[i], 3, 0, 1, 2);
STOP_TIMER("put_cabac_ueg")
}
put_cabac_terminate(&c, 1);
ff_init_cabac_decoder(&c, b, SIZE);
memset(state, 0, sizeof(state));
for(i=0; i<SIZE; i++){
START_TIMER
if( (r[i]&1) != get_cabac_bypass(&c) )
printf("CABAC bypass failure at %d\n", i);
STOP_TIMER("get_cabac_bypass")
}
for(i=0; i<SIZE; i++){
START_TIMER
if( (r[i]&1) != get_cabac(&c, state) )
printf("CABAC failure at %d\n", i);
STOP_TIMER("get_cabac")
}
for(i=0; i<SIZE; i++){
START_TIMER
if( r[i] != get_cabac_u(&c, state, (i&1) ? 6 : 7, 3, i&1) )
printf("CABAC unary (truncated) binarization failure at %d\n", i);
STOP_TIMER("get_cabac_u")
}
for(i=0; i<SIZE; i++){
START_TIMER
if( r[i] != get_cabac_ueg(&c, state, 3, 0, 1, 2))
printf("CABAC unary (truncated) binarization failure at %d\n", i);
STOP_TIMER("get_cabac_ueg")
}
if(!get_cabac_terminate(&c))
printf("where's the Terminator?\n");
return 0;
}
int main(void) {
CABACContext c;
uint8_t b[9*SIZE];
uint8_t r[9*SIZE];
int i;
uint8_t state[10]= {0};
AVLFG prng;
av_lfg_init(&prng, 1);
ff_init_cabac_encoder(&c, b, SIZE);
ff_init_cabac_states(&c);
for(i=0; i<SIZE; i++) {
r[i] = av_lfg_get(&prng) % 7;
}
for(i=0; i<SIZE; i++) {
START_TIMER
put_cabac_bypass(&c, r[i]&1);
STOP_TIMER("put_cabac_bypass")
}
for(i=0; i<SIZE; i++) {
START_TIMER
put_cabac(&c, state, r[i]&1);
STOP_TIMER("put_cabac")
}
for(i=0; i<SIZE; i++) {
START_TIMER
put_cabac_u(&c, state, r[i], 6, 3, i&1);
STOP_TIMER("put_cabac_u")
}
for(i=0; i<SIZE; i++) {
START_TIMER
put_cabac_ueg(&c, state, r[i], 3, 0, 1, 2);
STOP_TIMER("put_cabac_ueg")
}
put_cabac_terminate(&c, 1);
ff_init_cabac_decoder(&c, b, SIZE);
memset(state, 0, sizeof(state));
for(i=0; i<SIZE; i++) {
START_TIMER
if( (r[i]&1) != get_cabac_bypass(&c) )
av_log(NULL, AV_LOG_ERROR, "CABAC bypass failure at %d\n", i);
STOP_TIMER("get_cabac_bypass")
}
for(i=0; i<SIZE; i++) {
START_TIMER
if( (r[i]&1) != get_cabac(&c, state) )
av_log(NULL, AV_LOG_ERROR, "CABAC failure at %d\n", i);
STOP_TIMER("get_cabac")
}
#if 0
for(i=0; i<SIZE; i++) {
START_TIMER
if( r[i] != get_cabac_u(&c, state, (i&1) ? 6 : 7, 3, i&1) )
av_log(NULL, AV_LOG_ERROR, "CABAC unary (truncated) binarization failure at %d\n", i);
STOP_TIMER("get_cabac_u")
}
for(i=0; i<SIZE; i++) {
START_TIMER
if( r[i] != get_cabac_ueg(&c, state, 3, 0, 1, 2))
av_log(NULL, AV_LOG_ERROR, "CABAC unary (truncated) binarization failure at %d\n", i);
STOP_TIMER("get_cabac_ueg")
}
#endif
if(!get_cabac_terminate(&c))
av_log(NULL, AV_LOG_ERROR, "where's the Terminator?\n");
return 0;
}
