Also consider red cards for detecting divA/B

RoboCup-SSL · Jun 23, 2021 · 6a75fc5 · 6a75fc5
1 parent 4019524
commit 6a75fc5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/simctl/handler_geometry.go b/internal/simctl/handler_geometry.go
@@ -57,7 +57,9 @@ func (r *GeometryHandler) handleGeometry() {
 		return
 	}
 
-	maxBots := int(*r.c.lastRefereeMsg.Yellow.MaxAllowedBots) + len(r.c.lastRefereeMsg.Yellow.YellowCardTimes)
+	maxBots := int(*r.c.lastRefereeMsg.Yellow.MaxAllowedBots) +
+		len(r.c.lastRefereeMsg.Yellow.YellowCardTimes) +
+		int(*r.c.lastRefereeMsg.Yellow.RedCards)
 	var geometry *vision.SSL_GeometryData
 	if maxBots == 6 {
 		geometry = r.geometryDivB