package net.minecraft.client.model;

import java.util.Set;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.client.model.geom.PartPose;
import net.minecraft.client.model.geom.builders.CubeDeformation;
import net.minecraft.client.model.geom.builders.CubeListBuilder;
import net.minecraft.client.model.geom.builders.MeshDefinition;
import net.minecraft.client.model.geom.builders.MeshTransformer;
import net.minecraft.client.model.geom.builders.PartDefinition;
import net.minecraft.client.renderer.entity.state.FelineRenderState;
import net.minecraft.util.Mth;

@Environment(EnvType.CLIENT)
public class FelineModel<T extends FelineRenderState> extends EntityModel<T> {
	public static final MeshTransformer BABY_TRANSFORMER = new BabyModelTransform(true, 10.0F, 4.0F, Set.of("head"));
	private static final float XO = 0.0F;
	private static final float YO = 16.0F;
	private static final float ZO = -9.0F;
	protected static final float BACK_LEG_Y = 18.0F;
	protected static final float BACK_LEG_Z = 5.0F;
	protected static final float FRONT_LEG_Y = 14.1F;
	private static final float FRONT_LEG_Z = -5.0F;
	private static final String TAIL_1 = "tail1";
	private static final String TAIL_2 = "tail2";
	protected final ModelPart leftHindLeg;
	protected final ModelPart rightHindLeg;
	protected final ModelPart leftFrontLeg;
	protected final ModelPart rightFrontLeg;
	protected final ModelPart tail1;
	protected final ModelPart tail2;
	protected final ModelPart head;
	protected final ModelPart body;

	public FelineModel(ModelPart root) {
		super(root);
		this.head = root.getChild("head");
		this.body = root.getChild("body");
		this.tail1 = root.getChild("tail1");
		this.tail2 = root.getChild("tail2");
		this.leftHindLeg = root.getChild("left_hind_leg");
		this.rightHindLeg = root.getChild("right_hind_leg");
		this.leftFrontLeg = root.getChild("left_front_leg");
		this.rightFrontLeg = root.getChild("right_front_leg");
	}

	public static MeshDefinition createBodyMesh(CubeDeformation cubeDeformation) {
		MeshDefinition meshDefinition = new MeshDefinition();
		PartDefinition partDefinition = meshDefinition.getRoot();
		CubeDeformation cubeDeformation2 = new CubeDeformation(-0.02F);
		partDefinition.addOrReplaceChild(
			"head",
			CubeListBuilder.create()
				.addBox("main", -2.5F, -2.0F, -3.0F, 5.0F, 4.0F, 5.0F, cubeDeformation)
				.addBox("nose", -1.5F, -0.001F, -4.0F, 3, 2, 2, cubeDeformation, 0, 24)
				.addBox("ear1", -2.0F, -3.0F, 0.0F, 1, 1, 2, cubeDeformation, 0, 10)
				.addBox("ear2", 1.0F, -3.0F, 0.0F, 1, 1, 2, cubeDeformation, 6, 10),
			PartPose.offset(0.0F, 15.0F, -9.0F)
		);
		partDefinition.addOrReplaceChild(
			"body",
			CubeListBuilder.create().texOffs(20, 0).addBox(-2.0F, 3.0F, -8.0F, 4.0F, 16.0F, 6.0F, cubeDeformation),
			PartPose.offsetAndRotation(0.0F, 12.0F, -10.0F, (float) (Math.PI / 2), 0.0F, 0.0F)
		);
		partDefinition.addOrReplaceChild(
			"tail1",
			CubeListBuilder.create().texOffs(0, 15).addBox(-0.5F, 0.0F, 0.0F, 1.0F, 8.0F, 1.0F, cubeDeformation),
			PartPose.offsetAndRotation(0.0F, 15.0F, 8.0F, 0.9F, 0.0F, 0.0F)
		);
		partDefinition.addOrReplaceChild(
			"tail2", CubeListBuilder.create().texOffs(4, 15).addBox(-0.5F, 0.0F, 0.0F, 1.0F, 8.0F, 1.0F, cubeDeformation2), PartPose.offset(0.0F, 20.0F, 14.0F)
		);
		CubeListBuilder cubeListBuilder = CubeListBuilder.create().texOffs(8, 13).addBox(-1.0F, 0.0F, 1.0F, 2.0F, 6.0F, 2.0F, cubeDeformation);
		partDefinition.addOrReplaceChild("left_hind_leg", cubeListBuilder, PartPose.offset(1.1F, 18.0F, 5.0F));
		partDefinition.addOrReplaceChild("right_hind_leg", cubeListBuilder, PartPose.offset(-1.1F, 18.0F, 5.0F));
		CubeListBuilder cubeListBuilder2 = CubeListBuilder.create().texOffs(40, 0).addBox(-1.0F, 0.0F, 0.0F, 2.0F, 10.0F, 2.0F, cubeDeformation);
		partDefinition.addOrReplaceChild("left_front_leg", cubeListBuilder2, PartPose.offset(1.2F, 14.1F, -5.0F));
		partDefinition.addOrReplaceChild("right_front_leg", cubeListBuilder2, PartPose.offset(-1.2F, 14.1F, -5.0F));
		return meshDefinition;
	}

