www.pudn.com > src.rar > convey.c~


/*---------------------------------------------------------------------------
*    Arnold transformation
* ---------------------------------------------------------------------------*/
#include 
#include "bmp.h" 

#define MAXPATHLEN 64
static 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 int Seed = 2007;
static int N = 11;

static Tv *tv = NULL;
static Tv *tv1=NULL;


void convey_matrix(int** r, int** c, int height, int width) 
{ 
    	int i, j, k, x, y; 
 
    	printf("\nConvey transform\n"); 
	int **s1, **flag; 
	s1 = int_matrix(height, width); 
	flag = int_matrix(height, width); 
 
   	/*initial state : s1*/
/*-------------------------------------------------------------------*/ 
   	srand(Seed); 
   	for (i = 0; i < height; i++) 
    	{ 
        	for(j = 0; j < width; j++) 
        	{ 
            		s1[i][j] = rand() % 2; 
			flag[i][j] = 0; 
        	} 
    	} 

/*-------------------------------------------------------------------*/
	int count = 0;
	int seq = 0;
	int max = height * width;
	int m,n;
/*---------------------------S0--------------------------------------
   	for (i = 0; i < height; i++) 
    	{ 
        	for(j = 0; j < width; j++) 
        	{
			if(s1[i][j] == 1 )
			{
				r[i][j] = seq / width;
				c[i][j] = seq % width;
				seq ++; 
				flag[i][j] = 1;
			} 
        	} 
    	}
/*---------------------------2007.05.09----------------------------------------*/
for(k = 0; k < N ; k ++)
{
   	for (i = 0; i < height; i++) 
    	{ 
        	for(j = 0; j < width; j++) 
        	{
			count = 0;
			for(m = -1; m < 2; m ++)
				for(n = -1; n < 2; n ++)
				{
					if((m+i) < 0 ||(j+n) < 0 ||(j+n) >= width || (i+m) >= height);
					else if (s1[i+m][j+n])
						count ++;
				}
			if(s1[i][j]) 
				count --;
			switch(count)
			{
				case 2:
					break;
				case 3:
					if(s1[i][j] == 0 && flag[i][j] == 0)
					{
						r[i][j] = seq / width;
						c[i][j] = seq % width;
						seq ++; 
						flag[i][j] = 1;
					}
					s1[i][j] = 1;
					break;
				default:
					s1[i][j]=0;
			} 
        	} 
    	}
	if(seq == max)
	{
		printf("k = %d \n",k+1);
		break;
	}
}
   	for (i = 0; i < height; i++) 
    	{ 
        	for(j = 0; j < width; j++) 
        	{
			if(flag[i][j] == 0 )
			{
				r[i][j] = seq / width;
				c[i][j] = seq % width;
				seq ++; 
				flag[i][j] = 1;
			} 
        	} 
    	}

	free_int_matrix(s1, height); 
	free_int_matrix(flag, height); 
} 


void r_convey_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);
	convey_matrix(trans_r, trans_c, height, width);
	printf("\nR_Convey 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);
}



void Convey_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("Convey : 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("Convey: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);
	convey_matrix(trans_r, trans_c, height, width);
/*------------------------------------------------*/ 
	printf("\n\n=== Convey 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];
	char *filename0 ="result/Convey/";
	strcpy(filename,filename0);

	strcat(filename,"Convey_");
	gcvt(Seed,5,string);
	strcat(filename, string);
	strcat(filename,"_");
	gcvt(N,5,string);
	strcat(filename, string);
	strcat(filename,"_");
	gcvt(k,5,string);
	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("=== Convey transform complete! ===\n"); 
} 

void R_Convey_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("Convey : 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_Convey: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_convey_matrix(trans_r, trans_c, height, width);
/*------------------------------------------------*/ 
	printf("\n\n=== R_Convey 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/Convey/";
	strcpy(filename,filename0);
	strcat(filename,file_name);
	strcat(filename,"+R_Convey_");
	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_Convey 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            Convey_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("Convey 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/Convey/gRGB.gif");

	tv = tv_create("Convey original"); 
     	tv1 = tv_create("Convey 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("N = ", &N, 5);
	tw_iglobal("Seed = ", &Seed, 5);
	tw_newrow();
	tw_iglobal("Scrambling times = ", &SCRAM_n, 5); 
	tw_newrow(); 
	tw_button("Convey", Convey_Trans,NULL);
	tw_button("R_Convey", R_Convey_Trans,NULL);
	tw_newrow();
	tw_newrow(); 
	tw_end_tool(); 
}