package net.minecraft.client.renderer.texture;

import com.mojang.blaze3d.textures.GpuTexture;
import com.mojang.blaze3d.vertex.VertexConsumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.SpriteCoordinateExpander;
import net.minecraft.resources.ResourceLocation;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class TextureAtlasSprite {
	private final ResourceLocation atlasLocation;
	private final SpriteContents contents;
	final int x;
	final int y;
	private final float u0;
	private final float u1;
	private final float v0;
	private final float v1;

	protected TextureAtlasSprite(ResourceLocation atlasLocation, SpriteContents contents, int originX, int originY, int x, int y) {
		this.atlasLocation = atlasLocation;
		this.contents = contents;
		this.x = x;
		this.y = y;
		this.u0 = (float)x / originX;
		this.u1 = (float)(x + contents.width()) / originX;
		this.v0 = (float)y / originY;
		this.v1 = (float)(y + contents.height()) / originY;
	}

	public int getX() {
		return this.x;
	}

	public int getY() {
		return this.y;
	}

	/**
	 * @return the minimum U coordinate to use when rendering this sprite
	 */
	public float getU0() {
		return this.u0;
	}

	/**
	 * @return the maximum U coordinate to use when rendering this sprite
	 */
	public float getU1() {
		return this.u1;
	}

	public SpriteContents contents() {
		return this.contents;
	}

	@Nullable
	public TextureAtlasSprite.Ticker createTicker() {
		final SpriteTicker spriteTicker = this.contents.createTicker();
		return spriteTicker != null ? new TextureAtlasSprite.Ticker() {
			@Override
			public void tickAndUpload(GpuTexture texture) {
				spriteTicker.tickAndUpload(TextureAtlasSprite.this.x, TextureAtlasSprite.this.y, texture);
			}

			@Override
			public void close() {
				spriteTicker.close();
			}
		} : null;
	}

	public float getU(float u) {
		float f = this.u1 - this.u0;
		return this.u0 + f * u;
	}

	public float getUOffset(float offset) {
		float f = this.u1 - this.u0;
		return (offset - this.u0) / f;
	}

	/**
	 * @return the minimum V coordinate to use when rendering this sprite
	 */
	public float getV0() {
		return this.v0;
	}

	/**
	 * @return the maximum V coordinate to use when rendering this sprite
	 */
	public float getV1() {
		return this.v1;
	}

	public float getV(float v) {
		float f = this.v1 - this.v0;
		return this.v0 + f * v;
	}

	public float getVOffset(float offset) {
		float f = this.v1 - this.v0;
		return (offset - this.v0) / f;
	}

	public ResourceLocation atlasLocation() {
		return this.atlasLocation;
	}

	public String toString() {
		return "TextureAtlasSprite{contents='" + this.contents + "', u0=" + this.u0 + ", u1=" + this.u1 + ", v0=" + this.v0 + ", v1=" + this.v1 + "}";
	}

	public void uploadFirstFrame(GpuTexture texture) {
		this.contents.uploadFirstFrame(this.x, this.y, texture);
	}

	private float atlasSize() {
		float f = this.contents.width() / (this.u1 - this.u0);
		float g = this.contents.height() / (this.v1 - this.v0);
		return Math.max(g, f);
	}

	public float uvShrinkRatio() {
		return 4.0F / this.atlasSize();
	}

	public VertexConsumer wrap(VertexConsumer consumer) {
		return new SpriteCoordinateExpander(consumer, this);
	}

	@Environment(EnvType.CLIENT)
	public interface Ticker extends AutoCloseable {
		void tickAndUpload(GpuTexture texture);

		void close();
	}
}