/*
        色の度数分布

    2000/ 8/30  宍戸　輝光
*/

#include <windows.h>

#define WM_ENDTHREAD (WM_USER+1)

LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM);
BOOL readBMP(LPCTSTR);
DWORD count(void);
DWORD WINAPI dosuu(LPVOID);
HINSTANCE hInst;
HWND hwMain;
DWORD dwWidth,dwHeight,dwLength;
BOOL loaded;
LPBYTE lpBuf,lpBMP;
LPBITMAPINFO lpInfo;
volatile boolean fGo;

int WINAPI WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance,
                    PSTR szCmdLine, int iCmdShow){

	MSG  msg;
	WNDCLASSEX  wndclass ;

	hInst=hInstance; /* プロセスのハンドルを保存 */

	wndclass.cbSize        = sizeof(wndclass);
	wndclass.style         = CS_HREDRAW | CS_VREDRAW;
	wndclass.lpfnWndProc   = WndProc;
	wndclass.cbClsExtra    = 0;
	wndclass.cbWndExtra    = 0;
	wndclass.hInstance     = hInstance;
	wndclass.hIcon         = LoadIcon (NULL, IDI_APPLICATION);
	wndclass.hCursor       = LoadCursor (NULL, IDC_ARROW);
	wndclass.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
	wndclass.lpszMenuName  = NULL;
	wndclass.lpszClassName = "CWindow";
	wndclass.hIconSm       = LoadIcon (NULL, IDI_APPLICATION);

	RegisterClassEx (&wndclass);

	hwMain = CreateWindow ("CWindow","色の度数分布表",
	  WS_OVERLAPPEDWINDOW,CW_USEDEFAULT,CW_USEDEFAULT,560,480,
      NULL,NULL,hInstance,NULL);

	DragAcceptFiles(hwMain,TRUE); /* ドラッグ＆ドロップ受入 */

	ShowWindow (hwMain,iCmdShow); /* ウインドウを表示 */
	UpdateWindow (hwMain);        /* 再描画 */

	while (GetMessage (&msg,NULL,0,0)) { /* メッセージループ */

		TranslateMessage(&msg);
		DispatchMessage(&msg);

	}

	return msg.wParam ;

}

LRESULT CALLBACK WndProc (HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam) {

	static OPENFILENAME ofn;
	static char lpszFn[256],lpszFt[256],lpszStr[64];
	static HWND hwOpenB,hwCountB;
	static int d;
	static HANDLE hThread;

	HDROP hDrop;
	HDC hdc;
	PAINTSTRUCT ps;

	/* 「ファイルを開く」ダイアログ設定 */
	ofn.lStructSize=sizeof(OPENFILENAME);
	ofn.hwndOwner=hwnd;
	ofn.hInstance=NULL;
	ofn.lpstrFilter= // フィルター
	  "ビットマップ(*.bmp)\0*.bmp\0すべてのファイル(*.*)\0*.*\0\0";
	ofn.lpstrCustomFilter=NULL;
	ofn.nMaxCustFilter=0;
	ofn.nFilterIndex=1; /* フィルターの初期位置 */
	ofn.lpstrFile=lpszFn; /* ファイル名用文字列バッファ */
	ofn.nMaxFile=255;  /* 文字列バッファのサイズ */
	ofn.lpstrFileTitle=lpszFt; /* タイトル用文字列バッファ */
	ofn.nMaxFileTitle=255; /* 文字列バッファのサイズ */
	ofn.lpstrInitialDir=NULL;
	ofn.lpstrTitle="読み込むファイルを選んでください"; /* タイトル */
	ofn.Flags=OFN_FILEMUSTEXIST;
	ofn.nFileOffset=0;
	ofn.nFileExtension=0;
	ofn.lpstrDefExt=NULL;
	ofn.lCustData=0;
	ofn.lpfnHook=NULL;
	ofn.lpTemplateName=NULL;
	
	switch (iMsg) {

		case WM_CREATE:

			hwOpenB=CreateWindow("Button","読み込み", /* ボタン作成 */
				WS_CHILD | WS_VISIBLE,16,8,96,32,hwnd,
				(HMENU)0,hInst,NULL);

			hwCountB=CreateWindow("Button","度数表作成", /* ボタン作成 */
				WS_CHILD | WS_VISIBLE,128,8,96,32,hwnd,
				(HMENU)1,hInst,NULL);

			loaded=FALSE; /* 読み込みフラグリセット */
			fGo=FALSE;

			EnableWindow(hwCountB,FALSE); /* 「色数測定」ボタン無効 */

			return 0;

		case WM_COMMAND:

			switch (LOWORD(wParam)) {

				case 0: /* 読み込みボタン */
	
					if (!GetOpenFileName(&ofn)) /* ファイルを選択 */
						return 0;

					EnableWindow(hwCountB,readBMP(lpszFn));

					return 0;

				case 1: /* 度数表作成ボタン */
	
					if (fGo)
						fGo=FALSE;
					else {

						EnableWindow(hwOpenB,FALSE);
						SetWindowText(hwCountB,"中止");
						fGo=TRUE;
						hThread=CreateThread(NULL,0,dosuu,0,0,&d);

					}

					break;

			}

			return 0;

		case WM_DROPFILES: /* ファイルがドロップされた時の処理 */

			hDrop=(HDROP)wParam; /* HDROPを取得 */

			DragQueryFile(hDrop,0,lpszFn,256); /* ファイル名を取得 */

			EnableWindow(hwCountB,readBMP(lpszFn));

			DragFinish(hDrop); /* 終了処理 */

			return 0;

		case WM_PAINT:

			hdc=BeginPaint(hwnd,&ps);

			if (loaded) /* ビットマップが読み込まれていれば */
				StretchDIBits(hdc,4,48,dwWidth,dwHeight,
				  0,0,dwWidth,dwHeight,lpBMP,lpInfo,
				  DIB_RGB_COLORS,SRCCOPY); /* DIBを画面に描画 */

			EndPaint(hwnd,&ps);

			return 0;

		case WM_ENDTHREAD:

			EnableWindow(hwOpenB,TRUE);
			SetWindowText(hwCountB,"度数表作成");
			fGo=FALSE;

			return 0;

		case WM_DESTROY :

			if (loaded)
				GlobalFree(lpBuf);

			PostQuitMessage(0);

			return 0;

	}

	return DefWindowProc (hwnd, iMsg, wParam, lParam) ;

}

