Reset block predicate when model is loaded

Should help invalidate Embeddium's cache
embeddedt · Dec 19, 2023 · 9c00f72 · 9c00f72
1 parent aa53a76
commit 9c00f72
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...mbeddedt/modernfix/forge/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java b/...mbeddedt/modernfix/forge/mixin/perf/dynamic_resources/ctm/CTMPackReloadListenerMixin.java
@@ -63,11 +63,15 @@ private void refreshLayerHacks() {
                 if(original == null)
                     original = DEFAULT_PREDICATE;
                 blockRenderChecks.put(block.delegate, original);
-                ItemBlockRenderTypes.setRenderLayer(block, type -> this.useOverrideIfPresent(block.delegate, type));
+                updateBlockPredicate(block);
             }
         }
     }
 
+    private void updateBlockPredicate(Block block) {
+        ItemBlockRenderTypes.setRenderLayer(block, type -> this.useOverrideIfPresent(block.delegate, type));
+    }
+
     private boolean useOverrideIfPresent(IRegistryDelegate<Block> delegate, RenderType type) {
         Predicate<RenderType> override = renderCheckOverrides.get(delegate);
         if(override == null)
@@ -97,6 +101,7 @@ public BakedModel onBakedModelLoad(ResourceLocation location, UnbakedModel baseM
             Predicate<RenderType> newPredicate = this.getLayerCheck(state, originalModel);
             if(newPredicate != null) {
                 renderCheckOverrides.put(block.delegate, newPredicate);
+                updateBlockPredicate(block);
                 return originalModel;
             }
         }