#import "ViewController.h"
#import <MobileCoreServices/MobileCoreServices.h>
#import <AVFoundation/AVFoundation.h>

@interface ViewController ()<UIImagePickerControllerDelegate,UINavigationControllerDelegate>

@property (nonatomic, strong) UIImagePickerController *imagePick;
@property (nonatomic, assign) int isVideo;
@property (nonatomic, weak)   IBOutlet UIImageView *photo;
@property (nonatomic, strong) AVPlayer *player;

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    _isVideo = YES;//是否是錄制視頻
}
- (IBAction)takeClick:(id)sender {
    [self presentViewController:self.imagePick animated:YES completion:nil];
}

#pragma mark - UIImagePickerControllerDelegate
-(void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary<NSString *,id> *)info
{
    NSString *mediaType = [info objectForKey:UIImagePickerControllerMediaType];
    if ([mediaType isEqualToString:(NSString *)kUTTypeImage]) {//如果是拍照
        UIImage *image;
        if (self.imagePick.allowsEditing) {
            image = [info objectForKey:UIImagePickerControllerEditedImage];//獲取編輯后的照片
        }else{
            image = [info objectForKey:UIImagePickerControllerOriginalImage];//獲取原始照片
        }
        [self.photo setImage:image];//顯示照片
        UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil);//保存到相冊(cè)
    }else if ([mediaType isEqualToString:(NSString *)kUTTypeMovie]){
        NSURL *url = [info objectForKey:UIImagePickerControllerMediaURL];
        NSString *urlStr = [url path];
        if (UIVideoAtPathIsCompatibleWithSavedPhotosAlbum(urlStr)) {
            UISaveVideoAtPathToSavedPhotosAlbum(urlStr, self, @selector(video:didFinishSavingWithError:contextInfo:), nil);
        }
    }
    
    [self dismissViewControllerAnimated:YES completion:nil];
}

#pragma mark - 私有方法
-(UIImagePickerController *)imagePick{
    if (_imagePick == nil) {
        _imagePick = [[UIImagePickerController alloc]init];
        _imagePick.sourceType = UIImagePickerControllerSourceTypeCamera;//設(shè)置image picker的來(lái)源
        _imagePick.cameraDevice = UIImagePickerControllerCameraDeviceFront;//設(shè)置攝像頭類(lèi)型
    
    if (self.isVideo) {
        _imagePick.mediaTypes = @[(NSString *)kUTTypeMovie];
        _imagePick.videoQuality = UIImagePickerControllerQualityTypeIFrame1280x720;
        _imagePick.cameraCaptureMode = UIImagePickerControllerCameraCaptureModeVideo;//設(shè)置攝像頭模式
    }else{
        _imagePick.cameraCaptureMode = UIImagePickerControllerCameraCaptureModePhoto;
    }
    _imagePick.allowsEditing = YES;//允許編輯
    _imagePick.delegate = self;//設(shè)置代理
    }
    return _imagePick;
}

//視屏保存后的回調(diào)
-(void)video:(NSString *)videoPath didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo{
    if (error) {
        NSLog(@"發(fā)生錯(cuò)誤");
    }else{
        NSURL *url = [NSURL fileURLWithPath:videoPath];
        _player = [AVPlayer playerWithURL:url];
        AVPlayerLayer *playerLayer = [AVPlayerLayer playerLayerWithPlayer:_player];
        playerLayer.frame = self.photo.frame;
        [self.photo.layer addSublayer:playerLayer];
        [_player play];
    }
}

@end

測(cè)試截圖.jpg