How can I get the pixel data from an area of the screen (much) faster in AHK?

后端 未结 2 547
长发绾君心
长发绾君心 2021-01-15 03:23

I understand that for some things it would be better to write certain things in C++, but I\'d really like to be able to do this in AHK:

I want to be able to retrieve

2条回答
  •  礼貌的吻别
    2021-01-15 03:59

    I found a way to do it 103 times faster than Forivin's solution :D

    SetBatchLines, -1
    CoordMode, Pixel, screen
    
    FileDelete, Log.txt
    
    searchSpace := 400
    jumpSize := 1 ; how many units to skip each interval
    total := Round(((searchSpace * searchSpace) / jumpSize), 0)
    startTimer := getTimestamp()
    getPixelMapSlow(searchSpace, jumpSize)
    endTimer := getTimestamp()
    duration := endTimer - startTimer
    rate := total / duration
    FileAppend, % "[getPixelMapSlow] Retrieved " . total . " pixels from bitmap, duration: " . duration . "ms at at a rate of " . rate . " pixels/ms.`n", Log.txt
    
    searchSpace := 400
    jumpSize := 1 ; how many units to skip each interval
    total := Round(((searchSpace * searchSpace) / jumpSize), 0)
    startTimer := getTimestamp()
    getPixelMapFast(searchSpace, jumpSize)
    endTimer := getTimestamp()
    duration := endTimer - startTimer
    rate := total / duration
    FileAppend, % "[getPixelMapFast] Retrieved " . total . " pixels from bitmap, duration: " . duration . "ms at at a rate of " . rate . " pixels/ms.`n", Log.txt
    
    
    getPixelMapFast(searchSpace, jumpSize){
        width := 1920
        height := 1080
        centerX := width / 2
        centerY := height / 2
        searchSpacehalf := searchSpace / 2
        searchCounterX := 0
        searchCounterY := 0
        pixelMap := {}
        pBitmap := Gdip_BitmapFromScreen((centerX - searchSpacehalf) . "|" . (centerY - searchSpacehalf) . "|" . searchSpace . "|" . searchSpace)
        E1 := Gdip_LockBits(pBitmap, 0, 0, Gdip_GetImageWidth(pBitmap), Gdip_GetImageHeight(pBitmap), Stride, Scan0, BitmapData)
        Loop, %searchSpace%
        {
            tick := A_Index * jumpSize
            if (tick < searchSpace) {
                New_Index_X := tick
                Loop, %searchSpace%
                {
                    tick := A_Index * jumpSize
                    if (tick < searchSpace) {
                        New_Index_Y := tick
                        color1ARGB := Gdip_GetLockBitPixel(Scan0, New_Index_X, New_Index_Y, Stride)
                        SetFormat, Integer, H
                        color1RGB := 0x00ffffff & color1ARGB
                        SetFormat, Integer, D
                        if (!pixelMap[New_Index_X]){
                            pixelMap[New_Index_X] := {}
                        }
                        pixelMap[New_Index_X][New_Index_Y] := color1RGB
                    }
                }
            }
        }
        Gdip_UnlockBits(pBitmap, BitmapData)
        Gdip_DisposeImage(pBitmap)
        return pixelMap
    }
    
    getPixelMapSlow(searchSpace, jumpSize){
        width := 1920
        height := 1080
        centerX := width / 2
        centerY := height / 2
        searchSpacehalf := searchSpace / 2
        searchCounterX := 0
        searchCounterY := 0
        pixelMap := {}
        pBitmap := Gdip_BitmapFromScreen((centerX - searchSpacehalf) . "|" . (centerY - searchSpacehalf) . "|" . searchSpace . "|" . searchSpace)
        Loop, %searchSpace%
        {
            tick := A_Index * jumpSize
            if (tick < searchSpace) {
                New_Index_X := tick
                Loop, %searchSpace%
                {
                    tick := A_Index * jumpSize
                    if (tick < searchSpace) {
                        New_Index_Y := tick
                        color1ARGB := Gdip_GetPixel(pBitmap, New_Index_X, New_Index_Y)
                        if (!pixelMap[New_Index_X]){
                            pixelMap[New_Index_X] := {}
                        }
                        color1RGB := ARGBtoRGB(color1ARGB)
                        pixelMap[New_Index_X][New_Index_Y] := color1RGB
                    }
                }
            }
        }
        Gdip_DisposeImage(pBitmap)
        return pixelMap
    }
    
    ARGBtoRGB( ARGB ) {
        VarSetCapacity( RGB,6,0 )
        DllCall( "msvcrt.dll\sprintf", Str,RGB, Str,"%06X", UInt,ARGB<<8 )
        Return "0x" RGB
    }
    
    getTimestamp()
    {
        DllCall("QueryPerformanceCounter", "Int64*", timestamp)
        DllCall("QueryPerformanceFrequency", "Int64*", frequency)
        return Round(timestamp * 1000 / frequency)
    }
    

    Of course include the relevant functions of the AHK Gdip library (found on Github) in your code for this to work.

    Log:

    [getPixelMapSlow] Retrieved 160000 pixels from bitmap, duration: 33161ms at at a rate of 4.824945 pixels/ms.
    [getPixelMapFast] Retrieved 160000 pixels from bitmap, duration: 321ms at at a rate of 498.442368 pixels/ms.
    

提交回复
热议问题