fixed hatch texture blocks not getting setwhen using custom IHatchStr…

…ucturePredicate
GregTech-Intergalactical · Dec 12, 2023 · 4e6c358 · 4e6c358
1 parent 768e828
commit 4e6c358
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/common/src/main/java/muramasa/antimatter/structure/AntimatterStructureUtility.java b/common/src/main/java/muramasa/antimatter/structure/AntimatterStructureUtility.java
@@ -16,9 +16,6 @@ public class AntimatterStructureUtility {
     public static <T extends BlockEntityBasicMultiMachine<T>> IStructureElement<T> ofHatch(HatchMachine machine){
         return ofHatch(machine, (t, world, pos, machine1, handler) -> {
             t.addComponent(machine1.getId(), handler);
-            if (world.getBlockEntity(pos) instanceof BlockEntityHatch<?> hatch) {
-                hatch.setTextureBlock(t.getHatchBlock(pos));
-            }
             return true;
         });
     }
@@ -44,7 +41,11 @@ public boolean check(T t, Level world, int x, int y, int z) {
                     if (component.getComponentHandler().isPresent()) {
                         IComponentHandler componentHandler = component.getComponentHandler().orElse(null);
                         if (machine.getId().equals(componentHandler.getId())) {
-                            return callback.test(t, world, pos, machine, componentHandler);
+                            boolean test = callback.test(t, world, pos, machine, componentHandler);
+                            if (test && world.getBlockEntity(pos) instanceof BlockEntityHatch<?> hatch) {
+                                hatch.setTextureBlock(t.getHatchBlock(pos));
+                            }
+                            return test;
                         }
                         return false;
                     }