From 710b0e99976eca96ed1094c23ae8843be17bed5c Mon Sep 17 00:00:00 2001
From: Miquel Massot <miquel.massot@gmail.com>
Date: Fri, 26 Jan 2024 06:07:26 +0000
Subject: [PATCH] Allow lazy rates

---
 src/zeroros/rate.py | 37 ++++++++++++++++++++++++++++++-------
 1 file changed, 30 insertions(+), 7 deletions(-)

diff --git a/src/zeroros/rate.py b/src/zeroros/rate.py
index 6ac8b7b..ba98163 100644
--- a/src/zeroros/rate.py
+++ b/src/zeroros/rate.py
@@ -20,7 +20,7 @@ def __init__(self, frequency_hz):
         @param frequency_hz: Frequency rate in Hz to determine sleeping
         @type  frequency_hz: float
         """
-        self.last_time = float(datetime.timestamp(datetime.utcnow()))
+        self.last_time = datetime.utcnow().timestamp()
         self.sleep_dur = 1.0 / frequency_hz
 
     def _remaining(self, curr_time):
@@ -45,16 +45,22 @@ def remaining(self):
         @return: time remaining
         @rtype: L{Time}
         """
-        curr_time = float(datetime.timestamp(datetime.utcnow()))
+        curr_time = datetime.utcnow().timestamp()
         return self._remaining(curr_time)
 
+    def reset(self):
+        """
+        Reset the rate timer.
+        """
+        self.last_time = datetime.utcnow().timestamp()
+
     def sleep(self):
         """
         Attempt sleep at the specified rate. sleep() takes into
         account the time elapsed since the last successful
         sleep().
         """
-        curr_time = float(datetime.timestamp(datetime.utcnow()))
+        curr_time = datetime.utcnow().timestamp()
         sleep(self._remaining(curr_time))
         self.last_time = self.last_time + self.sleep_dur
 
@@ -67,13 +73,30 @@ def sleep(self):
 if __name__ == "__main__":
     import random
 
+    count = 0
+
     r = Rate(10.0)
-    curr_time = float(datetime.timestamp(datetime.utcnow()))
-    while True:
-        diff = 1.0 / (float(datetime.timestamp(datetime.utcnow())) - curr_time)
-        curr_time = float(datetime.timestamp(datetime.utcnow()))
+    curr_time = datetime.utcnow().timestamp()
+    while True and count < 10:
+        diff = 1.0 / (datetime.utcnow().timestamp() - curr_time)
+        curr_time = datetime.utcnow().timestamp()
         x = random.random() / 10.0
         print(diff, x)
         # Random time-lenght long computation
         time.sleep(x)
         r.sleep()
+        count += 1
+
+    r = Rate(1.0)
+    count = 0
+    while True:
+        x = random.random() / 10.0
+        time.sleep(x)
+        if r.remaining() <= 0:
+            diff = 1.0 / (datetime.utcnow().timestamp() - curr_time)
+            curr_time = datetime.utcnow().timestamp()
+            print(diff, x, r.remaining())  # Random time-lenght long computation
+            r.reset()
+            count += 1
+        if count >= 10:
+            break