package iscs_sys

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

// WaterPumpSystem 水泵
// 如：冷冻水泵、冷却水循环泵、超声波型水泵、浮球型水泵
// 注：浮球型水泵通过浮球来控制水泵的开启关闭
type WaterPumpSystem struct {
	query *ecs.Query
}

func NewWaterPumpSystem() *WaterPumpSystem {
	return &WaterPumpSystem{
		query: ecs.NewQuery(filter.Contains(component.WaterPumpType, component.FluidDriverType, component.DeviceExceptionType)),
	}
}
func (s *WaterPumpSystem) Update(w ecs.World) {
	s.query.Each(w, func(entry *ecs.Entry) {
		pump := component.WaterPumpType.Get(entry)
		fluidDriver := component.FluidDriverType.Get(entry)
		//浮球型水泵，水箱水位控制浮球开关从而控制水泵开关
		if entry.HasComponent(component.FloatBallSwitchType) {
			fbs := component.FloatBallSwitchType.Get(entry)
			pump.Running = fbs.On
		}
		//
		de := component.DeviceExceptionType.Get(entry)
		//异常停止运行
		if de.Exception != consts.DeviceExceptionNon {
			pump.Running = false
		}
		//
		fluidDriver.On = pump.Running
	})
}