watermarkdlg.cpp

数字水印源程序

	int num = 0;
	for(i=WaterHeight-1; i>=0; i--)
	{
		for(j=0; j<WaterWidth; j++)
		{
			m_watermarkdata[i][j].rgbBlue  = colorTable[count++];
			m_watermarkdata[i][j].rgbGreen = colorTable[count++];
			m_watermarkdata[i][j].rgbRed   = colorTable[count++];
			m_watermarkdata[i][j].rgbReserved = 0;
			count += byteBitCount-3;
			num+=1; //TRACE("\n@%d @%d @%d",m_watermarkdata[i][j].rgbRed,m_watermarkdata[i][j].rgbBlue,m_watermarkdata[i][j].rgbRed);//
		}
		count += (4-(WaterWidth*byteBitCount)%4)%4;		
	}
}
//转换公式为Gray(i,j)=0.11*R(i,j)+0.59*G(i,j)+0.3*B(i,j)，
//其中Gray(i,j)为转换后的黑白图像在（i,j）点处的灰度值，
//其中绿色所占的比重最大，所以转换时可以直接使用G值。
void CWaterMarkDlg::OnWaterEmbed() 
{
	// TODO: Add your control notification handler code here
	if ((bm[0].bmHeight*bm[0].bmWidth)<3*(bm[1].bmHeight*bm[1].bmWidth))
	{	
		//不能嵌入水印		
		MessageBox("水印图象过大请更换较大宿主图象","错误",MB_OK);
		Mark=FALSE;
	}
	if(!Mark)
		return ;	
	if(biBitCount<biBitCount1)
		{
		MessageBox("水印受限制");
		return ;
		}

	RGBQUAD *m_copymater;//定义一个水印信息的数据副本
	m_copymater = new RGBQUAD [WaterHeight*WaterWidth];
	
	int i,j;
	int count=0;
	BYTE m_r,m_b;
	if(m_waterEmbed !=NULL)//m_waterEmbed是水印.嵌入后.的位图数据全局变量
		if(m_Ih!=0)
		for(int l =0;l<m_Ih;l++)
			delete [] m_waterEmbed[l];//?
	m_waterEmbed = new RGBQUAD*[ImageHeight];
	for(i=0;i<ImageHeight;i++)
		m_waterEmbed[i] = new RGBQUAD [ImageWidth];	
	for(i=0;i<ImageHeight;i++)//将原来的图象的数据复制到m_waterEmbed里
		for(j=0;j<ImageWidth;j++)
			m_waterEmbed[i][j]=m_tOriPixelArray[i][j];
	
	for(i=0;i<WaterHeight;i++)//将水印信息的数据复制到一维数组m_copymater水印信息的数据副本里
		for(j=0;j<WaterWidth;j++)
		{
			m_copymater[i*WaterHeight+j]=m_watermarkdata[i][j];
			count++;
		}

	number=count;//记录嵌入的点数
	//
	int k=0;		
	if (biBitCount==24)//Operate_Byte()嵌入水印信息位操作
	{	
		for(i=0;i<ImageHeight&&count>0;i++)
			for(j=0;j<ImageWidth&&count>0;j++,count--)//下面可以用一条语句的 但为了测试还是多句
			{	
				m_r=m_waterEmbed[i][j].rgbRed;
				m_b=m_waterEmbed[i][j].rgbBlue;
				BYTE &a=m_r;
				BYTE &b=m_b;
				BYTE c=m_copymater[i*ImageHeight+j].rgbGreen;
				Operate_Byte(a,b,c);
				m_waterEmbed[i][j].rgbRed=m_r;
				m_waterEmbed[i][j].rgbBlue=m_b;				
				//TRACE("\n%d",c);k++;
			}
		PutInWaterMessage();//
	}
	else 
		return;	

	//TRACE("\n次数是%d",k);

	CDC *dc=GetDC();//将象素打印出来
	RECT m_rect;
	float w,h,p,q;
	GetDlgItem(IDC_STATIC2)->GetClientRect(&m_rect);
	if(m_rect.right<ImageWidth&&m_rect.bottom<ImageHeight)
	{
		w=(float)m_rect.right/(float)ImageWidth;
		h=(float)m_rect.bottom/(float)ImageHeight;
	}
	else
		w=h=1;
	for(  p=0,i=0;i<ImageHeight;i++,p++)
		for( j=0, q=0;j<ImageWidth;j++,q++)
		dc->SetPixel(2+(int)(q*w),235+(int)(p*h),RGB(m_waterEmbed[i][j].rgbRed,
			 m_waterEmbed[i][j].rgbGreen,m_waterEmbed[i][j].rgbBlue));

	ReleaseDC(dc);
	
	delete [] m_copymater;
	
	m_Ih=ImageHeight;//为保证直接更换水印信息而没有更换宿主图象提供原宽高 void error bug
	m_Iw=ImageWidth;
	m_Wh=WaterHeight;
	m_Ww=WaterWidth;
	TRACE("\n%d\n",ImageHeight);
	GetDlgItem(IDC_BUTTON4)->EnableWindow(TRUE);
	GetDlgItem(IDC_BUTTON5)->EnableWindow(TRUE);
}



