131 lines
3.5 KiB
C++
131 lines
3.5 KiB
C++
// ScreenSpy.cpp: implementation of the CScreenSpy class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#include "stdafx.h"
|
|
#include "ScreenSpy.h"
|
|
#include "Common.h"
|
|
#include <stdio.h>
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Construction/Destruction
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
CScreenSpy::CScreenSpy(ULONG ulbiBitCount, BYTE algo, int gop) : ScreenCapture(algo)
|
|
{
|
|
m_GOP = gop;
|
|
int m_ulbiBitCount = (ulbiBitCount == 16 || ulbiBitCount == 32) ? ulbiBitCount : 16;
|
|
|
|
m_BitmapInfor_Full = new BITMAPINFO();
|
|
memset(m_BitmapInfor_Full, 0, sizeof(BITMAPINFO));
|
|
BITMAPINFOHEADER* BitmapInforHeader = &(m_BitmapInfor_Full->bmiHeader);
|
|
BitmapInforHeader->biSize = sizeof(BITMAPINFOHEADER);
|
|
BitmapInforHeader->biWidth = m_ulFullWidth; //1080
|
|
BitmapInforHeader->biHeight = m_ulFullHeight; //1920
|
|
BitmapInforHeader->biPlanes = 1;
|
|
BitmapInforHeader->biBitCount = ulbiBitCount; //ͨ³£Îª32
|
|
BitmapInforHeader->biCompression = BI_RGB;
|
|
BitmapInforHeader->biSizeImage =
|
|
((BitmapInforHeader->biWidth * BitmapInforHeader->biBitCount + 31) / 32) * 4 * BitmapInforHeader->biHeight;
|
|
|
|
m_hDeskTopWnd = GetDesktopWindow();
|
|
m_hFullDC = GetDC(m_hDeskTopWnd);
|
|
|
|
m_BitmapData_Full = NULL;
|
|
m_hFullMemDC = CreateCompatibleDC(m_hFullDC);
|
|
m_BitmapHandle = ::CreateDIBSection(m_hFullDC, m_BitmapInfor_Full, DIB_RGB_COLORS, &m_BitmapData_Full, NULL, NULL);
|
|
::SelectObject(m_hFullMemDC, m_BitmapHandle);
|
|
m_FirstBuffer = (LPBYTE)m_BitmapData_Full;
|
|
|
|
m_DiffBitmapData_Full = NULL;
|
|
m_hDiffMemDC = CreateCompatibleDC(m_hFullDC);
|
|
m_DiffBitmapHandle = ::CreateDIBSection(m_hFullDC, m_BitmapInfor_Full, DIB_RGB_COLORS, &m_DiffBitmapData_Full, NULL, NULL);
|
|
::SelectObject(m_hDiffMemDC, m_DiffBitmapHandle);
|
|
|
|
m_RectBufferOffset = 0;
|
|
m_RectBuffer = new BYTE[m_BitmapInfor_Full->bmiHeader.biSizeImage * 2];
|
|
}
|
|
|
|
|
|
CScreenSpy::~CScreenSpy()
|
|
{
|
|
if (m_BitmapInfor_Full != NULL)
|
|
{
|
|
delete m_BitmapInfor_Full;
|
|
m_BitmapInfor_Full = NULL;
|
|
}
|
|
|
|
ReleaseDC(m_hDeskTopWnd, m_hFullDC);
|
|
|
|
if (m_hFullMemDC!=NULL)
|
|
{
|
|
DeleteDC(m_hFullMemDC);
|
|
|
|
::DeleteObject(m_BitmapHandle);
|
|
if (m_BitmapData_Full!=NULL)
|
|
{
|
|
m_BitmapData_Full = NULL;
|
|
}
|
|
|
|
m_hFullMemDC = NULL;
|
|
}
|
|
|
|
if (m_hDiffMemDC!=NULL)
|
|
{
|
|
DeleteDC(m_hDiffMemDC);
|
|
|
|
::DeleteObject(m_DiffBitmapHandle);
|
|
if (m_DiffBitmapData_Full!=NULL)
|
|
{
|
|
m_DiffBitmapData_Full = NULL;
|
|
}
|
|
}
|
|
|
|
if (m_RectBuffer)
|
|
{
|
|
delete[] m_RectBuffer;
|
|
m_RectBuffer = NULL;
|
|
}
|
|
|
|
m_RectBufferOffset = 0;
|
|
}
|
|
|
|
LPBYTE CScreenSpy::GetFirstScreenData(ULONG* ulFirstScreenLength)
|
|
{
|
|
//ÓÃÓÚ´ÓÔÉ豸Öи´ÖÆλͼµ½Ä¿±êÉ豸
|
|
::BitBlt(m_hFullMemDC, 0, 0, m_ulFullWidth, m_ulFullHeight, m_hFullDC, 0, 0, SRCCOPY);
|
|
|
|
m_RectBuffer[0] = TOKEN_FIRSTSCREEN;
|
|
memcpy(1 + m_RectBuffer, m_BitmapData_Full, m_BitmapInfor_Full->bmiHeader.biSizeImage);
|
|
if (m_bAlgorithm == ALGORITHM_GRAY) {
|
|
ToGray(1 + m_RectBuffer, 1 + m_RectBuffer, m_BitmapInfor_Full->bmiHeader.biSizeImage);
|
|
}
|
|
*ulFirstScreenLength = m_BitmapInfor_Full->bmiHeader.biSizeImage;
|
|
|
|
return m_RectBuffer; //ÄÚ´æ
|
|
}
|
|
|
|
|
|
VOID CScreenSpy::ScanScreen(HDC hdcDest, HDC hdcSour, ULONG ulWidth, ULONG ulHeight)
|
|
{
|
|
AUTO_TICK(70);
|
|
#if COPY_ALL
|
|
BitBlt(hdcDest, 0, 0, ulWidth, ulHeight, hdcSour, 0, 0, SRCCOPY);
|
|
#else
|
|
const ULONG ulJumpLine = 50;
|
|
const ULONG ulJumpSleep = ulJumpLine / 10;
|
|
|
|
for (int i = 0, ulToJump = 0; i < ulHeight; i += ulToJump)
|
|
{
|
|
ULONG ulv1 = ulHeight - i;
|
|
|
|
if (ulv1 > ulJumpLine)
|
|
ulToJump = ulJumpLine;
|
|
else
|
|
ulToJump = ulv1;
|
|
BitBlt(hdcDest, 0, i, ulWidth, ulToJump, hdcSour,0, i, SRCCOPY);
|
|
Sleep(ulJumpSleep);
|
|
}
|
|
#endif
|
|
}
|