diff --git a/src/main/java/org/embeddedt/modernfix/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java b/src/main/java/org/embeddedt/modernfix/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java index 63c7404b..bd980f94 100644 --- a/src/main/java/org/embeddedt/modernfix/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java +++ b/src/main/java/org/embeddedt/modernfix/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java @@ -32,12 +32,18 @@ import team.chisel.ctm.client.model.AbstractCTMBakedModel; import team.chisel.ctm.client.util.CTMPackReloadListener; import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; import java.util.function.Predicate; @Mixin(CTMPackReloadListener.class) public abstract class CTMPackReloadListenerMixin { + /* caches the original render checks */ @Shadow @Final private static Map, Predicate> blockRenderChecks; + private static Map, Predicate> renderCheckOverrides = new ConcurrentHashMap<>(); + + private static Predicate DEFAULT_PREDICATE = type -> type == RenderType.solid(); + @Shadow protected abstract Predicate getLayerCheck(BlockState state, BakedModel model); @Shadow protected abstract Predicate getExistingRenderCheck(Block block); @@ -49,8 +55,23 @@ public abstract class CTMPackReloadListenerMixin { @Overwrite(remap = false) private void refreshLayerHacks() { - blockRenderChecks.forEach((b, p) -> ItemBlockRenderTypes.setRenderLayer((Block) b.get(), p)); - blockRenderChecks.clear(); + renderCheckOverrides.clear(); + if(blockRenderChecks.isEmpty()) { + for(Block block : ForgeRegistries.BLOCKS.getValues()) { + Predicate original = this.getExistingRenderCheck(block); + if(original == null) + original = DEFAULT_PREDICATE; + blockRenderChecks.put(block.delegate, original); + ItemBlockRenderTypes.setRenderLayer(block, type -> this.useOverrideIfPresent(block.delegate, type)); + } + } + } + + private boolean useOverrideIfPresent(IRegistryDelegate delegate, RenderType type) { + Predicate override = renderCheckOverrides.get(delegate); + if(override == null) + override = blockRenderChecks.get(delegate); + return override.test(type); } private void onModelBake(DynamicModelBakeEvent event) { @@ -58,7 +79,7 @@ public abstract class CTMPackReloadListenerMixin { return; /* we construct a new ResourceLocation because an MRL is coming in */ Block block = ForgeRegistries.BLOCKS.getValue(new ResourceLocation(event.getLocation().getNamespace(), event.getLocation().getPath())); - if(block == null || block == Blocks.AIR || blockRenderChecks.containsKey(block.delegate)) + if(block == null || block == Blocks.AIR || renderCheckOverrides.containsKey(block.delegate)) return; /* find all states that match this MRL */ ImmutableList allStates; @@ -71,8 +92,7 @@ public abstract class CTMPackReloadListenerMixin { for(BlockState state : allStates) { Predicate newPredicate = this.getLayerCheck(state, event.getModel()); if(newPredicate != null) { - blockRenderChecks.put(block.delegate, this.getExistingRenderCheck(block)); - ItemBlockRenderTypes.setRenderLayer(block, newPredicate); + renderCheckOverrides.put(block.delegate, newPredicate); return; } }