Go语言实现找图返回坐标

转自：Go语言实现找图返回坐标_Ksam的博客-CSDN博客_golang 找图 

1. 需求来源
   在一次开发脚本的时候，突发奇想使用go语言来完成，其中一个功能就是找图返回坐标，话不多说，上代码。
2. 具体实现

/*
Authorization：Aluncca
p:16685034116
*/
import (
	"encoding/base64"
	"os"
	"image"
	"image/color"
	"image/draw"
	)
type Position struct {
	X	int
	Y	int
}
func FindImgPos(dst string,src string) (Position,bool) {
	var pos Position
	// 打开 A 图片
	a, err := os.Open(src)
	if err != nil {
		return pos,false
	}
	defer a.Close()

	// 打开 B 图片
	b, err := os.Open(dst)
	if err != nil {
		return pos,false
	}
	defer b.Close()

	// 解码 A 图片
	imgA, err := png.Decode(a)
	if err != nil {
		return pos,false
	}

	// 解码 B 图片
	imgB, err := png.Decode(b)
	if err != nil {
		return pos,false
	}

	// 将 A 图片的类型转换为 *image.NRGBA
	imgANRGBA := image.NewNRGBA(imgA.Bounds())
	if imgANRGBA == nil {
		imgANRGBA = image.NewNRGBA(imgA.Bounds())
	}
	color.NRGBA{}.RGBA()
	draw.Draw(imgANRGBA, imgANRGBA.Bounds(), imgA, imgA.Bounds().Min, draw.Src)


	// 将 B 图片的类型转换为 *image.NRGBA
	imgBNRGBA := image.NewNRGBA(imgB.Bounds())
	if imgBNRGBA == nil {
		imgBNRGBA = image.NewNRGBA(imgB.Bounds())
	}
	color.NRGBA{}.RGBA()
	draw.Draw(imgBNRGBA, imgBNRGBA.Bounds(), imgB, imgB.Bounds().Min, draw.Src)

	// 获取 A 图片的像素数据
	boundsA := imgANRGBA.Bounds()
	pixA := make([]byte, boundsA.Dx()*boundsA.Dy()*4)
	copy(pixA, imgANRGBA.Pix)

	// 获取 B 图片的像素数据
	boundsB := imgBNRGBA.Bounds()
	pixB := make([]byte, boundsB.Dx()*boundsB.Dy()*4)
	copy(pixB, imgBNRGBA.Pix)

	// 查找 B 图片在 A 图片中的位置
	found := false
	for y := 0; y < boundsA.Dy()-boundsB.Dy(); y++ {
		for x := 0; x < boundsA.Dx()-boundsB.Dx(); x++ {
			match := true
			for j := 0; j < boundsB.Dy() && match; j++ {
				for i := 0; i < boundsB.Dx() && match; i++ {
					if pixA[(y+j)*boundsA.Dx()*4+(x+i)*4] != pixB[j*boundsB.Dx()*4+i*4] ||
						pixA[(y+j)*boundsA.Dx()*4+(x+i)*4+1] != pixB[j*boundsB.Dx()*4+i*4+1] ||
						pixA[(y+j)*boundsA.Dx()*4+(x+i)*4+2] != pixB[j*boundsB.Dx()*4+i*4+2] ||
						pixA[(y+j)*boundsA.Dx()*4+(x+i)*4+3] != pixB[j*boundsB.Dx()*4+i*4+3] {
						match = false
					}
				}
			}

			if match {
				pos.X=x
				pos.Y=y
				fmt.Printf("B 图片在 A 图片中的位置为 (%d, %d)\n", x, y)
				found = true
			}
		}
	}

	if !found {
		return pos,false
	}
	return pos,true
}