Refactor TypedBlockQueue.

shogo4405 · Jul 28, 2024 · 8dd8abd · 8dd8abd
1 parent 0a319d7
commit 8dd8abd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/Sources/Screen/ScreenObject.swift b/Sources/Screen/ScreenObject.swift
@@ -226,7 +226,7 @@ public final class VideoTrackScreenObject: ScreenObject {
         super.init()
         horizontalAlignment = .center
         do {
-            queue = TypedBlockQueue(try CMBufferQueue(capacity: 1, handlers: .outputPTSSortedSampleBuffers))
+            queue = try TypedBlockQueue(capacity: 1, handlers: .outputPTSSortedSampleBuffers)
         } catch {
             logger.error(error)
         }

diff --git a/Sources/Util/TypedBlockQueue.swift b/Sources/Util/TypedBlockQueue.swift
@@ -3,6 +3,7 @@ import Foundation
 
 final class TypedBlockQueue<T: AnyObject> {
     private let queue: CMBufferQueue
+    private let capacity: CMItemCount
 
     @inlinable @inline(__always) var head: T? {
         guard let head = queue.head else {
@@ -19,8 +20,9 @@ final class TypedBlockQueue<T: AnyObject> {
         queue.duration
     }
 
-    init(_ queue: CMBufferQueue) {
-        self.queue = queue
+    init(capacity: CMItemCount, handlers: CMBufferQueue.Handlers) throws {
+        self.capacity = capacity
+        self.queue = try CMBufferQueue(capacity: capacity, handlers: handlers)
     }
 
     @inlinable