/***************************************
*******************************
*
* 函数名称：
*     AddPepperSaltNoise()
*
* 参数：
*     void
*
* 返回值：
*     void
*
* 说明：在view中为图像添加椒盐噪声
*     
*
********************************************
**************************/
void CImgEnhance::AddPepperSaltNoise()
{
unsigned char* pDIB;
int bytecount,i;
if(m_pImgDataOut!=NULL)
{
delete []m_pImgDataOut;
m_pImgDataOut=NULL;
}
pDIB=m_pImgData;
if(m_nBitCount==8||m_nBitCount==24)
{
bytecount = m_imgWidth * m_imgHeight *m_nBitCount/8; 
//获取位图数据区的字节数
}
else
{
AfxMessageBox("只能处理真彩色和8位灰度图像!");
return ;
}
if (m_nBitCount==8)      //处理灰度图像
{  
for (i=0;i<bytecount;i++)
{
if(rand()>32000)  pDIB[i]=0;
if(rand()<200)    pDIB[i]=255;
}
}
else                  //处理24位真彩色图像
{
for (i=0;i<bytecount;i=i+3)
{
int num=rand();
if (num>32000) 
{
pDIB[i]=(rand())%255;        //处理每一个像素的RGB值
pDIB[i+1]=(rand())%255;
pDIB[i+2]=(rand())%255;
}
if (num<200) 
{
pDIB[i]=(rand())%255;
pDIB[i+1]=(rand())%255;
pDIB[i+2]=(rand())%255;
}
}
}
//生成新的DIB位图
m_nBitCountOut=m_nBitCount;
int lineByteOut=(m_imgWidth*m_nBitCountOut/8+3)/4*4;
if (!m_pImgDataOut)
{
m_pImgDataOut=new unsigned char
[lineByteOut*m_imgHeight];
}
int j, pixelByte=m_nBitCountOut/8;
for(i=0;i<m_imgHeight;i++){
for(j=0;j<m_imgWidth*pixelByte;j++)
*(m_pImgDataOut+i*lineByteOut+j)= *
(pDIB+i*lineByteOut+j);
}
}

 /***************************************
*******************************
*
* 函数名称：
*     AddRandomNoise()
*
* 参数：
*     void
*
* 返回值：
*     void
*
* 说明：在view中为图像添加随机噪声
*     
*
***************************************
*******************************/
void CImgEnhance::AddRandomNoise()
{
unsigned char* pDIB;
int bytecount,i;
if(m_pImgDataOut!=NULL)
{
delete []m_pImgDataOut;
m_pImgDataOut=NULL;
}
pDIB=m_pImgData;
if(m_nBitCount==8)
{
bytecount = m_imgWidth * m_imgHeight *m_nBitCount/8; 
//获取位图数据区的字节数
}
else
{
AfxMessageBox("只能处理8位灰度图像!");
return ;
}
//在DIB中添加随机噪声 
for( i=0;i<bytecount;i++)
{
LONG Temp=rand();
Temp=pDIB[i]*224/256+Temp/512;
pDIB[i]=Temp>=255 255:Temp;
}

m_nBitCountOut=m_nBitCount;
int lineByteOut=(m_imgWidth*m_nBitCountOut/8+3)/4*4;
if (!m_pImgDataOut)
{
m_pImgDataOut=new unsigned char
[lineByteOut*m_imgHeight];
}

int j, pixelByte=m_nBitCountOut/8;
for(i=0;i<m_imgHeight;i++){
for(j=0;j<m_imgWidth*pixelByte;j++)
*(m_pImgDataOut+i*lineByteOut+j)= *
(pDIB+i*lineByteOut+j);
}

}

void CDemoView::OnRandom() 
{
CDemoDoc *pDoc=GetDocument();
ImgCenterDib *pDib=pDoc->GetPDib();

if(pDib->m_nBitCount!=8
::MessageBox(0,"只处理灰度图像",MB_OK,0);
return ;
}

CImgEnhance imgnoise(pDib->GetDimensions(),
pDib->m_nBitCount,
pDib->m_lpColorTable, pDib->m_pImgData);
imgnoise.AddRandomNoise();

CMainFrame* pFrame = (CMainFrame *)
(AfxGetApp()->m_pMainWnd);
pFrame->SendMessage(WM_COMMAND, ID_FILE_NEW);

CDemoView* pView=(CDemoView*)pFrame->
MDIGetActive()->GetActiveView();
CDemoDoc* pDocNew=pView->GetDocument();
ImgCenterDib *dibNew=pDocNew->GetPDib();

dibNew->ReplaceDib(imgnoise.GetDimensions(),
imgnoise.m_nBitCountOut,imgnoise.
m_lpColorTable, imgnoise.m_pImgDataOut);
pDocNew->SetModifiedFlag(TRUE);
pDocNew->UpdateAllViews(pView);

Invalidate();

}

void CDemoView::OnAveSmooth()
{
CDemoDoc *pDoc=GetDocument();
ImgCenterDib *pDib=pDoc->GetPDib();

if(pDib->m_nBitCount!=8
::MessageBox(0,"只处理灰度图像",MB_OK,0);
return ;
}

CImgEnhance imgnoise(pDib->GetDimensions(),pDib->m_nBitCount,
pDib->m_lpColorTable, pDib->m_pImgData);
int TH, TW, TCX, TCY;
float fCoef;
float T[9];
//设模板为平均模板
T[0]=1.0;
T[1]=1.0;
T[2]=1.0;
T[3]=1.0;
T[4]=1.0;
T[5]=1.0;
T[6]=1.0;
T[7]=1.0;
T[8]=1.0;
//初始化对话框变量
TH=3;
TW=3;
TCX=1;
TempCY=1;
fCoef=(float)(1.0/9.0);
imgnoise.AvgTemplate(TempH,TempW,TempCX,TempCY,Temp,fCoef);

CMainFrame* pFrame = (CMainFrame *)(AfxGetApp()->m_pMainWnd);
pFrame->SendMessage(WM_COMMAND, ID_FILE_NEW);

CDemoView* pView=(CDemoView*)pFrame->MDIGetActive()->GetActiveView();
CDemoDoc* pDocNew=pView->GetDocument();
ImgCenterDib *dibNew=pDocNew->GetPDib();

dibNew->ReplaceDib(imgnoise.GetDimensions(),
imgnoise.m_nBitCountOut,imgnoise.m_lpColorTable, imgnoise.m_pImgDataOut);
pDocNew->SetModifiedFlag(TRUE);
pDocNew->UpdateAllViews(pView);

Invalidate();
return;
}