66 lines
2.0 KiB
Java
66 lines
2.0 KiB
Java
import java.awt.image.BufferedImage;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
|
|
import javax.imageio.ImageIO;
|
|
|
|
public class InventoryItem {
|
|
|
|
private BufferedImage itemImage;
|
|
private int minimumNumberOfMatchingPixels;
|
|
|
|
public InventoryItem(String itemDirectoryPath, String itemName) throws IOException {
|
|
initializeImage(itemDirectoryPath, itemName);
|
|
this.minimumNumberOfMatchingPixels = 250;
|
|
}
|
|
|
|
private void initializeImage(String itemDirectoryPath, String itemName) throws IOException {
|
|
File itemImageFile = new File(joinPathWithFileName(itemDirectoryPath, itemName));
|
|
this.itemImage = ImageIO.read(itemImageFile);
|
|
}
|
|
|
|
private String joinPathWithFileName(String itemDirectoryPath, String itemName) {
|
|
return (itemDirectoryPath + itemName);
|
|
}
|
|
|
|
public boolean itemMatchesImage(BufferedImage itemImageToCompare) {
|
|
if (imagesAreTheSameSize(itemImageToCompare)) {
|
|
return imagesMatch(itemImageToCompare, 5);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private boolean imagesAreTheSameSize(BufferedImage itemImageToCompare) {
|
|
return (itemImage.getWidth() == itemImageToCompare.getWidth()) && (itemImage.getHeight() == itemImageToCompare.getHeight());
|
|
}
|
|
|
|
|
|
private boolean imagesMatch(BufferedImage itemImageToCompare, int pixelTolerance) {
|
|
int numberOfMatchingPixels = 0;
|
|
for (int row = 0; row < itemImageToCompare.getWidth(); row++) {
|
|
for (int col = 0; col < itemImageToCompare.getHeight(); col++) {
|
|
if (pixelsAreWithinRGBTolerance(itemImage.getRGB(row, col), itemImageToCompare.getRGB(row, col))) {
|
|
numberOfMatchingPixels++;
|
|
}
|
|
}
|
|
}
|
|
return (numberOfMatchingPixels > this.minimumNumberOfMatchingPixels);
|
|
}
|
|
|
|
private boolean pixelsAreWithinRGBTolerance(int rgb1, int rgb2) {
|
|
int[] colors1 = getRGBValuesFromPixel(rgb1);
|
|
int[] colors2 = getRGBValuesFromPixel(rgb2);
|
|
for (int i = 0; i < 3; i++) {
|
|
if (Math.abs(colors1[i] - colors2[i]) > 3) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
private int[] getRGBValuesFromPixel(int pixel) {
|
|
int[] colors = {(pixel)&0xFF, (pixel>>8)&0xFF, (pixel>>16)&0xFF, (pixel>>24)&0xFF};
|
|
return colors;
|
|
}
|
|
}
|