From 28b6df864f0e8f5e6f479a84c06143ac19b45daa Mon Sep 17 00:00:00 2001
From: amv <basil96@users.noreply.github.com>
Date: Wed, 16 Sep 2020 22:18:20 -0400
Subject: [PATCH] Add pause/resume functionality to FPS

* Does not break existing API.

* Add `pause` and `resume` methods to FPS in order to pause and resume the timer in an accurate way.

* Calls to `update` will not increment frame count while the timer is paused.

* Also, convert old comments to proper docstrings.
---
 imutils/video/fps.py | 42 ++++++++++++++++++++++++++++++------------
 1 file changed, 30 insertions(+), 12 deletions(-)

diff --git a/imutils/video/fps.py b/imutils/video/fps.py
index cad5b89..935e313 100644
--- a/imutils/video/fps.py
+++ b/imutils/video/fps.py
@@ -1,33 +1,51 @@
 # import the necessary packages
 import datetime
 
+
 class FPS:
 	def __init__(self):
-		# store the start time, end time, and total number of frames
-		# that were examined between the start and end intervals
+		'''Create a new FPS timer.'''
+		# Store the total elapsed time and the total number of frames
+		# that were examined during that elapsed time.
+		# Start time and end time apply only to a given segment.
+		self._elapsed = datetime.timedelta()
 		self._start = None
 		self._end = None
 		self._numFrames = 0
+		self._is_paused = False
 
 	def start(self):
-		# start the timer
+		'''Start the timer.'''
 		self._start = datetime.datetime.now()
 		return self
 
 	def stop(self):
-		# stop the timer
+		'''Stop the timer.'''
 		self._end = datetime.datetime.now()
+		self._elapsed += (self._end - self._start)
+
+	def pause(self):
+		'''Pause the timer.'''
+		self.stop()
+		self._is_paused = True
+
+	def resume(self):
+		'''Resume the timer from a pause.'''
+		self.start()
+		self._is_paused = False
 
 	def update(self):
-		# increment the total number of frames examined during the
-		# start and end intervals
-		self._numFrames += 1
+		'''Increment the total number of frames examined during the
+		timing intervals.'''
+		# ignore this call while we're paused
+		if not self._is_paused:
+			self._numFrames += 1
 
 	def elapsed(self):
-		# return the total number of seconds between the start and
-		# end interval
-		return (self._end - self._start).total_seconds()
+		'''Return the total number of seconds during the
+		timing intervals.'''
+		return self._elapsed.total_seconds()
 
 	def fps(self):
-		# compute the (approximate) frames per second
-		return self._numFrames / self.elapsed()
\ No newline at end of file
+		'''Return the (approximate) frames per second.'''
+		return self._numFrames / self.elapsed()