void CWaterMarkDlg::ShowBitmap(CString filename)
{

}

void CWaterMarkDlg::SetBitmap2Pixel()
{
	if(ImageWidth*ImageHeight==0)
		return	;
	CDC *dc=GetDC();
	for(int i=0;i<ImageHeight;i++)
		for(int j=0;j<ImageWidth;j++)
		dc->SetPixel(5+j,i+220,RGB(m_tOriPixelArray[i][j].rgbRed,
			 m_tOriPixelArray[i][j].rgbGreen,m_tOriPixelArray[i][j].rgbBlue));

	ReleaseDC(dc);
}

void CWaterMarkDlg::SetWaterBitmap2Pixel()
{
	if(ImageWidth*ImageHeight==0)
		return	;
	CDC *dc1=GetDC();
	for(int i=0;i<WaterHeight;i++)
		for(int j=0;j<WaterWidth;j++)
		dc1->SetPixel(5+j,i+220,RGB(m_watermarkdata[i][j].rgbRed,
			 m_watermarkdata[i][j].rgbGreen,m_watermarkdata[i][j].rgbBlue));
	
	ReleaseDC(dc1);
}

void CWaterMarkDlg::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
		if (show[0])
		{
			::StretchBlt(hDesDC[0], rect[0].left, rect[0].top, rect[0].right, \
				rect[0].bottom, hSrcDC[0], 0, 0, bm[0].bmWidth, bm[0].bmHeight,+SRCCOPY);
		}
	
		if (show[1])
		{
			::StretchBlt(hDesDC[1], rect[1].left, rect[1].top, rect[1].right, \
				rect[1].bottom, hSrcDC[1], 0, 0, bm[1].bmWidth, bm[1].bmHeight,+SRCCOPY);
		}
		if(show[2])
		{
			::StretchBlt(hDesDC[2], rect[2].left, rect[2].top, rect[2].right, \
				rect[2].bottom, hSrcDC[2], 0, 0, bm[2].bmWidth, bm[2].bmHeight,+SRCCOPY);
		}
		CDialog::OnTimer(nIDEvent);
}


void CWaterMarkDlg::Operate_Byte(BYTE &operate1, BYTE &operate2, BYTE m_operate)
{
	//将m_operate最低四位取出来放到L里最高四位放到H里，
	//最后将H赋值给operate1最后四位 L赋值给operate2最后四位
	int i;
	BYTE M=240,t=m_operate;//1111 0000
	BYTE H,L,x[8],y[4];
	y[3]=8;y[2]=4;y[1]=2;y[0]=1;

	for(i=0;i<8;i++)
	{
		x[i]=m_operate&1;
		m_operate>>=1;
	}
	
	H=x[7]*y[3]+x[6]*y[2]+x[5]*y[1]+x[4]*y[0];//hight
	L=x[3]*y[3]+x[2]*y[2]+x[1]*y[1]+x[0]*y[0];//low

	operate1&=M;
	operate2&=M;
	operate1+=H;
	operate2+=L;
}


