Merge pull request #1232 from shogo4405/feature/video-codec-delegate

Invoke delegate the stream(_ stream: NetStream, didOutput video: CMSa…
shogo4405 · Jul 11, 2023 · a1cc00f · a1cc00f
2 parents 9b4bb11 + 4a94f6d
commit a1cc00f
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/Examples/iOS/PlaybackViewController.swift b/Examples/iOS/PlaybackViewController.swift
@@ -16,6 +16,7 @@ final class PlaybackViewController: UIViewController {
     override func viewDidLoad() {
         super.viewDidLoad()
         rtmpStream = RTMPStream(connection: rtmpConnection)
+        rtmpStream.delegate = self
     }
 
     override func viewWillAppear(_ animated: Bool) {
@@ -119,3 +120,31 @@ extension PlaybackViewController: AVPictureInPictureSampleBufferPlaybackDelegate
         completionHandler()
     }
 }
+
+extension PlaybackViewController: NetStreamDelegate {
+    // MARK: NetStreamDelegate
+    func stream(_ stream: NetStream, didOutput audio: AVAudioBuffer, presentationTimeStamp: CMTime) {
+    }
+
+    func stream(_ stream: NetStream, didOutput video: CMSampleBuffer) {
+    }
+
+    func stream(_ stream: NetStream, sessionWasInterrupted session: AVCaptureSession, reason: AVCaptureSession.InterruptionReason) {
+    }
+
+    func stream(_ stream: NetStream, sessionInterruptionEnded session: AVCaptureSession, reason: AVCaptureSession.InterruptionReason) {
+    }
+
+    func stream(_ stream: NetStream, videoCodecErrorOccurred error: VideoCodec.Error) {
+    }
+
+    func stream(_ stream: NetStream, audioCodecErrorOccurred error: HaishinKit.AudioCodec.Error) {
+    }
+
+    func streamWillDropFrame(_ stream: NetStream) -> Bool {
+        return false
+    }
+
+    func streamDidOpen(_ stream: NetStream) {
+    }
+}
diff --git a/Sources/Media/IOMixer.swift b/Sources/Media/IOMixer.swift
@@ -287,6 +287,7 @@ extension IOMixer: IOUnitDecoding {
 extension IOMixer: MediaLinkDelegate {
     // MARK: MediaLinkDelegate
     func mediaLink(_ mediaLink: MediaLink, dequeue sampleBuffer: CMSampleBuffer) {
+        delegate?.mixer(self, didOutput: sampleBuffer)
         drawable?.enqueue(sampleBuffer)
     }