58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
|
package device_sys
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/yohamta/donburi"
|
||
|
|
"joylink.club/ecs"
|
||
|
|
"joylink.club/ecs/filter"
|
||
|
|
"joylink.club/rtsssimulation/component"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TrainSectionSystem 列车所在区段更新系统
|
||
|
|
type TrainSectionSystem struct {
|
||
|
|
trainQuery *ecs.Query
|
||
|
|
axleSectionQuery *ecs.Query
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTrainSectionSystem() *TrainSectionSystem {
|
||
|
|
return &TrainSectionSystem{trainQuery: ecs.NewQuery(filter.Contains(component.UidType, component.TrainPositionInfoType)),
|
||
|
|
axleSectionQuery: ecs.NewQuery(filter.Contains(component.UidType, component.AxlePhysicalSectionType))}
|
||
|
|
}
|
||
|
|
func (s *TrainSectionSystem) Update(w ecs.World) {
|
||
|
|
//key-sectionId,统计区段上有车的情况
|
||
|
|
sectionTrainMap := make(map[string]*trainCount)
|
||
|
|
//所有列车
|
||
|
|
s.trainQuery.Each(w, func(entry *donburi.Entry) {
|
||
|
|
tp := component.TrainPositionInfoType.Get(entry)
|
||
|
|
for _, sectionId := range tp.SectionIds { //车所在区段
|
||
|
|
tc, find := sectionTrainMap[sectionId]
|
||
|
|
if !find {
|
||
|
|
tc = newTrainCount()
|
||
|
|
sectionTrainMap[sectionId] = tc
|
||
|
|
}
|
||
|
|
tc.add()
|
||
|
|
}
|
||
|
|
})
|
||
|
|
//计轴区段
|
||
|
|
s.axleSectionQuery.Each(w, func(entry *donburi.Entry) {
|
||
|
|
axleSectionId := component.UidType.Get(entry).Id
|
||
|
|
axleSection := component.AxlePhysicalSectionType.Get(entry)
|
||
|
|
tc, find := sectionTrainMap[axleSectionId]
|
||
|
|
if find {
|
||
|
|
axleSection.UpdateCount(int(tc.count))
|
||
|
|
} else {
|
||
|
|
axleSection.UpdateCount(0)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
type trainCount struct {
|
||
|
|
count int8
|
||
|
|
}
|
||
|
|
|
||
|
|
func newTrainCount() *trainCount {
|
||
|
|
return &trainCount{count: 0}
|
||
|
|
}
|
||
|
|
func (c *trainCount) add() {
|
||
|
|
c.count++
|
||
|
|
}
|