GPUImage源碼分析與使用（三） 使用GPUImage拍照添加濾鏡 使用GPUImage拍視頻添加濾鏡

使用GPUImage拍照添加濾鏡

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view from its nib.
    
    //拍照添加濾鏡
    //1.添加按鈕拍照
    UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
    button.frame = CGRectMake((self.view.bounds.size.width-80)*0.5, self.view.bounds.size.height-120, 80, 40);
    button.backgroundColor = [UIColor blueColor];
    [button setTitle:@"拍照" forState:UIControlStateNormal];
    [button addTarget:self action:@selector(takePhoto) forControlEvents:UIControlEventTouchUpInside];
    [self.view addSubview:button];
    [self.view bringSubviewToFront:button];
    
    //2.照相機添加濾鏡
    [self addFilterCamera];
}

- (void)addFilterCamera {
    //1.獲取照相機，AVCaptureSessionPreset640x480照片尺寸，AVCaptureDevicePositionFront前置攝像頭
    _sCamera = [[GPUImageStillCamera alloc] initWithSessionPreset:AVCaptureSessionPreset640x480 cameraPosition:AVCaptureDevicePositionFront];
    
    //2.切換攝像頭
    [_sCamera rotateCamera];
    
    //3.豎屏
    _sCamera.outputImageOrientation = UIInterfaceOrientationPortrait;
    
    //4.初始化濾鏡---灰度濾鏡
    _grayFilter = [[GPUImageGrayscaleFilter alloc] init];
    [_sCamera addTarget:_grayFilter];
    
    //5.顯示在GPUImageView
    _gpuImageView = [[GPUImageView alloc] initWithFrame:self.jingImageView.frame];
    [_grayFilter addTarget:_gpuImageView];
    
    [self.view addSubview:_gpuImageView];
    
    //6.開始拍照
    [_sCamera startCameraCapture];
    
    
}

- (void)takePhoto {
    //7.拍照後處理照片，添加相冊
    [_sCamera capturePhotoAsJPEGProcessedUpToFilter:_grayFilter withCompletionHandler:^(NSData *processedJPEG, NSError *error) {
        
        [[PHPhotoLibrary sharedPhotoLibrary] performChanges:^{
            [[PHAssetCreationRequest creationRequestForAsset] addResourceWithType:PHAssetResourceTypePhoto data:processedJPEG options:nil];
        } completionHandler:^(BOOL success, NSError * _Nullable error) {
            
        }];
        
        //獲取拍攝的圖片預覽
        UIImage *newImage = [UIImage imageWithData:processedJPEG];
        
        [self.view bringSubviewToFront:self.jingImageView];
        self.jingImageView.image = newImage;
    }];
    
}

使用GPUImage拍視頻添加濾鏡

VideoManager.h

@protocol videoDelegate <NSObject>

- (void)didStartRecordVideo;

- (void)didCompressingVideo;

- (void)didEndRecordVideoWithTime:(CGFloat)totalTime outputFile:(NSString *)filePath;

@end

@interface VideoManager : NSObject

@property (nonatomic, assign) CGRect frame;
@property (nonatomic, weak) id<videoDelegate> delegate;

- (void)showWithFrame:(CGRect *)rect superView:(UIView *)view;

- (void)startRecording;
- (void)stopRecording;

@end

VideoManager.m

//視頻錄製單例
+ (instancetype)manager {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        _manager = [[VideoManager alloc] init];
    });
    return _manager;
}

//開始錄製
- (void)startRecording {
    //錄製路徑
    NSString *defaultPath = [self getVideoPathCache];
    self.moviePath = [defaultPath stringByAppendingPathComponent:[self getVideoNameWithType:@"mp4"]];
    self.movieUrl = [NSURL fileURLWithPath:self.moviePath];
    
    self.movieWriter = [[GPUImageMovieWriter alloc] initWithMovieURL:self.movieUrl size:CGSizeMake(480, 640)];
    self.movieWriter.encodingLiveVideo = YES;
    self.movieWriter.shouldPassthroughAudio = YES;
    
    [self.saturationFilter addTarget:self.movieWriter];
    
    self.videoCamera.audioEncodingTarget = self.movieWriter;
    
    //開始錄製
    [self.movieWriter startRecording];
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(didStartRecordVideo)]) {
        [self.delegate didStartRecordVideo];
    }
    [self.timer setFireDate:[NSDate distantPast]];
    [self.timer fire];
}

//結束錄製
- (void)stopRecording {
    if (self.videoCamera) {
        
        [self.timer invalidate];
        self.timer = nil;
        
        [self.movieWriter finishRecording];
        
        //移除濾鏡
        [self.saturationFilter removeTarget:self.movieWriter];
        
        self.videoCamera.audioEncodingTarget = nil;
        
        if (self.recordSecond > self.maxTime) {
            //可以清除錄製的內容
        }else {
            //壓縮視頻
            if (self.delegate && [self.delegate respondsToSelector:@selector(didCompressingVideo)]) {
                [self.delegate didCompressingVideo]; //提示正在壓縮
            }
            
            //開始壓縮
            __weak typeof(self) weakSelf = self;
            [self compressVideoWithUrl:self.movieUrl compressionType:AVAssetExportPresetMediumQuality filePath:^(NSString *resultPath, float memorySize, NSString *videoImagePath, int seconds){
                
                NSData *data = [NSData dataWithContentsOfFile:resultPath];
                CGFloat totalTime = (CGFloat)data.length/1024/1024;
                
                if (weakSelf.delegate && [weakSelf.delegate respondsToSelector:@selector(didEndRecordVideoWithTime:outputFile:)]) {
                    [weakSelf.delegate didEndRecordVideoWithTime:totalTime outputFile:resultPath];
                }
            }];
        }
    }
}

