78 lines
2.3 KiB
Java
78 lines
2.3 KiB
Java
import java.awt.AWTException;
|
|
import java.awt.Rectangle;
|
|
import java.awt.Robot;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.IOException;
|
|
|
|
public class Inventory {
|
|
|
|
public static final int INVENTORY_OFFSET_WIDTH = 655; //top left corner of inventory, fromm top left corner of screen
|
|
public static final int INVENTORY_OFFSET_HEIGHT = 290;
|
|
public static final int INVENTORY_WIDTH = 820 - 649;// 820
|
|
public static final int INVENTORY_HEIGHT = 350; // 530
|
|
|
|
public static final int NUM_ROWS = 4;
|
|
public static final int NUM_COLUMNS = 7;
|
|
|
|
Robot robot;
|
|
Rectangle inventoryRectangleToCapture;
|
|
InventorySlot[][] inventorySlots;
|
|
InventoryItems items;
|
|
|
|
public Inventory() throws AWTException, IOException {
|
|
initializeInventoryRectangle();
|
|
initializeInventorySlots();
|
|
initializeItems();
|
|
robot = new Robot();
|
|
}
|
|
|
|
private void initializeInventoryRectangle() {
|
|
inventoryRectangleToCapture = new Rectangle(INVENTORY_OFFSET_WIDTH, INVENTORY_OFFSET_HEIGHT, INVENTORY_WIDTH, INVENTORY_HEIGHT);
|
|
}
|
|
|
|
private void initializeInventorySlots() {
|
|
inventorySlots = new InventorySlot[4][7];
|
|
for (int row = 0; row < 4; row++) {
|
|
for (int column = 0; column < 7; column++) {
|
|
inventorySlots[row][column] = new InventorySlot(row, column);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void initializeItems() throws IOException {
|
|
items = new InventoryItems("/home/dpapp/Desktop/RunescapeAIPics/Items/");
|
|
}
|
|
|
|
public void update() throws IOException {
|
|
BufferedImage image = robot.createScreenCapture(this.inventoryRectangleToCapture);
|
|
updateAllInventorySlots(image);
|
|
}
|
|
|
|
public void updateWithFakeImageForTests(BufferedImage testImage) throws IOException {
|
|
updateAllInventorySlots(testImage);
|
|
}
|
|
|
|
private void updateAllInventorySlots(BufferedImage image) throws IOException {
|
|
for (int row = 0; row < 4; row++) {
|
|
for (int column = 0; column < 7; column++) {
|
|
inventorySlots[row][column].updateInventorySlot(image);
|
|
}
|
|
}
|
|
}
|
|
|
|
public String getItemNameInInventorySlot(int row, int column) {
|
|
return inventorySlots[row][column].getItemNameInInventorySlot(items);
|
|
}
|
|
|
|
public boolean isInventoryFull() {
|
|
for (int row = 0; row < 4; row++) {
|
|
for (int column = 0; column < 7; column++) {
|
|
if (!inventorySlots[row][column].isInventorySlotEmpty(items)) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|