erpsuiteios/Redant Drivers/Apex And Drivers/Camera/RACameraViewController.m

//
//  RACameraViewController.m
//  Apex And Drivers
//
//  Created by Jack on 2018/6/5.
//  Copyright © 2018年 USAI. All rights reserved.
//

#import "RACameraViewController.h"
#import <AVFoundation/AVFoundation.h>
#import "RATakePhotoPreviewController.h"

@interface RACameraViewController ()

@property (nonatomic,strong) IBOutlet UIView *previewContainer;
@property (nonatomic,strong) IBOutlet UIView *controlContanier;

#pragma mark - Camera

@property (nonatomic,strong) AVCaptureDevice *captureDevice;
@property (nonatomic,strong) AVCaptureSession *captureSession;
@property (nonatomic,strong) AVCaptureDeviceInput *captureInput;
@property (nonatomic,strong) AVCaptureStillImageOutput *captureOutput;
@property (nonatomic,strong) AVCaptureVideoPreviewLayer *previewLayer;

@property (nonatomic,assign) BOOL cameraInitial;

@end

@implementation RACameraViewController

+ (instancetype)viewControllerFromStoryboard {
    RACameraViewController *cameraVC = [[UIStoryboard storyboardWithName:@"Camera" bundle:nil] instantiateViewControllerWithIdentifier:[self storyboardID]];
    return cameraVC;
}

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    
    [self initCapture];
    [self initTakePicture];
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

- (void)viewWillAppear:(BOOL)animated {
    [super viewWillAppear:animated];
    
    if (self.cameraInitial) {
        [self.captureSession startRunning];
    }
}

- (void)viewWillDisappear:(BOOL)animated {
    [super viewWillDisappear:animated];
    
    if ([self.captureSession isRunning]) {
        [self.captureSession stopRunning];
    }
}

- (void)viewDidLayoutSubviews {
    [super viewDidLayoutSubviews];
    
    self.previewLayer.frame = self.previewContainer.bounds;
    
    if (self.previewLayer.connection.isVideoOrientationSupported) {
        self.previewLayer.connection.videoOrientation = [self captureVideoOrientation];
    }
    
    if ([self.captureOutput connectionWithMediaType:AVMediaTypeVideo].isVideoOrientationSupported) {
        [self.captureOutput connectionWithMediaType:AVMediaTypeVideo].videoOrientation = [self captureVideoOrientation];
    }
}

#pragma mark - Capture

- (void)initCapture {
    
    self.cameraInitial = NO;
    
    self.captureSession = [[AVCaptureSession alloc] init];
    
    self.captureDevice = [self videoDevicePosition:AVCaptureDevicePositionBack];
    if (!self.captureDevice) {
        NSLog(@"there is no capture device while init camera");
        return;
    }
    
    NSError *error;
    self.captureInput = [[AVCaptureDeviceInput alloc] initWithDevice:self.captureDevice error:&error];
    
    if (error) {
        NSLog(@"init camera error: %@",error);
        return;
    }
    
    [self.captureSession beginConfiguration];
    
    if ([_captureSession canSetSessionPreset:AVCaptureSessionPresetHigh]) {
        _captureSession.sessionPreset = AVCaptureSessionPresetHigh;
    }
    
    if ([self.captureSession canAddInput:self.captureInput]) {
        
        
        [self.captureSession addInput:self.captureInput];
        
//        if ([self.captureSession canSetSessionPreset:AVCaptureSessionPreset1920x1080]) {
//            self.captureSession.sessionPreset = AVCaptureSessionPreset1920x1080;
//        } else {
//            if ([self.captureSession canSetSessionPreset:AVCaptureSessionPresetHigh]) {
//                self.captureSession.sessionPreset = AVCaptureSessionPresetHigh;
//            }
//        }
        
    } else {
        NSLog(@"init camera can't add input");
        return;
    }
    
    [self.captureSession commitConfiguration];
    
    self.previewLayer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.captureSession];
    self.previewLayer.videoGravity = AVLayerVideoGravityResizeAspectFill;
    [self.previewContainer.layer addSublayer:self.previewLayer];
    
    self.cameraInitial = YES;
}

- (void)initTakePicture {
    if (!self.cameraInitial) {
        return;
    }
    self.captureOutput = [[AVCaptureStillImageOutput alloc] init];
    
    NSDictionary *setting = @{AVVideoCodecKey:AVVideoCodecJPEG};
    [self.captureOutput setOutputSettings:setting];
    
    if ([self.captureSession canAddOutput:self.captureOutput]) {
        [self.captureSession addOutput:self.captureOutput];
    }
    
    AVCaptureConnection *connection = [self.captureOutput connectionWithMediaType:AVMediaTypeVideo];
    if (connection.isVideoOrientationSupported) { // addOutput之后判断，否则一直都是false
        connection.videoOrientation = [self captureVideoOrientation];
    }
    
}

- (AVCaptureDevice *) videoDevicePosition:(AVCaptureDevicePosition)position {

    NSArray *videoDevices = [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo];
    for (AVCaptureDevice *device in videoDevices) {
        if ([device position] == position) {
            return device;
        }
    }
    
    return nil;
}

- (AVCaptureVideoOrientation)captureVideoOrientation {
    AVCaptureVideoOrientation result;
    
    UIDeviceOrientation deviceOrientation = [UIDevice currentDevice].orientation;
    switch (deviceOrientation) {
        case UIDeviceOrientationPortrait:
        case UIDeviceOrientationFaceUp:
        case UIDeviceOrientationFaceDown:
            result = AVCaptureVideoOrientationPortrait;
            break;
        case UIDeviceOrientationPortraitUpsideDown:
            //如果这里设置成AVCaptureVideoOrientationPortraitUpsideDown，则视频方向和拍摄时的方向是相反的。
            result = AVCaptureVideoOrientationPortrait;
            break;
        case UIDeviceOrientationLandscapeLeft:
            result = AVCaptureVideoOrientationLandscapeRight;
            break;
        case UIDeviceOrientationLandscapeRight:
            result = AVCaptureVideoOrientationLandscapeLeft;
            break;
        default:
            result = AVCaptureVideoOrientationPortrait;
            break;
    }
    
    return result;
}

#pragma mark - Action

- (IBAction)takePictureBtnClick:(UIButton *)sender {
    
    if (!self.cameraInitial) {
        return;
    }
    
    AVCaptureConnection *connection = [self.captureOutput connectionWithMediaType:AVMediaTypeVideo];
    [self.captureOutput captureStillImageAsynchronouslyFromConnection:connection completionHandler:^(CMSampleBufferRef  _Nullable imageDataSampleBuffer, NSError * _Nullable error) {
        
        if (imageDataSampleBuffer) {
            NSData *imageData=[AVCaptureStillImageOutput jpegStillImageNSDataRepresentation:imageDataSampleBuffer];
            UIImage *image=[UIImage imageWithData:imageData];
            
            __weak typeof(self) weakSelf = self;
            RATakePhotoPreviewController *preVC = [RATakePhotoPreviewController viewControllerFromStoryboard];
            preVC.photoHandler = ^(UIImage *img){
                if (weakSelf) {
                    __strong typeof(weakSelf) strongSelf = weakSelf;
                    if (strongSelf.completion) {
                        strongSelf.completion(img);
                    }
                }
            };
            preVC.preImage = image;
            preVC.popTo = self.fromVC;
            [self.navigationController pushViewController:preVC animated:YES];
            
        } else {
            NSLog(@"take picture failed: %@",error);
        }
        
    }];
}


@end