package net.minecraft.world.level.redstone;

import com.google.common.collect.Sets;
import java.util.Set;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.RedStoneWireBlock;
import net.minecraft.world.level.block.state.BlockState;
import org.jspecify.annotations.Nullable;

public class DefaultRedstoneWireEvaluator extends RedstoneWireEvaluator {
	public DefaultRedstoneWireEvaluator(final RedStoneWireBlock wireBlock) {
		super(wireBlock);
	}

	@Override
	public void updatePowerStrength(
		final Level level, final BlockPos pos, final BlockState state, @Nullable final Orientation orientation, final boolean skipShapeUpdates
	) {
		int targetStrength = this.calculateTargetStrength(level, pos);
		if ((Integer)state.getValue(RedStoneWireBlock.POWER) != targetStrength) {
			if (level.getBlockState(pos) == state) {
				level.setBlock(pos, state.setValue(RedStoneWireBlock.POWER, targetStrength), 2);
			}

			Set<BlockPos> toUpdate = Sets.<BlockPos>newHashSet();
			toUpdate.add(pos);

			for (Direction direction : Direction.values()) {
				toUpdate.add(pos.relative(direction));
			}

			for (BlockPos blockPos : toUpdate) {
				level.updateNeighborsAt(blockPos, this.wireBlock);
			}
		}
	}

	private int calculateTargetStrength(final Level level, final BlockPos pos) {
		int blockSignal = this.getBlockSignal(level, pos);
		return blockSignal == 15 ? blockSignal : Math.max(blockSignal, this.getIncomingWireSignal(level, pos));
	}
}