一個iOS圖片選擇器的DEMO(實現圖片添加，宮格排列，圖片長按刪除，以及圖片替換等功能)

　　在開發中，經常用到選擇多張圖片進行上傳或作其他處理等等，以下DEMO滿足了此功能中的大部分功能，可直接使用到項目中。

主要功能如下：

　　1，圖片九宮格排列（可自動設置）

　　2，圖片長按抖動（仿蘋果軟件刪除時，圖標抖動效果），可進入刪除狀態，再次單擊進入普通狀態

　　3，圖片設置最大上限，加號按鈕自動隱藏

　　4，已選圖片可單擊進行重新選擇

　　5，無需代理，直接調用對應屬性就可獲取所有圖片，並與顯示順序保持一致

　　效果圖如下：

//
//  SZAddImage.h
//  addImage
//
//  Created by mac on 14-5-21.
//  Copyright (c) 2014年 shunzi. All rights reserved.
//

/**
 * 使用說明:直接創建此view添加到你需要放置的位置即可.
 * 屬性images可以獲取到當前選中的所有圖片對象.
 */

#import <UIKit/UIKit.h>

@interface SZAddImage : UIView

/**
 *  存儲所有的照片(UIImage)
 */
@property (nonatomic, strong) NSMutableArray *images;

@end

//
//  SZAddImage.m
//  addImage
//
//  Created by mac on 14-5-21.
//  Copyright (c) 2014年 shunzi. All rights reserved.
//

#define imageH 100 // 圖片高度
#define imageW 75 // 圖片寬度
#define kMaxColumn 3 // 每行顯示數量
#define MaxImageCount 9 // 最多顯示圖片個數
#define deleImageWH 25 // 刪除按鈕的寬高
#define kAdeleImage @"close.png" // 刪除按鈕圖片
#define kAddImage @"add.png" // 添加按鈕圖片



#import "SZAddImage.h"
@interface SZAddImage()<UINavigationControllerDelegate, UIImagePickerControllerDelegate>
{
    // 標識被編輯的按鈕 -1 為添加新的按鈕
    NSInteger editTag;
}
@end

@implementation SZAddImage

- (id)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self) {
        UIButton *btn = [self createButtonWithImage:kAddImage andSeletor:@selector(addNew:)];
        [self addSubview:btn];
    }
    return self;
}

-(NSMutableArray *)images
{
    if (_images == nil) {
        _images = [NSMutableArray array];
    }
    return _images;
}

// 添加新的控件
- (void)addNew:(UIButton *)btn
{
    // 標識為添加一個新的圖片
    
    if (![self deleClose:btn]) {
        editTag = -1;
        [self callImagePicker];
    }
   
    
}

// 修改舊的控件
- (void)changeOld:(UIButton *)btn
{
    // 標識為修改(tag為修改標識)
    if (![self deleClose:btn]) {
        editTag = btn.tag;
        [self callImagePicker];
    }
}

// 刪除"刪除按鈕"
- (BOOL)deleClose:(UIButton *)btn
{
    if (btn.subviews.count == 2) {
        [[btn.subviews lastObject] removeFromSuperview];
        [self stop:btn];
        return YES;
    }
    
    return NO;
}

// 調用圖片選擇器
- (void)callImagePicker
{
    UIImagePickerController *pc = [[UIImagePickerController alloc] init];
    pc.allowsEditing = YES;
    pc.delegate = self;
    [self.window.rootViewController presentViewController:pc animated:YES completion:nil];
}


// 根據圖片名稱或者圖片創建一個新的顯示控件
- (UIButton *)createButtonWithImage:(id)imageNameOrImage andSeletor : (SEL)selector
{
    UIImage *addImage = nil;
    if ([imageNameOrImage isKindOfClass:[NSString class]]) {
       addImage = [UIImage imageNamed:imageNameOrImage];
    }
    else if([imageNameOrImage isKindOfClass:[UIImage class]])
    {
        addImage = imageNameOrImage;
    }
    UIButton *addBtn = [UIButton buttonWithType:UIButtonTypeCustom];
    [addBtn setImage:addImage forState:UIControlStateNormal];
    [addBtn addTarget:self action:selector forControlEvents:UIControlEventTouchUpInside];
    addBtn.tag = self.subviews.count;
   
    // 添加長按手勢,用作刪除.加號按鈕不添加
    if(addBtn.tag != 0)
    {
    UILongPressGestureRecognizer *gester = [[UILongPressGestureRecognizer alloc] initWithTarget:self action:@selector(longPress:)];
    [addBtn addGestureRecognizer:gester];
    }
    return addBtn;
    
}


