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

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.math.Axis;
import net.minecraft.client.model.ArmedModel;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.entity.state.ArmedEntityRenderState;
import net.minecraft.client.renderer.item.ItemStackRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.world.entity.HumanoidArm;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

@OnlyIn(Dist.CLIENT)
public class ItemInHandLayer<S extends ArmedEntityRenderState, M extends EntityModel<S> & ArmedModel> extends RenderLayer<S, M> {
    public ItemInHandLayer(RenderLayerParent<S, M> p_234846_) {
        super(p_234846_);
    }

    public void render(PoseStack p_117204_, MultiBufferSource p_117205_, int p_117206_, S p_375467_, float p_117208_, float p_117209_) {
        this.renderArmWithItem(p_375467_, p_375467_.rightHandItem, HumanoidArm.RIGHT, p_117204_, p_117205_, p_117206_);
        this.renderArmWithItem(p_375467_, p_375467_.leftHandItem, HumanoidArm.LEFT, p_117204_, p_117205_, p_117206_);
    }

    protected void renderArmWithItem(
        S p_377398_, ItemStackRenderState p_377304_, HumanoidArm p_117188_, PoseStack p_117189_, MultiBufferSource p_117190_, int p_117191_
    ) {
        if (!p_377304_.isEmpty()) {
            p_117189_.pushPose();
            this.getParentModel().translateToHand(p_117188_, p_117189_);
            p_117189_.mulPose(Axis.XP.rotationDegrees(-90.0F));
            p_117189_.mulPose(Axis.YP.rotationDegrees(180.0F));
            boolean flag = p_117188_ == HumanoidArm.LEFT;
            p_117189_.translate((flag ? -1 : 1) / 16.0F, 0.125F, -0.625F);
            p_377304_.render(p_117189_, p_117190_, p_117191_, OverlayTexture.NO_OVERLAY);
            p_117189_.popPose();
        }
    }
}