swift3.0自定義相機界面

這是公司上上上一個項目的自定義相機界面,原來是swift2.0寫的,今天改為swift3.0,記錄一下.

效果圖如下:

                                                                                        　　　　　　　　　　　　　　　　　　　　　　　　　　                                                                                                    

使用AVFoundation制作.

屬性:

    var device:AVCaptureDevice!   //獲取設備:如攝像頭
    var input:AVCaptureDeviceInput!   //輸入流
    var photoOutput:AVCaptureStillImageOutput! //輸出流
    var  output:AVCaptureMetadataOutput! //當啟動攝像頭開始捕獲輸入
    var  session:AVCaptureSession!//會話,協調着intput到output的數據傳輸,input和output的橋梁
    var  previewLayer:AVCaptureVideoPreviewLayer! //圖像預覽層，實時顯示捕獲的圖像
    
    var photoButton: UIButton?   //拍照按鈕
    var imageView: UIImageView?   //拍照后的成像
    var image: UIImage?   //拍照后的成像
    var isJurisdiction: Bool?   //是否獲取了拍照標示
    var flashBtn:UIButton?  //閃光燈按鈕

在viewDidLoad里面判斷下權限,沒給權限給個提示

        isJurisdiction = canUserCamear()
        if isJurisdiction! {
            customCamera()  //自定義相機
            customUI()  //自定義相機按鈕
        }
        else {
            return
        }

    // MARK: - 檢查相機權限
    func canUserCamear() -> Bool {
        let authStatus: AVAuthorizationStatus = AVCaptureDevice.authorizationStatus(forMediaType: AVMediaTypeVideo)
        if authStatus == .denied {
            let alertView = UIAlertView(title: "請打開相機權限", message: "設置-隱私-相機", delegate: self, cancelButtonTitle: "確定", otherButtonTitles: "取消")
            alertView.tag = 100
            alertView.show()
            return false
        }
        else {
            return true
        }
        return true
    }

自定義相機:

 //MARK: 初始化自定義相機
    func customCamera(){
       guard let devices = AVCaptureDevice.devices(withMediaType: AVMediaTypeVideo) as? [AVCaptureDevice] else { return } //初始化攝像頭設備
        guard let device = devices.filter({ return $0.position == .back }).first else{ return}
        self.device = device
        //輸入流初始化
        self.input = try? AVCaptureDeviceInput(device: device)
        //照片輸出流初始化
        self.photoOutput = AVCaptureStillImageOutput.init()
        //輸出流初始化
        self.output = AVCaptureMetadataOutput.init()
        //生成會話
        self.session = AVCaptureSession.init()
        if(self.session.canSetSessionPreset("AVCaptureSessionPreset1280x720")){
            self.session.sessionPreset = "AVCaptureSessionPreset1280x720"
        }
        if(self.session.canAddInput(self.input)){
            self.session.addInput(self.input)
        }
        if(self.session.canAddOutput(self.photoOutput)){
            self.session.addOutput(self.photoOutput)
        }
        //使用self.session，初始化預覽層，self.session負責驅動input進行信息的采集，layer負責把圖像渲染顯示
        self.previewLayer = AVCaptureVideoPreviewLayer.init(session: self.session)
        self.previewLayer.frame  = CGRect.init(x: 0, y: 0, width: kScreenWH.width, height: kScreenWH.height)
        self.previewLayer.videoGravity = "AVLayerVideoGravityResizeAspectFill"
        self.view.layer.addSublayer(self.previewLayer)
        //啟動
        self.session.startRunning()
        if ((try? device.lockForConfiguration()) != nil) {
            if device.isFlashModeSupported(.auto) {
                device.flashMode = .auto
            }
            //自動白平衡
            if device.isWhiteBalanceModeSupported(.autoWhiteBalance) {
                device.whiteBalanceMode = .autoWhiteBalance
            }
            device.unlockForConfiguration()
        }
        
        //閃光燈
        do{ try device.lockForConfiguration() }catch{ }
        if device.hasFlash == false { return }
        device.flashMode = AVCaptureFlashMode.auto
        device.unlockForConfiguration()
        
    }

自定義相機界面按鈕:拍照,取消,閃光燈,前后拍照切換按鈕

