|
|
package com.example.jmacro;
|
|
|
|
|
|
import cn.hutool.core.img.ImgUtil;
|
|
|
import cn.hutool.json.JSONUtil;
|
|
|
import org.w3c.dom.css.Rect;
|
|
|
|
|
|
import java.awt.*;
|
|
|
import java.awt.image.BufferedImage;
|
|
|
import java.io.File;
|
|
|
|
|
|
public class Main {
|
|
|
|
|
|
static Robot robot;
|
|
|
|
|
|
static {
|
|
|
try {
|
|
|
robot = new Robot();
|
|
|
} catch (AWTException e) {
|
|
|
throw new RuntimeException(e);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
public static void main(String[] args) throws AWTException {
|
|
|
int[] aa = findImage(loadScreen(), loadAnchor("野外"));
|
|
|
|
|
|
System.out.println(JSONUtil.toJsonStr(aa));
|
|
|
}
|
|
|
|
|
|
public static Rect findImage(BufferedImage bigImage, BufferedImage smallImage) {
|
|
|
int bigWidth = bigImage.getWidth();
|
|
|
int bigHeight = bigImage.getHeight();
|
|
|
int smallWidth = smallImage.getWidth();
|
|
|
int smallHeight = smallImage.getHeight();
|
|
|
int[][] bigData = getImageRGB(bigImage);
|
|
|
int[][] smallData = getImageRGB(smallImage);
|
|
|
int[] target = {-1, -1};
|
|
|
int yEnd = bigHeight - smallHeight;
|
|
|
int xEnd = bigWidth - smallWidth;
|
|
|
|
|
|
for (int y = 0; y < yEnd; y++) {
|
|
|
for (int x = 0; x < xEnd; x++) {
|
|
|
// 对关键点进行先期匹配，降低运算复杂度。如果关键点本身就不匹配，就没必要再去匹配小图的每一个像素点
|
|
|
if (ColorUtil.isSimilar(bigData[x][y], smallData[0][0]) && // 左上角
|
|
|
ColorUtil.isSimilar(bigData[x + smallWidth - 1][y], smallData[smallWidth - 1][0]) &&// 右上角
|
|
|
ColorUtil.isSimilar(bigData[x][y + smallHeight - 1], smallData[0][smallHeight - 1]) &&// 左下角
|
|
|
ColorUtil.isSimilar(bigData[x + smallWidth - 1][y + smallHeight - 1], smallData[smallWidth - 1][smallHeight - 1]) && // 右下角
|
|
|
ColorUtil.isSimilar(bigData[x + smallWidth / 2][y + smallHeight / 2], smallData[smallWidth / 2][smallHeight / 2]) //中心点
|
|
|
) {
|
|
|
|
|
|
// 进行全像素匹配
|
|
|
boolean isMatched = checkAllMatch(x, y, smallHeight, smallWidth, bigData, smallData);
|
|
|
if (isMatched) {
|
|
|
System.out.println("像素点X" + x + " : Y" + y + "，对应的值为：" + bigData[x][y]);
|
|
|
|
|
|
// 获取小图的中心位置的点
|
|
|
int centerX = x + smallWidth / 2;
|
|
|
int centerY = y + smallHeight / 2;
|
|
|
target[0] = centerX;
|
|
|
target[1] = centerY;
|
|
|
return new int[][]{{x,y},};
|
|
|
return new Rect
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
|
|
}
|
|
|
|
|
|
return target;
|
|
|
}
|
|
|
|
|
|
public static int[][] getImageRGB(BufferedImage bfImage) {
|
|
|
int width = bfImage.getWidth();
|
|
|
int height = bfImage.getHeight();
|
|
|
int[][] result = new int[width][height];
|
|
|
for (int y = 0; y < height; y++) {
|
|
|
for (int x = 0; x < width; x++) {
|
|
|
// 对某个像素点的RGB编码并存入数据库
|
|
|
result[x][y] = bfImage.getRGB(x, y) & 0xFFFFFF;
|
|
|
|
|
|
// 单独获取每一个像素点的Red，Green，和Blue的值。
|
|
|
// int r = (bfImage.getRGB(x, y) & 0xFF0000) >> 16;
|
|
|
// int g = (bfImage.getRGB(x, y) & 0xFF00) >> 8;
|
|
|
// int b = bfImage.getRGB(x, y) & 0xFF;
|
|
|
}
|
|
|
}
|
|
|
return result;
|
|
|
}
|
|
|
|
|
|
private static boolean checkAllMatch(int x, int y, int smallHeight, int smallWidth, int[][] bigData, int[][] smallData) {
|
|
|
boolean isMatched = true;
|
|
|
for (int smallY = 0; smallY < smallHeight; smallY++) {
|
|
|
for (int smallX = 0; smallX < smallWidth; smallX++) {
|
|
|
// 如果发现有一个像素点，两者的值不一样，则认为不相等，如果不相等，则没必要继续比较其它点.
|
|
|
if (!ColorUtil.isSimilar(bigData[x + smallX][y + smallY], smallData[smallX][smallY])) {
|
|
|
isMatched = false;
|
|
|
return isMatched;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
return isMatched;
|
|
|
}
|
|
|
|
|
|
|
|
|
// 获取屏幕截图
|
|
|
static BufferedImage loadScreen() {
|
|
|
// 获取缺省工具包
|
|
|
Toolkit tk = Toolkit.getDefaultToolkit();
|
|
|
// 屏幕尺寸规格
|
|
|
Dimension di = tk.getScreenSize();
|
|
|
LogUtil.info("获取屏幕尺寸（{},{}）", di.width, di.height);
|
|
|
Rectangle rec = new Rectangle(0, 0, di.width, di.height);
|
|
|
return robot.createScreenCapture(rec);
|
|
|
}
|
|
|
|
|
|
static BufferedImage loadAnchor(String anchor) {
|
|
|
return ImgUtil.read(new File("anchor", anchor + ".png"));
|
|
|
}
|
|
|
}
|
