#include<math.h>
#include"hensu.h"

void gradient(LPDWORD image_in,LPDWORD image_out,double amp,int ope)
{
	int cx[9],cy[9];

	switch(ope){
		case 1:
			cx[0]= 0; cx[1]= 0; cx[2]= 0;
			cx[3]= 0; cx[4]= 1; cx[5]=-1;
			cx[6]= 0; cx[7]= 0; cx[8]= 0;

			cy[0]= 0; cy[1]= 0; cy[2]= 0;
			cy[3]= 0; cy[4]= 1; cy[5]= 0;
			cy[6]= 0; cy[7]=-1; cy[8]= 0;
			break;
		case 2:
			cx[0]= 0; cx[1]= 0; cx[2]= 0;
			cx[3]= 0; cx[4]= 1; cx[5]= 0;
			cx[6]= 0; cx[7]= 0; cx[8]=-1;

			cy[0]= 0; cy[1]= 0; cy[2]= 0;
			cy[3]= 0; cy[4]= 0; cy[5]= 1;
			cy[6]= 0; cy[7]=-1; cy[8]= 0;
			break;
		default:
			cx[0]=-1; cx[1]= 0; cx[2]= 1;
			cx[3]=-2; cx[4]= 0; cx[5]= 2;
			cx[6]=-1; cx[7]= 0; cx[8]= 1;

			cy[0]=-1; cy[1]=-2; cy[2]=-1;
			cy[3]= 0; cy[4]= 0; cy[5]= 1;
			cy[6]= 1; cy[7]= 2; cy[8]= 1;
			break;
	}

	int d[9];
	int i,j,dat;
	double xx,yy,zz;

	for(j=1;j<iHeight-1;j++)
		for(i=1;i<iWidth-1;i++){
			d[0]=(BYTE)image_in[i-1+(j-1)*iWidth];
			d[1]=(BYTE)image_in[i+(j-1)*iWidth];
			d[2]=(BYTE)image_in[i+1+(j-1)*iWidth];
			d[3]=(BYTE)image_in[i-1+j*iWidth];
			d[4]=(BYTE)image_in[i+j*iWidth];
			d[5]=(BYTE)image_in[i+1+j*iWidth];
			d[6]=(BYTE)image_in[i-1+(j+1)*iWidth];
			d[7]=(BYTE)image_in[i+(j+1)*iWidth];
			d[8]=(BYTE)image_in[i+1+(j+1)*iWidth];
			xx=cx[0]*d[0]+cx[1]*d[1]+cx[2]*d[2]
			  +cx[3]*d[3]+cx[4]*d[4]+cx[5]*d[5]
			  +cx[6]*d[6]+cx[7]*d[7]+cx[8]*d[8];
			yy=cy[0]*d[0]+cy[1]*d[1]+cy[2]*d[2]
			  +cy[3]*d[3]+cy[4]*d[4]+cy[5]*d[5]
			  +cy[6]*d[6]+cy[7]*d[7]+cy[8]*d[8];
		    zz=amp*sqrt(xx*xx+yy*yy);
			dat=(int)zz;
			if(dat>255) dat=255;
			image_out[i+j*iWidth]=RGB(dat,dat,dat);
		}
}

void laplacian(LPDWORD image_in,LPDWORD image_out,double amp,int ope)
{
	int c[9];

	switch(ope){
		case 1:
			c[0]= 0; c[1]=-1; c[2]= 0;
			c[3]=-1; c[4]= 4; c[5]=-1;
			c[6]= 0; c[7]=-1; c[8]= 0;
			break;
		case 2:
			c[0]=-1; c[1]=-1; c[2]=-1;
			c[3]=-1; c[4]= 8; c[5]=-1;
			c[6]=-1; c[7]=-1; c[8]=-1;
			break;
		default:
			c[0]= 1; c[1]=-2; c[2]= 1;
			c[3]=-2; c[4]= 4; c[5]=-2;
			c[6]= 1; c[7]=-2; c[8]= 1;
			break;
	}

	int d[9];
	int i,j,dat;
	double z,zz;

	for(j=1;j<iHeight-1;j++)
		for(i=1;i<iWidth-1;i++){
			d[0]=(BYTE)image_in[i-1+(j-1)*iWidth];
			d[1]=(BYTE)image_in[i+(j-1)*iWidth];
			d[2]=(BYTE)image_in[i+1+(j-1)*iWidth];
			d[3]=(BYTE)image_in[i-1+j*iWidth];
			d[4]=(BYTE)image_in[i+j*iWidth];
			d[5]=(BYTE)image_in[i+1+j*iWidth];
			d[6]=(BYTE)image_in[i-1+(j+1)*iWidth];
			d[7]=(BYTE)image_in[i+(j+1)*iWidth];
			d[8]=(BYTE)image_in[i+1+(j+1)*iWidth];
			z=c[0]*d[0]+c[1]*d[1]+c[2]*d[2]
			 +c[3]*d[3]+c[4]*d[4]+c[5]*d[5]
			 +c[6]*d[6]+c[7]*d[7]+c[8]*d[8];
		    zz=amp*z;
			dat=(int)zz;
			if(dat<0) dat=-dat;
			else if(dat>255) dat=255;
			image_out[i+j*iWidth]=RGB(dat,dat,dat);
		}
}

void mytemplate(LPDWORD image_in,LPDWORD image_out,double amp)
{
	int d[9];
	int i,j,k,max,dat;
	int m[8];
	double zz;

	for(j=1;j<iHeight-1;j++)
		for(i=1;i<iWidth-1;i++){
			d[0]=(BYTE)image_in[i-1+(j-1)*iWidth];
			d[1]=(BYTE)image_in[i+(j-1)*iWidth];
			d[2]=(BYTE)image_in[i+1+(j-1)*iWidth];
			d[3]=(BYTE)image_in[i-1+j*iWidth];
			d[4]=(BYTE)image_in[i+j*iWidth];
			d[5]=(BYTE)image_in[i+1+j*iWidth];
			d[6]=(BYTE)image_in[i-1+(j+1)*iWidth];
			d[7]=(BYTE)image_in[i+(j+1)*iWidth];
			d[8]=(BYTE)image_in[i+1+(j+1)*iWidth];
			m[0]=d[0]+d[1]+d[2]+d[3]-2*d[4]+d[5]-d[6]-d[7]-d[8];
			m[1]=d[0]+d[1]+d[2]+d[3]-2*d[4]-d[5]+d[6]-d[7]-d[8];
			m[2]=d[0]+d[1]-d[2]+d[3]-2*d[4]-d[5]+d[6]+d[7]-d[8];
			m[3]=d[0]-d[1]-d[2]+d[3]-2*d[4]-d[5]+d[6]+d[7]+d[8];
			m[4]=-d[0]-d[1]-d[2]+d[3]-2*d[4]+d[5]+d[6]+d[7]+d[8];
			m[5]=-d[0]-d[1]+d[2]-d[3]-2*d[4]+d[5]+d[6]+d[7]+d[8];
			m[6]=-d[0]+d[1]+d[2]-d[3]-2*d[4]+d[5]-d[6]+d[7]+d[8];
			m[7]=d[0]+d[1]+d[2]-d[3]-2*d[4]+d[5]-d[6]-d[7]+d[8];
			max=0;
			for(k=0;k<8;k++)
				if(max<m[k]) max=m[k];
			zz=amp*max;
			dat=(int)zz;
			if(dat>255) dat=255;
			image_out[i+j*iWidth]=RGB(dat,dat,dat);
		}
}