NotMyWing
308 lines
15 KiB
Java
308 lines
15 KiB
Java
package appeng.integration.modules.jei;
|
|
|
|
import appeng.api.AEApi;
|
|
import appeng.api.config.FuzzyMode;
|
|
import appeng.api.storage.channels.IItemStorageChannel;
|
|
import appeng.api.storage.data.IAEItemStack;
|
|
import appeng.api.storage.data.IItemList;
|
|
import appeng.container.implementations.ContainerMEMonitorable;
|
|
import appeng.helpers.IContainerCraftingPacket;
|
|
import appeng.util.Platform;
|
|
import appeng.util.item.AEItemStack;
|
|
import mezz.jei.api.gui.IGuiIngredient;
|
|
import mezz.jei.api.gui.IRecipeLayout;
|
|
import mezz.jei.api.recipe.transfer.IRecipeTransferError;
|
|
import mezz.jei.gui.TooltipRenderer;
|
|
import mezz.jei.gui.recipes.RecipeLayout;
|
|
import mezz.jei.gui.recipes.RecipeTransferButton;
|
|
import net.minecraft.client.Minecraft;
|
|
import net.minecraft.client.gui.ScaledResolution;
|
|
import net.minecraft.inventory.Container;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraftforge.items.IItemHandler;
|
|
import net.minecraftforge.items.wrapper.PlayerMainInvWrapper;
|
|
|
|
import javax.annotation.Nonnull;
|
|
import java.awt.*;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
|
|
import static mezz.jei.api.recipe.transfer.IRecipeTransferError.Type.USER_FACING;
|
|
import static net.minecraft.client.resources.I18n.format;
|
|
|
|
public class JEIMissingItem implements IRecipeTransferError {
|
|
private final IRecipeLayout recipeLayout;
|
|
private boolean errored;
|
|
public long lastUpdate;
|
|
private final List<Integer> craftableSlots = new ArrayList<>();
|
|
private final List<Integer> foundSlots = new ArrayList<>();
|
|
|
|
IItemList<IAEItemStack> available = AEApi.instance().storage().getStorageChannel(IItemStorageChannel.class).createList();
|
|
|
|
IItemList<IAEItemStack> used = AEApi.instance().storage().getStorageChannel(IItemStorageChannel.class).createList();
|
|
private boolean foundAny;
|
|
|
|
JEIMissingItem(Container container, @Nonnull IRecipeLayout recipeLayout) {
|
|
this.recipeLayout = recipeLayout;
|
|
|
|
if (container instanceof ContainerMEMonitorable) {
|
|
IItemList<IAEItemStack> ir = ((ContainerMEMonitorable) container).items;
|
|
|
|
IItemList<IAEItemStack> available = mergeInventories(ir, (ContainerMEMonitorable) container);
|
|
|
|
boolean found;
|
|
this.errored = false;
|
|
recipeLayout.getItemStacks().addTooltipCallback(new CraftableCallBack(container, available));
|
|
|
|
IItemList<IAEItemStack> used = AEApi.instance().storage().getStorageChannel(IItemStorageChannel.class).createList();
|
|
for (IGuiIngredient<?> i : recipeLayout.getItemStacks().getGuiIngredients().values()) {
|
|
found = false;
|
|
if (i.isInput() && !i.getAllIngredients().isEmpty()) {
|
|
List<?> allIngredients = i.getAllIngredients();
|
|
for (Object allIngredient : allIngredients) {
|
|
if (allIngredient instanceof ItemStack stack) {
|
|
if (!stack.isEmpty()) {
|
|
IAEItemStack search = AEItemStack.fromItemStack(stack);
|
|
if (stack.getItem().isDamageable() || Platform.isGTDamageableItem(stack.getItem())) {
|
|
Collection<IAEItemStack> fuzzy = available.findFuzzy(search, FuzzyMode.IGNORE_ALL);
|
|
if (!fuzzy.isEmpty()) {
|
|
for (IAEItemStack itemStack : fuzzy) {
|
|
if (itemStack.getStackSize() > 0) {
|
|
if (Platform.isGTDamageableItem(stack.getItem())) {
|
|
if (!(stack.getMetadata() == itemStack.getDefinition().getMetadata())) {
|
|
continue;
|
|
}
|
|
}
|
|
found = true;
|
|
used.add(itemStack.copy().setStackSize(1));
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
IAEItemStack ext = available.findPrecise(search);
|
|
if (ext != null) {
|
|
IAEItemStack usedStack = used.findPrecise(ext);
|
|
if (ext.getStackSize() > 0 && (usedStack == null || ext.getStackSize() > usedStack.getStackSize())) {
|
|
used.add(ext.copy().setStackSize(1));
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
if (!found) {
|
|
this.errored = true;
|
|
} else{
|
|
this.foundAny = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public Type getType() {
|
|
// Workaround. Re-enable the button if not errored.
|
|
if (this.errored && this.foundAny && this.recipeLayout instanceof RecipeLayout castedRecipeLayout) {
|
|
var recipeTransferButton = castedRecipeLayout.getRecipeTransferButton();
|
|
if (recipeTransferButton != null) {
|
|
recipeTransferButton.enabled = true;
|
|
}
|
|
}
|
|
return USER_FACING;
|
|
}
|
|
|
|
@Override
|
|
public void showError(Minecraft minecraft, int mouseX, int mouseY, @Nonnull IRecipeLayout recipeLayout, int recipeX, int recipeY) {
|
|
Container c = minecraft.player.openContainer;
|
|
if (c instanceof ContainerMEMonitorable container) {
|
|
IItemList<IAEItemStack> ir = ((ContainerMEMonitorable) c).items;
|
|
boolean found;
|
|
boolean craftable;
|
|
|
|
int foundMissing = 0;
|
|
int foundCraftables = 0;
|
|
int currentSlot = 0;
|
|
|
|
this.errored = false;
|
|
this.foundAny = false;
|
|
|
|
if (System.currentTimeMillis() - lastUpdate > 1000) {
|
|
lastUpdate = System.currentTimeMillis();
|
|
available = mergeInventories(ir, container);
|
|
this.foundSlots.clear();
|
|
this.craftableSlots.clear();
|
|
} else {
|
|
for (IGuiIngredient<?> i : recipeLayout.getItemStacks().getGuiIngredients().values()) {
|
|
if (i.isInput()) {
|
|
if (!foundSlots.contains(currentSlot)) {
|
|
if (craftableSlots.contains(currentSlot)) {
|
|
i.drawHighlight(minecraft, new Color(0.0f, 0.0f, 1.0f, 0.4f), recipeX, recipeY);
|
|
} else {
|
|
i.drawHighlight(minecraft, new Color(1.0f, 0.0f, 0.0f, 0.4f), recipeX, recipeY);
|
|
}
|
|
}
|
|
}
|
|
currentSlot++;
|
|
}
|
|
return;
|
|
}
|
|
this.used.resetStatus();
|
|
|
|
for (IGuiIngredient<?> i : recipeLayout.getItemStacks().getGuiIngredients().values()) {
|
|
found = false;
|
|
craftable = false;
|
|
IItemList<IAEItemStack> valid = AEApi.instance().storage().getStorageChannel(IItemStorageChannel.class).createList();
|
|
|
|
if (i.isInput()) {
|
|
List<?> allIngredients = i.getAllIngredients();
|
|
for (Object allIngredient : allIngredients) {
|
|
if (allIngredient instanceof ItemStack stack) {
|
|
if (!stack.isEmpty()) {
|
|
IAEItemStack search = AEItemStack.fromItemStack(stack);
|
|
if (stack.getItem().isDamageable() || Platform.isGTDamageableItem(stack.getItem())) {
|
|
Collection<IAEItemStack> fuzzy = available.findFuzzy(search, FuzzyMode.IGNORE_ALL);
|
|
if (!fuzzy.isEmpty()) {
|
|
for (IAEItemStack itemStack : fuzzy) {
|
|
if (itemStack.getStackSize() > 0) {
|
|
if (Platform.isGTDamageableItem(stack.getItem())) {
|
|
if (!(stack.getMetadata() == itemStack.getDefinition().getMetadata())) {
|
|
continue;
|
|
}
|
|
}
|
|
found = true;
|
|
used.add(itemStack.copy().setStackSize(1));
|
|
valid.add(itemStack.copy().setStackSize(1));
|
|
} else {
|
|
if (itemStack.isCraftable()) {
|
|
craftable = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
IAEItemStack ext = available.findPrecise(search);
|
|
if (ext != null) {
|
|
IAEItemStack usedStack = used.findPrecise(ext);
|
|
if (ext.getStackSize() > 0 && (usedStack == null || usedStack.getStackSize() < ext.getStackSize())) {
|
|
used.add(ext.copy().setStackSize(1));
|
|
if (craftable) {
|
|
valid.resetStatus();
|
|
}
|
|
valid.add(ext.copy().setStackSize(1));
|
|
found = true;
|
|
} else if (ext.isCraftable()) {
|
|
valid.add(ext.copy().setStackSize(1));
|
|
craftable = true;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
if (i.getAllIngredients().isEmpty()) {
|
|
currentSlot++;
|
|
continue;
|
|
}
|
|
ArrayList<ItemStack> validStacks = new ArrayList<>();
|
|
valid.forEach(v -> {
|
|
if (v.getStackSize() > 0) {
|
|
ItemStack validStack = v.createItemStack();
|
|
validStack.setCount(1);
|
|
validStacks.add(validStack);
|
|
}
|
|
});
|
|
if (!found) {
|
|
if (craftable) {
|
|
i.drawHighlight(minecraft, new Color(0.0f, 0.0f, 1.0f, 0.4f), recipeX, recipeY);
|
|
this.craftableSlots.add(currentSlot);
|
|
recipeLayout.getItemStacks().set(currentSlot, validStacks);
|
|
foundCraftables++;
|
|
} else {
|
|
i.drawHighlight(minecraft, new Color(1.0f, 0.0f, 0.0f, 0.4f), recipeX, recipeY);
|
|
}
|
|
this.errored = true;
|
|
foundMissing++;
|
|
} else {
|
|
foundAny = true;
|
|
this.foundSlots.add(currentSlot);
|
|
recipeLayout.getItemStacks().set(currentSlot, validStacks);
|
|
}
|
|
}
|
|
currentSlot++;
|
|
}
|
|
RecipeTransferButton b = ((RecipeLayout) recipeLayout).getRecipeTransferButton();
|
|
if (b != null) {
|
|
List<String> tooltipLines = new ArrayList<>();
|
|
b.init(c, minecraft.player);
|
|
if (errored && foundAny) {
|
|
tooltipLines.add(format("gui.tooltips.appliedenergistics2.PartialTransfer"));
|
|
b.enabled = true;
|
|
b.visible = true;
|
|
|
|
tooltipLines.add(format("gui.tooltips.appliedenergistics2.CondenseItems"));
|
|
}
|
|
|
|
if (foundMissing > 0) {
|
|
tooltipLines.add(format("gui.tooltips.appliedenergistics2.MissingItem", String.valueOf(foundMissing)));
|
|
}
|
|
if (foundCraftables > 0) {
|
|
tooltipLines.add(format("gui.tooltips.appliedenergistics2.CraftableItem", foundCraftables));
|
|
}
|
|
|
|
var longestStringWidth = minecraft.fontRenderer.getStringWidth(tooltipLines.stream()
|
|
.max(Comparator.comparingInt(String::length)).get());
|
|
|
|
var background = ((RecipeLayout) recipeLayout).getRecipeCategory().getBackground();
|
|
var scaledresolution = new ScaledResolution(minecraft);
|
|
|
|
// Mostly reverse-engineered Minecraft code.
|
|
final int offset;
|
|
if (mouseX + longestStringWidth + 4 + 12 > scaledresolution.getScaledWidth()) {
|
|
// The tooltip will appear to the left of the mouse cursor.
|
|
// Need to offset Y so that the tooltip doesn't block the ingredients.
|
|
offset = background.getHeight() + recipeY
|
|
+ (minecraft.fontRenderer.FONT_HEIGHT * tooltipLines.size()
|
|
+ 2 * (tooltipLines.size() - 1)) / 2
|
|
+ 4;
|
|
} else {
|
|
offset = mouseY;
|
|
}
|
|
TooltipRenderer.drawHoveringText(minecraft, tooltipLines, mouseX, offset);
|
|
}
|
|
}
|
|
}
|
|
|
|
IItemList<IAEItemStack> mergeInventories(IItemList<IAEItemStack> repo, ContainerMEMonitorable containerCraftingTerm) {
|
|
IItemList<IAEItemStack> itemList = AEApi.instance().storage().getStorageChannel(IItemStorageChannel.class).createList();
|
|
for (IAEItemStack i : repo) {
|
|
itemList.addStorage(i);
|
|
}
|
|
|
|
PlayerMainInvWrapper invWrapper = new PlayerMainInvWrapper(containerCraftingTerm.getPlayerInv());
|
|
for (int i = 0; i < invWrapper.getSlots(); i++) {
|
|
itemList.addStorage(AEItemStack.fromItemStack(invWrapper.getStackInSlot(i)));
|
|
}
|
|
|
|
if (containerCraftingTerm instanceof IContainerCraftingPacket) {
|
|
IItemHandler itemHandler = ((IContainerCraftingPacket) containerCraftingTerm).getInventoryByName("crafting");
|
|
for (int i = 0; i < itemHandler.getSlots(); i++) {
|
|
itemList.addStorage(AEItemStack.fromItemStack(itemHandler.getStackInSlot(i)));
|
|
}
|
|
}
|
|
return itemList;
|
|
}
|
|
|
|
public boolean errored() {
|
|
return errored;
|
|
}
|
|
}
|