rts-sim-module/sys/device_sys/dbq.go

package device_sys

import (
	"github.com/yohamta/donburi/filter"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/consts"
)

// 断相保护器
type DBQSys struct {
	query *ecs.Query
}

func NewDBQSys() *DBQSys {
	return &DBQSys{
		query: ecs.NewQuery(filter.Contains(component.DBQTag, component.DBQStateType)),
	}
}

func (q *DBQSys) Update(w ecs.World) {
	q.query.Each(w, func(entry *ecs.Entry) {
		// 电路是否通电
		dbq := component.DBQStateType.Get(entry)
		if dbq.Td && !dbq.Dzkg { // 通电但开关未开,启动
			dbq.Dzkg = true
			if entry.HasComponent(component.CounterType) {
				// 定时器启动
				counter := component.CounterType.Get(entry)
				counter.Val = 0
				counter.Step = w.Tick()
			}
		} else if dbq.Td && dbq.Dzkg { // 已经启动，判定延时切断
			if entry.HasComponent(component.CounterType) {
				// 判断定时器是否超时，如果超时，断开电子开关
				counter := component.CounterType.Get(entry)
				if counter.Val >= consts.DBQTimeout {
					counter.Step = 0
					dbq.Dzkg = false
				}
			}
		} else if !dbq.Td && dbq.Dzkg { // 断电，断开电子开关
			dbq.Dzkg = false
			if entry.HasComponent(component.CounterType) { // 断电，停止计时
				counter := component.CounterType.Get(entry)
				counter.Step = 0
			}
		}
	})
}