WaterMarkDlg.cpp 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789
  1. // WaterMarkDlg.cpp : implementation file
  2. // Download by http://www.NewXing.com
  3. #include "stdafx.h"
  4. #include "WaterMark.h"
  5. #include "WaterMarkDlg.h"
  6. #include "Dib.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  12. /////////////////////////////////////////////////////////////////////////////
  13. // CAboutDlg dialog used for App About
  14. class CAboutDlg : public CDialog
  15. {
  16. public:
  17. CAboutDlg();
  18. // Dialog Data
  19. //{{AFX_DATA(CAboutDlg)
  20. enum { IDD = IDD_ABOUTBOX };
  21. //}}AFX_DATA
  22. // ClassWizard generated virtual function overrides
  23. //{{AFX_VIRTUAL(CAboutDlg)
  24. protected:
  25. virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
  26. //}}AFX_VIRTUAL
  27. // Implementation
  28. protected:
  29. //{{AFX_MSG(CAboutDlg)
  30. //}}AFX_MSG
  31. DECLARE_MESSAGE_MAP()
  32. };
  33. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  34. {
  35. //{{AFX_DATA_INIT(CAboutDlg)
  36. //}}AFX_DATA_INIT
  37. }
  38. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  39. {
  40. CDialog::DoDataExchange(pDX);
  41. //{{AFX_DATA_MAP(CAboutDlg)
  42. //}}AFX_DATA_MAP
  43. }
  44. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  45. //{{AFX_MSG_MAP(CAboutDlg)
  46. // No message handlers
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  49. /////////////////////////////////////////////////////////////////////////////
  50. // CWaterMarkDlg dialog
  51. CWaterMarkDlg::CWaterMarkDlg(CWnd* pParent /*=NULL*/)
  52. : CDialog(CWaterMarkDlg::IDD, pParent)
  53. {
  54. //{{AFX_DATA_INIT(CWaterMarkDlg)
  55. // NOTE: the ClassWizard will add member initialization here
  56. //}}AFX_DATA_INIT
  57. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  58. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  59. show[0]=show[1]=show[2]=FALSE;
  60. hwnd[0]=hwnd[1]=hwnd[3]=NULL;
  61. hSrcDC[0]=hSrcDC[1]=hSrcDC[2]=hDesDC[0]=hDesDC[1]=hDesDC[2]=NULL;
  62. m_dib = new CDib();
  63. m_dib1 = new CDib();
  64. pEdit=pEdit1=NULL;
  65. pStatic=pStatic1=NULL;
  66. Mark=FALSE;
  67. m_tOriPixelArray=m_watermarkdata=m_waterEmbed=m_pickupwater=NULL;
  68. ImageHeight=m_Ih=ImageWidth=m_Iw=WaterHeight=m_Wh=WaterWidth=m_Ww=0;
  69. number=0;
  70. }
  71. void CWaterMarkDlg::DoDataExchange(CDataExchange* pDX)
  72. {
  73. CDialog::DoDataExchange(pDX);
  74. //{{AFX_DATA_MAP(CWaterMarkDlg)
  75. DDX_Control(pDX, IDC_STATIC0, m_pic);
  76. //}}AFX_DATA_MAP
  77. }
  78. BEGIN_MESSAGE_MAP(CWaterMarkDlg, CDialog)
  79. //{{AFX_MSG_MAP(CWaterMarkDlg)
  80. ON_WM_SYSCOMMAND()
  81. ON_WM_PAINT()
  82. ON_WM_QUERYDRAGICON()
  83. ON_BN_CLICKED(IDC_BUTTON1, OnOpenFile)
  84. ON_WM_TIMER()
  85. ON_BN_CLICKED(IDC_BUTTON2, OnOpenWaterMark)
  86. ON_BN_CLICKED(IDC_BUTTON3, OnWaterEmbed)
  87. ON_BN_CLICKED(IDC_BUTTON5, OnPick_Up)
  88. ON_BN_CLICKED(IDC_BUTTON4, OnSaveWEBitmap)
  89. ON_BN_CLICKED(IDC_BUTTON6, OnOpenW_E_Bitmap)
  90. //}}AFX_MSG_MAP
  91. END_MESSAGE_MAP()
  92. /////////////////////////////////////////////////////////////////////////////
  93. // CWaterMarkDlg message handlers
  94. BOOL CWaterMarkDlg::OnInitDialog()
  95. {
  96. CDialog::OnInitDialog();
  97. // Add "About..." menu item to system menu.
  98. // IDM_ABOUTBOX must be in the system command range.
  99. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  100. ASSERT(IDM_ABOUTBOX < 0xF000);
  101. CMenu* pSysMenu = GetSystemMenu(FALSE);
  102. if (pSysMenu != NULL)
  103. {
  104. CString strAboutMenu;
  105. strAboutMenu.LoadString(IDS_ABOUTBOX);
  106. if (!strAboutMenu.IsEmpty())
  107. {
  108. pSysMenu->AppendMenu(MF_SEPARATOR);
  109. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  110. }
  111. }
  112. // Set the icon for this dialog. The framework does this automatically
  113. // when the application's main window is not a dialog
  114. SetIcon(m_hIcon, TRUE); // Set big icon
  115. SetIcon(m_hIcon, FALSE); // Set small icon
  116. // TODO: Add extra initialization here
  117. return TRUE; // return TRUE unless you set the focus to a control
  118. }
  119. void CWaterMarkDlg::OnSysCommand(UINT nID, LPARAM lParam)
  120. {
  121. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  122. {
  123. CAboutDlg dlgAbout;
  124. dlgAbout.DoModal();
  125. }
  126. else
  127. {
  128. CDialog::OnSysCommand(nID, lParam);
  129. }
  130. }
  131. // If you add a minimize button to your dialog, you will need the code below
  132. // to draw the icon. For MFC applications using the document/view model,
  133. // this is automatically done for you by the framework.
  134. void CWaterMarkDlg::OnPaint()
  135. {
  136. if (IsIconic())
  137. {
  138. CPaintDC dc(this); // device context for painting
  139. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  140. // Center icon in client rectangle
  141. int cxIcon = GetSystemMetrics(SM_CXICON);
  142. int cyIcon = GetSystemMetrics(SM_CYICON);
  143. CRect rect;
  144. GetClientRect(&rect);
  145. int x = (rect.Width() - cxIcon + 1) / 2;
  146. int y = (rect.Height() - cyIcon + 1) / 2;
  147. // Draw the icon
  148. dc.DrawIcon(x, y, m_hIcon);
  149. }
  150. else
  151. {
  152. CDialog::OnPaint();
  153. }
  154. }
  155. // The system calls this to obtain the cursor to display while the user drags
  156. // the minimized window.
  157. HCURSOR CWaterMarkDlg::OnQueryDragIcon()
  158. {
  159. return (HCURSOR) m_hIcon;
  160. }
  161. void CWaterMarkDlg::OnOpenFile()
  162. {
  163. // TODO: Add your control notification handler code here
  164. CFileDialog dlg(TRUE, "", NULL,
  165. OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_ALLOWMULTISELECT,
  166. "(*.bmp)|*.bmp|所有文件(*.*)|*.*||",AfxGetMainWnd());
  167. //读取图象的文件名CString filename;
  168. if(dlg.DoModal()==IDOK)
  169. {
  170. //Bitmap2Data()将图象的转换为数据保存在二维数组m_tOriPixelArray
  171. POSITION pos = dlg.GetStartPosition();
  172. filename = dlg.GetNextPathName(pos);
  173. Bitmap2Data();
  174. //////////////////////////////////
  175. if(hwnd[0]!=NULL)
  176. hwnd[0]=NULL;
  177. if(hSrcDC[0]!=NULL)
  178. hSrcDC[0]=NULL;
  179. if(hDesDC[0]!=NULL)
  180. hDesDC[0]=NULL;
  181. hwnd[0] = GetDlgItem(IDC_STATIC0);
  182. hDesDC[0] = hwnd[0]->GetDC()->m_hDC;
  183. hSrcDC[0] = CreateCompatibleDC(hDesDC[0]);
  184. filename=dlg.GetPathName();
  185. hBitmap[0]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),filename,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
  186. GetObject(hBitmap[0], sizeof BITMAP, &bm[0]);
  187. SelectObject(hSrcDC[0], hBitmap[0]);
  188. hwnd[0]->GetClientRect(&rect[0]);
  189. ::SetStretchBltMode(hDesDC[0],COLORONCOLOR);
  190. ::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);
  191. show[0]=TRUE;
  192. SetTimer(NULL,50,0);
  193. GetDlgItem(IDC_BUTTON2)->EnableWindow(TRUE);
  194. }
  195. else
  196. {/* GetDlgItem(IDC_STATIC0)->MoveWidow(rect(0,0,bm.bmWidth,bm.bmHeight));
  197. hwnd->MoveWidow(rect(0,0,bm.bmWidth,bm.bmHeight));
  198. m_pic.SetBitmap(hBitmap);
  199. this->RedrawWindow();*/
  200. TRACE("错误");
  201. }
  202. CString s,s0,s1,s2;
  203. s.Format("%2d",bm[0].bmHeight);
  204. s1.Format("%2d",bm[0].bmWidth);
  205. s0.Format("%d",m_dib->GetBiBitCount());
  206. s2="宿主图象"+s0+"位H×W:"+s+"X"+s1;
  207. if(pEdit!=NULL)
  208. pEdit=NULL;
  209. r.left=35;r.top=rect[0].bottom+2;r.right=200+r.left;r.bottom=r.top+20;
  210. pEdit=new CEdit;
  211. pEdit->Create(ES_CENTER|WS_VISIBLE|ES_READONLY,r,this,1);
  212. CFont * cFont=new CFont;
  213. cFont->CreateFont(16,0,0,0,FW_SEMIBOLD,FALSE,FALSE,0,
  214. ANSI_CHARSET,OUT_DEFAULT_PRECIS,
  215. CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY,
  216. DEFAULT_PITCH&FF_SWISS,"Arial");
  217. pEdit->SetFont(cFont,TRUE);
  218. pEdit->SetWindowText(s2);
  219. }
  220. void CWaterMarkDlg::OnOpenWaterMark()
  221. {
  222. // TODO: Add your control notification handler code here
  223. CFileDialog dlg1(TRUE,"bmp",".bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"位图文件(*.bmp)|*.bmp|JPEG 文件(*.jpg)|*.jpg|GIF文件(*.gif)|*.gif||");
  224. if(dlg1.DoModal()==IDOK)
  225. {
  226. if(hwnd[1]!=NULL)
  227. hwnd[1]=NULL;
  228. if(hSrcDC[1]!=NULL)
  229. hSrcDC[1]=NULL;
  230. if(hDesDC[1]!=NULL)
  231. hDesDC[1]=NULL;
  232. hwnd[1] = GetDlgItem(IDC_STATIC1);
  233. hDesDC[1] = hwnd[1]->GetDC()->m_hDC;
  234. hSrcDC[1] = CreateCompatibleDC(hDesDC[1]);
  235. watermark=dlg1.GetPathName();
  236. hBitmap[1]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),watermark,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
  237. GetObject(hBitmap[1], sizeof BITMAP, &bm[1]);
  238. SelectObject(hSrcDC[1], hBitmap[1]);
  239. hwnd[1]->GetClientRect(&rect[1]);
  240. ::SetStretchBltMode(hDesDC[1],COLORONCOLOR);
  241. ::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);
  242. show[1]=TRUE;
  243. if ((bm[0].bmHeight*bm[0].bmWidth)<2*(bm[1].bmHeight*bm[1].bmWidth))
  244. {
  245. //不能嵌入水印
  246. MessageBox("水印图象过大请更换较大宿主图象","错误",MB_OK);
  247. Mark=FALSE;
  248. }
  249. else
  250. {
  251. SetTimer(NULL,50,0);
  252. //WaterBitmap2Data()将图象的转换为数据保存在二维数组_watermarkdata
  253. POSITION pos = dlg1.GetStartPosition();
  254. watermark = dlg1.GetNextPathName(pos);;
  255. WaterBitmap2Data();//GetDlgItem(IDC_STOP_BUTTON)->EnableWindow(FALSE);
  256. Mark=TRUE;
  257. GetDlgItem(IDC_BUTTON3)->EnableWindow(TRUE);
  258. }
  259. CString s,s0,s1,s2;
  260. s.Format("%2d",bm[1].bmHeight);
  261. s1.Format("%2d",bm[1].bmWidth);
  262. s0.Format("%d",m_dib1->GetBiBitCount());
  263. s2="水印图象"+s0+"位H×W:"+s+"X"+s1;
  264. if(pEdit1!=NULL)
  265. pEdit1=NULL;
  266. r1.left=rect[0].right+35;r1.top=rect[1].bottom+2;r1.right=200+r1.left;r1.bottom=r1.top+20;
  267. pStatic1 = new CStatic;
  268. pStatic1->Create(s2,ES_CENTER|WS_VISIBLE|ES_READONLY,r1,this,4);
  269. }
  270. }
  271. void CWaterMarkDlg::Bitmap2Data()//将宿主图象的转换为数据保存在二维数组m_tOriPixelArray
  272. {
  273. int i,j;
  274. if(m_tOriPixelArray !=NULL)
  275. if(m_Ih!=0)
  276. for(i=0; i<m_Ih; i++)
  277. delete [] m_tOriPixelArray[i];
  278. else
  279. for(i=0; i<ImageHeight; i++)
  280. delete [] m_tOriPixelArray[i];
  281. if(filename.GetLength()==0)
  282. {
  283. MessageBox("无效的文件");
  284. return ;
  285. }
  286. m_dib->Open(filename);
  287. ImageWidth = m_dib->GetWidth();
  288. ImageHeight = m_dib->GetHeight();
  289. biBitCount = m_dib->GetBiBitCount();
  290. BYTE *colorTable;
  291. colorTable =(BYTE *) m_dib->m_pDibBits;
  292. int byteBitCount = m_dib->GetBiBitCount()/8;
  293. m_tOriPixelArray =new RGBQUAD*[ImageHeight];
  294. for (int l=0;l<ImageHeight;l++)
  295. m_tOriPixelArray[l]=new RGBQUAD [ImageWidth];
  296. int count = 0;
  297. int num = 0;
  298. for(i=ImageHeight-1; i>=0; i--)
  299. {
  300. for(j=0; j<ImageWidth; j++)
  301. {
  302. m_tOriPixelArray[i][j].rgbBlue =colorTable[count++];
  303. m_tOriPixelArray[i][j].rgbGreen=colorTable[count++];
  304. m_tOriPixelArray[i][j].rgbRed =colorTable[count++];
  305. m_tOriPixelArray[i][j].rgbReserved = 0;
  306. count += byteBitCount-3;
  307. num+=1;
  308. }
  309. count += (4-(ImageWidth*byteBitCount)%4)%4;
  310. }
  311. show[2]=TRUE;
  312. }
  313. void CWaterMarkDlg::WaterBitmap2Data()//将水印图象的转换为数据保存在二维数组m_watermarkdata
  314. {
  315. int i,j;
  316. if(m_watermarkdata !=NULL)
  317. if(m_Wh!=0)
  318. for(i=0; i<m_Wh; i++)
  319. delete [] m_watermarkdata[i];
  320. else
  321. return ;
  322. if(watermark.GetLength()==0)
  323. {
  324. MessageBox("无效的水印文件");
  325. return ;
  326. }
  327. m_dib1->Open(watermark);if(m_dib1->GetBiBitCount()>8) MessageBox("目前仅对8位的水印信息有效");
  328. WaterWidth=m_dib1->GetWidth();
  329. WaterHeight=m_dib1->GetHeight();
  330. biBitCount1 = m_dib1->GetBiBitCount();
  331. BYTE *colorTable;
  332. colorTable =(BYTE *) m_dib1->m_pDibBits;
  333. int byteBitCount = m_dib1->GetBiBitCount()/8;
  334. m_watermarkdata =new RGBQUAD*[WaterHeight];
  335. for (int l=0;l<WaterHeight;l++)
  336. m_watermarkdata[l]=new RGBQUAD [WaterWidth];
  337. int count = 0;
  338. int num = 0;
  339. for(i=WaterHeight-1; i>=0; i--)
  340. {
  341. for(j=0; j<WaterWidth; j++)
  342. {
  343. m_watermarkdata[i][j].rgbBlue = colorTable[count++];
  344. m_watermarkdata[i][j].rgbGreen = colorTable[count++];
  345. m_watermarkdata[i][j].rgbRed = colorTable[count++];
  346. m_watermarkdata[i][j].rgbReserved = 0;
  347. count += byteBitCount-3;
  348. num+=1; //TRACE("\n@%d @%d @%d",m_watermarkdata[i][j].rgbRed,m_watermarkdata[i][j].rgbBlue,m_watermarkdata[i][j].rgbRed);//
  349. }
  350. count += (4-(WaterWidth*byteBitCount)%4)%4;
  351. }
  352. }
  353. //转换公式为Gray(i,j)=0.11*R(i,j)+0.59*G(i,j)+0.3*B(i,j),
  354. //其中Gray(i,j)为转换后的黑白图像在(i,j)点处的灰度值,
  355. //其中绿色所占的比重最大,所以转换时可以直接使用G值。
  356. void CWaterMarkDlg::OnWaterEmbed()
  357. {
  358. // TODO: Add your control notification handler code here
  359. if ((bm[0].bmHeight*bm[0].bmWidth)<3*(bm[1].bmHeight*bm[1].bmWidth))
  360. {
  361. //不能嵌入水印
  362. MessageBox("水印图象过大请更换较大宿主图象","错误",MB_OK);
  363. Mark=FALSE;
  364. }
  365. if(!Mark)
  366. return ;
  367. if(biBitCount<biBitCount1)
  368. {
  369. MessageBox("水印受限制");
  370. return ;
  371. }
  372. RGBQUAD *m_copymater;//定义一个水印信息的数据副本
  373. m_copymater = new RGBQUAD [WaterHeight*WaterWidth];
  374. int i,j;
  375. int count=0;
  376. BYTE m_r,m_b;
  377. if(m_waterEmbed !=NULL)//m_waterEmbed是水印.嵌入后.的位图数据全局变量
  378. if(m_Ih!=0)
  379. for(int l =0;l<m_Ih;l++)
  380. delete [] m_waterEmbed[l];//?
  381. m_waterEmbed = new RGBQUAD*[ImageHeight];
  382. for(i=0;i<ImageHeight;i++)
  383. m_waterEmbed[i] = new RGBQUAD [ImageWidth];
  384. for(i=0;i<ImageHeight;i++)//将原来的图象的数据复制到m_waterEmbed里
  385. for(j=0;j<ImageWidth;j++)
  386. m_waterEmbed[i][j]=m_tOriPixelArray[i][j];
  387. for(i=0;i<WaterHeight;i++)//将水印信息的数据复制到一维数组m_copymater水印信息的数据副本里
  388. for(j=0;j<WaterWidth;j++)
  389. {
  390. m_copymater[i*WaterHeight+j]=m_watermarkdata[i][j];
  391. count++;
  392. }
  393. number=count;//记录嵌入的点数
  394. //
  395. int k=0;
  396. if (biBitCount==24)//Operate_Byte()嵌入水印信息位操作
  397. {
  398. for(i=0;i<ImageHeight&&count>0;i++)
  399. for(j=0;j<ImageWidth&&count>0;j++,count--)//下面可以用一条语句的 但为了测试还是多句
  400. {
  401. m_r=m_waterEmbed[i][j].rgbRed;
  402. m_b=m_waterEmbed[i][j].rgbBlue;
  403. BYTE &a=m_r;
  404. BYTE &b=m_b;
  405. BYTE c=m_copymater[i*ImageHeight+j].rgbGreen;
  406. Operate_Byte(a,b,c);
  407. m_waterEmbed[i][j].rgbRed=m_r;
  408. m_waterEmbed[i][j].rgbBlue=m_b;
  409. //TRACE("\n%d",c);k++;
  410. }
  411. PutInWaterMessage();//
  412. }
  413. else
  414. return;
  415. //TRACE("\n次数是%d",k);
  416. CDC *dc=GetDC();//将象素打印出来
  417. RECT m_rect;
  418. float w,h,p,q;
  419. GetDlgItem(IDC_STATIC2)->GetClientRect(&m_rect);
  420. if(m_rect.right<ImageWidth&&m_rect.bottom<ImageHeight)
  421. {
  422. w=(float)m_rect.right/(float)ImageWidth;
  423. h=(float)m_rect.bottom/(float)ImageHeight;
  424. }
  425. else
  426. w=h=1;
  427. for( p=0,i=0;i<ImageHeight;i++,p++)
  428. for( j=0, q=0;j<ImageWidth;j++,q++)
  429. dc->SetPixel(2+(int)(q*w),235+(int)(p*h),RGB(m_waterEmbed[i][j].rgbRed,
  430. m_waterEmbed[i][j].rgbGreen,m_waterEmbed[i][j].rgbBlue));
  431. ReleaseDC(dc);
  432. delete [] m_copymater;
  433. m_Ih=ImageHeight;//为保证直接更换水印信息而没有更换宿主图象提供原宽高 void error bug
  434. m_Iw=ImageWidth;
  435. m_Wh=WaterHeight;
  436. m_Ww=WaterWidth;
  437. TRACE("\n%d\n",ImageHeight);
  438. GetDlgItem(IDC_BUTTON4)->EnableWindow(TRUE);
  439. GetDlgItem(IDC_BUTTON5)->EnableWindow(TRUE);
  440. }
  441. void CWaterMarkDlg::ShowBitmap(CString filename)
  442. {
  443. }
  444. void CWaterMarkDlg::SetBitmap2Pixel()
  445. {
  446. if(ImageWidth*ImageHeight==0)
  447. return ;
  448. CDC *dc=GetDC();
  449. for(int i=0;i<ImageHeight;i++)
  450. for(int j=0;j<ImageWidth;j++)
  451. dc->SetPixel(5+j,i+220,RGB(m_tOriPixelArray[i][j].rgbRed,
  452. m_tOriPixelArray[i][j].rgbGreen,m_tOriPixelArray[i][j].rgbBlue));
  453. ReleaseDC(dc);
  454. }
  455. void CWaterMarkDlg::SetWaterBitmap2Pixel()
  456. {
  457. if(ImageWidth*ImageHeight==0)
  458. return ;
  459. CDC *dc1=GetDC();
  460. for(int i=0;i<WaterHeight;i++)
  461. for(int j=0;j<WaterWidth;j++)
  462. dc1->SetPixel(5+j,i+220,RGB(m_watermarkdata[i][j].rgbRed,
  463. m_watermarkdata[i][j].rgbGreen,m_watermarkdata[i][j].rgbBlue));
  464. ReleaseDC(dc1);
  465. }
  466. void CWaterMarkDlg::OnTimer(UINT nIDEvent)
  467. {
  468. // TODO: Add your message handler code here and/or call default
  469. if (show[0])
  470. {
  471. ::StretchBlt(hDesDC[0], rect[0].left, rect[0].top, rect[0].right, \
  472. rect[0].bottom, hSrcDC[0], 0, 0, bm[0].bmWidth, bm[0].bmHeight,+SRCCOPY);
  473. }
  474. if (show[1])
  475. {
  476. ::StretchBlt(hDesDC[1], rect[1].left, rect[1].top, rect[1].right, \
  477. rect[1].bottom, hSrcDC[1], 0, 0, bm[1].bmWidth, bm[1].bmHeight,+SRCCOPY);
  478. }
  479. if(show[2])
  480. {
  481. ::StretchBlt(hDesDC[2], rect[2].left, rect[2].top, rect[2].right, \
  482. rect[2].bottom, hSrcDC[2], 0, 0, bm[2].bmWidth, bm[2].bmHeight,+SRCCOPY);
  483. }
  484. CDialog::OnTimer(nIDEvent);
  485. }
  486. void CWaterMarkDlg::Operate_Byte(BYTE &operate1, BYTE &operate2, BYTE m_operate)
  487. {
  488. //将m_operate最低四位取出来放到L里最高四位放到H里,
  489. //最后将H赋值给operate1最后四位 L赋值给operate2最后四位
  490. int i;
  491. BYTE M=240,t=m_operate;//1111 0000
  492. BYTE H,L,x[8],y[4];
  493. y[3]=8;y[2]=4;y[1]=2;y[0]=1;
  494. for(i=0;i<8;i++)
  495. {
  496. x[i]=m_operate&1;
  497. m_operate>>=1;
  498. }
  499. H=x[7]*y[3]+x[6]*y[2]+x[5]*y[1]+x[4]*y[0];//hight
  500. L=x[3]*y[3]+x[2]*y[2]+x[1]*y[1]+x[0]*y[0];//low
  501. operate1&=M;
  502. operate2&=M;
  503. operate1+=H;
  504. operate2+=L;
  505. }
  506. void CWaterMarkDlg::OnPick_Up()
  507. {
  508. // TODO: Add your control notification handler code here
  509. PickUpWaterMessage(m_Wh,m_Ww);
  510. }
  511. void CWaterMarkDlg::PickUpWaterMessage(long h,long w)
  512. {
  513. int i,j;
  514. RGBQUAD *m_color;
  515. m_color = new RGBQUAD[h*w];
  516. int count=number;
  517. if(m_Ih!=0)
  518. for(i=0;i<m_Ih&&count>0;i++)
  519. for(j=0;j<m_Iw&&count>0;j++,count--)
  520. m_color[i*m_Ih+j]=m_waterEmbed[i][j];
  521. if(m_pickupwater!=NULL)
  522. if(m_pwh!=0)
  523. for(i=0;i<m_pwh;i++)
  524. delete [] m_pickupwater[i];
  525. m_pickupwater = new RGBQUAD *[h];
  526. for(i=0;i<h;i++)
  527. m_pickupwater[i] = new RGBQUAD[w];
  528. count=number;
  529. int k=0;
  530. for(i=0;i<h&&count>0;i++)
  531. for(j=0;j<w&&count>0;j++,count--)
  532. {
  533. m_pickupwater[i][j].rgbGreen=Operate_ByteOut(m_color[i*h+j].rgbRed,m_color[i*h+j].rgbBlue);
  534. m_pickupwater[i][j].rgbRed=m_pickupwater[i][j].rgbGreen;
  535. m_pickupwater[i][j].rgbBlue=m_pickupwater[i][j].rgbGreen;
  536. //TRACE("\n#%d\n",m_pickupwater[i][j].rgbGreen);k++;
  537. }
  538. //TRACE("\n提取的次数是%d\n",k);
  539. CDC *dc=GetDC();//将象素打印出来
  540. RECT m_rect;
  541. GetDlgItem(IDC_STATIC3)->GetClientRect(&m_rect);
  542. for( i=0;i<h;i++)
  543. for( j=0;j<w;j++)
  544. dc->SetPixel(m_rect.right+2+j,235+i,RGB(m_pickupwater[i][j].rgbRed,
  545. m_pickupwater[i][j].rgbGreen,m_pickupwater[i][j].rgbBlue));
  546. ReleaseDC(dc);
  547. m_pwh=h;//获得最后一次水印的信息位图的宽高信息
  548. m_pww=w;
  549. }
  550. void CWaterMarkDlg::PutInWaterMessage()
  551. {
  552. long a,b,c,d;
  553. a = ImageHeight;
  554. b = ImageWidth;
  555. c = WaterHeight;
  556. d = WaterWidth;
  557. }
  558. BYTE CWaterMarkDlg::Operate_ByteOut(BYTE operate1, BYTE operate2)
  559. {
  560. //相对与Operate_Byte()
  561. int i = 0;
  562. BYTE x[8],y[8],data;
  563. for (i=0;i<4;i++)
  564. {
  565. x[i]=operate2&1;
  566. operate2>>=1;//取出最低的四位
  567. }
  568. for (i=4;i<8;i++)
  569. {
  570. x[i]=operate1&1;
  571. operate1>>=1;//取出最高的四位
  572. }
  573. y[0]=1;y[1]=2;y[2]=4;y[3]=8;y[4]=16;y[5]=32;y[6]=64;y[7]=128;
  574. 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];
  575. return data;
  576. }
  577. void CWaterMarkDlg::OnSaveWEBitmap()
  578. {
  579. // TODO: Add your control notification handler code here
  580. CFileDialog dlg(FALSE,"未标题bmp","未标题.bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"位图文件(*.bmp)|*.bmp");
  581. if(dlg.DoModal()!=IDOK)
  582. return ;
  583. CString s;
  584. CDib *m_dib_save;
  585. m_dib_save = new CDib();
  586. m_dib_save = m_dib;
  587. s=dlg.GetPathName();
  588. int i,j;
  589. int byteBitCount = m_dib_save->GetBiBitCount()/8;
  590. long H=m_dib_save->GetHeight();
  591. long W=m_dib_save->GetWidth();
  592. int ct =0;
  593. for(i=H-1;i>=0;i--)//0.11*R(i,j)+0.59*G(i,j)+0.3*B(i,j)
  594. {
  595. for(j=0;j<W;j++)
  596. {
  597. ((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbBlue;
  598. ((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbGreen;
  599. ((BYTE*) m_dib_save->m_pDibBits)[ct++] = m_waterEmbed[i][j].rgbRed;
  600. ct +=byteBitCount -3;
  601. }
  602. ct +=(4-(W*byteBitCount)%4)%4;
  603. }
  604. m_dib_save->Save(s);
  605. }
  606. void CWaterMarkDlg::OnOpenW_E_Bitmap()
  607. {
  608. // TODO: Add your control notification handler code here
  609. CFileDialog dlg(TRUE, "", NULL,
  610. OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_ALLOWMULTISELECT,
  611. "(*.bmp)|*.bmp|所有文件(*.*)|*.*||",AfxGetMainWnd());
  612. if(dlg.DoModal()!=IDOK)
  613. return ;
  614. if(hwnd[2]!=NULL)
  615. hwnd[2]=NULL;
  616. if(hSrcDC[2]!=NULL)
  617. hSrcDC[2]=NULL;
  618. if(hDesDC[2]!=NULL)
  619. hDesDC[2]=NULL;
  620. hwnd[2] = GetDlgItem(IDC_STATIC2);
  621. hDesDC[2] = hwnd[2]->GetDC()->m_hDC;
  622. hSrcDC[2] = CreateCompatibleDC(hDesDC[2]);
  623. CString file;
  624. file=dlg.GetPathName();
  625. POSITION pos = dlg.GetStartPosition();
  626. file = dlg.GetNextPathName(pos);
  627. Embed2Data( file);
  628. hBitmap[2]=(HBITMAP)LoadImage(AfxGetInstanceHandle(),file,IMAGE_BITMAP,0,0,LR_LOADFROMFILE|LR_CREATEDIBSECTION);
  629. GetObject(hBitmap[2], sizeof BITMAP, &bm[2]);
  630. SelectObject(hSrcDC[2], hBitmap[2]);
  631. hwnd[2]->GetClientRect(&rect[2]);
  632. ::SetStretchBltMode(hDesDC[2],COLORONCOLOR);
  633. ::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);
  634. show[2]=TRUE;
  635. SetTimer(NULL,50,0);
  636. }
  637. void CWaterMarkDlg::Embed2Data(CString m_filename)//需要加入代码避免错误
  638. {
  639. int i,j,H,W;
  640. m_dib->Open(m_filename);
  641. H = m_dib->GetHeight();
  642. W = m_dib->GetWidth();
  643. if(m_waterEmbed !=NULL)//m_waterEmbed是水印.嵌入后.的位图数据全局变量
  644. if(m_Ih!=0)
  645. for(int l =0;l<m_Ih;l++)
  646. delete [] m_waterEmbed[l];//?
  647. else
  648. {
  649. MessageBox("Please add right code!");
  650. return;
  651. }
  652. m_waterEmbed = new RGBQUAD*[H];
  653. for(i=0;i<H;i++)
  654. m_waterEmbed[i] = new RGBQUAD [W];
  655. BYTE *color ;
  656. color =(BYTE*) m_dib->m_pDibBits;
  657. int byteBitCount = m_dib->GetBiBitCount()/8;
  658. int count =0;
  659. for(i=H-1;i>=0;i--)
  660. {
  661. for(j=0;j<W;j++)
  662. {
  663. m_waterEmbed[i][j].rgbBlue = color[count++];
  664. m_waterEmbed[i][j].rgbGreen = color[count++];
  665. m_waterEmbed[i][j].rgbRed = color[count++];
  666. m_waterEmbed[i][j].rgbReserved=0;
  667. count +=byteBitCount -3;
  668. }
  669. count +=(4-(W*byteBitCount)%4)%4;
  670. }
  671. }