diff --git a/src/main/java/com/refinedmods/refinedstorage/RSBlocks.java b/src/main/java/com/refinedmods/refinedstorage/RSBlocks.java index f87eb99ad..831efe2a3 100644 --- a/src/main/java/com/refinedmods/refinedstorage/RSBlocks.java +++ b/src/main/java/com/refinedmods/refinedstorage/RSBlocks.java @@ -14,13 +14,15 @@ import net.minecraftforge.registries.DeferredRegister; import net.minecraftforge.registries.ForgeRegistries; import java.util.ArrayList; -import java.util.HashMap; +import java.util.EnumMap; import java.util.List; import java.util.Map; public final class RSBlocks { private static final DeferredRegister BLOCKS = DeferredRegister.create(ForgeRegistries.BLOCKS, RS.ID); + private static final String GRID_SUFFIX = "_grid"; + public static final RegistryObject IMPORTER; public static final RegistryObject EXPORTER; @@ -29,8 +31,8 @@ public final class RSBlocks { public static final RegistryObject CABLE; public static final RegistryObject DISK_DRIVE; public static final RegistryObject EXTERNAL_STORAGE; - public static final Map> STORAGE_BLOCKS = new HashMap<>(); - public static final Map> FLUID_STORAGE_BLOCKS = new HashMap<>(); + public static final Map> STORAGE_BLOCKS = new EnumMap<>(ItemStorageType.class); + public static final Map> FLUID_STORAGE_BLOCKS = new EnumMap<>(FluidStorageType.class); public static final RegistryObject INTERFACE; public static final RegistryObject FLUID_INTERFACE; public static final RegistryObject STORAGE_MONITOR; @@ -83,9 +85,9 @@ public final class RSBlocks { } GRID.registerBlocks("grid", () -> new GridBlock(GridType.NORMAL)); - CRAFTING_GRID.registerBlocks(GridType.CRAFTING.getString() + "_grid", () -> new GridBlock(GridType.CRAFTING)); - PATTERN_GRID.registerBlocks(GridType.PATTERN.getString() + "_grid", () -> new GridBlock(GridType.PATTERN)); - FLUID_GRID.registerBlocks(GridType.FLUID.getString() + "_grid", () -> new GridBlock(GridType.FLUID)); + CRAFTING_GRID.registerBlocks(GridType.CRAFTING.getString() + GRID_SUFFIX, () -> new GridBlock(GridType.CRAFTING)); + PATTERN_GRID.registerBlocks(GridType.PATTERN.getString() + GRID_SUFFIX, () -> new GridBlock(GridType.PATTERN)); + FLUID_GRID.registerBlocks(GridType.FLUID.getString() + GRID_SUFFIX, () -> new GridBlock(GridType.FLUID)); CONTROLLER.registerBlocks("controller", () -> new ControllerBlock(NetworkType.NORMAL)); CREATIVE_CONTROLLER.registerBlocks("creative_controller", () -> new ControllerBlock(NetworkType.CREATIVE)); NETWORK_RECEIVER.registerBlocks("network_receiver", NetworkReceiverBlock::new); @@ -100,7 +102,10 @@ public final class RSBlocks { DETECTOR.registerBlocks("detector", DetectorBlock::new); } + private RSBlocks() { + } + public static void register() { BLOCKS.register(FMLJavaModLoadingContext.get().getModEventBus()); } -} \ No newline at end of file +}