iOS使用视听媒体框架AVFoundation实现照片拍摄

#import "HWPhotoVC.h"
#import <AVFoundation/AVFoundation.h>
 
@interface HWPhotoVC ()
 
@property (nonatomic, strong) AVCaptureSession *captureSession;//负责输入和输出设备之间的数据传递
@property (nonatomic, strong) AVCaptureDeviceInput *captureDeviceInput;//负责从AVCaptureDevice获得输入数据
@property (nonatomic, strong) AVCaptureStillImageOutput *captureStillImageOutput;//照片输出流
@property (nonatomic, strong) AVCaptureVideoPreviewLayer *captureVideoPreviewLayer;//相机拍摄预览图层
@property (nonatomic, weak) UIView *containerView;//内容视图
@property (nonatomic, weak) UIImageView *focusCursor;//聚焦按钮
@property (nonatomic, weak) UIImageView *imgView;//拍摄照片
 
@end
 
@implementation HWPhotoVC
 
- (void)viewDidLoad {
 [super viewDidLoad];
 
 self.navigationItem.title = @"拍照";
 self.view.backgroundColor = [UIColor whiteColor];
 
 //创建控件
 [self creatControl];
}
 
- (void)viewWillAppear:(BOOL)animated
{
 [super viewWillAppear:animated];
 
 //初始化信息
 [self initPhotoInfo];
}
 
- (void)viewDidAppear:(BOOL)animated
{
 [super viewDidAppear:animated];
 
 [self.captureSession startRunning];
}
 
- (void)viewDidDisappear:(BOOL)animated
{
 [super viewDidDisappear:animated];
 
 [self.captureSession stopRunning];
}
 
- (void)creatControl
{
 CGFloat btnW = 150.f;
 CGFloat btnH = 40.f;
 CGFloat marginY = 20.f;
 CGFloat w = [UIScreen mainScreen].bounds.size.width;
 CGFloat h = [UIScreen mainScreen].bounds.size.height;
 
 //内容视图
 CGFloat containerViewH = h - 64 - btnH - marginY * 3;
 UIView *containerView = [[UIView alloc] initWithFrame:CGRectMake(10, 64 + marginY, w - 20, containerViewH)];
 containerView.backgroundColor = [UIColor whiteColor];
 containerView.layer.borderWidth = 1.f;
 containerView.layer.borderColor = [[UIColor grayColor] CGColor];
 [self.view addSubview:containerView];
 _containerView = containerView;
 
 //摄像头切换按钮
 CGFloat cameraSwitchBtnW = 50.f;
 CGFloat cameraSwitchBtnMargin = 10.f;
 UIButton *cameraSwitchBtn = [[UIButton alloc] initWithFrame:CGRectMake(containerView.bounds.size.width - cameraSwitchBtnW - cameraSwitchBtnMargin, 64 + marginY + cameraSwitchBtnMargin, cameraSwitchBtnW, cameraSwitchBtnW)];
 [cameraSwitchBtn setImage:[UIImage imageNamed:@"camera_switch"] forState:UIControlStateNormal];
 [cameraSwitchBtn addTarget:self action:@selector(cameraSwitchBtnOnClick) forControlEvents:UIControlEventTouchUpInside];
 [self.view addSubview:cameraSwitchBtn];
 
 //聚焦图片
 UIImageView *focusCursor = [[UIImageView alloc] initWithFrame:CGRectMake(50, 50, 75, 75)];
 focusCursor.alpha = 0;
 focusCursor.image = [UIImage imageNamed:@"camera_focus_red"];
 [containerView addSubview:focusCursor];
 _focusCursor = focusCursor;
 
 //拍摄照片容器
 UIImageView *imgView = [[UIImageView alloc] initWithFrame:containerView.frame];
 imgView.hidden = YES;
 imgView.layer.borderWidth = 1.f;
 imgView.layer.borderColor = [[UIColor grayColor] CGColor];
 imgView.contentMode = UIViewContentModeScaleAspectFill;
 imgView.clipsToBounds = YES;
 [self.view addSubview:imgView];
 _imgView = imgView;
 
 //按钮
 NSArray *titleArray = @[@"拍摄照片", @"重新拍摄"];
 CGFloat btnY = CGRectGetMaxY(containerView.frame) + marginY;
 CGFloat margin = (w - btnW * titleArray.count) / (titleArray.count + 1);
 for (int i = 0; i < titleArray.count; i++) {
  CGFloat btnX = margin + (margin + btnW) * i;
  UIButton *btn = [[UIButton alloc] initWithFrame:CGRectMake(btnX, btnY, btnW, btnH)];
  btn.tag = 1000 + i;
  [btn setTitle:titleArray[i] forState:UIControlStateNormal];
  btn.backgroundColor = [UIColor orangeColor];
  btn.layer.cornerRadius = 2.0f;
  btn.layer.masksToBounds = YES;
  if (i == 1) {
   btn.hidden = YES;
  }
  [btn addTarget:self action:@selector(btnOnClick:) forControlEvents:UIControlEventTouchUpInside];
  [self.view addSubview:btn];
 }
}
 
