开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 书籍源码 > 查看源码
CaptureDlg.cpp
资源名称：c++时尚编程百例__1-20.rar　[点击查看]
上传用户：aokegd
上传日期：2009-12-14
资源大小：1276k
文件大小：9k
源码类别：

书籍源码

开发平台：

Visual C++

CaptureDlg.cpp：源码内容
  1. // CaptureDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Capture.h"
  6. #include "CaptureDlg.h"
  7. #include <windowsx.h>
  8. #pragma comment(lib,"hook.lib")
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  14. #define IDM_SHELL WM_USER+1
  15. BOOL __declspec(dllexport)__stdcall  AddHotkey(HWND,UCHAR key,UCHAR mask);
  16. BOOL __declspec(dllexport)__stdcall  DeleteHotkey(HWND,UCHAR key,UCHAR mask);
  17. /////////////////////////////////////////////////////////////////////////////
  18. // CAboutDlg dialog used for App About
  19. UCHAR Key_Table[]={0x78,0x79,0x7a,0x7b,0x6a,0x30,0x31,0x32,0x33,0x34,0x35,0x36,0x37,0x38,0x39};
  20. class CAboutDlg : public CDialog
  21. {
  22. public:
  23. CAboutDlg();
  25. // Dialog Data
  26. //{{AFX_DATA(CAboutDlg)
  27. enum { IDD = IDD_ABOUTBOX };
  28. //}}AFX_DATA
  30. // ClassWizard generated virtual function overrides
  31. //{{AFX_VIRTUAL(CAboutDlg)
  32. protected:
  33. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  34. //}}AFX_VIRTUAL
  36. // Implementation
  37. protected:
  38. //{{AFX_MSG(CAboutDlg)
  39. //}}AFX_MSG
  40. DECLARE_MESSAGE_MAP()
  41. };
  43. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  44. {
  45. //{{AFX_DATA_INIT(CAboutDlg)
  46. //}}AFX_DATA_INIT
  47. }
  49. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  50. {
  51. CDialog::DoDataExchange(pDX);
  52. //{{AFX_DATA_MAP(CAboutDlg)
  53. //}}AFX_DATA_MAP
  54. }
  56. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  57. //{{AFX_MSG_MAP(CAboutDlg)
  58. // No message handlers
  59. //}}AFX_MSG_MAP
  60. END_MESSAGE_MAP()
  62. /////////////////////////////////////////////////////////////////////////////
  63. // CCaptureDlg dialog
  65. CCaptureDlg::CCaptureDlg(CWnd* pParent /*=NULL*/)
  66. : CDialog(CCaptureDlg::IDD, pParent)
  67. {
  68. //{{AFX_DATA_INIT(CCaptureDlg)
  69. m_bControl = FALSE;
  70. m_bAlt = FALSE;
  71. m_bShift = FALSE;
  72. m_Path = _T("c:\");
  73. m_Number = _T("0 picture captured.");
  74. nCount=0;
  75. bRegistered=FALSE;
  76. bTray=FALSE;
  77. //}}AFX_DATA_INIT
  78. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  79. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  80. }
  82. void CCaptureDlg::DoDataExchange(CDataExchange* pDX)
  83. {
  84. CDialog::DoDataExchange(pDX);
  85. //{{AFX_DATA_MAP(CCaptureDlg)
  86. DDX_Control(pDX, IDC_KEY, m_Key);
  87. DDX_Check(pDX, IDC_CONTROL, m_bControl);
  88. DDX_Check(pDX, IDC_ALT, m_bAlt);
  89. DDX_Check(pDX, IDC_SHIFT, m_bShift);
  90. DDX_Text(pDX, IDC_PATH, m_Path);
  91. DDX_Text(pDX, IDC_NUMBER, m_Number);
  92. //}}AFX_DATA_MAP
  93. }
  95. BEGIN_MESSAGE_MAP(CCaptureDlg, CDialog)
  96. //{{AFX_MSG_MAP(CCaptureDlg)
  97. ON_WM_SYSCOMMAND()
  98. ON_WM_PAINT()
  99. ON_WM_QUERYDRAGICON()
  100. ON_BN_CLICKED(ID_ABOUT, OnAbout)
  101. ON_BN_CLICKED(IDC_BROWSE, OnBrowse)
  102. ON_BN_CLICKED(ID_CHANGE, OnChange)
  103. //}}AFX_MSG_MAP
  104. END_MESSAGE_MAP()
  106. /////////////////////////////////////////////////////////////////////////////
  107. // CCaptureDlg message handlers
  109. BOOL CCaptureDlg::OnInitDialog()
  110. {
  111. CDialog::OnInitDialog();
  113. // Add "About..." menu item to system menu.
  115. // IDM_ABOUTBOX must be in the system command range.
  116. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  117. ASSERT(IDM_ABOUTBOX < 0xF000);
  119. CMenu* pSysMenu = GetSystemMenu(FALSE);
  120. if (pSysMenu != NULL)
  121. {
  122. CString strAboutMenu;
  123. strAboutMenu.LoadString(IDS_ABOUTBOX);
  124. if (!strAboutMenu.IsEmpty())
  125. {
  126. pSysMenu->AppendMenu(MF_SEPARATOR);
  127. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  128. }
  129. }
  131. // Set the icon for this dialog.  The framework does this automatically
  132. //  when the application's main window is not a dialog
  133. SetIcon(m_hIcon, TRUE); // Set big icon
  134. SetIcon(m_hIcon, FALSE); // Set small icon
  135. m_Key.SetCurSel(0);
  136. RegisterHotkey();
  137. CMenu* pMenu=GetSystemMenu(FALSE);
  138. pMenu->DeleteMenu(SC_MAXIMIZE,MF_BYCOMMAND);
  139. pMenu->DeleteMenu(SC_SIZE,MF_BYCOMMAND);
  140. pMenu->DeleteMenu(SC_RESTORE,MF_BYCOMMAND);
  141. return TRUE;  // return TRUE  unless you set the focus to a control
  142. }
  144. void CCaptureDlg::OnSysCommand(UINT nID, LPARAM lParam)
  145. {
  146. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  147. {
  148. CAboutDlg dlgAbout;
  149. dlgAbout.DoModal();
  150. }
  151. else
  152. {
  153. CDialog::OnSysCommand(nID, lParam);
  154. }
  155. }
  157. // If you add a minimize button to your dialog, you will need the code below
  158. //  to draw the icon.  For MFC applications using the document/view model,
  159. //  this is automatically done for you by the framework.
  161. void CCaptureDlg::OnPaint() 
  162. {
  163. if (IsIconic())
  164. {
  165. CPaintDC dc(this); // device context for painting
  167. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  169. // Center icon in client rectangle
  170. int cxIcon = GetSystemMetrics(SM_CXICON);
  171. int cyIcon = GetSystemMetrics(SM_CYICON);
  172. CRect rect;
  173. GetClientRect(&rect);
  174. int x = (rect.Width() - cxIcon + 1) / 2;
  175. int y = (rect.Height() - cyIcon + 1) / 2;
  177. // Draw the icon
  178. dc.DrawIcon(x, y, m_hIcon);
  179. }
  180. else
  181. {
  182. CDialog::OnPaint();
  183. }
  184. }
  186. // The system calls this to obtain the cursor to display while the user drags
  187. //  the minimized window.
  188. HCURSOR CCaptureDlg::OnQueryDragIcon()
  189. {
  190. return (HCURSOR) m_hIcon;
  191. }
  193. void CCaptureDlg::OnCancel() 
  194. {
  195. if(bTray)
  196. DeleteIcon();
  197. CDialog::OnCancel();
  198. }
  200. void CCaptureDlg::OnAbout() 
  201. {
  202. CAboutDlg dlg;
  203. dlg.DoModal();
  204. }
  206. void CCaptureDlg::OnBrowse() 
  207. {
  208. CString str;
  209. BROWSEINFO bi;
  210. char name[MAX_PATH];
  211. ZeroMemory(&bi,sizeof(BROWSEINFO));
  212. bi.hwndOwner=GetSafeHwnd();
  213. bi.pszDisplayName=name;
  214. bi.lpszTitle="Select folder";
  215. bi.ulFlags=BIF_RETURNONLYFSDIRS;
  216. LPITEMIDLIST idl=SHBrowseForFolder(&bi);
  217. if(idl==NULL)
  218. return;
  219. SHGetPathFromIDList(idl,str.GetBuffer(MAX_PATH));
  220. str.ReleaseBuffer();
  221. m_Path=str;
  222. if(str.GetAt(str.GetLength()-1)!='\')
  223. m_Path+="\";
  224. UpdateData(FALSE);
  225. }
  227. void CCaptureDlg::SaveBmp()
  228. {
  229. CDC dc;
  230. dc.CreateDC("DISPLAY",NULL,NULL,NULL);
  231. CBitmap bm;
  232. int Width=GetSystemMetrics(SM_CXSCREEN);
  233. int Height=GetSystemMetrics(SM_CYSCREEN);
  234. bm.CreateCompatibleBitmap(&dc,Width,Height);
  235. CDC tdc;
  236. tdc.CreateCompatibleDC(&dc);
  237. CBitmap*pOld=tdc.SelectObject(&bm);
  238. tdc.BitBlt(0,0,Width,Height,&dc,0,0,SRCCOPY);
  239. tdc.SelectObject(pOld);
  240. BITMAP btm;
  241. bm.GetBitmap(&btm);
  242. DWORD size=btm.bmWidthBytes*btm.bmHeight;
  243. LPSTR lpData=(LPSTR)GlobalAllocPtr(GPTR,size);
  244. /////////////////////////////////////////////
  245. BITMAPINFOHEADER bih;
  246. bih.biBitCount=btm.bmBitsPixel;
  247. bih.biClrImportant=0;
  248. bih.biClrUsed=0;
  249. bih.biCompression=0;
  250. bih.biHeight=btm.bmHeight;
  251. bih.biPlanes=1;
  252. bih.biSize=sizeof(BITMAPINFOHEADER);
  253. bih.biSizeImage=size;
  254. bih.biWidth=btm.bmWidth;
  255. bih.biXPelsPerMeter=0;
  256. bih.biYPelsPerMeter=0;
  257. ///////////////////////////////////
  258. GetDIBits(dc,bm,0,bih.biHeight,lpData,(BITMAPINFO*)&bih,DIB_RGB_COLORS);
  259. // bm.GetBitmapBits(size,lpData); //此函数在处理5-5-5模式的16位色下会出现颜色混乱
  260. //////////////////////////////
  261. static int filecount=0;
  262. CString name;
  263. name.Format("pict%04d.bmp",filecount++);
  264. name=m_Path+name;
  265. BITMAPFILEHEADER bfh;
  266. bfh.bfReserved1=bfh.bfReserved2=0;
  267. bfh.bfType=((WORD)('M'<< 8)|'B');
  268. bfh.bfSize=54+size;
  269. bfh.bfOffBits=54;
  270. CFile bf;
  271. if(bf.Open(name,CFile::modeCreate|CFile::modeWrite)){
  272. bf.WriteHuge(&bfh,sizeof(BITMAPFILEHEADER));
  273. bf.WriteHuge(&bih,sizeof(BITMAPINFOHEADER));
  274. bf.WriteHuge(lpData,size);
  275. bf.Close();
  276. nCount++;
  277. }
  278. GlobalFreePtr(lpData);
  279. if(nCount==1)
  280. m_Number.Format("%d picture captured.",nCount);
  281. else
  282. m_Number.Format("%d pictures captured.",nCount);
  283. UpdateData(FALSE);
  284. }
  286. BOOL CCaptureDlg::PreTranslateMessage(MSG* pMsg) 
  287. {
  288.     if(pMsg -> message == WM_KEYDOWN)
  289. {
  290.         if(pMsg -> wParam == VK_ESCAPE)
  291. return TRUE;
  292. if(pMsg -> wParam == VK_RETURN)
  293. return TRUE;
  294. }
  295. return CDialog::PreTranslateMessage(pMsg);
  296. }
  298. LRESULT CCaptureDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
  299. {
  300. if(message==WM_HOTKEY&&lParam==WM_KEYDOWN){
  301. SaveBmp();
  302. return FALSE;
  303. }
  304. if(message==IDM_SHELL&&lParam==WM_RBUTTONUP){
  305. CMenu pop;
  306. pop.LoadMenu(MAKEINTRESOURCE(IDR_MENU1));
  307. CMenu*pMenu=pop.GetSubMenu(0);
  308. pMenu->SetDefaultItem(ID_EXITICON);
  309. CPoint pt;
  310. GetCursorPos(&pt);
  311. int id=pMenu->TrackPopupMenu(TPM_RIGHTALIGN|TPM_NONOTIFY|TPM_RETURNCMD|TPM_LEFTBUTTON,
  312. pt.x,pt.y,this);
  313. if(id==ID_EXITICON)
  314. DeleteIcon();
  315. else if(id==ID_EXIT)
  316. OnCancel();
  317. return FALSE;
  318. }
  319. LRESULT res= CDialog::WindowProc(message, wParam, lParam);
  320. if(message==WM_SYSCOMMAND&&wParam==SC_MINIMIZE)
  321. AddIcon();
  322. return res;
  323. }
  325. void CCaptureDlg::AddIcon()
  326. {
  327. NOTIFYICONDATA data;
  328. data.cbSize=sizeof(NOTIFYICONDATA);
  329. CString tip;
  330. tip.LoadString(IDS_ICONTIP);
  331. data.hIcon=GetIcon(0);
  332. data.hWnd=GetSafeHwnd();
  333. strcpy(data.szTip,tip);
  334. data.uCallbackMessage=IDM_SHELL;
  335. data.uFlags=NIF_ICON|NIF_MESSAGE |NIF_TIP ;
  336. data.uID=98;
  337. Shell_NotifyIcon(NIM_ADD,&data);
  338. ShowWindow(SW_HIDE);
  339. bTray=TRUE;
  340. }
  342. void CCaptureDlg::DeleteIcon()
  343. {
  344. NOTIFYICONDATA data;
  345. data.cbSize=sizeof(NOTIFYICONDATA);
  346. data.hWnd=GetSafeHwnd();
  347. data.uID=98;
  348. Shell_NotifyIcon(NIM_DELETE,&data);
  349. ShowWindow(SW_SHOW);
  350. SetForegroundWindow();
  351. ShowWindow(SW_SHOWNORMAL);
  352. bTray=FALSE;
  353. }
  355. void CCaptureDlg::OnChange() 
  356. {
  357. RegisterHotkey();
  358. }
  360. BOOL CCaptureDlg::RegisterHotkey()
  361. {
  362. UpdateData();
  363. UCHAR mask=0;
  364. UCHAR key=0;
  365. if(m_bControl)
  366. mask|=4;
  367. if(m_bAlt)
  368. mask|=2;
  369. if(m_bShift)
  370. mask|=1;
  371. key=Key_Table[m_Key.GetCurSel()];
  372. if(bRegistered){
  373. DeleteHotkey(GetSafeHwnd(),cKey,cMask);
  374. bRegistered=FALSE;
  375. }
  376. cMask=mask;
  377. cKey=key;
  378. bRegistered=AddHotkey(GetSafeHwnd(),cKey,cMask);
  379. return bRegistered;
  380. }