// 長按添加刪除按鈕
- (void)longPress : (UIGestureRecognizer *)gester
{
    if (gester.state == UIGestureRecognizerStateBegan)
    {
    UIButton *btn = (UIButton *)gester.view;
    
    UIButton *dele = [UIButton buttonWithType:UIButtonTypeCustom];
    dele.bounds = CGRectMake(0, 0, deleImageWH, deleImageWH);
    [dele setImage:[UIImage imageNamed:kAdeleImage] forState:UIControlStateNormal];
    [dele addTarget:self action:@selector(deletePic:) forControlEvents:UIControlEventTouchUpInside];
    dele.frame = CGRectMake(btn.frame.size.width - dele.frame.size.width, 0, dele.frame.size.width, dele.frame.size.height);
    
    [btn addSubview:dele];
    [self start : btn];
        
        
    }
    
}

// 長按開始抖動
- (void)start : (UIButton *)btn {
    double angle1 = -5.0 / 180.0 * M_PI;
    double angle2 = 5.0 / 180.0 * M_PI;
    CAKeyframeAnimation *anim = [CAKeyframeAnimation animation];
    anim.keyPath = @"transform.rotation";
    
    anim.values = @[@(angle1),  @(angle2), @(angle1)];
    anim.duration = 0.25;
    // 動畫的重復執行次數
    anim.repeatCount = MAXFLOAT;
    
    // 保持動畫執行完畢后的狀態
    anim.removedOnCompletion = NO;
    anim.fillMode = kCAFillModeForwards;
    
    [btn.layer addAnimation:anim forKey:@"shake"];
}

// 停止抖動
- (void)stop : (UIButton *)btn{
    [btn.layer removeAnimationForKey:@"shake"];
}

// 刪除圖片
- (void)deletePic : (UIButton *)btn
{
    [self.images removeObject:[(UIButton *)btn.superview imageForState:UIControlStateNormal]];
    [btn.superview removeFromSuperview];
    if ([[self.subviews lastObject] isHidden]) {
        [[self.subviews lastObject] setHidden:NO];
    }
    
    
}

// 對所有子控件進行布局
- (void)layoutSubviews
{
    [super layoutSubviews];
    int count = self.subviews.count;
    CGFloat btnW = imageW;
    CGFloat btnH = imageH;
    int maxColumn = kMaxColumn > self.frame.size.width / imageW ? self.frame.size.width / imageW : kMaxColumn;
    CGFloat marginX = (self.frame.size.width - maxColumn * btnW) / (count + 1);
    CGFloat marginY = marginX;
    for (int i = 0; i < count; i++) {
        UIButton *btn = self.subviews[i];
        CGFloat btnX = (i % maxColumn) * (marginX + btnW) + marginX;
        CGFloat btnY = (i / maxColumn) * (marginY + btnH) + marginY;
        btn.frame = CGRectMake(btnX, btnY, btnW, btnH);
    }
    
}

#pragma mark - UIImagePickerController 代理方法
-(void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary *)info
{
    UIImage *image = info[UIImagePickerControllerEditedImage];
    if (editTag == -1) {
        // 創建一個新的控件
        UIButton *btn = [self createButtonWithImage:image andSeletor:@selector(changeOld:)];
        [self insertSubview:btn atIndex:self.subviews.count - 1];
        [self.images addObject:image];
        if (self.subviews.count - 1 == MaxImageCount) {
            [[self.subviews lastObject] setHidden:YES];
           
        }
    }
    else
    {
        // 根據tag修改需要編輯的控件
        UIButton *btn = (UIButton *)[self viewWithTag:editTag];
        int index = [self.images indexOfObject:[btn imageForState:UIControlStateNormal]];
        [self.images removeObjectAtIndex:index];
        [btn setImage:image forState:UIControlStateNormal];
        [self.images insertObject:image atIndex:index];
    }
    // 退出圖片選擇控制器
    [picker dismissViewControllerAnimated:YES completion:nil];
}



@end