1 package com.jebysun.test.imagefind; 2 3 import java.awt.AWTException; 4 import java.awt.Rectangle; 5 import java.awt.Robot; 6 import java.awt.Toolkit; 7 import java.awt.image.BufferedImage; 8 import java.io.File; 9 import java.io.IOException; 10 11 import javax.imageio.ImageIO; 12 /** 13 * 屏幕上查找指定圖片 14 * @author Jeby Sun 15 * @date 2014-09-13 16 * @website http://www.jebysun.com 17 */ 18 public class ImageFindDemo { 19 20 BufferedImage screenShotImage; //屏幕截圖 21 BufferedImage keyImage; //查找目標圖片 22 23 int scrShotImgWidth; //屏幕截圖寬度 24 int scrShotImgHeight; //屏幕截圖高度 25 26 int keyImgWidth; //查找目標圖片寬度 27 int keyImgHeight; //查找目標圖片高度 28 29 int[][] screenShotImageRGBData; //屏幕截圖RGB數據 30 int[][] keyImageRGBData; //查找目標圖片RGB數據 31 32 int[][][] findImgData; //查找結果，目標圖標位于屏幕截圖上的坐標數據 33 34 35 public ImageFindDemo(String keyImagePath) { 36 screenShotImage = this.getFullScreenShot(); 37 keyImage = this.getBfImageFromPath(keyImagePath); 38 screenShotImageRGBData = this.getImageGRB(screenShotImage); 39 keyImageRGBData = this.getImageGRB(keyImage); 40 scrShotImgWidth = screenShotImage.getWidth(); 41 scrShotImgHeight = screenShotImage.getHeight(); 42 keyImgWidth = keyImage.getWidth(); 43 keyImgHeight = keyImage.getHeight(); 44 45 //開始查找 46 this.findImage(); 47 48 } 49 50 /** 51 * 全屏截圖 52 * @return 返回BufferedImage 53 */ 54 public BufferedImage getFullScreenShot() { 55 BufferedImage bfImage = null; 56 int width = (int) Toolkit.getDefaultToolkit().getScreenSize().getWidth(); 57 int height = (int) Toolkit.getDefaultToolkit().getScreenSize().getHeight(); 58 try { 59 Robot robot = new Robot(); 60 bfImage = robot.createScreenCapture(new Rectangle(0, 0, width, height)); 61 } catch (AWTException e) { 62 e.printStackTrace(); 63 } 64 return bfImage; 65 } 66 67 /** 68 * 從本地文件讀取目標圖片 69 * @param keyImagePath - 圖片絕對路徑 70 * @return 本地圖片的BufferedImage對象 71 */ 72 public BufferedImage getBfImageFromPath(String keyImagePath) { 73 BufferedImage bfImage = null; 74 try { 75 bfImage = ImageIO.read(new File(keyImagePath)); 76 } catch (IOException e) { 77 e.printStackTrace(); 78 } 79 return bfImage; 80 } 81 82 /** 83 * 根據BufferedImage獲取圖片RGB數組 84 * @param bfImage 85 * @return 86 */ 87 public int[][] getImageGRB(BufferedImage bfImage) { 88 int width = bfImage.getWidth(); 89 int height = bfImage.getHeight(); 90 int[][] result = new int[height][width]; 91 for (int h = 0; h < height; h++) { 92 for (int w = 0; w < width; w++) { 93 //使用getRGB(w, h)獲取該點的顏色值是ARGB，而在實際應用中使用的是RGB，所以需要將ARGB轉化成RGB，即bufImg.getRGB(w, h) & 0xFFFFFF。 94 result[h][w] = bfImage.getRGB(w, h) & 0xFFFFFF; 95 } 96 } 97 return result; 98 } 99 100 101 /** 102 * 查找圖片 103 */ 104 public void findImage() { 105 findImgData = new int[keyImgHeight][keyImgWidth][2]; 106 //遍歷屏幕截圖像素點數據 107 for(int y=0; y<scrShotImgHeight-keyImgHeight; y++) { 108 for(int x=0; x<scrShotImgWidth-keyImgWidth; x++) { 109 //根據目標圖的尺寸，得到目標圖四個角映射到屏幕截圖上的四個點， 110 //判斷截圖上對應的四個點與圖B的四個角像素點的值是否相同， 111 //如果相同就將屏幕截圖上映射范圍內的所有的點與目標圖的所有的點進行比較。 112 if((keyImageRGBData[0][0]^screenShotImageRGBData[y][x])==0 113 && (keyImageRGBData[0][keyImgWidth-1]^screenShotImageRGBData[y][x+keyImgWidth-1])==0 114 && (keyImageRGBData[keyImgHeight-1][keyImgWidth-1]^screenShotImageRGBData[y+keyImgHeight-1][x+keyImgWidth-1])==0 115 && (keyImageRGBData[keyImgHeight-1][0]^screenShotImageRGBData[y+keyImgHeight-1][x])==0) { 116 117 boolean isFinded = isMatchAll(y, x); 118 //如果比較結果完全相同，則說明圖片找到，填充查找到的位置坐標數據到查找結果數組。 119 if(isFinded) { 120 for(int h=0; h<keyImgHeight; h++) { 121 for(int w=0; w<keyImgWidth; w++) { 122 findImgData[h][w][0] = y+h; 123 findImgData[h][w][1] = x+w; 124 } 125 } 126 return; 127 } 128 } 129 } 130 } 131 } 132 133 /** 134 * 判斷屏幕截圖上目標圖映射范圍內的全部點是否全部和小圖的點一一對應。 135 * @param y - 與目標圖左上角像素點想匹配的屏幕截圖y坐標 136 * @param x - 與目標圖左上角像素點想匹配的屏幕截圖x坐標 137 * @return 138 */ 139 public boolean isMatchAll(int y, int x) { 140 int biggerY = 0; 141 int biggerX = 0; 142 int xor = 0; 143 for(int smallerY=0; smallerY<keyImgHeight; smallerY++) { 144 biggerY = y+smallerY; 145 for(int smallerX=0; smallerX<keyImgWidth; smallerX++) { 146 biggerX = x+smallerX; 147 if(biggerY>=scrShotImgHeight || biggerX>=scrShotImgWidth) { 148 return false; 149 } 150 xor = keyImageRGBData[smallerY][smallerX]^screenShotImageRGBData[biggerY][biggerX]; 151 if(xor!=0) { 152 return false; 153 } 154 } 155 biggerX = x; 156 } 157 return true; 158 } 159 160 /** 161 * 輸出查找到的坐標數據 162 */ 163 private void printFindData() { 164 for(int y=0; y<keyImgHeight; y++) { 165 for(int x=0; x<keyImgWidth; x++) { 166 System.out.print("("+this.findImgData[y][x][0]+", "+this.findImgData[y][x][1]+")"); 167 } 168 System.out.println(); 169 } 170 } 171 172 173 public static void main(String[] args) { 174 String keyImagePath = "D:/key.png"; 175 ImageFindDemo demo = new ImageFindDemo(keyImagePath); 176 demo.printFindData(); 177 } 178 179 } |