/*
      ビットマップの色数２

     1999/ 7/19  宍戸　輝光
*/

#include <windows.h>

LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM);
BOOL readBMP(char fn[]);
DWORD count(void);
HINSTANCE hInst;
HWND hwnd;
DWORD dwWidth,dwHeight,dwLength;
BOOL loaded;
LPBYTE lpBuf,lpBMP;
LPBITMAPINFO lpInfo;

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);

	hwnd = CreateWindow ("CWindow","ビットマップの色数計算",
	  WS_OVERLAPPEDWINDOW,CW_USEDEFAULT,CW_USEDEFAULT,560,480,
      NULL,NULL,hInstance,NULL);

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

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

	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;

	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);

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

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

			break;

		case WM_COMMAND:

			switch (LOWORD(wParam)) {

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

					EnableWindow(hwCountB,readBMP(lpszFn));

					break;

			}

			break;

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

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

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

			EnableWindow(hwCountB,readBMP(lpszFn));

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

			break;

		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);

			break;

		case WM_DESTROY :

			if (loaded)
				GlobalFree(lpBuf);

			PostQuitMessage(0);
			break;

	}

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

}

BOOL readBMP(char 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(hwnd,"Error!!");

		return FALSE;

	}

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

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

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

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

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

	InvalidateRgn(hwnd,NULL,FALSE);
	UpdateWindow (hwnd);             /* 再描画 */

	return TRUE;

}

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

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

	fColors=GlobalAlloc(GPTR,256*256*32);

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

			/* (j,i)の色をdwPixelに３２ビットで格納 */
			CopyMemory(&dwPixel,lpBMP+j*3+i*dwLength,3);

			/* dwPixelビット目の位置を計算 */
			dwByte=dwPixel/8;
			dwBit=dwPixel%8;

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

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

			}

	}

	GlobalFree(fColors);

	return dwCount;

}