/*---------------------------------------------------------------------------*    hilbert transformation* ---------------------------------------------------------------------------*/#include <tina/all_tina.h>#include "bmp.h"
#define MAXPATHLEN 64static char directory_name[MAXPATHLEN];static char file_name[MAXPATHLEN];static char pathname[MAXPATHLEN];
static void *pdir=0, *pfile=0;static int SCRAM_n = 1;static Tv *tv = NULL;static Tv *tv1=NULL;
static int x = -1, y = 0;
static int i = 0, j = 0;
static int width;

static void step(int dir, int **r, int **c)
{
	/*int x0 = x, y0 = y;*/

	switch(dir & 3)
	{
		case 0: x = x + 1; break;
		case 1: y = y + 1; break;
		case 2: x = x - 1; break;
		case 3: y = y - 1; break;
	}
	/*if(x0 >= 0) tv_linexy(tv, x0*10, y0*10, x*10, y*10);*/
	if(j == width)
	{
		j = 0;
		++i;
	}
	r[y][x] = i;
	c[y][x] = j++;
}

static void hilbert(int dir, int rot, int order, int **r, int **c)
{
	if(order == 0) return;

	dir += rot;
	hilbert(dir, -rot, order-1, r, c);
	step(dir, r, c);
	dir = dir - rot;
	hilbert(dir, rot, order-1, r, c);
	step(dir, r, c);
	hilbert(dir, rot, order-1, r, c);
	dir = dir - rot;
	step(dir, r, c);
	hilbert(dir, -rot, order-1, r, c);
}

