122 lines
3.0 KiB
Go
122 lines
3.0 KiB
Go
package fi
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"joylink.club/ecs"
|
|
"joylink.club/rtsssimulation/component"
|
|
"joylink.club/rtsssimulation/component/component_proto"
|
|
"joylink.club/rtsssimulation/entity"
|
|
)
|
|
|
|
// 道岔功能接口
|
|
|
|
// 设置道岔故障
|
|
func SetTurnoutFault(w ecs.World, id string, fault component_proto.Turnout_Fault) error {
|
|
return updateTrunoutFault(w, id, fault, true)
|
|
}
|
|
|
|
// 取消道岔故障
|
|
func CancelTurnoutFault(w ecs.World, id string, fault component_proto.Turnout_Fault) error {
|
|
return updateTrunoutFault(w, id, fault, false)
|
|
}
|
|
|
|
func updateTrunoutFault(w ecs.World, id string, fault component_proto.Turnout_Fault, set bool) error {
|
|
result := <-ecs.Request[ecs.EmptyType](w, func() ecs.Result[ecs.EmptyType] {
|
|
wd := entity.GetWorldData(w)
|
|
entry, ok := wd.EntityMap[id]
|
|
if ok {
|
|
ct := component.GetTurnoutFaultType(fault)
|
|
switch fault {
|
|
case component_proto.Turnout_JC: // 挤岔故障
|
|
zzjs := component.TurnoutZzjType.Get(entry)
|
|
for _, e := range zzjs.ZzjList {
|
|
if set {
|
|
if !e.HasComponent(ct) {
|
|
e.AddComponent(ct)
|
|
}
|
|
} else {
|
|
e.RemoveComponent(ct)
|
|
}
|
|
}
|
|
default:
|
|
if set {
|
|
if !entry.HasComponent(ct) {
|
|
entry.AddComponent(ct)
|
|
}
|
|
} else {
|
|
entry.RemoveComponent(ct)
|
|
}
|
|
}
|
|
} else {
|
|
return ecs.NewErrResult(fmt.Errorf("未找到id=%s的道岔", id))
|
|
}
|
|
return ecs.NewOkEmptyResult()
|
|
})
|
|
return result.Err
|
|
}
|
|
|
|
// 驱动道岔转辙机
|
|
// id - 道岔uid
|
|
// dc - 定操/反操
|
|
// on - 设置/取消
|
|
func driveTurnoutZzj(w ecs.World, id string, dc bool, on bool) error {
|
|
result := <-ecs.Request[ecs.EmptyType](w, func() ecs.Result[ecs.EmptyType] {
|
|
wd := entity.GetWorldData(w)
|
|
entry, ok := wd.EntityMap[id]
|
|
if ok {
|
|
if entry.HasComponent(component.Zdj9TwoDriveType) { // 有电路，驱动继电器
|
|
drive := component.Zdj9TwoDriveType.Get(entry)
|
|
if dc {
|
|
drive.YCJ = on
|
|
drive.DCJ = on
|
|
if on {
|
|
drive.FCJ = !on
|
|
}
|
|
} else {
|
|
drive.YCJ = on
|
|
drive.FCJ = on
|
|
if on {
|
|
drive.DCJ = !on
|
|
}
|
|
}
|
|
} else { // 无电路，直接驱动转辙机
|
|
tz := component.TurnoutZzjType.Get(entry)
|
|
for _, zzj := range tz.ZzjList {
|
|
state := component.ZzjStateType.Get(zzj)
|
|
state.Td = on
|
|
if dc {
|
|
state.Dw = true
|
|
} else {
|
|
state.Dw = false
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
return ecs.NewErrResult(fmt.Errorf("未找到id=%s的道岔", id))
|
|
}
|
|
return ecs.NewOkEmptyResult()
|
|
})
|
|
return result.Err
|
|
}
|
|
|
|
// 设置道岔转辙机定操
|
|
func DriveTurnoutDCOn(w ecs.World, id string) error {
|
|
return driveTurnoutZzj(w, id, true /* dc */, true /* on */)
|
|
}
|
|
|
|
// 取消道岔转辙机定操
|
|
func DriveTurnoutDCOff(w ecs.World, id string) error {
|
|
return driveTurnoutZzj(w, id, true /* dc */, false /* on */)
|
|
}
|
|
|
|
// 设置道岔转辙机反操
|
|
func DriveTurnoutFCOn(w ecs.World, id string) error {
|
|
return driveTurnoutZzj(w, id, false /* dc */, true /* on */)
|
|
}
|
|
|
|
// 取消道岔转辙机反操
|
|
func DriveTurnoutFCOff(w ecs.World, id string) error {
|
|
return driveTurnoutZzj(w, id, false /* dc */, false /* on */)
|
|
}
|