From 4019524f8556f4ee4ea04a406af9129f3885ef8d Mon Sep 17 00:00:00 2001
From: Nicolai Ommer <nicolai.ommer@gmail.com>
Date: Tue, 22 Jun 2021 19:38:28 +0200
Subject: [PATCH] Also reset geometry after game started

---
 internal/simctl/handler_geometry.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/internal/simctl/handler_geometry.go b/internal/simctl/handler_geometry.go
index 2578a18..c88a95b 100644
--- a/internal/simctl/handler_geometry.go
+++ b/internal/simctl/handler_geometry.go
@@ -47,8 +47,8 @@ func (r *GeometryHandler) loadGeometry() {
 }
 
 func (r *GeometryHandler) handleGeometry() {
-	if *r.c.lastRefereeMsg.Stage != referee.Referee_NORMAL_FIRST_HALF_PRE {
-		// Only before the game starts
+	if *r.c.lastRefereeMsg.Command != referee.Referee_HALT {
+		// Only during HALT
 		return
 	}
 
@@ -57,7 +57,7 @@ func (r *GeometryHandler) handleGeometry() {
 		return
 	}
 
-	maxBots := *r.c.lastRefereeMsg.Yellow.MaxAllowedBots
+	maxBots := int(*r.c.lastRefereeMsg.Yellow.MaxAllowedBots) + len(r.c.lastRefereeMsg.Yellow.YellowCardTimes)
 	var geometry *vision.SSL_GeometryData
 	if maxBots == 6 {
 		geometry = r.geometryDivB