BOOL readBMP(LPCTSTR fn) { /* ビットマップ読み込み */

	char lpStr[280];
	DWORD offset,dummy;

	HANDLE fh;

	if (loaded) /* 以前確保したバッファを解放 */
		GlobalFree(lpBuf);

	fh=CreateFile(fn,GENERIC_READ,0,NULL,OPEN_EXISTING,
		FILE_ATTRIBUTE_NORMAL,NULL); /* ファイルオープン */

	lpBuf=(LPBYTE)GlobalAlloc
	    (GPTR,GetFileSize(fh,NULL)); /* バッファ確保 */

	ReadFile(fh,lpBuf,GetFileSize(fh,NULL),&dummy,NULL);

	lpInfo=(LPBITMAPINFO)(lpBuf+sizeof(BITMAPFILEHEADER));

	offset=*(LPDWORD)(lpBuf+10);
	lpBMP=lpBuf+offset; /* ビットマップバッファの先頭アドレス */

	CloseHandle(fh);

	/* ２４ビットフルカラービットマップでなければ無効 */
	if (lpBuf[0]!='B' || lpInfo->bmiHeader.biBitCount!=24) {

		GlobalFree(lpBuf);
		loaded=FALSE;

		SetWindowText(hwMain,"Error!!");

		return FALSE;

	}

	/* ビットマップの大きさ保存 */
	dwWidth=lpInfo->bmiHeader.biWidth;
	dwHeight=lpInfo->bmiHeader.biHeight;

	if (dwWidth % 4==0) /* バッファの１ラインの長さを計算 */
		dwLength=dwWidth*3;
	else
		dwLength=dwWidth*3+(4-(dwWidth*3) % 4);

	loaded=TRUE; /* 読み込みフラグセット */

	InvalidateRgn(hwMain,NULL,TRUE);
	UpdateWindow (hwMain);             /* 再描画 */

	wsprintf(lpStr,"%s %d*%dPixel %dBit %d色",fn,dwWidth,dwHeight,
	  lpInfo->bmiHeader.biBitCount,count());

	SetWindowText(hwMain,lpStr); /* ビットマップの情報表示 */

	return TRUE;

}

