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.HangingSignRenderState;
import net.minecraft.client.renderer.blockentity.state.SignRenderState;
import net.minecraft.client.renderer.feature.ModelFeatureRenderer.CrumblingOverlay;
import net.minecraft.core.Direction;
import net.minecraft.world.level.block.CeilingHangingSignBlock;
import net.minecraft.world.level.block.HangingSignBlock;
import net.minecraft.world.level.block.WallHangingSignBlock;
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 HangingSignRenderer extends AbstractSignRenderer<HangingSignRenderState> {
	private static final float TEXT_RENDER_SCALE = 0.9F;
	private static final Vector3fc TEXT_OFFSET = new Vector3f(0.0F, -0.32F, 0.073F);
	public static final WallAndGroundTransformations<SignRenderState.SignTransformations> TRANSFORMATIONS = new WallAndGroundTransformations<>(
		HangingSignRenderer::createWallTransformation, HangingSignRenderer::createGroundTransformation, 16
	);

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

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

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

	private static Transformation textTransformation(final float angle, final boolean isFrontText) {
		Matrix4f result = new Matrix4f().translation(0.5F, 0.9375F, 0.5F).rotate(Axis.YP.rotationDegrees(-angle)).translate(0.0F, -0.3125F, 0.0F);
		if (!isFrontText) {
			result.rotate(Axis.YP.rotationDegrees(180.0F));
		}

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

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

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

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