自定义相机二


import UIKit
import SnapKit
import AVFoundation

class IDCustomCameraVC: UIViewController, AVCapturePhotoCaptureDelegate {
    
    private let captureSession = AVCaptureSession()
    private var previewLayer: AVCaptureVideoPreviewLayer!
    private var capturePhotoOutput: AVCapturePhotoOutput!
    
    private let bgView = UIView()
    private let closeBtn = UIButton(type: .custom)
    private let photographBtn = UIButton(type: .custom)
    
    private let cancelBtn = UIButton(type: .custom)
    private let confirmBtn = UIButton(type: .custom)
    
    private let plaseLab = UILabel()
    
    
    typealias IDPhotoFinishBlock = (_ image: UIImage?)->Void
    var idPhotoFinishBlock: IDPhotoFinishBlock?
    

    override func viewDidLoad() {
        super.viewDidLoad()
        
        setupCamera()
        
        view.addSubview(bgView)
        bgView.backgroundColor = .black.withAlphaComponent(0.6)
        bgView.snp.makeConstraints { make in
            make.edges.equalToSuperview()
        }
        
        view.addSubview(closeBtn)
        closeBtn.backgroundColor = .red
        closeBtn.snp.makeConstraints { make in
            make.size.equalTo(CGSizeMake(25, 25))
            make.right.equalTo(view.snp.right).offset(-20)
            make.top.equalTo(view.snp.topMargin).offset(20)
        }
        
        view.addSubview(photographBtn)
        photographBtn.backgroundColor = .blue
        photographBtn.snp.makeConstraints { make in
            make.centerX.equalToSuperview()
            make.size.equalTo(CGSize(width: 64, height: 64))
            make.bottom.equalTo(view.snp.bottomMargin).offset(-15)
        }
        
        view.addSubview(cancelBtn)
        view.addSubview(confirmBtn)
        
        cancelBtn.backgroundColor = .green
        confirmBtn.backgroundColor = .brown
        
        cancelBtn.snp.makeConstraints { make in
            make.left.equalTo(88)
            make.centerY.equalTo(photographBtn)
            make.size.equalTo(CGSize(width: 52, height: 52))
        }
        
        confirmBtn.snp.makeConstraints { make in
            make.right.equalTo(-88)
            make.centerY.equalTo(photographBtn)
            make.size.equalTo(CGSize(width: 52, height: 52))
        }
        
        view.addSubview(plaseLab)
        plaseLab.numberOfLines = 0
        plaseLab.textColor = .white
        plaseLab.textAlignment = .center
        plaseLab.font = .systemFont(ofSize: 14)
        plaseLab.frame = CGRect(x: 0, y: 0, width: 350, height: 60)
        plaseLab.text = "Please place the ID photo in the frame and\nmake sure it is clear and complete"
        
        plaseLab.transform = CGAffineTransform(rotationAngle: CGFloat.pi / 2)
        plaseLab.frame.origin = CGPoint(x: view.bounds.midX+plaseLab.frame.size.width*1.9, y: (view.frame.size.height - plaseLab.frame.size.height)/2)
        
        cancelBtn.isHidden = true
        confirmBtn.isHidden = true
        
        closeBtn.addTarget(self, action: #selector(cancelBtnClick(_:)), for: .touchUpInside)
        cancelBtn.addTarget(self, action: #selector(cancelBtnClick(_:)), for: .touchUpInside)
        confirmBtn.addTarget(self, action: #selector(confirmBtnClick(_:)), for: .touchUpInside)
        photographBtn.addTarget(self, action: #selector(photographBtnClick(_:)), for: .touchUpInside)
            
    }
    
    @objc func cancelBtnClick(_ btn: UIButton) {
        
        idPhotoFinishBlock?(nil)
        self.navigationController?.popViewController(animated: true)
    }
    
    @objc func confirmBtnClick(_ btn: UIButton) {
        
        
    }
    
    @objc func photographBtnClick(_ btn: UIButton) {
        
        let settings = AVCapturePhotoSettings()
        capturePhotoOutput.capturePhoto(with: settings, delegate: self)
        
        btn.isHidden = true
        cancelBtn.isHidden = !btn.isHidden
        confirmBtn.isHidden = !btn.isHidden
    }

    
    func setupCamera() {
        
        captureSession.sessionPreset = .photo
        
        guard let frontCamera = AVCaptureDevice.default(.builtInWideAngleCamera, for: .video, position: .back) else {
            print("Unable to access front camera")
            return
        }
        
        do {
            let input = try AVCaptureDeviceInput(device: frontCamera)
            captureSession.addInput(input)
            capturePhotoOutput = AVCapturePhotoOutput()
            captureSession.addOutput(capturePhotoOutput)
        } catch {
            print(error.localizedDescription)
        }
        
        previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
        previewLayer.videoGravity = .resizeAspectFill
        previewLayer.frame = view.layer.bounds
        view.layer.addSublayer(previewLayer)
        
        captureSession.startRunning()
    }
    
    func photoOutput(_ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: Error?) {
        
        if let error = error {
            print("Error capturing photo: \(error.localizedDescription)")
            return
        }

        guard let imageData = photo.fileDataRepresentation() else {
            print("Unable to get photo data")
            return
        }

        var capturedImage = UIImage(data: imageData)

        if let cgImage = capturedImage?.cgImage {
            capturedImage = UIImage(cgImage: cgImage, scale: capturedImage?.scale ?? 1.0, orientation: .leftMirrored)
        }
        
        captureSession.stopRunning()
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容