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 =
            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 $.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: , y: , 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 - , y: , width: , height: )
         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 *  / 2.0 - , y: kScreenWH.height - , width: , height: )
         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: , y: , width: , height: )
         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: , y: kScreenWH.height - , width: , height: )
         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 $.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 :
             device!.flashMode = AVCaptureFlashMode.on
             flashBtn?.setImage(#imageLiteral(resourceName: "flash-on"), for: .normal)
             break
         case :
         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