Merge pull request #896 from Gu-ZT/releases/1.20.1

实现Emi拖拽设置过滤

Author: Gu-ZT

common/src/main/java/dev/dubhe/anvilcraft/api/depository/FilteredItemDepository.java

@@ -19,9 +19,9 @@
 public class FilteredItemDepository extends ItemDepository {
 
     public static final Codec<FilteredItemDepository> CODEC = RecordCodecBuilder.create(ins -> ins.group(
-            Codec.BOOL.fieldOf("filterEnabled").forGetter(o -> o.filterEnabled),
-            ItemStack.CODEC.listOf().fieldOf("filteredItems").forGetter(o -> o.filteredItems),
-            Codec.BOOL.listOf().fieldOf("disabled").forGetter(o -> o.disabled)
+        Codec.BOOL.fieldOf("filterEnabled").forGetter(o -> o.filterEnabled),
+        ItemStack.CODEC.listOf().fieldOf("filteredItems").forGetter(o -> o.filteredItems),
+        Codec.BOOL.listOf().fieldOf("disabled").forGetter(o -> o.disabled)
     ).apply(ins, FilteredItemDepository::new));
 
     private boolean filterEnabled = false;
@@ -140,10 +140,12 @@ public boolean isFiltered(int slot, ItemStack stack) {
      * @param slot  槽位
      * @param stack 过滤物品堆栈（不检查NBT）
      */
-    public void setFilter(int slot, @NotNull ItemStack stack) {
-        if (stack.isEmpty()) return;
+    public boolean setFilter(int slot, @NotNull ItemStack stack) {
+        if (slot < 0 || slot >= this.filteredItems.size()) return false;
+        if (stack.isEmpty()) return false;
         this.setSlotDisabled(slot, false);
         this.filteredItems.set(slot, new ItemStack(stack.getItem(), 1));
+        return true;
     }
 
     /**
@@ -203,18 +205,18 @@ public void deserializeNbt(@NotNull CompoundTag tag) {
      */
     public CompoundTag serializeFiltering() {
         return (CompoundTag) CODEC.encodeStart(NbtOps.INSTANCE, this)
-                .getOrThrow(false, e -> {
-                });
+            .getOrThrow(false, e -> {
+            });
     }
 
     /**
      *
      */
     public void deserializeFiltering(@NotNull CompoundTag tag) {
         FilteredItemDepository depository = CODEC.decode(NbtOps.INSTANCE, tag)
-                .getOrThrow(false, s -> {
-                })
-                .getFirst();
+            .getOrThrow(false, s -> {
+            })
+            .getFirst();
         if (this.getSize() != depository.getSize()) throw new IllegalArgumentException("Depository size mismatch");
         this.filterEnabled = tag.getBoolean("filterEnabled");
         int size = depository.filteredItems.size();

common/src/main/java/dev/dubhe/anvilcraft/block/entity/IFilterBlockEntity.java

@@ -76,7 +76,7 @@ default ItemStack getFilter(int slot) {
      * @param slot   槽位
      * @param filter 过滤
      */
-    default void setFilter(int slot, ItemStack filter) {
-        this.getFilteredItemDepository().setFilter(slot, filter);
+    default boolean setFilter(int slot, ItemStack filter) {
+        return this.getFilteredItemDepository().setFilter(slot, filter);
     }
 }

common/src/main/java/dev/dubhe/anvilcraft/client/gui/screen/inventory/AutoCrafterScreen.java

@@ -4,7 +4,6 @@
 import dev.dubhe.anvilcraft.api.depository.ItemDepositorySlot;
 import dev.dubhe.anvilcraft.client.gui.component.EnableFilterButton;
 import dev.dubhe.anvilcraft.inventory.AutoCrafterMenu;
-import dev.dubhe.anvilcraft.inventory.IFilterMenu;
 import dev.dubhe.anvilcraft.network.SlotDisableChangePack;
 import lombok.Getter;
 import net.minecraft.client.gui.GuiGraphics;
@@ -17,7 +16,7 @@
 
 import java.util.function.BiFunction;
 
-public class AutoCrafterScreen extends BaseMachineScreen<AutoCrafterMenu> implements IFilterScreen {
+public class AutoCrafterScreen extends BaseMachineScreen<AutoCrafterMenu> implements IFilterScreen<AutoCrafterMenu> {
     private static final ResourceLocation CONTAINER_LOCATION =
         AnvilCraft.of("textures/gui/container/machine/background/auto_crafter.png");
     BiFunction<Integer, Integer, EnableFilterButton> enableFilterButtonSupplier = this
@@ -67,7 +66,7 @@ protected void renderSlotTooltip(@NotNull GuiGraphics guiGraphics, int x, int y)
     }
 
     @Override
-    public IFilterMenu getFilterMenu() {
+    public AutoCrafterMenu getFilterMenu() {
         return this.menu;
     }
 
@@ -92,4 +91,14 @@ protected void slotClicked(@NotNull Slot slot, int slotId, int mouseButton, @Not
         }
         super.slotClicked(slot, slotId, mouseButton, type);
     }
+
+    @Override
+    public int getOffsetX() {
+        return (this.width - this.imageWidth) / 2;
+    }
+
+    @Override
+    public int getOffsetY() {
+        return (this.height - this.imageHeight) / 2;
+    }
 }

common/src/main/java/dev/dubhe/anvilcraft/client/gui/screen/inventory/ChuteScreen.java

@@ -5,7 +5,6 @@
 import dev.dubhe.anvilcraft.api.depository.ItemDepositorySlot;
 import dev.dubhe.anvilcraft.client.gui.component.EnableFilterButton;
 import dev.dubhe.anvilcraft.inventory.ChuteMenu;
-import dev.dubhe.anvilcraft.inventory.IFilterMenu;
 import dev.dubhe.anvilcraft.network.SlotDisableChangePack;
 import lombok.Getter;
 import net.minecraft.client.gui.GuiGraphics;
@@ -19,7 +18,7 @@
 
 import java.util.function.BiFunction;
 
-public class ChuteScreen extends BaseMachineScreen<ChuteMenu> implements IFilterScreen {
+public class ChuteScreen extends BaseMachineScreen<ChuteMenu> implements IFilterScreen<ChuteMenu> {
     private static final ResourceLocation CONTAINER_LOCATION =
         AnvilCraft.of("textures/gui/container/machine/background/chute.png");
 
@@ -72,7 +71,7 @@ protected void renderSlotTooltip(@NotNull GuiGraphics guiGraphics, int x, int y)
     }
 
     @Override
-    public IFilterMenu getFilterMenu() {
+    public ChuteMenu getFilterMenu() {
         return menu;
     }
 
@@ -98,4 +97,14 @@ protected void slotClicked(@NotNull Slot slot, int slotId, int mouseButton, @Not
         }
         super.slotClicked(slot, slotId, mouseButton, type);
     }
+
+    @Override
+    public int getOffsetX() {
+        return (this.width - this.imageWidth) / 2;
+    }
+
+    @Override
+    public int getOffsetY() {
+        return (this.height - this.imageHeight) / 2;
+    }
 }

common/src/main/java/dev/dubhe/anvilcraft/client/gui/screen/inventory/IFilterScreen.java

@@ -8,6 +8,7 @@
 import dev.dubhe.anvilcraft.network.MachineEnableFilterPack;
 import net.minecraft.client.gui.GuiGraphics;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.inventory.AbstractContainerMenu;
 import net.minecraft.world.inventory.Slot;
 import net.minecraft.world.item.ItemStack;
 import org.jetbrains.annotations.NotNull;
@@ -17,10 +18,10 @@
 /**
  * 有过滤的 GUI
  */
-public interface IFilterScreen {
+public interface IFilterScreen<T extends AbstractContainerMenu & IFilterMenu> {
     ResourceLocation DISABLED_SLOT = AnvilCraft.of("textures/gui/container/machine/disabled_slot.png");
 
-    IFilterMenu getFilterMenu();
+    T getFilterMenu();
 
     /**
      * 获取是否开启过滤
@@ -65,8 +66,8 @@ default boolean isSlotDisabled(int slot) {
      * @param slot   槽位
      * @param filter 过滤
      */
-    default void setFilter(int slot, ItemStack filter) {
-        this.getFilterMenu().setFilter(slot, filter);
+    default boolean setFilter(int slot, ItemStack filter) {
+        return this.getFilterMenu().setFilter(slot, filter);
     }
 
     /**
@@ -143,4 +144,12 @@ default void renderFilterItem(@NotNull GuiGraphics guiGraphics, @NotNull Slot sl
         guiGraphics.renderFakeItem(stack, i, j);
         guiGraphics.fill(i, j, i + 16, j + 16, 0x80ffaaaa);
     }
+
+    default int getOffsetY() {
+        return 0;
+    }
+
+    default int getOffsetX() {
+        return 0;
+    }
 }

common/src/main/java/dev/dubhe/anvilcraft/client/gui/screen/inventory/ItemCollectorScreen.java

@@ -5,7 +5,6 @@
 import dev.dubhe.anvilcraft.client.gui.component.EnableFilterButton;
 import dev.dubhe.anvilcraft.client.gui.component.ItemCollectorButton;
 import dev.dubhe.anvilcraft.client.gui.component.TextWidget;
-import dev.dubhe.anvilcraft.inventory.IFilterMenu;
 import dev.dubhe.anvilcraft.inventory.ItemCollectorMenu;
 import dev.dubhe.anvilcraft.network.SlotDisableChangePack;
 import lombok.Getter;
@@ -21,7 +20,8 @@
 
 import java.util.function.BiFunction;
 
-public class ItemCollectorScreen extends AbstractContainerScreen<ItemCollectorMenu> implements IFilterScreen {
+public class ItemCollectorScreen extends AbstractContainerScreen<ItemCollectorMenu>
+    implements IFilterScreen<ItemCollectorMenu> {
     private static final ResourceLocation CONTAINER_LOCATION =
             AnvilCraft.of("textures/gui/container/machine/background/item_collector.png");
     BiFunction<Integer, Integer, EnableFilterButton> enableFilterButtonSupplier =
@@ -123,7 +123,7 @@ protected void renderSlotTooltip(@NotNull GuiGraphics guiGraphics, int x, int y)
     }
 
     @Override
-    public IFilterMenu getFilterMenu() {
+    public ItemCollectorMenu getFilterMenu() {
         return this.menu;
     }
 
@@ -148,4 +148,14 @@ protected void slotClicked(@NotNull Slot slot, int slotId, int mouseButton, @Not
         }
         super.slotClicked(slot, slotId, mouseButton, type);
     }
+
+    @Override
+    public int getOffsetX() {
+        return (this.width - this.imageWidth) / 2;
+    }
+
+    @Override
+    public int getOffsetY() {
+        return (this.height - this.imageHeight) / 2;
+    }
 }

common/src/main/java/dev/dubhe/anvilcraft/integration/emi/AnvilCraftEmiPlugin.java

@@ -1,6 +1,9 @@
 package dev.dubhe.anvilcraft.integration.emi;
 
 import dev.dubhe.anvilcraft.api.recipe.AnvilRecipeManager;
+import dev.dubhe.anvilcraft.client.gui.screen.inventory.AutoCrafterScreen;
+import dev.dubhe.anvilcraft.client.gui.screen.inventory.ChuteScreen;
+import dev.dubhe.anvilcraft.client.gui.screen.inventory.ItemCollectorScreen;
 import dev.dubhe.anvilcraft.init.ModBlocks;
 import dev.dubhe.anvilcraft.integration.emi.recipe.AnvilProcessEmiRecipe;
 import dev.emi.emi.api.EmiEntrypoint;
@@ -16,6 +19,10 @@ public class AnvilCraftEmiPlugin implements EmiPlugin {
     public void register(@NotNull EmiRegistry registry) {
         AnvilRecipeCategory.ALL.forEach(registry::addCategory);
 
+        registry.addDragDropHandler(AutoCrafterScreen.class, new GhostIngredientHandler<>());
+        registry.addDragDropHandler(ChuteScreen.class, new GhostIngredientHandler<>());
+        registry.addDragDropHandler(ItemCollectorScreen.class, new GhostIngredientHandler<>());
+
         registry.addWorkstation(AnvilRecipeCategory.STAMPING, EmiStack.of(ModBlocks.STAMPING_PLATFORM.get()));
         registry.addWorkstation(AnvilRecipeCategory.STAMPING, EmiStack.of(ModBlocks.ROYAL_ANVIL));
         registry.addWorkstation(AnvilRecipeCategory.STAMPING, EmiStack.of(Blocks.ANVIL));

common/src/main/java/dev/dubhe/anvilcraft/integration/emi/GhostIngredientHandler.java

@@ -0,0 +1,37 @@
+package dev.dubhe.anvilcraft.integration.emi;
+
+import dev.dubhe.anvilcraft.client.gui.screen.inventory.IFilterScreen;
+import dev.dubhe.anvilcraft.network.SlotFilterChangePack;
+import dev.emi.emi.api.EmiDragDropHandler;
+import dev.emi.emi.api.stack.EmiIngredient;
+import dev.emi.emi.api.stack.EmiStack;
+import net.minecraft.client.gui.screens.Screen;
+import net.minecraft.client.renderer.Rect2i;
+import net.minecraft.world.inventory.Slot;
+import net.minecraft.world.item.ItemStack;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.List;
+
+public class GhostIngredientHandler<T extends Screen & IFilterScreen<?>> implements EmiDragDropHandler<T> {
+    @Override
+    public boolean dropStack(
+        @NotNull T screen, @NotNull EmiIngredient ingredient, int x, int y
+    ) {
+        if (!screen.isFilterEnabled()) return false;
+        List<EmiStack> stacks = ingredient.getEmiStacks();
+        if (stacks.size() != 1) return false;
+        ItemStack stack = stacks.get(0).getItemStack();
+        if (stack.isEmpty()) return false;
+        for (Slot slot : screen.getFilterMenu().slots) {
+            Rect2i rect2i = new Rect2i(screen.getOffsetX() + slot.x, screen.getOffsetY() + slot.y, 16, 16);
+            if (rect2i.contains(x, y)) {
+                if (screen.setFilter(screen.getFilterMenu().getFilterSlotIndex(slot), stack)) {
+                    new SlotFilterChangePack(screen.getFilterMenu().getFilterSlotIndex(slot), stack).send();
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+}

common/src/main/java/dev/dubhe/anvilcraft/inventory/AutoCrafterMenu.java

@@ -177,6 +177,11 @@ public IFilterBlockEntity getFilterBlockEntity() {
         return this.blockEntity;
     }
 
+    @Override
+    public int getFilterSlotIndex(@NotNull Slot slot) {
+        return slot.index - 36;
+    }
+
     @Override
     public void setItem(int slotId, int stateId, @NotNull ItemStack stack) {
         super.setItem(slotId, stateId, stack);

common/src/main/java/dev/dubhe/anvilcraft/inventory/ChuteMenu.java

@@ -134,4 +134,9 @@ public boolean stillValid(@NotNull Player player) {
     public IFilterBlockEntity getFilterBlockEntity() {
         return blockEntity;
     }
+
+    @Override
+    public int getFilterSlotIndex(@NotNull Slot slot) {
+        return slot.index - 36;
+    }
 }