iOS 修改头像，几行代码实现从相册选择照片
我的GitHub：点击打开链接
SelectPhotoManager.h
#import <Foundation/Foundation.h>
#import <UIKit/UIKit.h>
typedef enum {
PhotoCamera = 0,
PhotoAlbum,
}SelectPhotoType;
@protocol selectPhotoDelegate <NSObject>
//照片选取成功
- (void)selectPhotoManagerDidFinishImage:(UIImage *)image;
//照片选取失败
- (void)selectPhotoManagerDidError:(NSError *)error;
@end
@interface SelectPhotoManager : NSObject<UINavigationControllerDelegate,UIImagePickerControllerDelegate,UIActionSheetDelegate>
//代理对象
@property(nonatomic, weak)__weak id<selectPhotoDelegate>delegate;
//是否开启照片编辑功能
@property(nonatomic, assign)BOOL canEditPhoto;
//跳转的控制器 可选参数
@property(nonatomic, weak)__weak UIViewController *superVC;
//照片选取成功回调
@property(nonatomic, strong)void (^successHandle)(SelectPhotoManager *manager, UIImage *image);
//照片选取失败回调
@property(nonatomic, strong)void (^errorHandle)(NSString *error);
//开始选取照片
- (void)startSelectPhotoWithImageName:(NSString *)imageName;
- (void)startSelectPhotoWithType:(SelectPhotoType )type andImageName:(NSString *)imageName;
@end
SelectPhotoManager.m 文件
#import "SelectPhotoManager.h"
@implementation SelectPhotoManager {
//图片名
NSString *_imageName;
}
- (instancetype)init {
self = [super init];
if (self) {
_canEditPhoto = YES;
}
return self;
}
//开始选择照片
- (void)startSelectPhotoWithImageName:(NSString *)imageName{
_imageName = imageName;
UIAlertController *alertController = [UIAlertController alertControllerWithTitle:@"修改我的头像" message:nil preferredStyle: UIAlertControllerStyleActionSheet];
UIAlertAction *cancelAction = [UIAlertAction actionWithTitle:@"取消" style:UIAlertActionStyleCancel handler:nil];
[alertController addAction: [UIAlertAction actionWithTitle: @"拍照" style: UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
[self selectPhotoWithType:0];
}]];
[alertController addAction: [UIAlertAction actionWithTitle: @"从相册获取" style: UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
[self selectPhotoWithType:1];
}]];
[alertController addAction:cancelAction];
[[self getCurrentVC] presentViewController:alertController animated:YES completion:nil];
}
//根据类型选取照片
- (void)startSelectPhotoWithType:(SelectPhotoType)type andImageName:(NSString *)imageName {
_imageName = imageName;
UIImagePickerController *ipVC = [[UIImagePickerController alloc] init];
//设置跳转方式
ipVC.modalTransitionStyle = UIModalTransitionStyleCoverVertical;
if (_canEditPhoto) {
//设置是否可对图片进行编辑
ipVC.allowsEditing = YES;
}
ipVC.delegate = self;
if (type == PhotoCamera) {
NSLog(@"相机");
BOOL isCamera = [UIImagePickerController isCameraDeviceAvailable:UIImagePickerControllerCameraDeviceRear];
if (!isCamera) {
NSLog(@"没有摄像头");
if (_errorHandle) {
_errorHandle(@"没有摄像头");
}
UIAlertController *alertController = [UIAlertController alertControllerWithTitle:@"提示" message:@"您的设备不支持拍照" preferredStyle:UIAlertControllerStyleAlert];
[alertController addAction: [UIAlertAction actionWithTitle: @"确定" style: UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
}]];
[[self getCurrentVC] presentViewController:alertController animated:YES completion:nil];
return ;
}else{
ipVC.sourceType = UIImagePickerControllerSourceTypeCamera;
}
}else{
NSLog(@"相册");
ipVC.sourceType = UIImagePickerControllerSourceTypePhotoLibrary;
}
[[self getCurrentVC] presentViewController:ipVC animated:YES completion:nil];
}
//获取当前屏幕显示的viewcontroller
- (UIViewController *)getCurrentVC {
if (_superVC) {
return _superVC;
}
UIViewController *result = nil;
UIWindow * window = [[UIApplication sharedApplication] keyWindow];
if (window.windowLevel != UIWindowLevelNormal)
{
NSArray *windows = [[UIApplication sharedApplication] windows];
for(UIWindow * tmpWin in windows)
{
if (tmpWin.windowLevel == UIWindowLevelNormal)
{
window = tmpWin;
break;
}
}
}
UIView *frontView = [[window subviews] objectAtIndex:0];
id nextResponder = [frontView nextResponder];
if ([nextResponder isKindOfClass:[UIViewController class]]) {
result = nextResponder;
}else{
result = window.rootViewController;
}
return result;
}
#pragma mark 方法
-(void)selectPhotoWithType:(int)type {
if (type == 2) {
NSLog(@"取消");
}else{
UIImagePickerController *ipVC = [[UIImagePickerController alloc] init];
//设置跳转方式
ipVC.modalTransitionStyle = UIModalTransitionStyleCoverVertical;
if (_canEditPhoto) {
//设置是否可对图片进行编辑
ipVC.allowsEditing = YES;
}
ipVC.delegate = self;
if (type == 0) {
NSLog(@"相机");
BOOL isCamera = [UIImagePickerController isCameraDeviceAvailable:UIImagePickerControllerCameraDeviceRear];
if (!isCamera) {
NSLog(@"没有摄像头");
if (_errorHandle) {
_errorHandle(@"没有摄像头");
}
UIAlertController *alertController = [UIAlertController alertControllerWithTitle:@"提示" message:@"您的设备不支持拍照" preferredStyle:UIAlertControllerStyleAlert];
[alertController addAction: [UIAlertAction actionWithTitle: @"确定" style: UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
}]];
[[self getCurrentVC] presentViewController:alertController animated:YES completion:nil];
return ;
}else{
ipVC.sourceType = UIImagePickerControllerSourceTypeCamera;
}
}else{
NSLog(@"相册");
ipVC.sourceType = UIImagePickerControllerSourceTypePhotoLibrary;
}
[[self getCurrentVC] presentViewController:ipVC animated:YES completion:nil];
}
}
#pragma mark -----------------imagePickerController协议方法
- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary *)info{
NSLog(@"info = %@",info);
UIImage *image = [info objectForKey:@"UIImagePickerControllerEditedImage"];
if (image == nil) {
image = [info objectForKey:@"UIImagePickerControllerOriginalImage"];
}
//图片旋转
if (image.imageOrientation != UIImageOrientationUp) {
//图片旋转
image = [self fixOrientation:image];
}
if (_imageName==nil || _imageName.length == 0) {
//获取当前时间,生成图片路径
NSDate *date = [NSDate date];
NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
[formatter setDateStyle:NSDateFormatterMediumStyle];
[formatter setTimeStyle:NSDateFormatterShortStyle];
[formatter setDateFormat:@"YYYY-MM-dd hh:mm:ss"];
NSString *dateStr = [formatter stringFromDate:date];
_imageName = [NSString stringWithFormat:@"photo_%@.png",dateStr];
}
[[self getCurrentVC] dismissViewControllerAnimated:YES completion:nil];
if (_delegate && [_delegate respondsToSelector:@selector(selectPhotoManagerDidFinishImage:)]) {
[_delegate selectPhotoManagerDidFinishImage:image];
}
if (_successHandle) {
_successHandle(self,image);
}
}
- (void)imagePickerControllerDidCancel:(UIImagePickerController *)picker {
[[self getCurrentVC] dismissViewControllerAnimated:YES completion:nil];
if (_delegate && [_delegate respondsToSelector:@selector(selectPhotoManagerDidError:)]) {
[_delegate selectPhotoManagerDidError:nil];
}
if (_errorHandle) {
_errorHandle(@"撤销");
}
}
#pragma mark 图片处理方法
//图片旋转处理
- (UIImage *)fixOrientation:(UIImage *)aImage {
CGAffineTransform transform = CGAffineTransformIdentity;
switch (aImage.imageOrientation) {
case UIImageOrientationDown:
case UIImageOrientationDownMirrored:
transform = CGAffineTransformTranslate(transform, aImage.size.width, aImage.size.height);
transform = CGAffineTransformRotate(transform, M_PI);
break;
case UIImageOrientationLeft:
case UIImageOrientationLeftMirrored:
transform = CGAffineTransformTranslate(transform, aImage.size.width, 0);
transform = CGAffineTransformRotate(transform, M_PI_2);
break;
case UIImageOrientationRight:
case UIImageOrientationRightMirrored:
transform = CGAffineTransformTranslate(transform, 0, aImage.size.height);
transform = CGAffineTransformRotate(transform, -M_PI_2);
break;
default:
break;
}
switch (aImage.imageOrientation) {
case UIImageOrientationUpMirrored:
case UIImageOrientationDownMirrored:
transform = CGAffineTransformTranslate(transform, aImage.size.width, 0);
transform = CGAffineTransformScale(transform, -1, 1);
break;
case UIImageOrientationLeftMirrored:
case UIImageOrientationRightMirrored:
transform = CGAffineTransformTranslate(transform, aImage.size.height, 0);
transform = CGAffineTransformScale(transform, -1, 1);
break;
default:
break;
}
// Now we draw the underlying CGImage into a new context, applying the transform
// calculated above.
CGContextRef ctx = CGBitmapContextCreate(NULL, aImage.size.width, aImage.size.height,
CGImageGetBitsPerComponent(aImage.CGImage), 0,
CGImageGetColorSpace(aImage.CGImage),
CGImageGetBitmapInfo(aImage.CGImage));
CGContextConcatCTM(ctx, transform);
switch (aImage.imageOrientation) {
case UIImageOrientationLeft:
case UIImageOrientationLeftMirrored:
case UIImageOrientationRight:
case UIImageOrientationRightMirrored:
// Grr...
CGContextDrawImage(ctx, CGRectMake(0,0,aImage.size.height,aImage.size.width), aImage.CGImage);
break;
default:
CGContextDrawImage(ctx, CGRectMake(0,0,aImage.size.width,aImage.size.height), aImage.CGImage);
break;
}
// And now we just create a new UIImage from the drawing context
CGImageRef cgimg = CGBitmapContextCreateImage(ctx);
UIImage *img = [UIImage imageWithCGImage:cgimg];
CGContextRelease(ctx);
CGImageRelease(cgimg);
return img;
}
@end
在controller里实现下方法即可
//头像点击事件
-(void)tapClick:(UITapGestureRecognizer *)recognizer{
if (!_photoManager) {
_photoManager =[[SelectPhotoManager alloc]init];
}
[_photoManager startSelectPhotoWithImageName:@"选择头像"];
__weak typeof(self)mySelf=self;
//选取照片成功
_photoManager.successHandle=^(SelectPhotoManager *manager,UIImage *image){
mySelf.headerImage.image = image;
//保存到本地
NSData *data = UIImagePNGRepresentation(image);
[[NSUserDefaults standardUserDefaults] setObject:data forKey:@"headerImage"];
};
}