40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
|
package iscs_sys
|
||
|
|
||
|
import (
|
||
|
"joylink.club/ecs"
|
||
|
"joylink.club/ecs/filter"
|
||
|
"joylink.club/rtsssimulation/component"
|
||
|
"joylink.club/rtsssimulation/entity"
|
||
|
"joylink.club/rtsssimulation/repository"
|
||
|
)
|
||
|
|
||
|
type LightningArresterSystem struct {
|
||
|
query *ecs.Query
|
||
|
}
|
||
|
|
||
|
func NewLightningArresterSystem() *LightningArresterSystem {
|
||
|
return &LightningArresterSystem{
|
||
|
query: ecs.NewQuery(filter.Contains(component.UidType, component.LightningArresterType)),
|
||
|
}
|
||
|
}
|
||
|
func (s *LightningArresterSystem) Update(w ecs.World) {
|
||
|
wd := entity.GetWorldData(w)
|
||
|
s.query.Each(w, func(entry *ecs.Entry) {
|
||
|
s.lightningArresterTransPower(wd, entry)
|
||
|
})
|
||
|
|
||
|
}
|
||
|
|
||
|
// 避雷器传递电能
|
||
|
func (s *LightningArresterSystem) lightningArresterTransPower(wd *component.WorldData, entry *ecs.Entry) {
|
||
|
arresterId := component.UidType.Get(entry).Id
|
||
|
closed := true
|
||
|
arresterModel := (wd.Repo.FindById(arresterId)).(*repository.LightningArrester)
|
||
|
//避雷器A端连接的管线
|
||
|
arresterPortA := arresterModel.PortA
|
||
|
//避雷器B端连接的管线
|
||
|
arresterPortB := arresterModel.PortB
|
||
|
//传递电能
|
||
|
towPipePortsTransPower(wd, closed, arresterPortA, arresterPortB)
|
||
|
}
|