Swift 自定义相机

之前都是用的Object-C，最近在学习Swift，就用Swift写一个拍照界面。
自定义相机会使用到以下类：
AVCaptureSession：管理输入输出音视频流；
AVCaptureDevice：相机硬件的接口，用于控制硬件特性，例如前后镜头、闪光灯等；
AVCaptureDeviceInput：从AVCaptureDevice捕获媒体的接口
AVCapturePhotoOutput：捕获照片的接口；
AVCaptureVideoPreviewLayer：用于预览的接口；

获取摄像头权限

switch AVCaptureDevice.authorizationStatus(for: .video) {
        case .authorized:
            print("已获取权限")
            break
        case .notDetermined:
            AVCaptureDevice.requestAccess(for: .video) { granted in
                print("获取权限\(granted ? "成功" : "失败")")
            }
        case .denied:
            print("获取权限失败")
        default:
            break
        }

获取音视频管理器

使用AVCaptureSession创建管理器：

fileprivate var session: AVCaptureSession!

session = AVCaptureSession()
session.sessionPreset = .photo

这里的sessionPreset指输出照片的分辨率，详细参数见文档。

获取摄像头

使用AVCaptureDevice类获取后置摄像头：
AVCaptureDevice是提供实时输入媒体数据（例如视频和音频）的物理设备。

fileprivate var device: AVCaptureDevice!

获取摄像头，

func cameraWithPosition(position: AVCaptureDevice.Position) -> AVCaptureDevice {
        let tmpdevice: AVCaptureDevice!
        let devices: AVCaptureDevice.DiscoverySession = AVCaptureDevice.DiscoverySession(deviceTypes: [AVCaptureDevice.DeviceType.builtInUltraWideCamera], mediaType: AVMediaType.video, position: position) // 获取广角镜头
        if devices.devices.count == 0 {
            tmpdevice = AVCaptureDevice.default(for: AVMediaType.video)! // 普通镜头
        } else {
            tmpdevice = devices.devices.last ?? AVCaptureDevice(uniqueID: "")! // 广角镜头
        }
        return tmpdevice
    }

因为我之前到项目里面，需要用的广角镜头，所以这里默认获取广角镜头，在没有广角镜头的时候，调用正常镜头。

device = cameraWithPosition(position: AVCaptureDevice.Position.back)

获取输入数据

使用AVCaptureDeviceInput类获取镜头输入的数据：
AVCaptureDeviceInput：从AVCaptureDevice设备获取数据（用于获取摄像头拍摄的数据），然后会被添加给AVCaptureSession管理。

fileprivate var input: AVCaptureDeviceInput!

从device中获取数据：

do {
	try input = AVCaptureDeviceInput(device: device)
   } catch {
   }

将input添加到session中：

if session.canAddInput(input) {
            session.addInput(input)
        }

获取输出照片

使用AVCapturePhotoOutput类获取照片：

fileprivate var imageOutput: AVCapturePhotoOutput!

将imageOutput添加到session中：

if session.canAddOutput(imageOutput) {
	session.addOutput(imageOutput)
	}

实时预览

使用AVCaptureVideoPreviewLayer预览：

fileprivate var previewLayer: AVCaptureVideoPreviewLayer!

实现预览效果

 previewLayer = AVCaptureVideoPreviewLayer(session: session)
        previewLayer.frame = CGRect(x: 0, y: 0, width: screenWidth, height: screenWidth * 4 / 3)
        previewLayer.videoGravity = .resizeAspectFill
        preview.layer.addSublayer(previewLayer)

因为我选择的分辨率是.photo，所以图片原始尺寸比例是4/3，所以我这边预览的比例也就设为4/3了，如果照片比例和预览比例不一致，会出现预览存在黑边或显示不全的情况。

运行音视频管理器

DispatchQueue.global(qos: .background).async {
            self.session.startRunning()
        }

注意：这串代码要放在全局线程里面运行，否则界面会有卡顿的危险

获取照片

do {
            try input.device.lockForConfiguration()
            let videoPreviewLayerVideoOrientation = previewLayer?.connection?.videoOrientation
            let photoOutputConnection = imageOutput.connection(with: .video)
            photoOutputConnection?.videoOrientation = videoPreviewLayerVideoOrientation ?? .portrait // 设置照片方向

            let photoSettings = AVCapturePhotoSettings(format: [AVVideoCodecKey: AVVideoCodecType.jpeg])
            if input.device.isFlashAvailable {//打开闪光灯
                photoSettings.flashMode = .on
            }
            imageOutput.capturePhoto(with: photoSettings, delegate: self)
            input.device.unlockForConfiguration()
        } catch {
        }

通过AVCapturePhotoOutput类的代理方法获取拍的照片

func photoOutput(_ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: (any Error)?) {
        let image: UIImage! = UIImage(data: photo.fileDataRepresentation()!) ?? UIImage(named: "AppIcon")
        UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil)
    }

最终效果如下：

后续会陆续添加二维码扫描，实时边缘检测等功能，期待一波吧！