- (void)initPhotoInfo
{
 //初始化会话
 _captureSession = [[AVCaptureSession alloc] init];
 
 //设置分辨率
 if ([_captureSession canSetSessionPreset:AVCaptureSessionPreset1280x720]) {
  _captureSession.sessionPreset = AVCaptureSessionPreset1280x720;
 }
 
 //获得输入设备,取得后置摄像头
 AVCaptureDevice *captureDevice = [self getCameraDeviceWithPosition:AVCaptureDevicePositionBack];
 if (!captureDevice) {
  NSLog(@"取得后置摄像头时出现问题");
  return;
 }
 
 NSError *error = nil;
 //根据输入设备初始化设备输入对象，用于获得输入数据
 _captureDeviceInput = [[AVCaptureDeviceInput alloc]initWithDevice:captureDevice error:&error];
 if (error) {
  NSLog(@"取得设备输入对象时出错，错误原因：%@", error.localizedDescription);
  return;
 }
 
 //初始化设备输出对象，用于获得输出数据
 _captureStillImageOutput = [[AVCaptureStillImageOutput alloc] init];
 NSDictionary *outputSettings = @{AVVideoCodecKey:AVVideoCodecJPEG};
 //输出设置
 [_captureStillImageOutput setOutputSettings:outputSettings];
 
 //将设备输入添加到会话中
 if ([_captureSession canAddInput:_captureDeviceInput]) {
  [_captureSession addInput:_captureDeviceInput];
 }
 
 //将设备输出添加到会话中
 if ([_captureSession canAddOutput:_captureStillImageOutput]) {
  [_captureSession addOutput:_captureStillImageOutput];
 }
 
 //创建视频预览层，用于实时展示摄像头状态
 _captureVideoPreviewLayer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.captureSession];
 
 //摄像头方向
 AVCaptureConnection *captureConnection = [self.captureVideoPreviewLayer connection];
 captureConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
 
 CALayer *layer = _containerView.layer;
 layer.masksToBounds = YES;
 
 _captureVideoPreviewLayer.frame = layer.bounds;
 //填充模式
 _captureVideoPreviewLayer.videoGravity = AVLayerVideoGravityResizeAspectFill;
 //将视频预览层添加到界面中
 [layer insertSublayer:_captureVideoPreviewLayer below:self.focusCursor.layer];
 
 [self addNotificationToCaptureDevice:captureDevice];
 [self addGenstureRecognizer];
}
 
- (void)btnOnClick:(UIButton *)btn
{
 if (btn.tag == 1000) {
  //拍摄照片
  [self photoBtnOnClick];
  
 }else if (btn.tag == 1001) {
  //重新拍摄
  [self resetPhoto];
 }
}
 
