fix(1.21.4): Fix LayerDefinition registration

fabric/src/main/java/net/blay09/mods/balm/fabric/client/rendering/FabricBalmRenderers.java

@@ -4,6 +4,7 @@
 import net.fabricmc.fabric.api.blockrenderlayer.v1.BlockRenderLayerMap;
 import net.fabricmc.fabric.api.client.rendering.v1.BlockEntityRendererRegistry;
 import net.fabricmc.fabric.api.client.rendering.v1.ColorProviderRegistry;
+import net.fabricmc.fabric.api.client.rendering.v1.EntityModelLayerRegistry;
 import net.fabricmc.fabric.api.client.rendering.v1.EntityRendererRegistry;
 import net.minecraft.client.color.block.BlockColor;
 import net.minecraft.client.model.geom.ModelLayerLocation;
@@ -14,30 +15,20 @@
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.entity.EntityType;
-import net.minecraft.world.level.ItemLike;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.entity.BlockEntityType;
 
-import java.util.HashMap;
-import java.util.Map;
 import java.util.function.Supplier;
-import java.util.stream.Collectors;
 
 public class FabricBalmRenderers implements BalmRenderers {
-    private final Map<ModelLayerLocation, Supplier<LayerDefinition>> layerDefinitions = new HashMap<>();
-
     @Override
     public ModelLayerLocation registerModel(ResourceLocation location, Supplier<LayerDefinition> layerDefinition) {
-        ModelLayerLocation modelLayerLocation = new ModelLayerLocation(location, "main");
-        layerDefinitions.put(modelLayerLocation, layerDefinition);
+        final var modelLayerLocation = new ModelLayerLocation(location, "main");
+        EntityModelLayerRegistry.registerModelLayer(modelLayerLocation, layerDefinition::get);
         return modelLayerLocation;
     }
 
-    public Map<ModelLayerLocation, LayerDefinition> createRoots() {
-        return layerDefinitions.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, it -> it.getValue().get()));
-    }
-
     @Override
     public <T extends Entity> void registerEntityRenderer(Supplier<EntityType<T>> type, EntityRendererProvider<? super T> provider) {
         EntityRendererRegistry.register(type.get(), provider);

fabric/src/main/resources/balm.fabric.mixins.json

@@ -30,7 +30,6 @@
     "ChatComponentMixin",
     "CheckboxAccessor",
     "EntityModelSetAccessor",
-    "LayerDefinitionsMixin",
     "GuiMixin",
     "ImageButtonAccessor",
     "ItemInHandRendererMixin",