60 lines
1.9 KiB
Go
60 lines
1.9 KiB
Go
package memory
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"joylink.club/bj-rtsts-server/dto/data_proto"
|
|
"joylink.club/rtsssimulation/fi"
|
|
)
|
|
|
|
// 操作IBP按钮
|
|
func ChangeIBPButtonState(sim *VerifySimulation, mapId int32, stationId, btnId uint32, pressDown bool) error {
|
|
uidMap, err := getIbpUidByMapIdAndStationId(mapId, stationId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if uidMap.IbpButtonIds[btnId] == nil {
|
|
return fmt.Errorf("车站【%d】按钮【%d】UID不存在", stationId, btnId)
|
|
}
|
|
stationUid := QueryUidByMidAndComId(mapId, stationId, &data_proto.Station{})
|
|
if pressDown {
|
|
return fi.PressDownButton(sim.World, stationUid+"_"+uidMap.IbpButtonIds[btnId].Uid)
|
|
} else {
|
|
return fi.PressUpButton(sim.World, stationUid+"_"+uidMap.IbpButtonIds[btnId].Uid)
|
|
}
|
|
}
|
|
|
|
// 操作IBP按钮
|
|
func ChangeIBPKeyState(sim *VerifySimulation, mapId int32, stationId, keyId uint32, gear int32) error {
|
|
uidMap, err := getIbpUidByMapIdAndStationId(mapId, stationId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if uidMap.IbpKeyIds[keyId] == nil {
|
|
return fmt.Errorf("车站【%d】钥匙【%d】UID不存在", stationId, keyId)
|
|
}
|
|
stationUid := QueryUidByMidAndComId(mapId, stationId, &data_proto.Station{})
|
|
return fi.SwitchKeyGear(sim.World, stationUid+"_"+uidMap.IbpKeyIds[keyId].Uid, gear)
|
|
}
|
|
|
|
// 根据平面布置图ID、列车ID获取IbpUid信息
|
|
func getIbpUidByMapIdAndStationId(mapId int32, stationId uint32) (*IBPUidStructure, error) {
|
|
giData := QueryGiData[*data_proto.RtssGraphicStorage](mapId)
|
|
var station *data_proto.Station
|
|
for _, s := range giData.Stations {
|
|
if GetMapElementId(s.Common) == stationId {
|
|
station = s
|
|
break
|
|
}
|
|
}
|
|
if station == nil {
|
|
return nil, fmt.Errorf("地图【%d】车站【%d】不存在", mapId, stationId)
|
|
}
|
|
|
|
if station.RefIbpMapCode == "" {
|
|
return nil, fmt.Errorf("车站【%s】未关联IBP地图", station.StationName)
|
|
}
|
|
ibpMapId, _ := QueryGiDataByName[*data_proto.IBPGraphicStorage](station.RefIbpMapCode)
|
|
return QueryUidStructure[*IBPUidStructure](ibpMapId), nil
|
|
}
|