package net.minecraft.client.model.monster.dragon;

import net.minecraft.client.model.EntityModel;
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.PartDefinition;
import net.minecraft.client.renderer.entity.state.EnderDragonRenderState;
import net.minecraft.util.Mth;
import net.minecraft.world.entity.boss.enderdragon.DragonFlightHistory.Sample;

public class EnderDragonModel extends EntityModel<EnderDragonRenderState> {
	private static final int NECK_PART_COUNT = 5;
	private static final int TAIL_PART_COUNT = 12;
	private final ModelPart head;
	private final ModelPart[] neckParts = new ModelPart[5];
	private final ModelPart[] tailParts = new ModelPart[12];
	private final ModelPart jaw;
	private final ModelPart body;
	private final ModelPart leftWing;
	private final ModelPart leftWingTip;
	private final ModelPart leftFrontLeg;
	private final ModelPart leftFrontLegTip;
	private final ModelPart leftFrontFoot;
	private final ModelPart leftRearLeg;
	private final ModelPart leftRearLegTip;
	private final ModelPart leftRearFoot;
	private final ModelPart rightWing;
	private final ModelPart rightWingTip;
	private final ModelPart rightFrontLeg;
	private final ModelPart rightFrontLegTip;
	private final ModelPart rightFrontFoot;
	private final ModelPart rightRearLeg;
	private final ModelPart rightRearLegTip;
	private final ModelPart rightRearFoot;

	private static String neckName(final int index) {
		return "neck" + index;
	}

	private static String tailName(final int index) {
		return "tail" + index;
	}

	public EnderDragonModel(final ModelPart root) {
		super(root);
		this.head = root.getChild("head");
		this.jaw = this.head.getChild("jaw");

		for (int i = 0; i < this.neckParts.length; i++) {
			this.neckParts[i] = root.getChild(neckName(i));
		}

		for (int i = 0; i < this.tailParts.length; i++) {
			this.tailParts[i] = root.getChild(tailName(i));
		}

		this.body = root.getChild("body");
		this.leftWing = this.body.getChild("left_wing");
		this.leftWingTip = this.leftWing.getChild("left_wing_tip");
		this.leftFrontLeg = this.body.getChild("left_front_leg");
		this.leftFrontLegTip = this.leftFrontLeg.getChild("left_front_leg_tip");
		this.leftFrontFoot = this.leftFrontLegTip.getChild("left_front_foot");
		this.leftRearLeg = this.body.getChild("left_hind_leg");
		this.leftRearLegTip = this.leftRearLeg.getChild("left_hind_leg_tip");
		this.leftRearFoot = this.leftRearLegTip.getChild("left_hind_foot");
		this.rightWing = this.body.getChild("right_wing");
		this.rightWingTip = this.rightWing.getChild("right_wing_tip");
		this.rightFrontLeg = this.body.getChild("right_front_leg");
		this.rightFrontLegTip = this.rightFrontLeg.getChild("right_front_leg_tip");
		this.rightFrontFoot = this.rightFrontLegTip.getChild("right_front_foot");
		this.rightRearLeg = this.body.getChild("right_hind_leg");
		this.rightRearLegTip = this.rightRearLeg.getChild("right_hind_leg_tip");
		this.rightRearFoot = this.rightRearLegTip.getChild("right_hind_foot");
	}

