package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.math.Axis;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.block.BlockRenderDispatcher;
import net.minecraft.client.renderer.entity.state.TntRenderState;
import net.minecraft.util.Mth;
import net.minecraft.world.entity.item.PrimedTnt;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

@OnlyIn(Dist.CLIENT)
public class TntRenderer extends EntityRenderer<PrimedTnt, TntRenderState> {
    private final BlockRenderDispatcher blockRenderer;

    public TntRenderer(EntityRendererProvider.Context p_174426_) {
        super(p_174426_);
        this.shadowRadius = 0.5F;
        this.blockRenderer = p_174426_.getBlockRenderDispatcher();
    }

    public void render(TntRenderState p_361297_, PoseStack p_116180_, MultiBufferSource p_116181_, int p_116182_) {
        p_116180_.pushPose();
        p_116180_.translate(0.0F, 0.5F, 0.0F);
        float f = p_361297_.fuseRemainingInTicks;
        if (p_361297_.fuseRemainingInTicks < 10.0F) {
            float f1 = 1.0F - p_361297_.fuseRemainingInTicks / 10.0F;
            f1 = Mth.clamp(f1, 0.0F, 1.0F);
            f1 *= f1;
            f1 *= f1;
            float f2 = 1.0F + f1 * 0.3F;
            p_116180_.scale(f2, f2, f2);
        }

        p_116180_.mulPose(Axis.YP.rotationDegrees(-90.0F));
        p_116180_.translate(-0.5F, -0.5F, 0.5F);
        p_116180_.mulPose(Axis.YP.rotationDegrees(90.0F));
        if (p_361297_.blockState != null) {
            TntMinecartRenderer.renderWhiteSolidBlock(this.blockRenderer, p_361297_.blockState, p_116180_, p_116181_, p_116182_, (int)f / 5 % 2 == 0);
        }

        p_116180_.popPose();
        super.render(p_361297_, p_116180_, p_116181_, p_116182_);
    }

    public TntRenderState createRenderState() {
        return new TntRenderState();
    }

    public void extractRenderState(PrimedTnt p_366432_, TntRenderState p_365560_, float p_367967_) {
        super.extractRenderState(p_366432_, p_365560_, p_367967_);
        p_365560_.fuseRemainingInTicks = p_366432_.getFuse() - p_367967_ + 1.0F;
        p_365560_.blockState = p_366432_.getBlockState();
    }
}