rts-sim-testing-service/third_party/can_btm/balise_detection.go

package can_btm

import (
	"fmt"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/fi"
	"joylink.club/rtsssimulation/repository"
	"joylink.club/rtsssimulation/repository/model/proto"
	"log/slog"
	"math"
	"sort"
	"sync"
	"time"
)

//通过提前预测来实现BTM天线扫描应答器的实时性
//根据当前列车运行信息预测出列车前方应答器被扫描到的时刻

// BtmAntennaRunningInfo 车载BTM天线中心点位置运行信息
// BTM天线一般在第一车轴后某个位置
type BtmAntennaRunningInfo struct {
	Up           bool    //车载BTM天线中心点运行方向
	LinkId       string  //车载BTM天线中心点所在轨道的id
	LinkOffset   int64   //车载BTM天线中心点所在轨道上的偏移，mm
	Speed        float32 //列车运行速度（m/s）
	Acceleration float32 //加速度(m/s^2)
}

const (
	BtmAntennaOffsetHead = int64(1000) //车载BTM天线距车头端点的距离，mm
)

// TrainHeadPositionInfo 列车车头运行位置信息
type TrainHeadPositionInfo = fi.TrainHeadPositionInfo
type BtmAntennaToBaliseInfo struct {
	Distance int64  //BTM天线中心到应答器的距离，mm
	BaliseId string //应答器id
}
type BtmAntennaScanningBaliseInfo struct {
	BaliseId string    //应答器id
	Time     time.Time //应答器预计被BTM天线激活的时刻
	active   bool      //true-激活过，即列车扫过
	telegram []byte    //应答器报文
}

// BaliseDetector 车载BTM天线，应答器探测器
type BaliseDetector struct {
	eq     [3]*BtmAntennaScanningBaliseInfo //预测将被BTM天线扫描的应答器队列，左边为头
	eqLock sync.Mutex
	//车载应答器天线功率放大器开关，true-开，false-关
	powerAmplifierSwitch bool
	//天线此时是否在应答器上方
	aboveBalise bool
	//应答器计数（每过一个应答器加一，在同一个应答器内不变）
	baliseCounter int
	//报文计数器（每解出一个应答器报文加一,应答器报文长度830bits）
	messageCounter int
	//BTM所在列车id
	trianId string
}

// 由于同一时间只能有一辆列车与CAN BTM绑定
// 当检测到重新绑定列车时，重置相关数据
func (t *BaliseDetector) tryRebind(th *TrainHeadPositionInfo) {
	if th.TrainId != t.trianId {
		t.trianId = th.TrainId
		t.clearExpectedBalise()
		t.baliseCounter = 0
		t.messageCounter = 0
		slog.Debug(fmt.Sprintf("列车[%s]与CAN-BTM绑定", t.trianId))
	}
}

func (t *BaliseDetector) detect(wd *component.WorldData, repo *repository.Repository, th *TrainHeadPositionInfo) {
	t.tryRebind(th)
	//if !t.powerAmplifierSwitch { //天线功率放大器未开启，不进行探测
	//	return
	//}
	curTime := time.Now()
	//BTM天线中心点运行信息
	curAntennaRi := t.createBtmAntennaRunningInfo(wd, repo, th)
	//预测BTM天线到最近一个应答器的时刻
	curExpect := t.timeScanNearestBalise(curTime, wd, repo, curAntennaRi)
	if curExpect != nil && curExpect.Time.UnixMilli()-curTime.UnixMilli() < 20 { //20ms
		//slog.Debug("将要激活应答器", "BaliseId", curExpect.BaliseId, "ActiveTime", dt)
		//记录即将经过的应答器
		if t.addExpectedBalise(curExpect) {
			t.baliseCounterAdd1() //应答器计数器
			telegram := t.rcvTelegram(wd, curExpect.BaliseId)
			if len(telegram) > 0 {
				curExpect.telegram = telegram
				t.baliseMessageCounterAdd1() //报文计数器
			}
		}
		//BTM天线即将经过应答器
		t.aboveBalise = true
	} else {
		t.aboveBalise = false
	}
}

// 应答器计数器加1，[0,255]
func (t *BaliseDetector) baliseCounterAdd1() {
	t.baliseCounter++
	if t.baliseCounter > 255 {
		t.baliseCounter = 0
	}
}

