Cleanup PhotoOutputSynchronizer

mrousavy · Aug 4, 2023 · bf059d3 · bf059d3
1 parent 8221fd2
commit bf059d3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/android/src/main/java/com/mrousavy/camera/CameraSession.kt b/android/src/main/java/com/mrousavy/camera/CameraSession.kt
@@ -177,7 +177,7 @@ class CameraSession(private val cameraManager: CameraManager,
     val timestamp = result[CaptureResult.SENSOR_TIMESTAMP]!!
     Log.i(TAG, "Photo capture 1/2 complete - received metadata with timestamp $timestamp")
     try {
-      val image = photoOutputSynchronizer[timestamp].await()
+      val image = photoOutputSynchronizer.await(timestamp)
       // TODO: Correctly get rotationDegrees and isMirrored
       val rotation = ExifUtils.computeExifOrientation(0, false)
 

diff --git a/android/src/main/java/com/mrousavy/camera/utils/PhotoOutputSynchronizer.kt b/android/src/main/java/com/mrousavy/camera/utils/PhotoOutputSynchronizer.kt
@@ -6,15 +6,21 @@ import kotlinx.coroutines.CompletableDeferred
 class PhotoOutputSynchronizer {
   private val photoOutputQueue = HashMap<Long, CompletableDeferred<Image>>()
 
-  operator fun get(key: Long): CompletableDeferred<Image> {
+  private operator fun get(key: Long): CompletableDeferred<Image> {
     if (!photoOutputQueue.containsKey(key)) {
       photoOutputQueue[key] = CompletableDeferred()
     }
     return photoOutputQueue[key]!!
   }
 
-  fun set(key: Long, image: Image) {
-    this[key].complete(image)
+  suspend fun await(timestamp: Long): Image {
+    val image = this[timestamp].await()
+    photoOutputQueue.remove(timestamp)
+    return image
+  }
+
+  fun set(timestamp: Long, image: Image) {
+    this[timestamp].complete(image)
   }
 
   fun clear() {