166 lines
3.3 KiB
Go
166 lines
3.3 KiB
Go
package repository
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"joylink.club/rtsssimulation/repository/model/proto"
|
||
)
|
||
|
||
type Turnout struct {
|
||
Identity
|
||
|
||
// 岔心的公里标
|
||
km *proto.Kilometer
|
||
|
||
// 道岔关联的道岔物理区段
|
||
section *PhysicalSection
|
||
|
||
// A/B/C端口关联的设备端口
|
||
aDevicePort DevicePort
|
||
bDevicePort DevicePort
|
||
cDevicePort DevicePort
|
||
|
||
// A/B/C端口的区段边界的公里标
|
||
aKm *proto.Kilometer
|
||
bKm *proto.Kilometer
|
||
cKm *proto.Kilometer
|
||
|
||
// A/B/C方向关联的设备（目前有信号机、应答器、(非区段边界)检测点）
|
||
aDevices []Identity
|
||
bDevices []Identity
|
||
cDevices []Identity
|
||
}
|
||
|
||
func NewTurnout(id string, km *proto.Kilometer) *Turnout {
|
||
return &Turnout{
|
||
Identity: identity{id, proto.DeviceType_DeviceType_Turnout},
|
||
km: km,
|
||
}
|
||
}
|
||
|
||
func (t *Turnout) PortNum() int {
|
||
return 3
|
||
}
|
||
|
||
func (t *Turnout) bindPhysicalSection(section *PhysicalSection) {
|
||
t.section = section
|
||
}
|
||
|
||
func (t *Turnout) bindDevicePort(port proto.Port, devicePort DevicePort) error {
|
||
_, isSectionPort := devicePort.(*PhysicalSectionPort)
|
||
_, isTurnoutPort := devicePort.(*TurnoutPort)
|
||
if !isSectionPort && !isTurnoutPort {
|
||
return errors.New(fmt.Sprintf("道岔不能与[%s]类型的设备端口关联", devicePort.Device().Type()))
|
||
}
|
||
switch port {
|
||
case proto.Port_A:
|
||
t.aDevicePort = devicePort
|
||
case proto.Port_B:
|
||
t.bDevicePort = devicePort
|
||
case proto.Port_C:
|
||
t.cDevicePort = devicePort
|
||
default:
|
||
return errors.New(fmt.Sprintf("道岔无端口[%s]", port))
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (t *Turnout) bindBoundaryKm(km *proto.Kilometer, port proto.Port) error {
|
||
switch port {
|
||
case proto.Port_A:
|
||
t.aKm = km
|
||
case proto.Port_B:
|
||
t.bKm = km
|
||
case proto.Port_C:
|
||
t.cKm = km
|
||
default:
|
||
return errors.New(fmt.Sprintf("道岔无端口[%s]", port))
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (t *Turnout) bindDevice(device Identity, port proto.Port) {
|
||
switch port {
|
||
case proto.Port_A:
|
||
t.aDevices = append(t.aDevices, device)
|
||
case proto.Port_B:
|
||
t.bDevices = append(t.bDevices, device)
|
||
case proto.Port_C:
|
||
t.cDevices = append(t.cDevices, device)
|
||
}
|
||
}
|
||
|
||
func (t *Turnout) checkPoints() []*CheckPoint {
|
||
var cps []*CheckPoint
|
||
for _, device := range t.aDevices {
|
||
cp, ok := device.(*CheckPoint)
|
||
if ok {
|
||
cps = append(cps, cp)
|
||
}
|
||
}
|
||
for _, device := range t.bDevices {
|
||
cp, ok := device.(*CheckPoint)
|
||
if ok {
|
||
cps = append(cps, cp)
|
||
}
|
||
}
|
||
for _, device := range t.cDevices {
|
||
cp, ok := device.(*CheckPoint)
|
||
if ok {
|
||
cps = append(cps, cp)
|
||
}
|
||
}
|
||
return cps
|
||
}
|
||
|
||
// 获取指定端口连接的设备端口
|
||
func (t *Turnout) getDevicePortByPort(port proto.Port) DevicePort {
|
||
switch port {
|
||
case proto.Port_A:
|
||
return t.aDevicePort
|
||
case proto.Port_B:
|
||
return t.bDevicePort
|
||
case proto.Port_C:
|
||
return t.cDevicePort
|
||
default:
|
||
return nil
|
||
}
|
||
}
|
||
|
||
func (t *Turnout) findBoundaryKmByPort(port proto.Port) *proto.Kilometer {
|
||
switch port {
|
||
case proto.Port_A:
|
||
return t.aKm
|
||
case proto.Port_B:
|
||
return t.bKm
|
||
case proto.Port_C:
|
||
return t.cKm
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (t *Turnout) getDevicesByPort(port proto.Port) []Identity {
|
||
switch port {
|
||
case proto.Port_A:
|
||
return t.aDevices
|
||
case proto.Port_B:
|
||
return t.bDevices
|
||
case proto.Port_C:
|
||
return t.cDevices
|
||
}
|
||
return nil
|
||
}
|
||
|
||
type TurnoutPort struct {
|
||
turnout *Turnout
|
||
port proto.Port
|
||
}
|
||
|
||
func (t *TurnoutPort) Port() proto.Port {
|
||
return t.port
|
||
}
|
||
|
||
func (t *TurnoutPort) Device() PortedDevice {
|
||
return t.turnout
|
||
}
|