bpeblockcoding.c

来自「CCSDS空间图像图像压缩标准c源代码」· C语言 代码 · 共 589 行 · 第 1/2 页

/*
Implementation of CCSDS 122.0-B-1 Recommended Standard
Please note:
(1)	Before you download and use the program, you must read and agree the license agreement carefully. 
(2)	We supply the source code and program WITHOUT ANY WARRANTIES. The users will be responsible 
        for any loses or damages caused by the use of the source code and the program. 

Author: 
Hongqiang Wang
Department of Electrical Engineering
University of Nebraska-Lincoln
Email: [e-mail], [e-mail]

Your comment and suggestions are welcome. Please report bugs to me via email and I would greatly appreciate it. 
Jan. 21, 2007
*/ 
#include <stdlib.h>
#include <math.h>
#include "global.h"

void BlockScanEncode(StructCodingPara *PtrCoding, 
					 BitPlaneBits *BlockInfo)
{
	int i = 0;
	int j = 0;
	int k = 0;
	int p = 0;
	short temp_x = 0;
	short temp_y = 0;
	short counter = 0;	
	UCHAR8 BitPlane = PtrCoding->BitPlane;
	UCHAR8 si = 0;
	WORD16 temp;
	DWORD32 BlockSeq;
	DWORD32 Bit_Set_Plane = (1 << (BitPlane - 1)) ;	
	long ** block = NULL;

	for (BlockSeq = 0;	BlockSeq < PtrCoding->PtrHeader->Header.Part3.S_20Bits; BlockSeq++)		
	{	
		si = 0;
		if (BlockInfo[BlockSeq].BitMaxAC < BitPlane)
			continue;

		PtrCoding->block_index = BlockSeq;
		block = BlockInfo[BlockSeq].PtrBlockAddress;
			
	// check the parents. 
	
		for (i = 0 ; i < 3; i++)
		{
			if(PtrCoding->PtrHeader->Header.Part4.DWTType == INTEGER_WAVELET)
			{
				if(((i == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL3_2bits >= BitPlane)) // HL3 band. 
					|| ((i == 1) && (PtrCoding->PtrHeader->Header.Part4.CustomWtLH3_2bits >= BitPlane)) // HL3 band. 
					|| ((i == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH3_2bits >= BitPlane))) // HL3 band. 
					continue;
			}

			temp_x = (i >= 1 ? 1 : 0);
			temp_y = (i != 1 ? 1 : 0);
			if ((BlockInfo[BlockSeq].StrPlaneHitHistory.TypeP & (1 << (2 - i))) == 0)
			{
				BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TYPE_P;
				BlockInfo[BlockSeq].SymbolsBlock[si].sym_len ++;	
				BlockInfo[BlockSeq].SymbolsBlock[si].sym_val <<= 1;	

				if((AMPLITUDE(block[temp_x][temp_y]) >= (1 << (BitPlane - 1))) &&
					(AMPLITUDE(block[temp_x][temp_y]) < (1 << BitPlane)))
				{
					BlockInfo[BlockSeq].StrPlaneHitHistory.TypeP += (1 << (2 - i));
					BlockInfo[BlockSeq].SymbolsBlock[si].sym_val += 1;	
					BlockInfo[BlockSeq].SymbolsBlock[si].sign <<= 1;
					BlockInfo[BlockSeq].SymbolsBlock[si].sign += SIGN(block[temp_x][temp_y]);		
				}
			}
			else
			{			
				temp = ( (AMPLITUDE(block[temp_x][temp_y])) & (Bit_Set_Plane)) > 0 ? 1 : 0 ;
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
				{
					if (((i == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL3_2bits < BitPlane)) 
						|| ((i == 1) && (PtrCoding->PtrHeader->Header.Part4.CustomWtLH3_2bits < BitPlane))
						|| ((i == 2) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHH3_2bits < BitPlane)))
					{
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentRefSymbol <<= 1;
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentRefSymbol += temp;
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentSymbolLength ++;
					}				
				}
				else 
				{				
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentRefSymbol <<= 1;
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentRefSymbol += temp;
						BlockInfo[BlockSeq].RefineBits.RefineParent.ParentSymbolLength ++;
				}
			}
		}  

		if (BlockInfo[BlockSeq].SymbolsBlock[si].sym_len != 0)
			si ++;
	// now determine the TranB, look into the descendent, i.e., TypeCi. 

		if (BlockInfo[BlockSeq].StrPlaneHitHistory.TranB == 0)
		{				
			for ( k = 0; k < 3; k ++)
			{					
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
				{
					if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL2_2bits >= BitPlane) &&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtHL1_2bits >= BitPlane)) // HL3 band.
						|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH2_2bits >= BitPlane)&&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtLH1_2bits >= BitPlane)) // HL3 band.
						|| ((k == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH2_2bits >= BitPlane)&&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtHH1_2bits >= BitPlane))) // HL3 band. 
						continue;
				}
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
				{
					if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL2_2bits >= BitPlane)) // HL3 band. 
						|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH2_2bits >= BitPlane)) // HL3 band. 
						|| ((k == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH2_2bits >= BitPlane))) // HL3 band. 
						goto CHECK_GC_B;
				}						
				temp_x = (k >= 1 ? 1 : 0);
				temp_x *= 2;
				temp_y = (k != 1 ? 1 : 0);
				temp_y *= 2;
				
				for ( i = temp_x ; i < temp_x + 2; i ++)
					for ( j =temp_y; j < temp_y + 2; j ++)
					{
						// check the BitPlane to see if the bit plane is one. 
						if(	(Bit_Set_Plane & AMPLITUDE(block[i][j])) > 0) 
						{
							BlockInfo[BlockSeq].StrPlaneHitHistory.TranB = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_len = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_val = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_B;
							si++;
							goto DS_Update;
						}
					}
