?? 堆棧的數(shù)據(jù)結構和操作.txt
字號:
//堆棧結構
typedef struct{
HGLOBAL hMem; //堆棧全局內(nèi)存句柄
POINT *lpMyStack; //指向該句柄的指針
LONG ElementsNum; //堆棧的大小
LONG ptr; //指向棧頂?shù)闹羔?}MYSTACK;
//初始化堆棧的操作,第二個參數(shù)指定堆棧的大小
BOOL InitStack(HWND hWnd,LONG StackLen)
{
SeedFillStack.ElementsNum=StackLen; //將堆棧的大小賦值
if((SeedFillStack.hMem=GlobalAlloc(GHND,SeedFillStack.
ElementsNum*sizeof(POINT)))==NULL)
{
//內(nèi)存分配錯誤,返回FALSE;
MessageBox(hWnd,"Error alloc memory!","ErrorMessage",MB_OK|
MB_ICONEXCLAMATION);
return FALSE;
}
SeedFillStack.lpMyStack=(POINT *)GlobalLock(SeedFillStack.hMem);
//緩沖區(qū)全部清零
memset(SeedFillStack.lpMyStack,0,SeedFillStack.ElementsNum*sizeof(POINT));
//堆頂指針為零
SeedFillStack.ptr=0;
//成功,返回TRUE
return TRUE;
}
//析構函數(shù)
void DeInitStack()
{
//釋放內(nèi)存,重置堆棧大小及棧頂指針。
GlobalUnlock(SeedFillStack.hMem);
GlobalFree(SeedFillStack.hMem);
SeedFillStack.ElementsNum=0;
SeedFillStack.ptr=0;
}
//push操作
BOOL MyPush(POINT p)
{
POINT *TempPtr;
if(SeedFillStack.ptr>=SeedFillStack.ElementsNum)
return FALSE; //棧已滿,返回FALSE
//進棧,棧頂指針加1
TempPtr=(POINT *)(SeedFillStack.lpMyStack+SeedFillStack.ptr++);
(*TempPtr).x=p.x;
(*TempPtr).y=p.y;
return TRUE;
}
//pop操作
POINT MyPop()
{
POINT InvalidP;
InvalidP.x=-1;
InvalidP.y=-1;
if(SeedFillStack.ptr<=0)
return InvalidP; //棧為空,返回無效點
SeedFillStack.ptr--; //棧頂指針減1
//返回棧頂點
return *(SeedFillStack.lpMyStack+SeedFillStack.ptr);
}
//判斷堆棧是否為空
BOOL IsStackEmpty()
{
return (SeedFillStack.ptr==0)?TRUE:FALSE;
}
如果讀者對堆棧的概念還不清楚,請參閱有關數(shù)據(jù)結構方面的書籍,這里就不詳述了。
要注意的是:1. 要填充的區(qū)域是封閉的;2. 我們處理的雖然是二值圖,但實際上是256級灰度圖,不過只用到了0和255兩種顏色;3.在菜單中選擇種子填充命令時,提示用戶用鼠標點取一個要填充區(qū)域中的點,處理是在WM_LBUTTONDOWN中。
MYSTACK SeedFillStack;
BOOL SeedFill(HWND hWnd)
{
DWORD BufSize;
LPBITMAPINFOHEADER lpImgData;
HLOCAL hTempImgData;
LPBITMAPINFOHEADER lpTempImgData;
LPSTR lpTempPtr,lpTempPtr1;
HDC hDc;
HFILE hf;
POINT CurP,NeighborP;
//我們處理的實際上是256級灰度圖,不過只用到了0
和255兩種顏色。
if( NumColors!=256){
MessageBox(hWnd,"Must be a mono bitmap with grayscale palette!",
"Error Message",MB_OK|MB_ICONEXCLAMATION);
return FALSE;
}
//原圖緩沖區(qū)的大小
BufSize=bf.bfSize-sizeof(BITMAPFILEHEADER);
//為新圖緩沖區(qū)分配內(nèi)存
if((hTempImgData=LocalAlloc(LHND,BufSize))==NULL)
{
MessageBox(hWnd,"Error alloc memory!","Error Message",MB_OK|
MB_ICONEXCLAMATION);
return FALSE;
}
lpImgData=(LPBITMAPINFOHEADER)GlobalLock(hImgData);
lpTempImgData=(LPBITMAPINFOHEADER)LocalLock(hTempImgData);
//拷貝頭信息和位圖數(shù)據(jù)
memcpy(lpTempImgData,lpImgData,BufSize);
if(!InitStack(hWnd,(LONG)bi.biHeight*bi.biWidth)){ //初始化堆棧
//若失敗,釋放內(nèi)存,返回
LocalUnlock(hTempImgData);
LocalFree(hTempImgData);
GlobalUnlock(hImgData);
return FALSE;
}
lpTempPtr=(char *)lpTempImgData+(BufSize-LineBytes-
SeedPoint.y*LineBytes)+SeedPoint.x;
if(*lpTempPtr==0){
//鼠標點到了黑點上,提示用戶不能選擇邊界上的點,返回FALSE
MessageBox(hWnd,"The point you select is a contour point!","Error Message",MB_OK|MB_ICONEXCLAMATION);
LocalUnlock(hTempImgData);
LocalFree(hTempImgData);
GlobalUnlock(hImgData);
DeInitStack();
return FALSE;
}
//push該點(用戶用鼠標選擇的,處理是在
WM_LBUTTONDOWN中
MyPush(SeedPoint);
while(!IsStackEmpty()) //堆棧不空則一直處理
{
CurP=MyPop(); //pop棧頂?shù)狞c
lpTempPtr=(char *)lpTempImgData+(BufSize-
LineBytes-CurP.y*LineBytes)+CurP.x;
//將該點涂黑
*lpTempPtr=(unsigned char)0;
//左鄰點
if(CurP.x>0) //注意判斷邊界
{
NeighborP.x=CurP.x-1;
NeighborP.y=CurP.y;
lpTempPtr1=lpTempPtr-1;
if(*lpTempPtr1!=0) //如果為白,表示還沒有填,進棧
MyPush(NeighborP);
}
//上鄰點
if(CurP.y>0) //注意判斷邊界
{
NeighborP.x=CurP.x;
NeighborP.y=CurP.y-1;
lpTempPtr1=lpTempPtr+LineBytes;
if(*lpTempPtr1!=0) //如果為白,表示還沒有填,進棧
MyPush(NeighborP);
}
//右鄰點
if(CurP.x<bi.biWidth-1) //注意判斷邊界
{
NeighborP.x=CurP.x+1;
NeighborP.y=CurP.y;
lpTempPtr1=lpTempPtr+1;
if(*lpTempPtr1!=0) //如果為白,表示還沒有填,進棧
MyPush(NeighborP);
}
//下鄰點
if(CurP.y<bi.biHeight-1) //注意判斷邊界
{
NeighborP.x=CurP.x;
NeighborP.y=CurP.y+1;
lpTempPtr1=lpTempPtr-LineBytes;
if(*lpTempPtr1!=0) //如果為白,表示還沒有填,進棧
MyPush(NeighborP);
}
}
//析構堆棧,釋放內(nèi)存
DeInitStack();
if(hBitmap!=NULL)
DeleteObject(hBitmap);
hDc=GetDC(hWnd);
//創(chuàng)建新的位圖
hBitmap=CreateDIBitmap(hDc,(LPBITMAPINFOHEADER)lpTempImgData,
(LONG)CBM_INIT,(LPSTR)lpTempImgData+sizeof(BITMAPINFOHEADER) +
NumColors*sizeof(RGBQUAD), (LPBITMAPINFO)lpTempImgData, DIB_RGB_COLORS);
hf=_lcreat("c:\\seed.bmp",0);
_lwrite(hf,(LPSTR)&bf,sizeof(BITMAPFILEHEADER));
_lwrite(hf,(LPSTR)lpTempImgData,BufSize);
_lclose(hf);
//釋放內(nèi)存和資源
ReleaseDC(hWnd,hDc);
LocalUnlock(hTempImgData);
LocalFree(hTempImgData);
GlobalUnlock(hImgData);
return TRUE;
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -