70 lines
3.4 KiB
Java
70 lines
3.4 KiB
Java
package net.minecraft.client.renderer.entity.layers;
|
|
|
|
import com.mojang.blaze3d.vertex.PoseStack;
|
|
import com.mojang.blaze3d.vertex.VertexConsumer;
|
|
import java.util.Map;
|
|
import net.fabricmc.api.EnvType;
|
|
import net.fabricmc.api.Environment;
|
|
import net.minecraft.client.model.Model;
|
|
import net.minecraft.client.model.WolfModel;
|
|
import net.minecraft.client.model.geom.EntityModelSet;
|
|
import net.minecraft.client.model.geom.ModelLayers;
|
|
import net.minecraft.client.renderer.MultiBufferSource;
|
|
import net.minecraft.client.renderer.RenderType;
|
|
import net.minecraft.client.renderer.entity.RenderLayerParent;
|
|
import net.minecraft.client.renderer.entity.state.WolfRenderState;
|
|
import net.minecraft.client.renderer.texture.OverlayTexture;
|
|
import net.minecraft.client.resources.model.EquipmentClientInfo;
|
|
import net.minecraft.core.component.DataComponents;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.world.entity.Crackiness;
|
|
import net.minecraft.world.entity.Crackiness.Level;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.equipment.EquipmentAsset;
|
|
import net.minecraft.world.item.equipment.Equippable;
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public class WolfArmorLayer extends RenderLayer<WolfRenderState, WolfModel> {
|
|
private final WolfModel adultModel;
|
|
private final WolfModel babyModel;
|
|
private final EquipmentLayerRenderer equipmentRenderer;
|
|
private static final Map<Level, ResourceLocation> ARMOR_CRACK_LOCATIONS = Map.of(
|
|
Level.LOW,
|
|
ResourceLocation.withDefaultNamespace("textures/entity/wolf/wolf_armor_crackiness_low.png"),
|
|
Level.MEDIUM,
|
|
ResourceLocation.withDefaultNamespace("textures/entity/wolf/wolf_armor_crackiness_medium.png"),
|
|
Level.HIGH,
|
|
ResourceLocation.withDefaultNamespace("textures/entity/wolf/wolf_armor_crackiness_high.png")
|
|
);
|
|
|
|
public WolfArmorLayer(RenderLayerParent<WolfRenderState, WolfModel> renderer, EntityModelSet entityModels, EquipmentLayerRenderer equipmentRenderer) {
|
|
super(renderer);
|
|
this.adultModel = new WolfModel(entityModels.bakeLayer(ModelLayers.WOLF_ARMOR));
|
|
this.babyModel = new WolfModel(entityModels.bakeLayer(ModelLayers.WOLF_BABY_ARMOR));
|
|
this.equipmentRenderer = equipmentRenderer;
|
|
}
|
|
|
|
public void render(PoseStack poseStack, MultiBufferSource multiBufferSource, int i, WolfRenderState wolfRenderState, float f, float g) {
|
|
ItemStack itemStack = wolfRenderState.bodyArmorItem;
|
|
Equippable equippable = itemStack.get(DataComponents.EQUIPPABLE);
|
|
if (equippable != null && !equippable.assetId().isEmpty()) {
|
|
WolfModel wolfModel = wolfRenderState.isBaby ? this.babyModel : this.adultModel;
|
|
wolfModel.setupAnim(wolfRenderState);
|
|
this.equipmentRenderer
|
|
.renderLayers(
|
|
EquipmentClientInfo.LayerType.WOLF_BODY, (ResourceKey<EquipmentAsset>)equippable.assetId().get(), wolfModel, itemStack, poseStack, multiBufferSource, i
|
|
);
|
|
this.maybeRenderCracks(poseStack, multiBufferSource, i, itemStack, wolfModel);
|
|
}
|
|
}
|
|
|
|
private void maybeRenderCracks(PoseStack poseStack, MultiBufferSource bufferSource, int packedLight, ItemStack armorItem, Model model) {
|
|
Level level = Crackiness.WOLF_ARMOR.byDamage(armorItem);
|
|
if (level != Level.NONE) {
|
|
ResourceLocation resourceLocation = (ResourceLocation)ARMOR_CRACK_LOCATIONS.get(level);
|
|
VertexConsumer vertexConsumer = bufferSource.getBuffer(RenderType.armorTranslucent(resourceLocation));
|
|
model.renderToBuffer(poseStack, vertexConsumer, packedLight, OverlayTexture.NO_OVERLAY);
|
|
}
|
|
}
|
|
}
|