CHECK_GC_B:;
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
					{
						if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL1_2bits >= BitPlane)) // HL3 band. 
							|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH1_2bits >= BitPlane)) // HL3 band. 
							|| ((k == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH1_2bits >= BitPlane))) // HL3 band. 
							continue;
					}		
				temp_x = (k >= 1 ? 1 : 0);
				temp_x *= 4;
				temp_y = (k != 1 ? 1 : 0);
 				temp_y *= 4;
				for ( i = temp_x; i < temp_x + 4; i ++)
					for ( j = temp_y; j < temp_y + 4; j++)
						if(	(Bit_Set_Plane & AMPLITUDE(block[i][j])) > 0) 
						{
							BlockInfo[BlockSeq].StrPlaneHitHistory.TranB = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_len = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_val = 1;
							BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_B;
							si++;
							goto DS_Update;
						}
			}
	DS_Update:;
		}


		if (BlockInfo[BlockSeq].StrPlaneHitHistory.TranB == 0)
		{
			BlockInfo[BlockSeq].SymbolsBlock[si].sym_len = 1;
			BlockInfo[BlockSeq].SymbolsBlock[si].sym_val = 0;
			BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_B;
			continue;
		}
		// TranB bits end

		if (BlockInfo[BlockSeq].SymbolsBlock[si].sym_len != 0)
			si ++;

		if (BlockInfo[BlockSeq].StrPlaneHitHistory.TranB == 1) // continue to scan
		{
			// determine if to scan the descendent. 

			for ( k = 0; k < 3; k ++)
			{
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
				{
					if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL2_2bits >= BitPlane)&&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtHL1_2bits >= BitPlane)) // HL3 band. 
						|| ((k == 1) &&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtLH2_2bits >= BitPlane)&&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtLH1_2bits >= BitPlane)) // HL3 band. 					
						|| ((k == 2) &&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtHH2_2bits >= BitPlane)&&
						(PtrCoding->PtrHeader->Header.Part4.CustomWtHH1_2bits >= BitPlane))) // HL3 band. 
						continue;
				}

				if ((BlockInfo[BlockSeq].StrPlaneHitHistory.TranD & ( 1 << ( 2 - k))) == 0)
				{
    				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
					{
						if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL2_2bits >= BitPlane)) // HL3 band. 
							|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH2_2bits >= BitPlane)) // HL3 band. 
							|| ((k == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH2_2bits >= BitPlane))) // HL3 band. 
							goto CHECK_GC;
					}	
					
					temp_x = (k >= 1 ? 1 : 0);
					temp_x *= 2;
					temp_y = (k != 1 ? 1 : 0);
					temp_y *= 2;

					for ( i = temp_x ; i < temp_x + 2; i ++)
						for ( j = temp_y; j < temp_y + 2; j ++)
						{
							// check the BitPlane to see if the bit plane is one. 
							if(	(Bit_Set_Plane & AMPLITUDE(block[i][j])) > 0) 
							{									
								BlockInfo[BlockSeq].StrPlaneHitHistory.TranD += (1 << ( 2 - k));
								BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_D;
								BlockInfo[BlockSeq].SymbolsBlock[si].sym_len ++;
								BlockInfo[BlockSeq].SymbolsBlock[si].sym_val <<= 1;
								BlockInfo[BlockSeq].SymbolsBlock[si].sym_val ++;
								goto NEW_B;
							}
						}

	CHECK_GC:;
						if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
						{
							if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL1_2bits >= BitPlane)) // HL3 band. 
								|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH1_2bits >= BitPlane)) // HL3 band. 
								|| ((k == 2) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHH1_2bits >= BitPlane))) // HL3 band. 
								continue;
						}

						temp_x = (k >= 1 ? 1 : 0);
						temp_x *= 4;
						temp_y = (k != 1 ? 1 : 0);
						temp_y *= 4;
						
						for (i = temp_x; i < temp_x + 4; i ++)
							for ( j = temp_y; j < temp_y + 4; j++)
							{
								if(	(Bit_Set_Plane & AMPLITUDE(block[i][j])) > 0) 
								{									
									BlockInfo[BlockSeq].StrPlaneHitHistory.TranD += (1 << ( 2 - k));
									BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_D;
									BlockInfo[BlockSeq].SymbolsBlock[si].sym_len ++;
									BlockInfo[BlockSeq].SymbolsBlock[si].sym_val <<= 1;
									BlockInfo[BlockSeq].SymbolsBlock[si].sym_val ++;
									goto NEW_B;
								}
							}
						// otherwise the transition bit D is 0. 
						if((BlockInfo[BlockSeq].StrPlaneHitHistory.TranD & (1 << ( 2 - k))) == 0) 
						{						
							BlockInfo[BlockSeq].SymbolsBlock[si].type = ENUM_TRAN_D;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_len ++;
							BlockInfo[BlockSeq].SymbolsBlock[si].sym_val <<= 1;								
						}
				}	
