abrasionmeasurementview.cpp

钢轨磨耗测量系统

   	
   	// 图像在计算机在存储中的实际大小
   	CSize sizeImageSave	= pDib->GetDibSaveDim();
   	
   	// 图像在内存中每一行象素占用的实际空间
   	int nSaveWidth = sizeImageSave.cx;
   	
   	// 应用Sobel算子求梯度
   	SobelOperator(pDib, pdGrad);
   	
   	
   	for(y=0; y<nHeight ; y++ )
   		for(x=0 ; x<nWidth ; x++ )
   		{
 			if(*(pdGrad+y*nWidth+x)>50)
 				*(	lpImage+y*nSaveWidth+x	)=0;
 			else
 				*(	lpImage+y*nSaveWidth+x	)=255;			
 		}
 		
   //释放内存空间
 	delete []pdGrad;
 	pdGrad=NULL;
 
 	// 恢复光标形状
 	EndWaitCursor();  	
 	// 设置脏标记
 	pDoc->SetModifiedFlag(TRUE);
 	
 	// 更新视图
 	pDoc->UpdateAllViews(NULL);
}

   
/*************************************************************************
*
* \函数名称：
*   OnEdgePrewitt()
*
* \输入参数:
*   无
*
* \返回值:
*   无
*
* \说明:
*   实现Prewitt算子
*
************************************************************************
*/
   void CAbrasionMeasurementView::OnEdgePrewitt() 
   {
   	// TODO: Add your command handler code here
   	
   	//更改光标形状
   	BeginWaitCursor(); 
   
   	// 循环控制变量
   	int y;
   	int x;
   	
   	CAbrasionMeasurementDoc * pDoc = (CAbrasionMeasurementDoc *)this->GetDocument();
	pDoc->OnFileReload1();//reload 提取前图像

   	CDib * pDib = pDoc->m_pDibInit;
   
   	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;
   	
   	// 判断是否是8-bpp位图
   	if (lpBMIH->biBitCount != 8)
   	{
   		// 提示用户
   		MessageBox("目前只支持256色位图的图象分割！", "系统提示" ,
   			MB_ICONINFORMATION | MB_OK);
   		
   		// 返回
   		return;
   	}
   	
   	// 图象的长宽大小
   	CSize sizeImage		= pDib->GetDimensions();
   	int nWidth			= sizeImage.cx		;
   	int nHeight			= sizeImage.cy		;
   	
   	// 指向梯度数据的指针
   	double * pdGrad;
   	
   	// 按照图像的大小开辟内存空间，存储梯度计算的结果
   	pdGrad=new double[nHeight*nWidth];
   	
   	//图像数据的指针
   	LPBYTE  lpImage = pDib->m_lpImage;
   	
   	// 图像在计算机在存储中的实际大小
   	CSize sizeImageSave	= pDib->GetDibSaveDim();
   	
   	// 图像在内存中每一行象素占用的实际空间
   	int nSaveWidth = sizeImageSave.cx;
   	
   	// 应用Prewitt算子求梯度
   	PrewittOperator(pDib, pdGrad);
   	
   	
   	for(y=0; y<nHeight ; y++ )
   		for(x=0 ; x<nWidth ; x++ )
   		{
   			if(*(pdGrad+y*nWidth+x)>50)
   				*(	lpImage+y*nSaveWidth+x	)=0;
   			else
   				*(	lpImage+y*nSaveWidth+x	)=255;			
   		}
   	
     //释放内存空间
   	delete []pdGrad;
   	pdGrad=NULL;
   	
   	// 恢复光标形状
   	EndWaitCursor(); 
   	
   	// 设置脏标记
   	pDoc->SetModifiedFlag(TRUE);
   	
   	// 更新视图
   	pDoc->UpdateAllViews(NULL);
   		
   }

