iOS获取相册/相机图片-------自定义获取图片小控件

@interface ZQImageToolsButton : UIButton


/**
 *  可以获得储存图片信息的字典，里面包含了沙盒里的图片完整路径(程序每次运行沙盒的路径可能会不同，但是图片名不变)
 *
 *  @return
 */
-(NSDictionary * )getImageInfro;

/**
 *  初始化方法，点击按钮可以选择从相册或是相机（如果有相机）选择图片并返回图片
 *
 *  @param title      按钮的名字
 *  @param frame      按钮的frame
 *  @param controller 按钮所在的控制器，弹出对话框的控制器
 *  @param block      获得图片后的回调
 *  @param imageName  储存在沙盒里图片的名字
 *
 *  @return 返回一个按钮
 */
-(instancetype)initWithTitle:(NSString *)title frame:(CGRect)frame controller:(UIViewController *)controller imageBlock:(void(^)(UIImage *))block imageName :(NSString *)imageName;


@end

@interface ZQImageToolsButton() <UIImagePickerControllerDelegate,UINavigationControllerDelegate>

@property(nonatomic,weak) UIViewController * controller;

@property(nonatomic,copy) void(^imageBlock)(UIImage *) ;

@property(nonatomic,strong) NSString * imageName;

@property(nonatomic,strong) NSDictionary * imageInfro;


@end



@implementation ZQImageToolsButton

-(NSDictionary * )getImageInfro
{
    return self.imageInfro;

}


-(instancetype)initWithTitle:(NSString *)title frame:(CGRect)frame controller:(UIViewController *)controller imageBlock:(void(^)(UIImage *))block imageName :(NSString *)imageName
{

    self = [[ZQImageToolsButton alloc]initWithFrame:frame];
    
    self.imageBlock = block;
    
    self.controller = controller;
    
    self.imageName = imageName;
    
    [self setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
    
    [self setTitle:title forState:UIControlStateNormal];
    
    [self addTarget:self action:@selector(btnClick:) forControlEvents:UIControlEventTouchUpInside];
    
    return self ;

}




- (void)btnClick:(UIButton *) btn
{
    
    // 创建图片管理器、 跳转到相机或相册页面
    UIImagePickerController *imagePickerController = [[UIImagePickerController alloc] init];
    
    imagePickerController.delegate = self;
    
    imagePickerController.allowsEditing = YES;
    
    
    //创建弹框
    
    UIAlertController * av = [UIAlertController alertControllerWithTitle:@"选择" message:@"从您的相册/相机选择" preferredStyle:UIAlertControllerStyleActionSheet];
    
    
    //设置image管理器的源类型
    
    __block NSInteger sourceType = 0;
    
    UIAlertAction * photo = [UIAlertAction actionWithTitle:@"我的相册" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        
        
        //图片源为相册
        
        sourceType = UIImagePickerControllerSourceTypeSavedPhotosAlbum;
        
        imagePickerController.sourceType = sourceType;
        
        [self.controller presentViewController:imagePickerController animated:YES completion:^{}];
        
    }];
    
    
    UIAlertAction * camera = [UIAlertAction actionWithTitle:@"我的相机" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        
        
        //图片源为相机
        
        sourceType = UIImagePickerControllerSourceTypeCamera;
        
        imagePickerController.sourceType = sourceType;
        
        [self.controller presentViewController:imagePickerController animated:YES completion:^{}];
        
        
    }];
    
    UIAlertAction * cancle = [UIAlertAction actionWithTitle:@"取消" style:UIAlertActionStyleCancel handler:^(UIAlertAction * _Nonnull action) {
        //取消
    }];
    
    
    
    //判断是否支持相机
    
    if([UIImagePickerController isSourceTypeAvailable:UIImagePickerControllerSourceTypeCamera])
    {
        [av addAction:photo];
        [av addAction:camera];
        [av addAction:cancle];
        
    }
    else
    {
        //如果不支持相机，就不用把相机按钮加入提示框了
        [av addAction:photo];
        [av addAction:cancle];
    }
  
    [self.controller presentViewController:av animated:YES completion:nil];
    
}

// 图片选择结束之后，走这个方法，字典存放所有图片信息
#pragma mark - image picker delegte 系统代理，返回从相册中选中的图片
- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary *)info
{
    [picker dismissViewControllerAnimated:YES completion:^{
    
        NSLog(@"选好图片了");
    
    }];
    
    //获得选中的图片了！
    UIImage *image = [info objectForKey:UIImagePickerControllerOriginalImage];
    
    //block回调，获得图片后，交给控制器
    self.imageBlock(image);
    
    // 保存图片至本地，用传进来的名字重新拼接图片名字
    
    NSString * imgName = [NSString stringWithFormat:@"SavedImage-%@.png",self.imageName];
    
    //拼接保存至沙盒的路径
    NSString *fullPath = [[NSHomeDirectory() stringByAppendingPathComponent:@"Documents"] stringByAppendingPathComponent:imgName];
    
    //记录一下图片的信息，存到字典中，需要的时候可以访问
    NSDictionary *dic = @{@"image":fullPath};
    self.imageInfro = dic;
    
    //保存到沙盒
    
    [self saveImage:image withName:imgName filePath:fullPath];
    
}

//保存至沙盒
- (void) saveImage:(UIImage *)currentImage withName:(NSString *)imageName filePath:(NSString *)filePath
{
    //压缩比例0.5
    NSData * imageData = UIImageJPEGRepresentation(currentImage, 0.5);

    // 将图片写入文件
    [imageData writeToFile:filePath atomically:NO];
}


@end