]> git.r.bdr.sh - rbdr/captura/blob - Captura/Domain/CapturaCaptureSession.swift
Complete initial release
[rbdr/captura] / Captura / Domain / CapturaCaptureSession.swift
1 import AppKit
2 import AVFoundation
3
4 class CapturaCaptureSession: AVCaptureSession, AVCaptureFileOutputRecordingDelegate, AVCaptureVideoDataOutputSampleBufferDelegate {
5
6 let videoOutput = AVCaptureVideoDataOutput()
7 let movieFileOutput = AVCaptureMovieFileOutput()
8 var receivedFrames = false
9
10 init(_ screen: NSScreen, box: NSRect) {
11 super.init()
12
13 let displayId = screen.deviceDescription[NSDeviceDescriptionKey("NSScreenNumber")] as! CGDirectDisplayID
14 let screenInput = AVCaptureScreenInput(displayID: displayId)
15 screenInput?.cropRect = box.insetBy(dx: 1, dy: 1)
16
17 if self.canAddInput(screenInput!) {
18 self.addInput(screenInput!)
19 }
20
21 videoOutput.setSampleBufferDelegate(self, queue: Dispatch.DispatchQueue(label: "sample buffer delegate", attributes: []))
22
23 if self.canAddOutput(videoOutput) {
24 self.addOutput(videoOutput)
25 }
26
27 if self.canAddOutput(movieFileOutput) {
28 self.addOutput(movieFileOutput)
29 }
30 }
31
32 func startRecording() {
33 receivedFrames = false
34 self.startRunning()
35
36 DispatchQueue.main.asyncAfter(deadline: .now() + 1) {
37 if !self.receivedFrames {
38 NotificationCenter.default.post(name: .failedToStart, object: nil, userInfo: nil)
39 }
40 }
41 }
42
43 func startRecording(to url: URL) {
44 self.startRecording()
45 movieFileOutput.startRecording(to: url, recordingDelegate: self)
46 }
47
48 // MARK: - AVCaptureVideoDataOutputSampleBufferDelegate Implementation
49
50 func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
51 receivedFrames = true
52
53 guard let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
54 NotificationCenter.default.post(name: .receivedFrame, object: nil, userInfo: ["frame": imageBuffer])
55 }
56
57 // MARK: - AVCaptureFileOutputRecordingDelegate Implementation
58
59 func fileOutput(_ output: AVCaptureFileOutput, didFinishRecordingTo outputFileURL: URL, from connections: [AVCaptureConnection], error: Error?) {}
60 }