#include<windows.h>
#include"resource.h"
#include"bmp_io.h"
#include"hensu.h"
#include"clip_board.h"
#include<commctrl.h>
#include<stdio.h>
#include"nichishori.h"
#include"tokuchou.h"
#include"rinkaku.h"
#include<windowsx.h>

LRESULT CALLBACK WndProc(HWND,UINT,WPARAM,LPARAM);
ATOM InitApp(HINSTANCE);
BOOL InitInstance(HINSTANCE,int);
LRESULT CALLBACK MyDlgProc(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp);
LRESULT CALLBACK MyDlgProc2(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp);
LRESULT CALLBACK MyDlgProc3(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp);

char szClassName[]="template";
int iWidth,iHeight;

HINSTANCE hInst;
int thresh=128,amp=10;
int min,max,radio=1;

int WINAPI WinMain(HINSTANCE hCurInst,HINSTANCE hPrevInst,
				   LPSTR lpsCmdLine,int nCmdShow)
{
	MSG msg;
	BOOL bRet;

	hInst=hCurInst;

	if(!InitApp(hCurInst))
		return FALSE;
	if(!InitInstance(hCurInst,nCmdShow))
		return FALSE;
	while((bRet=GetMessage(&msg,NULL,0,0))!=0){
		if(bRet==-1){
			MessageBox(NULL,"GetMessageƒGƒ‰[","Error",MB_OK);
			break;
		}else{
			TranslateMessage(&msg);
			DispatchMessage(&msg);
		}
	}
	return (int)msg.wParam;
}

ATOM InitApp(HINSTANCE hInst)
{
	WNDCLASSEX wc;
	wc.cbSize=sizeof(WNDCLASSEX);
	wc.style=CS_HREDRAW | CS_VREDRAW;
	wc.lpfnWndProc=WndProc;
	wc.cbClsExtra=0;
	wc.cbWndExtra=0;
	wc.hInstance=hInst;
	wc.hIcon=(HICON)LoadImage(NULL,
		MAKEINTRESOURCE(IDI_APPLICATION),
		IMAGE_ICON,
		0,0,
		LR_DEFAULTSIZE|LR_SHARED);
	wc.hCursor=(HCURSOR)LoadImage(NULL,
		MAKEINTRESOURCE(IDC_ARROW),
		IMAGE_CURSOR,
		0,0,
		LR_DEFAULTSIZE|LR_SHARED);
	wc.hbrBackground=(HBRUSH)GetStockObject(WHITE_BRUSH);
	wc.lpszMenuName="MYMENU";
	wc.lpszClassName=(LPCTSTR)szClassName;
	wc.hIconSm=(HICON)LoadImage(NULL,
		MAKEINTRESOURCE(IDI_APPLICATION),
		IMAGE_ICON,
		0,0,
		LR_DEFAULTSIZE|LR_SHARED);

	return RegisterClassEx(&wc);
}

BOOL InitInstance(HINSTANCE hInst,int nCmdShow)
{
	HWND hWnd;

	hWnd=CreateWindow(szClassName,
		"‘ๆ‚Vอ@“ม’ฅ‚๐’ฒ‚ื‚้",
		WS_OVERLAPPEDWINDOW,
		CW_USEDEFAULT,
		CW_USEDEFAULT,
		CW_USEDEFAULT,
		CW_USEDEFAULT,
		NULL,
		NULL,
		hInst,
		NULL);
	if(!hWnd)
		return FALSE;
	ShowWindow(hWnd,nCmdShow);
	UpdateWindow(hWnd);
	return TRUE;
}

