rts-sim-module/sys/circuit_sys/signal_3xh3.go

package circuit_sys

import (
	"joylink.club/ecs"
	"joylink.club/ecs/filter"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/entity"
)

type Signal3XH3System struct {
	query *ecs.Query
}

func NewSignal3XH3System() *Signal3XH3System {
	return &Signal3XH3System{query: ecs.NewQuery(filter.Contains(
		component.Signal3XH3ElectronicType,
		component.Signal3XH3LsqType,
		component.Signal3XH3LscType,
		component.SignalLightsType))}
}

// Update world 执行
func (s *Signal3XH3System) Update(w ecs.World) {
	wd := entity.GetWorldData(w)
	s.query.Each(w, func(entry *ecs.Entry) {
		circuit := component.Signal3XH3ElectronicType.Get(entry)
		excite3xh3ByCiDrive(circuit, wd)
		//lsq := component.Signal3XH3LsqType.Get(entry)
		//lsc := component.Signal3XH3LscType.Get(entry)
		lights := component.SignalLightsType.Get(entry)
		Z3XH3_H := lights.GetLightByTag(component.HdTag)
		Z3XH3_U := lights.GetLightByTag(component.UdTag)

		//s.calculateLsq(circuit, lsq)
		s.calculateU(circuit, Z3XH3_U)
		s.calculateH(circuit, Z3XH3_H)
		s.calculateDJ(circuit, Z3XH3_U, Z3XH3_H)
		s.calculate2DJ(circuit, Z3XH3_U)
		//s.calculateLsc(circuit, lsc)
	})
}

func excite3xh3ByCiDrive(circuit *component.Signal3XH3Electronic, wd *component.WorldData) {
	bit, err := wd.QueryQdBit(component.UidType.Get(circuit.Z3XH3_DDJ).Id)
	if err == nil {
		component.RelayDriveType.Get(circuit.Z3XH3_DDJ).Td = bit
	}

	bit, err = wd.QueryQdBit(component.UidType.Get(circuit.Z3XH3_LXJ).Id)
	if err == nil {
		component.RelayDriveType.Get(circuit.Z3XH3_LXJ).Td = bit
	}

	bit, err = wd.QueryQdBit(component.UidType.Get(circuit.Z3XH3_YXJ).Id)
	if err == nil {
		component.RelayDriveType.Get(circuit.Z3XH3_YXJ).Td = bit
	}
}

// 联锁驱
func (s *Signal3XH3System) calculateLsq(state *component.Signal3XH3Electronic, lsq *component.Signal3XH3Lsq) {
	ddj := component.RelayDriveType.Get(state.Z3XH3_DDJ)
	lxj := component.RelayDriveType.Get(state.Z3XH3_LXJ)
	yxj := component.RelayDriveType.Get(state.Z3XH3_YXJ)
	//
	ddjQ := lsq.Z3XH3_DDJ_Q
	ddj.Td = ddjQ
	ddj.Xq = ddjQ
	//
	lxjQ := lsq.Z3XH3_LXJ_Q
	lxj.Td = lxjQ
	lxj.Xq = lxjQ
	//
	yxjQ := lsq.Z3XH3_YXJ_Q
	yxj.Td = yxjQ
	yxj.Xq = yxjQ
}

// 联锁采
func (s *Signal3XH3System) calculateLsc(state *component.Signal3XH3Electronic, lsc *component.Signal3XH3Lsc) {
	ddj := component.BitStateType.Get(state.Z3XH3_DDJ)
	lxj := component.BitStateType.Get(state.Z3XH3_LXJ)
	dj := component.BitStateType.Get(state.Z3XH3_DJ)
	edj := component.BitStateType.Get(state.Z3XH3_2DJ)
	yxj := component.BitStateType.Get(state.Z3XH3_YXJ)
	//
	lsc.Z3XH3_2DJ_Xq = edj.Val
	lsc.Z3XH3_DJ_Xq = dj.Val
	lsc.Z3XH3_DDJ_Lx = !ddj.Val
	lsc.Z3XH3_LXJ_Xq = lxj.Val
	lsc.Z3XH3_YXJ_Xq = yxj.Val
}
func (s *Signal3XH3System) calculateU(state *component.Signal3XH3Electronic, Z3XH3_U *ecs.Entry) {
	ddj := component.BitStateType.Get(state.Z3XH3_DDJ)
	lxj := component.BitStateType.Get(state.Z3XH3_LXJ)
	dj := component.BitStateType.Get(state.Z3XH3_DJ)
	yxj := component.BitStateType.Get(state.Z3XH3_YXJ)
	isU := !ddj.Val && !lxj.Val && dj.Val && yxj.Val || !ddj.Val && lxj.Val
	driveU := component.LightDriveType.Get(Z3XH3_U)
	driveU.Td = isU

}
func (s *Signal3XH3System) calculateH(state *component.Signal3XH3Electronic, Z3XH3_H *ecs.Entry) {
	ddj := component.BitStateType.Get(state.Z3XH3_DDJ)
	lxj := component.BitStateType.Get(state.Z3XH3_LXJ)
	isH := !ddj.Val && !lxj.Val
	driveH := component.LightDriveType.Get(Z3XH3_H)
	driveH.Td = isH
}
func (s *Signal3XH3System) calculateDJ(state *component.Signal3XH3Electronic, Z3XH3_U *ecs.Entry, Z3XH3_H *ecs.Entry) {
	ddj := component.BitStateType.Get(state.Z3XH3_DDJ)
	lxj := component.BitStateType.Get(state.Z3XH3_LXJ)
	ud := component.BitStateType.Get(Z3XH3_U)
	hd := component.BitStateType.Get(Z3XH3_H)
	isDJ := ud.Val && !ddj.Val && lxj.Val || hd.Val && !ddj.Val && !lxj.Val
	drive := component.RelayDriveType.Get(state.Z3XH3_DJ)
	drive.Td = isDJ
	drive.Xq = isDJ
}
func (s *Signal3XH3System) calculate2DJ(state *component.Signal3XH3Electronic, Z3XH3_U *ecs.Entry) {
	ddj := component.BitStateType.Get(state.Z3XH3_DDJ)
	lxj := component.BitStateType.Get(state.Z3XH3_LXJ)
	dj := component.BitStateType.Get(state.Z3XH3_DJ)
	yxj := component.BitStateType.Get(state.Z3XH3_YXJ)
	ud := component.BitStateType.Get(Z3XH3_U)
	is2DJ := ud.Val && !ddj.Val && !lxj.Val && dj.Val && yxj.Val
	drive := component.RelayDriveType.Get(state.Z3XH3_2DJ)
	drive.Td = is2DJ
	drive.Xq = is2DJ
}