[fix] Fix Render Crash issue with cobblemon

Author: MarbleGateKeeper

src/main/java/plus/dragons/createenchantmentindustry/client/CEIClient.java

@@ -21,7 +21,6 @@
 import net.createmod.ponder.foundation.PonderIndex;
 import net.neoforged.api.distmarker.Dist;
 import net.neoforged.bus.api.IEventBus;
-import net.neoforged.bus.api.SubscribeEvent;
 import net.neoforged.fml.common.Mod;
 import net.neoforged.fml.event.lifecycle.FMLClientSetupEvent;
 import plus.dragons.createenchantmentindustry.client.model.CEIPartialModels;
@@ -31,12 +30,15 @@
 @Mod(value = CEICommon.ID, dist = Dist.CLIENT)
 public class CEIClient {
     public CEIClient(IEventBus modBus) {
-        modBus.register(this);
+        // CEIPartialModels must be registered here,
+        // or when PartialModelEventHandler#onRegisterAdditional triggered,
+        // PartialModel.ALL won't include all partial model in 'some cases'
+        // AllPartialModels#ini does not do this since AllPartialModels is already triggered at AllBlocks.TRACK
+        CEIPartialModels.register();
+        modBus.addListener(CEIClient::setup);
     }
 
-    @SubscribeEvent
-    public void setup(final FMLClientSetupEvent event) {
-        CEIPartialModels.register();
+    public static void setup(final FMLClientSetupEvent event) {
         PonderIndex.addPlugin(new CEIPonderPlugin());
     }
 }

src/main/java/plus/dragons/createenchantmentindustry/client/model/CEIPartialModels.java

@@ -27,7 +27,6 @@ public class CEIPartialModels {
     public static final PartialModel PRINTER_NOZZLE_TOP = block("printer/nozzle_top");
     public static final PartialModel PRINTER_NOZZLE_BOTTOM = block("printer/nozzle_bottom");
     public static final PartialModel PRINTER_PISTON = block("printer/piston");
-    public static final PartialModel BLAZE_ITEM = PartialModel.of(Create.asResource("block/blaze_burner/block_with_blaze"));
     public static final PartialModel BLAZE_ENCHANTER_HAT = block("blaze/enchanter_hat");
     public static final PartialModel BLAZE_ENCHANTER_HAT_SMALL = block("blaze/enchanter_hat_small");
     public static final PartialModel BLAZE_FORGER_HAT = block("blaze/forger_hat");

src/main/java/plus/dragons/createenchantmentindustry/common/processing/enchanter/BlazeEnchanterItemRenderer.java

@@ -46,7 +46,7 @@ public static void register(RegisterClientExtensionsEvent event) {
 
     @Override
     protected void render(ItemStack stack, CustomRenderedItemModel model, PartialItemModelRenderer renderer, ItemDisplayContext transformType, PoseStack poseStack, MultiBufferSource bufferSource, int light, int overlay) {
-        renderer.render(CEIPartialModels.BLAZE_ITEM.get(), light);
+        renderer.render(model.getOriginalModel(), light);
         poseStack.pushPose();
         poseStack.translate(.5f, .75f, .5f);
         renderer.render(CEIPartialModels.BLAZE_ENCHANTER_HAT.get(), light);

src/main/java/plus/dragons/createenchantmentindustry/common/processing/forger/BlazeForgerItemRenderer.java

@@ -46,7 +46,7 @@ public static void register(RegisterClientExtensionsEvent event) {
 
     @Override
     protected void render(ItemStack stack, CustomRenderedItemModel model, PartialItemModelRenderer renderer, ItemDisplayContext transformType, PoseStack poseStack, MultiBufferSource bufferSource, int light, int overlay) {
-        renderer.render(CEIPartialModels.BLAZE_ITEM.get(), light);
+        renderer.render(model.getOriginalModel(), light);
         poseStack.pushPose();
         poseStack.translate(.5f, .75f, .5f);
         renderer.render(CEIPartialModels.BLAZE_FORGER_HAT.get(), light);