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;
import net.minecraft.core.particles.SimpleParticleType;
import net.minecraft.util.LightCoordsUtil;
import net.minecraft.util.RandomSource;

@Environment(EnvType.CLIENT)
public class PortalParticle extends SingleQuadParticle {
	private final double xStart;
	private final double yStart;
	private final double zStart;

	protected PortalParticle(
		final ClientLevel level, final double x, final double y, final double z, final double xd, final double yd, final double zd, final TextureAtlasSprite sprite
	) {
		super(level, x, y, z, sprite);
		this.xd = xd;
		this.yd = yd;
		this.zd = zd;
		this.x = x;
		this.y = y;
		this.z = z;
		this.xStart = this.x;
		this.yStart = this.y;
		this.zStart = this.z;
		this.quadSize = 0.1F * (this.random.nextFloat() * 0.2F + 0.5F);
		float br = this.random.nextFloat() * 0.6F + 0.4F;
		this.rCol = br * 0.9F;
		this.gCol = br * 0.3F;
		this.bCol = br;
		this.lifetime = (int)(this.random.nextFloat() * 10.0F) + 40;
	}

	@Override
	public SingleQuadParticle.Layer getLayer() {
		return SingleQuadParticle.Layer.OPAQUE;
	}

	@Override
	public void move(final double xa, final double ya, final double za) {
		this.setBoundingBox(this.getBoundingBox().move(xa, ya, za));
		this.setLocationFromBoundingbox();
	}

	@Override
	public float getQuadSize(final float a) {
		float s = (this.age + a) / this.lifetime;
		s = 1.0F - s;
		s *= s;
		s = 1.0F - s;
		return this.quadSize * s;
	}

	@Override
	public int getLightCoords(final float a) {
		float brightness = (float)this.age / this.lifetime;
		brightness *= brightness;
		brightness *= brightness;
		return LightCoordsUtil.addSmoothBlockEmission(super.getLightCoords(a), brightness);
	}

	@Override
	public void tick() {
		this.xo = this.x;
		this.yo = this.y;
		this.zo = this.z;
		if (this.age++ >= this.lifetime) {
			this.remove();
		} else {
			float pos = (float)this.age / this.lifetime;
			float var3 = -pos + pos * pos * 2.0F;
			float var4 = 1.0F - var3;
			this.x = this.xStart + this.xd * var4;
			this.y = this.yStart + this.yd * var4 + (1.0F - pos);
			this.z = this.zStart + this.zd * var4;
		}
	}

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

		public Provider(final SpriteSet sprite) {
			this.sprite = sprite;
		}

		public Particle createParticle(
			final SimpleParticleType options,
			final ClientLevel level,
			final double x,
			final double y,
			final double z,
			final double xAux,
			final double yAux,
			final double zAux,
			final RandomSource random
		) {
			return new PortalParticle(level, x, y, z, xAux, yAux, zAux, this.sprite.get(random));
		}
	}
}