在 Golang 中将 GDI32.dll 位图保存到磁盘

Question

我的第一个 SO 问题 :-) 我希望通过在 Windows 机器上调用 User32.dll 和 GDI32.dll 从 Golang 截取屏幕截图（项目要求）。

我有一个包含屏幕截图像​​素的位图句柄。但是，我不知道如何访问其数据或如何将其保存到磁盘。有谁知道如何将 GDI 位图映射到 Golang []byte 然后另存为 JPG 或 PNG？

package main
import "syscall"

var (
    user32            = syscall.NewLazyDLL("user32.dll")
    procGetClientRect = user32.NewProc("GetClientRect")
    // etc...
    gdi32             = syscall.NewLazyDLL("gdi32.dll")
    procCreateDC      = gdi32.NewProc("CreateDC")
    SRCCOPY      uint = 13369376
   //etc...
)

//
// omitted for brevity
//
func TakeDesktopScreenshotViaWinAPI() {

    // these are all calls to user32.dll or gdi32.dll
    hDesktop := GetDesktopWindow()
    desktopRect := GetClientRect(hDesktop)
    width := int(desktopRect.Right - desktopRect.Left)
    height := int(desktopRect.Bottom - desktopRect.Top)

    // create device contexts
    srcDC := GetDC(hDesktop)
    targetDC := CreateCompatibleDC(srcDC)

    // create bitmap to copy to
    hBitmap := CreateCompatibleBitmap(targetDC, width, height)

    // select the bitmap into target DC
    hOldSelection := SelectObject(targetDC, HGDIOBJ(hBitmap))

    //bit block transfer from src to target
    BitBlt(targetDC, 0, 0, width, height, srcDC, 0, 0, SRCCOPY)

   // how to save the  the data in
   // *hBitmap ???

   // restore selection
   SelectObject(targetDC, hOldSelection)

   // clean up
   DeleteDC(HDC(targetDC))
   ReleaseDC(hDesktop, srcDC)
   DeleteObject(HGDIOBJ(hBitmap))
}

Answer 1

您可以直接使用 vova616 的the screenshot library，或查看screenshot_windows.go 以了解所需的转换方法。

根据提供的示例：

package main

import (
    "github.com/vova616/screenshot"
    "image/png"
    "os"
)

func main() {
    img, err := screenshot.CaptureScreen()
    if err != nil {
        panic(err)
    }
    f, err := os.Create("./ss.png")
    if err != nil {
        panic(err)
    }
    err = png.Encode(f, img)
    if err != nil {
        panic(err)
    }
    f.Close()
}