	public static LayerDefinition createBodyLayer() {
		MeshDefinition mesh = new MeshDefinition();
		PartDefinition root = mesh.getRoot();
		float zo = -16.0F;
		PartDefinition head = root.addOrReplaceChild(
			"head",
			CubeListBuilder.create()
				.addBox("upperlip", -6.0F, -1.0F, -24.0F, 12, 5, 16, 176, 44)
				.addBox("upperhead", -8.0F, -8.0F, -10.0F, 16, 16, 16, 112, 30)
				.mirror()
				.addBox("scale", -5.0F, -12.0F, -4.0F, 2, 4, 6, 0, 0)
				.addBox("nostril", -5.0F, -3.0F, -22.0F, 2, 2, 4, 112, 0)
				.mirror()
				.addBox("scale", 3.0F, -12.0F, -4.0F, 2, 4, 6, 0, 0)
				.addBox("nostril", 3.0F, -3.0F, -22.0F, 2, 2, 4, 112, 0),
			PartPose.offset(0.0F, 20.0F, -62.0F)
		);
		head.addOrReplaceChild("jaw", CubeListBuilder.create().addBox("jaw", -6.0F, 0.0F, -16.0F, 12, 4, 16, 176, 65), PartPose.offset(0.0F, 4.0F, -8.0F));
		CubeListBuilder spineCubes = CubeListBuilder.create()
			.addBox("box", -5.0F, -5.0F, -5.0F, 10, 10, 10, 192, 104)
			.addBox("scale", -1.0F, -9.0F, -3.0F, 2, 4, 6, 48, 0);

		for (int i = 0; i < 5; i++) {
			root.addOrReplaceChild(neckName(i), spineCubes, PartPose.offset(0.0F, 20.0F, -12.0F - i * 10.0F));
		}

		for (int i = 0; i < 12; i++) {
			root.addOrReplaceChild(tailName(i), spineCubes, PartPose.offset(0.0F, 10.0F, 60.0F + i * 10.0F));
		}

		PartDefinition body = root.addOrReplaceChild(
			"body",
			CubeListBuilder.create()
				.addBox("body", -12.0F, 1.0F, -16.0F, 24, 24, 64, 0, 0)
				.addBox("scale", -1.0F, -5.0F, -10.0F, 2, 6, 12, 220, 53)
				.addBox("scale", -1.0F, -5.0F, 10.0F, 2, 6, 12, 220, 53)
				.addBox("scale", -1.0F, -5.0F, 30.0F, 2, 6, 12, 220, 53),
			PartPose.offset(0.0F, 3.0F, 8.0F)
		);
		PartDefinition leftWing = body.addOrReplaceChild(
			"left_wing",
			CubeListBuilder.create().mirror().addBox("bone", 0.0F, -4.0F, -4.0F, 56, 8, 8, 112, 88).addBox("skin", 0.0F, 0.0F, 2.0F, 56, 0, 56, -56, 88),
			PartPose.offset(12.0F, 2.0F, -6.0F)
		);
		leftWing.addOrReplaceChild(
			"left_wing_tip",
			CubeListBuilder.create().mirror().addBox("bone", 0.0F, -2.0F, -2.0F, 56, 4, 4, 112, 136).addBox("skin", 0.0F, 0.0F, 2.0F, 56, 0, 56, -56, 144),
			PartPose.offset(56.0F, 0.0F, 0.0F)
		);
		PartDefinition leftFrontLeg = body.addOrReplaceChild(
			"left_front_leg",
			CubeListBuilder.create().addBox("main", -4.0F, -4.0F, -4.0F, 8, 24, 8, 112, 104),
			PartPose.offsetAndRotation(12.0F, 17.0F, -6.0F, 1.3F, 0.0F, 0.0F)
		);
		PartDefinition leftFrontLegTip = leftFrontLeg.addOrReplaceChild(
			"left_front_leg_tip",
			CubeListBuilder.create().addBox("main", -3.0F, -1.0F, -3.0F, 6, 24, 6, 226, 138),
			PartPose.offsetAndRotation(0.0F, 20.0F, -1.0F, -0.5F, 0.0F, 0.0F)
		);
		leftFrontLegTip.addOrReplaceChild(
			"left_front_foot",
			CubeListBuilder.create().addBox("main", -4.0F, 0.0F, -12.0F, 8, 4, 16, 144, 104),
			PartPose.offsetAndRotation(0.0F, 23.0F, 0.0F, 0.75F, 0.0F, 0.0F)
		);
		PartDefinition leftRearLeg = body.addOrReplaceChild(
			"left_hind_leg",
			CubeListBuilder.create().addBox("main", -8.0F, -4.0F, -8.0F, 16, 32, 16, 0, 0),
			PartPose.offsetAndRotation(16.0F, 13.0F, 34.0F, 1.0F, 0.0F, 0.0F)
		);
		PartDefinition leftRearLegTip = leftRearLeg.addOrReplaceChild(
			"left_hind_leg_tip",
			CubeListBuilder.create().addBox("main", -6.0F, -2.0F, 0.0F, 12, 32, 12, 196, 0),
			PartPose.offsetAndRotation(0.0F, 32.0F, -4.0F, 0.5F, 0.0F, 0.0F)
		);
		leftRearLegTip.addOrReplaceChild(
			"left_hind_foot",
			CubeListBuilder.create().addBox("main", -9.0F, 0.0F, -20.0F, 18, 6, 24, 112, 0),
			PartPose.offsetAndRotation(0.0F, 31.0F, 4.0F, 0.75F, 0.0F, 0.0F)
		);
		PartDefinition rightWing = body.addOrReplaceChild(
			"right_wing",
			CubeListBuilder.create().addBox("bone", -56.0F, -4.0F, -4.0F, 56, 8, 8, 112, 88).addBox("skin", -56.0F, 0.0F, 2.0F, 56, 0, 56, -56, 88),
			PartPose.offset(-12.0F, 2.0F, -6.0F)
		);
		rightWing.addOrReplaceChild(
			"right_wing_tip",
			CubeListBuilder.create().addBox("bone", -56.0F, -2.0F, -2.0F, 56, 4, 4, 112, 136).addBox("skin", -56.0F, 0.0F, 2.0F, 56, 0, 56, -56, 144),
			PartPose.offset(-56.0F, 0.0F, 0.0F)
		);
		PartDefinition rightFrontLeg = body.addOrReplaceChild(
			"right_front_leg",
			CubeListBuilder.create().addBox("main", -4.0F, -4.0F, -4.0F, 8, 24, 8, 112, 104),
			PartPose.offsetAndRotation(-12.0F, 17.0F, -6.0F, 1.3F, 0.0F, 0.0F)
		);
		PartDefinition rightFrontLegTip = rightFrontLeg.addOrReplaceChild(
			"right_front_leg_tip",
			CubeListBuilder.create().addBox("main", -3.0F, -1.0F, -3.0F, 6, 24, 6, 226, 138),
			PartPose.offsetAndRotation(0.0F, 20.0F, -1.0F, -0.5F, 0.0F, 0.0F)
		);
		rightFrontLegTip.addOrReplaceChild(
			"right_front_foot",
			CubeListBuilder.create().addBox("main", -4.0F, 0.0F, -12.0F, 8, 4, 16, 144, 104),
			PartPose.offsetAndRotation(0.0F, 23.0F, 0.0F, 0.75F, 0.0F, 0.0F)
		);
		PartDefinition rightRearLeg = body.addOrReplaceChild(
			"right_hind_leg",
			CubeListBuilder.create().addBox("main", -8.0F, -4.0F, -8.0F, 16, 32, 16, 0, 0),
			PartPose.offsetAndRotation(-16.0F, 13.0F, 34.0F, 1.0F, 0.0F, 0.0F)
		);
		PartDefinition rightRearLegTip = rightRearLeg.addOrReplaceChild(
			"right_hind_leg_tip",
			CubeListBuilder.create().addBox("main", -6.0F, -2.0F, 0.0F, 12, 32, 12, 196, 0),
			PartPose.offsetAndRotation(0.0F, 32.0F, -4.0F, 0.5F, 0.0F, 0.0F)
		);
		rightRearLegTip.addOrReplaceChild(
			"right_hind_foot",
			CubeListBuilder.create().addBox("main", -9.0F, 0.0F, -20.0F, 18, 6, 24, 112, 0),
			PartPose.offsetAndRotation(0.0F, 31.0F, 4.0F, 0.75F, 0.0F, 0.0F)
		);
		return LayerDefinition.create(mesh, 256, 256);
	}

