基于AVCaptureSession的视频录制工具

基于AVCaptureSession的视频录制工具封装
视频输出使用的AVAssetWriter写入沙盒

使用：

//初始化
self.cTool = [CameraTool cameraWithPreView:self.baseView.preView AndFinishShootingBlock:^(NSURL * url) {
  //url 视频输出地址
}];
//开始预览并录制（注意下block循环引用问题）
 [self.cTool startCapture:^{
                //开始录制
                [self.cTool startRecord];
                //......
                //结束录制        
                [self.cTool stopRecord];
}];

CameraTool.h

//
//  CameraTool.h
//  mvvm
//
//  Created by 朱鑫华 on 2020/8/16.
//  Copyright © 2020 朱鑫华. All rights reserved.
//

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

typedef void (^CameraFinishedBlock)(NSURL *);

@interface CameraTool : NSObject
/**
  preView - 父容器
  block - 结束操作
 */
+(instancetype)cameraWithPreView:(UIView *)preView AndFinishShootingBlock:(CameraFinishedBlock) block;

-(void)startCapture:(void(^)(void))block;
-(void)stopCapture;
-(void)startRecord;
-(void)stopRecord;

@end

NS_ASSUME_NONNULL_END

CameraTool.m

//
//  CameraTool.m
//  mvvm
//
//  Created by 朱鑫华 on 2020/8/16.
//  Copyright © 2020 朱鑫华. All rights reserved.
//

#import "CameraTool.h"
#import <AVFoundation/AVFoundation.h>

@interface CameraTool() <AVCaptureVideoDataOutputSampleBufferDelegate,AVCaptureAudioDataOutputSampleBufferDelegate>

@property (nonatomic , strong) AVCaptureSession *captureSession;
@property (nonatomic , strong) AVCaptureDeviceInput *videoInput;
@property (nonatomic , strong) AVCaptureDeviceInput *audioInput;
@property (nonatomic , strong) AVCaptureVideoDataOutput *videoOutput;
@property (nonatomic , strong) AVCaptureAudioDataOutput *audioOutput;
@property (nonatomic , strong) dispatch_queue_t sessionQeue;
@property (nonatomic , strong)  AVCaptureVideoPreviewLayer *preViewLayer;
@property (nonatomic , assign) BOOL isRecording;
@property (nonatomic , assign) BOOL canWrite;
@property (nonatomic , strong) AVAssetWriter *writer;
@property (nonatomic , strong) AVAssetWriterInput *writerAudioInput;
@property (nonatomic , strong) AVAssetWriterInput *writerVideoInput;
@property (nonatomic , strong) AVPlayer *player;
@property (nonatomic , strong) AVPlayerLayer *playerLayer;
@property (nonatomic , strong) UIButton *palyBtn;

@property (nonatomic , strong) CALayer *maskLayer;

@property (nonatomic , copy) CameraFinishedBlock shootingFinishBlcok;

@end


@implementation CameraTool

+(instancetype)cameraWithPreView:(UIView *)preView AndFinishShootingBlock:(nonnull CameraFinishedBlock)block{
    CameraTool *tool = [[self alloc] initWithPreView:preView];
    tool.shootingFinishBlcok = block;
    return tool;
}

-(instancetype)initWithPreView:(UIView *)preview{
    if (self = [super init]) {
        [self initializeDataWithPreView:preview];
    }
    return self;
}

-(void)initializeDataWithPreView:(UIView *)preView{
    _sessionQeue = dispatch_queue_create("cameraQueue", NULL);
    
    _captureSession = [[AVCaptureSession alloc] init];
    if ([_captureSession canSetSessionPreset:AVCaptureSessionPresetHigh]) {
        [_captureSession setSessionPreset:AVCaptureSessionPresetHigh];
    }else if([_captureSession canSetSessionPreset:AVCaptureSessionPresetMedium]){
        [_captureSession setSessionPreset:AVCaptureSessionPresetMedium];
    }else if ([_captureSession canSetSessionPreset:AVCaptureSessionPresetLow]){
        [_captureSession setSessionPreset:AVCaptureSessionPresetLow];
    }
    
    _preViewLayer = [AVCaptureVideoPreviewLayer layerWithSession:_captureSession];
    _preViewLayer.masksToBounds = YES;
    _preViewLayer.videoGravity = AVLayerVideoGravityResizeAspectFill;
    _preViewLayer.frame = preView.bounds;
    [preView.layer addSublayer:_preViewLayer];
    
    [self createMaskLayer:preView.bounds];
    [self setCaptureInput];
    [self setCaptureOutput];
}

