33 lines
551 B
Go
33 lines
551 B
Go
package capture
|
|
|
|
import (
|
|
"image"
|
|
"sync"
|
|
)
|
|
|
|
var rgbaPool sync.Pool
|
|
|
|
func GetRGBA(w, h int) *image.RGBA {
|
|
need := w * h * 4
|
|
if need <= 0 {
|
|
return image.NewRGBA(image.Rect(0, 0, w, h))
|
|
}
|
|
if v := rgbaPool.Get(); v != nil {
|
|
if buf, ok := v.([]byte); ok && cap(buf) >= need {
|
|
return &image.RGBA{
|
|
Pix: buf[:need],
|
|
Stride: w * 4,
|
|
Rect: image.Rect(0, 0, w, h),
|
|
}
|
|
}
|
|
}
|
|
return image.NewRGBA(image.Rect(0, 0, w, h))
|
|
}
|
|
|
|
func PutRGBA(img *image.RGBA) {
|
|
if img == nil || len(img.Pix) == 0 {
|
|
return
|
|
}
|
|
rgbaPool.Put(img.Pix)
|
|
}
|