Train Admin

- Clean up /dumpRailways
- Trains can now be removed via a command

src/main/java/com/simibubi/create/content/logistics/trains/GlobalRailwayManager.java

@@ -227,6 +227,7 @@ public class GlobalRailwayManager {
 			if (train.invalid) {
 				iterator.remove();
 				trains.remove(train.id);
+				AllPackets.channel.send(PacketDistributor.ALL.noArg(), new TrainPacket(train, false));
 				continue;
 			}
 
@@ -242,6 +243,7 @@ public class GlobalRailwayManager {
 			if (train.invalid) {
 				iterator.remove();
 				trains.remove(train.id);
+				AllPackets.channel.send(PacketDistributor.ALL.noArg(), new TrainPacket(train, false));
 				continue;
 			}
 
@@ -251,7 +253,6 @@ public class GlobalRailwayManager {
 			iterator.remove();
 		}
 
-		
 	}
 
 	public void tickSignalOverlay() {

src/main/java/com/simibubi/create/content/logistics/trains/TrackGraph.java

@@ -9,6 +9,7 @@ import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
+import java.util.Objects;
 import java.util.Random;
 import java.util.Set;
 import java.util.UUID;
@@ -465,6 +466,17 @@ public class TrackGraph {
 		return connections.put(node2, edge) == null;
 	}
 
+	public float distanceToLocationSqr(Level level, Vec3 location) {
+		float nearest = Float.MAX_VALUE;
+		for (TrackNodeLocation tnl : nodes.keySet()) {
+			if (!Objects.equals(tnl.dimension, level.dimension()))
+				continue;
+			nearest = Math.min(nearest, (float) tnl.getLocation()
+				.distanceToSqr(location));
+		}
+		return nearest;
+	}
+
 	public void deferIntersectionUpdate(TrackEdge edge) {
 		deferredIntersectionUpdates.add(edge);
 	}

src/main/java/com/simibubi/create/content/logistics/trains/entity/CarriageContraptionEntity.java

@@ -266,6 +266,10 @@ public class CarriageContraptionEntity extends OrientedContraptionEntity {
 
 		carriageData.approach(this, carriage, 1f / getType().updateInterval());
 
+		if (!Create.RAILWAYS.sided(null).trains.containsKey(carriage.train.id)) {
+			discard();
+			return;
+		}
 		if (!carriage.train.derailed)
 			carriage.updateContraptionAnchors();
 

src/main/java/com/simibubi/create/content/logistics/trains/entity/Train.java

@@ -1204,4 +1204,15 @@ public class Train {
 		}
 	}
 
+	public float distanceToLocationSqr(Level level, Vec3 location) {
+		float distance = Float.MAX_VALUE;
+		for (Carriage carriage : carriages) {
+			DimensionalCarriageEntity dce = carriage.getDimensionalIfPresent(level.dimension());
+			if (dce == null || dce.positionAnchor == null)
+				continue;
+			distance = Math.min(distance, (float) dce.positionAnchor.distanceToSqr(location));
+		}
+		return distance;
+	}
+
 }

src/main/java/com/simibubi/create/foundation/command/AllCommands.java

@@ -29,6 +29,7 @@ public class AllCommands {
 				.then(DumpRailwaysCommand.register())
 				.then(FixLightingCommand.register())
 				.then(HighlightCommand.register())
+				.then(KillTrainCommand.register())
 				.then(CouplingCommand.register())
 				.then(ConfigCommand.register())
 				.then(PonderCommand.register())

src/main/java/com/simibubi/create/foundation/command/DumpRailwaysCommand.java

@@ -1,9 +1,9 @@
 package com.simibubi.create.foundation.command;
 
 import java.util.Collection;
-import java.util.Map.Entry;
-import java.util.UUID;
+import java.util.List;
 import java.util.function.BiConsumer;
+import java.util.function.Consumer;
 
 import com.mojang.brigadier.builder.ArgumentBuilder;
 import com.simibubi.create.Create;
@@ -14,32 +14,36 @@ import com.simibubi.create.content.logistics.trains.management.edgePoint.EdgePoi
 import com.simibubi.create.content.logistics.trains.management.edgePoint.signal.SignalBoundary;
 import com.simibubi.create.content.logistics.trains.management.edgePoint.station.GlobalStation;
 import com.simibubi.create.content.logistics.trains.management.schedule.ScheduleRuntime;
-import com.simibubi.create.content.logistics.trains.management.schedule.ScheduleRuntime.State;
 
 import net.minecraft.ChatFormatting;
 import net.minecraft.commands.CommandSourceStack;
 import net.minecraft.commands.Commands;
-import net.minecraft.core.BlockPos;
+import net.minecraft.network.chat.ClickEvent;
 import net.minecraft.network.chat.Component;
+import net.minecraft.network.chat.ComponentUtils;
+import net.minecraft.network.chat.HoverEvent;
 import net.minecraft.network.chat.TextComponent;
 import net.minecraft.server.level.ServerLevel;
 import net.minecraft.util.Mth;
 import net.minecraft.world.entity.LivingEntity;
+import net.minecraft.world.phys.Vec3;
 
 public class DumpRailwaysCommand {
 
 	static ArgumentBuilder<CommandSourceStack, ?> register() {
 		return Commands.literal("dumpRailways")
-			.requires(cs -> cs.hasPermission(1))
+			.requires(cs -> cs.hasPermission(2))
 			.executes(ctx -> {
 				CommandSourceStack source = ctx.getSource();
-				fillReport(source.getLevel(),
-					(s, f) -> source.sendSuccess(new TextComponent(s).withStyle(st -> st.withColor(f)), false));
+				fillReport(source.getLevel(), source.getPosition(),
+					(s, f) -> source.sendSuccess(new TextComponent(s).withStyle(st -> st.withColor(f)), false),
+					(c) -> source.sendSuccess(c, false));
 				return 1;
 			});
 	}
 
-	static void fillReport(ServerLevel level, BiConsumer<String, Integer> chat) {
+	static void fillReport(ServerLevel level, Vec3 location, BiConsumer<String, Integer> chat,
+		Consumer<Component> chatRaw) {
 		GlobalRailwayManager railways = Create.RAILWAYS;
 		int white = ChatFormatting.WHITE.getColor();
 		int blue = 0xD3DEDC;
@@ -47,81 +51,103 @@ public class DumpRailwaysCommand {
 		int bright = 0xFFEFEF;
 		int orange = 0xFFAD60;
 
-		for (int i = 0; i < 10; i++)
 		chat.accept("", white);
 		chat.accept("-+------<< Railways Summary: >>------+-", white);
-		chat.accept("Track Networks: " + railways.trackNetworks.size(), blue);
+		int graphCount = railways.trackNetworks.size();
+		chat.accept("Track Networks: " + graphCount, blue);
 		chat.accept("Signal Groups: " + railways.signalEdgeGroups.size(), blue);
-		chat.accept("Trains: " + railways.trains.size(), blue);
+		int trainCount = railways.trains.size();
+		chat.accept("Trains: " + trainCount, blue);
 		chat.accept("", white);
 
-		for (Entry<UUID, TrackGraph> entry : railways.trackNetworks.entrySet()) {
-			TrackGraph graph = entry.getValue();
-			UUID id = entry.getKey();
-			chat.accept(id.toString()
-				.substring(0, 5) + ": Track Graph, "
+		List<TrackGraph> nearest = railways.trackNetworks.values()
+			.stream()
+			.sorted((tg1, tg2) -> Float.compare(tg1.distanceToLocationSqr(level, location),
+				tg2.distanceToLocationSqr(level, location)))
+			.limit(5)
+			.toList();
+
+		if (graphCount > 0) {
+			chat.accept("Nearest Graphs: ", orange);
+			chat.accept("", white);
+			for (TrackGraph graph : nearest) {
+				chat.accept(graph.id.toString()
+					.substring(0, 5) + " with "
 					+ graph.getNodes()
 						.size()
-				+ " Nodes", graph.color.getRGB());
+					+ " Nodes", white);
 				Collection<SignalBoundary> signals = graph.getPoints(EdgePointType.SIGNAL);
 				if (!signals.isEmpty())
-				chat.accept(" -> " + signals.size() + " registered Signals", blue);
-			for (GlobalStation globalStation : graph.getPoints(EdgePointType.STATION)) {
-				BlockPos pos = globalStation.getTilePos();
-				if (pos == null)
-					pos = BlockPos.ZERO;
-				chat.accept(" -> " + globalStation.name + " (" + globalStation.id.toString()
-					.substring(0, 5) + ") [" + pos.getX() + "," + pos.getY() + "," + pos.getZ() + "]", darkBlue);
-				if (globalStation.getPresentTrain() != null) {
-					chat.accept("  > Currently Occupied by " + globalStation.getPresentTrain().name.getString(), blue);
-				} else {
-					Train imminentTrain = globalStation.getImminentTrain();
-					if (imminentTrain != null) {
-						chat.accept("  > Reserved by " + imminentTrain.name.getString() + " ("
-							+ Mth.floor(imminentTrain.navigation.distanceToDestination) + "m away)", blue);
-					}
-				}
+					chat.accept(" -> " + signals.size() + " Signals", blue);
+				Collection<GlobalStation> stations = graph.getPoints(EdgePointType.STATION);
+				if (!stations.isEmpty())
+					chat.accept(" -> " + stations.size() + " Stations", blue);
 			}
 			chat.accept("", white);
+			if (graphCount > 5) {
+				chat.accept("[...]", white);
+				chat.accept("", white);
+			}
 		}
 
-		for (Train train : railways.trains.values()) {
+		List<Train> nearestTrains = railways.trains.values()
+			.stream()
+			.sorted((t1, t2) -> Float.compare(t1.distanceToLocationSqr(level, location),
+				t2.distanceToLocationSqr(level, location)))
+			.limit(5)
+			.toList();
+
+		if (trainCount > 0 && !nearestTrains.isEmpty()) {
+			chat.accept("Nearest Trains: ", orange);
+			chat.accept("", white);
+			for (Train train : nearestTrains) {
 				chat.accept(train.id.toString()
-				.substring(0, 5) + ": " + train.name.getString() + ", " + train.carriages.size() + " Wagons", bright);
-			if (train.graph != null)
-				chat.accept(" -> On Track: " + train.graph.id.toString()
-					.substring(0, 5), train.graph.color.getRGB());
+					.substring(0, 5) + ": " + train.name.getString() + ", " + train.carriages.size() + " Wagons",
+					bright);
 				if (train.derailed)
 					chat.accept(" -> Derailed", orange);
+				else if (train.graph != null)
+					chat.accept(" -> On Track: " + train.graph.id.toString()
+						.substring(0, 5), blue);
 				LivingEntity owner = train.getOwner(level);
 				if (owner != null)
-				chat.accept("  > Owned by " + owner.getName()
+					chat.accept(" -> Owned by " + owner.getName()
 						.getString(), blue);
 				GlobalStation currentStation = train.getCurrentStation();
 				if (currentStation != null) {
-				chat.accept("  > Waiting at: " + currentStation.name, blue);
-			} else if (train.navigation.destination != null) {
-				chat.accept("  > Travelling to " + train.navigation.destination.name + " ("
+					chat.accept(" -> Waiting at: " + currentStation.name, blue);
+				} else if (train.navigation.destination != null)
+					chat.accept(" -> Travelling to " + train.navigation.destination.name + " ("
 						+ Mth.floor(train.navigation.distanceToDestination) + "m away)", darkBlue);
-			}
 				ScheduleRuntime runtime = train.runtime;
 				if (runtime.getSchedule() != null) {
-				chat.accept("  > Schedule, Entry " + runtime.currentEntry + ", "
+					chat.accept(" -> Schedule, Entry " + runtime.currentEntry + ", "
 						+ (runtime.paused ? "Paused"
 							: runtime.state.name()
 								.replaceAll("_", " ")),
 						runtime.paused ? darkBlue : blue);
-				if (!runtime.paused && runtime.state != State.POST_TRANSIT) {
-					for (Component component : runtime.getSchedule().entries.get(runtime.currentEntry).instruction
-						.getTitleAs("destination")) {
-						chat.accept("   - " + component.getString(), blue);
-					}
-				}
 				} else
-				chat.accept("  > Idle, No Schedule", darkBlue);
+					chat.accept(" -> Idle, No Schedule", darkBlue);
+				chatRaw.accept(createDeleteButton(train));
 				chat.accept("", white);
 			}
+			if (trainCount > 5) {
+				chat.accept("[...]", white);
+				chat.accept("", white);
+			}
+		}
+
 		chat.accept("-+--------------------------------+-", white);
 	}
 
+	private static Component createDeleteButton(Train train) {
+		return ComponentUtils.wrapInSquareBrackets((new TextComponent("Remove")).withStyle((p_180514_) -> {
+			return p_180514_.withColor(0xFFAD60)
+				.withClickEvent(new ClickEvent(ClickEvent.Action.RUN_COMMAND, "/c killTrain " + train.id.toString()))
+				.withHoverEvent(new HoverEvent(HoverEvent.Action.SHOW_TEXT,
+					new TextComponent("Click to remove ").append(train.name)))
+				.withInsertion("/c killTrain " + train.id.toString());
+		}));
+	}
+
 }

src/main/java/com/simibubi/create/foundation/command/KillTrainCommand.java

@@ -0,0 +1,40 @@
+package com.simibubi.create.foundation.command;
+
+import java.util.UUID;
+
+import com.mojang.brigadier.builder.ArgumentBuilder;
+import com.simibubi.create.Create;
+import com.simibubi.create.content.logistics.trains.entity.Train;
+
+import net.minecraft.commands.CommandSourceStack;
+import net.minecraft.commands.Commands;
+import net.minecraft.commands.arguments.UuidArgument;
+import net.minecraft.network.chat.TextComponent;
+
+public class KillTrainCommand {
+
+	static ArgumentBuilder<CommandSourceStack, ?> register() {
+		return Commands.literal("killTrain")
+			.requires(cs -> cs.hasPermission(2))
+			.then(Commands.argument("train", UuidArgument.uuid())
+				.executes(ctx -> {
+					CommandSourceStack source = ctx.getSource();
+					run(source, UuidArgument.getUuid(ctx, "train"));
+					return 1;
+				}));
+	}
+
+	private static void run(CommandSourceStack source, UUID argument) {
+		Train train = Create.RAILWAYS.trains.get(argument);
+		if (train == null) {
+			source.sendFailure(new TextComponent("No Train with id " + argument.toString()
+				.substring(0, 5) + "[...] was found"));
+			return;
+		}
+
+		train.invalid = true;
+		source.sendSuccess(new TextComponent("Train '").append(train.name)
+			.append("' removed successfully"), true);
+	}
+
+}