Fix: Consider removal time for yellow cards only during RUNNING

RoboCup-SSL · Apr 23, 2022 · 9e0b16c · 9e0b16c
1 parent 7d4b60c
commit 9e0b16c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/app/engine/process_botremoved.go b/internal/app/engine/process_botremoved.go
@@ -38,9 +38,13 @@ func (p *BotNumberProcessor) processBotNumber() {
 func (e *Engine) processBotNumberPerTeam(team state.Team) {
 
 	teamInfo := e.currentState.TeamState[team.String()]
+	removalTime := e.gameConfig.YellowCardBotRemovalTime
+	if *e.currentState.GameState.Type != state.GameState_RUNNING {
+		removalTime = 0
+	}
 	newCards := newActiveYellowCards(
 		teamInfo.YellowCards,
-		e.gameConfig.YellowCardDuration-e.gameConfig.YellowCardBotRemovalTime,
+		e.gameConfig.YellowCardDuration-removalTime,
 	)
 
 	numBots := e.gcState.TrackerStateGc.NumTeamRobots(team)