【修改NCC更新逻辑】

This commit is contained in:
weizhihong 2023-03-02 17:55:51 +08:00
parent b8cae4edfa
commit 0396b88c2b
6 changed files with 103 additions and 100 deletions

@ -1 +1 @@
Subproject commit 3ff3c0da726665b5bd057c75bd86cd0072c3da31
Subproject commit 12bfb1243559f366ddaa1f36b0a8cbaed3a3c694

View File

@ -5,10 +5,10 @@ import club.joylink.ecs.WorldManage;
import club.joylink.rtss.simulation.ncc.dto.NccMapDataDto;
import club.joylink.rtss.simulation.ncc.exception.NccExceptionAssertEnum;
import club.joylink.rtss.simulation.ncc.module.proto.StorageProto;
import club.joylink.rtss.simulation.ncc.module.proto.TrainStateProto;
import club.joylink.rtss.simulation.ncc.repository.IRtsNccMapDataRepository;
import club.joylink.rtss.simulation.ncc.simulation.event.NCCCreateEvent;
import club.joylink.rtss.simulation.ncc.simulation.event.NCCDestroyEvent;
import club.joylink.rtss.simulation.ncc.simulation.event.NCCResetEvent;
import club.joylink.rtss.simulation.ncc.simulation.event.NCCUpdateEvent;
import club.joylink.rtss.simulation.ncc.simulation.system.TrainStateSystem;
import com.google.protobuf.InvalidProtocolBufferException;
@ -40,6 +40,11 @@ public class NCCManagerService {
World world = WorldManage.create(generateWorldId(event.getSimulationId()));
// 将构建信息放入world
world.getStorage().insertSingleton(storage);
// 构建world实体
storage.getLinesList().forEach(line -> world.getStorage().createEntity(line.getCode(), line));
storage.getStationsList().forEach(station -> world.getStorage().createEntity(station.getCode(), station));
storage.getRoadSectionsList().forEach(roadSection -> world.getStorage().createEntity(roadSection.getCode(), roadSection));
storage.getTrainList().forEach(train -> world.getStorage().createEntity(train.getCode(), train, TrainStateProto.TrainState.newBuilder()));
// 添加系统
world.getTaskControl().addSystem(new TrainStateSystem());
world.start();
@ -62,20 +67,12 @@ public class NCCManagerService {
}
}
/**
* 重置world数据
*/
@EventListener
public void resetWorld(NCCResetEvent event) {
}
/**
* 销毁world
*/
@EventListener
public void destroyWorld(NCCDestroyEvent event) {
WorldManage.destroy(generateWorldId(event.getSimulationId()));
}
/**

View File

@ -3,11 +3,8 @@ package club.joylink.rtss.simulation.ncc.simulation;
import club.joylink.ecs.World;
import club.joylink.ecs.api.Entity;
import club.joylink.rtss.simulation.ncc.exception.NccExceptionAssertEnum;
import club.joylink.rtss.simulation.ncc.module.proto.CommandStatusProto;
import club.joylink.rtss.simulation.ncc.module.proto.RunStatusProto;
import club.joylink.rtss.simulation.ncc.module.proto.TrainStateCommandProto;
import club.joylink.rtss.simulation.ncc.module.proto.TrainStateProto;
import club.joylink.rtss.simulation.ncc.module.proto.dto.TrainStateDtoProto;
import com.google.protobuf.InvalidProtocolBufferException;
import java.util.Optional;
@ -46,7 +43,4 @@ public enum NccStateType {
}
public abstract void handle(World world, byte[] data);
}

View File

@ -1,14 +0,0 @@
package club.joylink.rtss.simulation.ncc.simulation.event;
import lombok.Getter;
/**
* NCC重置事件
*/
@Getter
public class NCCResetEvent extends NCCCommEvent {
public NCCResetEvent(Object source, String simulationId) {
super(source, simulationId);
}
}

View File

