C++实现屏幕截图
更新时间:2020年4月25日 17:27 点击:2190
上回分享了一个全屏截图的代码,保存为BMP,参考:C++实现屏幕截图(全屏截图)
实际使用的过程中我发现截图文件实在大,无奈又整成了PNG截图,现在分享出来。
MakePNG.h
//MakePNG.h #pragma once #include <GdiPlus.h> using namespace Gdiplus; #pragma comment(lib,"GdiPlus.lib") class CMakePNG { public: CMakePNG(void); ~CMakePNG(void); BOOL MakePNG(HDC hDC,CRect rect,CString strFilePath); BOOL BMptoPNG(LPCWSTR StrBMp,LPCWSTR StrPNG); BOOL PNGtoBMp(LPCWSTR StrPNG,LPCWSTR StrBMp); BOOL GetEncoderClsid(WCHAR* pFormat,CLSID* pClsid); private: GdiplusStartupInput m_gdiplusStartupInput; ULONG_PTR m_pGdiToken; };
MakePNG.cpp
//MakePNG.cpp #include "StdAfx.h" #include "MakePNG.h" CMakePNG::CMakePNG(void) { GdiplusStartup(&m_pGdiToken,&m_gdiplusStartupInput,NULL); } CMakePNG::~CMakePNG(void) { } /***************************************************************************/ /* 功能: 根据rect屏幕抓图,保存为文件名为strFilePath的PNG图像文件 */ /* 输入参数: HDC hDC 屏幕HDC; */ /* CRect rect 需要的矩形; */ /* CString strFilePath 保存文件全路径(含后缀名); */ /***************************************************************************/ BOOL CMakePNG::MakePNG(HDC hDC, CRect rect, CString strFilePath) { BITMAP bmp; PBITMAPINFO pbmi; PBITMAPINFOHEADER pbih; // bitmap info-header BITMAPFILEHEADER hdr; // bitmap file-header WORD cClrBits; LPBYTE lpBits; // memory pointer DWORD dwTmp; DWORD cb; // incremental count of bytes BYTE *hp; // byte pointer HANDLE hfile; // file handle CString szBMPFilename = strFilePath.Left(strFilePath.GetLength() - 3) + _T("bmp");//先保存成位图 HDC hdcCompatible = CreateCompatibleDC(hDC); HBITMAP hbmScreen = CreateCompatibleBitmap(hDC, rect.Width(), rect.Height()); if (hbmScreen == NULL) { AfxMessageBox(_T("CreateCompatibleBitmap() error")); return FALSE; } // Select the bitmaps into the compatible DC. if (!SelectObject(hdcCompatible, hbmScreen)) { AfxMessageBox(_T("Compatible Bitmap Selection error")); return FALSE; } //Copy color data for the entire display into a //bitmap that is selected into a compatible DC. if (!BitBlt(hdcCompatible, 0,0, rect.Width(), rect.Height(), hDC, rect.left,rect.top, SRCCOPY)) { AfxMessageBox(_T("Screen to Compat Blt Failed")); return FALSE; } // Retrieve the bitmap's color format, width, and height. if (!GetObject(hbmScreen, sizeof(BITMAP), (LPSTR)&bmp)) { AfxMessageBox(_T("GetObject()出错!")); return FALSE; } // Convert the color format to a count of bits. cClrBits = (WORD)(bmp.bmPlanes * bmp.bmBitsPixel); if (cClrBits == 1) cClrBits = 1; else if (cClrBits <= 4) cClrBits = 4; else if (cClrBits <= 8) cClrBits = 8; else if (cClrBits <= 16) cClrBits = 16; else if (cClrBits <= 24) cClrBits = 24; else cClrBits = 32; // Allocate memory for the BITMAPINFO structure. (This structure // contains a BITMAPINFOHEADER structure and an array of RGBQUAD // data structures.) if (cClrBits != 24) pbmi = (PBITMAPINFO) LocalAlloc(LPTR, sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * (1<< cClrBits)); // There is no RGBQUAD array for the 24-bit-per-pixel format. else pbmi = (PBITMAPINFO) LocalAlloc(LPTR, sizeof(BITMAPINFOHEADER)); // Initialize the fields in the BITMAPINFO structure. pbmi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER); pbmi->bmiHeader.biWidth = bmp.bmWidth; pbmi->bmiHeader.biHeight = bmp.bmHeight; pbmi->bmiHeader.biPlanes = bmp.bmPlanes; pbmi->bmiHeader.biBitCount = bmp.bmBitsPixel; if (cClrBits < 24) pbmi->bmiHeader.biClrUsed = (1<<cClrBits); // If the bitmap is not compressed, set the BI_RGB flag. pbmi->bmiHeader.biCompression = BI_RGB; // Compute the number of bytes in the array of color // indices and store the result in biSizeImage. pbmi->bmiHeader.biSizeImage = ((pbmi->bmiHeader.biWidth * cClrBits +31) & ~31) /8 * pbmi->bmiHeader.biHeight; // Set biClrImportant to 0, indicating that all of the device colors are important. pbmi->bmiHeader.biClrImportant = 0; pbih = (PBITMAPINFOHEADER) pbmi; lpBits = (LPBYTE) GlobalAlloc(GMEM_FIXED, pbih->biSizeImage); if (!lpBits) { AfxMessageBox(_T("内存分配错误!")); return FALSE; } // Retrieve the color table (RGBQUAD array) and the bits // (array of palette indices) from the DIB. if (!GetDIBits(hDC, hbmScreen, 0, (WORD) pbih->biHeight, lpBits, pbmi, DIB_RGB_COLORS)) { AfxMessageBox(_T("GetDIBits() error")); return FALSE; } // Create the .BMP file. hfile = CreateFile(szBMPFilename, GENERIC_READ | GENERIC_WRITE, (DWORD) 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, (HANDLE) NULL); if (hfile == INVALID_HANDLE_VALUE) { AfxMessageBox(_T("创建文件失败")); return false; } hdr.bfType = 0x4d42; // 0x42 = "B" 0x4d = "M" // Compute the size of the entire file. hdr.bfSize = (DWORD) (sizeof(BITMAPFILEHEADER) + pbih->biSize + pbih->biClrUsed * sizeof(RGBQUAD) + pbih->biSizeImage); hdr.bfReserved1 = 0; hdr.bfReserved2 = 0; // Compute the offset to the array of color indices. hdr.bfOffBits = (DWORD) sizeof(BITMAPFILEHEADER) + pbih->biSize + pbih->biClrUsed * sizeof (RGBQUAD); // Copy the BITMAPFILEHEADER into the .BMP file. if (!WriteFile(hfile, (LPVOID) &hdr, sizeof(BITMAPFILEHEADER), (LPDWORD) &dwTmp, NULL)) { AfxMessageBox(_T("写BMP文件头失败")); return FALSE; } // Copy the BITMAPINFOHEADER and RGBQUAD array into the file. if (!WriteFile(hfile, (LPVOID) pbih, sizeof(BITMAPINFOHEADER) + pbih->biClrUsed * sizeof (RGBQUAD), (LPDWORD) &dwTmp, ( NULL))) { AfxMessageBox(_T("写BMP文件头失败")); return FALSE; } // Copy the array of color indices into the .BMP file. cb = pbih->biSizeImage; hp = lpBits; if (!WriteFile(hfile, (LPSTR) hp, (int) cb, (LPDWORD) &dwTmp,NULL)) { AfxMessageBox(_T("写入BMP文件失败")); return FALSE; } // Close the .BMP file. if (!CloseHandle(hfile)) { AfxMessageBox(_T("Can't close BMP file.")); } // Free memory. GlobalFree((HGLOBAL)lpBits); //转换成PNG if(!BMptoPNG(szBMPFilename,strFilePath)) { DeleteFile(szBMPFilename); return FALSE; } DeleteObject(hbmScreen); DeleteFile(szBMPFilename); return TRUE; } // //转换BMP文件为PNG文件 BOOL CMakePNG::BMptoPNG(LPCWSTR StrBMp,LPCWSTR StrPNG) { CLSID encoderClsid; Status stat; Image* image = NULL; image = Bitmap::FromFile(StrBMp,TRUE); if (!GetEncoderClsid(L"image/png",&encoderClsid)) { return FALSE; } stat = image->Save(StrPNG,&encoderClsid,NULL); if (stat != Ok) { return FALSE; } delete image; return TRUE; } // 功能描述: 转换PNG文件为BMP文件 BOOL CMakePNG::PNGtoBMp(LPCWSTR StrPNG,LPCWSTR StrBMp) { CLSID encoderClsid; Status stat; Image* pImage; pImage = Bitmap::FromFile(StrPNG,TRUE); if (!GetEncoderClsid(L"image/bmp",&encoderClsid)) { return FALSE; } stat = pImage->Save(StrBMp,&encoderClsid,NULL); if (stat != Ok) { return FALSE; } delete pImage; return TRUE; } BOOL CMakePNG::GetEncoderClsid(WCHAR* pFormat,CLSID* pClsid) { UINT num = 0,size = 0; ImageCodecInfo* pImageCodecInfo = NULL; GetImageEncodersSize(&num,&size); if (size == 0) { return FALSE; } pImageCodecInfo = (ImageCodecInfo*)(malloc(size)); if (pImageCodecInfo == NULL) { return FALSE; } GetImageEncoders(num,size,pImageCodecInfo); BOOL bfound = FALSE; for (UINT i = 0;!bfound && i < num; i++) { if (_wcsicmp(pImageCodecInfo[i].MimeType,pFormat) == 0) { *pClsid = pImageCodecInfo[i].Clsid; bfound = TRUE; } } free(pImageCodecInfo); return bfound; }
以上两个文件实际上是CMakePNG类,使用时需要把他们添加到项目中,调用方法如下:
wstring GetAppPathW() { wchar_t szExePath[MAX_PATH] = {0}; GetModuleFileNameW(NULL, szExePath, MAX_PATH); wchar_t *pstr = wcsrchr(szExePath, '\\'); memset(pstr + 1, 0, 2); wstring strAppPath(szExePath); return strAppPath; } // 屏幕截图 CString CDemoDlg::ScreenShot(void) { CWnd *pDesktop = GetDesktopWindow(); CDC *pDC = pDesktop->GetDC(); CRect rect; //获取窗口的大小 pDesktop->GetClientRect(&rect); //保存到的文件名 CString strFileName(GetAppPathW().c_str()); strFileName += _T("ScreenShot\\"); CreateDirectory((LPCTSTR)strFileName,NULL); CTime t = CTime::GetCurrentTime(); CString tt = t.Format("%Y%m%d_%H%M%S"); strFileName += tt; strFileName += _T(".PNG"); //保存为PNG CMakePNG MakePNG; MakePNG.MakePNG(pDC->m_hDC,rect,strFileName); ReleaseDC(pDC); return strFileName; }
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持猪先飞。
上一篇: C++常见获取随机数的方法小结
相关文章
- vector是表示可以改变大小的数组的序列容器,本文主要介绍了C++STL标准库std::vector的使用详解,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2022-03-06
- 这篇文章主要介绍了C++中取余运算的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-23
- 这篇文章主要介绍了C++ string常用截取字符串方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25
- 本文通过例子,讲述了C++调用C#的DLL程序的方法,作出了以下总结,下面就让我们一起来学习吧。...2020-06-25
- 本篇文章主要介绍了C++中四种加密算法之AES源代码,小编觉得挺不错的,现在分享给大家,也给大家做个参考。...2020-04-25
- 整数拆分,指把一个整数分解成若干个整数的和。本文重点给大家介绍C++ 整数拆分方法详解,非常不错,感兴趣的朋友一起学习吧...2020-04-25
- 这篇文章主要介绍了C++中Sort函数详细解析,sort函数是algorithm库下的一个函数,sort函数是不稳定的,即大小相同的元素在排序后相对顺序可能发生改变...2022-08-18
- 这篇文章主要介绍了C++万能库头文件在vs中的安装步骤(图文),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-23
- 这篇文章主要介绍了C++ bitset用法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25
- 本篇文章小编并不是为大家讲解string类型的用法,而是讲解我个人比较好奇的问题,就是string 类型占几个字节...2020-04-25
- 这篇文章主要为大家详细介绍了C++ Eigen库计算矩阵特征值及特征向量,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2020-04-25
- 这篇文章主要介绍了C++ pair的用法实例详解的相关资料,需要的朋友可以参考下...2020-04-25
- 这篇文章主要介绍了VSCode C++多文件编译的简单使用方法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-03-29
- 虽然C++11引入了智能指针的,但是开发人员在与内存的斗争问题上并没有解放,如果我门实用不当仍然有内存泄漏问题,其中智能指针的循环引用缺陷是最大的问题。下面通过实例代码给大家介绍c++中的循环引用,一起看看吧...2020-04-25
- 这篇文章主要给大家介绍了关于C++随机点名生成器的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25
- map容器是C++ STL中的重要一员,删除map容器中value为指定元素的问题是我们经常与遇到的一个问题,下面这篇文章主要给大家介绍了关于利用C++如何删除map容器中指定值的元素的相关资料,需要的朋友可以参考借鉴,下面来一起看看吧。...2020-04-25
- 这篇文章主要介绍了C++ 约瑟夫环问题案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下...2021-08-15
- 这篇文章主要介绍了C++中cin的用法详细,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25
- 本篇文章是对C++中的常见编译错误进行了详细的分析介绍,需要的朋友参考下...2020-04-25
- 这篇文章主要介绍了c++优先队列(priority_queue)用法详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25