package net.minecraft.client.particle;

import net.minecraft.client.Camera;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.client.renderer.state.level.QuadParticleRenderState;
import net.minecraft.client.renderer.texture.TextureAtlasSprite;
import net.minecraft.core.particles.SimpleParticleType;
import net.minecraft.util.LightCoordsUtil;
import net.minecraft.util.RandomSource;

public class FlyTowardsPositionParticle extends SingleQuadParticle {
	private final double xStart;
	private final double yStart;
	private final double zStart;
	private final boolean isGlowing;
	private final Particle.LifetimeAlpha lifetimeAlpha;

	private FlyTowardsPositionParticle(
		final ClientLevel level, final double x, final double y, final double z, final double xd, final double yd, final double zd, final TextureAtlasSprite sprite
	) {
		this(level, x, y, z, xd, yd, zd, false, Particle.LifetimeAlpha.ALWAYS_OPAQUE, sprite);
	}

	private FlyTowardsPositionParticle(
		final ClientLevel level,
		final double x,
		final double y,
		final double z,
		final double xd,
		final double yd,
		final double zd,
		final boolean isGlowing,
		final Particle.LifetimeAlpha lifetimeAlpha,
		final TextureAtlasSprite sprite
	) {
		super(level, x, y, z, sprite);
		this.isGlowing = isGlowing;
		this.lifetimeAlpha = lifetimeAlpha;
		this.setAlpha(lifetimeAlpha.startAlpha());
		this.xd = xd;
		this.yd = yd;
		this.zd = zd;
		this.xStart = x;
		this.yStart = y;
		this.zStart = z;
		this.xo = x + xd;
		this.yo = y + yd;
		this.zo = z + zd;
		this.x = this.xo;
		this.y = this.yo;
		this.z = this.zo;
		this.quadSize = 0.1F * (this.random.nextFloat() * 0.5F + 0.2F);
		float br = this.random.nextFloat() * 0.6F + 0.4F;
		this.rCol = 0.9F * br;
		this.gCol = 0.9F * br;
		this.bCol = br;
		this.hasPhysics = false;
		this.lifetime = (int)(this.random.nextFloat() * 10.0F) + 30;
	}

	@Override
	public SingleQuadParticle.Layer getLayer() {
		return this.lifetimeAlpha.isOpaque() ? SingleQuadParticle.Layer.OPAQUE : SingleQuadParticle.Layer.TRANSLUCENT;
	}

	@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 int getLightCoords(final float a) {
		if (this.isGlowing) {
			return LightCoordsUtil.withBlock(super.getLightCoords(a), 15);
		} else {
			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;
			pos = 1.0F - pos;
			float pp = 1.0F - pos;
			pp *= pp;
			pp *= pp;
			this.x = this.xStart + this.xd * pos;
			this.y = this.yStart + this.yd * pos - pp * 1.2F;
			this.z = this.zStart + this.zd * pos;
		}
	}

	@Override
	public void extract(final QuadParticleRenderState particleTypeRenderState, final Camera camera, final float partialTickTime) {
		this.setAlpha(this.lifetimeAlpha.currentAlphaForAge(this.age, this.lifetime, partialTickTime));
		super.extract(particleTypeRenderState, camera, partialTickTime);
	}

	public static class EnchantProvider implements ParticleProvider<SimpleParticleType> {
		private final SpriteSet sprite;

		public EnchantProvider(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 FlyTowardsPositionParticle(level, x, y, z, xAux, yAux, zAux, this.sprite.get(random));
		}
	}

	public static class NautilusProvider implements ParticleProvider<SimpleParticleType> {
		private final SpriteSet sprite;

		public NautilusProvider(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 FlyTowardsPositionParticle(level, x, y, z, xAux, yAux, zAux, this.sprite.get(random));
		}
	}

	public static class VaultConnectionProvider implements ParticleProvider<SimpleParticleType> {
		private final SpriteSet sprite;

		public VaultConnectionProvider(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
		) {
			FlyTowardsPositionParticle particle = new FlyTowardsPositionParticle(
				level, x, y, z, xAux, yAux, zAux, true, new Particle.LifetimeAlpha(0.0F, 0.6F, 0.25F, 1.0F), this.sprite.get(random)
			);
			particle.scale(1.5F);
			return particle;
		}
	}
}