watermarkdlg.cpp

程序完成数字图像的加密解密过程

// WaterMarkDlg.cpp : implementation file
//

#include "stdafx.h"
#include "WaterMark.h"
#include "WaterMarkDlg.h"
#include "Dib.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWaterMarkDlg dialog

CWaterMarkDlg::CWaterMarkDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWaterMarkDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CWaterMarkDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	show[0]=show[1]=show[2]=FALSE;
	hwnd[0]=hwnd[1]=hwnd[3]=NULL;
	hSrcDC[0]=hSrcDC[1]=hSrcDC[2]=hDesDC[0]=hDesDC[1]=hDesDC[2]=NULL;
	m_dib = new CDib();
	m_dib1 = new CDib();
	pEdit=pEdit1=NULL;
	pStatic=pStatic1=NULL;
	Mark=FALSE;
	m_tOriPixelArray=m_watermarkdata=m_waterEmbed=m_pickupwater=NULL;
	ImageHeight=m_Ih=ImageWidth=m_Iw=WaterHeight=m_Wh=WaterWidth=m_Ww=0;
	number=0;
}

void CWaterMarkDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CWaterMarkDlg)
	DDX_Control(pDX, IDC_STATIC0, m_pic);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CWaterMarkDlg, CDialog)
	//{{AFX_MSG_MAP(CWaterMarkDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON1, OnOpenFile)
	ON_WM_TIMER()
	ON_BN_CLICKED(IDC_BUTTON2, OnOpenWaterMark)
	ON_BN_CLICKED(IDC_BUTTON3, OnWaterEmbed)
	ON_BN_CLICKED(IDC_BUTTON5, OnPick_Up)
	ON_BN_CLICKED(IDC_BUTTON4, OnSaveWEBitmap)
	ON_BN_CLICKED(IDC_BUTTON6, OnOpenW_E_Bitmap)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWaterMarkDlg message handlers

BOOL CWaterMarkDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CWaterMarkDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CWaterMarkDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CWaterMarkDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CWaterMarkDlg::OnOpenFile() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE, "", NULL, 
					OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_ALLOWMULTISELECT, 
					"(*.bmp)|*.bmp|所有文件(*.*)|*.*||",AfxGetMainWnd());
	//读取图象的文件名CString filename;
	if(dlg.DoModal()==IDOK)
	{
	//Bitmap2Data()将图象的转换为数据保存在二维数组m_tOriPixelArray
	POSITION pos = dlg.GetStartPosition();
	filename = dlg.GetNextPathName(pos);
	Bitmap2Data();
	//////////////////////////////////
	if(hwnd[0]!=NULL)
		hwnd[0]=NULL;
	if(hSrcDC[0]!=NULL)
		hSrcDC[0]=NULL;
	if(hDesDC[0]!=NULL)
		hDesDC[0]=NULL;

	hwnd[0] = GetDlgItem(IDC_STATIC0);
	hDesDC[0] = hwnd[0]->GetDC()->m_hDC;
	hSrcDC[0] = CreateCompatibleDC(hDesDC[0]);			
		
	filename=dlg.GetPathName();
	hBitmap[0]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),filename,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
	GetObject(hBitmap[0], sizeof BITMAP, &bm[0]);		
	SelectObject(hSrcDC[0], hBitmap[0]);
	hwnd[0]->GetClientRect(&rect[0]);
		
	::SetStretchBltMode(hDesDC[0],COLORONCOLOR);       
	::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);
	show[0]=TRUE;
	SetTimer(NULL,50,0);
				
	GetDlgItem(IDC_BUTTON2)->EnableWindow(TRUE);
	}	
	else
	{/*	GetDlgItem(IDC_STATIC0)->MoveWidow(rect(0,0,bm.bmWidth,bm.bmHeight));
		hwnd->MoveWidow(rect(0,0,bm.bmWidth,bm.bmHeight));
		m_pic.SetBitmap(hBitmap);
		this->RedrawWindow();*/
		TRACE("错误");
	}

	CString s,s0,s1,s2;
	s.Format("%2d",bm[0].bmHeight);
	s1.Format("%2d",bm[0].bmWidth);
	s0.Format("%d",m_dib->GetBiBitCount());
	s2="宿主图象"+s0+"位H×W:"+s+"X"+s1;
		
	if(pEdit!=NULL)
		pEdit=NULL;
	r.left=35;r.top=rect[0].bottom+2;r.right=200+r.left;r.bottom=r.top+20;
	pEdit=new CEdit;
	pEdit->Create(ES_CENTER|WS_VISIBLE|ES_READONLY,r,this,1);
	CFont * cFont=new CFont;
	cFont->CreateFont(16,0,0,0,FW_SEMIBOLD,FALSE,FALSE,0, 
                                         ANSI_CHARSET,OUT_DEFAULT_PRECIS,
                                         CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY,
                                         DEFAULT_PITCH&FF_SWISS,"Arial");
	pEdit->SetFont(cFont,TRUE);
	pEdit->SetWindowText(s2);
}


