rts-sim-module/entities/switch_entity.go

package entities

import (
	"github.com/yohamta/donburi/component"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/repository"
	"joylink.club/rtsssimulation/simulation"
	"joylink.club/rtsssimulation/system"
)

type Position int

const (
	SK Position = iota //四开（不会起名字）
	D                  //定位
	F                  //反位
)

// CreateSwitch2jzdj9Entity 双机zdj9
// 默认定位
func CreateSwitch2jzdj9Entity(w ecs.World, switchId string) *ecs.Entry {
	e := w.Create(system.EntityIdentityComponent,
		system.Switch2jZdj9StateComponent,
		system.PercentageDeviceState1Component,
		system.PercentageDeviceState2Component)
	system.EntityIdentityComponent.Set(e, &system.EntityIdentity{Id: switchId})
	//电路
	system.Switch2jZdj9StateComponent.Set(e, system.NewSwitch2jZdj9State())
	//J1
	j1 := system.NewPercentageDeviceStateM() //定位
	system.PercentageDeviceState1Component.Set(e, j1)
	//J2
	j2 := system.NewPercentageDeviceStateM() //定位
	system.PercentageDeviceState2Component.Set(e, j2)
	return e
}

func CreateTurnoutEntries(world ecs.World, turnouts []*repository.Turnout) []*ecs.Entry {
	var entries []*ecs.Entry
	for _, turnout := range turnouts {
		if len(turnout.RelayGroups()) == 0 {
			continue
		}
		var components []component.IComponentType
		components = append(components, system.EntityIdentityComponent)
		components = append(components, system.Switch2jZdj9StateComponent)
		entry := world.Create(components...)
		entries = append(entries, entry)
		system.EntityIdentityComponent.Set(entry, &system.EntityIdentity{Id: turnout.Id()})
		system.Switch2jZdj9StateComponent.Set(entry, system.NewSwitch2jZdj9State())
	}
	return entries
}

func TurnToNormal(worldId ecs.WorldId, turnoutId string) {
	sim := simulation.FindSimulation(worldId)
	turnoutEntry := system.FindEntityById(sim.World(), turnoutId)
	state := system.Switch2jZdj9StateComponent.Get(turnoutEntry)
	state.YCJ = true
	state.DCJ = true
}

func TurnToReverse(worldId ecs.WorldId, turnoutId string) {
	sim := simulation.FindSimulation(worldId)
	turnoutEntry := system.FindEntityById(sim.World(), turnoutId)
	state := system.Switch2jZdj9StateComponent.Get(turnoutEntry)
	state.YCJ = true
	state.FCJ = true
}

func GetState(worldId ecs.WorldId, turnoutId string) Position {
	sim := simulation.FindSimulation(worldId)
	turnoutEntry := system.FindEntityById(sim.World(), turnoutId)
	state := system.Switch2jZdj9StateComponent.Get(turnoutEntry)
	if state.ZDBJ {
		return D
	} else if state.ZFBJ {
		return F
	} else {
		return SK
	}
}