#pragma mark 拍照
- (void)photoBtnOnClick
{
 //根据设备输出获得连接
 AVCaptureConnection *captureConnection = [self.captureStillImageOutput connectionWithMediaType:AVMediaTypeVideo];
 captureConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
 
 //根据连接取得设备输出的数据
 [self.captureStillImageOutput captureStillImageAsynchronouslyFromConnection:captureConnection completionHandler:^(CMSampleBufferRef imageDataSampleBuffer, NSError *error) {
  if (imageDataSampleBuffer) {
   NSData *imageData = [AVCaptureStillImageOutput jpegStillImageNSDataRepresentation:imageDataSampleBuffer];
   UIImage *image = [UIImage imageWithData:imageData];
   _imgView.image = image;
   _imgView.hidden = NO;
  }
 }];
 
 UIButton *btn = (UIButton *)[self.view viewWithTag:1001];
 btn.hidden = NO;
}
 
//重新拍摄
- (void)resetPhoto
{
 _imgView.hidden = YES;
 UIButton *btn = (UIButton *)[self.view viewWithTag:1001];
 btn.hidden = YES;
}
 
#pragma mark - 通知
//给输入设备添加通知
- (void)addNotificationToCaptureDevice:(AVCaptureDevice *)captureDevice
{
 //注意添加区域改变捕获通知必须首先设置设备允许捕获
 [self changeDeviceProperty:^(AVCaptureDevice *captureDevice) {
  captureDevice.subjectAreaChangeMonitoringEnabled = YES;
 }];
 
 //捕获区域发生改变
 [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(areaChange:) name:AVCaptureDeviceSubjectAreaDidChangeNotification object:captureDevice];
}
 
- (void)removeNotificationFromCaptureDevice:(AVCaptureDevice *)captureDevice
{
 [[NSNotificationCenter defaultCenter] removeObserver:self name:AVCaptureDeviceSubjectAreaDidChangeNotification object:captureDevice];
}
 
//移除所有通知
- (void)removeNotification
{
 [[NSNotificationCenter defaultCenter] removeObserver:self];
}
 
//设备连接成功
- (void)deviceConnected:(NSNotification *)notification
{
 NSLog(@"设备已连接...");
}
 
//设备连接断开
- (void)deviceDisconnected:(NSNotification *)notification
{
 NSLog(@"设备已断开.");
}
 
//捕获区域改变
- (void)areaChange:(NSNotification *)notification
{
 NSLog(@"捕获区域改变...");
}
 
#pragma mark - 私有方法
//取得指定位置的摄像头
- (AVCaptureDevice *)getCameraDeviceWithPosition:(AVCaptureDevicePosition )position
{
 NSArray *cameras = [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo];
 for (AVCaptureDevice *camera in cameras) {
  if ([camera position] == position) {
   return camera;
  }
 }
 
 return nil;
}
 
#pragma mark 切换前后摄像头
- (void)cameraSwitchBtnOnClick
{
 AVCaptureDevice *currentDevice = [self.captureDeviceInput device];
 AVCaptureDevicePosition currentPosition = [currentDevice position];
 [self removeNotificationFromCaptureDevice:currentDevice];
 
 AVCaptureDevice *toChangeDevice;
 AVCaptureDevicePosition toChangePosition = AVCaptureDevicePositionFront;
 if (currentPosition == AVCaptureDevicePositionUnspecified || currentPosition == AVCaptureDevicePositionFront) {
  toChangePosition = AVCaptureDevicePositionBack;
 }
 toChangeDevice = [self getCameraDeviceWithPosition:toChangePosition];
 [self addNotificationToCaptureDevice:toChangeDevice];
 //获得要调整的设备输入对象
 AVCaptureDeviceInput *toChangeDeviceInput = [[AVCaptureDeviceInput alloc] initWithDevice:toChangeDevice error:nil];
 
 //改变会话的配置前一定要先开启配置，配置完成后提交配置改变
 [self.captureSession beginConfiguration];
 //移除原有输入对象
 [self.captureSession removeInput:self.captureDeviceInput];
 //添加新的输入对象
 if ([self.captureSession canAddInput:toChangeDeviceInput]) {
  [self.captureSession addInput:toChangeDeviceInput];
  self.captureDeviceInput = toChangeDeviceInput;
 }
 //提交会话配置
 [self.captureSession commitConfiguration];
}
 
