skippyall
79 lines
3.2 KiB
Java
79 lines
3.2 KiB
Java
package io.github.skippyall.minions.command;
|
|
|
|
import com.mojang.brigadier.builder.LiteralArgumentBuilder;
|
|
import com.mojang.brigadier.context.CommandContext;
|
|
import io.github.skippyall.minions.Minions;
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
import net.minecraft.commands.arguments.coordinates.BlockPosArgument;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.block.Blocks;
|
|
|
|
import java.util.Collection;
|
|
import java.util.HashSet;
|
|
|
|
import static net.minecraft.commands.Commands.argument;
|
|
import static net.minecraft.commands.Commands.literal;
|
|
|
|
public class TestSubcommand {
|
|
public static LiteralArgumentBuilder<CommandSourceStack> TEST = literal("test")
|
|
.then(argument("pos", BlockPosArgument.blockPos())
|
|
.executes(TestSubcommand::execute)
|
|
);
|
|
|
|
|
|
private static int execute(CommandContext<CommandSourceStack> context) {
|
|
try {
|
|
BlockPos pos = BlockPosArgument.getBlockPos(context, "pos");
|
|
Collection<BlockPos> result = findInputs(context.getSource().getLevel(), pos);
|
|
for (BlockPos resultPos : result) {
|
|
context.getSource().sendSuccess(() -> Component.literal(resultPos.toShortString()), false);
|
|
}
|
|
} catch (Throwable e) {
|
|
Minions.LOGGER.error("Error", e);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private static Collection<BlockPos> findInputs(Level world, BlockPos pos) {
|
|
//positions that are already processed
|
|
Collection<BlockPos> visitedPositions = new HashSet<>();
|
|
//positions we are currently looking at
|
|
Collection<BlockPos> currentPositions = new HashSet<>();
|
|
//new positions are added here
|
|
Collection<BlockPos> newPositions = new HashSet<>();
|
|
//resulting machines
|
|
Collection<BlockPos> machines = new HashSet<>();
|
|
|
|
currentPositions.add(pos);
|
|
|
|
while(!currentPositions.isEmpty()) {
|
|
for(BlockPos currentPosition : currentPositions) {
|
|
for(Direction dir : Direction.values()) {
|
|
//check each neighbor of the current positions
|
|
BlockPos newPos = currentPosition.relative(dir);
|
|
//Do not check blocks that were already checked
|
|
if(!visitedPositions.contains(newPos)) {
|
|
if (world.getBlockState(newPos).getBlock() == Blocks.REDSTONE_BLOCK) {
|
|
//Add pipes to positions for the next iteration
|
|
newPositions.add(newPos);
|
|
} else if (world.getBlockState(newPos).getBlock() == Blocks.IRON_BLOCK) {
|
|
//Add machines to output set
|
|
machines.add(newPos);
|
|
}
|
|
//Add checked blocks here so that they are not checked again
|
|
visitedPositions.add(newPos);
|
|
}
|
|
}
|
|
}
|
|
//Check the new positions in the next iteration
|
|
currentPositions = newPositions;
|
|
newPositions = new HashSet<>();
|
|
}
|
|
|
|
return machines;
|
|
}
|
|
}
|