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

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
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.EntityRenderState;
import net.minecraft.util.Mth;

@Environment(EnvType.CLIENT)
public class EndermiteModel extends EntityModel<EntityRenderState> {
	private static final int BODY_COUNT = 4;
	private static final int[][] BODY_SIZES = new int[][]{{4, 3, 2}, {6, 4, 5}, {3, 3, 1}, {1, 2, 1}};
	private static final int[][] BODY_TEXS = new int[][]{{0, 0}, {0, 5}, {0, 14}, {0, 18}};
	private final ModelPart[] bodyParts = new ModelPart[4];

	public EndermiteModel(final ModelPart root) {
		super(root);

		for (int i = 0; i < 4; i++) {
			this.bodyParts[i] = root.getChild(createSegmentName(i));
		}
	}

	private static String createSegmentName(final int i) {
		return "segment" + i;
	}

	public static LayerDefinition createBodyLayer() {
		MeshDefinition mesh = new MeshDefinition();
		PartDefinition root = mesh.getRoot();
		float placement = -3.5F;

		for (int i = 0; i < 4; i++) {
			root.addOrReplaceChild(
				createSegmentName(i),
				CubeListBuilder.create()
					.texOffs(BODY_TEXS[i][0], BODY_TEXS[i][1])
					.addBox(BODY_SIZES[i][0] * -0.5F, 0.0F, BODY_SIZES[i][2] * -0.5F, BODY_SIZES[i][0], BODY_SIZES[i][1], BODY_SIZES[i][2]),
				PartPose.offset(0.0F, 24 - BODY_SIZES[i][1], placement)
			);
			if (i < 3) {
				placement += (BODY_SIZES[i][2] + BODY_SIZES[i + 1][2]) * 0.5F;
			}
		}

		return LayerDefinition.create(mesh, 64, 32);
	}

	public void setupAnim(final EntityRenderState state) {
		super.setupAnim(state);

		for (int i = 0; i < this.bodyParts.length; i++) {
			this.bodyParts[i].yRot = Mth.cos(state.ageInTicks * 0.9F + i * 0.15F * (float) Math.PI) * (float) Math.PI * 0.01F * (1 + Math.abs(i - 2));
			this.bodyParts[i].x = Mth.sin(state.ageInTicks * 0.9F + i * 0.15F * (float) Math.PI) * (float) Math.PI * 0.1F * Math.abs(i - 2);
		}
	}
}