package net.minecraft.client.renderer.entity;

import com.mojang.blaze3d.vertex.PoseStack;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.model.object.crystal.EndCrystalModel;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.culling.Frustum;
import net.minecraft.client.renderer.entity.state.EndCrystalRenderState;
import net.minecraft.client.renderer.state.level.CameraRenderState;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.core.BlockPos;
import net.minecraft.resources.Identifier;
import net.minecraft.util.Mth;
import net.minecraft.world.entity.boss.enderdragon.EndCrystal;
import net.minecraft.world.phys.Vec3;

public class EndCrystalRenderer extends EntityRenderer<EndCrystal, EndCrystalRenderState> {
	private static final Identifier END_CRYSTAL_LOCATION = Identifier.withDefaultNamespace("textures/entity/end_crystal/end_crystal.png");
	private final EndCrystalModel model;

	public EndCrystalRenderer(final EntityRendererProvider.Context context) {
		super(context);
		this.shadowRadius = 0.5F;
		this.model = new EndCrystalModel(context.bakeLayer(ModelLayers.END_CRYSTAL));
	}

	public void submit(final EndCrystalRenderState state, final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector, final CameraRenderState camera) {
		poseStack.pushPose();
		poseStack.scale(2.0F, 2.0F, 2.0F);
		poseStack.translate(0.0F, -0.5F, 0.0F);
		submitNodeCollector.submitModel(this.model, state, poseStack, END_CRYSTAL_LOCATION, state.lightCoords, OverlayTexture.NO_OVERLAY, state.outlineColor, null);
		poseStack.popPose();
		Vec3 beamOffset = state.beamOffset;
		if (beamOffset != null) {
			float crystalY = getY(state.ageInTicks);
			float deltaX = (float)beamOffset.x;
			float deltaY = (float)beamOffset.y;
			float deltaZ = (float)beamOffset.z;
			poseStack.translate(beamOffset);
			EnderDragonRenderer.submitCrystalBeams(-deltaX, -deltaY + crystalY, -deltaZ, state.ageInTicks, poseStack, submitNodeCollector, state.lightCoords);
		}

		super.submit(state, poseStack, submitNodeCollector, camera);
	}

	public static float getY(final float timeInTicks) {
		float hh = Mth.sin(timeInTicks * 0.2F) / 2.0F + 0.5F;
		hh = (hh * hh + hh) * 0.4F;
		return hh - 1.4F;
	}

	public EndCrystalRenderState createRenderState() {
		return new EndCrystalRenderState();
	}

	public void extractRenderState(final EndCrystal entity, final EndCrystalRenderState state, final float partialTicks) {
		super.extractRenderState(entity, state, partialTicks);
		state.ageInTicks = entity.time + partialTicks;
		state.showsBottom = entity.showsBottom();
		BlockPos beamTarget = entity.getBeamTarget();
		if (beamTarget != null) {
			state.beamOffset = Vec3.atCenterOf(beamTarget).subtract(entity.getPosition(partialTicks));
		} else {
			state.beamOffset = null;
		}
	}

	public boolean shouldRender(final EndCrystal entity, final Frustum culler, final double camX, final double camY, final double camZ) {
		return super.shouldRender(entity, culler, camX, camY, camZ) || entity.getBeamTarget() != null;
	}
}