// ■ハノイの塔問題(改良マルチスレッド版)
//   描画用スレッドを生成せず，メインスレッドのタイマー処理で描画
//
#include "myWin.h"
#include "math.h"
#include "process.h"
#define RED(X)   ( X      & 0xFF)
#define GREEN(X) ((X>> 8) & 0xFF)
#define BLUE(X)  ((X>>16) & 0xFF)
#define MAX 5      //円盤の数
#define SLEEP 100  //Sleepする時間（円盤の動く速度を変えるときはこれを変更する）
static int WFlag, ExeThread=false;
static int colDisk[3][MAX], numDisk[3];
static int Nst;
static HDC hBuff; static HBITMAP hBM;
static int diskCtab[]={
	0xFF77FF, 0x00FFFF, 0x00FF00, 0x0000FF, 0xFF5555,
	0xFFFF00, 0x7777FF, 0xFFFF77, 0x77FF77, 0xFF00FF};
int colorBright(int color, double BR){
	if(BR>1)BR=1;
	int R = (int)(BR*RED(color)), G=(int)(BR*GREEN(color)),B=(int)(BR*BLUE(color));
	return RGB(R,G,B);
}
void drawLine(int X1, int Y1, int X2, int Y2, int color){//線表示
	HPEN pen=CreatePen(PS_SOLID, 1, color); SelectObject(hBuff, pen);
	MoveToEx(hBuff,X1,Y1,NULL); LineTo(hBuff,X2,Y2);
	DeleteObject(pen);
}
void drawRectangle(int X1, int Y1, int X2, int Y2, int color){//床表示
	HPEN pen=CreatePen(PS_SOLID, 1, 0); SelectObject(hBuff, pen);
	HBRUSH brush=CreateSolidBrush(color); SelectObject(hBuff, brush);
	Rectangle(hBuff,X1,Y1,X2,Y2);
	DeleteObject(brush);DeleteObject(pen);
}
void drawDisk(int X0, int Y0, int R, int TH, int color){        //円盤または柱の
	double DR = R, RR =DR*DR; int X1=X0-R, X2=X0+R, Y2=Y0-TH;   //グラデーション付き
	for(int X=X1; X<=X2; X++){									//の四角形を描く
		double XX=X-X0;double Bright=sqrt((double)(RR-XX*XX))/R;
		drawLine(X, Y0, X, Y0 - TH, colorBright(color, Bright));
	}
	drawLine(X1,Y0, X1,Y2, 0);drawLine(X1,Y2, X2,Y2, 0);//円盤または柱の枠表示
	drawLine(X2,Y2, X2,Y0, 0);drawLine(X2,Y0, X1,Y0, 0);
}
void crearBitmap(){//描画用ビットマップの準備
	SelectObject(hBuff,GetStockObject(NULL_PEN));
	PatBlt(hBuff,0, 0,1000,400,WHITENESS);
}
void drawState(){
	int DX=Nst*20,XST=Nst*15, poleH=10*Nst+10, Xed=70*Nst, Yed=20*Nst+40;
	crearBitmap();	drawRectangle(5, Yed, Xed, Yed+20, 0x5577);      //床描画
	for(int i=0; i<3;i++) drawDisk(i*DX+XST, Yed,3, poleH, 0x225577);//３本の柱描画
	for(int i=0;i<3;i++)for(int j=0; j<numDisk[i];j++){              //円盤を描画
		int N=colDisk[i][j];
		drawDisk(i*DX+XST, Yed-j*10, 10*N, 10, diskCtab[N%10]);
	}
}
void initState(int N){
	for(int i=0;i<N;i++) colDisk[0][i] = N - i;//左の柱に円盤を揃える
	numDisk[0]=Nst=N; numDisk[1]=numDisk[2]=0; //中央，右の柱の円盤 0 枚
}
void moveTo(int n, int x, int y){//円盤の移動(move n-th disk from x to y)
	numDisk[x]--; colDisk[y][numDisk[y]++]=n;WFlag=true;
	while(WFlag) Sleep(10);//描画が終わるまで待つ
}
void hanoi(int n, int x, int y){//ハノイの塔の解を求める
	if(n<=0)return;
	hanoi( n-1,x, 3-x-y); moveTo( n,x,y); hanoi( n-1,3-x-y,y);
}
void hanoiThread(void *a){//ハノイの塔用スレッド
	int N=(int)a; hanoi(N,0, 2); ExeThread=false;
}
void procPaint(HWND hw, WPARAM wp,LPARAM lp){//ペイント処理
	PAINTSTRUCT ps;	HDC hdc=BeginPaint(hw, &ps);
	BitBlt(hdc, 0,0,1000,400,hBuff,0,0,SRCCOPY);
	EndPaint(hw,&ps); 
}
void start(HWND hw){//開始処理
	MoveWindow(hw,0,0,80*MAX,20*MAX+150,TRUE);
	WFlag=false;initState(MAX);	drawState();
	InvalidateRect(hw,NULL,TRUE);
}
void procTimer(HWND hw, WPARAM wp,LPARAM lp){//タイマー処理(描画)
	if(!ExeThread){KillTimer(hw,1);return;}
	while(!WFlag && ExeThread) Sleep(10);//円盤が動く／ハノイの塔の処理が終わるまで
	if(!ExeThread) return ;				 //ハノイの塔の処理が終わっていれば描画終了
	drawState(); InvalidateRect(hw,NULL,TRUE);
	WFlag=false;
}
void procLButtonDown(HWND hw, WPARAM wp,LPARAM lp){//左ボタンプッシュで開始
	if(ExeThread) return;
	DWORD dwID;	int N=MAX; start(hw);ExeThread=true;
	HANDLE H1=(HANDLE) _beginthread( hanoiThread, 0, (void *)N);
	SetTimer(hw,1, SLEEP,NULL);
}
void initBitmap(HWND hw){
	HDC hdc=GetDC(hw); hBM=CreateCompatibleBitmap(hdc,1000,400);
	hBuff=CreateCompatibleDC(hdc); SelectObject(hBuff,hBM);
	crearBitmap();
}
void procCreate(HWND hw, WPARAM wp,LPARAM lp){
	SetWindowText(hw,TEXT("ハノイの塔"));
	MoveWindow(hw,0,0,800,300,TRUE);
	initBitmap(hw);	start(hw);
}
LRESULT CALLBACK WndProc(HWND hw, UINT msg, WPARAM wp,LPARAM lp){
	switch(msg){
	case WM_DESTROY       : PostQuitMessage(0)         ; return 0;
	case WM_CREATE        : procCreate(hw,wp,lp)       ; return 0;
	case WM_TIMER         : procTimer(hw,wp,lp)        ; return 0;
	case WM_PAINT         : procPaint(hw,wp,lp)        ; return 0;
	case WM_LBUTTONDOWN   : procLButtonDown(hw,wp,lp)  ; break;
	}
	return DefWindowProc(hw,msg,wp,lp);
}