package eu.midnightdust.hats.christmas;

import net.minecraft.client.model.*;
import net.minecraft.client.render.VertexConsumer;
import net.minecraft.client.render.entity.model.SinglePartEntityModel;
import net.minecraft.client.util.math.MatrixStack;
import net.minecraft.entity.LivingEntity;

public class ChristmasHatModel<T extends LivingEntity> extends SinglePartEntityModel<T> {
	private final ModelPart headwear;

	public ChristmasHatModel(ModelPart root) {
		headwear = root;
		root.setPivot(5.0F, -9.0F, -5.0F);

		ModelPart bone = headwear.getChild("bone");
		bone.setPivot(-8.5F, -0.1F, 1.5F);
		setRotationAngle(bone, -0.0524F, 0.0F, 0.0349F);

		ModelPart bone2 = bone.getChild("bone2");
		bone2.setPivot(1.5F, -4.0F, 1.5F);
		setRotationAngle(bone2, -0.1222F, 0.0F, 0.0698F);

		ModelPart bone3 = bone2.getChild("bone3");
		bone3.setPivot(1.5F, -4.0F, 1.5F);
		setRotationAngle(bone3, -0.2618F, 0.0F, 0.1047F);
	}
    public static ModelData getModelData(){
        ModelData modelData = new ModelData();
        ModelPartData modelPartData = modelData.getRoot();
        modelPartData.addChild("headwear", ModelPartBuilder.create().uv(0, 0).cuboid(-10.0F, -0.1F, 0.0F, 10.0F, 2.0F, 10.0F), ModelTransform.NONE);
		ModelPartData modelPartData2 = modelPartData.addChild("bone", ModelPartBuilder.create().uv(0, 12).cuboid(0.0F, -4.0F, 0.0F, 7.0F, 4.0F, 7.0F), ModelTransform.rotation(-0.0524F, 0.0F, 0.0349F));
		ModelPartData modelPartData3 = modelPartData2.addChild("bone2", ModelPartBuilder.create().uv(0, 23).cuboid(0.0F, -4.0F, 0.0F, 4.0F, 4.0F, 4.0F), ModelTransform.rotation(-0.1222F, 0.0F, 0.0698F));
		modelPartData3.addChild("bone3", ModelPartBuilder.create().uv(21, 12).cuboid(0.0F, -3.0F, 0.0F, 3.0F, 3.0F, 3.0F), ModelTransform.rotation(-0.2618F, 0.0F, 0.1047F));

        return modelData;
    }
    public ModelPart getPart() {
        return this.headwear;
    }

	@Override
	public void setAngles(T entity, float limbAngle, float limbDistance, float animationProgress, float headYaw, float headPitch) {
	}

	@Override
	public void render(MatrixStack matrixStack, VertexConsumer buffer, int packedLight, int packedOverlay, float red, float green, float blue, float alpha){
		headwear.render(matrixStack, buffer, packedLight, packedOverlay);
	}

	public void setRotationAngle(ModelPart bone, float x, float y, float z) {
		bone.pitch = x;
		bone.yaw = y;
		bone.roll = z;
	}
}