void CWaterMarkDlg::OnPick_Up() 
{
	// TODO: Add your control notification handler code here
	
	PickUpWaterMessage(m_Wh,m_Ww);
}

void CWaterMarkDlg::PickUpWaterMessage(long h,long w)
{
	int i,j;
	RGBQUAD *m_color;
	m_color = new RGBQUAD[h*w];
	
	int count=number;
	if(m_Ih!=0)
	for(i=0;i<m_Ih&&count>0;i++)
		for(j=0;j<m_Iw&&count>0;j++,count--)
			m_color[i*m_Ih+j]=m_waterEmbed[i][j];
			


	if(m_pickupwater!=NULL)
		if(m_pwh!=0)
			for(i=0;i<m_pwh;i++)
			delete [] m_pickupwater[i];
	m_pickupwater = new RGBQUAD *[h];
	for(i=0;i<h;i++)
		m_pickupwater[i] = new RGBQUAD[w];
	
	count=number;
	int k=0;
	for(i=0;i<h&&count>0;i++)
		for(j=0;j<w&&count>0;j++,count--)
		{
			m_pickupwater[i][j].rgbGreen=Operate_ByteOut(m_color[i*h+j].rgbRed,m_color[i*h+j].rgbBlue);
			m_pickupwater[i][j].rgbRed=m_pickupwater[i][j].rgbGreen;
			m_pickupwater[i][j].rgbBlue=m_pickupwater[i][j].rgbGreen;
			//TRACE("\n#%d\n",m_pickupwater[i][j].rgbGreen);k++;
		}

	//TRACE("\n提取的次数是%d\n",k);

	CDC *dc=GetDC();//将象素打印出来
	RECT m_rect;
	GetDlgItem(IDC_STATIC3)->GetClientRect(&m_rect);

	for(  i=0;i<h;i++)
		for( j=0;j<w;j++)
		dc->SetPixel(m_rect.right+2+j,235+i,RGB(m_pickupwater[i][j].rgbRed,
			 m_pickupwater[i][j].rgbGreen,m_pickupwater[i][j].rgbBlue));

	ReleaseDC(dc);

	
	m_pwh=h;//获得最后一次水印的信息位图的宽高信息
	m_pww=w;
}

void CWaterMarkDlg::PutInWaterMessage()
{
	long a,b,c,d;
	a = ImageHeight;
	b = ImageWidth;
	c = WaterHeight;
	d = WaterWidth;
	
}

BYTE CWaterMarkDlg::Operate_ByteOut(BYTE operate1, BYTE operate2)
{
	//相对与Operate_Byte()
	int i = 0;
	BYTE x[8],y[8],data;
	
	for (i=0;i<4;i++)
	{
		x[i]=operate2&1;
		operate2>>=1;//取出最低的四位
	}
	
	for (i=4;i<8;i++)
	{
		x[i]=operate1&1;
		operate1>>=1;//取出最高的四位
	}

	y[0]=1;y[1]=2;y[2]=4;y[3]=8;y[4]=16;y[5]=32;y[6]=64;y[7]=128;
	data=x[0]*y[0]+x[1]*y[1]+x[2]*y[2]+x[3]*y[3]+x[4]*y[4]+x[5]*y[5]+x[6]*y[6]+x[7]*y[7];
	
	return data;
}

