package net.minecraft.client.model;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.animation.definitions.SnifferAnimation;
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.LayerDefinition;
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.SnifferRenderState;

@Environment(EnvType.CLIENT)
public class SnifferModel extends EntityModel<SnifferRenderState> {
	public static final MeshTransformer BABY_TRANSFORMER = MeshTransformer.scaling(0.5F);
	private static final float WALK_ANIMATION_SPEED_MAX = 9.0F;
	private static final float WALK_ANIMATION_SCALE_FACTOR = 100.0F;
	private final ModelPart head;

	public SnifferModel(ModelPart root) {
		super(root);
		this.head = root.getChild("bone").getChild("body").getChild("head");
	}

	public static LayerDefinition createBodyLayer() {
		MeshDefinition meshDefinition = new MeshDefinition();
		PartDefinition partDefinition = meshDefinition.getRoot();
		PartDefinition partDefinition2 = partDefinition.addOrReplaceChild("bone", CubeListBuilder.create(), PartPose.offset(0.0F, 5.0F, 0.0F));
		PartDefinition partDefinition3 = partDefinition2.addOrReplaceChild(
			"body",
			CubeListBuilder.create()
				.texOffs(62, 68)
				.addBox(-12.5F, -14.0F, -20.0F, 25.0F, 29.0F, 40.0F, new CubeDeformation(0.0F))
				.texOffs(62, 0)
				.addBox(-12.5F, -14.0F, -20.0F, 25.0F, 24.0F, 40.0F, new CubeDeformation(0.5F))
				.texOffs(87, 68)
				.addBox(-12.5F, 12.0F, -20.0F, 25.0F, 0.0F, 40.0F, new CubeDeformation(0.0F)),
			PartPose.offset(0.0F, 0.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			"right_front_leg",
			CubeListBuilder.create().texOffs(32, 87).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(-7.5F, 10.0F, -15.0F)
		);
		partDefinition2.addOrReplaceChild(
			"right_mid_leg",
			CubeListBuilder.create().texOffs(32, 105).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(-7.5F, 10.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			"right_hind_leg",
			CubeListBuilder.create().texOffs(32, 123).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(-7.5F, 10.0F, 15.0F)
		);
		partDefinition2.addOrReplaceChild(
			"left_front_leg",
			CubeListBuilder.create().texOffs(0, 87).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(7.5F, 10.0F, -15.0F)
		);
		partDefinition2.addOrReplaceChild(
			"left_mid_leg",
			CubeListBuilder.create().texOffs(0, 105).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(7.5F, 10.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			"left_hind_leg",
			CubeListBuilder.create().texOffs(0, 123).addBox(-3.5F, -1.0F, -4.0F, 7.0F, 10.0F, 8.0F, new CubeDeformation(0.0F)),
			PartPose.offset(7.5F, 10.0F, 15.0F)
		);
		PartDefinition partDefinition4 = partDefinition3.addOrReplaceChild(
			"head",
			CubeListBuilder.create()
				.texOffs(8, 15)
				.addBox(-6.5F, -7.5F, -11.5F, 13.0F, 18.0F, 11.0F, new CubeDeformation(0.0F))
				.texOffs(8, 4)
				.addBox(-6.5F, 7.5F, -11.5F, 13.0F, 0.0F, 11.0F, new CubeDeformation(0.0F)),
			PartPose.offset(0.0F, 6.5F, -19.48F)
		);
		partDefinition4.addOrReplaceChild(
			"left_ear",
			CubeListBuilder.create().texOffs(2, 0).addBox(0.0F, 0.0F, -3.0F, 1.0F, 19.0F, 7.0F, new CubeDeformation(0.0F)),
			PartPose.offset(6.51F, -7.5F, -4.51F)
		);
		partDefinition4.addOrReplaceChild(
			"right_ear",
			CubeListBuilder.create().texOffs(48, 0).addBox(-1.0F, 0.0F, -3.0F, 1.0F, 19.0F, 7.0F, new CubeDeformation(0.0F)),
			PartPose.offset(-6.51F, -7.5F, -4.51F)
		);
		partDefinition4.addOrReplaceChild(
			"nose",
			CubeListBuilder.create().texOffs(10, 45).addBox(-6.5F, -2.0F, -9.0F, 13.0F, 2.0F, 9.0F, new CubeDeformation(0.0F)),
			PartPose.offset(0.0F, -4.5F, -11.5F)
		);
		partDefinition4.addOrReplaceChild(
			"lower_beak",
			CubeListBuilder.create().texOffs(10, 57).addBox(-6.5F, -7.0F, -8.0F, 13.0F, 12.0F, 9.0F, new CubeDeformation(0.0F)),
			PartPose.offset(0.0F, 2.5F, -12.5F)
		);
		return LayerDefinition.create(meshDefinition, 192, 192);
	}

	public void setupAnim(SnifferRenderState snifferRenderState) {
		super.setupAnim(snifferRenderState);
		this.head.xRot = snifferRenderState.xRot * (float) (Math.PI / 180.0);
		this.head.yRot = snifferRenderState.yRot * (float) (Math.PI / 180.0);
		if (snifferRenderState.isSearching) {
			this.animateWalk(SnifferAnimation.SNIFFER_SNIFF_SEARCH, snifferRenderState.walkAnimationPos, snifferRenderState.walkAnimationSpeed, 9.0F, 100.0F);
		} else {
			this.animateWalk(SnifferAnimation.SNIFFER_WALK, snifferRenderState.walkAnimationPos, snifferRenderState.walkAnimationSpeed, 9.0F, 100.0F);
		}

		this.animate(snifferRenderState.diggingAnimationState, SnifferAnimation.SNIFFER_DIG, snifferRenderState.ageInTicks);
		this.animate(snifferRenderState.sniffingAnimationState, SnifferAnimation.SNIFFER_LONGSNIFF, snifferRenderState.ageInTicks);
		this.animate(snifferRenderState.risingAnimationState, SnifferAnimation.SNIFFER_STAND_UP, snifferRenderState.ageInTicks);
		this.animate(snifferRenderState.feelingHappyAnimationState, SnifferAnimation.SNIFFER_HAPPY, snifferRenderState.ageInTicks);
		this.animate(snifferRenderState.scentingAnimationState, SnifferAnimation.SNIFFER_SNIFFSNIFF, snifferRenderState.ageInTicks);
		if (snifferRenderState.isBaby) {
			this.applyStatic(SnifferAnimation.BABY_TRANSFORM);
		}
	}
}