LRESULT CALLBACK WndProc(HWND hWnd,UINT msg,WPARAM wp,LPARAM lp)
{
	static BMP_IO mybmp;

	static LPBITMAPINFO lpbiInfo;
	static LPDWORD lpPixel,lpPixelNichi,lpPixelBibun,lpPixelShow,lpPixelTemp;
	static LPDWORD lpPixelMask,lpPixelMask2;
	
	HDROP hDrop;
	char szFileName[256];

	HMENU hMenu;
	MENUITEMINFO mi;

	INITCOMMONCONTROLSEX ic;

	static double w,h;

	int x,y;

	HDC hdc;
	PAINTSTRUCT ps;

	static int cnt;
	static double size[256],ratio[256];
	static char buf[256*256],text[64];

	static RECT rc;

	switch(msg){
		case WM_CREATE:
			DragAcceptFiles(hWnd,TRUE);

			ic.dwSize=sizeof(INITCOMMONCONTROLSEX);
			ic.dwICC=ICC_UPDOWN_CLASS;
			InitCommonControlsEx(&ic);

			w=1; h=1;
			rc.left=0;
			rc.top=0;
			break;
		case WM_SIZE:
			rc.right=LOWORD(lp);
			rc.bottom=HIWORD(lp);
			break;
		case WM_DROPFILES:
			hDrop=(HDROP)wp;
			DragQueryFile(hDrop,0,szFileName,sizeof(szFileName));
			DragFinish(hDrop);

			if(!mybmp.bmp_in(szFileName,hWnd)) break;

			lpbiInfo=&mybmp.biInfo;
			lpPixel=mybmp.lpPixel;

			iWidth=mybmp.iWidth;
			iHeight=mybmp.iHeight;

			if(lpPixelShow!=NULL)
				HeapFree(GetProcessHeap(),0,lpPixelShow);

			lpPixelShow=(LPDWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,6*iWidth*iHeight*4);
			lpPixelNichi=lpPixelShow+iWidth*iHeight;
			lpPixelBibun=lpPixelShow+iWidth*iHeight*2;
			lpPixelMask=lpPixelShow+iWidth*iHeight*3;
			lpPixelMask2=lpPixelShow+iWidth*iHeight*4;
			lpPixelTemp=lpPixelShow+iWidth*iHeight*5;
			
			for(y=0;y<iHeight;y++)
				for(x=0;x<iWidth;x++){
					if(y>=2 && y<iHeight-2 && x>=2 && x<iWidth-2)
						continue;
					lpPixel[x+y*iWidth]=0;
				}

			CopyMemory(lpPixelShow,lpPixel,iWidth*iHeight*4);
			CopyMemory(lpPixelNichi,lpPixel,iWidth*iHeight*4);
			CopyMemory(lpPixelBibun,lpPixel,iWidth*iHeight*4);
			CopyMemory(lpPixelMask,lpPixel,iWidth*iHeight*4);
			CopyMemory(lpPixelMask2,lpPixel,iWidth*iHeight*4);
			CopyMemory(lpPixelTemp,lpPixel,iWidth*iHeight*4);

			w=1; h=1;
			rc.left=(int)(w*iWidth);

			InvalidateRect(hWnd,NULL,TRUE);
			break;
		case WM_PAINT:
			hdc=BeginPaint(hWnd,&ps);
			DrawText(hdc,buf,-1,&rc,DT_WORDBREAK);
			StretchDIBits(hdc,0,0,(int)(w*iWidth),(int)(h*iHeight),
				0,0,iWidth,iHeight,lpPixelShow,lpbiInfo,
				DIB_RGB_COLORS,SRCCOPY);
			EndPaint(hWnd,&ps);
			break;
		case WM_COMMAND:
			switch(LOWORD(wp)){
				case IDM_COPY:
					CopyDIB(hWnd,lpPixelShow);
					break;
				case IDM_PASTE:
					if(!mybmp.PasteDIBSection(hWnd)) break;

					lpbiInfo=&mybmp.biInfo;
					lpPixel=mybmp.lpPixel;

					iWidth=mybmp.iWidth;
					iHeight=mybmp.iHeight;

					if(lpPixelShow!=NULL)
						HeapFree(GetProcessHeap(),0,lpPixelShow);

					lpPixelShow=(LPDWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,6*iWidth*iHeight*4);
					lpPixelNichi=lpPixelShow+iWidth*iHeight;
					lpPixelBibun=lpPixelShow+iWidth*iHeight*2;
					lpPixelMask=lpPixelShow+iWidth*iHeight*3;
					lpPixelMask2=lpPixelShow+iWidth*iHeight*4;
					lpPixelTemp=lpPixelShow+iWidth*iHeight*5;
					
					for(y=0;y<iHeight;y++)
						for(x=0;x<iWidth;x++){
							if(y>=2 && y<iHeight-2 && x>=2 && x<iWidth-2)
								continue;
							lpPixel[x+y*iWidth]=0;
						}

					CopyMemory(lpPixelShow,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelNichi,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelBibun,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelMask,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelMask2,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelTemp,lpPixel,iWidth*iHeight*4);

					w=1; h=1;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_NICHI:
					if(DialogBox(hInst,"MYDLG",hWnd,(DLGPROC)MyDlgProc)==IDOK){
						CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
						threshold(lpPixelBibun,lpPixelNichi,thresh,1);
						CopyMemory(lpPixelShow,lpPixelNichi,iWidth*iHeight*4);
					}
					break;
				case IDM_LABEL:
                    CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
					labeling(lpPixelNichi,lpPixelMask,&cnt,buf);
					CopyMemory(lpPixelShow,lpPixelMask,iWidth*iHeight*4);
					break;
				case IDM_FEATURE:
                    CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
					features(lpPixelMask,lpPixelMask2,cnt,size,ratio,buf);
					CopyMemory(lpPixelShow,lpPixelMask2,iWidth*iHeight*4);
					break;
				case IDM_RATIO_SIZE:
					if(DialogBox(hInst,"MYDLG3",hWnd,(DLGPROC)MyDlgProc3)==IDOK){
                        CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
						switch(radio){
							case 1:
								ratio_extract(lpPixelMask,lpPixelMask2,cnt,ratio,(double)min/100,(double)max/100);
								sprintf(text,"‰~Œ`“xFลฌ’l=%.2lf`ล‘ๅ’l=%.2lf‚ฬ”อˆอ‚๐’Šo",(double)min/100,(double)max/100);
								break;
							case 2:
								size_extract(lpPixelMask,lpPixelMask2,cnt,size,min,max);
								sprintf(text,"–สฯFลฌ’l=%d`ล‘ๅ’l=%d‚ฬ”อˆอ‚๐’Šo",min,max);
								break;
						}
						CopyMemory(lpPixelShow,lpPixelMask2,iWidth*iHeight*4);
						SetWindowText(hWnd,text);
					}
					break;
				case IDM_MASK:
					CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
					masking(lpPixel,lpPixelShow,lpPixelMask2);
					break;
				case IDM_GRADIENT:
					if(DialogBox(hInst,"MYDLG2",hWnd,(DLGPROC)MyDlgProc2)==IDOK){
						CopyMemory(lpPixelTemp,lpPixelShow,iWidth*iHeight*4);
						gradient(lpPixel,lpPixelBibun,(double)amp/10,1);
						CopyMemory(lpPixelShow,lpPixelBibun,iWidth*iHeight*4);
					}
					break;
				case IDM_BEFORE:
					CopyMemory(lpPixelShow,lpPixelTemp,iWidth*iHeight*4);
					break;
				case IDM_ORIGINAL:
					CopyMemory(lpPixelShow,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelNichi,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelBibun,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelMask,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelMask2,lpPixel,iWidth*iHeight*4);
					CopyMemory(lpPixelTemp,lpPixel,iWidth*iHeight*4);
					break;
				case IDM_ZOOM1:
					w=1; h=1;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM2:
					w=2; h=2;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM3:
					w=3; h=3;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM4:
					w=4; h=4;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM02:
					w=1.0/2; h=1.0/2;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM03:
					w=1.0/3; h=1.0/3;
					rc.left=(int)(w*iWidth);
					break;
				case IDM_ZOOM04:
					w=1.0/4; h=1.0/4;
					rc.left=(int)(w*iWidth);
					break;
			}
			InvalidateRect(hWnd,NULL,TRUE);
			break;
		case WM_INITMENU:
			hMenu=GetMenu(hWnd);

			mi.cbSize=sizeof(MENUITEMINFO);
			mi.fMask=MIIM_STATE;
	
			if(IsClipboardFormatAvailable(CF_DIB)){
				mi.fState=MFS_ENABLED;
                SetMenuItemInfo(hMenu,IDM_PASTE,FALSE,&mi);
			}else{
				mi.fState=MFS_GRAYED;
                SetMenuItemInfo(hMenu,IDM_PASTE,FALSE,&mi);
			}

			if(w==1){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM1,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM1,FALSE,&mi);
			}
			if(w==2){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM2,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM2,FALSE,&mi);
			}
			if(w==3){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM3,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM3,FALSE,&mi);
			}
			if(w==4){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM4,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM4,FALSE,&mi);
			}
			if(w==1.0/2){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM02,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM02,FALSE,&mi);
			}
			if(w==1.0/3){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM03,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM03,FALSE,&mi);
			}
			if(w==1.0/4){
				mi.fState=MFS_CHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM04,FALSE,&mi);
			}else{
				mi.fState=MFS_UNCHECKED;
                SetMenuItemInfo(hMenu,IDM_ZOOM04,FALSE,&mi);
			}
			DrawMenuBar(hWnd);
			break;
		case WM_CLOSE:
			HeapFree(GetProcessHeap(),0,lpPixelShow);
			DestroyWindow(hWnd);
			break;
		case WM_DESTROY:
			PostQuitMessage(0);
			break;
		default:
			return DefWindowProc(hWnd,msg,wp,lp);
	}
	return 0;
}