- (void)compressVideoWithUrl:(NSURL *)url compressionType:(NSString *)type filePath:(void(^)(NSString *resultPath, float memorySize, NSString *videoImagePath, int seconds))resultBlock {
    
    NSString *resultPath;
    
    // 視頻壓縮前大小
    NSData *data = [NSData dataWithContentsOfURL:url];
    CGFloat totalSize = (float)data.length / 1024 / 1024;
    NSLog(@"壓縮前大小：%.2fM",totalSize);
    AVURLAsset *avAsset = [AVURLAsset URLAssetWithURL:url options:nil];
    
    CMTime time = [avAsset duration];
    
    // 視頻時長
    int seconds = ceil(time.value / time.timescale);
    
    NSArray *compatiblePresets = [AVAssetExportSession exportPresetsCompatibleWithAsset:avAsset];
    if ([compatiblePresets containsObject:type]) {
        
        // 中等質量
        AVAssetExportSession *session = [[AVAssetExportSession alloc] initWithAsset:avAsset presetName:AVAssetExportPresetMediumQuality];
        
        // 用時間給文件命名 防止存儲被覆蓋
        NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
        [formatter setDateFormat:@"yyyy-MM-dd-HH:mm:ss"];
        
        // 若壓縮路徑不存在重新創建
        NSFileManager *manager = [NSFileManager defaultManager];
        BOOL isExist = [manager fileExistsAtPath:COMPRESSEDVIDEOPATH];
        if (!isExist) {
            [manager createDirectoryAtPath:COMPRESSEDVIDEOPATH withIntermediateDirectories:YES attributes:nil error:nil];
        }
        resultPath = [COMPRESSEDVIDEOPATH stringByAppendingPathComponent:[NSString stringWithFormat:@"user%outputVideo-%@.mp4",arc4random_uniform(10000),[formatter stringFromDate:[NSDate date]]]];
        
        session.outputURL = [NSURL fileURLWithPath:resultPath];
        session.outputFileType = AVFileTypeMPEG4;
        session.shouldOptimizeForNetworkUse = YES;
        [session exportAsynchronouslyWithCompletionHandler:^{
            
            switch (session.status) {
                case AVAssetExportSessionStatusUnknown:
                    break;
                case AVAssetExportSessionStatusWaiting:
                    break;
                case AVAssetExportSessionStatusExporting:
                    break;
                case AVAssetExportSessionStatusCancelled:
                    break;
                case AVAssetExportSessionStatusFailed:
                    break;
                case AVAssetExportSessionStatusCompleted:{
                    
                    NSData *data = [NSData dataWithContentsOfFile:resultPath];
                    // 壓縮過後的大小
                    float compressedSize = (float)data.length / 1024 / 1024;
                    resultBlock(resultPath,compressedSize,@"",seconds);
                    NSLog(@"壓縮後大小：%.2f",compressedSize);
                }
                default:
                    break;
            }
        }];
    }
}

- (void)showWithFrame:(CGRect)frame superView:(UIView *)superView {
    _frame = frame;
    [self.saturationFilter addTarget:self.displayView];
    [self.videoCamera addTarget:self.saturationFilter];
    
    [superView addSubview:self.displayView];
    [self.videoCamera startCameraCapture];
}

#pragma mark - 攝像頭輸出
- (void)willOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer {
    //返回已經添加濾鏡的sampleBuffer
}

VideoViewController.m

- (void)setupVideoManager {
    _manager = [[VideoManager alloc] init];
    _manager.delegate = self;
    [_manager showWithFrame:CGRectMake(20, 120, LMW, LMH) superView:self.view];
    
}

- (void)shootVideo {
    [_manager startRecording];
}

- (void)stopVideo {
    [_manager stopRecording];
}

- (void)playVideo {
    _player = [[AVPlayerViewController alloc] init];
    _player.player = [[AVPlayer alloc] initWithURL:[NSURL fileURLWithPath:_filePath]];
    _player.videoGravity = AVLayerVideoGravityResizeAspect;
    [self presentViewController:_player animated:NO completion:nil];
}

- (void)didStartRecordVideo {
    NSLog(@"開始錄製");
}

- (void)didCompressingVideo {
    NSLog(@"視頻壓縮中");
}

- (void)didEndRecordVideoWithTime:(CGFloat)totalTime outputFile:(NSString *)filePath {
    NSLog(@"錄製完畢---%f秒，%@", totalTime, filePath);
    _filePath = filePath;
}