ref: a6631746659a7d54bc342b1d7e56bbce1b6305a1
dir: /libentcode/ectest.c/
#include <stdio.h> #include <math.h> #include "probenc.h" #include "probdec.h" #include "bitrenc.h" int main(int _argc,char **_argv){ ec_byte_buffer buf; ec_enc enc; ec_dec dec; ec_probmod mod; ec_uint64 sym64; long nbits; long nbits2; double entropy; int ft; int ftb; int sym; int sz; int s; int i; entropy=0; /*Testing encoding of raw bit values.*/ ec_byte_writeinit(&buf); ec_enc_init(&enc,&buf); for(ft=0;ft<1024;ft++){ for(i=0;i<ft;i++){ entropy+=log(ft)*M_LOG2E; ec_enc_uint(&enc,i,ft); entropy+=log(ft)*M_LOG2E+30; ec_enc_uint64(&enc,(ec_uint64)i<<30|i,(ec_uint64)ft<<30); } } /*Testing encoding of raw bit values.*/ for(ftb=0;ftb<16;ftb++){ for(i=0;i<(1<<ftb);i++){ entropy+=ftb; nbits=ec_enc_tell(&enc,0); ec_enc_bits(&enc,i,ftb); nbits2=ec_enc_tell(&enc,0); if(nbits2-nbits!=ftb){ fprintf(stderr,"Used %li bits to encode %i bits directly.\n", nbits2-nbits,ftb); } entropy+=ftb+30; nbits=nbits2; ec_enc_bits64(&enc,(ec_uint64)i<<30|i,ftb+30); nbits2=ec_enc_tell(&enc,0); if(nbits2-nbits!=ftb+30){ fprintf(stderr,"Used %li bits to encode %i bits directly.\n", nbits2-nbits,ftb+30); } } } for(sz=1;sz<256;sz++){ ec_probmod_init_full(&mod,sz,1,sz+(sz>>1),NULL); for(i=0;i<sz;i++){ s=((unsigned)(i*45678901+7))%sz; entropy+=(log(mod.ft)-log(ec_bitree_get_freq(mod.bitree,s)))*M_LOG2E; ec_probmod_write(&mod,&enc,s); } ec_probmod_clear(&mod); } for(sz=11;sz<256;sz++){ ec_probmod_init_full(&mod,sz,1,sz+(sz>>1),NULL); for(i=0;i<sz;i++){ s=((unsigned)(i*45678901+7))%sz; entropy+=(log(ec_bitree_get_cumul(mod.bitree,EC_MINI(s+6,sz))- ec_bitree_get_cumul(mod.bitree,EC_MAXI(s-5,0)))- log(ec_bitree_get_freq(mod.bitree,s)))*M_LOG2E; ec_probmod_write_range(&mod,&enc,s,EC_MAXI(s-5,0),EC_MINI(s+6,sz)); } ec_probmod_clear(&mod); } nbits=ec_enc_tell(&enc,4); ec_enc_done(&enc); fprintf(stderr, "Encoded %0.2lf bits of entropy to %0.2lf bits (%0.3lf%% wasted).\n", entropy,ldexp(nbits,-4),100*(nbits-ldexp(entropy,4))/nbits); fprintf(stderr,"Packed to %li bytes.\n",(long)(buf.ptr-buf.buf)); ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf)); ec_dec_init(&dec,&buf); for(ft=0;ft<1024;ft++){ for(i=0;i<ft;i++){ sym=ec_dec_uint(&dec,ft); if(sym!=i){ fprintf(stderr,"Decoded %i instead of %i with ft of %i.\n",sym,i,ft); return -1; } sym64=ec_dec_uint64(&dec,(ec_uint64)ft<<30); if(sym64!=((ec_uint64)i<<30|i)){ fprintf(stderr,"Decoded %lli instead of %lli with ft of %lli.\n",sym64, (ec_uint64)i<<30|i,(ec_uint64)ft<<30); } } } for(ftb=0;ftb<16;ftb++){ for(i=0;i<(1<<ftb);i++){ sym=ec_dec_bits(&dec,ftb); if(sym!=i){ fprintf(stderr,"Decoded %i instead of %i with ftb of %i.\n",sym,i,ftb); return -1; } sym64=ec_dec_bits64(&dec,ftb+30); if(sym64!=((ec_uint64)i<<30|i)){ fprintf(stderr,"Decoded %lli instead of %lli with ftb of %i.\n", sym64,(ec_uint64)i<<30|i,ftb+30); } } } for(sz=1;sz<256;sz++){ ec_probmod_init_full(&mod,sz,1,sz+(sz>>1),NULL); for(i=0;i<sz;i++){ s=((unsigned)(i*45678901+7))%sz; sym=ec_probmod_read(&mod,&dec); if(sym!=s){ fprintf(stderr,"Decoded %i instead of %i with sz of %i.\n",sym,s,sz); return -1; } } ec_probmod_clear(&mod); } for(sz=11;sz<256;sz++){ ec_probmod_init_full(&mod,sz,1,sz+(sz>>1),NULL); for(i=0;i<sz;i++){ s=((unsigned)(i*45678901+7))%sz; sym=ec_probmod_read_range(&mod,&dec,EC_MAXI(s-5,0),EC_MINI(s+6,sz)); if(sym!=s){ fprintf(stderr,"Decoded %i instead of %i with sz of %i.\n",sym,s,sz); return -1; } } ec_probmod_clear(&mod); } nbits2=ec_dec_tell(&dec,4); if(nbits!=nbits2){ fprintf(stderr, "Reported number of bits used was %0.2lf, should be %0.2lf.\n", ldexp(nbits2,-4),ldexp(nbits,-4)); } ec_byte_writeclear(&buf); fprintf(stderr,"All tests passed.\n"); return 0; }