rts-sim-module/entity/relay.go

package entity

import (
	"strings"

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

// 创建继电器实体
func NewRelayEntity(w ecs.World, relay *repository.Relay, entityMap map[string]*ecs.Entry) *ecs.Entry {
	uid := relay.Id()
	model := proto.Relay_Model_name[int32(relay.Model())]
	entry, ok := entityMap[uid]
	if !ok {
		entry = w.Entry(w.Create(component.RelayTag, component.UidType, component.RelayDriveType, component.BitStateType))
		component.UidType.SetValue(entry, component.Uid{Id: uid})
		if strings.Contains(model, "Y") { // 有极继电器
			entry.AddComponent(component.YjRelayTag)
		} else if strings.Contains(model, "W") || strings.Contains(model, "Z") || strings.Contains(model, "P") { // 无极继电器
			entry.AddComponent(component.WjRelayTag)
		}
		if strings.Contains(model, "H") { // 缓放继电器
			entry.AddComponent(component.HfRelayTag)
		}
		entityMap[uid] = entry
		//设置继电器初始状态
		switch relay.DefaultPos() {
		case proto.Relay_Pos_Q:
			component.RelayDriveType.Get(entry).Td = true
		case proto.Relay_Pos_H:
			component.RelayDriveType.Get(entry).Td = false
		}
	}
	return entry
}