package entity

import (
	"fmt"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/repository"
)

// LoadAxlePhysicalSections 加载计轴区段
func LoadAxlePhysicalSections(w ecs.World) error {
	data := GetWorldData(w)
	sections := data.Repo.PhysicalSectionList()
	for _, section := range sections {
		if is, se := section.IsAxleSection(); se == nil && is {
			fadcDeviceId := createFaDcAxleDeviceId(section.CentralizedStation())
			fadcDeviceEntry, find := data.EntityMap[fadcDeviceId]
			if !find {
				fadcDeviceEntry = createFadcDeviceEntity(w, section, data)
				data.EntityMap[fadcDeviceId] = fadcDeviceEntry
			}
			fadcDevice := component.FaDcAxleDeviceType.Get(fadcDeviceEntry)
			counter := createAxleCounterEntity(w, section, data)
			axleSec := createAxleSectionEntity(w, section, data)
			fadcDevice.AddAxleSection(counter, axleSec)
		}
	}
	return nil
}

// 联锁集中站计轴管理设备实体
func createFadcDeviceEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
	uid := createFaDcAxleDeviceId(axleSection.CentralizedStation())
	entry, ok := worldData.EntityMap[uid]
	if !ok {
		entry = w.Entry(w.Create(component.UidType, component.FaDcAxleDeviceType))
		//
		component.UidType.SetValue(entry, component.Uid{Id: uid})
		component.FaDcAxleDeviceType.Set(entry, &component.FaDcAxleDevice{CounterMap: make(map[string]*ecs.Entry)})
		//
		worldData.EntityMap[uid] = entry
	}
	return entry
}

// 计轴区段实体
func createAxleSectionEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
	uid := axleSection.Id()
	entry, ok := worldData.EntityMap[uid]
	if !ok {
		entry = w.Entry(w.Create(component.UidType, component.AxleSectionType, component.AxleSectionFaultType))
		//
		component.UidType.SetValue(entry, component.Uid{Id: uid})
		component.AxleSectionType.Set(entry, &component.AxleSection{Occ: false})
		component.AxleSectionFaultType.Set(entry, &component.AxleSectionFault{SectionFault: false})
		//
		worldData.EntityMap[uid] = entry
	}
	return entry
}

// 计轴器实体
func createAxleCounterEntity(w ecs.World, axleSection *repository.PhysicalSection, worldData *component.WorldData) *ecs.Entry {
	entry := w.Entry(w.Create(component.AxleCounterType, component.AxleCounterRuntimeType, component.AxleSectionFlag))
	component.AxleCounterType.Set(entry, component.NewAxleCounter())
	component.AxleCounterRuntimeType.Set(entry, &component.AxleCounterRuntime{Rac: false, Rjo: false, Rjt: false, Drst: false, Pdrst: false, DoingPdrst: false})
	component.AxleSectionFlag.SetValue(entry, component.Uid{Id: axleSection.Id()})
	return entry
}

// 设备集中站计轴管理设备id（通过联锁集中站centralizedStation来构建）
func createFaDcAxleDeviceId(centralizedStation string) string {
	return fmt.Sprintf("fadc-axle-device-%s", centralizedStation)
}