diff --git a/common/src/main/java/org/embeddedt/modernfix/dynamicresources/ModelBakeryHelpers.java b/common/src/main/java/org/embeddedt/modernfix/dynamicresources/ModelBakeryHelpers.java index c12859c1..53ac8767 100644 --- a/common/src/main/java/org/embeddedt/modernfix/dynamicresources/ModelBakeryHelpers.java +++ b/common/src/main/java/org/embeddedt/modernfix/dynamicresources/ModelBakeryHelpers.java @@ -2,13 +2,17 @@ package org.embeddedt.modernfix.dynamicresources; import com.google.common.base.Splitter; import com.google.common.collect.ImmutableList; -import net.minecraft.client.resources.model.ModelResourceLocation; +import com.mojang.datafixers.util.Pair; +import net.minecraft.client.resources.model.*; +import net.minecraft.resources.ResourceLocation; import net.minecraft.world.level.block.Block; import net.minecraft.world.level.block.state.BlockState; import net.minecraft.world.level.block.state.StateDefinition; import net.minecraft.world.level.block.state.properties.Property; +import org.embeddedt.modernfix.api.entrypoint.ModernFixClientIntegration; import java.util.*; +import java.util.function.BiFunction; public class ModelBakeryHelpers { /** @@ -78,4 +82,13 @@ public class ModelBakeryHelpers { } return ImmutableList.copyOf(finalList); } + + public static ModernFixClientIntegration bakedModelWrapper(BiFunction, BakedModel> consumer) { + return new ModernFixClientIntegration() { + @Override + public BakedModel onBakedModelLoad(ResourceLocation location, UnbakedModel baseModel, BakedModel originalModel, ModelState state, ModelBakery bakery) { + return consumer.apply(location, Pair.of(baseModel, originalModel)); + } + }; + } }