DWORD count(void) { /* ビットマップ内の色数を数える */

	DWORD i,j,dwBit,dwByte,dwCount=0,dwPixel=0,dwR,dwG,dwB;
	LPBYTE fColors;

	fColors=GlobalAlloc(GPTR,256*256*256/8);

	for (i=0;i<dwHeight;i++)
		for (j=0;j<dwWidth;j++) {

			/* (j,i)の色をdwPixelに３２ビットで格納 */
			dwR=lpBMP[j*3+i*dwLength];
			dwG=lpBMP[j*3+i*dwLength+1];
			dwB=lpBMP[j*3+i*dwLength+2];
			dwPixel=(dwR << 16)+(dwG << 8)+dwB;
				
			/* dwPixelビット目の位置を計算 */
			dwByte=dwPixel/8;
			dwBit=dwPixel%8;

			/* フラグ領域のdwPixelビット目を検査 */
			if ((fColors[dwByte] & (1 << dwBit))==0) {

				/* フラグ領域のdwPixelビット目を立てる */
				fColors[dwByte]=fColors[dwByte]|(1 << dwBit);
				dwCount++; /* 色数カウンタ更新 */

			}

	}

	return dwCount;

}

DWORD WINAPI dosuu(LPVOID lpV) { /* ビットマップ内の色数を数える */

	typedef struct {

		DWORD dwRGB,dwNum;

	} sRGB;

	sRGB wData,*lpData;
	DWORD i,j,k,dwBit,dwByte,dwCount=0,dwPixel=0,dwR,dwG,dwB;
	LPBYTE fColors;
	LPTSTR wrk,str;

	fColors=GlobalAlloc(GPTR,256*256*256/8);

	/* ビットマップ内の色数カウント＆色フラグ作成 */
	for (i=0;i<dwHeight && fGo;i++)
		for (j=0;j<dwWidth && fGo;j++) {

			/* (j,i)の色をdwPixelに３２ビットで格納 */
			dwR=lpBMP[j*3+i*dwLength+2];
			dwG=lpBMP[j*3+i*dwLength+1];
			dwB=lpBMP[j*3+i*dwLength];
			dwPixel=(dwR << 16)+(dwG << 8)+dwB;
				
			/* dwPixelビット目の位置を計算 */
			dwByte=dwPixel/8;
			dwBit=dwPixel%8;

			/* フラグ領域のdwPixelビット目を検査 */
			if ((fColors[dwByte] & (1 << dwBit))==0) {

				/* フラグ領域のdwPixelビット目を立てる */
				fColors[dwByte]=fColors[dwByte]|(1 << dwBit);
				dwCount++; /* 色数カウンタ更新 */

			}

			if (!fGo)
				break;

	}

	lpData=(sRGB*)GlobalAlloc(GPTR,dwCount*sizeof(sRGB));

	j=0;

	/* ビットマップ内の色をdwRGBに記録 */
	for (i=0;(i<256*256*256) && fGo;i++)
		if((fColors[i/8] & (1 << (i % 8)))!=0)
			lpData[j++].dwRGB=i;

	for (i=0;i<dwHeight && fGo;i++)
		for (j=0;j<dwWidth && fGo;j++) {

			/* (j,i)の色をdwPixelに３２ビットで格納 */
			dwR=lpBMP[j*3+i*dwLength+2];
			dwG=lpBMP[j*3+i*dwLength+1];
			dwB=lpBMP[j*3+i*dwLength];
			dwPixel=(dwR << 16)+(dwG << 8)+dwB;
			
			for (k=0;k<dwCount;k++)
				if (dwPixel==lpData[k].dwRGB) {

					lpData[k].dwNum++;
					break;

				}

		}

		for (i=0;i<dwCount && fGo;i++)
			for (j=i+1;j<dwCount && fGo;j++)
				if (lpData[j].dwNum>lpData[i].dwNum) {

					wData=lpData[j];
					lpData[j]=lpData[i];
					lpData[i]=wData;

				}

	wrk=(LPTSTR)GlobalAlloc(GPTR,sizeof(TCHAR)*360);
	str=wrk+40;

	if (fGo) {

		for (i=0;i<8;i++)
			if (i<dwCount) {

				wsprintf(wrk,"%08x-%4d\n",lpData[i].dwRGB,lpData[i].dwNum);
				lstrcat(str,wrk);

			}

		MessageBox(NULL,str,"色の度数分布上位８色",MB_OK);

	}

	GlobalFree(fColors);
	GlobalFree(lpData);
	GlobalFree(wrk);

	PostMessage(hwMain,WM_ENDTHREAD,0,0);

	ExitThread(0);

	return 0;

}