-(void)createMaskLayer:(CGRect)rect{
    CAShapeLayer *shapeLayer = [CAShapeLayer layer];
    _maskLayer = shapeLayer;
    UIBezierPath *apath = [UIBezierPath bezierPathWithRoundedRect:rect cornerRadius:0];
    [apath appendPath:[UIBezierPath bezierPathWithArcCenter:CGPointMake(rect.size.width * 0.5 , rect.size.height * 0.5) radius:115 startAngle:0 endAngle:2*M_PI clockwise:NO]];
    shapeLayer.path = apath.CGPath;
    shapeLayer.strokeColor = [UIColor clearColor].CGColor;
    shapeLayer.fillColor = [UIColor colorWithWhite:0 alpha:0.3].CGColor;
    
    CAShapeLayer *shape2 = [CAShapeLayer layer];
    UIBezierPath *spath = [UIBezierPath bezierPathWithArcCenter:CGPointMake(rect.size.width * 0.5 , rect.size.height * 0.5) radius:115 startAngle:0 endAngle:2*M_PI clockwise:NO];
    shape2.path = spath.CGPath;
    shape2.strokeColor = ColorHexString(@"#0C2340").CGColor;
    shape2.lineWidth = 4;
    shape2.fillColor = [UIColor clearColor].CGColor;
    [shapeLayer addSublayer:shape2];
}

///输入设置
-(void)setCaptureInput{
    //视频输入
     AVCaptureDevice *device = [self cameraWithPosition:AVCaptureDevicePositionFront];
    self.videoInput = [AVCaptureDeviceInput deviceInputWithDevice:device error:nil];
    if ([self.captureSession canAddInput:self.videoInput]) {
        [self.captureSession addInput:self.videoInput];
    }
    
    //音频输入
    AVCaptureDevice *audioDevice = [[AVCaptureDevice devicesWithMediaType:AVMediaTypeAudio] firstObject];
    self.audioInput = [AVCaptureDeviceInput deviceInputWithDevice:audioDevice error:nil];
    if ([self.captureSession canAddInput:self.audioInput]) {
        [self.captureSession addInput:self.audioInput];
    }
}

///输入设置
-(void)setCaptureOutput{
    self.videoOutput = [[AVCaptureVideoDataOutput alloc] init];
    [self.videoOutput setVideoSettings:@{
        (id)kCVPixelBufferPixelFormatTypeKey : @(kCVPixelFormatType_32BGRA)
    }];
    [self.videoOutput setSampleBufferDelegate:self queue:self.sessionQeue];
    if ([self.captureSession canAddOutput:self.videoOutput]) {
        [self.captureSession addOutput:self.videoOutput];
    }
    
    self.audioOutput = [[AVCaptureAudioDataOutput alloc] init];
    [self.audioOutput setSampleBufferDelegate:self queue:self.sessionQeue];
    if ([self.captureSession canAddOutput:self.audioOutput]) {
        [self.captureSession addOutput:self.audioOutput];
    }
    
//    AVCaptureConnection *connection = [self.videoOutput connectionWithMediaType:AVMediaTypeVideo];
//    // 设置前置摄像头拍照不镜像
//       AVCaptureDevicePosition currentPosition=[[self.videoInput device] position];
//    if (currentPosition == AVCaptureDevicePositionUnspecified || currentPosition == AVCaptureDevicePositionFront) {
//         connection.videoMirrored = YES;
//    } else {
//         connection.videoMirrored = NO;
//    }

}

#pragma mark - 预览相关
-(void)startCapture:(void (^)(void))block{
    if (![self.captureSession isRunning]) {
        __weak typeof(self) weakSelf = self;
        [self->_maskLayer removeFromSuperlayer];
        [self->_preViewLayer addSublayer:_maskLayer];
        dispatch_async(self.sessionQeue, ^{
            [weakSelf.captureSession startRunning];
            dispatch_async(dispatch_get_main_queue(), ^{
                block();
            });
        });
    }
}
-(void)stopCapture{
    if ([self.captureSession isRunning]) {
        __weak typeof(self) weakSelf = self;
        dispatch_async(self.sessionQeue, ^{
            [weakSelf.captureSession stopRunning];
        });
    }
}