void CWaterMarkDlg::OnSaveWEBitmap() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(FALSE,"未标题bmp","未标题.bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"位图文件(*.bmp)|*.bmp");
	if(dlg.DoModal()!=IDOK)
		return ;
	CString s;
	CDib *m_dib_save;
	m_dib_save = new CDib();
	m_dib_save = m_dib;
	s=dlg.GetPathName();
	int i,j;
	int byteBitCount = m_dib_save->GetBiBitCount()/8;
	long H=m_dib_save->GetHeight();
	long W=m_dib_save->GetWidth();
	int ct =0;
	
	for(i=H-1;i>=0;i--)//0.11*R(i,j)+0.59*G(i,j)+0.3*B(i,j)
	{
		for(j=0;j<W;j++)
		{	
			((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbBlue;
			((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbGreen;
			((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbRed;
			ct +=byteBitCount -3;
		}
		ct +=(4-(W*byteBitCount)%4)%4;
	}
	m_dib_save->Save(s);
}

void CWaterMarkDlg::OnOpenW_E_Bitmap() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE, "", NULL, 
					OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_ALLOWMULTISELECT, 
					"(*.bmp)|*.bmp|所有文件(*.*)|*.*||",AfxGetMainWnd());
	if(dlg.DoModal()!=IDOK)
		return ;


	if(hwnd[2]!=NULL)
		hwnd[2]=NULL;
	if(hSrcDC[2]!=NULL)
		hSrcDC[2]=NULL;
	if(hDesDC[2]!=NULL)
		hDesDC[2]=NULL;

	hwnd[2] = GetDlgItem(IDC_STATIC2);
	hDesDC[2] = hwnd[2]->GetDC()->m_hDC;
	hSrcDC[2] = CreateCompatibleDC(hDesDC[2]);			
	CString file;
	file=dlg.GetPathName();

	POSITION pos = dlg.GetStartPosition();
	file = dlg.GetNextPathName(pos);

	Embed2Data( file);
	hBitmap[2]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),file,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
	GetObject(hBitmap[2], sizeof BITMAP, &bm[2]);		
	SelectObject(hSrcDC[2], hBitmap[2]);
	hwnd[2]->GetClientRect(&rect[2]);
		
	::SetStretchBltMode(hDesDC[2],COLORONCOLOR);       
	::StretchBlt(hDesDC[2], rect[2].left, rect[2].top, rect[2].right, rect[2].bottom, hSrcDC[2], 0, 0, bm[2].bmWidth, bm[2].bmHeight,+SRCCOPY);
	show[2]=TRUE;
	SetTimer(NULL,50,0);				
}

void CWaterMarkDlg::Embed2Data(CString m_filename)//需要加入代码避免错误
{
	int i,j,H,W;
	m_dib->Open(m_filename);
	H = m_dib->GetHeight();
	W = m_dib->GetWidth();
			
	if(m_waterEmbed !=NULL)//m_waterEmbed是水印.嵌入后.的位图数据全局变量
		if(m_Ih!=0)
		for(int l =0;l<m_Ih;l++)
			delete [] m_waterEmbed[l];//?
	else
	{
		MessageBox("Please add right code!");
		return;
	}
	m_waterEmbed = new RGBQUAD*[H];
	for(i=0;i<H;i++)
		m_waterEmbed[i] = new RGBQUAD [W];	

	
	BYTE *color ;
	color	=(BYTE*) m_dib->m_pDibBits;
	int byteBitCount = m_dib->GetBiBitCount()/8;
	
	int count =0;
	for(i=H-1;i>=0;i--)
	{
		for(j=0;j<W;j++)
		{
			m_waterEmbed[i][j].rgbBlue	= color[count++];
			m_waterEmbed[i][j].rgbGreen	= color[count++];
			m_waterEmbed[i][j].rgbRed	= color[count++];
			m_waterEmbed[i][j].rgbReserved=0;
			count +=byteBitCount -3;
		}
		count +=(4-(W*byteBitCount)%4)%4;
	}
}