	public void setupAnim(final EnderDragonRenderState state) {
		super.setupAnim(state);
		float flapTime = state.flapTime * (float) (Math.PI * 2);
		this.jaw.xRot = (Mth.sin(flapTime) + 1.0F) * 0.2F;
		float bounce = Mth.sin(flapTime - 1.0F) + 1.0F;
		bounce = (bounce * bounce + bounce * 2.0F) * 0.05F;
		this.root.y = (bounce - 2.0F) * 16.0F;
		this.root.z = -48.0F;
		this.root.xRot = bounce * 2.0F * (float) (Math.PI / 180.0);
		float xx = this.neckParts[0].x;
		float yy = this.neckParts[0].y;
		float zz = this.neckParts[0].z;
		float rotScale = 1.5F;
		Sample start = state.getHistoricalPos(6);
		float rot2 = Mth.wrapDegrees(state.getHistoricalPos(5).yRot() - state.getHistoricalPos(10).yRot());
		float rot = Mth.wrapDegrees(state.getHistoricalPos(5).yRot() + rot2 / 2.0F);

		for (int i = 0; i < 5; i++) {
			ModelPart neck = this.neckParts[i];
			Sample point = state.getHistoricalPos(5 - i);
			float neckXRot = Mth.cos(i * 0.45F + flapTime) * 0.15F;
			neck.yRot = Mth.wrapDegrees(point.yRot() - start.yRot()) * (float) (Math.PI / 180.0) * 1.5F;
			neck.xRot = neckXRot + state.getHeadPartYOffset(i, start, point) * (float) (Math.PI / 180.0) * 1.5F * 5.0F;
			neck.zRot = -Mth.wrapDegrees(point.yRot() - rot) * (float) (Math.PI / 180.0) * 1.5F;
			neck.y = yy;
			neck.z = zz;
			neck.x = xx;
			xx -= Mth.sin(neck.yRot) * Mth.cos(neck.xRot) * 10.0F;
			yy += Mth.sin(neck.xRot) * 10.0F;
			zz -= Mth.cos(neck.yRot) * Mth.cos(neck.xRot) * 10.0F;
		}

		this.head.y = yy;
		this.head.z = zz;
		this.head.x = xx;
		Sample current = state.getHistoricalPos(0);
		this.head.yRot = Mth.wrapDegrees(current.yRot() - start.yRot()) * (float) (Math.PI / 180.0);
		this.head.xRot = Mth.wrapDegrees(state.getHeadPartYOffset(6, start, current)) * (float) (Math.PI / 180.0) * 1.5F * 5.0F;
		this.head.zRot = -Mth.wrapDegrees(current.yRot() - rot) * (float) (Math.PI / 180.0);
		this.body.zRot = -rot2 * 1.5F * (float) (Math.PI / 180.0);
		this.leftWing.xRot = 0.125F - Mth.cos(flapTime) * 0.2F;
		this.leftWing.yRot = -0.25F;
		this.leftWing.zRot = -(Mth.sin(flapTime) + 0.125F) * 0.8F;
		this.leftWingTip.zRot = (Mth.sin(flapTime + 2.0F) + 0.5F) * 0.75F;
		this.rightWing.xRot = this.leftWing.xRot;
		this.rightWing.yRot = -this.leftWing.yRot;
		this.rightWing.zRot = -this.leftWing.zRot;
		this.rightWingTip.zRot = -this.leftWingTip.zRot;
		this.poseLimbs(bounce, this.leftFrontLeg, this.leftFrontLegTip, this.leftFrontFoot, this.leftRearLeg, this.leftRearLegTip, this.leftRearFoot);
		this.poseLimbs(bounce, this.rightFrontLeg, this.rightFrontLegTip, this.rightFrontFoot, this.rightRearLeg, this.rightRearLegTip, this.rightRearFoot);
		float tailXRot = 0.0F;
		yy = this.tailParts[0].y;
		zz = this.tailParts[0].z;
		xx = this.tailParts[0].x;
		start = state.getHistoricalPos(11);

		for (int i = 0; i < 12; i++) {
			Sample point = state.getHistoricalPos(12 + i);
			tailXRot += Mth.sin(i * 0.45F + flapTime) * 0.05F;
			ModelPart tail = this.tailParts[i];
			tail.yRot = (Mth.wrapDegrees(point.yRot() - start.yRot()) * 1.5F + 180.0F) * (float) (Math.PI / 180.0);
			tail.xRot = tailXRot + (float)(point.y() - start.y()) * (float) (Math.PI / 180.0) * 1.5F * 5.0F;
			tail.zRot = Mth.wrapDegrees(point.yRot() - rot) * (float) (Math.PI / 180.0) * 1.5F;
			tail.y = yy;
			tail.z = zz;
			tail.x = xx;
			yy += Mth.sin(tail.xRot) * 10.0F;
			zz -= Mth.cos(tail.yRot) * Mth.cos(tail.xRot) * 10.0F;
			xx -= Mth.sin(tail.yRot) * Mth.cos(tail.xRot) * 10.0F;
		}
	}

	private void poseLimbs(
		final float bounce,
		final ModelPart frontLeg,
		final ModelPart frontLegTip,
		final ModelPart frontFoot,
		final ModelPart rearLeg,
		final ModelPart rearLegTip,
		final ModelPart rearFoot
	) {
		rearLeg.xRot = 1.0F + bounce * 0.1F;
		rearLegTip.xRot = 0.5F + bounce * 0.1F;
		rearFoot.xRot = 0.75F + bounce * 0.1F;
		frontLeg.xRot = 1.3F + bounce * 0.1F;
		frontLegTip.xRot = -0.5F - bounce * 0.1F;
		frontFoot.xRot = 0.75F + bounce * 0.1F;
	}
}