package net.minecraft.client.renderer.blockentity;

import com.mojang.math.Axis;
import com.mojang.math.Transformation;
import net.minecraft.client.renderer.blockentity.BlockEntityRendererProvider.Context;
import net.minecraft.client.renderer.blockentity.state.SignRenderState;
import net.minecraft.client.renderer.blockentity.state.StandingSignRenderState;
import net.minecraft.client.renderer.feature.ModelFeatureRenderer.CrumblingOverlay;
import net.minecraft.core.Direction;
import net.minecraft.world.level.block.PlainSignBlock;
import net.minecraft.world.level.block.StandingSignBlock;
import net.minecraft.world.level.block.WallSignBlock;
import net.minecraft.world.level.block.PlainSignBlock.Attachment;
import net.minecraft.world.level.block.entity.SignBlockEntity;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.properties.RotationSegment;
import net.minecraft.world.phys.Vec3;
import org.joml.Matrix4f;
import org.joml.Vector3f;
import org.joml.Vector3fc;
import org.jspecify.annotations.Nullable;

public class StandingSignRenderer extends AbstractSignRenderer<StandingSignRenderState> {
	private static final float RENDER_SCALE = 0.6666667F;
	private static final Vector3fc TEXT_OFFSET = new Vector3f(0.0F, 0.33333334F, 0.046666667F);
	public static final WallAndGroundTransformations<SignRenderState.SignTransformations> TRANSFORMATIONS = new WallAndGroundTransformations<>(
		StandingSignRenderer::createWallTransformation, StandingSignRenderer::createGroundTransformation, 16
	);

	public StandingSignRenderer(final Context context) {
		super(context);
	}

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

	public void extractRenderState(
		final SignBlockEntity blockEntity,
		final StandingSignRenderState state,
		final float partialTicks,
		final Vec3 cameraPosition,
		@Nullable final CrumblingOverlay breakProgress
	) {
		super.extractRenderState(blockEntity, state, partialTicks, cameraPosition, breakProgress);
		BlockState blockState = blockEntity.getBlockState();
		state.attachmentType = PlainSignBlock.getAttachmentPoint(blockState);
		if (blockState.getBlock() instanceof WallSignBlock) {
			state.transformations = TRANSFORMATIONS.wallTransformation((Direction)blockState.getValue(WallSignBlock.FACING));
		} else {
			state.transformations = TRANSFORMATIONS.freeTransformations((Integer)blockState.getValue(StandingSignBlock.ROTATION));
		}
	}

	private static Transformation textTransformation(final Attachment attachmentType, final float angle, final boolean isFrontText) {
		Matrix4f result = new Matrix4f().translate(0.5F, 0.5F, 0.5F).rotate(Axis.YP.rotationDegrees(-angle));
		if (attachmentType == Attachment.WALL) {
			result.translate(0.0F, -0.3125F, -0.4375F);
		}

		if (!isFrontText) {
			result.rotate(Axis.YP.rotationDegrees(180.0F));
		}

		float s = 0.010416667F;
		return new Transformation(result.translate(TEXT_OFFSET).scale(0.010416667F, -0.010416667F, 0.010416667F));
	}

	private static SignRenderState.SignTransformations createTransformations(final Attachment attachmentType, final float angle) {
		return new SignRenderState.SignTransformations(textTransformation(attachmentType, angle, true), textTransformation(attachmentType, angle, false));
	}

	private static SignRenderState.SignTransformations createGroundTransformation(final int segment) {
		return createTransformations(Attachment.GROUND, RotationSegment.convertToDegrees(segment));
	}

	private static SignRenderState.SignTransformations createWallTransformation(final Direction direction) {
		return createTransformations(Attachment.WALL, direction.toYRot());
	}
}