rts-sim-module/entity/balise.go

package entity

import (
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/component/component_data"
	"joylink.club/rtsssimulation/repository"
	"joylink.club/rtsssimulation/repository/model/proto"
)

var BaliseBaseComponentTypeArr = []ecs.IComponentType{component.UidType, component.BaliseFixedTelegramType,
	component.LinkPositionType, component.KmType, component.BaliseWorkStateType}

// LoadBalises 加载应答器实体
func LoadBalises(w ecs.World) error {
	data := GetWorldData(w)
	balises := data.Repo.ResponderList()
	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(BaliseBaseComponentTypeArr...))
		component.UidType.SetValue(entry, component.Uid{Id: uid})
		component.BaliseWorkStateType.SetValue(entry, component.BaliseWorkState{Work: true})
		component.BaliseFixedTelegramType.Set(entry, &component.BaliseTelegram{
			Telegram:     td.FixedTelegram(),
			UserTelegram: td.FixedUserTelegram(),
		})
		if proto.Transponder_IB == td.BaliseType() || proto.Transponder_VB == td.BaliseType() {
			entry.AddComponent(component.BaliseVariableTelegramType)
		}
		linkPosition := td.LinkPosition()
		component.LinkPositionType.SetValue(entry, component_data.LinkPosition{
			LinkId: linkPosition.Link().Id(),
			Offset: linkPosition.Offset(),
		})
		component.KmType.Set(entry, td.Km())
		worldData.EntityMap[uid] = entry
	}
	return entry
}