46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
|
package fi
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"joylink.club/ecs"
|
|||
|
|
"joylink.club/rtsssimulation/component"
|
|||
|
|
"joylink.club/rtsssimulation/component/component_proto"
|
|||
|
|
"joylink.club/rtsssimulation/entity"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// UpdateSignalLightFault 设置信号机灯故障
|
|||
|
|
//
|
|||
|
|
// light : 灯泡
|
|||
|
|
// faultType ：故障类型
|
|||
|
|
// set : true-设置故障，false-取消故障
|
|||
|
|
func UpdateSignalLightFault(w ecs.World, signalId string, light component_proto.Light_Color, faultType component_proto.Signal_Fault, set bool) error {
|
|||
|
|
r := <-ecs.Request[ecs.EmptyType](w, func() ecs.Result[ecs.EmptyType] {
|
|||
|
|
wd := entity.GetWorldData(w)
|
|||
|
|
signalEntry, ok := wd.EntityMap[signalId]
|
|||
|
|
if ok {
|
|||
|
|
switch faultType {
|
|||
|
|
case component_proto.Signal_DS: //断丝故障
|
|||
|
|
{
|
|||
|
|
lightEntry := component.SignalLightsType.Get(signalEntry).GetLightByColor(light)
|
|||
|
|
if lightEntry == nil {
|
|||
|
|
return ecs.NewErrResult(fmt.Errorf("信号机[%s]没有色灯[%d]", signalId, light))
|
|||
|
|
}
|
|||
|
|
if set {
|
|||
|
|
if !lightEntry.HasComponent(component.LightFaultDsType) {
|
|||
|
|
lightEntry.AddComponent(component.LightFaultDsType)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
if lightEntry.HasComponent(component.LightFaultDsType) {
|
|||
|
|
lightEntry.RemoveComponent(component.LightFaultDsType)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
default:
|
|||
|
|
return ecs.NewErrResult(fmt.Errorf("信号机[%s]不支持该故障类型[%s]", signalId, faultType))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return ecs.NewOkEmptyResult()
|
|||
|
|
})
|
|||
|
|
return r.Err
|
|||
|
|
}
|