static void hilbert_curve(int order, int **r, int **c)
{
	x = -1, y = 0;
	i = 0, j = 0;
	width = 1;
	int k;
	for(k = 0; k < order; k++) width *= 2;
	step(0, r, c);
	hilbert(0, 1, order, r, c);
}/*---------------------------------------------------------------------------     To generate the scrambling mapping-matrix of hilbert transformation. ---------------------------------------------------------------------------*/void hilbert_matrix(int** r, int** c, int height, int width)
{
	int i, lev = 1;
    while((int)pow(2,lev) < height) lev++;
	printf("\nHilbert transform\n");
	hilbert_curve(lev, r, c);
} /*---------------------------------------------------------------------------  To generate the inverse scrambling mapping-matrix of hilbert transformation. ---------------------------------------------------------------------------*/void r_hilbert_matrix(int** src_r,int** src_c, int height, int width){	int i,j,r,c;	int **trans_r, **trans_c;
	trans_r = int_matrix(height, width);
	trans_c = int_matrix(height, width);	hilbert_matrix(trans_r, trans_c, height, width);	printf("\nR_Hilbert transform\n");	for(i = 0; i < height; i++)
	{
		for(j = 0; j < width; j++)
		{
			r = trans_r[i][j];
			c = trans_c[i][j];
			src_r[r][c] = i;
			src_c[r][c] = j;
		}
	}	free_int_matrix(trans_r, height);
	free_int_matrix(trans_c, height);}static 	Imrect *read_image(void){	Imrect *srcIm;/*----------------Read the image---------------------------*/  	(void) strip_spaces(file_name);  	(void) strip_spaces(directory_name);  	(void) string_append(pathname, directory_name, "/", file_name, NULL);	srcIm = ReadGIF(pathname, 1);/*------------------------------------------------*/	return(srcIm);} /*---------------------------------------------------------------------------     hilbert transformation ---------------------------------------------------------------------------*/void Hilbert_Trans(int nTrans)
{
	Imrect *srcIm,*destIm,*tmpIm;
	int width, height;
	Imregion *roi;	int i,j,k;	int type;
	BYTE pix;/*	if (stack_check_types(IMRECT, NULL) == false)
	{
		error("Hilbert : wrong types on stack", warning);
		return;
	}	srcIm = (Imrect *) stack_pop(&type);*/	srcIm = read_image();	//srcIm = DrawRGB_proc();
	width = srcIm->width;
    	height = srcIm->height;
	if (width != height)
	{
		error("Hilbert:width != height", warning);
		return;
	}

	roi = srcIm->region;
	if (roi == NULL) return ;/*------------------------------------------------*/	int **trans_r, **trans_c;
	trans_r = int_matrix(height, width);
	trans_c = int_matrix(height, width);	hilbert_matrix(trans_r, trans_c, height, width);/*------------------------------------------------*/
	printf("\n\n=== Hilbert Transform begin: ===\n");	destIm = im_alloc( height, width, roi, int_v );	int r,c;
	for( k = 0; k < SCRAM_n; k++)
	{			for (i = 0; i < height  ; i++)
		{
			for (j = 0; j< width ; j++)
			{								IM_PIX_GET(srcIm, i, j, pix);
				IM_PIX_SET(destIm, trans_r[i][j], trans_c[i][j], pix);
			}
		}
 		tmpIm = srcIm;
		srcIm = destIm;
		destIm = tmpIm;	}/*----------------------Save the image as .gif--------------------------*/	char filename[64];	char string[25];	gcvt(k,5,string);	printf("string=%s\n",string);	char *filename0 ="result/Hilbert/";	strcpy(filename,filename0);	//strcat(filename,file_name);	strcat(filename,"Hilbert_");	strcat(filename, string);	strcat(filename,".gif");	//gif_write_file(srcIm,filename);	gif_write_file_color(srcIm,filename);	//strcat(filename,".bmp");	//ImrectToBMP8(srcIm,filename);//}/*------------------------------------------------*/	tv_imrect2(tv1, srcIm);	im_free(destIm);	free_int_matrix(trans_r, height);
	free_int_matrix(trans_c, height);	printf("=== Hilbert transform complete! ===\n");
}
/*---------------------------------------------------------------------------    inverse hilbert transformation ---------------------------------------------------------------------------*/void R_Hilbert_Trans(int nTrans)
{
	Imrect *srcIm,*destIm,*tmpIm;
	int width, height;
	Imregion *roi;	int i,j,k;	int type;
	BYTE pix;/*	if (stack_check_types(IMRECT, NULL) == false)
	{
		error("Hilbert : wrong types on stack", warning);
		return;
	}	srcIm = (Imrect *) stack_pop(&type);*/	srcIm = read_image();
	width = srcIm->width;
    	height = srcIm->height;
	if (width != height)
	{
		error("R_Hilbert:width != height", warning);
		return;
	}

	roi = srcIm->region;
	if (roi == NULL) return ;
/*------------------------------------------------*/	int **trans_r, **trans_c;
	trans_r = int_matrix(height, width);
	trans_c = int_matrix(height, width);	r_hilbert_matrix(trans_r, trans_c, height, width);/*------------------------------------------------*/
	printf("\n\n=== R_Hilbert Transform begin: ===\n");	destIm = im_alloc( height, width, roi, int_v );	int r,c;
	for( k = 0; k < SCRAM_n; k++)
	{			for (i = 0; i < height  ; i++)
		{
			for (j = 0; j< width ; j++)
			{								IM_PIX_GET(srcIm, i, j, pix);
				IM_PIX_SET(destIm, trans_r[i][j], trans_c[i][j], pix);
			}
		}
 		tmpIm = srcIm;
		srcIm = destIm;
		destIm = tmpIm;	}/*----------------------Save the image as .gif--------------------------*/	char filename[64];	char string[25];	gcvt(k,5,string);	printf("string=%s\n",string);	char *filename0 ="result/Hilbert/";	strcpy(filename,filename0);	strcat(filename,file_name);	strcat(filename,"+R_Hilbert_");	strcat(filename, string);	strcat(filename,".gif");	gif_write_file(srcIm,filename);/*------------------------------------------------*/	tv_imrect2(tv1, srcIm);	im_free(destIm);	free_int_matrix(trans_r, height);
	free_int_matrix(trans_c, height);	printf("=== R_Hilbert transform complete! ===\n");
}static void tv_choice_proc(choice)
int     choice;
{
    switch (choice)
    	{
    	case 0: tv_set_next(tv);
		break;
    	case 1: tv_set_next(tv1);
		break;
    	default:error("tv_choice_proc: unknown choice\n", warning);
		break;
    	}}static void scan_proc(void){    scan_files(directory_name,file_name);    tw_sglobal_reset(pdir);    tw_sglobal_reset(pfile);}void            Hilbert_tool(int x, int y)
{
	static void *tool = NULL;
	static void tv_choice_proc();	if (tool)
	{
		tw_show_tool(tool);
		return;
	}
	tool = (void *)tw_tool("Hilbert Scrambling tool", x, y);    	/* Initialise pathname from environment variable (or #define) */    	(void) environ_pathname_get(directory_name, file_name,				"TINA_IMAGE_DEFAULT",				"/home/wzz/newimscramdegree/result/Hilbert/gRGB.gif");	tv = tv_create("Hilbert original");
     	tv1 = tv_create("Hilbert scrambled");
        tw_choice("Tv choice", tv_choice_proc, 0, "original","scrambled", NULL);	tw_newrow();
	tw_newrow();	/*---------------------------*/  	pdir = (void *) tw_sglobal("Directory:", directory_name, 32);    	tw_button("scan", scan_proc, NULL);  	tw_newrow();  	pfile = (void*) tw_sglobal("File:", file_name, 32);  	tw_newrow();	/*---------------------------*/	tw_iglobal("Scrambling times = ", &SCRAM_n, 5);
	tw_newrow();
	tw_button("Hilbert", Hilbert_Trans,NULL);	tw_button("R_Hilbert", R_Hilbert_Trans,NULL);	tw_newrow();	tw_newrow();
	tw_end_tool();
}