Temp

submodules/TelegramUI/Sources/SampleBufferPool.swift

@@ -0,0 +1,64 @@
+import Foundation
+import UIKit
+import AVFoundation
+import SwiftSignalKit
+
+private final class SampleBufferLayerImplNullAction: NSObject, CAAction {
+    @objc func run(forKey event: String, object anObject: Any, arguments dict: [AnyHashable : Any]?) {
+    }
+}
+
+private final class SampleBufferLayerImpl: AVSampleBufferDisplayLayer {
+    override func action(forKey event: String) -> CAAction? {
+        return SampleBufferLayerImplNullAction()
+    }
+}
+
+final class SampleBufferLayer {
+    let layer: AVSampleBufferDisplayLayer
+    private let enqueue: (AVSampleBufferDisplayLayer) -> Void
+    
+    
+    var isFreed: Bool = false
+    fileprivate init(layer: AVSampleBufferDisplayLayer, enqueue: @escaping (AVSampleBufferDisplayLayer) -> Void) {
+        self.layer = layer
+        self.enqueue = enqueue
+    }
+    
+    deinit {
+        if !isFreed {
+            self.enqueue(self.layer)
+        }
+    }
+}
+
+private let pool = Atomic<[AVSampleBufferDisplayLayer]>(value: [])
+
+func clearSampleBufferLayerPoll() {
+    let _ = pool.modify { _ in return [] }
+}
+
+func takeSampleBufferLayer() -> SampleBufferLayer {
+    var layer: AVSampleBufferDisplayLayer?
+    let _ = pool.modify { list in
+        var list = list
+        if !list.isEmpty {
+            layer = list.removeLast()
+        }
+        return list
+    }
+    if layer == nil {
+        layer = SampleBufferLayerImpl()
+    }
+    return SampleBufferLayer(layer: layer!, enqueue: { layer in
+        Queue.mainQueue().async {
+            layer.flushAndRemoveImage()
+            layer.setAffineTransform(CGAffineTransform.identity)
+            let _ = pool.modify { list in
+                var list = list
+                list.append(layer)
+                return list
+            }
+        }
+    })
+}