LRESULT CALLBACK MyDlgProc(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp)
{
	static HWND hEdit,hStatic,hUpdown;
	static int nNo;
	char szNo[16];

	switch(msg){
		case WM_INITDIALOG:
			nNo=thresh;

			hEdit=GetDlgItem(hDlg,IDC_EDIT1);
			hStatic=GetDlgItem(hDlg,IDC_STATIC1);
			hUpdown=GetDlgItem(hDlg,IDC_SPIN1);

			SendMessage(hUpdown,UDM_SETBUDDY,(WPARAM)hEdit,0);
			SendMessage(hUpdown,UDM_SETRANGE32,(WPARAM)0,(LPARAM)255);
			SendMessage(hUpdown,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo,0));
			
			sprintf(szNo,"่‡’lF%d",nNo);
			SetWindowText(hStatic,szNo);
			return TRUE;
		case WM_COMMAND:
			switch(LOWORD(wp)){
				case IDOK:
					thresh=nNo;
					EndDialog(hDlg,IDOK);
					return TRUE;
				case IDC_CANCEL:
					EndDialog(hDlg,IDC_CANCEL);
					return TRUE;
			}
			if((HWND)lp==hEdit && HIWORD(wp)==EN_CHANGE){
				nNo=(int)SendMessage(hUpdown,UDM_GETPOS,0,0);
				sprintf(szNo,"่‡’lF%d",nNo);
				SetWindowText(hStatic,szNo);
				return TRUE;
			}
			return FALSE;
	}
	return FALSE;
}

