package net.minecraft.client.renderer.entity.layers;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import com.mojang.math.Axis;
import net.minecraft.client.model.SnowGolemModel;
import net.minecraft.client.renderer.ItemBlockRenderTypes;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.block.BlockRenderDispatcher;
import net.minecraft.client.renderer.block.ModelBlockRenderer;
import net.minecraft.client.renderer.block.model.BlockStateModel;
import net.minecraft.client.renderer.entity.LivingEntityRenderer;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.SnowGolemRenderState;
import net.minecraft.client.renderer.texture.TextureAtlas;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

@OnlyIn(Dist.CLIENT)
public class SnowGolemHeadLayer extends RenderLayer<SnowGolemRenderState, SnowGolemModel> {
    private final BlockRenderDispatcher blockRenderer;

    public SnowGolemHeadLayer(RenderLayerParent<SnowGolemRenderState, SnowGolemModel> p_234871_, BlockRenderDispatcher p_234872_) {
        super(p_234871_);
        this.blockRenderer = p_234872_;
    }

    public void render(PoseStack p_117483_, MultiBufferSource p_117484_, int p_117485_, SnowGolemRenderState p_376455_, float p_117487_, float p_117488_) {
        if (p_376455_.hasPumpkin) {
            if (!p_376455_.isInvisible || p_376455_.appearsGlowing) {
                p_117483_.pushPose();
                this.getParentModel().getHead().translateAndRotate(p_117483_);
                float f = 0.625F;
                p_117483_.translate(0.0F, -0.34375F, 0.0F);
                p_117483_.mulPose(Axis.YP.rotationDegrees(180.0F));
                p_117483_.scale(0.625F, -0.625F, -0.625F);
                BlockState blockstate = Blocks.CARVED_PUMPKIN.defaultBlockState();
                BlockStateModel blockstatemodel = this.blockRenderer.getBlockModel(blockstate);
                int i = LivingEntityRenderer.getOverlayCoords(p_376455_, 0.0F);
                p_117483_.translate(-0.5F, -0.5F, -0.5F);
                VertexConsumer vertexconsumer = p_376455_.appearsGlowing && p_376455_.isInvisible
                    ? p_117484_.getBuffer(RenderType.outline(TextureAtlas.LOCATION_BLOCKS))
                    : p_117484_.getBuffer(ItemBlockRenderTypes.getRenderType(blockstate));
                ModelBlockRenderer.renderModel(p_117483_.last(), vertexconsumer, blockstatemodel, 0.0F, 0.0F, 0.0F, p_117485_, i);
                p_117483_.popPose();
            }
        }
    }
}