@ -19,14 +19,27 @@ public class TrainStateSystem implements WorldSystem<Tuple1<TrainStateProto.Trai
@Override
public void update(WorldQuery<Tuple1<TrainStateProto.TrainState.Builder>> query, Tuple1<StorageProto.Storage> storageTuple1) {
// 所有资源
StorageProto.Storage storage = storageTuple1.val1().get();
// 查找需要更新的车状态信息
query.forEach(tuple -> {
TrainStateProto.TrainState.Builder trainState = tuple.val1().get();
TrainStateCommandProto.TrainStateCommand.Builder command = trainState.getCommandBuilder();
if (command != null && Objects.equals(command.getStatus(), CommandStatusProto.CommandStatus.UNDO)) {
CommandStatusProto.CommandStatus status = command.getStatus();
if (Objects.equals(status, CommandStatusProto.CommandStatus.UNDO)
|| Objects.equals(status, CommandStatusProto.CommandStatus.RESET)) {
// 标识状态
command.setStatus(CommandStatusProto.CommandStatus.DOING);
if (CommandStatusProto.CommandStatus.RESET.equals(status)) {
trainState.clear();
} else {
update(trainState, command, storageTuple1.val1().get());
}
// 完成标识
command.setStatus(CommandStatusProto.CommandStatus.DONE);
}
});
}
private void update(TrainStateProto.TrainState.Builder trainState, TrainStateCommandProto.TrainStateCommand.Builder command, StorageProto.Storage storage) {
// 更新车次信息
if (StringUtils.isNoneBlank(command.getTripNumber())) {
trainState.setTripNumber(command.getTripNumber());
@ -87,12 +100,7 @@ public class TrainStateSystem implements WorldSystem<Tuple1<TrainStateProto.Trai
double totalDistance = Math.abs(roadSection.getMaxDistance() - roadSection.getMinDistance());
double distanceRatio = command.getDistance() / totalDistance;
trainState.setDistanceRatio(distanceRatio);
log.info("列车行驶至车站:" + trainState.getStartStation().getName() + "距离 " + trainState.getDistance() + ",行驶比例:" + distanceRatio);
}
// 完成标识
command.setStatus(CommandStatusProto.CommandStatus.DONE);
}
});
}
}

View File

@ -43,6 +43,14 @@ public final class CommandStatusProto {
* <code>DONE = 2;</code>
*/
DONE(2),
/**
* <pre>
* 重置
* </pre>
*
* <code>RESET = 3;</code>
*/
RESET(3),
UNRECOGNIZED(-1),
;
@ -70,6 +78,14 @@ public final class CommandStatusProto {
* <code>DONE = 2;</code>
*/
public static final int DONE_VALUE = 2;
/**
* <pre>
* 重置
* </pre>
*
* <code>RESET = 3;</code>
*/
public static final int RESET_VALUE = 3;
public final int getNumber() {
@ -99,6 +115,7 @@ public final class CommandStatusProto {
case 0: return UNDO;
case 1: return DOING;
case 2: return DONE;
case 3: return RESET;
default: return null;
}
}
@ -164,10 +181,11 @@ public final class CommandStatusProto {
descriptor;
static {
java.lang.String[] descriptorData = {
"\n\024command_status.proto\022\005proto*.\n\rCommand" +
"Status\022\010\n\004UNDO\020\000\022\t\n\005DOING\020\001\022\010\n\004DONE\020\002BC\n" +
"-club.joylink.rtss.simulation.ncc.module" +
".protoB\022CommandStatusProtob\006proto3"
"\n\024command_status.proto\022\005proto*9\n\rCommand" +
"Status\022\010\n\004UNDO\020\000\022\t\n\005DOING\020\001\022\010\n\004DONE\020\002\022\t\n" +
"\005RESET\020\003BC\n-club.joylink.rtss.simulation" +
".ncc.module.protoB\022CommandStatusProtob\006p" +
"roto3"
};
descriptor = com.google.protobuf.Descriptors.FileDescriptor
.internalBuildGeneratedFileFrom(descriptorData,