package net.minecraft.client.gui.render.pip;

import com.mojang.blaze3d.platform.Lighting;
import com.mojang.blaze3d.vertex.PoseStack;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.renderer.SubmitNodeCollector;
import net.minecraft.client.renderer.entity.EntityRenderDispatcher;
import net.minecraft.client.renderer.state.gui.pip.GuiEntityRenderState;
import net.minecraft.client.renderer.state.level.CameraRenderState;
import org.joml.Quaternionf;
import org.joml.Quaternionfc;
import org.joml.Vector3fc;

@Environment(EnvType.CLIENT)
public class GuiEntityRenderer extends PictureInPictureRenderer<GuiEntityRenderState> {
	private final EntityRenderDispatcher entityRenderDispatcher;

	public GuiEntityRenderer(final EntityRenderDispatcher entityRenderDispatcher) {
		this.entityRenderDispatcher = entityRenderDispatcher;
	}

	@Override
	public Class<GuiEntityRenderState> getRenderStateClass() {
		return GuiEntityRenderState.class;
	}

	protected void renderToTexture(final GuiEntityRenderState entityState, final PoseStack poseStack, final SubmitNodeCollector submitNodeCollector) {
		Minecraft.getInstance().gameRenderer.lighting().setupFor(Lighting.Entry.ENTITY_IN_UI);
		Vector3fc translation = entityState.translation();
		poseStack.translate(translation.x(), translation.y(), translation.z());
		poseStack.mulPose(entityState.rotation());
		Quaternionfc overriddenCameraAngle = entityState.overrideCameraAngle();
		CameraRenderState cameraRenderState = new CameraRenderState();
		if (overriddenCameraAngle != null) {
			cameraRenderState.orientation = overriddenCameraAngle.conjugate(new Quaternionf()).rotateY((float) Math.PI);
		}

		this.entityRenderDispatcher.submit(entityState.renderState(), cameraRenderState, 0.0, 0.0, 0.0, poseStack, submitNodeCollector);
	}

	@Override
	protected float getTranslateY(final int height, final int guiScale) {
		return height / 2.0F;
	}

	@Override
	protected String getTextureLabel() {
		return "entity";
	}
}