package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.minecraft.core.BlockPos;
import net.minecraft.core.particles.ColorParticleOption;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.util.ExtraCodecs;
import net.minecraft.util.ParticleUtils;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.state.BlockBehaviour;

public class TintedParticleLeavesBlock extends LeavesBlock {
	public static final MapCodec<TintedParticleLeavesBlock> CODEC = RecordCodecBuilder.mapCodec(
		i -> i.group(ExtraCodecs.floatRange(0.0F, 1.0F).fieldOf("leaf_particle_chance").forGetter(e -> e.leafParticleChance), propertiesCodec())
			.apply(i, TintedParticleLeavesBlock::new)
	);

	public TintedParticleLeavesBlock(final float leafParticleChance, final BlockBehaviour.Properties properties) {
		super(leafParticleChance, properties);
	}

	@Override
	protected void spawnFallingLeavesParticle(final Level level, final BlockPos pos, final RandomSource random) {
		ColorParticleOption particle = ColorParticleOption.create(ParticleTypes.TINTED_LEAVES, level.getClientLeafTintColor(pos));
		ParticleUtils.spawnParticleBelow(level, pos, random, particle);
	}

	@Override
	public MapCodec<? extends TintedParticleLeavesBlock> codec() {
		return CODEC;
	}
}