package net.minecraft.client.particle;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.core.particles.SimpleParticleType;
import net.minecraft.util.Mth;

@Environment(EnvType.CLIENT)
public class DragonBreathParticle extends TextureSheetParticle {
	private static final int COLOR_MIN = 11993298;
	private static final int COLOR_MAX = 14614777;
	private static final float COLOR_MIN_RED = 0.7176471F;
	private static final float COLOR_MIN_GREEN = 0.0F;
	private static final float COLOR_MIN_BLUE = 0.8235294F;
	private static final float COLOR_MAX_RED = 0.8745098F;
	private static final float COLOR_MAX_GREEN = 0.0F;
	private static final float COLOR_MAX_BLUE = 0.9764706F;
	private boolean hasHitGround;
	private final SpriteSet sprites;

	DragonBreathParticle(ClientLevel level, double x, double y, double z, double xSpeed, double ySpeed, double zSpeed, SpriteSet sprites) {
		super(level, x, y, z);
		this.friction = 0.96F;
		this.xd = xSpeed;
		this.yd = ySpeed;
		this.zd = zSpeed;
		this.rCol = Mth.nextFloat(this.random, 0.7176471F, 0.8745098F);
		this.gCol = Mth.nextFloat(this.random, 0.0F, 0.0F);
		this.bCol = Mth.nextFloat(this.random, 0.8235294F, 0.9764706F);
		this.quadSize *= 0.75F;
		this.lifetime = (int)(20.0 / (this.random.nextFloat() * 0.8 + 0.2));
		this.hasHitGround = false;
		this.hasPhysics = false;
		this.sprites = sprites;
		this.setSpriteFromAge(sprites);
	}

	@Override
	public void tick() {
		this.xo = this.x;
		this.yo = this.y;
		this.zo = this.z;
		if (this.age++ >= this.lifetime) {
			this.remove();
		} else {
			this.setSpriteFromAge(this.sprites);
			if (this.onGround) {
				this.yd = 0.0;
				this.hasHitGround = true;
			}

			if (this.hasHitGround) {
				this.yd += 0.002;
			}

			this.move(this.xd, this.yd, this.zd);
			if (this.y == this.yo) {
				this.xd *= 1.1;
				this.zd *= 1.1;
			}

			this.xd = this.xd * this.friction;
			this.zd = this.zd * this.friction;
			if (this.hasHitGround) {
				this.yd = this.yd * this.friction;
			}
		}
	}

	@Override
	public ParticleRenderType getRenderType() {
		return ParticleRenderType.PARTICLE_SHEET_OPAQUE;
	}

	@Override
	public float getQuadSize(float scaleFactor) {
		return this.quadSize * Mth.clamp((this.age + scaleFactor) / this.lifetime * 32.0F, 0.0F, 1.0F);
	}

	@Environment(EnvType.CLIENT)
	public static class Provider implements ParticleProvider<SimpleParticleType> {
		private final SpriteSet sprites;

		public Provider(SpriteSet sprites) {
			this.sprites = sprites;
		}

		public Particle createParticle(SimpleParticleType type, ClientLevel level, double x, double y, double z, double xSpeed, double ySpeed, double zSpeed) {
			return new DragonBreathParticle(level, x, y, z, xSpeed, ySpeed, zSpeed, this.sprites);
		}
	}
}