75 lines
2.9 KiB
Go
75 lines
2.9 KiB
Go
package entity
|
||
|
||
import (
|
||
"fmt"
|
||
"joylink.club/ecs"
|
||
"joylink.club/rtsssimulation/component"
|
||
"joylink.club/rtsssimulation/repository"
|
||
)
|
||
|
||
// LoadAxlePhysicalSections 加载计轴区段
|
||
func LoadAxlePhysicalSections(w ecs.World) error {
|
||
data := GetWorldData(w)
|
||
sections := data.Repo.PhysicalSectionList()
|
||
for _, section := range sections {
|
||
if is, se := section.IsAxleSection(); se == nil && is {
|
||
fadcDeviceId := createFaDcAxleDeviceId(section.CentralizedStation())
|
||
fadcDeviceEntry, find := data.EntityMap[fadcDeviceId]
|
||
if !find {
|
||
fadcDeviceEntry = createFadcDeviceEntity(w, section, data)
|
||
data.EntityMap[fadcDeviceId] = fadcDeviceEntry
|
||
}
|
||
fadcDevice := component.FaDcAxleDeviceType.Get(fadcDeviceEntry)
|
||
counter := createAxleCounterEntity(w, section, data)
|
||
axleSec := createAxleSectionEntity(w, section, data)
|
||
fadcDevice.AddAxleSection(counter, axleSec)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 联锁集中站计轴管理设备实体
|
||
func createFadcDeviceEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
|
||
uid := createFaDcAxleDeviceId(axleSection.CentralizedStation())
|
||
entry, ok := worldData.EntityMap[uid]
|
||
if !ok {
|
||
entry = w.Entry(w.Create(component.UidType, component.FaDcAxleDeviceType))
|
||
//
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
component.FaDcAxleDeviceType.Set(entry, &component.FaDcAxleDevice{CounterMap: make(map[string]*ecs.Entry)})
|
||
//
|
||
worldData.EntityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 计轴区段实体
|
||
func createAxleSectionEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
|
||
uid := axleSection.Id()
|
||
entry, ok := worldData.EntityMap[uid]
|
||
if !ok {
|
||
entry = w.Entry(w.Create(component.UidType, component.AxleSectionType, component.AxleSectionFaultType))
|
||
//
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
component.AxleSectionType.Set(entry, &component.AxleSection{Occ: false})
|
||
component.AxleSectionFaultType.Set(entry, &component.AxleSectionFault{SectionFault: false})
|
||
//
|
||
worldData.EntityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 计轴器实体
|
||
func createAxleCounterEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
|
||
entry := w.Entry(w.Create(component.AxleCounterType, component.AxleCounterRuntimeType, component.AxleSectionFlag))
|
||
component.AxleCounterType.Set(entry, component.NewAxleCounter())
|
||
component.AxleCounterRuntimeType.Set(entry, &component.AxleCounterRuntime{Rac: false, Rjo: false, Rjt: false, Drst: false, Pdrst: false, DoingPdrst: false})
|
||
component.AxleSectionFlag.SetValue(entry, component.Uid{Id: axleSection.Id()})
|
||
return entry
|
||
}
|
||
|
||
// 设备集中站计轴管理设备id（通过联锁集中站centralizedStation来构建）
|
||
func createFaDcAxleDeviceId(centralizedStation string) string {
|
||
return fmt.Sprintf("fadc-axle-device-%s", centralizedStation)
|
||
}
|