[bugfix] Count time during breaks

RoboCup-SSL · Aug 25, 2018 · e7dd7fc · e7dd7fc
1 parent 49eafe3
commit e7dd7fc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/internal/app/controller/engine.go b/internal/app/controller/engine.go
@@ -170,8 +170,12 @@ func (e *Engine) loadStages() {
 	e.StageTimes[StageShootoutBreak] = e.config.Overtime.BreakAfter
 }
 
+func (e *Engine) countStageTime() bool {
+	return e.State.Stage.IsPausedStage() || e.State.GameState() == GameStateRunning
+}
+
 func (e *Engine) updateTimes(delta time.Duration) {
-	if e.State.GameState() == GameStateRunning {
+	if e.countStageTime() {
 		e.State.StageTimeElapsed += delta
 		e.State.StageTimeLeft -= delta
 

diff --git a/internal/app/controller/state.go b/internal/app/controller/state.go
@@ -130,6 +130,14 @@ func (s Stage) IsPreStage() bool {
 	return false
 }
 
+func (s Stage) IsPausedStage() bool {
+	switch s {
+	case StageHalfTime, StageOvertimeBreak, StageOvertimeHalfTime, StageShootoutBreak:
+		return true
+	}
+	return false
+}
+
 // RefCommand is a command to be send to the teams
 type RefCommand string