robocars/robocar-tools
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[road] Fix contour road display

Browse Source
This commit is contained in:
Cyrille Nofficial 2020-01-19 17:15:36 +01:00
parent 135c69ddc2
commit 11c3c4139b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
part/part.go
View File

@ -30,7 +30,7 @@ func NewPart(client mqtt.Client, frameTopic, objectsTopic, roadTopic string, wit
} }
type FramePart struct { type FramePart struct {
client mqtt.Client client mqtt.Client
frameTopic, objectsTopic, roadTopic string frameTopic, objectsTopic, roadTopic string
window *gocv.Window window *gocv.Window
@ -165,6 +165,9 @@ func (p *FramePart) drawObjects(img *gocv.Mat, objects *events.ObjectsMessage) {
func (p *FramePart) drawRoad(img *gocv.Mat, road *events.RoadMessage) { func (p *FramePart) drawRoad(img *gocv.Mat, road *events.RoadMessage) {
cntr := make([]image.Point, 0, len(road.GetContour())) cntr := make([]image.Point, 0, len(road.GetContour()))
if road.GetContour() == nil || len(road.GetContour()) < 3 {
return
}
for _, pt := range road.GetContour() { for _, pt := range road.GetContour() {
cntr = append(cntr, image.Point{X: int(pt.GetX()), Y: int(pt.GetY())}) cntr = append(cntr, image.Point{X: int(pt.GetX()), Y: int(pt.GetY())})
} }
@ -175,6 +178,7 @@ func (p *FramePart) drawRoad(img *gocv.Mat, road *events.RoadMessage) {
0, 0,
color.RGBA{R: 255, G: 0, B: 0, A: 128,}, color.RGBA{R: 255, G: 0, B: 0, A: 128,},
-1) -1)
} }
func StopService(name string, client mqtt.Client, topics ...string) { func StopService(name string, client mqtt.Client, topics ...string) {