From 6107b939a51eb9f8d54034fef5ba242251b8e91f Mon Sep 17 00:00:00 2001
From: MRH0 <mrhmail1@gmail.com>
Date: Fri, 31 Dec 2021 16:29:22 +0100
Subject: [PATCH] cleanup

---
 .../flywheel/engine/FurnaceEngineBlock.java   |  2 +-
 .../engine/FurnaceEngineModifiers.java        | 26 ++++++++++++++-----
 .../engine/FurnaceEngineTileEntity.java       |  4 +--
 3 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineBlock.java b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineBlock.java
index 09629ecf5..5aae6789f 100644
--- a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineBlock.java
+++ b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineBlock.java
@@ -33,7 +33,7 @@ public class FurnaceEngineBlock extends EngineBlock implements ITE<FurnaceEngine
 
 	@Override
 	protected boolean isValidBaseBlock(BlockState baseBlock, BlockGetter world, BlockPos pos) {
-		return FurnaceEngineModifiers.INSTANCE.getEngineState(baseBlock).isEngine();
+		return FurnaceEngineModifiers.get().getEngineState(baseBlock).isEngine();
 	}
 
 	@Override
diff --git a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineModifiers.java b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineModifiers.java
index 73a992cf2..0354e1384 100644
--- a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineModifiers.java
+++ b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineModifiers.java
@@ -7,16 +7,20 @@ import java.util.function.Function;
 import net.minecraft.world.level.block.AbstractFurnaceBlock;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.Blocks;
-import net.minecraft.world.level.block.RedstoneLampBlock;
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraftforge.registries.IRegistryDelegate;
 
 public class FurnaceEngineModifiers {
 
-	public final static FurnaceEngineModifiers INSTANCE = new FurnaceEngineModifiers();
+	private final static FurnaceEngineModifiers INSTANCE = new FurnaceEngineModifiers();
+	
+	protected FurnaceEngineModifiers() {
+		blockModifiers = new HashMap<>();
+		blockActivators = new HashMap<>();
+	}
 
-	protected Map<IRegistryDelegate<Block>, Float> blockModifiers = new HashMap<>();
-	protected Map<IRegistryDelegate<Block>, Function<BlockState, EngineState>> blockActivators = new HashMap<>();
+	private final Map<IRegistryDelegate<Block>, Float> blockModifiers;
+	private final Map<IRegistryDelegate<Block>, Function<BlockState, EngineState>> blockActivators;
 
 	public void register(IRegistryDelegate<Block> block, float modifier) {
 		this.blockModifiers.put(block, modifier);
@@ -40,12 +44,22 @@ public class FurnaceEngineModifiers {
 	}
 	
 	public EngineState getEngineState(BlockState state) {
-		return getEngineStateOrDefault(state, s -> s.getBlock() instanceof AbstractFurnaceBlock && s.hasProperty(AbstractFurnaceBlock.LIT) ? (s.getValue(AbstractFurnaceBlock.LIT) ? EngineState.ACTIVE : EngineState.VALID) : EngineState.EMPTY).apply(state);
+		return getEngineStateOrDefault(state, 
+				s -> s.getBlock() instanceof AbstractFurnaceBlock && s.hasProperty(AbstractFurnaceBlock.LIT) ? 
+						(s.getValue(AbstractFurnaceBlock.LIT) ? EngineState.ACTIVE : EngineState.VALID) : EngineState.EMPTY).apply(state);
 	}
 
 	public static void register() {
 		INSTANCE.register(Blocks.BLAST_FURNACE.delegate, 2f);
-		//INSTANCE.register(Blocks.REDSTONE_LAMP.delegate, 1f, s -> s.getBlock() instanceof RedstoneLampBlock && s.hasProperty(RedstoneLampBlock.LIT) ? (s.getValue(RedstoneLampBlock.LIT) ? EngineState.ACTIVE : EngineState.VALID) : EngineState.EMPTY);
+		
+		// 	Example:
+		//	INSTANCE.register(Blocks.REDSTONE_LAMP.delegate, 1f, 
+		//		s -> s.getBlock() instanceof RedstoneLampBlock && s.hasProperty(RedstoneLampBlock.LIT) ? 
+		//			(s.getValue(RedstoneLampBlock.LIT) ? EngineState.ACTIVE : EngineState.VALID) : EngineState.EMPTY);
+	}
+	
+	public static FurnaceEngineModifiers get() {
+		return INSTANCE;
 	}
 	
 	public enum EngineState {
diff --git a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineTileEntity.java b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineTileEntity.java
index 0dbe13f2e..b838b55e6 100644
--- a/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineTileEntity.java
+++ b/src/main/java/com/simibubi/create/content/contraptions/components/flywheel/engine/FurnaceEngineTileEntity.java
@@ -22,11 +22,11 @@ public class FurnaceEngineTileEntity extends EngineTileEntity {
 
 	public void updateFurnace() {
 		BlockState state = level.getBlockState(EngineBlock.getBaseBlockPos(getBlockState(), worldPosition));
-		EngineState engineState = FurnaceEngineModifiers.INSTANCE.getEngineState(state);
+		EngineState engineState = FurnaceEngineModifiers.get().getEngineState(state);
 		if (engineState.isEmpty())
 			return;
 
-		float modifier = FurnaceEngineModifiers.INSTANCE.getModifier(state);
+		float modifier = FurnaceEngineModifiers.get().getModifier(state);
 		boolean active = engineState.isActive();
 		float speed = active ? 16 * modifier : 0;
 		float capacity =