package net.minecraft.client.model;

import java.util.Arrays;
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.PartDefinition;
import net.minecraft.client.renderer.entity.state.EntityRenderState;
import net.minecraft.util.Mth;

@Environment(EnvType.CLIENT)
public class SilverfishModel extends EntityModel<EntityRenderState> {
	private static final int BODY_COUNT = 7;
	private final ModelPart[] bodyParts = new ModelPart[7];
	private final ModelPart[] bodyLayers = new ModelPart[3];
	private static final int[][] BODY_SIZES = new int[][]{{3, 2, 2}, {4, 3, 2}, {6, 4, 3}, {3, 3, 3}, {2, 2, 3}, {2, 1, 2}, {1, 1, 2}};
	private static final int[][] BODY_TEXS = new int[][]{{0, 0}, {0, 4}, {0, 9}, {0, 16}, {0, 22}, {11, 0}, {13, 4}};

	public SilverfishModel(ModelPart root) {
		super(root);
		Arrays.setAll(this.bodyParts, i -> root.getChild(getSegmentName(i)));
		Arrays.setAll(this.bodyLayers, i -> root.getChild(getLayerName(i)));
	}

	private static String getLayerName(int index) {
		return "layer" + index;
	}

	private static String getSegmentName(int index) {
		return "segment" + index;
	}

	public static LayerDefinition createBodyLayer() {
		MeshDefinition meshDefinition = new MeshDefinition();
		PartDefinition partDefinition = meshDefinition.getRoot();
		float[] fs = new float[7];
		float f = -3.5F;

		for (int i = 0; i < 7; i++) {
			partDefinition.addOrReplaceChild(
				getSegmentName(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], f)
			);
			fs[i] = f;
			if (i < 6) {
				f += (BODY_SIZES[i][2] + BODY_SIZES[i + 1][2]) * 0.5F;
			}
		}

		partDefinition.addOrReplaceChild(
			getLayerName(0),
			CubeListBuilder.create().texOffs(20, 0).addBox(-5.0F, 0.0F, BODY_SIZES[2][2] * -0.5F, 10.0F, 8.0F, BODY_SIZES[2][2]),
			PartPose.offset(0.0F, 16.0F, fs[2])
		);
		partDefinition.addOrReplaceChild(
			getLayerName(1),
			CubeListBuilder.create().texOffs(20, 11).addBox(-3.0F, 0.0F, BODY_SIZES[4][2] * -0.5F, 6.0F, 4.0F, BODY_SIZES[4][2]),
			PartPose.offset(0.0F, 20.0F, fs[4])
		);
		partDefinition.addOrReplaceChild(
			getLayerName(2),
			CubeListBuilder.create().texOffs(20, 18).addBox(-3.0F, 0.0F, BODY_SIZES[4][2] * -0.5F, 6.0F, 5.0F, BODY_SIZES[1][2]),
			PartPose.offset(0.0F, 19.0F, fs[1])
		);
		return LayerDefinition.create(meshDefinition, 64, 32);
	}

	@Override
	public void setupAnim(EntityRenderState renderState) {
		super.setupAnim(renderState);

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

		this.bodyLayers[0].yRot = this.bodyParts[2].yRot;
		this.bodyLayers[1].yRot = this.bodyParts[4].yRot;
		this.bodyLayers[1].x = this.bodyParts[4].x;
		this.bodyLayers[2].yRot = this.bodyParts[1].yRot;
		this.bodyLayers[2].x = this.bodyParts[1].x;
	}
}