package net.minecraft.core;

import io.netty.buffer.ByteBuf;
import net.minecraft.nbt.FloatTag;
import net.minecraft.nbt.ListTag;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.Mth;

public class Rotations {
	public static final StreamCodec<ByteBuf, Rotations> STREAM_CODEC = new StreamCodec<ByteBuf, Rotations>() {
		public Rotations decode(ByteBuf byteBuf) {
			return new Rotations(byteBuf.readFloat(), byteBuf.readFloat(), byteBuf.readFloat());
		}

		public void encode(ByteBuf byteBuf, Rotations rotations) {
			byteBuf.writeFloat(rotations.x);
			byteBuf.writeFloat(rotations.y);
			byteBuf.writeFloat(rotations.z);
		}
	};
	/**
	 * Rotation on the X axis
	 */
	protected final float x;
	/**
	 * Rotation on the Y axis
	 */
	protected final float y;
	/**
	 * Rotation on the Z axis
	 */
	protected final float z;

	public Rotations(float x, float y, float z) {
		this.x = !Float.isInfinite(x) && !Float.isNaN(x) ? x % 360.0F : 0.0F;
		this.y = !Float.isInfinite(y) && !Float.isNaN(y) ? y % 360.0F : 0.0F;
		this.z = !Float.isInfinite(z) && !Float.isNaN(z) ? z % 360.0F : 0.0F;
	}

	public Rotations(ListTag tag) {
		this(tag.getFloat(0), tag.getFloat(1), tag.getFloat(2));
	}

	public ListTag save() {
		ListTag listTag = new ListTag();
		listTag.add(FloatTag.valueOf(this.x));
		listTag.add(FloatTag.valueOf(this.y));
		listTag.add(FloatTag.valueOf(this.z));
		return listTag;
	}

	public boolean equals(Object object) {
		return !(object instanceof Rotations rotations) ? false : this.x == rotations.x && this.y == rotations.y && this.z == rotations.z;
	}

	/**
	 * @return the X axis rotation
	 */
	public float getX() {
		return this.x;
	}

	/**
	 * @return the Y axis rotation
	 */
	public float getY() {
		return this.y;
	}

	/**
	 * @return the Z axis rotation
	 */
	public float getZ() {
		return this.z;
	}

	public float getWrappedX() {
		return Mth.wrapDegrees(this.x);
	}

	public float getWrappedY() {
		return Mth.wrapDegrees(this.y);
	}

	public float getWrappedZ() {
		return Mth.wrapDegrees(this.z);
	}
}