iOS 10自定义相机功能

//// TGCameraVC.swift// TGPhotoPicker//// Created by targetcloud on 2017/7/25.// Copyright © 2017年 targetcloud. All rights reserved.// import UIKitimport AVFoundationimport Photos @available(iOS 10.0, *)class TGCameraVC: UIViewController {  var callbackPicutureData: ((Data?) -> ())?  private var device: AVCaptureDevice? private var input: AVCaptureDeviceInput? private var imageOutput: AVCapturePhotoOutput? private var session: AVCaptureSession? private var previewLayer: AVCaptureVideoPreviewLayer? fileprivate var showImageContainerView: UIView? fileprivate var showImageView: UIImageView? fileprivate var picData: Data? private var flashMode: AVCaptureFlashMode = .auto private weak var flashButton: UIButton?  override func viewDidLoad() {  super.viewDidLoad()   setupCamera()  setupUI()    if #available(iOS 9.0, *) {   let isVCBased = Bundle.main.infoDictionary?["UIViewControllerBasedStatusBarAppearance"] as? Bool ?? false   if !isVCBased{    UIApplication.shared.setStatusBarHidden(false, with: .none)   }  }else {   UIApplication.shared.statusBarStyle = .lightContent   UIApplication.shared.setStatusBarHidden(false, with: .none)  } }  override var prefersStatusBarHidden: Bool{  return false }  override var preferredStatusBarStyle: UIStatusBarStyle {  return .lightContent }  private func setupCamera() {  AVCaptureDevice.requestAccess(forMediaType: AVMediaTypeVideo) { success in   if !success {    let alertVC = UIAlertController(title: TGPhotoPickerConfig.shared.cameraUsage, message: TGPhotoPickerConfig.shared.cameraUsageTip, preferredStyle: .actionSheet)    alertVC.addAction(UIAlertAction(title: TGPhotoPickerConfig.shared.confirmTitle, style: .default, handler: nil))    self.present(alertVC, animated: true, completion: nil)   }  }  device = cameraWithPosistion(.back)  input = try? AVCaptureDeviceInput(device: device)  guard input != nil else {   return  }    imageOutput = AVCapturePhotoOutput()  session = AVCaptureSession()  session?.beginConfiguration()  session?.sessionPreset = TGPhotoPickerConfig.shared.sessionPreset  if session!.canAddInput(input) {   session!.addInput(input)  }  if session!.canAddOutput(imageOutput) {   session!.addOutput(imageOutput)  }  previewLayer = AVCaptureVideoPreviewLayer(session: session)  previewLayer?.frame = view.bounds  previewLayer?.videoGravity = TGPhotoPickerConfig.shared.videoGravity  view.layer.addSublayer(previewLayer!)  session?.commitConfiguration()  session?.startRunning() }  private func cameraWithPosistion(_ position: AVCaptureDevicePosition) -> AVCaptureDevice {  let type = AVCaptureDeviceType(rawValue: TGPhotoPickerConfig.shared.captureDeviceType.rawValue)  return AVCaptureDevice.defaultDevice(withDeviceType: type, mediaType: AVMediaTypeVideo, position: position) }  private func setupUI() {  let takeButton = UIButton(frame: CGRect(x: 0, y: 0, width: TGPhotoPickerConfig.shared.takeWH, height: TGPhotoPickerConfig.shared.takeWH))  takeButton.center = CGPoint(x: UIScreen.main.bounds.width / 2, y: UIScreen.main.bounds.height - TGPhotoPickerConfig.shared.buttonEdge.bottom)  takeButton.setImage(UIImage.size(width: TGPhotoPickerConfig.shared.takeWH, height: TGPhotoPickerConfig.shared.takeWH).border(width: 3).border(color: .white).color(.clear).corner(radius: TGPhotoPickerConfig.shared.takeWH / 2).image +   UIImage.size(width: TGPhotoPickerConfig.shared.takeWH - 10, height: TGPhotoPickerConfig.shared.takeWH - 10).color(UIColor(white: 0.95, alpha: 1) ).corner(radius: (TGPhotoPickerConfig.shared.takeWH - 10) / 2).image, for: .normal)  takeButton.setImage(UIImage.size(width: TGPhotoPickerConfig.shared.takeWH, height: TGPhotoPickerConfig.shared.takeWH).border(width: 3).border(color: .white).color(.clear).corner(radius: TGPhotoPickerConfig.shared.takeWH / 2).image +   UIImage.size(width: TGPhotoPickerConfig.shared.takeWH - 10, height: TGPhotoPickerConfig.shared.takeWH - 10).color(UIColor(white: 0.8, alpha: 1) ).corner(radius: (TGPhotoPickerConfig.shared.takeWH - 10) / 2).image, for: .highlighted)  takeButton.addTarget(self, action: #selector(takePhotoAction), for: .touchUpInside)  view.addSubview(takeButton)    let cameraChangeButton = UIButton(frame: CGRect(x: 0, y: 0, width: TGPhotoPickerConfig.shared.takeWH * 0.6, height: TGPhotoPickerConfig.shared.takeWH * 0.6))  cameraChangeButton.setImage(TGPhotoPickerConfig.getImageNo2x3xSuffix("camera"), for: .normal)  cameraChangeButton.center = CGPoint(x: UIScreen.main.bounds.width - TGPhotoPickerConfig.shared.buttonEdge.right, y: takeButton.center.y)  cameraChangeButton.addTarget(self, action: #selector(changeCameraPositionAction), for: .touchUpInside)  cameraChangeButton.contentMode = .scaleAspectFit  view.addSubview(cameraChangeButton)    let flashChangeButton = UIButton(frame: CGRect(x: 0, y: 0, width: TGPhotoPickerConfig.shared.takeWH * 0.5, height: TGPhotoPickerConfig.shared.takeWH * 0.5))  flashChangeButton.center = CGPoint(x: cameraChangeButton.center.x, y: TGPhotoPickerConfig.shared.buttonEdge.top)  flashChangeButton.setImage(TGPhotoPickerConfig.getImageNo2x3xSuffix("flashauto"), for: .normal)  flashChangeButton.addTarget(self, action: #selector(flashChangeAction), for: .touchUpInside)  flashChangeButton.contentMode = .scaleAspectFit  flashButton = flashChangeButton  view.addSubview(flashChangeButton)    let backButton = UIButton(frame: CGRect(x: 0, y: 0, width: TGPhotoPickerConfig.shared.takeWH * 0.4, height: TGPhotoPickerConfig.shared.takeWH * 0.4))  backButton.center = CGPoint(x: TGPhotoPickerConfig.shared.buttonEdge.left , y: flashChangeButton.center.y)  backButton.setImage(UIImage.size(width: TGPhotoPickerConfig.shared.takeWH * 0.4, height: TGPhotoPickerConfig.shared.takeWH * 0.4)   .corner(radius: TGPhotoPickerConfig.shared.takeWH * 0.2)   .color(.clear)   .border(color: UIColor.white.withAlphaComponent(0.7))   .border(width: TGPhotoPickerConfig.shared.isShowBorder ? TGPhotoPickerConfig.shared.checkboxLineW : 0)   .image   .with({ context in    context.setLineCap(.round)    UIColor.white.setStroke()    context.setLineWidth(TGPhotoPickerConfig.shared.checkboxLineW)    let WH = TGPhotoPickerConfig.shared.takeWH * 0.4    context.move(to: CGPoint(x: WH * 0.6, y: WH * 0.2))    context.addLine(to: CGPoint(x: WH * 0.35, y: WH * 0.5))    context.move(to: CGPoint(x: WH * 0.35, y: WH * 0.5))    context.addLine(to: CGPoint(x: WH * 0.6, y: WH * 0.8))    context.strokePath()   }), for: .normal)  backButton.contentMode = .scaleAspectFit  backButton.addTarget(self, action: #selector(backAction), for: .touchUpInside)  view.addSubview(backButton)    showImageContainerView = UIView(frame: view.bounds)  showImageContainerView?.backgroundColor = TGPhotoPickerConfig.shared.previewBGColor  view.addSubview(showImageContainerView!)    let height = showImageContainerView!.bounds.height - TGPhotoPickerConfig.shared.takeWH - TGPhotoPickerConfig.shared.buttonEdge.bottom - TGPhotoPickerConfig.shared.previewPadding * 2  showImageView = UIImageView(frame: CGRect(x: TGPhotoPickerConfig.shared.previewPadding, y: TGPhotoPickerConfig.shared.previewPadding * 2, width: showImageContainerView!.bounds.width - 2 * TGPhotoPickerConfig.shared.previewPadding, height: height))  showImageView?.contentMode = .scaleAspectFit  showImageContainerView?.addSubview(showImageView!)  showImageContainerView?.isHidden = true    let giveupButton = createImageOperatorButton(nil, CGPoint(x: TGPhotoPickerConfig.shared.takeWH * 1.5, y: showImageContainerView!.bounds.height - TGPhotoPickerConfig.shared.takeWH * 1.5), TGPhotoPickerConfig.shared.getCheckboxImage(true, true, .circle, TGPhotoPickerConfig.shared.takeWH * 0.7).unselect)  giveupButton.addTarget(self, action: #selector(giveupImageAction), for: .touchUpInside)  showImageContainerView?.addSubview(giveupButton)    let ensureButton = createImageOperatorButton(nil, CGPoint(x: showImageContainerView!.bounds.width - TGPhotoPickerConfig.shared.takeWH * 1.5, y: showImageContainerView!.bounds.height - TGPhotoPickerConfig.shared.takeWH * 1.5), TGPhotoPickerConfig.shared.getCheckboxImage(true, false, .circle, TGPhotoPickerConfig.shared.takeWH * 0.7).select)  ensureButton.addTarget(self, action: #selector(useImageAction), for: .touchUpInside)  showImageContainerView?.addSubview(ensureButton) }  private func createImageOperatorButton(_ title: String?, _ center: CGPoint, _ img: UIImage?) -> UIButton {  let btn = UIButton(frame: CGRect(x: 0, y: 0, width: TGPhotoPickerConfig.shared.takeWH * 0.7, height: TGPhotoPickerConfig.shared.takeWH * 0.7))  btn.center = center  btn.setTitle(title, for: .normal)  btn.setImage(img, for: .normal)  btn.contentMode = .scaleAspectFit  return btn }  @objc private func flashChangeAction() {  switch flashMode {  case .auto:   flashMode = .on   flashButton?.setImage(TGPhotoPickerConfig.getImageNo2x3xSuffix("flash"), for: .normal)  case .on:   flashMode = .off   flashButton?.setImage(TGPhotoPickerConfig.getImageNo2x3xSuffix("flashno"), for: .normal)  case .off:   flashMode = .auto   flashButton?.setImage(TGPhotoPickerConfig.getImageNo2x3xSuffix("flashauto"), for: .normal)  } }  @objc private func backAction() {  dismiss(animated: true, completion: nil) }  @objc private func takePhotoAction() {  let connection = imageOutput?.connection(withMediaType: AVMediaTypeVideo)  guard connection != nil else {   return  }  let photoSettings = AVCapturePhotoSettings()  photoSettings.flashMode = flashMode  imageOutput?.capturePhoto(with: photoSettings, delegate: self) }  @objc private func changeCameraPositionAction() {  let animation = CATransition()  animation.duration = 0.5  animation.timingFunction = CAMediaTimingFunction(name: kCAMediaTimingFunctionEaseInEaseOut)  animation.type = TGPhotoPickerConfig.shared.transitionType    let newDevice: AVCaptureDevice!  let newInput: AVCaptureDeviceInput?  let position = input?.device.position  if position == .front {   newDevice = cameraWithPosistion(.back)   animation.subtype = kCATransitionFromLeft  } else {   newDevice = cameraWithPosistion(.front)   animation.subtype = kCATransitionFromRight  }  newInput = try? AVCaptureDeviceInput(device: newDevice)  guard newInput != nil else{   return  }    previewLayer?.add(animation, forKey: nil)    session?.beginConfiguration()  session?.removeInput(input)  if session!.canAddInput(newInput) {   session?.addInput(newInput!)   input = newInput  } else {   session?.addInput(input)  }  session?.commitConfiguration() }  @objc private func giveupImageAction() {  showImageView?.image = UIImage()  showImageContainerView?.isHidden = true }  @objc private func useImageAction() {  callbackPicutureData?(picData)  dismiss(animated: true, completion: nil) }} @available(iOS 10.0, *)extension TGCameraVC: AVCapturePhotoCaptureDelegate { func capture(_ captureOutput: AVCapturePhotoOutput, didFinishProcessingPhotoSampleBuffer photoSampleBuffer: CMSampleBuffer?, previewPhotoSampleBuffer: CMSampleBuffer?, resolvedSettings: AVCaptureResolvedPhotoSettings, bracketSettings: AVCaptureBracketedStillImageSettings?, error: Error?) {  if error != nil {   print("error = /(String(describing: error?.localizedDescription))")  } else {   if let imageData = AVCapturePhotoOutput.jpegPhotoDataRepresentation(forJPEGSampleBuffer: photoSampleBuffer!, previewPhotoSampleBuffer: previewPhotoSampleBuffer){    picData = imageData    showImageContainerView?.isHidden = false    showImageView?.image = UIImage(data: imageData)    if TGPhotoPickerConfig.shared.saveImageToPhotoAlbum{     self.saveImageToPhotoAlbum(UIImage(data: imageData)!)    }   }  } }  fileprivate func saveImageToPhotoAlbum(_ savedImage:UIImage){  UIImageWriteToSavedPhotosAlbum(savedImage, self, #selector(imageDidFinishSavingWithErrorContextInfo), nil) }  @objc fileprivate func imageDidFinishSavingWithErrorContextInfo(image:UIImage,error:NSError?,contextInfo:UnsafeMutableRawPointer?){  if canUseAlbum(){   let msg = (error != nil) ? TGPhotoPickerConfig.shared.saveImageFailTip : TGPhotoPickerConfig.shared.saveImageSuccessTip   let alert = UIAlertView(title: TGPhotoPickerConfig.shared.saveImageTip, message: msg, delegate: self, cancelButtonTitle: TGPhotoPickerConfig.shared.confirmTitle)   alert.show()  } }  fileprivate func canUseAlbum()-> Bool{  if PHPhotoLibrary.authorizationStatus() != PHAuthorizationStatus.authorized {   let alertView = UIAlertView(title: TGPhotoPickerConfig.shared.PhotoLibraryUsage, message: TGPhotoPickerConfig.shared.PhotoLibraryUsageTip, delegate: nil, cancelButtonTitle: TGPhotoPickerConfig.shared.confirmTitle, otherButtonTitles: TGPhotoPickerConfig.shared.cancelTitle)   alertView.tag = TGPhotoPickerConfig.shared.alertViewTag   alertView.show()   return false  }else{   return true  } }}