void CWaterMarkDlg::OnOpenWaterMark() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg1(TRUE,"bmp",".bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"位图文件(*.bmp)|*.bmp|JPEG 文件(*.jpg)|*.jpg|GIF文件(*.gif)|*.gif||");	
	if(dlg1.DoModal()==IDOK)
	{
	if(hwnd[1]!=NULL)
		hwnd[1]=NULL;
	if(hSrcDC[1]!=NULL)
		hSrcDC[1]=NULL;
	if(hDesDC[1]!=NULL)
		hDesDC[1]=NULL;
	hwnd[1] = GetDlgItem(IDC_STATIC1);
	hDesDC[1] = hwnd[1]->GetDC()->m_hDC;
	hSrcDC[1] = CreateCompatibleDC(hDesDC[1]);			
	watermark=dlg1.GetPathName();
	hBitmap[1]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),watermark,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
	GetObject(hBitmap[1], sizeof BITMAP, &bm[1]);

	SelectObject(hSrcDC[1], hBitmap[1]);
	hwnd[1]->GetClientRect(&rect[1]);
	::SetStretchBltMode(hDesDC[1],COLORONCOLOR);       
	::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);
	show[1]=TRUE;
		
	if ((bm[0].bmHeight*bm[0].bmWidth)<2*(bm[1].bmHeight*bm[1].bmWidth))
	{	
		//不能嵌入水印		
		MessageBox("水印图象过大请更换较大宿主图象","错误",MB_OK);
		Mark=FALSE;
	}
	else
	{
		SetTimer(NULL,50,0);
		//WaterBitmap2Data()将图象的转换为数据保存在二维数组_watermarkdata
		POSITION pos = dlg1.GetStartPosition();
		watermark = dlg1.GetNextPathName(pos);;
		WaterBitmap2Data();//GetDlgItem(IDC_STOP_BUTTON)->EnableWindow(FALSE);
		Mark=TRUE;
		GetDlgItem(IDC_BUTTON3)->EnableWindow(TRUE);
	}

	CString s,s0,s1,s2;
	s.Format("%2d",bm[1].bmHeight);
	s1.Format("%2d",bm[1].bmWidth);
	s0.Format("%d",m_dib1->GetBiBitCount());
	s2="水印图象"+s0+"位H×W:"+s+"X"+s1;
		
	if(pEdit1!=NULL)
		pEdit1=NULL;
	r1.left=rect[0].right+35;r1.top=rect[1].bottom+2;r1.right=200+r1.left;r1.bottom=r1.top+20;
	pStatic1 = new CStatic;
	pStatic1->Create(s2,ES_CENTER|WS_VISIBLE|ES_READONLY,r1,this,4);
	}
		
}

void CWaterMarkDlg::Bitmap2Data()//将宿主图象的转换为数据保存在二维数组m_tOriPixelArray
{
	int i,j;
	
	if(m_tOriPixelArray !=NULL)
		if(m_Ih!=0)
			for(i=0; i<m_Ih; i++)
				delete [] m_tOriPixelArray[i];
		else
			for(i=0; i<ImageHeight; i++)
				delete [] m_tOriPixelArray[i];
			
	if(filename.GetLength()==0)
	{
		MessageBox("无效的文件");
		return ;
	}
	
	m_dib->Open(filename);
	ImageWidth = m_dib->GetWidth();
	ImageHeight = m_dib->GetHeight();
	biBitCount = m_dib->GetBiBitCount(); 

	BYTE  *colorTable;
	colorTable =(BYTE *) m_dib->m_pDibBits;
	int byteBitCount = m_dib->GetBiBitCount()/8;

	m_tOriPixelArray =new RGBQUAD*[ImageHeight];
	
	for (int l=0;l<ImageHeight;l++)
	m_tOriPixelArray[l]=new RGBQUAD [ImageWidth];
	
	int count = 0;
	int num = 0;
	for(i=ImageHeight-1; i>=0; i--)
	{
		for(j=0; j<ImageWidth; j++)
		{
			m_tOriPixelArray[i][j].rgbBlue =colorTable[count++];
			m_tOriPixelArray[i][j].rgbGreen=colorTable[count++];
			m_tOriPixelArray[i][j].rgbRed  =colorTable[count++];
			m_tOriPixelArray[i][j].rgbReserved = 0;
			count += byteBitCount-3;
			num+=1;
		}
		count += (4-(ImageWidth*byteBitCount)%4)%4;		
	}
	show[2]=TRUE;

}

void CWaterMarkDlg::WaterBitmap2Data()//将水印图象的转换为数据保存在二维数组m_watermarkdata
{
	int i,j;	
	if(m_watermarkdata !=NULL)
		if(m_Wh!=0)
		for(i=0; i<m_Wh; i++)
			delete [] m_watermarkdata[i];
		else
			return ;

	if(watermark.GetLength()==0)
	{
		MessageBox("无效的水印文件");
		return ;
	}
	m_dib1->Open(watermark);if(m_dib1->GetBiBitCount()>8) MessageBox("目前仅对8位的水印信息有效");
	
	WaterWidth=m_dib1->GetWidth();
	WaterHeight=m_dib1->GetHeight();
	biBitCount1 = m_dib1->GetBiBitCount();
	
	BYTE  *colorTable;
	colorTable =(BYTE *) m_dib1->m_pDibBits;
	int byteBitCount = m_dib1->GetBiBitCount()/8;

	m_watermarkdata =new RGBQUAD*[WaterHeight];	
	for (int l=0;l<WaterHeight;l++)
	m_watermarkdata[l]=new RGBQUAD [WaterWidth];
	
	int count = 0;