CaptureViewController.m

#import <Foundation/Foundation.h>
#import <AVFoundation/AVFoundation.h>
#import "CaptureViewController.h"
#import "LWCheckBoxView.h"
#import "LWVideoCaptureSession.h"

@interface CaptureViewController () <LWVideoCaptureSessionDelegate, LWCheckBoxViewDelegate>
@property (strong, nonatomic) UIView* displayView;
@property (strong, nonatomic) LWCheckBoxView* checkBoxView;
@property (strong, nonatomic) UILabel* captureTypeLabel;
@property (strong, nonatomic) UIButton* startBtn;
@property (strong, nonatomic) UIButton* stopBtn;

@property (strong, nonatomic) LWVideoCaptureConfig* captureSessionConfig;
@property (strong, nonatomic) LWVideoCaptureSession* captureSession;
@end

@implementation CaptureViewController

- (void) viewDidLoad {
    int displayViewX = 0;
    int displayViewY = 20;
    int displayViewW = [UIScreen mainScreen].bounds.size.width;
    int displayViewH = displayViewW / 16 * 9;
    self.displayView = [[UIView alloc] initWithFrame:CGRectMake(displayViewX, displayViewY, displayViewW, displayViewH)];
    self.displayView.backgroundColor = [UIColor blackColor];
    [self.view addSubview:self.displayView];
    
    int checkBoxViewX = 100;
    int checkBoxViewY = displayViewH + displayViewY + 10;
    int checkBoxViewW = 270;
    int checkBoxViewH = 30;
    LWCheckBoxItem* item1 = [[LWCheckBoxItem alloc] initWithName:@"前置"];
    LWCheckBoxItem* item2 = [[LWCheckBoxItem alloc] initWithName:@"后置"];
    NSArray<LWCheckBoxItem*>* array = [NSArray arrayWithObjects:item1, item2, nil];
    self.checkBoxView = [[LWCheckBoxView alloc] initWithItems:array];
    self.checkBoxView.frame = CGRectMake(checkBoxViewX, checkBoxViewY, checkBoxViewW, checkBoxViewH);
    self.checkBoxView.displayType = kLWCheckBoxViewDisplayType_LR;
    self.checkBoxView.delegate = self;
    [self.view addSubview:self.checkBoxView];
    
    self.captureTypeLabel = [[UILabel alloc] initWithFrame:CGRectMake(0, checkBoxViewY, 100, 30)];
    self.captureTypeLabel.text = @"摄像头选择";
    self.captureTypeLabel.backgroundColor = [UIColor whiteColor];
    self.captureTypeLabel.textColor = [UIColor blackColor];
    [self.view addSubview:self.captureTypeLabel];
    
    self.startBtn = [[UIButton alloc] initWithFrame:CGRectMake(0, checkBoxViewY + checkBoxViewH + 10, [UIScreen mainScreen].bounds.size.width / 2, 30)];
//    self.startBtn.enabled = YES;
    [self.startBtn setTitle:@"start" forState:UIControlStateNormal];
    [self.startBtn setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
    [self.startBtn setBackgroundColor:[UIColor whiteColor]];
    [self.startBtn addTarget:self action:@selector(startSender:) forControlEvents:UIControlEventTouchUpInside];
    [self.view addSubview:self.startBtn];
    
    self.stopBtn = [[UIButton alloc] initWithFrame:CGRectMake([UIScreen mainScreen].bounds.size.width / 2, checkBoxViewY + checkBoxViewH + 10, [UIScreen mainScreen].bounds.size.width / 2, 30)];
//    self.startBtn.enabled = YES;
    [self.stopBtn setTitle:@"stop" forState:UIControlStateNormal];
    [self.stopBtn setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
    [self.stopBtn setBackgroundColor:[UIColor whiteColor]];
    [self.stopBtn addTarget:self action:@selector(stopSender:) forControlEvents:UIControlEventTouchUpInside];
    [self.view addSubview:self.stopBtn];
}

- (void)viewDidDisappear:(BOOL)animated {
    
}

- (IBAction)startSender:(id)sender {
    self.captureSessionConfig = [[LWVideoCaptureConfig alloc] init];
    self.captureSession = [[LWVideoCaptureSession alloc] initWithConfig:self.captureSessionConfig];
    
    [self.captureSession startCapture];
    
    AVCaptureVideoPreviewLayer* layer = [self.captureSession layer];
    layer.frame = CGRectMake(0, 0, self.displayView.frame.size.width, self.displayView.frame.size.height);
    layer.videoGravity = AVLayerVideoGravityResizeAspectFill;
    [self.displayView.layer addSublayer:layer];
}

- (IBAction)stopSender:(id)sender {
    AVCaptureVideoPreviewLayer* layer = [self.captureSession layer];
    layer.frame = CGRectMake(0, 0, self.displayView.frame.size.width, self.displayView.frame.size.height);
    layer.videoGravity = AVLayerVideoGravityResizeAspectFill;
    [layer removeFromSuperlayer];
    
    [self.captureSession stopCapture];
    self.captureSession = nil;
}

#pragma mark LWVideoCaptureSessionDelegate
- (void)onVideoWithSampleBuffer:(CMSampleBufferRef)sampleBuffer {
    
}

#pragma mark LWCheckBoxViewDelegate
- (void)onSelect:(int)index item:(LWCheckBoxItem*)item {
    if ([item.checkBoxName isEqual:@"前置"]) {
        self.captureSessionConfig.position = AVCaptureDevicePositionFront;
    } else {
        self.captureSessionConfig.position = AVCaptureDevicePositionBack;
    }
}

@end