// 报文计数器加1，[0,255]
func (t *BaliseDetector) baliseMessageCounterAdd1() {
	t.messageCounter++
	if t.messageCounter > 255 {
		t.messageCounter = 0
	}
}

// BTM天线接收应答器报文(线程不安全)
func (t *BaliseDetector) rcvTelegram(wd *component.WorldData, baliseId string) []byte {
	entry, ok := wd.EntityMap[baliseId]
	if ok {
		if component.BaliseWorkStateType.Get(entry).Work {
			return component.BaliseFixedTelegramType.Get(entry).Telegram
		}
	}
	return nil
}

// true-新增；false-更新
func (t *BaliseDetector) addExpectedBalise(curExpect *BtmAntennaScanningBaliseInfo) bool {
	//
	t.eqLock.Lock()
	defer t.eqLock.Unlock()
	////更新,去重
	//for i, e := range t.eq {
	//	if e != nil && e.BaliseId == curExpect.BaliseId {
	//		t.eq[i].Time = curExpect.Time
	//		return false
	//	}
	//}
	//检查是否已经记录过
	eq := t.eq[len(t.eq)-1]
	if eq != nil && eq.BaliseId == curExpect.BaliseId {
		return false
	}
	//左移
	for i := 1; i < len(t.eq); i++ {
		t.eq[i-1] = t.eq[i]
	}
	//存入队尾
	t.eq[len(t.eq)-1] = curExpect
	return true
}
func (t *BaliseDetector) clearExpectedBalise() {
	//
	t.eqLock.Lock()
	defer t.eqLock.Unlock()
	//
	for i := 0; i < len(t.eq); i++ {
		t.eq[i] = nil
	}
}
func (t *BaliseDetector) doScan() *BtmAntennaScanningBaliseInfo {
	//
	t.eqLock.Lock()
	defer t.eqLock.Unlock()
	//
	var rt *BtmAntennaScanningBaliseInfo
	for i := 0; i < len(t.eq); i++ {
		if t.eq[i] != nil && !t.eq[i].active {
			rt = t.eq[i]
			t.eq[i].active = true
			break
		}
	}
	return rt
}

// 计算列车在当前运行状态下，预测到最近一个应答器的时刻
func (t *BaliseDetector) timeScanNearestBalise(curTime time.Time, wd *component.WorldData, repo *repository.Repository, ba *BtmAntennaRunningInfo) *BtmAntennaScanningBaliseInfo {
	expectedBalise := t.findBaliseWillScanByBtmAntenna(wd, repo, ba)
	if expectedBalise != nil {
		curV := float64(ba.Speed)
		curAc := float64(ba.Acceleration)
		s := float64(expectedBalise.Distance) / 1000
		st, ok := t.calculateBtmAntennaScanNextBaliseTime(curTime, curV, curAc, s)
		if ok {
			return &BtmAntennaScanningBaliseInfo{BaliseId: expectedBalise.BaliseId, Time: st}
		}
	}
	return nil
}

// 预测BTM天线到运行方向的最近应答器的时刻
// curV-当前时刻BTM天线速度,m/s
// curAc-当前时刻BTM天线加速度,m/s^2
// s-BTM天线从当前时刻所处位置到运行方向最近一个应答器的位移,m
func (t *BaliseDetector) calculateBtmAntennaScanNextBaliseTime(curTime time.Time, curV float64, curAc float64, s float64) (time.Time, bool) {
	a := 0.5 * curAc
	b := curV
	c := -s
	//nt 单位秒
	nt, ok := t.calculateQuadratic(a, b, c)
	if ok {
		return curTime.Add(time.Millisecond * time.Duration(nt*1000)), true
	}
	return curTime, false
}

