SLAvCaptureTool.m 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649
  1. //
  2. // SLAvCaptureTool.m
  3. // DarkMode
  4. //
  5. // Created by wsl on 2019/9/20.
  6. // Copyright © 2019 wsl. All rights reserved.
  7. //
  8. #import "SLAvCaptureTool.h"
  9. #import <CoreMotion/CoreMotion.h>
  10. @interface SLAvCaptureTool () <AVCapturePhotoCaptureDelegate, AVCaptureVideoDataOutputSampleBufferDelegate, AVCaptureAudioDataOutputSampleBufferDelegate>
  11. @property (nonatomic, strong) AVCaptureSession *session; //采集会话
  12. @property (nonatomic, strong) AVCaptureVideoPreviewLayer *previewLayer;//摄像头采集内容展示区域
  13. @property (nonatomic, strong) AVCaptureDeviceInput *audioInput; //音频输入流
  14. @property (nonatomic, strong) AVCaptureDeviceInput *videoInput; //视频输入流
  15. @property (nonatomic, strong) AVCapturePhotoOutput *capturePhotoOutput; //照片输出流
  16. @property (nonatomic, strong) AVCaptureStillImageOutput *captureStillImageOutput; // iOS10之前的照片输出流
  17. @property (nonatomic, strong) AVCaptureVideoDataOutput *videoDataOutput; //视频数据帧输出流
  18. @property (nonatomic, strong) AVCaptureAudioDataOutput *audioDataOutput; //音频数据帧输出流
  19. @property (nonatomic, strong) AVAssetWriter *assetWriter; //音视频数据流文件写入
  20. @property (nonatomic, strong) AVAssetWriterInput *assetWriterVideoInput; //写入视频文件
  21. @property (nonatomic, strong) AVAssetWriterInput *assetWriterAudioInput; //写入音频文件
  22. @property (nonatomic, strong) NSDictionary *videoCompressionSettings; //视频写入配置
  23. @property (nonatomic, strong) NSDictionary *audioCompressionSettings; //音频写入配置
  24. @property (nonatomic, assign) BOOL canWrite; //是否能写入
  25. @property (nonatomic, copy) NSURL *outputFileURL; //音视频文件输出路径
  26. @property (nonatomic, assign) BOOL isRecording; //是否正在录制
  27. @property (nonatomic, assign) SLAvCaptureType avCaptureType; //音视频捕获类型 默认 SLAvCaptureTypeAv
  28. @property (nonatomic, assign) UIDeviceOrientation shootingOrientation; //拍摄录制时的手机方向
  29. @property (nonatomic, strong) CMMotionManager *motionManager; //运动传感器 监测设备方向
  30. @end
  31. @implementation SLAvCaptureTool
  32. + (instancetype)sharedAvCaptureTool {
  33. static SLAvCaptureTool *avCaptureTool = nil;
  34. static dispatch_once_t onceToken;
  35. dispatch_once(&onceToken, ^{
  36. avCaptureTool = [[SLAvCaptureTool alloc] init];
  37. });
  38. return avCaptureTool;
  39. }
  40. #pragma mark - Override
  41. - (instancetype)init {
  42. self = [super init];
  43. if (self) {
  44. self.videoSize = CGSizeMake([UIScreen mainScreen].bounds.size.width, [UIScreen mainScreen].bounds.size.height);
  45. }
  46. return self;
  47. }
  48. - (void)dealloc {
  49. [self stopRunning];
  50. }
  51. #pragma mark - HelpMethods
  52. //获取指定位置的摄像头
  53. - (AVCaptureDevice *)getCameraDeviceWithPosition:(AVCaptureDevicePosition)positon{
  54. if (@available(iOS 10.2, *)) {
  55. AVCaptureDeviceDiscoverySession *dissession = [AVCaptureDeviceDiscoverySession discoverySessionWithDeviceTypes:@[AVCaptureDeviceTypeBuiltInDualCamera,AVCaptureDeviceTypeBuiltInTelephotoCamera,AVCaptureDeviceTypeBuiltInWideAngleCamera] mediaType:AVMediaTypeVideo position:positon];
  56. for (AVCaptureDevice *device in dissession.devices) {
  57. if ([device position] == positon) {
  58. return device;
  59. }
  60. }
  61. } else {
  62. NSArray *devices = [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo];
  63. for (AVCaptureDevice *device in devices) {
  64. if ([device position] == positon) {
  65. return device;
  66. }
  67. }
  68. }
  69. return nil;
  70. }
  71. //最小缩放值 焦距
  72. - (CGFloat)minZoomFactor {
  73. CGFloat minZoomFactor = 1.0;
  74. if (@available(iOS 11.0, *)) {
  75. minZoomFactor = [self.videoInput device].minAvailableVideoZoomFactor;
  76. }
  77. return minZoomFactor;
  78. }
  79. //最大缩放值 焦距
  80. - (CGFloat)maxZoomFactor {
  81. CGFloat maxZoomFactor = [self.videoInput device].activeFormat.videoMaxZoomFactor;
  82. if (@available(iOS 11.0, *)) {
  83. maxZoomFactor = [self.videoInput device].maxAvailableVideoZoomFactor;
  84. }
  85. if (maxZoomFactor > 6) {
  86. maxZoomFactor = 6.0;
  87. }
  88. return maxZoomFactor;
  89. }
  90. #pragma mark - Getter
  91. - (AVCaptureSession *)session{
  92. if (_session == nil){
  93. _session = [[AVCaptureSession alloc] init];
  94. //高质量采集率
  95. [_session setSessionPreset:AVCaptureSessionPreset1280x720];
  96. if([_session canAddInput:self.videoInput]) [_session addInput:self.videoInput]; //添加视频输入流
  97. if([_session canAddInput:self.audioInput]) [_session addInput:self.audioInput]; //添加音频输入流
  98. if (@available(iOS 10.0, *)) {
  99. if([_session canAddOutput:self.capturePhotoOutput]) [_session addOutput:self.capturePhotoOutput]; //添加照片输出流
  100. } else {
  101. if([_session canAddOutput:self.captureStillImageOutput]) [_session addOutput:self.captureStillImageOutput]; //添加照片输出流
  102. }
  103. if([_session canAddOutput:self.videoDataOutput]) [_session addOutput:self.videoDataOutput]; //视频数据输出流 纯画面
  104. if([_session canAddOutput:self.audioDataOutput]) [_session addOutput:self.audioDataOutput]; //音频数据输出流
  105. AVCaptureConnection * captureVideoConnection = [self.videoDataOutput connectionWithMediaType:AVMediaTypeVideo];
  106. // 设置是否为镜像,前置摄像头采集到的数据本来就是翻转的,这里设置为镜像把画面转回来
  107. if (self.devicePosition == AVCaptureDevicePositionFront && captureVideoConnection.supportsVideoMirroring) {
  108. captureVideoConnection.videoMirrored = YES;
  109. }
  110. captureVideoConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
  111. }
  112. return _session;
  113. }
  114. - (AVCaptureDeviceInput *)videoInput {
  115. if (_videoInput == nil) {
  116. //添加一个视频输入设备 默认是后置摄像头
  117. AVCaptureDevice *videoCaptureDevice = [self getCameraDeviceWithPosition:AVCaptureDevicePositionBack];
  118. //创建视频输入流
  119. _videoInput = [AVCaptureDeviceInput deviceInputWithDevice:videoCaptureDevice error:nil];
  120. if (!_videoInput){
  121. NSLog(@"获得摄像头失败");
  122. return nil;
  123. }
  124. }
  125. return _videoInput;
  126. }
  127. - (AVCaptureDeviceInput *)audioInput {
  128. if (_audioInput == nil) {
  129. NSError * error = nil;
  130. //添加一个音频输入/捕获设备
  131. AVCaptureDevice * audioCaptureDevice = [AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeAudio];
  132. _audioInput = [[AVCaptureDeviceInput alloc] initWithDevice:audioCaptureDevice error:&error];
  133. if (error) {
  134. NSLog(@"获得音频输入设备失败:%@",error.localizedDescription);
  135. }
  136. }
  137. return _audioInput;
  138. }
  139. - (AVCapturePhotoOutput *)capturePhotoOutput {
  140. if (_capturePhotoOutput == nil) {
  141. _capturePhotoOutput = [[AVCapturePhotoOutput alloc] init];
  142. // _capturePhotoOutput.livePhotoCaptureEnabled = NO; //是否拍摄 live Photo
  143. }
  144. return _capturePhotoOutput;
  145. }
  146. - (AVCaptureStillImageOutput *)captureStillImageOutput {
  147. if (_captureStillImageOutput == nil) {
  148. _captureStillImageOutput = [[AVCaptureStillImageOutput alloc] init];
  149. }
  150. return _captureStillImageOutput;
  151. }
  152. - (AVCaptureVideoDataOutput *)videoDataOutput {
  153. if (_videoDataOutput == nil) {
  154. _videoDataOutput = [[AVCaptureVideoDataOutput alloc] init];
  155. [_videoDataOutput setSampleBufferDelegate:self queue:dispatch_get_global_queue(0, 0)];
  156. }
  157. return _videoDataOutput;
  158. }
  159. - (AVCaptureAudioDataOutput *)audioDataOutput {
  160. if (_audioDataOutput == nil) {
  161. _audioDataOutput = [[AVCaptureAudioDataOutput alloc] init];
  162. [_audioDataOutput setSampleBufferDelegate:self queue:dispatch_get_global_queue(0, 0)];
  163. }
  164. return _audioDataOutput;
  165. }
  166. - (AVCaptureVideoPreviewLayer *)previewLayer {
  167. if (_previewLayer == nil) {
  168. _previewLayer = [AVCaptureVideoPreviewLayer layerWithSession:self.session];
  169. _previewLayer.videoGravity = AVLayerVideoGravityResizeAspect;
  170. }
  171. return _previewLayer;
  172. }
  173. - (AVAssetWriterInput *)assetWriterVideoInput {
  174. if (!_assetWriterVideoInput) {
  175. //写入视频大小
  176. NSInteger numPixels = self.videoSize.width * [UIScreen mainScreen].scale * self.videoSize.height * [UIScreen mainScreen].scale;
  177. //每像素比特
  178. CGFloat bitsPerPixel = 12.0;
  179. NSInteger bitsPerSecond = numPixels * bitsPerPixel;
  180. // 码率和帧率设置
  181. NSDictionary *compressionProperties = @{ AVVideoAverageBitRateKey : @(bitsPerSecond),
  182. AVVideoExpectedSourceFrameRateKey : @(15),
  183. AVVideoMaxKeyFrameIntervalKey : @(15),
  184. AVVideoProfileLevelKey : AVVideoProfileLevelH264High40 };
  185. CGFloat width = self.videoSize.width * [UIScreen mainScreen].scale;
  186. CGFloat height = self.videoSize.height * [UIScreen mainScreen].scale;
  187. //视频属性
  188. self.videoCompressionSettings = @{ AVVideoCodecKey : AVVideoCodecH264,
  189. AVVideoWidthKey : @(width ),
  190. AVVideoHeightKey : @(height),
  191. AVVideoScalingModeKey : AVVideoScalingModeResizeAspectFill,
  192. AVVideoCompressionPropertiesKey : compressionProperties };
  193. _assetWriterVideoInput = [AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeVideo outputSettings:self.videoCompressionSettings];
  194. //expectsMediaDataInRealTime 必须设为yes,需要从capture session 实时获取数据
  195. _assetWriterVideoInput.expectsMediaDataInRealTime = YES;
  196. }
  197. return _assetWriterVideoInput;
  198. }
  199. - (AVAssetWriterInput *)assetWriterAudioInput {
  200. if (_assetWriterAudioInput == nil) {
  201. // 音频设置
  202. _audioCompressionSettings = @{ AVEncoderBitRatePerChannelKey : @(28000),
  203. AVFormatIDKey : @(kAudioFormatMPEG4AAC),
  204. AVNumberOfChannelsKey : @(1),
  205. AVSampleRateKey : @(22050) };
  206. _assetWriterAudioInput = [AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeAudio outputSettings:self.audioCompressionSettings];
  207. _assetWriterAudioInput.expectsMediaDataInRealTime = YES;
  208. }
  209. return _assetWriterAudioInput;
  210. }
  211. - (CMMotionManager *)motionManager {
  212. if (!_motionManager) {
  213. _motionManager = [[CMMotionManager alloc] init];
  214. }
  215. return _motionManager;
  216. }
  217. - (BOOL)isRunning {
  218. return self.session.isRunning;
  219. }
  220. - (AVCaptureDevicePosition)devicePosition {
  221. if([[self.videoInput device] position] == AVCaptureDevicePositionUnspecified) {
  222. return AVCaptureDevicePositionBack;
  223. }
  224. return [[self.videoInput device] position];
  225. }
  226. - (CGFloat)videoZoomFactor {
  227. return [self.videoInput device].videoZoomFactor;
  228. }
  229. #pragma mark - Setter
  230. - (void)setOutputFileURL:(NSURL *)outputFileURL {
  231. _outputFileURL = outputFileURL;
  232. if (self.avCaptureType == SLAvCaptureTypeAudio) {
  233. _assetWriter = [AVAssetWriter assetWriterWithURL:outputFileURL fileType:AVFileTypeAC3 error:nil];
  234. } else if (self.avCaptureType == SLAvCaptureTypeVideo || self.avCaptureType == SLAvCaptureTypeAv) {
  235. _assetWriter = [AVAssetWriter assetWriterWithURL:outputFileURL fileType:AVFileTypeMPEG4 error:nil];
  236. }
  237. }
  238. - (void)setPreview:(nullable UIView *)preview {
  239. if (preview == nil) {
  240. [self.previewLayer removeFromSuperlayer];
  241. }else {
  242. self.previewLayer.frame = preview.bounds;
  243. [preview.layer addSublayer:self.previewLayer];
  244. }
  245. _preview = preview;
  246. }
  247. - (void)setVideoZoomFactor:(CGFloat)videoZoomFactor {
  248. NSError *error = nil;
  249. if (videoZoomFactor <= self.maxZoomFactor &&
  250. videoZoomFactor >= self.minZoomFactor){
  251. if ([[self.videoInput device] lockForConfiguration:&error] ) {
  252. [self.videoInput device].videoZoomFactor = videoZoomFactor;
  253. [[self.videoInput device] unlockForConfiguration];
  254. } else {
  255. NSLog( @"调节焦距失败: %@", error );
  256. }
  257. }
  258. }
  259. - (void)setShootingOrientation:(UIDeviceOrientation)shootingOrientation {
  260. if (_shootingOrientation == shootingOrientation) {
  261. return;
  262. }
  263. _shootingOrientation = shootingOrientation;
  264. }
  265. #pragma mark - EventsHandle
  266. ///启动捕获
  267. - (void)startRunning {
  268. if(!self.session.isRunning) {
  269. [self.session startRunning];
  270. }
  271. [self startUpdateDeviceDirection];
  272. }
  273. ///结束捕获
  274. - (void)stopRunning {
  275. if (self.session.isRunning) {
  276. [self.session stopRunning];
  277. }
  278. [self stopUpdateDeviceDirection];
  279. }
  280. //设置聚焦点和模式 默认连续自动聚焦和自动曝光模式
  281. - (void)focusAtPoint:(CGPoint)focalPoint {
  282. //将UI坐标转化为摄像头坐标 (0,0) -> (1,1)
  283. CGPoint cameraPoint = [self.previewLayer captureDevicePointOfInterestForPoint:focalPoint];
  284. AVCaptureDevice *captureDevice = [self.videoInput device];
  285. NSError * error;
  286. //注意改变设备属性前一定要首先调用lockForConfiguration:调用完之后使用unlockForConfiguration方法解锁
  287. if ([captureDevice lockForConfiguration:&error]) {
  288. if ([captureDevice isFocusModeSupported:AVCaptureFocusModeContinuousAutoFocus]) {
  289. if ([captureDevice isFocusPointOfInterestSupported]) {
  290. [captureDevice setFocusPointOfInterest:cameraPoint];
  291. }
  292. [captureDevice setFocusMode:AVCaptureFocusModeContinuousAutoFocus];
  293. }
  294. //曝光模式
  295. if ([captureDevice isExposureModeSupported:AVCaptureExposureModeContinuousAutoExposure]) {
  296. if ([captureDevice isExposurePointOfInterestSupported]) {
  297. [captureDevice setExposurePointOfInterest:cameraPoint];
  298. }
  299. [captureDevice setExposureMode:AVCaptureExposureModeContinuousAutoExposure];
  300. }
  301. [captureDevice unlockForConfiguration];
  302. } else {
  303. NSLog(@"设置聚焦点错误:%@", error.localizedDescription);
  304. }
  305. }
  306. //切换前/后置摄像头
  307. - (void)switchsCamera:(AVCaptureDevicePosition)devicePosition {
  308. //当前设备方向
  309. if (self.devicePosition == devicePosition) {
  310. return;
  311. }
  312. AVCaptureDeviceInput *videoInput = [AVCaptureDeviceInput deviceInputWithDevice:[self getCameraDeviceWithPosition:devicePosition] error:nil];
  313. //先开启配置,配置完成后提交配置改变
  314. [self.session beginConfiguration];
  315. //移除原有输入对象
  316. [self.session removeInput:self.videoInput];
  317. //添加新的输入对象
  318. if ([self.session canAddInput:videoInput]) {
  319. [self.session addInput:videoInput];
  320. self.videoInput = videoInput;
  321. }
  322. //视频输入对象发生了改变 视频输出的链接也要重新初始化
  323. AVCaptureConnection * captureConnection = [self.videoDataOutput connectionWithMediaType:AVMediaTypeVideo];
  324. if ([captureConnection isVideoStabilizationSupported]) {
  325. //视频稳定模式
  326. captureConnection.preferredVideoStabilizationMode = AVCaptureVideoStabilizationModeAuto;
  327. }
  328. if (self.devicePosition == AVCaptureDevicePositionFront && captureConnection.supportsVideoMirroring) {
  329. captureConnection.videoMirrored = YES;
  330. }
  331. captureConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
  332. //提交新的输入对象
  333. [self.session commitConfiguration];
  334. }
  335. //拍照 输出图片
  336. - (void)outputPhoto {
  337. //获得图片输出连接
  338. AVCaptureConnection * captureConnection = [self.capturePhotoOutput connectionWithMediaType:AVMediaTypeVideo];
  339. // 设置是否为镜像,前置摄像头采集到的数据本来就是翻转的,这里设置为镜像把画面转回来
  340. if (self.devicePosition == AVCaptureDevicePositionFront && captureConnection.supportsVideoMirroring) {
  341. captureConnection.videoMirrored = YES;
  342. }
  343. if (self.shootingOrientation == UIDeviceOrientationLandscapeRight) {
  344. captureConnection.videoOrientation = AVCaptureVideoOrientationLandscapeLeft;
  345. } else if (self.shootingOrientation == UIDeviceOrientationLandscapeLeft) {
  346. captureConnection.videoOrientation = AVCaptureVideoOrientationLandscapeRight;
  347. } else if (self.shootingOrientation == UIDeviceOrientationPortraitUpsideDown) {
  348. captureConnection.videoOrientation = AVCaptureVideoOrientationPortraitUpsideDown;
  349. } else {
  350. captureConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
  351. }
  352. //输出样式设置 AVVideoCodecKey:AVVideoCodecJPEG等
  353. AVCapturePhotoSettings *capturePhotoSettings = [AVCapturePhotoSettings photoSettings];
  354. // capturePhotoSettings.highResolutionPhotoEnabled = YES; //高分辨率
  355. capturePhotoSettings.flashMode = _flashMode; //闪光灯 根据环境亮度自动决定是否打开闪光灯
  356. if (@available(iOS 10.0, *)) {
  357. [self.capturePhotoOutput capturePhotoWithSettings:capturePhotoSettings delegate:self];
  358. } else {
  359. //从 AVCaptureStillImageOutput 中取得 AVCaptureConnection
  360. AVCaptureConnection *connection = [self.captureStillImageOutput connectionWithMediaType:AVMediaTypeVideo];
  361. [self.captureStillImageOutput captureStillImageAsynchronouslyFromConnection:connection completionHandler:^(CMSampleBufferRef _Nullable imageDataSampleBuffer, NSError * _Nullable error) {
  362. if (imageDataSampleBuffer != nil) {
  363. NSData * data = [AVCaptureStillImageOutput jpegStillImageNSDataRepresentation:imageDataSampleBuffer];
  364. UIImage *image = [[UIImage alloc]initWithData:data];
  365. //重点:捕捉图片成功后,将图片传递出去
  366. if ([self.delegate respondsToSelector:@selector(captureTool:didOutputPhoto:error:)]) {
  367. [self.delegate captureTool:self didOutputPhoto:image error:error];
  368. }else {
  369. NSLog(@"请实现代理方法:captureTool:didOutputPhoto:error:");
  370. }
  371. }
  372. }];
  373. }
  374. }
  375. //开始录制视频
  376. - (void)startRecordVideoToOutputFileAtPath:(NSString *)path recordType:(SLAvCaptureType)avCaptureType{
  377. self.avCaptureType = avCaptureType;
  378. //移除重复文件
  379. if ([[NSFileManager defaultManager] fileExistsAtPath:path]) {
  380. [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
  381. }
  382. self.outputFileURL = [NSURL fileURLWithPath:path];
  383. [self stopUpdateDeviceDirection];
  384. //获得视频输出连接
  385. AVCaptureConnection * captureConnection = [self.videoDataOutput connectionWithMediaType:AVMediaTypeVideo];
  386. // 设置是否为镜像,前置摄像头采集到的数据本来就是翻转的,这里设置为镜像把画面转回来
  387. if (self.devicePosition == AVCaptureDevicePositionFront && captureConnection.supportsVideoMirroring) {
  388. captureConnection.videoMirrored = YES;
  389. }
  390. //这个API 每次开始录制时设置视频输出方向,会造成摄像头的短暂黑暗,有问题的代码可以看此类文件夹下的SLAvCaptureTool-bug副本;切换摄像头时设置此属性没有较大的影响
  391. // captureConnection.videoOrientation = AVCaptureVideoOrientationPortrait;
  392. //由于上述原因,故采用在写入输出视频时调整方向
  393. if (self.shootingOrientation == UIDeviceOrientationLandscapeRight) {
  394. self.assetWriterVideoInput.transform = CGAffineTransformMakeRotation(M_PI/2);
  395. } else if (self.shootingOrientation == UIDeviceOrientationLandscapeLeft) {
  396. self.assetWriterVideoInput.transform = CGAffineTransformMakeRotation(-M_PI/2);
  397. } else if (self.shootingOrientation == UIDeviceOrientationPortraitUpsideDown) {
  398. self.assetWriterVideoInput.transform = CGAffineTransformMakeRotation(M_PI);
  399. } else {
  400. self.assetWriterVideoInput.transform = CGAffineTransformMakeRotation(0);
  401. }
  402. if ([self.assetWriter canAddInput:self.assetWriterVideoInput]) {
  403. [self.assetWriter addInput:self.assetWriterVideoInput];
  404. } else {
  405. NSLog(@"视频写入失败");
  406. }
  407. if ([self.assetWriter canAddInput:self.assetWriterAudioInput] && self.avCaptureType == SLAvCaptureTypeAv) {
  408. [self.assetWriter addInput:self.assetWriterAudioInput];
  409. } else {
  410. NSLog(@"音频写入失败");
  411. }
  412. _isRecording = YES;
  413. }
  414. /// 结束捕获视频帧
  415. - (void)stopRecordVideo {
  416. if (_isRecording) {
  417. _isRecording = NO;
  418. __weak typeof(self) weakSelf = self;
  419. if(_assetWriter && self.canWrite && self.assetWriter.status != AVAssetWriterStatusUnknown) {
  420. [_assetWriter finishWritingWithCompletionHandler:^{
  421. if ([weakSelf.delegate respondsToSelector:@selector(captureTool:didFinishRecordingToOutputFileAtURL:error:)]) {
  422. SL_DISPATCH_ON_MAIN_THREAD(^{
  423. [weakSelf.delegate captureTool:weakSelf didFinishRecordingToOutputFileAtURL:weakSelf.outputFileURL error:weakSelf.assetWriter.error];
  424. });
  425. }
  426. weakSelf.canWrite = NO;
  427. weakSelf.assetWriter = nil;
  428. weakSelf.assetWriterAudioInput = nil;
  429. weakSelf.assetWriterVideoInput = nil;
  430. }];
  431. }
  432. }
  433. }
  434. //开始录制音频
  435. - (void)startRecordAudioToOutputFileAtPath:(NSString *)path {
  436. self.avCaptureType = SLAvCaptureTypeAudio;
  437. //移除重复文件
  438. if ([[NSFileManager defaultManager] fileExistsAtPath:path]) {
  439. [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
  440. }
  441. self.outputFileURL = [NSURL fileURLWithPath:path];
  442. [self stopUpdateDeviceDirection];
  443. [self.session beginConfiguration];
  444. //移除视频输出对象
  445. [self.session removeOutput:self.videoDataOutput];
  446. [self.session commitConfiguration];
  447. if ([self.assetWriter canAddInput:self.assetWriterAudioInput]) {
  448. [self.assetWriter addInput:self.assetWriterAudioInput];
  449. } else {
  450. NSLog(@"音频写入失败");
  451. }
  452. _isRecording = YES;
  453. }
  454. //结束音频录制
  455. - (void)stopRecordAudio {
  456. if (_isRecording) {
  457. _isRecording = NO;
  458. __weak typeof(self) weakSelf = self;
  459. if(_assetWriter && self.canWrite && self.assetWriter.status != AVAssetWriterStatusUnknown) {
  460. [_assetWriter finishWritingWithCompletionHandler:^{
  461. if ([weakSelf.delegate respondsToSelector:@selector(captureTool:didFinishRecordingToOutputFileAtURL:error:)]) {
  462. SL_DISPATCH_ON_MAIN_THREAD(^{
  463. [weakSelf.delegate captureTool:weakSelf didFinishRecordingToOutputFileAtURL:weakSelf.outputFileURL error:weakSelf.assetWriter.error];
  464. });
  465. }
  466. weakSelf.canWrite = NO;
  467. weakSelf.assetWriter = nil;
  468. weakSelf.assetWriterAudioInput = nil;
  469. weakSelf.assetWriterVideoInput = nil;
  470. }];
  471. }
  472. }
  473. }
  474. #pragma mark - 重力感应监测设备方向
  475. ///开始监听设备方向
  476. - (void)startUpdateDeviceDirection {
  477. if ([self.motionManager isAccelerometerAvailable] == YES) {
  478. //回调会一直调用,建议获取到就调用下面的停止方法,需要再重新开始,当然如果需求是实时不间断的话可以等离开页面之后再stop
  479. [self.motionManager setAccelerometerUpdateInterval:1.0];
  480. __weak typeof(self) weakSelf = self;
  481. [self.motionManager startAccelerometerUpdatesToQueue:[NSOperationQueue currentQueue] withHandler:^(CMAccelerometerData *accelerometerData, NSError *error) {
  482. double x = accelerometerData.acceleration.x;
  483. double y = accelerometerData.acceleration.y;
  484. if ((fabs(y) + 0.1f) >= fabs(x)) {
  485. if (y >= 0.1f) {
  486. // NSLog(@"Down");
  487. if (weakSelf.shootingOrientation == UIDeviceOrientationPortraitUpsideDown) {
  488. return ;
  489. }
  490. weakSelf.shootingOrientation = UIDeviceOrientationPortraitUpsideDown;
  491. } else {
  492. // NSLog(@"Portrait");
  493. if (weakSelf.shootingOrientation == UIDeviceOrientationPortrait) {
  494. return ;
  495. }
  496. weakSelf.shootingOrientation = UIDeviceOrientationPortrait;
  497. }
  498. } else {
  499. if (x >= 0.1f) {
  500. // NSLog(@"Right");
  501. if (weakSelf.shootingOrientation == UIDeviceOrientationLandscapeRight) {
  502. return ;
  503. }
  504. weakSelf.shootingOrientation = UIDeviceOrientationLandscapeRight;
  505. } else if (x <= 0.1f) {
  506. // NSLog(@"Left");
  507. if (weakSelf.shootingOrientation == UIDeviceOrientationLandscapeLeft) {
  508. return ;
  509. }
  510. weakSelf.shootingOrientation = UIDeviceOrientationLandscapeLeft;
  511. } else {
  512. // NSLog(@"Portrait");
  513. if (weakSelf.shootingOrientation == UIDeviceOrientationPortrait) {
  514. return ;
  515. }
  516. weakSelf.shootingOrientation = UIDeviceOrientationPortrait;
  517. }
  518. }
  519. }];
  520. }
  521. }
  522. /// 停止监测方向
  523. - (void)stopUpdateDeviceDirection {
  524. if ([self.motionManager isAccelerometerActive] == YES) {
  525. [self.motionManager stopAccelerometerUpdates];
  526. _motionManager = nil;
  527. }
  528. }
  529. #pragma mark - AVCapturePhotoCaptureDelegate 图片输出代理
  530. /// 捕获拍摄图片的回调
  531. - (void)captureOutput:(AVCapturePhotoOutput *)output didFinishProcessingPhoto:(AVCapturePhoto *)photo error:(nullable NSError *)error API_AVAILABLE(ios(11.0)) {
  532. NSData *data = [photo fileDataRepresentation];
  533. UIImage *image = [UIImage imageWithData:data];
  534. if ([self.delegate respondsToSelector:@selector(captureTool:didOutputPhoto:error:)]) {
  535. [self.delegate captureTool:self didOutputPhoto:image error:error];
  536. }else {
  537. NSLog(@"请实现代理方法:captureTool:didOutputPhoto:error:");
  538. }
  539. }
  540. #pragma mark - AVCaptureVideoDataOutputSampleBufferDelegate AVCaptureAudioDataOutputSampleBufferDelegate 实时输出帧内容
  541. /// 实时输出采集到的音视频帧内容
  542. - (void)captureOutput:(AVCaptureOutput *)output didOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection {
  543. //提供对外接口,方便自定义处理
  544. if (output == self.videoDataOutput) {
  545. if([self.delegate respondsToSelector:@selector(captureTool:didOutputVideoSampleBuffer:fromConnection:)]) {
  546. [self.delegate captureTool:self didOutputVideoSampleBuffer:sampleBuffer fromConnection:connection];
  547. }
  548. }
  549. if (output == self.audioDataOutput) {
  550. if([self.delegate respondsToSelector:@selector(captureTool:didOutputAudioSampleBuffer:fromConnection:)]) {
  551. [self.delegate captureTool:self didOutputAudioSampleBuffer:sampleBuffer fromConnection:connection];
  552. }
  553. }
  554. if(!_isRecording || sampleBuffer == NULL) {
  555. return;
  556. }
  557. if (output == self.videoDataOutput) {
  558. //写入视频
  559. [self writerVideoSampleBuffer:sampleBuffer fromConnection:connection];
  560. }
  561. if (output == self.audioDataOutput) {
  562. //写入音频
  563. [self writerAudioSampleBuffer:sampleBuffer fromConnection:connection];
  564. }
  565. }
  566. //实时写入输出的视频
  567. - (void)writerVideoSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection {
  568. @autoreleasepool {
  569. @synchronized(self) {
  570. if (!self.canWrite && (self.avCaptureType == SLAvCaptureTypeAv || self.avCaptureType == SLAvCaptureTypeVideo)) {
  571. [self.assetWriter startWriting];
  572. [self.assetWriter startSessionAtSourceTime:CMSampleBufferGetPresentationTimeStamp(sampleBuffer)];
  573. self.canWrite = YES;
  574. }
  575. //写入视频数据
  576. if (self.assetWriterVideoInput.readyForMoreMediaData) {
  577. BOOL success = [self.assetWriterVideoInput appendSampleBuffer:sampleBuffer];
  578. if (!success) {
  579. NSLog(@"视频写入失败, 错误:%@" ,self.assetWriter.error.localizedDescription);
  580. }
  581. // if (!success) {
  582. // @synchronized (self) {
  583. // [self stopRecordVideo];
  584. // }
  585. // }
  586. }
  587. }
  588. }
  589. }
  590. //实时写入输出的音频
  591. - (void)writerAudioSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection {
  592. @autoreleasepool {
  593. @synchronized(self) {
  594. if (!self.canWrite && self.avCaptureType == SLAvCaptureTypeAudio) {
  595. [self.assetWriter startWriting];
  596. [self.assetWriter startSessionAtSourceTime:CMSampleBufferGetPresentationTimeStamp(sampleBuffer)];
  597. self.canWrite = YES;
  598. }
  599. if (self.assetWriterAudioInput.readyForMoreMediaData && self.canWrite) {
  600. //写入音频数据
  601. BOOL success = [self.assetWriterAudioInput appendSampleBuffer:sampleBuffer];
  602. if (!success) {
  603. NSLog(@"音频写入失败, 错误:%@" ,self.assetWriter.error.localizedDescription);
  604. }
  605. // if (!success) {
  606. // @synchronized (self) {
  607. // if (self.avCaptureType == SLAvCaptureTypeAudio) {
  608. // [self stopRecordAudio];
  609. // }else if (self.avCaptureType == SLAvCaptureTypeAv || self.avCaptureType == SLAvCaptureTypeVideo) {
  610. // [self stopRecordVideo];
  611. // }
  612. // }
  613. // }
  614. }
  615. }
  616. }
  617. }
  618. /// 丢帧
  619. - (void)captureOutput:(AVCaptureOutput *)output didDropSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection API_AVAILABLE(ios(6.0)) {
  620. }
  621. @end