Give world context to block conversion.

This commit is contained in:
GTNH-Colen committed 2023-12-20 02:00:48 +00:00
1 parent 15a5806887
commit 1b96a07a74
2 files changed
+15 -12

No files matched your search

@@ -2,30 +2,31 @@ package com.colen.postea.API;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;
import java.util.function.BiFunction;
import net.minecraft.world.World;
import com.colen.postea.Utility.BlockConversionInfo;
public class BlockReplacementManager {
private static final Map<String, Function<BlockConversionInfo, BlockConversionInfo>> blockReplacementMap = new HashMap<>();
private static final Map<String, BiFunction<BlockConversionInfo, World, BlockConversionInfo>> blockReplacementMap = new HashMap<>();
@SuppressWarnings("unused")
public static void addBlockReplacement(String blockNameIn,
Function<BlockConversionInfo, BlockConversionInfo> transformer) {
BiFunction<BlockConversionInfo, World, BlockConversionInfo> transformer) {
blockReplacementMap.put(blockNameIn, transformer);
}
@SuppressWarnings("unused")
public static BlockConversionInfo getBlockReplacement(BlockConversionInfo blockConversionInfo) {
public static BlockConversionInfo getBlockReplacement(BlockConversionInfo blockConversionInfo, World world) {
Function<BlockConversionInfo, BlockConversionInfo> transformer = blockReplacementMap
BiFunction<BlockConversionInfo, World, BlockConversionInfo> transformer = blockReplacementMap
.getOrDefault(blockConversionInfo.blockName, null);
if (transformer == null) {
return null;
} else {
return transformer.apply(blockConversionInfo);
return transformer.apply(blockConversionInfo, world);
}
}