package net.minecraft.world.level.block;

import com.mojang.math.OctahedralGroup;
import com.mojang.serialization.Codec;
import io.netty.buffer.ByteBuf;
import java.util.List;
import java.util.function.IntFunction;
import net.minecraft.Util;
import net.minecraft.core.Direction;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.ByIdMap;
import net.minecraft.util.ExtraCodecs;
import net.minecraft.util.RandomSource;
import net.minecraft.util.StringRepresentable;
import net.minecraft.util.ByIdMap.OutOfBoundsStrategy;

public enum Rotation implements StringRepresentable {
	NONE(0, "none", OctahedralGroup.IDENTITY),
	CLOCKWISE_90(1, "clockwise_90", OctahedralGroup.ROT_90_Y_NEG),
	CLOCKWISE_180(2, "180", OctahedralGroup.ROT_180_FACE_XZ),
	COUNTERCLOCKWISE_90(3, "counterclockwise_90", OctahedralGroup.ROT_90_Y_POS);

	public static final IntFunction<Rotation> BY_ID = ByIdMap.continuous(Rotation::getIndex, values(), OutOfBoundsStrategy.WRAP);
	public static final Codec<Rotation> CODEC = StringRepresentable.fromEnum(Rotation::values);
	public static final StreamCodec<ByteBuf, Rotation> STREAM_CODEC = ByteBufCodecs.idMapper(BY_ID, Rotation::getIndex);
	@Deprecated
	public static final Codec<Rotation> LEGACY_CODEC = ExtraCodecs.legacyEnum(Rotation::valueOf);
	private final int index;
	private final String id;
	private final OctahedralGroup rotation;

	private Rotation(final int index, final String id, final OctahedralGroup rotation) {
		this.index = index;
		this.id = id;
		this.rotation = rotation;
	}

	public Rotation getRotated(Rotation rotation) {
		return switch (rotation) {
			case CLOCKWISE_90 -> {
				switch (this) {
					case NONE:
						yield CLOCKWISE_90;
					case CLOCKWISE_90:
						yield CLOCKWISE_180;
					case CLOCKWISE_180:
						yield COUNTERCLOCKWISE_90;
					case COUNTERCLOCKWISE_90:
						yield NONE;
					default:
						throw new MatchException(null, null);
				}
			}
			case CLOCKWISE_180 -> {
				switch (this) {
					case NONE:
						yield CLOCKWISE_180;
					case CLOCKWISE_90:
						yield COUNTERCLOCKWISE_90;
					case CLOCKWISE_180:
						yield NONE;
					case COUNTERCLOCKWISE_90:
						yield CLOCKWISE_90;
					default:
						throw new MatchException(null, null);
				}
			}
			case COUNTERCLOCKWISE_90 -> {
				switch (this) {
					case NONE:
						yield COUNTERCLOCKWISE_90;
					case CLOCKWISE_90:
						yield NONE;
					case CLOCKWISE_180:
						yield CLOCKWISE_90;
					case COUNTERCLOCKWISE_90:
						yield CLOCKWISE_180;
					default:
						throw new MatchException(null, null);
				}
			}
			default -> this;
		};
	}

	public OctahedralGroup rotation() {
		return this.rotation;
	}

	public Direction rotate(Direction facing) {
		if (facing.getAxis() == Direction.Axis.Y) {
			return facing;
		} else {
			return switch (this) {
				case CLOCKWISE_90 -> facing.getClockWise();
				case CLOCKWISE_180 -> facing.getOpposite();
				case COUNTERCLOCKWISE_90 -> facing.getCounterClockWise();
				default -> facing;
			};
		}
	}

	public int rotate(int rotation, int positionCount) {
		return switch (this) {
			case CLOCKWISE_90 -> (rotation + positionCount / 4) % positionCount;
			case CLOCKWISE_180 -> (rotation + positionCount / 2) % positionCount;
			case COUNTERCLOCKWISE_90 -> (rotation + positionCount * 3 / 4) % positionCount;
			default -> rotation;
		};
	}

	/**
	 * Chooses a random rotation.
	 */
	public static Rotation getRandom(RandomSource random) {
		return Util.getRandom(values(), random);
	}

	/**
	 * Get a list of all rotations in random order.
	 */
	public static List<Rotation> getShuffled(RandomSource random) {
		return Util.shuffledCopy(values(), random);
	}

	@Override
	public String getSerializedName() {
		return this.id;
	}

	private int getIndex() {
		return this.index;
	}
}