31ad3aa671
- Train station peripherals can now assemble and disassemble trains, check if the station is in assembly mode, set the assembly mode of the station, get and change the station name, check if a train is present at the station and get and change the currently present train name. - Refactored StationEditPacket. Moved most of the logic that was previously in StationEditPacket to StationTileEntity. This allows us to call this logic without having to send a packet. - Made Train#owner nullable. This is needed so that computers can assemble trains. All Train#owner is currently used for is to display the train status to the correct play.
120 lines
3.4 KiB
Java
120 lines
3.4 KiB
Java
package com.simibubi.create.content.logistics.trains.entity;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
import java.util.function.Supplier;
|
|
|
|
import com.simibubi.create.CreateClient;
|
|
import com.simibubi.create.content.logistics.trains.IBogeyBlock;
|
|
import com.simibubi.create.foundation.networking.SimplePacketBase;
|
|
import com.simibubi.create.foundation.utility.Couple;
|
|
import com.simibubi.create.foundation.utility.Iterate;
|
|
import com.simibubi.create.foundation.utility.RegisteredObjects;
|
|
import net.minecraft.network.FriendlyByteBuf;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.world.level.block.Block;
|
|
import net.minecraftforge.network.NetworkEvent.Context;
|
|
import net.minecraftforge.registries.ForgeRegistries;
|
|
|
|
public class TrainPacket extends SimplePacketBase {
|
|
|
|
UUID trainId;
|
|
Train train;
|
|
boolean add;
|
|
|
|
public TrainPacket(Train train, boolean add) {
|
|
this.train = train;
|
|
this.add = add;
|
|
}
|
|
|
|
public TrainPacket(FriendlyByteBuf buffer) {
|
|
add = buffer.readBoolean();
|
|
trainId = buffer.readUUID();
|
|
|
|
if (!add)
|
|
return;
|
|
|
|
UUID owner = null;
|
|
if (buffer.readBoolean())
|
|
owner = buffer.readUUID();
|
|
|
|
List<Carriage> carriages = new ArrayList<>();
|
|
List<Integer> carriageSpacing = new ArrayList<>();
|
|
|
|
int size = buffer.readVarInt();
|
|
for (int i = 0; i < size; i++) {
|
|
Couple<CarriageBogey> bogies = Couple.create(null, null);
|
|
for (boolean first : Iterate.trueAndFalse) {
|
|
if (!first && !buffer.readBoolean())
|
|
continue;
|
|
IBogeyBlock type = (IBogeyBlock) ForgeRegistries.BLOCKS.getValue(buffer.readResourceLocation());
|
|
bogies.set(first, new CarriageBogey(type, new TravellingPoint(), new TravellingPoint()));
|
|
}
|
|
int spacing = buffer.readVarInt();
|
|
carriages.add(new Carriage(bogies.getFirst(), bogies.getSecond(), spacing));
|
|
}
|
|
|
|
size = buffer.readVarInt();
|
|
for (int i = 0; i < size; i++)
|
|
carriageSpacing.add(buffer.readVarInt());
|
|
|
|
boolean doubleEnded = buffer.readBoolean();
|
|
train = new Train(trainId, owner, null, carriages, carriageSpacing, doubleEnded);
|
|
|
|
train.name = Component.Serializer.fromJson(buffer.readUtf());
|
|
train.icon = TrainIconType.byId(buffer.readResourceLocation());
|
|
}
|
|
|
|
@Override
|
|
public void write(FriendlyByteBuf buffer) {
|
|
buffer.writeBoolean(add);
|
|
buffer.writeUUID(train.id);
|
|
|
|
if (!add)
|
|
return;
|
|
|
|
buffer.writeBoolean(train.owner != null);
|
|
if (train.owner != null)
|
|
buffer.writeUUID(train.owner);
|
|
|
|
buffer.writeVarInt(train.carriages.size());
|
|
for (Carriage carriage : train.carriages) {
|
|
for (boolean first : Iterate.trueAndFalse) {
|
|
if (!first) {
|
|
boolean onTwoBogeys = carriage.isOnTwoBogeys();
|
|
buffer.writeBoolean(onTwoBogeys);
|
|
if (!onTwoBogeys)
|
|
continue;
|
|
}
|
|
CarriageBogey bogey = carriage.bogeys.get(first);
|
|
buffer.writeResourceLocation(RegisteredObjects.getKeyOrThrow((Block) bogey.type));
|
|
}
|
|
buffer.writeVarInt(carriage.bogeySpacing);
|
|
}
|
|
|
|
buffer.writeVarInt(train.carriageSpacing.size());
|
|
train.carriageSpacing.forEach(buffer::writeVarInt);
|
|
|
|
buffer.writeBoolean(train.doubleEnded);
|
|
buffer.writeUtf(Component.Serializer.toJson(train.name));
|
|
buffer.writeResourceLocation(train.icon.id);
|
|
}
|
|
|
|
@Override
|
|
public void handle(Supplier<Context> context) {
|
|
context.get()
|
|
.enqueueWork(() -> {
|
|
Map<UUID, Train> trains = CreateClient.RAILWAYS.trains;
|
|
if (add)
|
|
trains.put(train.id, train);
|
|
else
|
|
trains.remove(trainId);
|
|
});
|
|
context.get()
|
|
.setPacketHandled(true);
|
|
}
|
|
|
|
}
|