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.client.model.geom.PartPose;
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.StriderRenderState;
import net.minecraft.util.Mth;

@Environment(EnvType.CLIENT)
public class StriderModel extends EntityModel<StriderRenderState> {
	public static final MeshTransformer BABY_TRANSFORMER = MeshTransformer.scaling(0.5F);
	private static final String RIGHT_BOTTOM_BRISTLE = "right_bottom_bristle";
	private static final String RIGHT_MIDDLE_BRISTLE = "right_middle_bristle";
	private static final String RIGHT_TOP_BRISTLE = "right_top_bristle";
	private static final String LEFT_TOP_BRISTLE = "left_top_bristle";
	private static final String LEFT_MIDDLE_BRISTLE = "left_middle_bristle";
	private static final String LEFT_BOTTOM_BRISTLE = "left_bottom_bristle";
	private final ModelPart rightLeg;
	private final ModelPart leftLeg;
	private final ModelPart body;
	private final ModelPart rightBottomBristle;
	private final ModelPart rightMiddleBristle;
	private final ModelPart rightTopBristle;
	private final ModelPart leftTopBristle;
	private final ModelPart leftMiddleBristle;
	private final ModelPart leftBottomBristle;

	public StriderModel(ModelPart root) {
		super(root);
		this.rightLeg = root.getChild("right_leg");
		this.leftLeg = root.getChild("left_leg");
		this.body = root.getChild("body");
		this.rightBottomBristle = this.body.getChild("right_bottom_bristle");
		this.rightMiddleBristle = this.body.getChild("right_middle_bristle");
		this.rightTopBristle = this.body.getChild("right_top_bristle");
		this.leftTopBristle = this.body.getChild("left_top_bristle");
		this.leftMiddleBristle = this.body.getChild("left_middle_bristle");
		this.leftBottomBristle = this.body.getChild("left_bottom_bristle");
	}

	public static LayerDefinition createBodyLayer() {
		MeshDefinition meshDefinition = new MeshDefinition();
		PartDefinition partDefinition = meshDefinition.getRoot();
		partDefinition.addOrReplaceChild(
			"right_leg", CubeListBuilder.create().texOffs(0, 32).addBox(-2.0F, 0.0F, -2.0F, 4.0F, 16.0F, 4.0F), PartPose.offset(-4.0F, 8.0F, 0.0F)
		);
		partDefinition.addOrReplaceChild(
			"left_leg", CubeListBuilder.create().texOffs(0, 55).addBox(-2.0F, 0.0F, -2.0F, 4.0F, 16.0F, 4.0F), PartPose.offset(4.0F, 8.0F, 0.0F)
		);
		PartDefinition partDefinition2 = partDefinition.addOrReplaceChild(
			"body", CubeListBuilder.create().texOffs(0, 0).addBox(-8.0F, -6.0F, -8.0F, 16.0F, 14.0F, 16.0F), PartPose.offset(0.0F, 1.0F, 0.0F)
		);
		partDefinition2.addOrReplaceChild(
			"right_bottom_bristle",
			CubeListBuilder.create().texOffs(16, 65).addBox(-12.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F, true),
			PartPose.offsetAndRotation(-8.0F, 4.0F, -8.0F, 0.0F, 0.0F, -1.2217305F)
		);
		partDefinition2.addOrReplaceChild(
			"right_middle_bristle",
			CubeListBuilder.create().texOffs(16, 49).addBox(-12.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F, true),
			PartPose.offsetAndRotation(-8.0F, -1.0F, -8.0F, 0.0F, 0.0F, -1.134464F)
		);
		partDefinition2.addOrReplaceChild(
			"right_top_bristle",
			CubeListBuilder.create().texOffs(16, 33).addBox(-12.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F, true),
			PartPose.offsetAndRotation(-8.0F, -5.0F, -8.0F, 0.0F, 0.0F, -0.87266463F)
		);
		partDefinition2.addOrReplaceChild(
			"left_top_bristle",
			CubeListBuilder.create().texOffs(16, 33).addBox(0.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F),
			PartPose.offsetAndRotation(8.0F, -6.0F, -8.0F, 0.0F, 0.0F, 0.87266463F)
		);
		partDefinition2.addOrReplaceChild(
			"left_middle_bristle",
			CubeListBuilder.create().texOffs(16, 49).addBox(0.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F),
			PartPose.offsetAndRotation(8.0F, -2.0F, -8.0F, 0.0F, 0.0F, 1.134464F)
		);
		partDefinition2.addOrReplaceChild(
			"left_bottom_bristle",
			CubeListBuilder.create().texOffs(16, 65).addBox(0.0F, 0.0F, 0.0F, 12.0F, 0.0F, 16.0F),
			PartPose.offsetAndRotation(8.0F, 3.0F, -8.0F, 0.0F, 0.0F, 1.2217305F)
		);
		return LayerDefinition.create(meshDefinition, 64, 128);
	}

