Add helpers for registering to structure template pools

Required for CyclopsMC/EvilCraft#1146

Author: rubensworks

loader-common/src/main/java/org/cyclops/cyclopscore/helper/IModHelpers.java

@@ -40,4 +40,6 @@ public static IModHelpers get() {
 
     public IGuiHelpers getGuiHelpers();
 
+    public IStructureHelpers getStructureHelpers();
+
 }

loader-common/src/main/java/org/cyclops/cyclopscore/helper/IStructureHelpers.java

@@ -0,0 +1,19 @@
+package org.cyclops.cyclopscore.helper;
+
+import net.minecraft.core.RegistryAccess;
+import net.minecraft.resources.ResourceLocation;
+
+/**
+ * @author rubensworks
+ */
+public interface IStructureHelpers {
+
+    /**
+     * Add the given structure template to a pool.
+     * @param structureTemplatePool A pool.
+     * @param structureTemplate The template to add.
+     * @param registryAccess A registry access.
+     */
+    public void addToStructureTemplatePool(ResourceLocation structureTemplatePool, ResourceLocation structureTemplate, RegistryAccess registryAccess);
+
+}

loader-common/src/main/java/org/cyclops/cyclopscore/helper/ModHelpersCommon.java

@@ -16,6 +16,7 @@ public abstract class ModHelpersCommon implements IModHelpers {
     private ICraftingHelpers craftingHelpers;
     private IWorldHelpers worldHelpers;
     private IGuiHelpers guiHelpers;
+    private IStructureHelpers structureHelpers;
 
     protected ModHelpersCommon() {
         this.initializeHelpers();
@@ -39,6 +40,7 @@ protected void initializeHelpers() {
         this.baseHelpers = new BaseHelpersCommon();
         this.craftingHelpers = new CraftingHelpersCommon(this);
         this.worldHelpers = new WorldHelpersCommon(this);
+        this.structureHelpers = new StructureHelpersCommon();
     }
 
     @Override
@@ -95,4 +97,9 @@ public IWorldHelpers getWorldHelpers() {
     public IGuiHelpers getGuiHelpers() {
         return this.guiHelpers;
     }
+
+    @Override
+    public IStructureHelpers getStructureHelpers() {
+        return this.structureHelpers;
+    }
 }

loader-common/src/main/java/org/cyclops/cyclopscore/helper/StructureHelpersCommon.java

@@ -0,0 +1,29 @@
+package org.cyclops.cyclopscore.helper;
+
+import com.mojang.datafixers.util.Pair;
+import net.minecraft.core.Registry;
+import net.minecraft.core.RegistryAccess;
+import net.minecraft.core.registries.Registries;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.level.levelgen.structure.pools.SinglePoolElement;
+import net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool;
+
+import java.util.ArrayList;
+import java.util.Objects;
+
+/**
+ * @author rubensworks
+ */
+public class StructureHelpersCommon implements IStructureHelpers {
+    @Override
+    public void addToStructureTemplatePool(ResourceLocation structureTemplatePool, ResourceLocation structureTemplate, RegistryAccess registryAccess) {
+        Registry<StructureTemplatePool> registry = registryAccess.registryOrThrow(Registries.TEMPLATE_POOL);
+        StructureTemplatePool pool = Objects.requireNonNull(registry.get(structureTemplatePool), structureTemplatePool.getPath());
+        if(!(pool.rawTemplates instanceof ArrayList)) {
+            pool.rawTemplates = new ArrayList<>(pool.rawTemplates);
+        }
+        SinglePoolElement addedElement = SinglePoolElement.single(structureTemplate.toString()).apply(StructureTemplatePool.Projection.RIGID);
+        pool.rawTemplates.add(Pair.of(addedElement, 1));
+        pool.templates.add(addedElement);
+    }
+}

loader-common/src/main/resources/META-INF/accesstransformer.cfg

@@ -110,3 +110,9 @@ public net.minecraft.client.renderer.texture.AbstractTexture blur
 public net.minecraft.client.renderer.texture.AbstractTexture f_117951_ # blur
 public net.minecraft.client.renderer.texture.AbstractTexture mipmap
 public net.minecraft.client.renderer.texture.AbstractTexture f_117952_ # mipmap
+
+# Structures
+public-f net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool rawTemplates
+public-f net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool f_210559_ # rawTemplates
+public net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool templates
+public net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool f_210560_ # templates

loader-common/src/main/resources/cyclopscore.accesswidener

@@ -54,4 +54,9 @@ accessible field net/minecraft/client/particle/ParticleEngine textureManager Lne
 
 # Rendering
 accessible field net/minecraft/client/renderer/texture/AbstractTexture blur Z
-accessible field net/minecraft/client/renderer/texture/AbstractTexture mipmap Z
+accessible field net/minecraft/client/renderer/texture/AbstractTexture mipmap Z
+
+# Structures
+accessible field net/minecraft/world/level/levelgen/structure/pools/StructureTemplatePool rawTemplates Ljava/util/List;
+mutable field net/minecraft/world/level/levelgen/structure/pools/StructureTemplatePool rawTemplates Ljava/util/List;
+accessible field net/minecraft/world/level/levelgen/structure/pools/StructureTemplatePool templates Lit/unimi/dsi/fastutil/objects/ObjectArrayList;

loader-neoforge/src/main/java/org/cyclops/cyclopscore/helper/IModHelpersNeoForge.java

@@ -20,4 +20,6 @@ public static IModHelpersNeoForge get() {
 
     public IGuiHelpersNeoForge getGuiHelpers();
 
+    public IStructureHelpersNeoForge getStructureHelpers();
+
 }

loader-neoforge/src/main/java/org/cyclops/cyclopscore/helper/IStructureHelpersNeoForge.java

@@ -0,0 +1,17 @@
+package org.cyclops.cyclopscore.helper;
+
+import net.minecraft.resources.ResourceLocation;
+
+/**
+ * @author rubensworks
+ */
+public interface IStructureHelpersNeoForge extends IStructureHelpers {
+
+    /**
+     * Add the given structure template to a pool.
+     * @param structureTemplatePool A pool.
+     * @param structureTemplate The template to add.
+     */
+    public void addToStructureTemplatePool(ResourceLocation structureTemplatePool, ResourceLocation structureTemplate);
+
+}

loader-neoforge/src/main/java/org/cyclops/cyclopscore/helper/ModHelpersNeoForge.java

@@ -14,6 +14,7 @@ public class ModHelpersNeoForge extends ModHelpersCommon implements IModHelpersN
     private IRenderHelpersNeoForge renderHelpers;
     private IRegistrationHelpers registrationHelpers;
     private IGuiHelpersNeoForge guiHelpers;
+    private IStructureHelpersNeoForge structureHelpers;
 
     private ModHelpersNeoForge() {}
 
@@ -32,6 +33,7 @@ protected void initializeHelpers() {
         this.capabilityHelpers = new CapabilityHelpersNeoForge(this);
         this.fluidHelpers = new FluidHelpersNeoForge();
         this.registrationHelpers = new RegistrationHelpersNeoForge();
+        this.structureHelpers = new StructureHelpersNeoForge();
     }
 
     @Override
@@ -68,4 +70,9 @@ public IRegistrationHelpers getRegistrationHelpers() {
     public IGuiHelpersNeoForge getGuiHelpers() {
         return this.guiHelpers;
     }
+
+    @Override
+    public IStructureHelpersNeoForge getStructureHelpers() {
+        return this.structureHelpers;
+    }
 }

loader-neoforge/src/main/java/org/cyclops/cyclopscore/helper/StructureHelpersNeoForge.java

@@ -0,0 +1,20 @@
+package org.cyclops.cyclopscore.helper;
+
+import net.minecraft.resources.ResourceLocation;
+import net.neoforged.neoforge.common.NeoForge;
+import net.neoforged.neoforge.event.TagsUpdatedEvent;
+
+/**
+ * @author rubensworks
+ */
+public class StructureHelpersNeoForge extends StructureHelpersCommon implements IStructureHelpersNeoForge {
+
+    @Override
+    public void addToStructureTemplatePool(ResourceLocation structureTemplatePool, ResourceLocation structureTemplate) {
+        NeoForge.EVENT_BUS.addListener((TagsUpdatedEvent event) -> {
+            if (event.getUpdateCause() == TagsUpdatedEvent.UpdateCause.SERVER_DATA_LOAD) {
+                addToStructureTemplatePool(structureTemplatePool, structureTemplate, event.getRegistryAccess());
+            }
+        });
+    }
+}