//改变设备属性的统一操作方法
- (void)changeDeviceProperty:(void (^)(AVCaptureDevice *))propertyChange
{
 AVCaptureDevice *captureDevice = [self.captureDeviceInput device];
 NSError *error;
 //注意改变设备属性前一定要首先调用lockForConfiguration:调用完之后使用unlockForConfiguration方法解锁
 if ([captureDevice lockForConfiguration:&error]) {
  propertyChange(captureDevice);
  [captureDevice unlockForConfiguration];
  
 }else {
  NSLog(@"设置设备属性过程发生错误，错误信息：%@", error.localizedDescription);
 }
}
 
//设置闪光灯模式
- (void)setFlashMode:(AVCaptureFlashMode)flashMode
{
 [self changeDeviceProperty:^(AVCaptureDevice *captureDevice) {
  if ([captureDevice isFlashModeSupported:flashMode]) {
   [captureDevice setFlashMode:flashMode];
  }
 }];
}
 
//设置聚焦模式
- (void)setFocusMode:(AVCaptureFocusMode)focusMode
{
 [self changeDeviceProperty:^(AVCaptureDevice *captureDevice) {
  if ([captureDevice isFocusModeSupported:focusMode]) {
   [captureDevice setFocusMode:focusMode];
  }
 }];
}
 
//设置曝光模式
- (void)setExposureMode:(AVCaptureExposureMode)exposureMode
{
 [self changeDeviceProperty:^(AVCaptureDevice *captureDevice) {
  if ([captureDevice isExposureModeSupported:exposureMode]) {
   [captureDevice setExposureMode:exposureMode];
  }
 }];
}
 
//设置聚焦点
- (void)focusWithMode:(AVCaptureFocusMode)focusMode exposureMode:(AVCaptureExposureMode)exposureMode atPoint:(CGPoint)point
{
 [self changeDeviceProperty:^(AVCaptureDevice *captureDevice) {
  if ([captureDevice isFocusModeSupported:focusMode]) {
   [captureDevice setFocusMode:AVCaptureFocusModeAutoFocus];
  }
  if ([captureDevice isFocusPointOfInterestSupported]) {
   [captureDevice setFocusPointOfInterest:point];
  }
  if ([captureDevice isExposureModeSupported:exposureMode]) {
   [captureDevice setExposureMode:AVCaptureExposureModeAutoExpose];
  }
  if ([captureDevice isExposurePointOfInterestSupported]) {
   [captureDevice setExposurePointOfInterest:point];
  }
 }];
}
 
//添加点按手势，点按时聚焦
- (void)addGenstureRecognizer
{
 [self.containerView addGestureRecognizer:[[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(tapScreen:)]];
}
 
- (void)tapScreen:(UITapGestureRecognizer *)tapGesture
{
 CGPoint point = [tapGesture locationInView:self.containerView];
 //将UI坐标转化为摄像头坐标
 CGPoint cameraPoint = [self.captureVideoPreviewLayer captureDevicePointOfInterestForPoint:point];
 [self setFocusCursorWithPoint:point];
 [self focusWithMode:AVCaptureFocusModeAutoFocus exposureMode:AVCaptureExposureModeAutoExpose atPoint:cameraPoint];
}
 
//设置聚焦光标位置
- (void)setFocusCursorWithPoint:(CGPoint)point
{
 self.focusCursor.center = point;
 self.focusCursor.transform = CGAffineTransformMakeScale(1.5, 1.5);
 self.focusCursor.alpha = 1.0;
 [UIView animateWithDuration:1.0 animations:^{
  self.focusCursor.transform = CGAffineTransformIdentity;
 } completion:^(BOOL finished) {
  self.focusCursor.alpha = 0;
 }];
}
 
- (void)dealloc
{
 [self removeNotification];
}
 
@end