Sky Chest TESR / Tile Entity Registration Changes

2020-06-01 21:59:08 +02:00
parent 36af72faeb
commit 271e8889a1
37 changed files with 503 additions and 404 deletions
@@ -0,0 +1,126 @@
+package appeng.bootstrap;
+
+import appeng.block.AEBaseTileBlock;
+import appeng.bootstrap.components.ITileEntityRegistrationComponent;
+import appeng.bootstrap.definitions.TileEntityDefinition;
+import appeng.core.AppEng;
+import appeng.core.features.AEFeature;
+import appeng.core.features.ActivityState;
+import appeng.core.features.BlockStackSrc;
+import appeng.tile.AEBaseTile;
+import appeng.util.Platform;
+import com.google.common.base.Preconditions;
+import net.minecraft.block.Block;
+import net.minecraft.client.renderer.tileentity.TileEntityRenderer;
+import net.minecraft.tileentity.TileEntityType;
+import net.minecraftforge.api.distmarker.Dist;
+import net.minecraftforge.api.distmarker.OnlyIn;
+import net.minecraftforge.fml.DistExecutor;
+import net.minecraftforge.fml.client.registry.ClientRegistry;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.EnumSet;
+import java.util.List;
+import java.util.function.Consumer;
+import java.util.function.Function;
+import java.util.function.Supplier;
+
+/**
+ * Used to define our tile entities and all of their properties that are relevant to registering them.
+ *
+ * @param <T>
+ */
+public class TileEntityBuilder<T extends AEBaseTile> {
+
+    private final FeatureFactory factory;
+
+    private final String registryName;
+
+    // The tile entity class
+    private final Class<T> tileClass;
+
+    private TileEntityType<T> type;
+
+    // The factory for creating tile entity objects
+    private final Function<TileEntityType<T>, T> supplier;
+
+    @OnlyIn(Dist.CLIENT)
+    private final TileEntityRendering<T> tileEntityRendering = new TileEntityRendering<T>();
+
+    private final List<Block> blocks = new ArrayList<>();
+
+    private final EnumSet<AEFeature> features = EnumSet.noneOf(AEFeature.class);
+
+    public TileEntityBuilder(FeatureFactory factory, String registryName, Class<T> tileClass, Function<TileEntityType<T>, T> supplier) {
+        this.factory = factory;
+        this.registryName = registryName;
+        this.tileClass = tileClass;
+        this.supplier = supplier;
+    }
+
+    public TileEntityBuilder<T> features(AEFeature... features) {
+        this.features.clear();
+        this.addFeatures(features);
+        return this;
+    }
+
+    public TileEntityBuilder<T> addFeatures(AEFeature... features) {
+        Collections.addAll(this.features, features);
+        return this;
+    }
+
+    public TileEntityBuilder<T> rendering(TileEntityRenderingCustomizer<T> customizer) {
+        DistExecutor.runWhenOn(Dist.CLIENT, () -> () -> customizer.customize(tileEntityRendering));
+        return this;
+    }
+
+    @SuppressWarnings("unchecked")
+    public TileEntityDefinition build() {
+
+        this.factory.addBootstrapComponent((ITileEntityRegistrationComponent) registry ->
+        {
+            if (blocks.isEmpty()) {
+                throw new IllegalStateException("No blocks make use of this tile entity: " + tileClass);
+            }
+
+            Supplier<T> factory = () -> supplier.apply(type);
+            type = TileEntityType.Builder.create(factory, blocks.toArray(new Block[0]))
+                    .build(null);
+            type.setRegistryName(AppEng.MOD_ID, registryName);
+            registry.register(type);
+
+            DistExecutor.runWhenOn(Dist.CLIENT, () -> this::registerClient);
+
+            AEBaseTile.registerTileItem(
+                    tileClass,
+                    new BlockStackSrc(blocks.get(0), ActivityState.Enabled));
+
+            for (Block block : blocks) {
+                if (block instanceof AEBaseTileBlock) {
+                    AEBaseTileBlock<T> baseTileBlock = (AEBaseTileBlock<T>) block;
+                    baseTileBlock.setTileEntity(tileClass, factory);
+                }
+            }
+
+        });
+
+        return new TileEntityDefinition(this::addBlock);
+
+    }
+
+    @OnlyIn(Dist.CLIENT)
+    private void registerClient() {
+
+        if (tileEntityRendering.tileEntityRenderer != null) {
+            ClientRegistry.bindTileEntityRenderer(type, tileEntityRendering.tileEntityRenderer);
+        }
+
+    }
+
+    private void addBlock(Block block) {
+        Preconditions.checkState(type == null, "No more blocks can be added after registration completed.");
+        this.blocks.add(block);
+    }
+
+}