admin/SimpleRemoter
1
0
Fork 0
mirror of https://github.com/yuanyuanxiang/SimpleRemoter.git synced 2026-01-21 23:13:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve: Using SSE2 to improve bitmap compare speed

Browse Source
This commit is contained in:
yuanyuanxiang
2025-11-30 08:53:31 +01:00
parent d6464b48b5
commit 8d4be0a580
4 changed files with 587 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

276
client/ScreenCapture.h
View File

@@ -1,4 +1,4 @@
#pragma once
#pragma once
#include "stdafx.h"
#include <assert.h>
#include "CursorInfo.h"
@@ -13,13 +13,14 @@
#include <condition_variable>
#include <functional>
#include <future>
#include <emmintrin.h> // SSE2
#include "X264Encoder.h"
class ThreadPool
{
public:
// <EFBFBD><EFBFBD><EFBFBD>캯<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>̶<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>߳<EFBFBD>
// 构造函数：创建固定数量的线程
ThreadPool(size_t numThreads) : stop(false)
{
for (size_t i = 0; i < numThreads; ++i) {
@@ -37,14 +38,14 @@ public:
try {
task();
} catch (...) {
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>쳣
// 处理异常
}
}
});
}
}
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>̳߳<EFBFBD>
// 析构函数：销毁线程池
~ThreadPool()
{
{
@@ -56,7 +57,7 @@ public:
worker.join();
}
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ύ
// 任务提交
template<typename F>
auto enqueue(F&& f) -> std::future<decltype(f())>
{
@@ -98,7 +99,7 @@ private:
MONITORINFOEX mi;
mi.cbSize = sizeof(MONITORINFOEX);
if (GetMonitorInfo(hMonitor, &mi)) {
monitors->push_back(mi); // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʾ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ
monitors->push_back(mi); // 保存显示器信息
}
return TRUE;
}
@@ -109,31 +110,32 @@ private:
return monitors;
}
public:
ThreadPool* m_ThreadPool; // <EFBFBD>̳߳<EFBFBD>
BYTE* m_FirstBuffer; // <EFBFBD><EFBFBD>һ֡<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
BYTE* m_RectBuffer; // <EFBFBD><EFBFBD>ǰ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
LPBYTE* m_BlockBuffers; // <EFBFBD>ֿ黺<EFBFBD><EFBFBD>
ULONG* m_BlockSizes; // <EFBFBD>ֿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_BlockNum; // <EFBFBD>ֿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_SendQuality; // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
ThreadPool* m_ThreadPool; // 线程池
BYTE* m_FirstBuffer; // 上一帧数据
BYTE* m_RectBuffer; // 当前缓存区
LPBYTE* m_BlockBuffers; // 分块缓存
ULONG* m_BlockSizes; // 分块差异像素数
int m_BlockNum; // 分块个数
int m_SendQuality; // 发送质量
LPBITMAPINFO m_BitmapInfor_Full; // BMP<EFBFBD><EFBFBD>Ϣ
BYTE m_bAlgorithm; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>㷨
LPBITMAPINFO m_BitmapInfor_Full; // BMP信息
BYTE m_bAlgorithm; // 屏幕差异算法
int m_iScreenX; // <EFBFBD><EFBFBD>ʼx<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_iScreenY; // <EFBFBD><EFBFBD>ʼy<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
ULONG m_ulFullWidth; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD>
ULONG m_ulFullHeight; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD>
bool m_bZoomed; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
double m_wZoom; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ű<EFBFBD>
double m_hZoom; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ű<EFBFBD>
int m_iScreenX; // 起始x坐标
int m_iScreenY; // 起始y坐标
ULONG m_ulFullWidth; // 屏幕宽
ULONG m_ulFullHeight; // 屏幕高
bool m_bZoomed; // 屏幕被缩放
double m_wZoom; // 屏幕横向缩放比
double m_hZoom; // 屏幕纵向缩放比
int m_biBitCount; // ÿ<EFBFBD><EFBFBD><EFBFBD>ر<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_FrameID; // ֡<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_GOP; // <EFBFBD>ؼ<EFBFBD>֡<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
bool m_SendKeyFrame; // <EFBFBD><EFBFBD><EFBFBD>͹ؼ<EFBFBD>֡
CX264Encoder *m_encoder; // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_nScreenCount; // <EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
int m_biBitCount; // 每像素比特数
int m_FrameID; // 帧序号
int m_GOP; // 关键帧间隔
bool m_SendKeyFrame; // 发送关键帧
CX264Encoder *m_encoder; // 编码器
int m_nScreenCount; // 屏幕数量
BOOL m_bEnableMultiScreen;// 多显示器支持
ScreenCapture(int n = 32, BYTE algo = ALGORITHM_DIFF, BOOL all = FALSE) :
m_ThreadPool(nullptr), m_FirstBuffer(nullptr), m_RectBuffer(nullptr),
@@ -147,6 +149,7 @@ public:
static auto monitors = GetAllMonitors();
static int index = 0;
m_nScreenCount = monitors.size();
m_bEnableMultiScreen = all;
if (all && !monitors.empty()) {
int idx = index++ % (monitors.size()+1);
if (idx == 0) {
@@ -162,8 +165,8 @@ public:
m_ulFullHeight = rt.bottom - rt.top;
}
} else {
//::GetSystemMetrics(SM_CXSCREEN/SM_CYSCREEN)<EFBFBD><EFBFBD>ȡ<EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD>С<EFBFBD><EFBFBD>׼
//<EFBFBD><EFBFBD><EFBFBD>統<EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD>ʾ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ125%ʱ<><CAB1><EFBFBD><EFBFBD>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ļ<EFBFBD><C4BB>С<EFBFBD><D0A1>Ҫ<EFBFBD><D2AA><EFBFBD><EFBFBD>1.25<EFBFBD>Ŷ<EFBFBD>
//::GetSystemMetrics(SM_CXSCREEN/SM_CYSCREEN)获取屏幕大小不准
//例如当屏幕显示比例为125%时，获取到的屏幕大小需要乘以1.25才对
DEVMODE devmode;
memset(&devmode, 0, sizeof(devmode));
devmode.dmSize = sizeof(DEVMODE);
@@ -174,7 +177,7 @@ public:
int w = GetSystemMetrics(SM_CXSCREEN), h = GetSystemMetrics(SM_CYSCREEN);
m_bZoomed = (w != m_ulFullWidth) || (h != m_ulFullHeight);
m_wZoom = double(m_ulFullWidth) / w, m_hZoom = double(m_ulFullHeight) / h;
Mprintf("=> <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ű<EFBFBD><EFBFBD><EFBFBD>: %.2f, %.2f\t<EFBFBD>ֱ<EFBFBD><EFBFBD>ʣ<EFBFBD>%d x %d\n", m_wZoom, m_hZoom, m_ulFullWidth, m_ulFullHeight);
Mprintf("=> 桌面缩放比例: %.2f, %.2f\t分辨率：%d x %d\n", m_wZoom, m_hZoom, m_ulFullWidth, m_ulFullHeight);
m_wZoom = 1.0 / m_wZoom, m_hZoom = 1.0 / m_hZoom;
}
if (ALGORITHM_H264 == m_bAlgorithm) {
@@ -212,6 +215,10 @@ public:
return m_nScreenCount;
}
virtual BOOL IsMultiScreenEnabled() const {
return m_bEnableMultiScreen;
}
virtual int SendQuality(int quality)
{
int old = m_SendQuality;
@@ -230,36 +237,134 @@ public:
}
public:
//*************************************** ͼ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>㷨<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>У<EFBFBD> *************************************
//*************************************** 图像差异算法 SSE2 优化版 *************************************
virtual ULONG CompareBitmap(LPBYTE CompareSourData, LPBYTE CompareDestData, LPBYTE szBuffer,
DWORD ulCompareLength, BYTE algo, int startPostion = 0)
{
// Windows<77>涨һ<E6B6A8><D2BB>ɨ<EFBFBD><C9A8><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ռ<EFBFBD><D5BC><EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>4<EFBFBD>ı<EFBFBD><C4B1><EFBFBD>, <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>DWORD<52>Ƚ<EFBFBD>
LPDWORD p1 = (LPDWORD)CompareDestData, p2 = (LPDWORD)CompareSourData;
LPBYTE p = szBuffer;
ULONG channel = algo == ALGORITHM_GRAY ? 1 : 4;
ULONG ratio = algo == ALGORITHM_GRAY ? 4 : 1;
for (ULONG i = 0; i < ulCompareLength; i += 4, ++p1, ++p2) {
if (*p1 == *p2)
continue;
ULONG index = i;
LPDWORD pos1 = p1++, pos2 = p2++;
// <20><><EFBFBD><EFBFBD><EFBFBD>м<EFBFBD><D0BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5>ͬ
for (i += 4; i < ulCompareLength && *p1 != *p2; i += 4, ++p1, ++p2);
ULONG ulCount = i - index;
memcpy(pos1, pos2, ulCount); // <20><><EFBFBD><EFBFBD>Ŀ<EFBFBD><C4BF><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
// SSE2: 每次比较 16 字节 (4 个像素)
const ULONG SSE_BLOCK = 16;
const ULONG alignedLength = ulCompareLength & ~(SSE_BLOCK - 1);
__m128i* v1 = (__m128i*)CompareDestData;
__m128i* v2 = (__m128i*)CompareSourData;
ULONG i = 0;
while (i < alignedLength) {
// SSE2 快速比较: 一次比较 16 字节
__m128i cmp = _mm_cmpeq_epi32(*v1, *v2);
int mask = _mm_movemask_epi8(cmp);
if (mask == 0xFFFF) {
// 16 字节完全相同，跳过
i += SSE_BLOCK;
++v1;
++v2;
continue;
}
// 发现差异，记录起始位置
ULONG index = i;
LPBYTE pos1 = (LPBYTE)v1;
LPBYTE pos2 = (LPBYTE)v2;
// 继续扫描连续的差异区域（带间隙容忍）
// GAP_TOLERANCE: 允许的最大间隙，小于此值的相同区域会被合并
// 设为 32 字节（8像素），因为每个差异区域头部开销是 8 字节
const ULONG GAP_TOLERANCE = 32;
// 首先必须前进一个块（当前块已知有差异）
i += SSE_BLOCK;
++v1;
++v2;
// 继续扫描更多差异，应用间隙容忍
ULONG gapCount = 0;
while (i < alignedLength) {
cmp = _mm_cmpeq_epi32(*v1, *v2);
mask = _mm_movemask_epi8(cmp);
if (mask == 0xFFFF) {
// 相同块 - 累计间隙
gapCount += SSE_BLOCK;
if (gapCount > GAP_TOLERANCE) {
// 间隙太大 - 停止（不包含间隙）
break;
}
// 间隙仍可接受 - 继续扫描
i += SSE_BLOCK;
++v1;
++v2;
} else {
// 差异块 - 重置间隙并包含它
gapCount = 0;
i += SSE_BLOCK;
++v1;
++v2;
}
}
// 排除末尾累积的间隙
if (gapCount > 0 && gapCount <= GAP_TOLERANCE) {
i -= gapCount;
v1 = (__m128i*)((LPBYTE)v1 - gapCount);
v2 = (__m128i*)((LPBYTE)v2 - gapCount);
}
ULONG ulCount = i - index;
// 更新目标缓冲区
memcpy(pos1, pos2, ulCount);
// 写入差异信息: [位置][长度][数据]
*(LPDWORD)(p) = index + startPostion;
*(LPDWORD)(p + sizeof(ULONG)) = ulCount / ratio;
p += 2 * sizeof(ULONG);
if (channel != 1) {
memcpy(p, pos2, ulCount);
p += ulCount;
} else {
for (LPBYTE end = p + ulCount / ratio; p < end; p += channel, ++pos2) {
LPBYTE src = (LPBYTE)pos2;
*p = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
// 灰度转换：使用优化的批量处理
ConvertToGray_SSE2(p, pos2, ulCount);
p += ulCount / ratio;
}
}
// 处理剩余的非对齐部分 (0-12 字节)
if (i < ulCompareLength) {
LPDWORD p1 = (LPDWORD)((LPBYTE)CompareDestData + i);
LPDWORD p2 = (LPDWORD)((LPBYTE)CompareSourData + i);
for (; i < ulCompareLength; i += 4, ++p1, ++p2) {
if (*p1 == *p2)
continue;
ULONG index = i;
LPDWORD pos1 = p1++;
LPDWORD pos2 = p2++;
for (i += 4; i < ulCompareLength && *p1 != *p2; i += 4, ++p1, ++p2);
ULONG ulCount = i - index;
memcpy(pos1, pos2, ulCount);
*(LPDWORD)(p) = index + startPostion;
*(LPDWORD)(p + sizeof(ULONG)) = ulCount / ratio;
p += 2 * sizeof(ULONG);
if (channel != 1) {
memcpy(p, pos2, ulCount);
p += ulCount;
} else {
// 剩余部分用标量处理
LPDWORD srcPtr = pos2;
for (LPBYTE end = p + ulCount / ratio; p < end; ++p, ++srcPtr) {
LPBYTE src = (LPBYTE)srcPtr;
*p = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
}
}
}
}
@@ -267,20 +372,20 @@ public:
return p - szBuffer;
}
//*************************************** ͼ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>㷨<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>У<EFBFBD> *************************************
//*************************************** 图像差异算法（并行） *************************************
ULONG MultiCompareBitmap(LPBYTE srcData, LPBYTE dstData, LPBYTE szBuffer,
DWORD ulCompareLength, BYTE algo)
{
int N = m_BlockNum;
ULONG blockLength = ulCompareLength / N; // ÿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֽ<EFBFBD><EFBFBD><EFBFBD>
ULONG remainingLength = ulCompareLength % N; // ʣ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֽ<EFBFBD><EFBFBD><EFBFBD>
ULONG blockLength = ulCompareLength / N; // 每个任务的基本字节数
ULONG remainingLength = ulCompareLength % N; // 剩余的字节数
std::vector<std::future<ULONG>> futures;
for (int blockY = 0; blockY < N; ++blockY) {
// <EFBFBD><EFBFBD><EFBFBD>㵱ǰ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֽ<EFBFBD><EFBFBD><EFBFBD>
// 计算当前任务的字节数
ULONG currentBlockLength = blockLength + (blockY == N - 1 ? remainingLength : 0);
// <EFBFBD><EFBFBD><EFBFBD>㵱ǰ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʼλ<EFBFBD><EFBFBD>
// 计算当前任务的起始位置
ULONG startPosition = blockY * blockLength;
futures.emplace_back(m_ThreadPool->enqueue([=]() -> ULONG {
@@ -288,24 +393,24 @@ public:
LPBYTE dstBlock = dstData + startPosition;
LPBYTE blockBuffer = m_BlockBuffers[blockY];
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ǰ<EFBFBD><EFBFBD><EFBFBD>񲢷<EFBFBD><EFBFBD>رȶ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݴ<EFBFBD>С
// 处理当前任务并返回比对数据大小
return m_BlockSizes[blockY] = CompareBitmap(srcBlock, dstBlock, blockBuffer, currentBlockLength, algo, startPosition);
}));
}
// <EFBFBD>ȴ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ɲ<EFBFBD><EFBFBD><EFBFBD>ȡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֵ
// 等待所有任务完成并获取返回值
for (auto& future : futures) {
future.get();
}
// <EFBFBD>ϲ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>п<EFBFBD><EFBFBD>IJ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ<EFBFBD><EFBFBD> szBuffer
// 合并所有块的差异信息到 szBuffer
ULONG offset = 0;
for (int blockY = 0; blockY < N; ++blockY) {
memcpy(szBuffer + offset, m_BlockBuffers[blockY], m_BlockSizes[blockY]);
offset += m_BlockSizes[blockY];
}
return offset; // <EFBFBD><EFBFBD><EFBFBD>ػ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD>С
return offset; // 返回缓冲区的大小
}
virtual int GetFrameID() const {
@@ -321,10 +426,41 @@ public:
return m_BitmapInfor_Full->bmiHeader.biSizeImage;
}
// SSE2 优化：BGRA 转单通道灰度，一次处理 4 个像素，输出 4 字节
// 灰度公式: Y = 0.299*R + 0.587*G + 0.114*B ≈ (306*R + 601*G + 117*B) >> 10
// 输入: BGRA 像素数据 (每像素 4 字节)
// 输出: 灰度值 (每像素 1 字节)
// count: 输入数据的字节数 (必须是 4 的倍数)
inline void ConvertToGray_SSE2(LPBYTE dst, LPBYTE src, ULONG count)
{
ULONG pixels = count / 4;
ULONG i = 0;
ULONG aligned = pixels & ~3; // 4 像素对齐
// 一次处理 4 个像素
for (; i < aligned; i += 4, src += 16, dst += 4) {
// 计算 4 个灰度值
dst[0] = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
dst[1] = (306 * src[6] + 601 * src[4] + 117 * src[5]) >> 10;
dst[2] = (306 * src[10] + 601 * src[8] + 117 * src[9]) >> 10;
dst[3] = (306 * src[14] + 601 * src[12] + 117 * src[13]) >> 10;
}
// 处理剩余像素
for (; i < pixels; i++, src += 4, dst++) {
*dst = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
}
}
// ToGray: BGRA 转 BGRA 灰度 (三通道相同值)，用于关键帧
// 直接标量处理，编译器会自动向量化
void ToGray(LPBYTE dst, LPBYTE src, int biSizeImage)
{
for (ULONG i = 0; i < biSizeImage; i += 4, dst += 4, src += 4) {
dst[0] = dst[1] = dst[2] = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
ULONG pixels = biSizeImage / 4;
for (ULONG i = 0; i < pixels; i++, src += 4, dst += 4) {
BYTE g = (306 * src[2] + 601 * src[0] + 117 * src[1]) >> 10;
dst[0] = dst[1] = dst[2] = g;
dst[3] = 0xFF;
}
}
@@ -342,7 +478,7 @@ public:
return bmpInfo;
}
// <EFBFBD>㷨+<2B><><EFBFBD><EFBFBD>λ<EFBFBD><CEBB>+<2B><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
// 算法+光标位置+光标类型
virtual LPBYTE GetNextScreenData(ULONG* ulNextSendLength)
{
BYTE algo = m_bAlgorithm;
@@ -350,26 +486,26 @@ public:
bool keyFrame = (frameID % m_GOP == 0);
m_RectBuffer[0] = keyFrame ? TOKEN_KEYFRAME : TOKEN_NEXTSCREEN;
LPBYTE data = m_RectBuffer + 1;
// д<EFBFBD><EFBFBD>ʹ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>㷨
// 写入使用了哪种算法
memcpy(data, (LPBYTE)&algo, sizeof(BYTE));
// д<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>λ<EFBFBD><EFBFBD>
// 写入光标位置
POINT CursorPos;
GetCursorPos(&CursorPos);
CursorPos.x /= m_wZoom;
CursorPos.y /= m_hZoom;
memcpy(data + sizeof(BYTE), (LPBYTE)&CursorPos, sizeof(POINT));
// д<EFBFBD>뵱ǰ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
// 写入当前光标类型
static CCursorInfo m_CursorInfor;
BYTE bCursorIndex = m_CursorInfor.getCurrentCursorIndex();
memcpy(data + sizeof(BYTE) + sizeof(POINT), &bCursorIndex, sizeof(BYTE));
ULONG offset = sizeof(BYTE) + sizeof(POINT) + sizeof(BYTE);
// <EFBFBD>ֶ<EFBFBD>ɨ<EFBFBD><EFBFBD>ȫ<EFBFBD><EFBFBD>Ļ <20><><EFBFBD>µ<EFBFBD>λͼ<CEBB><CDBC><EFBFBD>뵽m_hDiffMemDC<EFBFBD><EFBFBD>
// 分段扫描全屏幕 将新的位图放入到m_hDiffMemDC中
LPBYTE nextData = ScanNextScreen();
if (nullptr == nextData) {
// ɨ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ֡ʧ<EFBFBD><EFBFBD>Ҳ<EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD><EFBFBD>͹<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ƶ<EFBFBD>
// 扫描下一帧失败也需要发送光标信息到控制端
*ulNextSendLength = 1 + offset;
return m_RectBuffer;
}
@@ -439,7 +575,7 @@ public:
return m_RectBuffer;
}
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>㷨
// 设置屏幕传输算法
virtual BYTE SetAlgorithm(int algo)
{
BYTE oldAlgo = m_bAlgorithm;
@@ -447,7 +583,7 @@ public:
return oldAlgo;
}
// <EFBFBD><EFBFBD><EFBFBD><EFBFBD>λ<EFBFBD><EFBFBD>ת<EFBFBD><EFBFBD>
// 鼠标位置转换
virtual void PointConversion(POINT& pt) const
{
if (m_bZoomed) {
@@ -458,17 +594,17 @@ public:
pt.y += m_iScreenY;
}
// <EFBFBD><EFBFBD>ȡλͼ<EFBFBD>ṹ<EFBFBD><EFBFBD>Ϣ
// 获取位图结构信息
virtual const LPBITMAPINFO& GetBIData() const
{
return m_BitmapInfor_Full;
}
public: // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӿ<EFBFBD>
public: // 纯虚接口
// <EFBFBD><EFBFBD>ȡ<EFBFBD><EFBFBD>һ֡<EFBFBD><EFBFBD>Ļ
// 获取第一帧屏幕
virtual LPBYTE GetFirstScreenData(ULONG* ulFirstScreenLength) = 0;
// <EFBFBD><EFBFBD>ȡ<EFBFBD><EFBFBD>һ֡<EFBFBD><EFBFBD>Ļ
// 获取下一帧屏幕
virtual LPBYTE ScanNextScreen() = 0;
};

5
client/ScreenManager.cpp
View File

@@ -1,4 +1,4 @@
// ScreenManager.cpp: implementation of the CScreenManager class.
// ScreenManager.cpp: implementation of the CScreenManager class.
//
//////////////////////////////////////////////////////////////////////
@@ -112,7 +112,8 @@ CScreenManager::CScreenManager(IOCPClient* ClientObject, int n, void* user):CMan
}
bool CScreenManager::SwitchScreen() {
if (m_ScreenSpyObject == NULL || m_ScreenSpyObject->GetScreenCount() <= 1)
if (m_ScreenSpyObject == NULL || m_ScreenSpyObject->GetScreenCount() <= 1 ||
!m_ScreenSpyObject->IsMultiScreenEnabled())
return false;
m_bIsWorking = FALSE;
DWORD s = WaitForSingleObject(m_hWorkThread, 3000);