📄 wvtpezw_tree_init_encode.cpp
字号:
/****************************************************************************/
/* MPEG4 Visual Texture Coding (VTC) Mode Software */
/* */
/* This software was jointly developed by the following participants: */
/* */
/* Single-quant, multi-quant and flow control */
/* are provided by Sarnoff Corporation */
/* Iraj Sodagar (iraj@sarnoff.com) */
/* Hung-Ju Lee (hjlee@sarnoff.com) */
/* Paul Hatrack (hatrack@sarnoff.com) */
/* Shipeng Li (shipeng@sarnoff.com) */
/* Bing-Bing Chai (bchai@sarnoff.com) */
/* B.S. Srinivas (bsrinivas@sarnoff.com) */
/* */
/* Bi-level is provided by Texas Instruments */
/* Jie Liang (liang@ti.com) */
/* */
/* Shape Coding is provided by OKI Electric Industry Co., Ltd. */
/* Zhixiong Wu (sgo@hlabs.oki.co.jp) */
/* Yoshihiro Ueda (yueda@hlabs.oki.co.jp) */
/* Toshifumi Kanamaru (kanamaru@hlabs.oki.co.jp) */
/* */
/* OKI, Sharp, Sarnoff, TI and Microsoft contributed to bitstream */
/* exchange and bug fixing. */
/* */
/* */
/* In the course of development of the MPEG-4 standard, this software */
/* module is an implementation of a part of one or more MPEG-4 tools as */
/* specified by the MPEG-4 standard. */
/* */
/* The copyright of this software belongs to ISO/IEC. ISO/IEC gives use */
/* of the MPEG-4 standard free license to use this software module or */
/* modifications thereof for hardware or software products claiming */
/* conformance to the MPEG-4 standard. */
/* */
/* Those intending to use this software module in hardware or software */
/* products are advised that use may infringe existing patents. The */
/* original developers of this software module and their companies, the */
/* subsequent editors and their companies, and ISO/IEC have no liability */
/* and ISO/IEC have no liability for use of this software module or */
/* modification thereof in an implementation. */
/* */
/* Permission is granted to MPEG members to use, copy, modify, */
/* and distribute the software modules ( or portions thereof ) */
/* for standardization activity within ISO/IEC JTC1/SC29/WG11. */
/* */
/* Copyright 1995, 1996, 1997, 1998 ISO/IEC */
/****************************************************************************/
/****************************************************************************/
/* Texas Instruments Predictive Embedded Zerotree (PEZW) Image Codec */
/* Developed by Jie Liang (liang@ti.com) */
/* */
/* Copyright 1996, 1997, 1998 Texas Instruments */
/****************************************************************************/
/****************************************************************************
File name: wvtpezw_tree_init_encode.c
Author: Jie Liang (liang@ti.com)
Functions: functions for initialization and closing of the
PEZW encoder.
Revisions: v1.0 (10/04/98)
*****************************************************************************/
#include <time.h>
#include <stdlib.h>
#include <math.h>
#include "wvtPEZW.hpp"
#include "PEZW_zerotree.hpp"
#include "wvtpezw_tree_codec.hpp"
#include "PEZW_functions.hpp"
void PEZW_encode_init (Int levels, Int Imgwidth, Int Imgheight)
{
int len;
int i,j;
int x,y;
int nsym=No_of_symbols;
int adapt=ADAPTATION_MODE;
int contexts;
int *freq;
int pos, bpos;
int hpos_start, hpos_end;
int vpos_start, vpos_end;
int dc_width, dc_height;
int hsize, vsize;
int band;
int npix;
int start,end;
int IsStream=1;
int Mbplane=0;
int NumTree=0;
#ifdef DEBUG_FILE
fp_debug = fopen("zerotree_debug_encode.txt","w");
if(DEBUG_SYMBOL)
fprintf(fp_debug,"\n********Encoder*********\n");
#endif
hsize = Imgwidth;
vsize = Imgheight;
/* define the global variables defined in
wvtpezw_tree_codec.h */
Max_Bitplane = MAX_BITPLANE; //保留足够的动态范围
Min_Bitplane = MIN_BITPLANE;
tree_depth = levels;
MaxValue=0;
//DC频带的宽、高值
dc_width = hsize>>levels;
dc_height = vsize>>levels;
//零树中每一个分解层的位置
len_tree_struct = 0;
level_pos = (short *)calloc(tree_depth,sizeof(short));
level_pos[0]=0;
for (i=1;i<levels;i++){
len_tree_struct += 1<<(2*(i-1));
level_pos[i] = len_tree_struct;
}
len_tree_struct += 1<<(2*(levels-1));
//数据空间大小的初始化
Init_Bufsize = (int **)calloc(tree_depth,sizeof(int *));
for (i=0;i<tree_depth;i++){
Init_Bufsize[i]=(int *)calloc(Max_Bitplane,sizeof(int));
for(j=0;j<Max_Bitplane;j++)
Init_Bufsize[i][j] = Initial_Bufsize;
}
//bits_to_go数据结构的初始化
bits_to_go_inBuffer = (unsigned char **)calloc(tree_depth,sizeof(char *));
for (i=0;i<tree_depth;i++)
bits_to_go_inBuffer[i]=(unsigned char *)calloc(Max_Bitplane,sizeof(char));
//信噪比控制
snr_weight = (int *)calloc(tree_depth,sizeof(int));
bitplane = (unsigned char *)calloc(tree_depth,sizeof(char));
//小波系数的数据结构初始化
the_wvt_tree = (WINT *)calloc(len_tree_struct,sizeof(WINT));
abs_wvt_tree = (WINT *)calloc(len_tree_struct,sizeof(WINT));
maskbit=(unsigned int *) calloc(tree_depth,sizeof(int));
//系数最大绝对值的数据结构
wvt_tree_maxval = (WINT *)calloc(len_tree_struct - (1<<(2*(levels-1))),
sizeof(WINT));
//小波系数读取的位置信息映射
hloc_map=(int *)calloc(len_tree_struct,sizeof(int));
vloc_map=(int *)calloc(len_tree_struct,sizeof(int));
hloc_map[0]=0;
vloc_map[0]=0;
for(i=1;i<tree_depth;i++){
npix=1<<(2*(i-1));
pos=level_pos[i];
start=level_pos[i-1];
end=level_pos[i];
for(j=start;j<end;j++)
{
hpos_start=2*hloc_map[j];
vpos_start=2*vloc_map[j];
hpos_end=hpos_start+2;
vpos_end=vpos_start+2;
for(y=vpos_start;y<vpos_end;y++)
for(x=hpos_start;x<hpos_end;x++){
hloc_map[pos]=x;
vloc_map[pos]=y;
pos++;
}
}
}
//开始零树扫描
len=2*(len_tree_struct-(1<<(2*(levels-1))));
ScanTrees = (short *)calloc(len,sizeof(short));
next_ScanTrees = (short *)calloc(len,sizeof(short));
//重要系数的位置
sig_pos = (short *)calloc(len_tree_struct,sizeof(short));
sig_layer = (char *)calloc(len_tree_struct,sizeof(char));
//初始化重要系数的数目
num_Sig = 0;
//先前零树的状态值
prev_label = (unsigned char*)calloc(len_tree_struct,sizeof(char));
//码流数据空间的指针初始化
PEZW_bitstream = (unsigned char***) calloc(tree_depth,sizeof(char **));
for(i=0;i<tree_depth;i++){
PEZW_bitstream[i]=(unsigned char**)calloc(Max_Bitplane,sizeof(char *));
for(bpos=Max_Bitplane-1;bpos>=0;bpos--)
PEZW_bitstream[i][bpos] = (unsigned char *) calloc(Init_Bufsize[i][bpos]
,sizeof(char));
}
//算术编码器结构
Encoder = (Ac_encoder **)calloc(tree_depth,sizeof(Ac_encoder *));
for(i=0;i<tree_depth;i++)
Encoder[i]=(Ac_encoder *)calloc(Max_Bitplane, sizeof(Ac_encoder));
//用于算术编码的上下文概率模型
context_model = (Ac_model *)calloc(Max_Bitplane*levels*NumContexts,sizeof(Ac_model));
for(bpos=Max_Bitplane-1;bpos>=0;bpos--)
for(i=0;i<tree_depth;i++)
for(j=0;j<NumContext_per_pixel;j++)
for (band=0;band<NumBands;band++)
{
nsym = 4;
contexts=bpos*tree_depth*NumContexts+i*NumContexts+j*NumBands+band;
if ((i==tree_depth-1)||(j==IZER))
freq = freq_dom2_IZER;
else
freq = freq_dom_ZTRZ;
Ac_model_init (&context_model[contexts], nsym, freq,
Max_frequency_TI, adapt);
}
model_sub = (Ac_model *)calloc(tree_depth*MAX_BITPLANE,sizeof(Ac_model));
model_sign = (Ac_model *)calloc(tree_depth*MAX_BITPLANE,sizeof(Ac_model));
for(i=0;i<tree_depth*MAX_BITPLANE;i++){
Ac_model_init (&model_sub[i],nsym,freq_dom2_IZER, Max_frequency_TI, adapt);
Ac_model_init (&model_sign[i],nsym,freq_dom2_IZER, Max_frequency_TI, adapt);
}
return;
}
/* end of coding processing (for each layer):
flush the arithmetic coder
output the pointer to the bitstream and
the lenght of the bitstream
bitstream and Init_Bufsize are the
return parameters (global variables ) */
void PEZW_encode_done ()
{
Int i, j;
Int Mbplane=0;
UChar *spbuffer;
Int bpos, band, contexts;
/* determin the maximum AC bitplane */
Mbplane=0;
for(i=Max_Bitplane-1;i>=0;i--){
if(MaxValue>>i){
Mbplane=i;
break;
}
}
if(MaxValue==0)
Max_Bitplane=0;
else
Max_Bitplane=Mbplane+1;
for(i=0;i<tree_depth;i++)
{
for(j=Max_Bitplane-1;j>=Min_Bitplane;j--)
{
Ac_encoder_done (&Encoder[i][j]);
bits_to_go_inBuffer[i][j] = Encoder[i][j].bits_to_go;
PEZW_bitstream[i][j] = Encoder[i][j].original_stream;
spbuffer = Encoder[i][j].stream;
Init_Bufsize[i][j] = spbuffer - PEZW_bitstream[i][j];
}
}
/* free memory */
free(level_pos);
free(bitplane);
free(snr_weight);
free(the_wvt_tree);
free(abs_wvt_tree);
free(wvt_tree_maxval);
free(maskbit);
free(ScanTrees);
free(next_ScanTrees);
free(sig_pos);
free(sig_layer);
free(prev_label);
for(i=0;i<tree_depth;i++){
free(Encoder[i]);
}
free(Encoder);
for(bpos=Max_Bitplane-1;bpos>=0;bpos--)
for(i=0;i<tree_depth;i++)
for(j=0;j<NumContext_per_pixel;j++)
for (band=0;band<NumBands;band++)
{
contexts=bpos*tree_depth*NumContexts+i*NumContexts+j*NumBands+band;
AC_free_model(&context_model[contexts]);
}
for(i=0;i<tree_depth*MAX_BITPLANE;i++){
AC_free_model(&model_sign[i]);
AC_free_model(&model_sub[i]);
}
free(model_sign);
free(model_sub);
free(context_model);
#ifdef DEBUG_FILE
fp_debug = fclose(fp_debug);
#endif
return;
}
void setbuffer_PEZW_encode ()
{
Int bpos, i;
/* set the bitstream buffer for encoder */
for(i=0;i<tree_depth;i++)
for(bpos=Max_Bitplane-1;bpos>=0;bpos--)
{
#ifdef BILEVEL_AC
buffer_ptr[i][bpos] = bitstream[i][bpos];
bpinitencode (&BAC_Encoder[i][bpos],bpputbyte);
#else
Ac_encoder_init (&Encoder[i][bpos],PEZW_bitstream[i][bpos],
Init_Bufsize[i][bpos], 1);
#endif
}
}
void reset_PEZW_encode ()
{
Int bpos;
Int i,j, band;
Int contexts;
Int nsym;
Int adapt=1;
int *freq;
int len;
unsigned char *spbuffer;
/* context models */
for(bpos=Max_Bitplane-1;bpos>=0;bpos--)
for(i=0;i<tree_depth;i++)
for(j=0;j<NumContext_per_pixel;j++)
for (band=0;band<NumBands;band++)
{
nsym = 4;
contexts=bpos*tree_depth*NumContexts+i*NumContexts+j*NumBands+band;
if ((i==tree_depth-1)||(j==IZER))
freq = freq_dom2_IZER;
else
freq = freq_dom_ZTRZ;
AC_free_model(&context_model[contexts]);
Ac_model_init (&context_model[contexts], nsym, freq,
Max_frequency_TI, adapt);
}
for(i=0;i<tree_depth*MAX_BITPLANE;i++){
AC_free_model(&model_sub[i]);
AC_free_model(&model_sign[i]);
Ac_model_init (&model_sub[i],nsym,freq_dom2_IZER, Max_frequency_TI, adapt);
Ac_model_init (&model_sign[i],nsym,freq_dom2_IZER, Max_frequency_TI, adapt);
}
/* re-initialize arithmetic coder */
for(i=0;i<tree_depth;i++)
{
for(j=Max_Bitplane-1;j>=Min_Bitplane;j--)
{
/* flush the arithmetic coder */
Ac_encoder_done (&Encoder[i][j]);
PEZW_bitstream[i][j] = Encoder[i][j].original_stream;
spbuffer = Encoder[i][j].stream;
/* set new buffer position
and initialize AC coders */
len = Encoder[i][j].space_left;
Ac_encoder_init (&Encoder[i][bpos],spbuffer,
len, 1);
}
}
return;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -