from mss import mss
import time

start = time.time()

with mss() as sct:
    # 截取主显示器
    sct.shot(output="screen.png")

print(f"截图耗时: {time.time() - start:.3f}秒")

#include <windows.h>
#include <gdiplus.h>
#include <chrono>
#include <iostream>

void CaptureScreen(const char* filename) {
    auto start = std::chrono::high_resolution_clock::now();
    
    HDC hdcScreen = GetDC(NULL);
    HDC hdcMem = CreateCompatibleDC(hdcScreen);
    
    int width = GetSystemMetrics(SM_CXSCREEN);
    int height = GetSystemMetrics(SM_CYSCREEN);
    
    HBITMAP hBitmap = CreateCompatibleBitmap(hdcScreen, width, height);
    SelectObject(hdcMem, hBitmap);
    BitBlt(hdcMem, 0, 0, width, height, hdcScreen, 0, 0, SRCCOPY);
    
    Gdiplus::Bitmap bitmap(hBitmap, NULL);
    CLSID clsid;
    GetEncoderClsid(L"image/png", &clsid);
    bitmap.Save(L"screen.png", &clsid, NULL);
    
    DeleteObject(hBitmap);
    DeleteDC(hdcMem);
    ReleaseDC(NULL, hdcScreen);
    
    auto end = std::chrono::high_resolution_clock::now();
    std::chrono::duration<double> elapsed = end - start;
    std::cout << "截图耗时: " << elapsed.count() << "秒" << std::endl;
}