-
-
Save dimroc/670898b0f98d4715161c41c871abd32e to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import UIKit | |
import AVFoundation | |
protocol FrameExtractorDelegate: class { | |
func captured(image: UIImage) | |
} | |
class FrameExtractor: NSObject, AVCaptureVideoDataOutputSampleBufferDelegate { | |
private let position = AVCaptureDevice.Position.front | |
private let quality = AVCaptureSession.Preset.hd1280x720 | |
private var permissionGranted = false | |
private let sessionQueue = DispatchQueue(label: "session queue") | |
private let captureSession = AVCaptureSession() | |
private let context = CIContext() | |
weak var delegate: FrameExtractorDelegate? | |
override init() { | |
super.init() | |
checkPermission() | |
sessionQueue.async { [unowned self] in | |
self.configureSession() | |
self.captureSession.startRunning() | |
} | |
} | |
// MARK: AVSession configuration | |
private func checkPermission() { | |
switch AVCaptureDevice.authorizationStatus(for: AVMediaType.video) { | |
case .authorized: | |
permissionGranted = true | |
case .notDetermined: | |
requestPermission() | |
default: | |
permissionGranted = false | |
} | |
} | |
private func requestPermission() { | |
sessionQueue.suspend() | |
AVCaptureDevice.requestAccess(for: AVMediaType.video) { [unowned self] granted in | |
self.permissionGranted = granted | |
self.sessionQueue.resume() | |
} | |
} | |
private func configureSession() { | |
guard permissionGranted else { return } | |
captureSession.sessionPreset = quality | |
guard let captureDevice = selectCaptureDevice() else { return } | |
guard let captureDeviceInput = try? AVCaptureDeviceInput(device: captureDevice) else { return } | |
guard captureSession.canAddInput(captureDeviceInput) else { return } | |
captureSession.addInput(captureDeviceInput) | |
let videoOutput = AVCaptureVideoDataOutput() | |
videoOutput.setSampleBufferDelegate(self, queue: DispatchQueue(label: "sample buffer")) | |
guard captureSession.canAddOutput(videoOutput) else { return } | |
captureSession.addOutput(videoOutput) | |
guard let connection = videoOutput.connection(with: AVFoundation.AVMediaType.video) else { return } | |
guard connection.isVideoOrientationSupported else { return } | |
guard connection.isVideoMirroringSupported else { return } | |
connection.videoOrientation = .portrait | |
connection.isVideoMirrored = position == .front | |
} | |
private func selectCaptureDevice() -> AVCaptureDevice? { | |
return AVCaptureDevice.default(.builtInWideAngleCamera, for: AVMediaType.video, position: self.position) | |
} | |
// MARK: Sample buffer to UIImage conversion | |
private func imageFromSampleBuffer(sampleBuffer: CMSampleBuffer) -> UIImage? { | |
guard let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return nil } | |
let ciImage = CIImage(cvPixelBuffer: imageBuffer) | |
guard let cgImage = context.createCGImage(ciImage, from: ciImage.extent) else { return nil } | |
return UIImage(cgImage: cgImage) | |
} | |
// MARK: AVCaptureVideoDataOutputSampleBufferDelegate | |
func captureOutput(_ captureOutput: AVCaptureOutput!, didOutputSampleBuffer sampleBuffer: CMSampleBuffer!, from connection: AVCaptureConnection!) { | |
guard let uiImage = imageFromSampleBuffer(sampleBuffer: sampleBuffer) else { return } | |
DispatchQueue.main.async { [unowned self] in | |
self.delegate?.captured(image: uiImage) | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment