#import "NJCamera.h" @interface NJCamera () @property (nonatomic) dispatch_queue_t sessionQueue; @property (strong, nonatomic) AVCaptureSession *session; // 预览页 @property (strong, nonatomic) AVCaptureVideoPreviewLayer *captureVideoPreviewLayer; // 摄像头 @property (strong, nonatomic) AVCaptureDevice *videoCaptureDevice; // 输入源 @property (nonatomic, strong) AVCaptureDeviceInput *captureDeviceInput; // 图像输出 @property (nonatomic, strong) AVCapturePhotoOutput *photoOutput; // 二维码输出 @property (nonatomic, strong) AVCaptureMetadataOutput *metadataOutput; // 视频流输出 @property (nonatomic, strong) AVCaptureVideoDataOutput *videoOutput; @property (copy, nonatomic) NSString *cameraQuality; //前后置摄像头 @property (nonatomic) NJCameraPosition cameraPosition; // 输出类型 @property (nonatomic) NJCameraOutputType outPutType; // 闪光灯 @property (nonatomic) NJCameraFlash flash; // 预览页方向 @property (nonatomic) AVCaptureVideoOrientation videoOrientation; @property (nonatomic, assign) BOOL isSupportAutoVideorientation; @end NSString *const NJCameraErrorDomain = @"NJCameraErrorDomain"; @implementation NJCamera - (BOOL)isFlashAvailable { return self.videoCaptureDevice.hasFlash && self.videoCaptureDevice.isFlashAvailable; } - (void)updateFlashMode:(NJCameraFlash)cameraFlash { self.flash = cameraFlash; } - (void)getQRCodeWith:(void (^)(NJCamera *camera, NSString *result))onQRCode { self.onQRCode = onQRCode; } // 照片输出 - (AVCapturePhotoOutput *)photoOutput { if (_photoOutput == nil) { _photoOutput = [[AVCapturePhotoOutput alloc] init]; } return _photoOutput; } // 二维码输出 -(AVCaptureMetadataOutput *)metadataOutput{ if (_metadataOutput == nil) { _metadataOutput = [[AVCaptureMetadataOutput alloc]init]; [_metadataOutput setMetadataObjectsDelegate:self queue:dispatch_get_main_queue()]; } return _metadataOutput; } // 实时视频流输出 -(AVCaptureVideoDataOutput *)videoOutput { if (_videoOutput == nil) { _videoOutput = [[AVCaptureVideoDataOutput alloc] init]; [_videoOutput setSampleBufferDelegate:self queue:self.sessionQueue]; _videoOutput.videoSettings = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithInt:kCVPixelFormatType_32BGRA], kCVPixelBufferPixelFormatTypeKey, nil]; } return _videoOutput; } - (UIImage *)imageConvert:(CMSampleBufferRef)sampleBuffer { if (sampleBuffer != nil) { if (CMSampleBufferIsValid(sampleBuffer) && CMSampleBufferGetImageBuffer(sampleBuffer) != nil) { CIImage *ciImage = [[CIImage alloc] initWithCVPixelBuffer:CMSampleBufferGetImageBuffer(sampleBuffer)]; return [[UIImage alloc] initWithCIImage:[ciImage imageByApplyingCGOrientation:(kCGImagePropertyOrientationRight)]]; } else { return nil; } } else { return nil; } } - (instancetype)initWithQuality:(NSString *)quality position:(NJCameraPosition)position OutputType:(NJCameraOutputType)type videoOrientation:(AVCaptureVideoOrientation)orientation isSupportAutoVideorientation:(BOOL)isSupportAutoVideorientation { if (self = [super initWithNibName:nil bundle:nil]) { [self setupWithQuality:quality position:position OutputType:type videoOrientation:orientation isSupportAutoVideorientation:isSupportAutoVideorientation]; } return self; } - (void)setupWithQuality:(NSString *)quality position:(NJCameraPosition)position OutputType:(NJCameraOutputType)type videoOrientation:(AVCaptureVideoOrientation)orientation isSupportAutoVideorientation:(BOOL)isSupportAutoVideorientation { _cameraQuality = quality; _cameraPosition = position; _outPutType = type; _videoOrientation = orientation; _isSupportAutoVideorientation = isSupportAutoVideorientation; self.sessionQueue = dispatch_queue_create("session queue", DISPATCH_QUEUE_SERIAL); } - (void)nj_attachToViewController:(UIViewController *)vc withFrame:(CGRect)frame { [vc addChildViewController:self]; self.view.frame = frame; [vc.view addSubview:self.view]; [self didMoveToParentViewController:vc]; } - (void)start { // 相机权限 AVAuthorizationStatus authorizationStatus = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo]; switch (authorizationStatus) { // 没有权限，发起授权许可 case AVAuthorizationStatusNotDetermined: { [self requestCameraPermission]; } break; // 用户已明确拒绝媒体捕获的权限。 case AVAuthorizationStatusDenied: { // 用户拒绝，无法拒绝 NSError *error = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeCameraPermission userInfo:nil]; [self passError:error]; } break; // 用户已明确授予媒体捕获权限，或者对于相关媒体类型不需要显式用户权限 case AVAuthorizationStatusAuthorized: { dispatch_async(self.sessionQueue, ^{ [self initialize]; }); } break; // 不允许用户访问媒体捕获设备。此状态通常不可见 - 用于发现设备的AVCaptureDevice类方法不会返回用户被限制访问的设备 case AVAuthorizationStatusRestricted: { NSError *error = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeCameraPermission userInfo:nil]; [self passError:error]; } break; default: { NSError *error = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeUNKownPermission userInfo:nil]; [self passError:error]; } break; } } - (void)requestCameraPermission { // 请求相机权限 dispatch_async(self.sessionQueue, ^{ if ([AVCaptureDevice respondsToSelector:@selector(requestAccessForMediaType:completionHandler:)]) { [AVCaptureDevice requestAccessForMediaType:AVMediaTypeVideo completionHandler:^(BOOL granted) { if (granted) { [self initialize]; } else { NSError *error = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeCameraPermission userInfo:nil]; [self passError:error]; } }]; } }); } - (void)stop { dispatch_async(self.sessionQueue, ^{ [self.session stopRunning]; self.session = nil; self.metadataOutput = nil; self.photoOutput = nil; dispatch_async(dispatch_get_main_queue(), ^{ [self.captureVideoPreviewLayer removeFromSuperlayer]; self.captureVideoPreviewLayer = nil; }); }); } // 初始化 - (void)initialize { if (!_session) { _session = [[AVCaptureSession alloc] init]; _session.sessionPreset = _cameraQuality; [_session beginConfiguration]; // 设备相关 AVCaptureDevicePosition devicePosition; switch (self.cameraPosition) { case NJCameraPositionRear: devicePosition = AVCaptureDevicePositionBack; break; case NJCameraPositionFront: devicePosition = AVCaptureDevicePositionFront; break; default: devicePosition = AVCaptureDevicePositionUnspecified; break; } if (devicePosition == AVCaptureDevicePositionUnspecified) { self.videoCaptureDevice = [AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeVideo]; } else { self.videoCaptureDevice = [self cameraWithPostion:devicePosition]; } // 输入源 NSError *error = nil; self.captureDeviceInput = [AVCaptureDeviceInput deviceInputWithDevice:self.videoCaptureDevice error:&error]; if (!self.captureDeviceInput) { [self passError:error]; [self.session commitConfiguration]; return; } if ([self.session canAddInput:self.captureDeviceInput]) { [self.session addInput:self.captureDeviceInput]; } else { NSError *sessionError = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeSession userInfo:nil]; [self passError:sessionError]; [self.session commitConfiguration]; return; } // 输出 switch (self.outPutType) { case NJCameraOutputTypPhoto: if ([self.session canAddOutput:self.photoOutput]) { [self.session addOutput:self.photoOutput]; } else { NSError *sessionError = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeSession userInfo:nil]; [self passError:sessionError]; [self.session commitConfiguration]; return; } break; case NJCameraOutputTypeQRCode: if([self.session canAddOutput:self.metadataOutput]){ [self.session addOutput:self.metadataOutput]; self.metadataOutput.metadataObjectTypes = self.metadataOutput.availableMetadataObjectTypes; } else { NSError *sessionError = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeSession userInfo:nil]; [self passError:sessionError]; [self.session commitConfiguration]; return; } break; case NJCameraOutputTypeVideo: if([self.session canAddOutput:self.videoOutput]){ [self.session addOutput:self.videoOutput]; } else { NSError *sessionError = [NSError errorWithDomain:NJCameraErrorDomain code:NJCameraErrorCodeSession userInfo:nil]; [self passError:sessionError]; [self.session commitConfiguration]; return; } break; } [self.session commitConfiguration]; [self.session startRunning]; dispatch_async(dispatch_get_main_queue(), ^{ // 预览layer CGRect bounds = self.view.layer.bounds; self.captureVideoPreviewLayer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.session]; self.captureVideoPreviewLayer.videoGravity = AVLayerVideoGravityResizeAspectFill; /// 判断是否支持自动修改方向 if (self.isSupportAutoVideorientation) { [self changeVideoOrientation:bounds]; } else { self.captureVideoPreviewLayer.connection.videoOrientation = self.videoOrientation; } self.captureVideoPreviewLayer.frame = bounds; self.captureVideoPreviewLayer.position = CGPointMake(CGRectGetMidX(bounds), CGRectGetMidY(bounds)); [self.view.layer insertSublayer:self.captureVideoPreviewLayer atIndex:0]; }); } } /// 修改摄像头方向 - (void)changeVideoOrientation:(CGRect)frame { self.captureVideoPreviewLayer.frame = frame; self.view.frame = frame; AVCaptureConnection *previewLayerConnection = self.captureVideoPreviewLayer.connection; UIInterfaceOrientation interfaceOrientation = [UIApplication sharedApplication].statusBarOrientation; if ([previewLayerConnection isVideoOrientationSupported]) { switch (interfaceOrientation) { case UIInterfaceOrientationPortrait: [previewLayerConnection setVideoOrientation:AVCaptureVideoOrientationPortrait]; self.videoOrientation = AVCaptureVideoOrientationPortrait; break; case UIInterfaceOrientationLandscapeRight: [previewLayerConnection setVideoOrientation:AVCaptureVideoOrientationLandscapeRight]; self.videoOrientation = AVCaptureVideoOrientationLandscapeRight; break; case UIInterfaceOrientationLandscapeLeft: [previewLayerConnection setVideoOrientation:AVCaptureVideoOrientationLandscapeLeft]; self.videoOrientation = AVCaptureVideoOrientationLandscapeLeft; break; case UIInterfaceOrientationPortraitUpsideDown: [previewLayerConnection setVideoOrientation:AVCaptureVideoOrientationPortraitUpsideDown]; self.videoOrientation = AVCaptureVideoOrientationPortraitUpsideDown; break; default: [previewLayerConnection setVideoOrientation:AVCaptureVideoOrientationPortrait]; self.videoOrientation = AVCaptureVideoOrientationPortrait; break; } } } - (AVCaptureDevice *)cameraWithPostion:(AVCaptureDevicePosition)position{ AVCaptureDeviceDiscoverySession *devicesIOS11 = [AVCaptureDeviceDiscoverySession discoverySessionWithDeviceTypes:@[AVCaptureDeviceTypeBuiltInWideAngleCamera] mediaType:AVMediaTypeVideo position:position]; NSArray *devicesIOS = devicesIOS11.devices; for (AVCaptureDevice *device in devicesIOS) { if ([device position] == position) { return device; } } return nil; } // 更改前后置 - (void)changePosition { dispatch_async(self.sessionQueue, ^{ if (self.session == nil) { return; } [self.session beginConfiguration]; [self.session removeInput:self.captureDeviceInput]; AVCaptureDevice *device = nil; if(self.captureDeviceInput.device.position == AVCaptureDevicePositionBack) { device = [self cameraWithPostion:AVCaptureDevicePositionFront]; self->_cameraPosition = NJCameraPositionFront; } else { device = [self cameraWithPostion:AVCaptureDevicePositionBack]; self->_cameraPosition = NJCameraPositionRear; } if(!device) { [self.session commitConfiguration]; return; } NSError *error = nil; AVCaptureDeviceInput *videoInput = [[AVCaptureDeviceInput alloc] initWithDevice:device error:&error]; if(error) { [self passError:error]; [self.session commitConfiguration]; return; } [self.session addInput:videoInput]; self.videoCaptureDevice = device; self.captureDeviceInput = videoInput; [self.session commitConfiguration]; }); } - (void)viewDidLoad { [super viewDidLoad]; // Do any additional setup after loading the view. } // 捕捉静态照片 - (void)startCapturePhotoAction:(void (^)(NJCamera * _Nonnull, UIImage * _Nonnull))onCapture { dispatch_async(self.sessionQueue, ^{ /// 捕获照片的时候增加容错判断 if (self.session == nil || self.photoOutput == nil || ![self.session.outputs containsObject:self.photoOutput]) { return; } AVCapturePhotoSettings *photoSettings; if ([self.photoOutput.availablePhotoCodecTypes containsObject:AVVideoCodecTypeJPEG]) { photoSettings = [AVCapturePhotoSettings photoSettingsWithFormat:@{AVVideoCodecKey:AVVideoCodecTypeJPEG}]; } else { photoSettings = [AVCapturePhotoSettings photoSettings]; } if ([self isFlashAvailable]) { AVCaptureFlashMode flashMode; if(self.flash == NJCameraFlashOn) { flashMode = AVCaptureFlashModeOn; } else if(self.flash == NJCameraFlashAuto) { flashMode = AVCaptureFlashModeAuto; } else { flashMode = AVCaptureFlashModeOff; } photoSettings.flashMode = flashMode; } [self.photoOutput capturePhotoWithSettings:photoSettings delegate:self]; self.onCapture = onCapture; }); } #pragma - AVCapturePhotoCaptureDelegate - (void)captureOutput:(AVCapturePhotoOutput *)output didFinishProcessingPhoto:(AVCapturePhoto *)photo error:(nullable NSError *)error { /// 捕获照片 if (error == nil) { NSData *imageData = photo.fileDataRepresentation; if (imageData != nil) { UIImage *image = [UIImage imageWithData:imageData]; if (self.onCapture) { /// 回到主线程 dispatch_async(dispatch_get_main_queue(), ^{ self.onCapture(self, image); }); } } } } #pragma - AVCaptureMetadataOutputObjectsDelegate - (void)captureOutput:(AVCaptureOutput *)output didOutputMetadataObjects:(NSArray<__kindof AVMetadataObject *> *)metadataObjects fromConnection:(AVCaptureConnection *)connection { if(metadataObjects.count > 0 && metadataObjects != nil){ AVMetadataMachineReadableCodeObject *metadataObject = [metadataObjects lastObject]; if ([metadataObject isKindOfClass:[AVMetadataMachineReadableCodeObject class]]) { NSString *result = metadataObject.stringValue; if (self.onQRCode) { self.onQRCode(self, result); } } } } #pragma - AVCaptureVideoDataOutputSampleBufferDelegate - (void)captureOutput:(AVCaptureOutput *)output didOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection { if (output == _videoOutput) { UIImage *image = [self imageConvert:sampleBuffer]; if (image != nil) { dispatch_async(dispatch_get_main_queue(), ^{ self.onCapture(self, [self getImageFrom:image]); }); } } } -(UIImage *)getImageFrom:(UIImage *)image{ CIContext *context = [CIContext contextWithOptions:nil]; CGImageRef imageRef = [context createCGImage:image.CIImage fromRect:[image.CIImage extent]]; UIImage *imageTmp = [UIImage imageWithCGImage:imageRef]; CGImageRelease(imageRef); return imageTmp; } - (BOOL)statusCheck{ if (![self.class isCameraAvailable]){ return NO; } if (![self.class isRearCameraAvailable] && ![self.class isFrontCameraAvailable]) { return NO; } if (![self.class isCameraAuthStatusCorrect]) { return NO; } return YES; } #pragma - Helpers - (void)passError:(NSError *)error { if(self.onError) { __weak typeof(self) weakSelf = self; self.onError(weakSelf, error); } } + (BOOL)isFrontCameraAvailable { return [UIImagePickerController isCameraDeviceAvailable:UIImagePickerControllerCameraDeviceFront]; } + (BOOL)isRearCameraAvailable { return [UIImagePickerController isCameraDeviceAvailable:UIImagePickerControllerCameraDeviceRear]; } + (BOOL)isCameraAvailable{ return [UIImagePickerController isSourceTypeAvailable:UIImagePickerControllerSourceTypeCamera]; } + (BOOL)isCameraAuthStatusCorrect{ AVAuthorizationStatus authStatus = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo]; if (authStatus == AVAuthorizationStatusAuthorized || authStatus == AVAuthorizationStatusNotDetermined) { return YES; } return NO; } @end