//MARK: 添加自定義按鈕等UI
    func customUI(){
       //前后攝像頭切換
        let changeBtn = UIButton.init()
        changeBtn.frame = CGRect.init(x: kScreenWH.width - 50, y: 20, width: 40, height: 40)
        changeBtn.setImage(#imageLiteral(resourceName: "change"), for: .normal)
        changeBtn.addTarget(self, action: #selector(self.changeCamera), for: .touchUpInside)
        view.addSubview(changeBtn)
        
        //拍照按鈕
        photoButton = UIButton(type: .custom)
        photoButton?.frame = CGRect(x: kScreenWH.width * 1 / 2.0 - 30, y: kScreenWH.height - 100, width: 60, height: 60)
        photoButton?.setImage(UIImage(named: "photograph"), for: .normal)
        photoButton?.setImage(UIImage(named: "photograph_Select"), for: .normal)
        photoButton?.addTarget(self, action: #selector(self.shutterCamera), for: .touchUpInside)
        view.addSubview(photoButton!)
        
      //閃光燈按鈕
        flashBtn = UIButton.init()
        flashBtn?.frame = CGRect.init(x: 10, y: 20, width: 40, height: 40)
        flashBtn?.addTarget(self, action: #selector(self.flashAction), for: .touchUpInside)
        flashBtn?.setImage(#imageLiteral(resourceName: "flash-A"), for: .normal)
        view.addSubview(flashBtn!)
        
       //取消
        let cancelBtn = UIButton.init()
        cancelBtn.frame = CGRect.init(x: 10, y: kScreenWH.height - 100, width: 60, height: 60)
        cancelBtn.setTitle("取消", for: .normal)
        cancelBtn.addTarget(self, action: #selector(self.cancelActin), for: .touchUpInside)
        view.addSubview(cancelBtn)
    }
    //MARK:前后攝像頭更改事件
    func changeCamera(){
        //獲取之前的鏡頭
        guard var position = input?.device.position else { return }
        //獲取當前應該顯示的鏡頭
        position = position == .front ? .back : .front
        //創建新的device
        guard let devices = AVCaptureDevice.devices(withMediaType: AVMediaTypeVideo) as? [AVCaptureDevice] else { return }
        // 1.2.取出獲取前置攝像頭
        let d = devices.filter({ return $0.position == position }).first
        device = d
        //input
        guard let videoInput = try? AVCaptureDeviceInput(device: d) else { return }

        //切換
        session.beginConfiguration()
        session.removeInput(self.input!)
        session.addInput(videoInput)
        session.commitConfiguration()
        self.input = videoInput
        
    }

拍照按鈕點擊事件:

    //MARK:拍照按鈕點擊事件
    func shutterCamera(){
        let videoConnection: AVCaptureConnection? = photoOutput.connection(withMediaType: AVMediaTypeVideo)
        if videoConnection == nil {
            print("take photo failed!")
            return
        }
        photoOutput.captureStillImageAsynchronously(from: videoConnection ?? AVCaptureConnection(), completionHandler: {(_ imageDataSampleBuffer: CMSampleBuffer?, _ error: Error?) -> Void in
            if imageDataSampleBuffer == nil {
                return
            }
            let imageData: Data? = AVCaptureStillImageOutput.jpegStillImageNSDataRepresentation(imageDataSampleBuffer)   //照片數據流
            self.image = UIImage(data: imageData!)
            self.session.stopRunning()
            self.imageView = UIImageView(frame: self.previewLayer.frame)
            self.view.insertSubview(self.imageView!, belowSubview: self.photoButton!)
            self.imageView?.layer.masksToBounds = true
            self.imageView?.image = self.image
            print("image size = \(NSStringFromCGSize((self.image?.size)!))")
        })
    }

閃光燈切換事件:

    //MARK: 閃光燈開關
    func flashAction(){
        try? device.lockForConfiguration()
        switch device.flashMode.rawValue {
        case 0:
            device!.flashMode = AVCaptureFlashMode.on
            flashBtn?.setImage(#imageLiteral(resourceName: "flash-on"), for: .normal)
            break
        case 1:
        device!.flashMode = AVCaptureFlashMode.auto
        flashBtn?.setImage(#imageLiteral(resourceName: "flash-A"), for: .normal)
            break
        default:
         device!.flashMode = AVCaptureFlashMode.off
            flashBtn?.setImage(#imageLiteral(resourceName: "flash-off"), for: .normal)
        }
        device.unlockForConfiguration()
        
    }

取消按鈕點擊事件:

    //MARK:取消按鈕
    func cancelActin(){
        self.imageView?.removeFromSuperview()
        self.session.startRunning()
    }

github源碼地址:https://github.com/pheromone/swift_custom_camera