首頁 >後端開發 >Golang >有效率地將RBGA影像轉換為RGB位元組數組

有效率地將RBGA影像轉換為RGB位元組數組

WBOY
WBOY轉載
2024-02-05 22:51:121107瀏覽

有效率地將RBGA影像轉換為RGB位元組數組

問題內容

我有一個 c 函式庫和函數,需要一個指向包含 rgb 格式的 24 位元位圖的位元組陣列的指標。 alpha 通道並不重要,可以被截斷。我嘗試過這樣的事情:

func load(filePath string) *image.RGBA {
    imgFile, err := os.Open(filePath)
    if err != nil {
        fmt.Printf("Cannot read file %v\n", err)
    }
    defer imgFile.Close()

    img, _, err := image.Decode(imgFile)
    if err != nil {
        fmt.Printf("Cannot decode file %v\n", err)
    }
    return img.(*image.RGBA)
}

    img := load("myimg.png")

    bounds := img.Bounds()
    width, height := bounds.Max.X, bounds.Max.Y

    // Convert to RGB? Probably not...
    newImg := image.NewNRGBA(image.Rect(0, 0, width, height))
    draw.Draw(newImg, newImg.Bounds(), img, bounds.Min, draw.Src)
    // Pass image pointer to C function.
    C.PaintOnImage(unsafe.Pointer(&newImg.Pix[0]), C.int(newImg.Bounds().Dy()), C.int(newImg.Bounds().Dx())

但是,nrgba 似乎也是建立在每個像素 4 個位元組的基礎上的。我可以透過使用 gocv 來解決這個問題,但這對於如此簡單的任務來說似乎有點矯枉過正。有沒有辦法在 go 中以簡單有效的方式做到這一點?


正確答案


標準庫中沒有 rgb 映像類型,但您可以非常輕鬆地組裝 rgb 陣列:

bounds := img.bounds()
rgb := make([]byte, bounds.dx()*bounds.dy()*3)
idx := 0
for y := bounds.min.y; y < bounds.max.y; y++ {
    for x := bounds.min.x; x < bounds.max.x; x++ {
        offs := img.pixoffset(x, y)
        copy(rgb[idx:], img.pix[offs:offs+3])
        idx += 3
    }
}

img.pix 資料保存 4 位元組 rgba 值。上面的程式碼只是複製所有像素的前 3 個位元組 rgb 值。

由於 pix 數組中的行是連續的,因此可以透過每行僅呼叫 pixoffset 一次來改進上述程式碼,並且每個像素前進 4 個位元組。另外,手動複製 3 個位元組可能比呼叫 copy() 更快(如果對您很重要,則為基準):

bounds := img.Bounds()
rgb := make([]byte, bounds.Dx()*bounds.Dy()*3)
idx := 0

for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
    offs := img.PixOffset(bounds.Min.X, y)
    for x := bounds.Min.X; x < bounds.Max.X; x++ {
        rgb[idx+0] = img.Pix[offs+0]
        rgb[idx+1] = img.Pix[offs+1]
        rgb[idx+2] = img.Pix[offs+2]
        idx += 3
        offs += 4
    }
}

以上是有效率地將RBGA影像轉換為RGB位元組數組的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:stackoverflow.com。如有侵權,請聯絡admin@php.cn刪除