Performance improvement.

shogo4405 · Dec 9, 2023 · a5c0b3a · a5c0b3a
1 parent bacaae1
commit a5c0b3a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/Examples/iOS/IngestViewController.swift b/Examples/iOS/IngestViewController.swift
@@ -69,7 +69,6 @@ final class IngestViewController: UIViewController {
         stream.attachAudio(AVCaptureDevice.default(for: .audio), automaticallyConfiguresApplicationAudioSession: false) { error in
             logger.warn(error)
         }
-
         if #available(iOS 13.0, *) {
             let front = AVCaptureDevice.default(.builtInWideAngleCamera, for: .video, position: .front)
             stream.videoCapture(for: 1)?.isVideoMirrored = true

diff --git a/Sources/IO/MTHKView.swift b/Sources/IO/MTHKView.swift
@@ -46,12 +46,15 @@ public class MTHKView: MTKView {
     }
 
     private var currentSampleBuffer: CMSampleBuffer?
+
     private let colorSpace: CGColorSpace = CGColorSpaceCreateDeviceRGB()
 
     private lazy var commandQueue: (any MTLCommandQueue)? = {
         return device?.makeCommandQueue()
     }()
 
+    private var context: CIContext?
+
     private var captureVideoPreview: View? {
         willSet {
             captureVideoPreview?.removeFromSuperview()
@@ -67,7 +70,9 @@ public class MTHKView: MTKView {
     private weak var currentStream: NetStream? {
         didSet {
             currentStream.map {
-                $0.context = CIContext(mtlDevice: device!)
+                if let context = self.context {
+                    $0.context = context
+                }
                 $0.drawable = self
             }
         }
@@ -91,6 +96,9 @@ public class MTHKView: MTKView {
         delegate = self
         framebufferOnly = false
         enableSetNeedsDisplay = true
+        if let device {
+            context = CIContext(mtlDevice: device)
+        }
     }
 }
 
@@ -129,9 +137,9 @@ extension MTHKView: MTKViewDelegate {
 
     public func draw(in view: MTKView) {
         guard
+            let context,
             let currentDrawable = currentDrawable,
-            let commandBuffer = commandQueue?.makeCommandBuffer(),
-            let context = currentStream?.context else {
+            let commandBuffer = commandQueue?.makeCommandBuffer() else {
             return
         }
         if