	public void setupAnim(StriderRenderState striderRenderState) {
		super.setupAnim(striderRenderState);
		float f = striderRenderState.walkAnimationPos;
		float g = Math.min(striderRenderState.walkAnimationSpeed, 0.25F);
		if (!striderRenderState.isRidden) {
			this.body.xRot = striderRenderState.xRot * (float) (Math.PI / 180.0);
			this.body.yRot = striderRenderState.yRot * (float) (Math.PI / 180.0);
		} else {
			this.body.xRot = 0.0F;
			this.body.yRot = 0.0F;
		}

		float h = 1.5F;
		this.body.zRot = 0.1F * Mth.sin(f * 1.5F) * 4.0F * g;
		this.body.y = 2.0F;
		this.body.y = this.body.y - 2.0F * Mth.cos(f * 1.5F) * 2.0F * g;
		this.leftLeg.xRot = Mth.sin(f * 1.5F * 0.5F) * 2.0F * g;
		this.rightLeg.xRot = Mth.sin(f * 1.5F * 0.5F + (float) Math.PI) * 2.0F * g;
		this.leftLeg.zRot = (float) (Math.PI / 18) * Mth.cos(f * 1.5F * 0.5F) * g;
		this.rightLeg.zRot = (float) (Math.PI / 18) * Mth.cos(f * 1.5F * 0.5F + (float) Math.PI) * g;
		this.leftLeg.y = 8.0F + 2.0F * Mth.sin(f * 1.5F * 0.5F + (float) Math.PI) * 2.0F * g;
		this.rightLeg.y = 8.0F + 2.0F * Mth.sin(f * 1.5F * 0.5F) * 2.0F * g;
		this.rightBottomBristle.zRot = -1.2217305F;
		this.rightMiddleBristle.zRot = -1.134464F;
		this.rightTopBristle.zRot = -0.87266463F;
		this.leftTopBristle.zRot = 0.87266463F;
		this.leftMiddleBristle.zRot = 1.134464F;
		this.leftBottomBristle.zRot = 1.2217305F;
		float i = Mth.cos(f * 1.5F + (float) Math.PI) * g;
		this.rightBottomBristle.zRot += i * 1.3F;
		this.rightMiddleBristle.zRot += i * 1.2F;
		this.rightTopBristle.zRot += i * 0.6F;
		this.leftTopBristle.zRot += i * 0.6F;
		this.leftMiddleBristle.zRot += i * 1.2F;
		this.leftBottomBristle.zRot += i * 1.3F;
		float j = 1.0F;
		float k = 1.0F;
		this.rightBottomBristle.zRot = this.rightBottomBristle.zRot + 0.05F * Mth.sin(striderRenderState.ageInTicks * 1.0F * -0.4F);
		this.rightMiddleBristle.zRot = this.rightMiddleBristle.zRot + 0.1F * Mth.sin(striderRenderState.ageInTicks * 1.0F * 0.2F);
		this.rightTopBristle.zRot = this.rightTopBristle.zRot + 0.1F * Mth.sin(striderRenderState.ageInTicks * 1.0F * 0.4F);
		this.leftTopBristle.zRot = this.leftTopBristle.zRot + 0.1F * Mth.sin(striderRenderState.ageInTicks * 1.0F * 0.4F);
		this.leftMiddleBristle.zRot = this.leftMiddleBristle.zRot + 0.1F * Mth.sin(striderRenderState.ageInTicks * 1.0F * 0.2F);
		this.leftBottomBristle.zRot = this.leftBottomBristle.zRot + 0.05F * Mth.sin(striderRenderState.ageInTicks * 1.0F * -0.4F);
	}
}