// 获取车载BTM天线中心点运行方向最近的1个应答器
func (t *BaliseDetector) findBaliseWillScanByBtmAntenna(wd *component.WorldData, repo *repository.Repository, ba *BtmAntennaRunningInfo) *BtmAntennaToBaliseInfo {
	//BTM天线中心点所在轨道
	baLink := repo.FindLink(ba.LinkId)
	rs1 := t.searchBalisesFromLinkPosition(repo, ba.LinkId, ba.Up, ba.LinkOffset)
	if ba.Up {
		if len(rs1) > 0 {
			return &BtmAntennaToBaliseInfo{BaliseId: rs1[0].Id(), Distance: rs1[0].LinkPosition().Offset() - ba.LinkOffset}
		} else {
			nextLinkPort := t.getNextLink(wd, repo, ba.LinkId, ba.Up)
			if nextLinkPort != nil {
				if nextLinkPort.IsPortA() {
					rs2 := t.searchBalisesFromLinkPosition(repo, nextLinkPort.Link().Id(), true, 0)
					if len(rs2) > 0 {
						return &BtmAntennaToBaliseInfo{BaliseId: rs2[0].Id(), Distance: baLink.Length() - ba.LinkOffset + rs2[0].LinkPosition().Offset()}
					}
				} else {
					rs2 := t.searchBalisesFromLinkPosition(repo, nextLinkPort.Link().Id(), false, nextLinkPort.Link().Length())
					if len(rs2) > 0 {
						return &BtmAntennaToBaliseInfo{BaliseId: rs2[0].Id(), Distance: baLink.Length() - ba.LinkOffset + nextLinkPort.Link().Length() - rs2[0].LinkPosition().Offset()}
					}
				}
			}
		}
	} else {
		if len(rs1) > 0 {
			return &BtmAntennaToBaliseInfo{BaliseId: rs1[0].Id(), Distance: ba.LinkOffset - rs1[0].LinkPosition().Offset()}
		} else {
			nextLinkPort := t.getNextLink(wd, repo, ba.LinkId, ba.Up)
			if nextLinkPort != nil {
				if nextLinkPort.IsPortA() {
					rs2 := t.searchBalisesFromLinkPosition(repo, nextLinkPort.Link().Id(), true, 0)
					if len(rs2) > 0 {
						return &BtmAntennaToBaliseInfo{BaliseId: rs2[0].Id(), Distance: ba.LinkOffset + rs2[0].LinkPosition().Offset()}
					}
				} else {
					rs2 := t.searchBalisesFromLinkPosition(repo, nextLinkPort.Link().Id(), false, nextLinkPort.Link().Length())
					if len(rs2) > 0 {
						return &BtmAntennaToBaliseInfo{BaliseId: rs2[0].Id(), Distance: ba.LinkOffset + nextLinkPort.Link().Length() - rs2[0].LinkPosition().Offset()}
					}
				}
			}
		}
	}
	return nil
}

// up-在轨道上的搜索方向
func (t *BaliseDetector) searchBalisesFromLinkPosition(repo *repository.Repository, linkId string, up bool, fromOffset int64) []*repository.Transponder {
	rs := repo.ResponderListByLink(linkId)
	if up {
		sort.SliceStable(rs, func(i, j int) bool {
			return rs[i].LinkPosition().Offset() < rs[j].LinkPosition().Offset()
		})
		//
		for i, r := range rs {
			if r.LinkPosition().Offset() >= fromOffset {
				return rs[i:]
			}
		}
	} else {
		sort.SliceStable(rs, func(i, j int) bool {
			return rs[j].LinkPosition().Offset() < rs[i].LinkPosition().Offset()
		})
		for i, r := range rs {
			if r.LinkPosition().Offset() <= fromOffset {
				return rs[i:]
			}
		}
	}
	return nil
}

