213 lines
6.6 KiB
Go
213 lines
6.6 KiB
Go
package entity
|
||
|
||
import (
|
||
"fmt"
|
||
"log/slog"
|
||
"strings"
|
||
"unsafe"
|
||
|
||
"joylink.club/ecs"
|
||
"joylink.club/rtsssimulation/component"
|
||
"joylink.club/rtsssimulation/repository"
|
||
"joylink.club/rtsssimulation/repository/model/proto"
|
||
)
|
||
|
||
// 创建按钮实体
|
||
func NewButtonEntity(w ecs.World, data *repository.Button, entityMap map[string]*ecs.Entry) *ecs.Entry {
|
||
uid := data.Id()
|
||
entry, ok := entityMap[uid]
|
||
if !ok {
|
||
var btnType ecs.IComponentType
|
||
switch data.GetBtnType() {
|
||
case proto.Button_NO_Reset_Press:
|
||
btnType = component.NoResetPressBtn
|
||
case proto.Button_NO_Reset_Up:
|
||
btnType = component.NoResetUpBtn
|
||
case proto.Button_Reset_Press:
|
||
btnType = component.ResetPressBtn
|
||
case proto.Button_Reset_Up:
|
||
btnType = component.ResetUpBtn
|
||
case proto.Button_Key_Knob:
|
||
btnType = component.KeyKnob
|
||
default:
|
||
btnType = component.UnknowBtn
|
||
}
|
||
entry = w.Entry(w.Create(component.ButtonTag, btnType, component.UidType, component.BitStateType))
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
// 带灯按钮
|
||
if data.HasLight() {
|
||
entry.AddComponent(component.SingleLightType, unsafe.Pointer(&component.SingleLight{Light: NewLightEntity(w)}))
|
||
}
|
||
entityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 创建蜂鸣器实体
|
||
func NewAlarmEntity(w ecs.World, alarm *repository.Alarm, entityMap map[string]*ecs.Entry) *ecs.Entry {
|
||
uid := alarm.Id()
|
||
entry, ok := entityMap[uid]
|
||
if !ok {
|
||
entry = w.Entry(w.Create(component.AlarmTag, component.UidType, component.AlarmDriveType, component.BitStateType))
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
entityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 创建IBP灯实体
|
||
func NewIBPLightEntity(w ecs.World, uid string, entityMap map[string]*ecs.Entry) *ecs.Entry {
|
||
entry, ok := entityMap[uid]
|
||
if !ok {
|
||
entry = w.Entry(w.Create(component.LightTag, component.UidType, component.LightDriveType, component.BitStateType))
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
entityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 创建IBP钥匙实体
|
||
func NewIBPKeyEntity(w ecs.World, uid string, entityMap map[string]*ecs.Entry) *ecs.Entry {
|
||
entry, ok := entityMap[uid]
|
||
if !ok {
|
||
entry = w.Entry(w.Create(component.KeyTag, component.UidType, component.GearStateType))
|
||
component.UidType.SetValue(entry, component.Uid{Id: uid})
|
||
entityMap[uid] = entry
|
||
}
|
||
return entry
|
||
}
|
||
|
||
// 构建人员防护实体
|
||
func LoadSPKEntity(w ecs.World, entry *ecs.Entry, datas []repository.IGroupedElectronicComponent, worldData *component.WorldData) error {
|
||
if len(datas) == 0 {
|
||
return nil
|
||
}
|
||
entityMap := worldData.EntityMap
|
||
spk := &component.SpkElectronic{}
|
||
for _, c := range datas {
|
||
switch c.Code() {
|
||
// 按钮组
|
||
case "SPKSXPLA":
|
||
spk.SPKSXPLA_BTN = NewButtonEntity(w, c.(*repository.Button), entityMap)
|
||
case "SPKSSPLA":
|
||
spk.SPKSSPLA_BTN = NewButtonEntity(w, c.(*repository.Button), entityMap)
|
||
case "SDA":
|
||
spk.SDA = NewButtonEntity(w, c.(*repository.Button), entityMap)
|
||
// 钥匙组、继电器
|
||
case "SPKSX1J":
|
||
r, ok := c.(*repository.Relay)
|
||
if ok {
|
||
spk.SPKSX1J = NewRelayEntity(w, r, entityMap)
|
||
} else {
|
||
spk.SPKSX1J_KEY = NewIBPKeyEntity(w, c.Id(), entityMap)
|
||
}
|
||
case "SPKSX3J":
|
||
r, ok := c.(*repository.Relay)
|
||
if ok {
|
||
spk.SPKSX3J = NewRelayEntity(w, r, entityMap)
|
||
} else {
|
||
spk.SPKSX3J_KEY = NewIBPKeyEntity(w, c.Id(), entityMap)
|
||
}
|
||
case "SPKSS2J":
|
||
r, ok := c.(*repository.Relay)
|
||
if ok {
|
||
spk.SPKSS2J = NewRelayEntity(w, r, entityMap)
|
||
} else {
|
||
spk.SPKSS2J_KEY = NewIBPKeyEntity(w, c.Id(), entityMap)
|
||
}
|
||
case "SPKSS4J":
|
||
r, ok := c.(*repository.Relay)
|
||
if ok {
|
||
spk.SPKSS4J = NewRelayEntity(w, r, entityMap)
|
||
} else {
|
||
spk.SPKSS4J_KEY = NewIBPKeyEntity(w, c.Id(), entityMap)
|
||
}
|
||
case "SPKSXPLAJ":
|
||
spk.SPKSXPLAJ = NewRelayEntity(w, c.(*repository.Relay), entityMap)
|
||
case "SPKSSPLAJ":
|
||
spk.SPKSSPLAJ = NewRelayEntity(w, c.(*repository.Relay), entityMap)
|
||
// 灯组
|
||
case "SPKSX1D":
|
||
spk.SPKSX1D = NewIBPLightEntity(w, c.Id(), entityMap)
|
||
case "SPKSX3D":
|
||
spk.SPKSX3D = NewIBPLightEntity(w, c.Id(), entityMap)
|
||
case "SPKSS2D":
|
||
spk.SPKSS2D = NewIBPLightEntity(w, c.Id(), entityMap)
|
||
case "SPKSS4D":
|
||
spk.SPKSS4D = NewIBPLightEntity(w, c.Id(), entityMap)
|
||
}
|
||
}
|
||
|
||
// 绑定组件
|
||
entry.AddComponent(component.SpkElectronicType, unsafe.Pointer(spk))
|
||
return nil
|
||
}
|
||
|
||
// 构建紧急关闭实体
|
||
func LoadEMPEntity(w ecs.World, entry *ecs.Entry, datas []*repository.ElectronicComponentGroup, worldData *component.WorldData) error {
|
||
if len(datas) == 0 {
|
||
return nil
|
||
}
|
||
entityMap := worldData.EntityMap
|
||
emp := &component.EmpElectronic{EMPJMap: make(map[string]*component.EmpDeviceElectronic)}
|
||
for _, g := range datas {
|
||
if g.Code() == "EMP" { // 独立设备
|
||
for _, c := range g.Components() {
|
||
switch c.Code() {
|
||
case "QBA":
|
||
emp.QBA = NewButtonEntity(w, c.(*repository.Button), entityMap)
|
||
case "SDA":
|
||
emp.SDA = NewButtonEntity(w, c.(*repository.Button), entityMap)
|
||
case "蜂鸣器":
|
||
emp.Alarm = NewAlarmEntity(w, c.(*repository.Alarm), entityMap)
|
||
default:
|
||
r, ok := c.(*repository.Relay)
|
||
if ok { // 如果是继电器
|
||
d := emp.EMPJMap[r.Code()]
|
||
if d == nil {
|
||
d = &component.EmpDeviceElectronic{}
|
||
}
|
||
d.EMPJ = NewRelayEntity(w, r, entityMap)
|
||
emp.EMPJMap[r.Code()] = d
|
||
} else {
|
||
slog.Warn("未实现设备[id:%s]逻辑", c.Id())
|
||
}
|
||
}
|
||
}
|
||
} else { // 组合设备
|
||
d := emp.EMPJMap[g.Code()]
|
||
if d == nil {
|
||
d = &component.EmpDeviceElectronic{}
|
||
}
|
||
for _, c := range g.Components() {
|
||
b, ok := c.(*repository.Button)
|
||
if strings.HasSuffix(c.Code(), "EMPD") {
|
||
d.EMPD = NewIBPLightEntity(w, c.Id(), entityMap)
|
||
} else if ok && strings.HasSuffix(c.Code(), "EMPFA") {
|
||
d.EMPFA_BTN = NewButtonEntity(w, b, entityMap)
|
||
} else if ok && (strings.HasPrefix(c.Code(), "EMP") || strings.HasPrefix(c.Code(), "ESB")) {
|
||
d.EMP_BTNS = append(d.EMP_BTNS, NewButtonEntity(w, b, entityMap))
|
||
} else {
|
||
slog.Warn("未实现设备[id:%s]逻辑", c.Id())
|
||
}
|
||
}
|
||
emp.EMPJMap[g.Code()] = d
|
||
}
|
||
}
|
||
if emp.Alarm == nil || emp.QBA == nil || emp.SDA == nil {
|
||
slog.Warn("EMP组合缺少元素，将不添加EMP电路组件")
|
||
return nil
|
||
}
|
||
for code, e := range emp.EMPJMap {
|
||
if e.EMPFA_BTN == nil || e.EMPD == nil || len(e.EMP_BTNS) == 0 {
|
||
slog.Warn(fmt.Sprintf("组合[%s]还原按钮未关联,请检查IBP地图组合数据", code))
|
||
return nil
|
||
} else if e.EMPJ == nil {
|
||
slog.Warn(fmt.Sprintf("组合[%s]继电器未关联,请检查继电器地图组合数据", code))
|
||
return nil
|
||
}
|
||
}
|
||
entry.AddComponent(component.EmpElectronicType, unsafe.Pointer(emp))
|
||
return nil
|
||
}
|