package train_pc_sim import ( "context" "encoding/hex" "fmt" "joylink.club/bj-rtsts-server/config" "joylink.club/bj-rtsts-server/dto/state_proto" "joylink.club/bj-rtsts-server/sys_error" "joylink.club/bj-rtsts-server/third_party/message" "joylink.club/bj-rtsts-server/third_party/tcp" "joylink.club/bj-rtsts-server/third_party/tpapi" "log/slog" "math" "sync" "time" ) type TrainControlEvent struct { Type byte Data []byte } //var FireTrainControlEventType = ecs.NewEventType[TrainControlEvent]() type TrainPcSim interface { tpapi.ThirdPartyApiService Start(pcSimManage TrainPcSimManage) Stop() // SendDriverActive Deprecated 发送驾驶端激活 SendDriverActive(train *state_proto.TrainState) // SendHandleSwitch 发送牵引制动手柄 SendHandleSwitch(oldTraction, oldBrakeForce int64, tractionState bool, train *state_proto.TrainState) // SendTrainDirection 列车运行方向 //因文档说明不清楚,在调用的时候目前是注释状态,现场调试可能会用到 SendTrainDirection(train *state_proto.TrainState, trainForward, trainBackward bool) //发送应答器信息数据 SendBaliseData(train *state_proto.TrainState, msgType byte, data []byte) //发布列车控制的相关事件 //PublishTrainControlEvent(train *state_proto.TrainState, events []TrainControlEvent) SendTrainControlMsg(train *state_proto.TrainState, baseMessage []message.TrainPcSimBaseMessage) // CreateOrRemoveSpeedPLace 创建或删除速度位置信息 //CreateOrRemoveSpeedPLace(train *state_proto.TrainState) // CreateOrRemoveTrain 创建或删除列车 CreateOrRemoveTrain(train *state_proto.TrainState, isCreate bool) error // TrainPluseCount 计算列车脉冲 TrainPluseCount(sta *state_proto.TrainState, h1, h2, t1, t2 float32) ResetPluseCount(sta *state_proto.TrainState) //FindAllThirdPartState() []tpapi.ThirdPartyApiService DsnAddAndReturn() byte ResetPlusePlace(sta *state_proto.TrainState) } type TrainPcSimManage interface { GetTrainPcSimConfig() []config.VehiclePCSimConfig //GetConnTrain() *state_proto.TrainState GetConnTrain2() []*state_proto.TrainState //获取列车模拟量数据 ObtainTrainDigitalMockData(train *state_proto.TrainState) []message.TrainPcSimBaseMessage // TrainPcSimDigitalOutInfoHandle 4.4.1. 车载输出数字量信息报文内容 TrainPcSimDigitalOutInfoHandle(train *state_proto.TrainState, trainInit bool, data []byte) bool // TrainPcSimDigitalReportHandle 4.4.2. 车载输出数字反馈量信息报文内容 TrainPcSimDigitalReportHandle(train *state_proto.TrainState, data []byte) FindConnTrain(ct state_proto.TrainConnState_TrainConnType) *state_proto.TrainState // TrainPcSimMockInfo 门模式 //TrainDoorModeHandle(state byte) //处理列车pc仿真模拟量数据 TrainPcSimMockInfo(train *state_proto.TrainState, data []byte) // TrainBtmQuery 处理列车btm查询 TrainBtmQuery(train *state_proto.TrainState, data []byte) } type trainPcSimService struct { state tpapi.ThirdPartyApiServiceState newPcSimclientMap map[string]*TrainPcReciverData cancleContext context.CancelFunc trainPcSimManage TrainPcSimManage configs []config.VehiclePCSimConfig btmDsn uint8 } var ( initLock = &sync.Mutex{} singleObj *trainPcSimService ) func Default() TrainPcSim { defer initLock.Unlock() initLock.Lock() if singleObj == nil { singleObj = &trainPcSimService{} } return singleObj } const Name = "车载pc仿真" func (d *trainPcSimService) Name() string { return "" } func (d *trainPcSimService) State() tpapi.ThirdPartyApiServiceState { return tpapi.ThirdPartyState_Closed } func (d *trainPcSimService) FindAppendApiService() []tpapi.ThirdPartyApiService { return d.findAllThirdPartState() } func (d *trainPcSimService) TrueService() bool { return false } func (d *trainPcSimService) ServiceDesc() string { return Name } func (d *trainPcSimService) DsnAddAndReturn() byte { defer initLock.Unlock() initLock.Lock() returnVal := d.btmDsn d.btmDsn = returnVal + 1 return returnVal } func FindTrainPcSimClientKey2(t *state_proto.TrainState) string { return t.ConnState.TypeName } func (d *trainPcSimService) findTrainConn(sta *state_proto.TrainState) (*TrainPcReciverData, error) { trainPcReciver := d.newPcSimclientMap[sta.ConnState.TypeName] if trainPcReciver == nil { return nil, fmt.Errorf("") } return trainPcReciver, nil } func (d *trainPcSimService) findAllThirdPartState() []tpapi.ThirdPartyApiService { services := make([]tpapi.ThirdPartyApiService, 0) for _, data := range d.newPcSimclientMap { services = append(services, data) } return services } // 速度(单位mm/s)对应的脉冲数:速度*200/pi/840; // 里程(单位mm)对应的脉冲:总里程*200/pi/840 func pluseCountSpeed(wheelDiameter int32, speedMeter float32) uint32 { s1 := speedMeter * 1000 pluseCountData := s1 * 200 / math.Pi / float32(wheelDiameter) return uint32(pluseCountData) } func (d *trainPcSimService) ResetPlusePlace(sta *state_proto.TrainState) { defer initLock.Unlock() initLock.Lock() if sd, err := d.findTrainConn(sta); err == nil { /*if !sd.speedPlace.Tmp { } else { sd.speedPlace.Tmp = false }*/ if sd.speedPlace.Tmp { //sd.speedPlace.PulseCount1 = sd.speedPlace.HistoryCount //sd.speedPlace.PulseCount2 = sd.speedPlace.HistoryCount //sd.speedPlace.Tmp = false } //sd.speedPlace.HistoryCount = 0 //slog.Info(fmt.Sprintf(" p1:%v,p2:%v,h:%v", sd.speedPlace.PulseCount1, sd.speedPlace.PulseCount2, sd.speedPlace.HistoryCount)) //sd.speedPlace.Tmp = true } } func (d *trainPcSimService) ResetPluseCount(sta *state_proto.TrainState) { defer initLock.Unlock() initLock.Lock() if sd, err := d.findTrainConn(sta); err == nil { sd.speedPlace.PulseCount1 = 0 sd.speedPlace.PulseCount2 = 0 } } func (d *trainPcSimService) pluseSpeed(sta *state_proto.TrainState) (uint32, float32) { defer initLock.Unlock() initLock.Lock() var sum float32 = 0 pcLen := len(sta.PluseCount.PulseCount3) if pcLen == 0 { return 0, 0 } for _, f := range sta.PluseCount.PulseCount3 { sum += f } d.TrainPluseCountReset(sta) speed := sum / float32(pcLen) return pluseCountSpeed(sta.WheelDiameter, speed), speed } func (d *trainPcSimService) TrainPluseCount(sta *state_proto.TrainState, h1, h2, t1, t2 float32) { defer initLock.Unlock() initLock.Lock() if sd, err := d.findTrainConn(sta); err == nil { sd.speedPlace.PulseCount1 += float32(sta.DynamicState.Displacement) sd.speedPlace.PulseCount2 = sd.speedPlace.PulseCount1 } if sta.TrainRunUp { if sta.TrainEndsA.SpeedSensorEnableA { sta.PluseCount.PulseCount1 = pluseCountSpeed(sta.WheelDiameter, h1) sta.PluseCount.PulseCount3 = append(sta.PluseCount.PulseCount3, h1) } if sta.TrainEndsA.SpeedSensorEnableB { sta.PluseCount.PulseCount2 = pluseCountSpeed(sta.WheelDiameter, h2) sta.PluseCount.PulseCount4 = append(sta.PluseCount.PulseCount3, h2) } } else { if sta.TrainEndsB.SpeedSensorEnableA { sta.PluseCount.PulseCount1 = pluseCountSpeed(sta.WheelDiameter, t1) sta.PluseCount.PulseCount3 = append(sta.PluseCount.PulseCount3, t1) } if sta.TrainEndsB.SpeedSensorEnableB { sta.PluseCount.PulseCount2 = pluseCountSpeed(sta.WheelDiameter, t2) sta.PluseCount.PulseCount4 = append(sta.PluseCount.PulseCount3, t2) } } } func (d *trainPcSimService) TrainPluseCountReset(sta *state_proto.TrainState) { sta.PluseCount.PulseCount1 = 0 sta.PluseCount.PulseCount2 = 0 sta.PluseCount.PulseCount3 = make([]float32, 0) sta.PluseCount.PulseCount4 = make([]float32, 0) } func (d *trainPcSimService) newCloseAllConn() { trains := d.trainPcSimManage.GetConnTrain2() for _, train := range trains { d.CreateOrRemoveTrain(train, false) } } func (d *trainPcSimService) newCloseConn(clientKey string) { rd := d.newPcSimclientMap[clientKey] if rd != nil { rd.tcpClient.Close() rd.tcpClient = nil rd.train = nil rd.speedPlace = nil } } func (d *trainPcSimService) findConfig(configName string) (*config.VehiclePCSimConfig, error) { for _, cfg := range d.configs { if cfg.Open && cfg.ConfigName == configName { return &cfg, nil } } return nil, fmt.Errorf("未找到对应的车载pc连接配置") } func (d *trainPcSimService) initConn(clientKey string) error { rd := d.newPcSimclientMap[clientKey] if rd != nil && rd.tcpClient != nil && rd.tcpClient.IsConning() { return nil } else { rd.trainInit = false rd.tcpClient = nil } cfg, cfgErr := d.findConfig(clientKey) if cfgErr != nil { errMsg := fmt.Sprintf("没找到对应的配置信息 key:%v", clientKey) //slog.Error(errMsg, cfgErr.Error()) rd.updateState(tpapi.ThirdPartyState_Broken) return sys_error.New(errMsg, cfgErr) } addr := fmt.Sprintf("%v:%v", cfg.PcSimIp, cfg.PcSimPort) client2, err := tcp.StartTcpClient(addr, rd.receiverDataHandle, rd.readError) if err != nil { connErrMsg := fmt.Sprintf("车载pc连接失败 clientKey:%v", clientKey) //slog.Error(connErrMsg, err.Error()) rd.updateState(tpapi.ThirdPartyState_Broken) return sys_error.New(connErrMsg, err) } else { rd.tcpClient = client2 } return nil } func (d *trainPcSimService) Start(pcSimManage TrainPcSimManage) { configs := pcSimManage.GetTrainPcSimConfig() d.newPcSimclientMap = make(map[string]*TrainPcReciverData) if len(configs) <= 0 { slog.Info("车载pc仿真配置未开启") return } closedCount := 0 for _, c := range configs { if !c.Open { closedCount++ } else { ck := c.ConfigName pcReciver := &TrainPcReciverData{clientKey: ck, pcSimManage: pcSimManage} pcReciver.updateState(tpapi.ThirdPartyState_Closed) d.newPcSimclientMap[ck] = pcReciver } } if closedCount == len(configs) { slog.Error("车载pc仿真配置未开启") return } d.configs = configs ctx, ctxFun := context.WithCancel(context.Background()) d.cancleContext = ctxFun d.trainPcSimManage = pcSimManage go d.sendTrainLocationAndSpeedTask(ctx) } func (d *trainPcSimService) Stop() { for _, data := range d.newPcSimclientMap { data.updateState(tpapi.ThirdPartyState_Closed) } if d.cancleContext != nil { d.cancleContext() d.cancleContext = nil } d.newCloseAllConn() } func (d *trainPcSimService) CreateOrRemoveTrain(train *state_proto.TrainState, isCreate bool) error { clientKey := FindTrainPcSimClientKey2(train) err := d.initConn(clientKey) if err != nil { d.newCloseConn(clientKey) return err } data := []byte{message.FLAG_CAMMAND_REMOVE_TRAIN} if isCreate { data[0] = message.FLAG_CAMMAND_CREATE_TRAIN } msg := &message.TrainPcSimBaseMessage{Data: data, Type: message.RECIVE_TRAIN_CREATE_REMOVE} rd := d.newPcSimclientMap[clientKey] if rd != nil { initTrainErr := d.initTrain(rd, train, isCreate, msg) if !isCreate { d.newCloseConn(clientKey) } if initTrainErr != nil { return initTrainErr } } return nil } func (d *trainPcSimService) initTrain(rd *TrainPcReciverData, train *state_proto.TrainState, isCreate bool, trains *message.TrainPcSimBaseMessage) error { msgs := make([]message.TrainPcSimBaseMessage, 0) sendMsg := make([]byte, 0) if isCreate { rd.speedPlace = &message.TrainSpeedPlaceReportMsg{} train.PluseCount = &state_proto.SensorSpeedPulseCount{} rd.train = train tcc := train.Tcc tcc.LineInitTimeStamp12 = 0 tcc.Line12ConnErr = false tmpMsgs := d.trainPcSimManage.ObtainTrainDigitalMockData(train) msgs = append(msgs, tmpMsgs...) msgs = append(msgs, message.TrainPcSimBaseMessage{Data: []byte{0x00}, Type: message.RECIVE_TRAIN_DOOR_MODE}) //门模式 msgs = append(msgs, message.TrainPcSimBaseMessage{Data: []byte{}, Type: message.RECIVE_TRAIN_BTN_CLEAR_ALL_PRE_DATA}) //清空应答器 } else { train.VobcState.Tc1Active = false train.VobcState.Tc2Active = false tcc := train.Tcc for _, key := range tcc.DriverKey { key.Val = false } msgs = append(msgs, message.TrainPcSimBaseMessage{Data: []byte{message.TRAIN_BRAKE_STATE, 0}, Type: message.SENDER_TRAIN_OUTR_INFO}) //驾驶室激活 msgs = append(msgs, message.TrainPcSimBaseMessage{Data: []byte{message.KEY_STATE, 0}, Type: message.SENDER_TRAIN_OUTR_INFO}) //驾驶室激活 msgs = append(msgs, message.TrainPcSimBaseMessage{Data: []byte{}, Type: message.SENDER_TRAIN_TC_NOT_ACTIVE}) //驾驶室激活 } for _, msg := range msgs { data := msg.Encode() sendMsg = append(sendMsg, data...) } sendMsg = append(sendMsg, trains.Encode()...) hexData := hex.EncodeToString(sendMsg) slog.Info(fmt.Sprintf("发送列车初始化消息:%v", hexData)) rd.tcpClient.Send(sendMsg) return nil } // 依据文档80ms发送列车速度位置 func (d *trainPcSimService) sendTrainLocationAndSpeedTask(ctx context.Context) { for { select { case <-ctx.Done(): return default: } trains := d.trainPcSimManage.GetConnTrain2() for _, train := range trains { if train.ConnState.Conn && train.PluseCount != nil { trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("pc仿真速度位置未找到对应的列车 id:%v", train.Id)) continue } connState := tpapi.ThirdPartyState_Normal if train.Tcc.Line12ConnErr { connState = tpapi.ThirdPartyState_Broken } trainClient.updateState(connState) //s1, s2 := train.PluseCount.PulseCount1, train.PluseCount.PulseCount2 s1, speed := d.pluseSpeed(train) s2 := s1 /*jjzdBtn := train.Tcc.Buttons["JJZD"] if jjzdBtn.Passed { s1 = 0 s2 = 0 }*/ //trainClient.speedPlace.ParsePulseCount1(s1, s2) //slog.Info(fmt.Sprintf("发送列车up:%v ,s1:%v,s2:%v,c1:%v,c2:%v, h位置消息:%v,eb:%v,tmp:%v", train.TrainRunUp, s1, s2, trainClient.speedPlace.PulseCount2, trainClient.speedPlace.PulseCount1, trainClient.speedPlace.HistoryCount, train.VobcState.EmergencyBrakingStatus, trainClient.speedPlace.Tmp)) //trainClient.speedPlace.PulseCount1 = uint32(train.DynamicState.Displacement) //trainClient.speedPlace.PulseCount2 = trainClient.speedPlace.PulseCount1 runDir := uint16(2) if train.Tcc.DirKey.Val == 1 { runDir = 1 } data := trainClient.speedPlace.Encode(runDir, s1, s2) bm := &message.TrainPcSimBaseMessage{Type: message.SENDER_TRAIN_LOCATION_INFO, Data: data} dataCode := bm.Encode() slog.Info(fmt.Sprintf("发送列车速度位置,列车:%v,列车速度:%v,计数脉冲: %v,累计里程: %v ,累计脉冲:%v,发送数据:%v", train.Id, speed, s1, trainClient.speedPlace.PulseCount1, trainClient.speedPlace.PcCount, hex.EncodeToString(dataCode))) err := trainClient.tcpClient.Send(dataCode) if err != nil { slog.Error(fmt.Sprintf("发送列车速度位置失败,列车:%v,发送数据:%v", train.Id, hex.EncodeToString(dataCode))) } } } time.Sleep(time.Millisecond * 80) } } // SendDriverActive Deprecated 发送驾驶激活 func (d *trainPcSimService) SendDriverActive(train *state_proto.TrainState) { trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("发送驾驶激活,未找到对应的列车连接,列车id:%v", train.Id)) return } vobc := train.VobcState defulatBuf := make([]byte, 0) msg := &message.TrainPcSimBaseMessage{Data: defulatBuf} if train.TrainRunUp { if vobc.Tc1Active { msg.Type = message.SENDER_TRAIN_TC_ACTIVE } else if vobc.Tc1Active == false { msg.Type = message.SENDER_TRAIN_TC_NOT_ACTIVE } } else if !train.TrainRunUp { if vobc.Tc2Active { msg.Type = message.SENDER_TRAIN_TC_ACTIVE } else if vobc.Tc2Active == false { msg.Type = message.SENDER_TRAIN_TC_NOT_ACTIVE } } msgs := make([]byte, 0) if msg.Type == message.SENDER_TRAIN_TC_ACTIVE { dd3 := message.TrainPcSimBaseMessage{Data: []byte{message.KEY_STATE, 1}, Type: message.SENDER_TRAIN_OUTR_INFO} msgs = append(msgs, dd3.Encode()...) } else { dd3 := message.TrainPcSimBaseMessage{Data: []byte{message.KEY_STATE, 0}, Type: message.SENDER_TRAIN_OUTR_INFO} msgs = append(msgs, dd3.Encode()...) } msgs = append(msgs, msg.Encode()...) hexData := hex.EncodeToString(msgs) //slog.Info(fmt.Sprintf("发送驾驶激活列车id:%v,数据:%v", train.Id, hexData)) err := trainClient.tcpClient.Send(msgs) if err != nil { slog.Error(fmt.Sprintf("发送驾驶激活失败列车id:%v,数据:%v,err:%v", train.Id, hexData, err.Error())) } } func (d *trainPcSimService) SendHandleSwitch(oldTraction, oldBrakeForce int64, tractionState bool, train *state_proto.TrainState) { trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("发送列车牵引知道失败,未找到对应的列车id:%v", train.Id)) return } tc := train.ConnState if tc.Conn { vobc := train.VobcState msg := &message.TrainPcSimBaseMessage{} newTraction := vobc.TractionForce if tractionState { if newTraction <= oldTraction && newTraction <= 0 { //手柄取消前进 msg.Type = message.RECIVE_TRAIN_HAND_KEY_CANCLE_FORWARD } else { //手柄前进 msg.Type = message.SENDER_TRAIN_HAND_KEY_FORWARD } } else { /*if newBrake >= newOldBrakeForce && newBrake == 0 { //手柄取消后退 msg.Type = message.RECIVE_TRAIN_HAND_KEY_CACLE_BACKWARD } else if newBrake < newOldBrakeForce { //手柄后退 msg.Type = message.RECIVE_TRAIN_HAND_KEY_BACKWARD } else { //手柄后退 msg.Type = message.RECIVE_TRAIN_HAND_KEY_BACKWARD }*/ msg.Type = message.RECIVE_TRAIN_HAND_KEY_BACKWARD } da := msg.Encode() //slog.Info("发送列车手柄消息", "msg", hex.EncodeToString(da)) err := trainClient.tcpClient.Send(da) //err := client.Send(da) if err != nil { slog.Error("发送列车手柄消息失败", "msg", hex.EncodeToString(da)) } } } func (d *trainPcSimService) SendTrainDirection(train *state_proto.TrainState, trainForward, trainBackward bool) { trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("发送列车方向失败,未找到列车连接,trainId:%s", train.Id)) return } baseMsgs := make([]*message.TrainPcSimBaseMessage, 0) if !trainForward && !trainBackward { baseMsgs = append(baseMsgs, &message.TrainPcSimBaseMessage{Type: message.RECIVE_TRAIN_HAND_KEY_CANCLE_FORWARD}) baseMsgs = append(baseMsgs, &message.TrainPcSimBaseMessage{Type: message.RECIVE_TRAIN_HAND_KEY_CACLE_BACKWARD}) } else if trainForward { baseMsgs = append(baseMsgs, &message.TrainPcSimBaseMessage{Type: message.SENDER_TRAIN_HAND_KEY_FORWARD}) } else if trainBackward { baseMsgs = append(baseMsgs, &message.TrainPcSimBaseMessage{Type: message.RECIVE_TRAIN_HAND_KEY_BACKWARD}) } for _, msg := range baseMsgs { da := msg.Encode() //slog.Info(fmt.Sprintf("发送列车方向列车:%v ,数据:%v", train.Id, hex.EncodeToString(da))) err := trainClient.tcpClient.Send(da) if err != nil { slog.Error(fmt.Sprintf("发送列车方向失败列车:%v ,数据:%v,err:%v", train.Id, hex.EncodeToString(da), err.Error())) } } } func (d *trainPcSimService) SendBaliseData(train *state_proto.TrainState, msgType byte, data []byte) { trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("发送列车PC仿真应答器信息失败,未找到列车连接,trainId:%v", train.Id)) return } msg := &message.TrainPcSimBaseMessage{} msg.Type = msgType msg.Data = data da := msg.Encode() //slog.Info(fmt.Sprintf("发送列车PC仿真应答器信息,数据类型:0x%x,源数据长度:%v,数据:%v", msgType, len(data), hex.EncodeToString(da))) err := trainClient.tcpClient.Send(da) if err != nil { slog.Info(fmt.Sprintf("发送列车PC仿真应答器信息失败,数据:%v", hex.EncodeToString(da))) } } func (d *trainPcSimService) SendTrainControlMsg(train *state_proto.TrainState, baseMessage []message.TrainPcSimBaseMessage) { if len(baseMessage) <= 0 { return } trainClient, trainDataErr := d.findTrainConn(train) if trainDataErr != nil { slog.Error(fmt.Sprintf("发送列车控制信息失败,无连接,列车Id:%v", train.Id)) return } for _, msg := range baseMessage { d.sendData(trainClient.tcpClient, msg.Encode()) } } func (d *trainPcSimService) sendData(client *tcp.TcpClient, data []byte) { err := client.Send(data) if err != nil { slog.Error(fmt.Sprintf("列车数字量信息发送失败,数据:%v", err.Error())) } }