NEW_B:;	
			}
		}

		if(BlockInfo[BlockSeq].StrPlaneHitHistory.TranD == 0)
			ErrorMsg(BPE_BLOCKSCAN_CODING_ERROR);//" Error encoding of TranD! Cannot be all Zero!\n");

	// Now we determine if we need TypeCi. 
		for ( k = 0; k < 3; k ++)
		{
			// TypeCi will be always scanned unless all its mementrs have been hit. 
			if ((BlockInfo[BlockSeq].StrPlaneHitHistory.TranD & (1 << (2 - k))) != 0)
			{			
				if((PtrCoding->PtrHeader->Header.Part4).DWTType == INTEGER_WAVELET)
				{
					if(((k == 0) && (PtrCoding->PtrHeader->Header.Part4.CustomWtHL2_2bits >= BitPlane)) // HL3 band. 
						|| ((k == 1) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtLH2_2bits >= BitPlane)) // HL3 band. 
						|| ((k == 2) &&	(PtrCoding->PtrHeader->Header.Part4.CustomWtHH2_2bits >= BitPlane))) // HL3 band. 
						continue;
				}

				temp_x = (k >= 1 ? 1 : 0);
				temp_x *= 2;
				temp_y = (k != 1 ? 1 : 0);
				temp_y *= 2;

				p = 1;
				for ( i = 0 ; i < 4; i ++)
					if ((BlockInfo[BlockSeq].StrPlaneHitHistory.TypeCi[k].TypeC << (1 << (3 - i))) != 1)