62 lines
2.1 KiB
Go
62 lines
2.1 KiB
Go
package entity
|
|
|
|
import (
|
|
"fmt"
|
|
"joylink.club/ecs"
|
|
"joylink.club/rtsssimulation/component"
|
|
"joylink.club/rtsssimulation/repository"
|
|
"strings"
|
|
)
|
|
|
|
func newAxleManageDevice(w ecs.World, data *component.WorldData, centralizedStation string) *ecs.Entry {
|
|
entry := w.Entry(w.Create(component.AxleManageDeviceType))
|
|
component.AxleManageDeviceType.Set(entry, component.NewAxleManageDevice(centralizedStation))
|
|
data.AxleManageDeviceEntities = append(data.AxleManageDeviceEntities, entry)
|
|
return entry
|
|
}
|
|
|
|
// 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 {
|
|
if len(strings.TrimSpace(section.CentralizedStation())) == 0 {
|
|
return fmt.Errorf("区段[%s]未设置所属集中站", section.Id())
|
|
}
|
|
amdEntry := newAxleManageDevice(w, data, section.CentralizedStation())
|
|
amd := component.AxleManageDeviceType.Get(amdEntry)
|
|
//
|
|
createAxleSectionEntity(w, section, data)
|
|
//
|
|
amd.Adrs[section.Id()] = component.NewAxleDeviceRuntime()
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
func FindAxleManageDevice(data *component.WorldData, centralizedStation string) *ecs.Entry {
|
|
for _, entry := range data.AxleManageDeviceEntities {
|
|
amd := component.AxleManageDeviceType.Get(entry)
|
|
if amd != nil && amd.CentralizedStation == centralizedStation {
|
|
return entry
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 计轴区段实体
|
|
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.PhysicalSectionStateType, component.AxlePhysicalSectionType))
|
|
//
|
|
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
|
component.PhysicalSectionStateType.Set(entry, &component.PhysicalSectionState{Occ: false})
|
|
component.AxlePhysicalSectionType.Set(entry, component.NewAxlePhysicalSection())
|
|
//
|
|
worldData.EntityMap[uid] = entry
|
|
}
|
|
return entry
|
|
}
|