趁着没什么事情，打算继续写点小轮子，便于以后项目中直接能用。这次写的是一个图片的展示器，相当于发微博时选完图片后出现的图片集视图。

代码与效果图：

a 默认情况

    __weak typeof(self) ws = self;
    YQEditPhotoContainerView *cv = [[YQEditPhotoContainerView alloc] initWithFrame:CGRectMake(0, 64, self.view.bounds.size.width, self.view.bounds.size.height - 64)];
    cv.addImageBlock = ^(NSInteger photoCount){
        [ws addImage];
    };
    [self.view addSubview:cv];

b 隐藏大按钮

    __weak typeof(self) ws = self;
    YQEditPhotoContainerView *cv = [[YQEditPhotoContainerView alloc] initWithFrame:CGRectMake(0, 64, self.view.bounds.size.width, self.view.bounds.size.height - 64)];
    cv.hiddenBigAddButton = YES;
    cv.addImageBlock = ^(NSInteger photoCount){
        [ws addImage];
    };
    [self.view addSubview:cv];

c 五列 十张

    __weak typeof(self) ws = self;
    YQEditPhotoContainerView *cv = [[YQEditPhotoContainerView alloc] initWithFrame:CGRectMake(0, 64, self.view.bounds.size.width, 150)];
    cv.totalCount = 10; // 如果要设置totalCount和colCount的话，务必要totalCount先设置，因为frame在colCount的set方法中计算，否则会数组越界而奔溃！
    cv.colCount = 5;
    cv.addImageBlock = ^(NSInteger photoCount){
        [ws addImage];
    };
    [self.view addSubview:cv];

使用简介

使用起来很简单，只需要通过- (instancetype)initWithFrame:(CGRect)frame方法初始化后添加到控制器的view中就搞定了。通过设置addImageBlock来设置点击添加图片按钮后的操作。

公开属性和方法

@property (nonatomic, strong) NSMutableArray *imageArray;              ///<  照片数组
@property (nonatomic, assign) NSInteger colCount;                      ///<  有多少列 默认3
@property (nonatomic, assign) NSInteger totalCount;                    ///<  总个数 默认9
@property (nonatomic, assign) BOOL hiddenBigAddButton;                 ///<  隐藏大的添加图片按钮 默认NO
@property (nonatomic, copy) AddImageBlock addImageBlock;               ///<  添加照片block

/**
 添加一张照片
 
 @param image 添加的照片

 */
- (void)addImage:(UIImage *)image;


/**
 添加几张照片
 
 @param imageArray 添加的照片
 
 */
- (void)addImageArray:(NSArray *)imageArray;


/**
 获得当前的照片数组
 
 */
- (NSArray *)getImageArray;

主要思路

在重写初始化的方法中设置好属性的默认值以及添加初始化按钮。并在- (void)setColCount:(NSInteger)colCount方法中通过列数和总数计算出各个imageView的frame并加入到imageViewFrameArr数组中。
每添加一张照片，就初始化一个imageView，每删除一张图片，就把imageView移除父视图。
最后重写- (void)layoutSubviews，在该方法�中拿到imageViewFrameArr数组，设置各个imageView和按钮的frame。同时也判断大小按钮是否需要隐藏。

注意

其中需要注意的是，如果totalCount和colCount都要设置的话，务必要totalCount先设置，因为imageview的frame是在colCount的set方法中计算的，否则会数组越界而奔溃！

最后

代码地址