Render enchanted upgrades with a glint (#1532)

projects/forge/src/client/java/dan200/computercraft/client/model/FoiledModel.java

@@ -0,0 +1,51 @@
+// SPDX-FileCopyrightText: 2023 The CC: Tweaked Developers
+//
+// SPDX-License-Identifier: MPL-2.0
+
+package dan200.computercraft.client.model;
+
+import dan200.computercraft.shared.util.ConsList;
+import net.minecraft.client.renderer.RenderType;
+import net.minecraft.client.renderer.block.model.BakedQuad;
+import net.minecraft.client.resources.model.BakedModel;
+import net.minecraft.core.Direction;
+import net.minecraft.util.RandomSource;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.block.state.BlockState;
+import net.minecraftforge.client.model.BakedModelWrapper;
+import net.minecraftforge.client.model.data.ModelData;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.List;
+
+/**
+ * A model wrapper which applies a glint/foil to the original model.
+ */
+public final class FoiledModel extends BakedModelWrapper<BakedModel> {
+    public FoiledModel(BakedModel model) {
+        super(model);
+    }
+
+    @Override
+    public List<BakedQuad> getQuads(@Nullable BlockState state, @Nullable Direction side, RandomSource rand, ModelData extraData, @Nullable RenderType renderType) {
+        return renderType == RenderType.glint()
+            ? super.getQuads(state, side, rand, extraData, null)
+            : super.getQuads(state, side, rand, extraData, renderType);
+    }
+
+    @Override
+    public List<RenderType> getRenderTypes(ItemStack itemStack, boolean fabulous) {
+        return new ConsList<>(fabulous ? RenderType.glintDirect() : RenderType.glint(), super.getRenderTypes(itemStack, fabulous));
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        return this == obj || (obj instanceof FoiledModel other && originalModel.equals(other.originalModel));
+    }
+
+    @Override
+    public int hashCode() {
+        return originalModel.hashCode() ^ 1;
+    }
+
+}

projects/forge/src/client/java/dan200/computercraft/client/platform/ClientPlatformHelperImpl.java

@@ -5,22 +5,53 @@
 package dan200.computercraft.client.platform;
 
 import com.google.auto.service.AutoService;
+import com.mojang.blaze3d.vertex.PoseStack;
+import dan200.computercraft.client.model.FoiledModel;
+import dan200.computercraft.client.render.ModelRenderer;
 import dan200.computercraft.shared.network.NetworkMessage;
 import dan200.computercraft.shared.network.server.ServerNetworkContext;
 import dan200.computercraft.shared.platform.NetworkHandler;
+import net.minecraft.client.renderer.MultiBufferSource;
 import net.minecraft.client.resources.model.BakedModel;
 import net.minecraft.client.resources.model.ModelManager;
+import net.minecraft.core.Direction;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.util.RandomSource;
+import net.minecraft.world.item.ItemStack;
+import net.minecraftforge.client.model.data.ModelData;
+
+import javax.annotation.Nullable;
+import java.util.Arrays;
 
 @AutoService(dan200.computercraft.impl.client.ClientPlatformHelper.class)
 public class ClientPlatformHelperImpl implements ClientPlatformHelper {
+    private static final RandomSource random = RandomSource.create(0);
+    private static final Direction[] directions = Arrays.copyOf(Direction.values(), 7);
+
     @Override
     public BakedModel getModel(ModelManager manager, ResourceLocation location) {
         return manager.getModel(location);
     }
 
+    @Override
+    public BakedModel createdFoiledModel(BakedModel model) {
+        return new FoiledModel(model);
+    }
+
     @Override
     public void sendToServer(NetworkMessage<ServerNetworkContext> message) {
         NetworkHandler.sendToServer(message);
     }
+
+    @Override
+    public void renderBakedModel(PoseStack transform, MultiBufferSource buffers, BakedModel model, int lightmapCoord, int overlayLight, @Nullable int[] tints) {
+        for (var renderType : model.getRenderTypes(ItemStack.EMPTY, true)) {
+            var buffer = buffers.getBuffer(renderType);
+            for (var face : directions) {
+                random.setSeed(42);
+                var quads = model.getQuads(null, face, random, ModelData.EMPTY, renderType);
+                ModelRenderer.renderQuads(transform, buffer, quads, lightmapCoord, overlayLight, tints);
+            }
+        }
+    }
 }