package net.minecraft.client.particle;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.client.renderer.texture.TextureAtlasSprite;

@Environment(EnvType.CLIENT)
public abstract class TextureSheetParticle extends SingleQuadParticle {
	protected TextureAtlasSprite sprite;

	protected TextureSheetParticle(ClientLevel clientLevel, double d, double e, double f) {
		super(clientLevel, d, e, f);
	}

	protected TextureSheetParticle(ClientLevel clientLevel, double d, double e, double f, double g, double h, double i) {
		super(clientLevel, d, e, f, g, h, i);
	}

	protected void setSprite(TextureAtlasSprite sprite) {
		this.sprite = sprite;
	}

	@Override
	protected float getU0() {
		return this.sprite.getU0();
	}

	@Override
	protected float getU1() {
		return this.sprite.getU1();
	}

	@Override
	protected float getV0() {
		return this.sprite.getV0();
	}

	@Override
	protected float getV1() {
		return this.sprite.getV1();
	}

	public void pickSprite(SpriteSet sprite) {
		this.setSprite(sprite.get(this.random));
	}

	public void setSpriteFromAge(SpriteSet sprite) {
		if (!this.removed) {
			this.setSprite(sprite.get(this.age, this.lifetime));
		}
	}
}