48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package entity
|
|
|
|
import (
|
|
"joylink.club/ecs"
|
|
"joylink.club/rtsssimulation/component"
|
|
"joylink.club/rtsssimulation/component/component_proto"
|
|
"joylink.club/rtsssimulation/repository"
|
|
"joylink.club/rtsssimulation/repository/model/proto"
|
|
)
|
|
|
|
// LoadBalises 加载应答器实体
|
|
func LoadBalises(w ecs.World) error {
|
|
data := GetWorldData(w)
|
|
balises := data.Repo.TransponderList()
|
|
for _, b := range balises {
|
|
be := newBaliseEntity(w, b, data)
|
|
switch b.TransponderType() {
|
|
case proto.Transponder_FB:
|
|
be.AddComponent(component.BaliseFB)
|
|
case proto.Transponder_WB:
|
|
be.AddComponent(component.BaliseWB)
|
|
case proto.Transponder_DB:
|
|
be.AddComponent(component.BaliseDB)
|
|
case proto.Transponder_VB:
|
|
be.AddComponent(component.BaliseVB)
|
|
case proto.Transponder_IB:
|
|
be.AddComponent(component.BaliseIB)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
func newBaliseEntity(w ecs.World, td *repository.Transponder, worldData *component.WorldData) *ecs.Entry {
|
|
uid := td.Id()
|
|
entry, ok := worldData.EntityMap[uid]
|
|
if !ok {
|
|
entry = w.Entry(w.Create(component.UidType, component.BaliseStateType, component.LinkPositionType, component.KmType))
|
|
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
|
component.BaliseStateType.Set(entry, &component.BaliseState{})
|
|
component.LinkPositionType.SetValue(entry, component_proto.LinkPosition{
|
|
LinkId: td.LinkPosition().Link().Id(),
|
|
Offset: td.LinkPosition().Offset(),
|
|
})
|
|
component.KmType.Set(entry, td.Km())
|
|
worldData.EntityMap[uid] = entry
|
|
}
|
|
return entry
|
|
}
|