/*************************************************************************
*
* \函数名称：
*   OnEdgeLaplace()
*
* \输入参数:
*   无
*
* \返回值:
*   无
*
* \说明:
*   实现拉普拉斯算子
*
************************************************************************
*/
   void CAbrasionMeasurementView::OnEdgeLaplace() 
   {
   	// TODO: Add your command handler code here
   	
   	//更改光标形状
   	BeginWaitCursor(); 
   	
   	// 循环控制变量
   	int y;
   	int x;
   	
   	CAbrasionMeasurementDoc * pDoc = (CAbrasionMeasurementDoc *)this->GetDocument();
	pDoc->OnFileReload1();//reload 提取前图像

   	CDib * pDib = pDoc->m_pDibInit;
   	
   	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;
   	
   	// 判断是否是8-bpp位图
   	if (lpBMIH->biBitCount != 8)
   	{
   		// 提示用户
   		MessageBox("目前只支持256色位图的图象分割！", "系统提示" ,
   			MB_ICONINFORMATION | MB_OK);
   		
   		// 返回
   		return;
   	}
   	
   	// 图象的长宽大小
   	CSize sizeImage		= pDib->GetDimensions();
   	int nWidth			= sizeImage.cx		;
   	int nHeight			= sizeImage.cy		;
   	
   	// 指向梯度数据的指针
   	double * pdGrad;
   	
   	// 按照图像的大小开辟内存空间，存储梯度计算的结果
   	pdGrad=new double[nHeight*nWidth];
   	
   	//图像数据的指针
   	LPBYTE  lpImage = pDib->m_lpImage;
   	
   	// 图像在计算机在存储中的实际大小
   	CSize sizeImageSave	= pDib->GetDibSaveDim();
   	
   	// 图像在内存中每一行象素占用的实际空间
   	int nSaveWidth = sizeImageSave.cx;
   	
   	// 应用Laplace算子求梯度
   	LaplacianOperator(pDib, pdGrad);
   	
   	
   	for(y=0; y<nHeight ; y++ )
   		for(x=0 ; x<nWidth ; x++ )
   		{
   			if(*(pdGrad+y*nWidth+x)>50)
   				*(	lpImage+y*nSaveWidth+x	)=0;
   			else
   				*(	lpImage+y*nSaveWidth+x	)=255;			
   		}
   	
   	//释放内存空间
   	delete []pdGrad;
   	pdGrad=NULL;
   	
   	// 恢复光标形状
   	EndWaitCursor(); 
   	
   	// 设置脏标记
   	pDoc->SetModifiedFlag(TRUE);
   	
   	// 更新视图
   	pDoc->UpdateAllViews(NULL);
   	
   }

/*************************************************************************
*
* \函数名称：
*   OnEdgeCanny()
*
* \输入参数:
*   无
*
* \返回值:
*   无
*
* \说明:
*   实现Canny算子
*
************************************************************************
*/






void CAbrasionMeasurementView::OnCanny()
{
	// TODO: 在此添加命令处理程序代码
//更改光标形状
	BeginWaitCursor(); 

	CAbrasionMeasurementDoc * pDoc = (CAbrasionMeasurementDoc *)this->GetDocument();
	pDoc->OnFileReload1();//reload 提取前图像

	CDib * pDib = pDoc->m_pDibInit;
	
	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;
	
	// 判断是否是8-bpp位图
	if (lpBMIH->biBitCount != 8)
	{
		// 提示用户
		MessageBox("目前只支持256色位图的图象分割！", "系统提示" ,
			MB_ICONINFORMATION | MB_OK);
		
		// 返回
		return;
	}

	// 循环控制变量
	int y; 
	int x;
	
	CSize sizeImage = pDib->GetDimensions();
	int nWidth = sizeImage.cx ;
	int nHeight= sizeImage.cy ;

	int nSaveWidth = pDib->GetDibSaveDim().cx;

	// 开辟内存，存储图象数据
	unsigned char * pUnchImage = new unsigned char[nWidth*nHeight];

	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			pUnchImage[y*nWidth+x] = pDib->m_lpImage[y*nSaveWidth+x];
		}
	}
	
	// canny算子计算后的结果
	unsigned char * pUnchEdge = new unsigned char[nWidth*nHeight];
	
	// 调用canny函数进行边界提取
	Canny(pUnchImage, nWidth, nHeight, 0.4, 0.4, 0.79, pUnchEdge) ;
	
	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			pDib->m_lpImage[y*nWidth+x]=(unsigned char)(255-pUnchEdge[y*nWidth+x]);
		}
	}

	delete []pUnchImage;
	pUnchImage = NULL  ;
	delete []pUnchEdge ;
	pUnchEdge = NULL   ;
	
	// 恢复光标形状
	EndWaitCursor(); 
	
	// 设置脏标记
	pDoc->SetModifiedFlag(TRUE);
	
	// 更新视图
	pDoc->UpdateAllViews(NULL);


}


//细化算法
void CAbrasionMeasurementView::OnMicro()
{
	// TODO: 在此添加命令处理程序代码

	//更改光标形状
	BeginWaitCursor(); 
	CAbrasionMeasurementDoc * pDoc=(CAbrasionMeasurementDoc *)this->GetDocument();
	HWND hWnd=NULL;//显示错误信息的窗口句柄 NULL为主窗口
    CThinning *micro=new CThinning();
	micro->MicroThin(hWnd,pDoc);


	// 恢复光标形状
	EndWaitCursor(); 
	
	// 设置脏标记
	pDoc->SetModifiedFlag(TRUE);
	// 更新视图
	pDoc->UpdateAllViews(NULL);

}