LRESULT CALLBACK MyDlgProc2(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp)
{
	static HWND hEdit,hStatic,hUpdown;
	static int nNo;
	char szNo[32];

	switch(msg){
		case WM_INITDIALOG:
			nNo=amp;

			hEdit=GetDlgItem(hDlg,IDC_EDIT1);
			hStatic=GetDlgItem(hDlg,IDC_STATIC1);
			hUpdown=GetDlgItem(hDlg,IDC_SPIN1);

			SendMessage(hUpdown,UDM_SETBUDDY,(WPARAM)hEdit,0);
			SendMessage(hUpdown,UDM_SETRANGE32,(WPARAM)0,(LPARAM)2550);
			SendMessage(hUpdown,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo,0));
			
			sprintf(szNo,"ƒŒƒxƒ‹’ฒฎF%.1lf",(double)nNo/10);
			SetWindowText(hStatic,szNo);
			return TRUE;
		case WM_COMMAND:
			switch(LOWORD(wp)){
				case IDOK:
					amp=nNo;
					EndDialog(hDlg,IDOK);
					return TRUE;
				case IDC_CANCEL:
					EndDialog(hDlg,IDC_CANCEL);
					return TRUE;
			}
			if((HWND)lp==hEdit && HIWORD(wp)==EN_CHANGE){
				nNo=(int)SendMessage(hUpdown,UDM_GETPOS,0,0);
				sprintf(szNo,"ƒŒƒxƒ‹’ฒฎF%.1lf",(double)nNo/10);
				SetWindowText(hStatic,szNo);
				return TRUE;
			}
			return FALSE;
	}
	return FALSE;
}