#pragma mark - 拍摄相关
-(void)startRecord{
    __weak typeof(self) weakSelf = self;
    dispatch_async(self.sessionQeue, ^{
        
        NSURL *fileUrl = [self createVideoPath];
        NSFileManager *fm = [NSFileManager defaultManager] ;
        BOOL res = [fm removeItemAtURL:fileUrl error:nil];
        
        NSError *error;
        //创建AVAssetWriter
        weakSelf.writer = [AVAssetWriter assetWriterWithURL:fileUrl fileType:AVFileTypeMPEG4 error:&error];
        NSDictionary *dict =  @{
                   AVEncoderBitRatePerChannelKey : @(28000),
                   AVFormatIDKey : @(kAudioFormatMPEG4AAC),
                   AVNumberOfChannelsKey : @(1),
                   AVSampleRateKey : @(22050)
               };
        NSDictionary *audioOutputSetting = dict;
        self.writerAudioInput = [[AVAssetWriterInput alloc] initWithMediaType:AVMediaTypeAudio outputSettings:audioOutputSetting];
        //输入是否调整处理成实时数据
        self.writerAudioInput.expectsMediaDataInRealTime = YES;
        
        NSDictionary *videoOutputSetting = @{
            AVVideoCodecKey : AVVideoCodecH264,
            AVVideoWidthKey : @(1280),
            AVVideoHeightKey : @(720),
            AVVideoCompressionPropertiesKey:@{
                    AVVideoAverageBitRateKey : @(1280*720*3),
            AVVideoExpectedSourceFrameRateKey : @(15),
            AVVideoMaxKeyFrameIntervalKey : @(15),
            AVVideoProfileLevelKey : AVVideoProfileLevelH264BaselineAutoLevel
            }
        };
        self.writerVideoInput = [[AVAssetWriterInput alloc] initWithMediaType:AVMediaTypeVideo outputSettings:videoOutputSetting];
        //输入是否调整处理成实时数据
        self.writerVideoInput.expectsMediaDataInRealTime = YES;
        
        //画面需要旋转90度
        weakSelf.writerVideoInput.transform = CGAffineTransformMakeRotation(M_PI / 2.0);
        if ([weakSelf.writer canAddInput:weakSelf.writerVideoInput]) {
            [weakSelf.writer addInput:weakSelf.writerVideoInput];
        }
        if ([weakSelf.writer canAddInput:weakSelf.writerAudioInput]) {
            [weakSelf.writer addInput:weakSelf.writerAudioInput];
        }
        weakSelf.isRecording = YES;
        weakSelf.canWrite = NO;
        
    });
}
-(void)stopRecord{
    __weak typeof(self) weakSelf = self;
    dispatch_async(self.sessionQeue, ^{
        weakSelf.isRecording = NO;
        if (weakSelf.writer.status == AVAssetWriterStatusWriting) {
            [weakSelf.writerVideoInput markAsFinished];
            [weakSelf.writerAudioInput markAsFinished];
            [weakSelf.writer finishWritingWithCompletionHandler:^{
                dispatch_async(dispatch_get_main_queue(), ^{
                     //通知代理写入完成，地址：weakSelf.writer.outputURL
                    NSLog(@"文件地址：%@",weakSelf.writer.outputURL.absoluteString);
                    [self stopCapture];
                    if (self.shootingFinishBlcok) {
                        self.shootingFinishBlcok(weakSelf.writer.outputURL);
                    }
                });
            }];
        }
    });
}

#pragma mark - 视频处理
-(void)captureOutput:(AVCaptureOutput *)output didOutputSampleBuffer:(nonnull CMSampleBufferRef)sampleBuffer fromConnection:(nonnull AVCaptureConnection *)connection{
    [self appendSampleBuffer:sampleBuffer];
}

-(void)appendSampleBuffer:(CMSampleBufferRef)sampleBuffer{
    if (self.isRecording == NO) {
        return;
    }
    //获取 mediaType
    CMTextFormatDescriptionRef formatDes = CMSampleBufferGetFormatDescription(sampleBuffer);
    CMMediaType mediaType = CMFormatDescriptionGetMediaType(formatDes);
    if (mediaType == kCMMediaType_Video) {
        if (!self.canWrite) {
            CMTime timestamp = CMSampleBufferGetPresentationTimeStamp(sampleBuffer);
            if ([self.writer startWriting]) {
//                NSLog(@"xie数据");
                [self.writer startSessionAtSourceTime:timestamp];
            }
            self.canWrite = YES;
        }
        
        if (self.canWrite && self.writerVideoInput.readyForMoreMediaData) {
//            NSLog(@"拼接视频数据");
            BOOL success = [self.writerVideoInput appendSampleBuffer:sampleBuffer];
            if (!success) {
//                NSLog(@"写入失败");
            }
        }
        
    }else if (mediaType == kCMMediaType_Audio){
        if (self.writerAudioInput.readyForMoreMediaData) {
//            NSLog(@"拼接饮品数据");
            BOOL success = [self.writerAudioInput appendSampleBuffer:sampleBuffer];
            if (!success) {
//                NSLog(@"==yinpin写入失败");
            }
        }
    }
}


-(NSURL *)createVideoPath{
    NSString *random = @"faceAuth";//NSString stringWithFormat:@"video%d_%d",arc4random()%999999,(int)[[NSDate date] timeIntervalSince1970] % 10000];
    NSString *path = [NSTemporaryDirectory() stringByAppendingPathComponent:[NSString stringWithFormat: @"%@.mp4",random]];
    return [NSURL fileURLWithPath:path];
}


- (AVCaptureDevice *)cameraWithPosition:(AVCaptureDevicePosition)position
{
    NSArray *devices = [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo];
    for ( AVCaptureDevice *device in devices )
        if ( device.position == position )
            return device;
    return nil;
}
-(void)dealloc{
    NSLog(@"%s dealloc",__func__);
}



@end
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 214,658评论 6赞 496
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 91,482评论 3赞 389
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 160,213评论 0赞 350
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 57,395评论 1赞 288
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 66,487评论 6赞 386
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 50,523评论 1赞 293
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 39,525评论 3赞 414
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 38,300评论 0赞 270
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 44,753评论 1赞 307
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,048评论 2赞 330
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,223评论 1赞 343
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 34,905评论 5赞 338
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 40,541评论 3赞 322
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,168评论 0赞 21
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,417评论 1赞 268
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 47,094评论 2赞 365
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 44,088评论 2赞 352