Timestamp Camera images between grab and decode (#36)

waggle-sensor · May 31, 2023 · 4b2d9a0 · 4b2d9a0
1 parent 786ee82
commit 4b2d9a0
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/waggle/data/vision.py b/src/waggle/data/vision.py
@@ -144,16 +144,22 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             self.capture.release()
 
     def snapshot(self):
-        timestamp = get_timestamp()
-        ok, data = self.capture.read()
+        ok = self.capture.grab()
         if not ok:
             raise RuntimeError("failed to take snapshot")
+        timestamp = get_timestamp()
+        ok, data = self.capture.retrieve()
+        if not ok:
+            raise RuntimeError("failed to retrieve the taken snapshot")
         return ImageSample(data=data, timestamp=timestamp, format=self.format)
 
     def stream(self):
         while True:
+            ok = self.capture.grab()
+            if not ok:
+                break
             timestamp = get_timestamp()
-            ok, data = self.capture.read()
+            ok, data = self.capture.retrieve()
             if not ok:
                 break
             yield ImageSample(data=data, timestamp=timestamp, format=self.format)