//
// QRCodeView.h
// QRCode
//
// Created by YouXianMing on 16/7/7.
// Copyright © 2016年 XianMing You. All rights reserved.
//

#import <UIKit/UIKit.h>
#import <AVFoundation/AVFoundation.h>
@class QRCodeView;

@protocol QRCodeViewDelegate <NSObject>

@optional

/**
 * 获取QR的扫描结果
 *
 * @param codeView QRCodeView实体对象
 * @param codeString 扫描字符串
 */
- (void)QRCodeView:(QRCodeView *)codeView codeString:(NSString *)codeString;

@end

@interface QRCodeView : UIView

/**
 * 代理
 */
@property (nonatomic, weak) id <QRCodeViewDelegate> delegate;

/**
 * 灯的状态,默认为关闭
 */
@property (nonatomic) AVCaptureTorchMode torchMode;

/**
 * 敏感区域,如果不设置,则为全部扫描区域
 */
@property (nonatomic) CGRect interestArea;

/**
 * 你用来添加自定义控件的view,尺寸与当前初始化的view一致
 */
@property (nonatomic, strong) UIView *contentView;

/**
 * 正在运行当中
 */
@property (nonatomic, readonly) BOOL isRunning;

/**
 * 开始扫描
 *
 * @return 如果成功,则返回YES,否则返回NO
 */
- (BOOL)start;

/**
 * 结束扫描
 */
- (void)stop;

@end

//
// QRCodeView.m
// QRCode
//
// Created by YouXianMing on 16/7/7.
// Copyright © 2016年 XianMing You. All rights reserved.
//

#import "QRCodeView.h"

@interface QRCodeView () <AVCaptureMetadataOutputObjectsDelegate>

@property (nonatomic) BOOL         isRunning;
@property (nonatomic, strong) UIView      *videoView;

@property (nonatomic, strong) AVCaptureDeviceInput  *deviceInput;
@property (nonatomic, strong) AVCaptureDevice    *captureDevice;
@property (nonatomic, strong) AVCaptureSession   *captureSession;
@property (nonatomic, strong) AVCaptureVideoPreviewLayer *videoPreviewLayer;
@property (nonatomic, strong) AVCaptureMetadataOutput  *captureMetadataOutput;

@end

@implementation QRCodeView

- (instancetype)initWithFrame:(CGRect)frame {
 
 if (self = [super initWithFrame:frame]) {
  
  self.videoView = [[UIView alloc] initWithFrame:self.bounds];
  [self addSubview:self.videoView];
  
  self.contentView = [[UIView alloc] initWithFrame:self.bounds];
  [self addSubview:self.contentView];
  
  self.captureDevice = [AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeVideo];
  
  _torchMode = AVCaptureTorchModeOff;
  
  [self addNotificationCenter];
 }
 
 return self;
}

#pragma mark - NSNotificationCenter related.

- (void)addNotificationCenter {
 
 [[NSNotificationCenter defaultCenter] addObserver:self
            selector:@selector(notificationCenterEvent:)
             name:AVCaptureInputPortFormatDescriptionDidChangeNotification
            object:nil];
}

- (void)removeNotificationCenter {
 
 [[NSNotificationCenter defaultCenter] removeObserver:self
             name:AVCaptureInputPortFormatDescriptionDidChangeNotification
             object:nil];
}

- (void)notificationCenterEvent:(NSNotification *)sender {
 
 if (self.interestArea.size.width && self.interestArea.size.height) {
  
  self.captureMetadataOutput.rectOfInterest = [self.videoPreviewLayer metadataOutputRectOfInterestForRect:self.interestArea];
  
 } else {
  
  self.captureMetadataOutput.rectOfInterest = CGRectMake(0, 0, 1, 1);
 }
}

#pragma mark - Start & Stop.

- (BOOL)start {
 
 // 初始化输入流
 BOOL  result = NO;
 NSError *error = nil;
 self.deviceInput = [AVCaptureDeviceInput deviceInputWithDevice:self.captureDevice error:&error];
 if (self.deviceInput == nil) {
  
  NSLog(@"%@", error);
  return result;
 }
 
 // 创建会话
 self.captureSession = [[AVCaptureSession alloc] init];
 
 // 添加输入流
 [self.captureSession addInput:self.deviceInput];
 
 // 初始化输出流
 self.captureMetadataOutput = [[AVCaptureMetadataOutput alloc] init];
 
 // 添加输出流
 [self.captureSession addOutput:self.captureMetadataOutput];
 
 // 创建queue.
 [self.captureMetadataOutput setMetadataObjectsDelegate:self queue:dispatch_queue_create(nil, nil)];
 self.captureMetadataOutput.metadataObjectTypes = @[AVMetadataObjectTypeQRCode];
 
 // 创建输出对象
 self.videoPreviewLayer = [[AVCaptureVideoPreviewLayer alloc] initWithSession:self.captureSession];
 self.videoPreviewLayer.videoGravity = AVLayerVideoGravityResizeAspectFill;
 self.videoPreviewLayer.frame = self.contentView.bounds;
 [self.videoView.layer addSublayer:self.videoPreviewLayer];
 
 // 开始
 [self.captureSession startRunning];
 self.isRunning = YES;
 result   = YES;
 
 return result;
}

- (void)stop {
 
 [self.captureSession stopRunning];
 self.isRunning  = NO;
 self.captureSession = nil;
}

#pragma mark - AVCaptureMetadataOutputObjectsDelegate

- (void)captureOutput:(AVCaptureOutput *)captureOutput didOutputMetadataObjects:(NSArray *)metadataObjects
  fromConnection:(AVCaptureConnection *)connection {
 
 if (metadataObjects.count > 0) {
  
  AVMetadataMachineReadableCodeObject *metadata = metadataObjects.firstObject;
  NSString       *result = nil;
  
  if ([metadata.type isEqualToString:AVMetadataObjectTypeQRCode]) {
   
   result = metadata.stringValue;
   
   if (_delegate && [_delegate respondsToSelector:@selector(QRCodeView:codeString:)]) {
    
    [_delegate QRCodeView:self codeString:result];
   }
  }
 }
}

#pragma mark - Setter & Getter.

- (void)setTorchMode:(AVCaptureTorchMode)torchMode {

 _torchMode = torchMode;
 
 if (_deviceInput && [self.captureDevice hasTorch]) {
  
  [self.captureDevice lockForConfiguration:nil];
  [self.captureDevice setTorchMode:torchMode];
  [self.captureDevice unlockForConfiguration];
 }
}

#pragma mark - System method.

- (void)dealloc {
 
 [self stop];
 [self removeNotificationCenter];
}

@end