package net.minecraft.client.particle;

import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.core.particles.GeyserBaseParticleOptions;
import net.minecraft.core.particles.GeyserParticleOptions;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.util.RandomSource;
import org.jspecify.annotations.Nullable;

public class GeyserEruptionParticle extends NoRenderParticle {
	public static final int BASE_PARTICLE_FREQUENCY = 2;
	public static final int BASE_PARTICLE_AMOUNT = 2;
	public static final int POOF_PARTICLE_FREQUENCY = 10;
	public static final int POOF_PARTICLE_AMOUNT = 20;
	private static final float BASE_BURST_IMPULSE = 1.5F;
	private static final float POOF_BURST_IMPULSE = 2.0F;
	private final int waterBlocks;
	private final double xa;
	private final double ya;
	private final double za;
	private final GeyserParticleOptions plumeParticle;
	private final GeyserBaseParticleOptions baseParticle;
	private final GeyserBaseParticleOptions poofParticle;

	protected GeyserEruptionParticle(
		final ClientLevel level,
		final double x,
		final double y,
		final double z,
		final double xAux,
		final double yAux,
		final double zAux,
		final GeyserParticleOptions options
	) {
		super(level, x, y, z);
		this.xa = xAux;
		this.ya = yAux;
		this.za = zAux;
		this.waterBlocks = options.waterBlocks();
		this.lifetime = 20;
		this.plumeParticle = new GeyserParticleOptions(ParticleTypes.GEYSER_PLUME, this.waterBlocks);
		this.baseParticle = new GeyserBaseParticleOptions(ParticleTypes.GEYSER_BASE, this.waterBlocks, 1.5F);
		this.poofParticle = new GeyserBaseParticleOptions(ParticleTypes.GEYSER_POOF, this.waterBlocks, 2.0F);
	}

	@Override
	public void tick() {
		super.tick();
		if (this.age % 2 == 0) {
			for (int i = 0; i < 2; i++) {
				this.level.addParticle(this.baseParticle, this.x, this.y, this.z, this.xa, this.ya, this.za);
			}
		}

		for (int i = 0; i < this.waterBlocks + 2; i++) {
			this.level.addParticle(this.plumeParticle, this.x, this.y, this.z, this.xa, this.ya, this.za);
		}

		if (this.age % 10 == 0) {
			for (int i = 0; i < 20; i++) {
				this.level.addParticle(this.poofParticle, this.x, this.y, this.z, this.xa, this.ya, this.za);
			}
		}
	}

	public static class Provider implements ParticleProvider<GeyserParticleOptions> {
		@Nullable
		public Particle createParticle(
			final GeyserParticleOptions 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 GeyserEruptionParticle(level, x, y, z, xAux, yAux, zAux, options);
		}
	}
}