package net.minecraft.client.renderer.entity.layers;

import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.entity.RenderLayerParent;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.PowerableMob;

@Environment(EnvType.CLIENT)
public abstract class EnergySwirlLayer<T extends Entity & PowerableMob, M extends EntityModel<T>> extends RenderLayer<T, M> {
	public EnergySwirlLayer(RenderLayerParent<T, M> renderer) {
		super(renderer);
	}

	@Override
	public void render(
		PoseStack poseStack,
		MultiBufferSource bufferSource,
		int packedLight,
		T livingEntity,
		float limbSwing,
		float limbSwingAmount,
		float partialTick,
		float ageInTicks,
		float netHeadYaw,
		float headPitch
	) {
		if (livingEntity.isPowered()) {
			float f = livingEntity.tickCount + partialTick;
			EntityModel<T> entityModel = this.model();
			entityModel.prepareMobModel(livingEntity, limbSwing, limbSwingAmount, partialTick);
			this.getParentModel().copyPropertiesTo(entityModel);
			VertexConsumer vertexConsumer = bufferSource.getBuffer(RenderType.energySwirl(this.getTextureLocation(), this.xOffset(f) % 1.0F, f * 0.01F % 1.0F));
			entityModel.setupAnim(livingEntity, limbSwing, limbSwingAmount, ageInTicks, netHeadYaw, headPitch);
			entityModel.renderToBuffer(poseStack, vertexConsumer, packedLight, OverlayTexture.NO_OVERLAY, -8355712);
		}
	}

	protected abstract float xOffset(float tickCount);

	protected abstract ResourceLocation getTextureLocation();

	protected abstract EntityModel<T> model();
}