Only update robot specs in pre-stages

RoboCup-SSL · Apr 17, 2021 · dd9ec29 · dd9ec29
1 parent ee8dd21
commit dd9ec29
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/simctl/robot_specs.go b/internal/simctl/robot_specs.go
@@ -50,8 +50,14 @@ func (r *RobotSpecSetter) LoadRobotSpecs(configFile string) {
 }
 
 func (r *RobotSpecSetter) handleRobotSpecs() {
-	if *r.c.lastRefereeMsg.Command != referee.Referee_HALT {
-		// Only during HALT
+	switch *r.c.lastRefereeMsg.Stage {
+	case referee.Referee_NORMAL_FIRST_HALF_PRE,
+		referee.Referee_NORMAL_SECOND_HALF_PRE,
+		referee.Referee_EXTRA_FIRST_HALF_PRE,
+		referee.Referee_EXTRA_SECOND_HALF_PRE:
+		// accept
+	default:
+		// Only in pre-stages
 		return
 	}