	public void setupAnim(T felineRenderState) {
		super.setupAnim(felineRenderState);
		float f = felineRenderState.ageScale;
		if (felineRenderState.isCrouching) {
			this.body.y += 1.0F * f;
			this.head.y += 2.0F * f;
			this.tail1.y += 1.0F * f;
			this.tail2.y += -4.0F * f;
			this.tail2.z += 2.0F * f;
			this.tail1.xRot = (float) (Math.PI / 2);
			this.tail2.xRot = (float) (Math.PI / 2);
		} else if (felineRenderState.isSprinting) {
			this.tail2.y = this.tail1.y;
			this.tail2.z += 2.0F * f;
			this.tail1.xRot = (float) (Math.PI / 2);
			this.tail2.xRot = (float) (Math.PI / 2);
		}

		this.head.xRot = felineRenderState.xRot * (float) (Math.PI / 180.0);
		this.head.yRot = felineRenderState.yRot * (float) (Math.PI / 180.0);
		if (!felineRenderState.isSitting) {
			this.body.xRot = (float) (Math.PI / 2);
			float g = felineRenderState.walkAnimationSpeed;
			float h = felineRenderState.walkAnimationPos;
			if (felineRenderState.isSprinting) {
				this.leftHindLeg.xRot = Mth.cos(h * 0.6662F) * g;
				this.rightHindLeg.xRot = Mth.cos(h * 0.6662F + 0.3F) * g;
				this.leftFrontLeg.xRot = Mth.cos(h * 0.6662F + (float) Math.PI + 0.3F) * g;
				this.rightFrontLeg.xRot = Mth.cos(h * 0.6662F + (float) Math.PI) * g;
				this.tail2.xRot = 1.7278761F + (float) (Math.PI / 10) * Mth.cos(h) * g;
			} else {
				this.leftHindLeg.xRot = Mth.cos(h * 0.6662F) * g;
				this.rightHindLeg.xRot = Mth.cos(h * 0.6662F + (float) Math.PI) * g;
				this.leftFrontLeg.xRot = Mth.cos(h * 0.6662F + (float) Math.PI) * g;
				this.rightFrontLeg.xRot = Mth.cos(h * 0.6662F) * g;
				if (!felineRenderState.isCrouching) {
					this.tail2.xRot = 1.7278761F + (float) (Math.PI / 4) * Mth.cos(h) * g;
				} else {
					this.tail2.xRot = 1.7278761F + 0.47123894F * Mth.cos(h) * g;
				}
			}
		}

		if (felineRenderState.isSitting) {
			this.body.xRot = (float) (Math.PI / 4);
			this.body.y += -4.0F * f;
			this.body.z += 5.0F * f;
			this.head.y += -3.3F * f;
			this.head.z += 1.0F * f;
			this.tail1.y += 8.0F * f;
			this.tail1.z += -2.0F * f;
			this.tail2.y += 2.0F * f;
			this.tail2.z += -0.8F * f;
			this.tail1.xRot = 1.7278761F;
			this.tail2.xRot = 2.670354F;
			this.leftFrontLeg.xRot = (float) (-Math.PI / 20);
			this.leftFrontLeg.y += 2.0F * f;
			this.leftFrontLeg.z -= 2.0F * f;
			this.rightFrontLeg.xRot = (float) (-Math.PI / 20);
			this.rightFrontLeg.y += 2.0F * f;
			this.rightFrontLeg.z -= 2.0F * f;
			this.leftHindLeg.xRot = (float) (-Math.PI / 2);
			this.leftHindLeg.y += 3.0F * f;
			this.leftHindLeg.z -= 4.0F * f;
			this.rightHindLeg.xRot = (float) (-Math.PI / 2);
			this.rightHindLeg.y += 3.0F * f;
			this.rightHindLeg.z -= 4.0F * f;
		}

		if (felineRenderState.lieDownAmount > 0.0F) {
			this.head.zRot = Mth.rotLerp(felineRenderState.lieDownAmount, this.head.zRot, -1.2707963F);
			this.head.yRot = Mth.rotLerp(felineRenderState.lieDownAmount, this.head.yRot, 1.2707963F);
			this.leftFrontLeg.xRot = -1.2707963F;
			this.rightFrontLeg.xRot = -0.47079635F;
			this.rightFrontLeg.zRot = -0.2F;
			this.rightFrontLeg.x += f;
			this.leftHindLeg.xRot = -0.4F;
			this.rightHindLeg.xRot = 0.5F;
			this.rightHindLeg.zRot = -0.5F;
			this.rightHindLeg.x += 0.8F * f;
			this.rightHindLeg.y += 2.0F * f;
			this.tail1.xRot = Mth.rotLerp(felineRenderState.lieDownAmountTail, this.tail1.xRot, 0.8F);
			this.tail2.xRot = Mth.rotLerp(felineRenderState.lieDownAmountTail, this.tail2.xRot, -0.4F);
		}

		if (felineRenderState.relaxStateOneAmount > 0.0F) {
			this.head.xRot = Mth.rotLerp(felineRenderState.relaxStateOneAmount, this.head.xRot, -0.58177644F);
		}
	}
}