主动化压测工具开发(MFC) - ToB企服应用市场:ToB评测及商务社交产业平台

bool CComTest::InitialCom(int iComID, int iComPort, DWORD iBaudRate) {
// 省略部分错误检查和资源清理代码...
// 配置串口参数
DCB dcbSerialParams = {0};
GetCommState(m_hCom, &dcbSerialParams);
dcbSerialParams.BaudRate = iBaudRate;
dcbSerialParams.ByteSize = 8;
dcbSerialParams.StopBits = STOPBITS_ONE;
dcbSerialParams.Parity = NOPARITY;
SetCommState(m_hCom, &dcbSerialParams);
// 设置串口超时
COMMTIMEOUTS timeouts = {0};
timeouts.ReadIntervalTimeout = MAXDWORD;
SetCommTimeouts(m_hCom, &timeouts);
// 完成串口初始化
return TRUE;
}

复制代码

DWORD CComTest::ReadCom(BYTE *pBuff, int nCount) {
DWORD dwRead = 0;
ReadFile(m_hCom, pBuff, nCount, &dwRead, NULL);
return dwRead;
}

复制代码

BOOL CComTest::WriteCom(BYTE *pBuff, int nCount) {
DWORD dwWritten = 0;
return WriteFile(m_hCom, pBuff, nCount, &dwWritten, NULL);
}

复制代码

BYTE pData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
WriteCom(m_iComID, pData, sizeof(pData));

复制代码

BOOL CCaptureDlg::OnInitDialog() {
CDialog::OnInitDialog();
// 屏幕截图并设置窗口大小
copyScreenToBitmap(m_ScreenBmp);
MoveWindow(-3, -3, GetSystemMetrics(SM_CXSCREEN) + 6, GetSystemMetrics(SM_CYSCREEN) + 6);
// 初始化橡皮筋跟踪器
m_rectTracker.m_nStyle = CRectTracker::resizeOutside | CRectTracker::dottedLine;
m_rectTracker.m_rect.SetRect(0, 0, 0, 0);
return TRUE;
}

复制代码

BOOL CCaptureDlg::OnEraseBkgnd(CDC *pDC) {
CDC memDC;
memDC.CreateCompatibleDC(pDC);
memDC.SelectObject(&m_ScreenBmp);
pDC->BitBlt(0, 0, GetClientRect().Width(), GetClientRect().Height(), &memDC, 0, 0, SRCCOPY);
memDC.DeleteDC();
return TRUE;
}

复制代码

void CCaptureDlg::OnLButtonDown(UINT nFlags, CPoint point) {
if (m_rectTracker.HitTest(point) == CRectTracker::hitNothing) {
m_rectTracker.TrackRubberBand(this, point, TRUE);
} else {
m_rectTracker.Track(this, point, TRUE);
m_rectTracker.m_rect.NormalizeRect();
}
Invalidate(TRUE);
CDialog::OnLButtonDown(nFlags, point);
}

复制代码

void CCaptureDlg::OnLButtonDblClk(UINT nFlags, CPoint point) {
if (m_rectTracker.HitTest(point) != CRectTracker::hitMiddle) {
MessageBox(_T("截图失败,请重新截图！"));
return;
}
// 检查双击是否在橡皮筋跟踪器的中间区域
if (m_rectTracker.HitTest(point) != CRectTracker::hitMiddle)
{
MessageBox(_T("截图失败, 请重新截图！"));
return;
}
// 保存双击时坐标
m_startPointX = point.x;
m_startPointY = point.y;
// 获取当前屏幕设备上下文
CDC* pScreenDC = GetDC();
if (!pScreenDC)
{
MessageBox(_T("无法获取屏幕设备上下文"));
return;
}
// 创建内存设备上下文与屏幕DC兼容，并选择m_ScreenBmp位图进该内存DC
CDC memDC;
memDC.CreateCompatibleDC(pScreenDC);
CBitmap* pOldBitmap = memDC.SelectObject(&m_ScreenBmp);
// 获取橡皮筋跟踪器定义的区域
CRect captureRect = m_rectTracker.m_rect;
// 创建一个兼容的位图用于保存截取的区域
CBitmap captureBmp;
captureBmp.CreateCompatibleBitmap(pScreenDC, captureRect.Width(), captureRect.Height());
// 创建另一个内存DC，选择刚才创建的兼容位图进这个DC
CDC captureDC;
captureDC.CreateCompatibleDC(pScreenDC);
CBitmap* pOldCaptureBmp = captureDC.SelectObject(&captureBmp);
// 从原屏幕位图中截取定义区域的图像到captureDC
captureDC.BitBlt(0, 0, captureRect.Width(), captureRect.Height(), &memDC, captureRect.left, captureRect.top, SRCCOPY);
// 使用CImage类来保存位图到文件
CImage img;
img.Attach(captureBmp);
CString strCapturePath = _T("你想要保存的路径\\capture1.jpg");
img.Save(strCapturePath);
// 清理资源
captureDC.SelectObject(pOldCaptureBmp);
memDC.SelectObject(pOldBitmap);
captureBmp.DeleteObject();
memDC.DeleteDC();
captureDC.DeleteDC();
ReleaseDC(pScreenDC);
// 截图操作完成后关闭对话框
CDialog::OnCancel();
}

复制代码

void CCaptureDlg::Screen(const CRect& cRect, int iNumber)
{
// 使用局部函数来自动释放资源
struct AutoRelease
{
HDC hDC;
~AutoRelease() { if (hDC) ::ReleaseDC(NULL, hDC); }
} autoReleaseDC;
// 获取屏幕设备上下文
autoReleaseDC.hDC = ::GetDC(NULL);
if (!autoReleaseDC.hDC)
{
MessageBox(_T("无法获取屏幕设备上下文。"));
return;
}
CDC memDC;
memDC.CreateCompatibleDC(NULL); // 创建与屏幕兼容的内存设备上下文
// 创建与屏幕兼容的位图
CBitmap memBitmap;
memBitmap.CreateCompatibleBitmap(autoReleaseDC.hDC, cRect.Width(), cRect.Height());
// 将新创建的位图选入内存设备上下文中
CBitmap* pOldBitmap = memDC.SelectObject(&memBitmap);
// 从屏幕设备上下文中复制图像到内存设备上下文中的位图
memDC.BitBlt(0, 0, cRect.Width(), cRect.Height(), CDC::FromHandle(autoReleaseDC.hDC), cRect.left, cRect.top, SRCCOPY);
// 使用CImage类来操作和保存位图
CImage img;
if (!img.Attach(memBitmap))
{
MessageBox(_T("无法将位图附加到CImage对象。"));
return;
}
// 构造文件保存路径
CString strCapturePath;
strCapturePath.Format(_T("保存路径\\capture%d.jpg"), iNumber);
// 保存位图到文件
if (!img.Save(strCapturePath))
{
AfxMessageBox(_T("保存图像失败。"));
}
// 清理资源
memDC.SelectObject(pOldBitmap); // 恢复原始位图
img.Detach(); // 从CImage对象中分离位图
}

复制代码

void SimulateMouseClick(int x, int y) {
SetCursorPos(x, y);
mouse_event(MOUSEEVENTF_LEFTDOWN, x, y, 0, 0);
mouse_event(MOUSEEVENTF_LEFTUP, x, y, 0, 0);
}

复制代码

void SimulateKeyboardInput(const char *szInput) {
for (int i = 0; szInput[i] != '\0'; ++i) {
keybd_event(szInput[i], 0, 0, 0);
keybd_event(szInput[i], 0, KEYEVENTF_KEYUP, 0);
}
}

复制代码