📄 coder_bp_zt2.c
字号:
/*****
coder_BP :
could get more speed out of arithc and soz
arithc could lose its struct and just code from locals
ALPHA_REMAP does seem to help speed a bit. The idea of it is to
make most common symbols lowest in alpha order, so that the
cheezy soz is as fast as possible
*****/
#include <stdio.h>
#include <stdlib.h>
#include <crblib/inc.h>
#include <crblib/arithc.h>
#include <crblib/soz.h>
#include "coder.h"
extern int tune_param;
static int bp_remap[] = {0,1,2,5,3,9,6,11,4,10,7,12,8,13,14,15};
static int bp_unmap[] = {0,1,2,4,8,3,6,10,12,5,9,7,11,13,14,15};
// #define bitModel(bit,P0,PT) do { PT += INC; if (!(bit)) P0 += INC; if ( PT > TOTMAX ) { PT >>= 1; P0 >>= 1; P0++; PT += 2; } } while(0)
#define bitModel(bit,P0,PT) do { PT += INC; P0 += ((bit)?0:INC); if ( PT > TOTMAX ) { PT >>= 1; P0 >>= 1; P0++; PT += 2; } } while(0)
#define bitEnc(bit,ari,P0,PT) do { arithEncBit(ari,P0,PT,bit); bitModel(bit,P0,PT); } while(0)
#define bitDec(bit,ari,P0,PT) do { bit = arithDecBit(ari,P0,PT); bitModel(bit,P0,PT); } while(0)
#define INC 20
#define TOTMAX 4000
#define ORDER1_CONTEXTS 5
#define ORDER1_ALPHABET 16 //4 bits
#define ORDER1_TOTMAX 15000
#define ORDER1_INC 30
void coderBPZT2_init(coder *c);
void coderBPZT2_free(coder *c);
void coderBPZT2_flush(coder *c);
void coderBPZT2_encodeBandZT(coder *me,int *band,int width,int height,int bits,int fullw,int **rows);
void coderBPZT2_decodeBandZT(coder *me,int *band,int width,int height,int bits,int fullw,int **rows);
void coderBPZT2_decodeBandZT_eatZTs(coder *me,int *band,int width,int height,int bits,int fullw,int **rows);
void coderBPZT2_decodeBandZT_noZTs(coder *me,int *band,int width,int height,int bits,int fullw,int **rows);
void coderBPZT2_decodeBandZT_ZTs(coder *me,int *band,int width,int height,int bits,int fullw,int **rows);
coder coderBPZT2 = {
"BP ZT 2",
coderBPZT2_init,
coderBPZT2_free,
NULL,NULL,NULL,NULL,
coderBPZT2_encodeBandZT,
coderBPZT2_decodeBandZT
};
typedef struct {
soz ** o1;
int zeros_p0[ORDER1_CONTEXTS],zeros_pt[ORDER1_CONTEXTS];
} myInfo;
void coderBPZT2_init(coder *c)
{
myInfo *d;
int i;
if ( (d = new(myInfo)) == NULL )
errexit("ozero init failed");
c->data = d;
if ( (d->o1 = newarray(void *,ORDER1_CONTEXTS)) == NULL )
errexit("Order1_Init failed!");
for(i=0;i<ORDER1_CONTEXTS;i++) {
if ( (d->o1[i] = sozCreate(c->arith,ORDER1_ALPHABET,ORDER1_TOTMAX,ORDER1_INC)) == NULL )
errexit("context creation failed!");
d->zeros_p0[i] = 1; d->zeros_pt[i] = 2;
}
}
void coderBPZT2_flush(coder *c)
{
myInfo *d;
int i;
d = c->data;
for(i=0;i<ORDER1_CONTEXTS;i++) {
sozReset(d->o1[i]);
d->zeros_p0[i] = 1; d->zeros_pt[i] = 2;
}
}
void coderBPZT2_free(coder *c)
{
if ( c->data ) {
myInfo *d;
d = c->data;
if ( d->o1 ) {
int i;
for(i=0;i<ORDER1_CONTEXTS;i++) {
if ( d->o1[i] ) sozFree(d->o1[i]);
}
}
free(d);
c->data = NULL;
}
}
void coderBPZT2_encodeBandZT(coder *me,int *band,int width,int height,int bits,int fullw,int **rows)
{
int x,y,context,block;
int *dp,*dpn;
soz **o1 = ((myInfo *)me->data)->o1;
arithInfo *ari = me->arith;
int donemask,nextmask,bitmask;
int *zeros_p0,*zeros_pt;
bool bottom;
zeros_p0 = ((myInfo *)me->data)->zeros_p0;
zeros_pt = ((myInfo *)me->data)->zeros_pt;
if ( width == (fullw >>1) ) bottom = true;
else bottom = false;
bitmask = 1<<bits;
for(x=bitmask,nextmask=0; x<(1<<CODE_MAX_BPN) ;x<<=1) nextmask += x;
donemask = nextmask - bitmask;
dp = band;
for(y=height>>1;y--;) {
dpn = dp + fullw;
for(x=width>>1;x--;) {
if ( dp[0] & FLAG_CHILD_ZERO ) { // the whole quartet is zero
dp[0] -= FLAG_CHILD_ZERO; // it's a zerotree child, don't code
dp[1] -= FLAG_CHILD_ZERO; // undo the evil unnecessary work of FindZeroTrees
dpn[0] -= FLAG_CHILD_ZERO;
dpn[1] -= FLAG_CHILD_ZERO;
} else {
context = ((dp[0] & donemask)?1:0) + ((dp[1] & donemask)?1:0) + ((dpn[0] & donemask)?1:0) + ((dpn[1] & donemask)?1:0);
/***
block = (( dp[0] & bitmask )?1:0)
+ (( dp[1] & bitmask )?2:0)
+ (( dpn[0] & bitmask )?4:0)
+ (( dpn[1] & bitmask )?8:0);
****/
block = ( (dp[0] >>bits) & 1 )
+ (( (dp[1] >>bits) & 1 )<<1)
+ (( (dpn[0] >>bits) & 1 )<<2)
+ (( (dpn[1] >>bits) & 1 )<<3);
sozEncode(o1[context],bp_remap[block]);
if ( ! bottom ) {
/** code zerotree flags on any zero bits **/
switch(block) {
case 15:
break;
case 14:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 13:
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 12:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 11:
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 10:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 9:
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 8:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 7:
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 6:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 5:
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 4:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 3:
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 2:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 1:
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
case 0:
if ( dp[0] & FLAG_ISOLATED_ZERO ) { dp[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dp[1] & FLAG_ISOLATED_ZERO ) { dp[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[0] & FLAG_ISOLATED_ZERO ) { dpn[0] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
if ( dpn[1] & FLAG_ISOLATED_ZERO ) { dpn[1] -= FLAG_ISOLATED_ZERO; bitEnc(0,ari,zeros_p0[context],zeros_pt[context]); } else { bitEnc(1,ari,zeros_p0[context],zeros_pt[context]); }
break;
}
}
}
dp += 2; dpn += 2;
}
dp += fullw + fullw - width;
}
}
void coderBPZT2_decodeBandZT(coder *me,int *band,int width,int height,int bits,int fullw,int **rows)
{
bool bottom;
if ( width == (fullw >>1) ) bottom = true;
else bottom = false;
if ( bottom ) {
coderBPZT2_decodeBandZT_noZTs(me,band,width,height,bits,fullw,rows);
} else {
coderBPZT2_decodeBandZT_ZTs(me,band,width,height,bits,fullw,rows);
}
}
void coderBPZT2_decodeBandZT_noZTs(coder *me,int *band,int width,int height,int bits,int fullw,int **rows)
{
int x,y,context,block;
int *dp,*dpn;
soz **o1 = ((myInfo *)me->data)->o1;
arithInfo *ari = me->arith;
int donemask,nextmask,bitmask;
bitmask = 1<<bits;
for(x=bitmask,nextmask=0; x<(1<<CODE_MAX_BPN) ;x<<=1) nextmask += x;
donemask = nextmask - bitmask;
/**
this is almost identical to the extremely simple pre-ZT BP coder
**/
dp = band;
for(y=0;y<height;y+=2) {
dpn = dp + fullw;
for(x=0;x<width;x+=2) {
if ( dp[0] & FLAG_ALREADY_DECODED ) {
dp[0] -= FLAG_ALREADY_DECODED;
} else {
context = ((dp[0] & donemask)?1:0) + ((dp[1] & donemask)?1:0) + ((dpn[0] & donemask)?1:0) + ((dpn[1] & donemask)?1:0);
block = bp_unmap[sozDecode(o1[context])];
dp[0 ] |= ( block & 1 )?bitmask:0;
dp[1 ] |= ( block & 2 )?bitmask:0;
dpn[0] |= ( block & 4 )?bitmask:0;
dpn[1] |= ( block & 8 )?bitmask:0;
}
dp += 2; dpn += 2;
}
dp += fullw + fullw - width;
}
}
void coderBPZT2_decodeBandZT_ZTs(coder *me,int *band,int width,int height,int bits,int fullw,int **rows)
{
int x,y,context,block;
int *dp,*dpn;
soz **o1 = ((myInfo *)me->data)->o1;
arithInfo *ari = me->arith;
int donemask,nextmask,bitmask;
int offwidth,offheight,*children;
int bit,tx,ty;
int *zeros_p0,*zeros_pt;
zeros_p0 = ((myInfo *)me->data)->zeros_p0;
zeros_pt = ((myInfo *)me->data)->zeros_pt;
bitmask = 1<<bits;
for(x=bitmask,nextmask=0; x<(1<<CODE_MAX_BPN) ;x<<=1) nextmask += x;
donemask = nextmask - bitmask;
offheight = (band - rows[0])/fullw;
offwidth = (band - rows[0]) - fullw*offheight;
dp = band;
for(y=0;y<height;y+=2) {
dpn = dp + fullw;
for(x=0;x<width;x+=2) {
if ( dp[0] & FLAG_ALREADY_DECODED ) {
dp[0] ^= FLAG_ALREADY_DECODED;
tx = (offwidth + x)<<1;
ty = (offheight+ y)<<1;
rows[ty ][tx ] |= FLAG_ALREADY_DECODED;
rows[ty ][tx+2] |= FLAG_ALREADY_DECODED;
rows[ty+2][tx ] |= FLAG_ALREADY_DECODED;
rows[ty+2][tx+2] |= FLAG_ALREADY_DECODED;
} else {
context = ((dp[0] & donemask)?1:0) + ((dp[1] & donemask)?1:0) + ((dpn[0] & donemask)?1:0) + ((dpn[1] & donemask)?1:0);
block = bp_unmap[sozDecode(o1[context])];
children = rows[(offheight+ y)<<1] + ((offwidth + x)<<1);
if ( block & 1 ) dp[0 ] |= bitmask;
else {
bitDec(bit,ari,zeros_p0[context],zeros_pt[context]);
children[0] |= bit?FLAG_ALREADY_DECODED:0;
}
if ( block & 2 ) dp[1 ] |= bitmask;
else {
bitDec(bit,ari,zeros_p0[context],zeros_pt[context]);
children[2] |= bit?FLAG_ALREADY_DECODED:0;
}
if ( block & 4 ) dpn[0] |= bitmask;
else {
bitDec(bit,ari,zeros_p0[context],zeros_pt[context]);
children[fullw+fullw] |= bit?FLAG_ALREADY_DECODED:0;
}
if ( block & 8 ) dpn[1] |= bitmask;
else {
bitDec(bit,ari,zeros_p0[context],zeros_pt[context]);
children[fullw+fullw+2] |= bit?FLAG_ALREADY_DECODED:0;
}
}
dp += 2; dpn += 2;
}
dp += fullw + fullw - width;
}
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -