package net.minecraft.client.model;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.world.entity.animal.Cat;

@Environment(EnvType.CLIENT)
public class CatModel<T extends Cat> extends OcelotModel<T> {
	private float lieDownAmount;
	private float lieDownAmountTail;
	private float relaxStateOneAmount;

	public CatModel(ModelPart root) {
		super(root);
	}

	public void prepareMobModel(T entity, float limbSwing, float limbSwingAmount, float partialTick) {
		this.lieDownAmount = entity.getLieDownAmount(partialTick);
		this.lieDownAmountTail = entity.getLieDownAmountTail(partialTick);
		this.relaxStateOneAmount = entity.getRelaxStateOneAmount(partialTick);
		if (this.lieDownAmount <= 0.0F) {
			this.head.xRot = 0.0F;
			this.head.zRot = 0.0F;
			this.leftFrontLeg.xRot = 0.0F;
			this.leftFrontLeg.zRot = 0.0F;
			this.rightFrontLeg.xRot = 0.0F;
			this.rightFrontLeg.zRot = 0.0F;
			this.rightFrontLeg.x = -1.2F;
			this.leftHindLeg.xRot = 0.0F;
			this.rightHindLeg.xRot = 0.0F;
			this.rightHindLeg.zRot = 0.0F;
			this.rightHindLeg.x = -1.1F;
			this.rightHindLeg.y = 18.0F;
		}

		super.prepareMobModel(entity, limbSwing, limbSwingAmount, partialTick);
		if (entity.isInSittingPose()) {
			this.body.xRot = (float) (Math.PI / 4);
			this.body.y += -4.0F;
			this.body.z += 5.0F;
			this.head.y += -3.3F;
			this.head.z++;
			this.tail1.y += 8.0F;
			this.tail1.z += -2.0F;
			this.tail2.y += 2.0F;
			this.tail2.z += -0.8F;
			this.tail1.xRot = 1.7278761F;
			this.tail2.xRot = 2.670354F;
			this.leftFrontLeg.xRot = (float) (-Math.PI / 20);
			this.leftFrontLeg.y = 16.1F;
			this.leftFrontLeg.z = -7.0F;
			this.rightFrontLeg.xRot = (float) (-Math.PI / 20);
			this.rightFrontLeg.y = 16.1F;
			this.rightFrontLeg.z = -7.0F;
			this.leftHindLeg.xRot = (float) (-Math.PI / 2);
			this.leftHindLeg.y = 21.0F;
			this.leftHindLeg.z = 1.0F;
			this.rightHindLeg.xRot = (float) (-Math.PI / 2);
			this.rightHindLeg.y = 21.0F;
			this.rightHindLeg.z = 1.0F;
			this.state = 3;
		}
	}

	/**
	 * Sets this entity's model rotation angles
	 */
	public void setupAnim(T entity, float limbSwing, float limbSwingAmount, float ageInTicks, float netHeadYaw, float headPitch) {
		super.setupAnim(entity, limbSwing, limbSwingAmount, ageInTicks, netHeadYaw, headPitch);
		if (this.lieDownAmount > 0.0F) {
			this.head.zRot = ModelUtils.rotlerpRad(this.head.zRot, -1.2707963F, this.lieDownAmount);
			this.head.yRot = ModelUtils.rotlerpRad(this.head.yRot, 1.2707963F, this.lieDownAmount);
			this.leftFrontLeg.xRot = -1.2707963F;
			this.rightFrontLeg.xRot = -0.47079635F;
			this.rightFrontLeg.zRot = -0.2F;
			this.rightFrontLeg.x = -0.2F;
			this.leftHindLeg.xRot = -0.4F;
			this.rightHindLeg.xRot = 0.5F;
			this.rightHindLeg.zRot = -0.5F;
			this.rightHindLeg.x = -0.3F;
			this.rightHindLeg.y = 20.0F;
			this.tail1.xRot = ModelUtils.rotlerpRad(this.tail1.xRot, 0.8F, this.lieDownAmountTail);
			this.tail2.xRot = ModelUtils.rotlerpRad(this.tail2.xRot, -0.4F, this.lieDownAmountTail);
		}

		if (this.relaxStateOneAmount > 0.0F) {
			this.head.xRot = ModelUtils.rotlerpRad(this.head.xRot, -0.58177644F, this.relaxStateOneAmount);
		}
	}
}