// 列车车头端点运行信息转换为车载BTM天线中心点运行信息
func (t *BaliseDetector) createBtmAntennaRunningInfo(wd *component.WorldData, repo *repository.Repository, head *TrainHeadPositionInfo) *BtmAntennaRunningInfo {
	headLink := repo.FindLink(head.Link)
	if head.Up {
		if head.LinkOffset >= BtmAntennaOffsetHead { //车头与BTM天线在同一个轨道上
			return &BtmAntennaRunningInfo{Up: head.Up, LinkId: head.Link, LinkOffset: head.LinkOffset - BtmAntennaOffsetHead, Speed: head.Speed, Acceleration: head.Acceleration}
		} else { //车头与BTM天线在同一个轨道上
			nextLinkPort := t.getNextLink(wd, repo, head.Link, !head.Up)
			nextLink := nextLinkPort.Link()
			if nextLinkPort.IsPortA() {
				return &BtmAntennaRunningInfo{Up: false, LinkId: nextLink.Id(), LinkOffset: BtmAntennaOffsetHead - head.LinkOffset, Speed: head.Speed, Acceleration: head.Acceleration}
			} else {
				return &BtmAntennaRunningInfo{Up: true, LinkId: nextLink.Id(), LinkOffset: nextLink.Length() - (BtmAntennaOffsetHead - head.LinkOffset), Speed: head.Speed, Acceleration: head.Acceleration}
			}
		}
	} else {
		if headLink.Length()-head.LinkOffset >= BtmAntennaOffsetHead { //车头与BTM天线在同一个轨道上
			return &BtmAntennaRunningInfo{Up: head.Up, LinkId: head.Link, LinkOffset: head.LinkOffset + BtmAntennaOffsetHead, Speed: head.Speed, Acceleration: head.Acceleration}
		} else {
			nextLinkPort := t.getNextLink(wd, repo, head.Link, !head.Up)
			nextLink := nextLinkPort.Link()
			if nextLinkPort.IsPortA() {
				return &BtmAntennaRunningInfo{Up: false, LinkId: nextLink.Id(), LinkOffset: BtmAntennaOffsetHead - headLink.Length() + head.LinkOffset, Speed: head.Speed, Acceleration: head.Acceleration}
			} else {
				return &BtmAntennaRunningInfo{Up: true, LinkId: nextLink.Id(), LinkOffset: nextLink.Length() - (BtmAntennaOffsetHead - headLink.Length() + head.LinkOffset), Speed: head.Speed, Acceleration: head.Acceleration}
			}
		}
	}
}
func (t *BaliseDetector) getNextLink(wd *component.WorldData, repo *repository.Repository, curLinkId string, searchDirection bool) *repository.LinkPort {
	var nextLinkPort *repository.LinkPort
	curLink := repo.FindLink(curLinkId)
	if searchDirection {
		bDc := curLink.BRelation()
		if bDc == nil || bDc.Turnout() == nil {
			return nil
		}
		//
		bDcDw := t.turnoutPosition(wd, bDc.Turnout().Id())
		switch bDc.Port() {
		case proto.Port_A: //link-b连接turnout-A
			if bDcDw {
				nextLinkPort = bDc.Turnout().FindLinkByPort(proto.Port_B)
			} else {
				nextLinkPort = bDc.Turnout().FindLinkByPort(proto.Port_C)
			}
		case proto.Port_B: //link-b连接turnout-B
			fallthrough
		case proto.Port_C: //link-b连接turnout-C
			nextLinkPort = bDc.Turnout().FindLinkByPort(proto.Port_A)
		}
	} else {
		aDc := curLink.ARelation()
		if aDc == nil || aDc.Turnout() == nil {
			return nil
		}
		//
		aDcDw := t.turnoutPosition(wd, aDc.Turnout().Id())
		switch aDc.Port() {
		case proto.Port_A: //link-a连接turnout-A
			if aDcDw {
				nextLinkPort = aDc.Turnout().FindLinkByPort(proto.Port_B)
			} else {
				nextLinkPort = aDc.Turnout().FindLinkByPort(proto.Port_C)
			}
		case proto.Port_B: //link-a连接turnout-B
			fallthrough
		case proto.Port_C: //link-a连接turnout-C
			nextLinkPort = aDc.Turnout().FindLinkByPort(proto.Port_A)
		}
	}
	//
	return nextLinkPort
}

// 获取道岔实际位置(线程不安全)
func (t *BaliseDetector) turnoutPosition(wd *component.WorldData, id string) bool {
	entry, ok := wd.EntityMap[id]
	if ok {
		return component.TurnoutPositionType.Get(entry).Dw
	} else {
		panic(fmt.Sprintf("道岔[%s]的实体不存在", id))
	}
}

// 解 ax²+bx+c=0
func (t *BaliseDetector) calculateQuadratic(a, b, c float64) (float64, bool) {
	//D=b²-4ac
	d := b*b - 4*a*c
	if d < 0 {
		return 0, false
	} else if d == 0 { //若D=0，则计算它的重根：x=-b/2a
		return -(b / (2 * a)), true
	} else { //若D>0，则计算它的两个根：x1=(-b+√D)/2a，x2=(-b-√D)/2a
		sqrtD := math.Sqrt(d)
		ax2 := 2 * a
		//选择正直输出
		x1 := (-b + sqrtD) / ax2
		x2 := (-b - sqrtD) / ax2
		if x1 >= 0 && x2 >= 0 {
			//panic("x1 >= 0 && x2 >= 0")
			return 0, false
		}
		if x1 >= 0 {
			return x1, true
		}
		if x2 >= 0 {
			return x2, true
		}
	}
	return 0, false
}