joylink/rts-sim-module
6
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

btm

Browse Source
This commit is contained in:
xzb 2023-11-23 15:12:16 +08:00
parent 8fdc707a97
commit 1d016f6fab
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
entity/train.go
View File

@ -13,3 +13,12 @@ func NewTrainEntity(w ecs.World, trainId string) *ecs.Entry {
data.EntityMap[trainId] = te
return te
}
func NewTrainWithBtmEntity(w ecs.World, trainId string) *ecs.Entry {
data := GetWorldData(w)
te := w.Entry(w.Create(component.UidType, component.TrainPositionInfoType, component.TrainBtmType))
component.UidType.SetValue(te, component.Uid{Id: trainId})
component.TrainPositionInfoType.Set(te, &component.TrainPositionInfo{})
component.TrainBtmType.Set(te, &component.TrainBtm{})
data.EntityMap[trainId] = te
return te
}

2
fi/train.go
View File

@ -13,7 +13,7 @@ func AddTrainToWorld(w ecs.World, trainId string) error {
wd := entity.GetWorldData(w)
_, find := wd.EntityMap[trainId]
if !find {
entity.NewTrainEntity(w, trainId)
entity.NewTrainWithBtmEntity(w, trainId)
}
return ecs.NewOkEmptyResult()
})