LRESULT CALLBACK MyDlgProc3(HWND hDlg,UINT msg,WPARAM wp,LPARAM lp)
{
	static HWND hEdit1,hEdit2,hUpdown1,hUpdown2,hRadio1,hRadio2;
	static int nNo1,nNo2;

	switch(msg){
		case WM_INITDIALOG:
			nNo1=min;
			nNo2=max;

			hEdit1=GetDlgItem(hDlg,IDC_EDIT1);
			hEdit2=GetDlgItem(hDlg,IDC_EDIT2);
			hUpdown1=GetDlgItem(hDlg,IDC_SPIN1);
			hUpdown2=GetDlgItem(hDlg,IDC_SPIN2);
			hRadio1=GetDlgItem(hDlg,IDC_RADIO1);
			hRadio2=GetDlgItem(hDlg,IDC_RADIO2);

			SendMessage(hUpdown1,UDM_SETBUDDY,(WPARAM)hEdit1,0);
			SendMessage(hUpdown2,UDM_SETBUDDY,(WPARAM)hEdit2,0);
			switch(radio){
				case 1:
					SendMessage(hUpdown1,UDM_SETRANGE32,(WPARAM)0,(LPARAM)100);
					SendMessage(hUpdown2,UDM_SETRANGE32,(WPARAM)0,(LPARAM)100);
					break;
				case 2:
					SendMessage(hUpdown1,UDM_SETRANGE32,(WPARAM)0,(LPARAM)32767);
					SendMessage(hUpdown2,UDM_SETRANGE32,(WPARAM)0,(LPARAM)32767);
					break;
			}
			SendMessage(hUpdown1,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo1,0));
			SendMessage(hUpdown2,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo2,0));

			if(radio==1)
				Button_SetCheck(hRadio1,BST_CHECKED);
			else if(radio==2)
				Button_SetCheck(hRadio2,BST_CHECKED);

			return TRUE;
		case WM_COMMAND:
			switch(LOWORD(wp)){
				case IDOK:
					min=nNo1;
					max=nNo2;
					EndDialog(hDlg,IDOK);
					return TRUE;
				case IDCANCEL:
					EndDialog(hDlg,IDC_CANCEL);
					return TRUE;
				case IDC_RADIO1:
					radio=1;
					SendMessage(hUpdown1,UDM_SETRANGE32,(WPARAM)0,(LPARAM)100);
					SendMessage(hUpdown2,UDM_SETRANGE32,(WPARAM)0,(LPARAM)100);
					SendMessage(hUpdown1,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo1,0));
					SendMessage(hUpdown2,UDM_SETPOS,0,(LPARAM)MAKELONG((short)nNo2,0));
					return TRUE;
				case IDC_RADIO2:
					radio=2;
					SendMessage(hUpdown1,UDM_SETRANGE32,(WPARAM)0,(LPARAM)32767);
					SendMessage(hUpdown2,UDM_SETRANGE32,(WPARAM)0,(LPARAM)32767);
					return TRUE;
			}
			if((HWND)lp==hEdit1 && HIWORD(wp)==EN_CHANGE){
				nNo1=(int)SendMessage(hUpdown1,UDM_GETPOS,0,0);
				return TRUE;
			}
			if((HWND)lp==hEdit2 && HIWORD(wp)==EN_CHANGE){
				nNo2=(int)SendMessage(hUpdown2,UDM_GETPOS,0,0);
				return TRUE;
			}
			return FALSE;
	}
	return FALSE;
}