photoWall自定义

- (instancetype)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self) {
        _scrollView = [[UIScrollView alloc] initWithFrame:self.bounds];
        [self addSubview:_scrollView];

        _cellArray = [NSMutableArray array];

        _numberOfColumns = 4;
        _spacing=5;
        _edgeInset=UIEdgeInsetsMake(20, 0, 20, 0);
    }

    return self;
}

- (void)reloadData
{
    //重新加载数据前需要将老数据清理掉

    //获取视图（小格子）的个数
    NSInteger count = [_dataSource numberOfViewInPhotoWall:self];

    CGFloat width = self.bounds.size.width / _numberOfColumns;
    CGFloat height = 44.0f;
    if (_dataSource && [_dataSource respondsToSelector:@selector(photoWall:cellHeightAtIndex:)]) {

    }

    //计算滚动视图的大小(有问题，自己算..... 高度不对)
    int heiCount;
    if(count%_numberOfColumns)
    {
        heiCount=count/_numberOfColumns+1;
    }
    else
    {
        heiCount=count/_numberOfColumns;
    }
    _scrollView.contentInset=_edgeInset;
    _scrollView.contentSize = CGSizeMake(_scrollView.frame.size.width,heiCount * height);

    //总数与index相差1
    for (NSInteger i = 0; i < count; i++) {
        MyPhotoWallCell *cell = [_dataSource photoWall:self cellAtIndex:i];
        [cell addTarget:self action:@selector(didClicked:) forControlEvents:UIControlEventTouchUpInside];
        cell.tag = i;
        if (_dataSource&&[_dataSource respondsToSelector:@selector(photoWall:cellHeightAtIndex:)]) {
            height = [_dataSource photoWall:self cellHeightAtIndex:i];
        }
        CGFloat y = (i / _numberOfColumns) * height;
        NSLog(@"%f--%f", width * i, y);
        cell.frame = CGRectMake(width * (i % _numberOfColumns), y, width-_spacing, height-_spacing);
        [_scrollView addSubview:cell];
    }
}

- (void)didClicked:(MyPhotoWallCell *)sender
{
    if (_delegate && [_delegate respondsToSelector:@selector(photoWall:didSelectedIndex:)]) {
        [_delegate photoWall:self didSelectedIndex:sender.tag];
    }
}

//
//  ViewController.m
//  MyCustomPhotoWall
//
//  Created by apple on 14-8-22.
//  Copyright (c) 2014年 戴维营教育. All rights reserved.
//

#import "ViewController.h"

#import "MyPhotoWallCell.h"
#import "MyPhotoWall.h"

@interface ViewController () <MyPhotoWallDataSource, MyPhotoWallDelegate>

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];

    MyPhotoWall *photoWall = [[MyPhotoWall alloc] initWithFrame:CGRectMake(10, 10, 300, 460)];
    photoWall.backgroundColor = [UIColor redColor];
    photoWall.dataSource = self;
    photoWall.delegate = self;
    [self.view addSubview:photoWall];

    [photoWall reloadData];
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

- (NSInteger)numberOfViewInPhotoWall:(MyPhotoWall *)photoWall
{
    return 100;
}

- (CGFloat)photoWall:(MyPhotoWall *)photoWall cellHeightAtIndex:(NSInteger)index
{
    if (index%2) {
        return 94;
    }
    else
    {
        return 55;
    }
}

- (MyPhotoWallCell *)photoWall:(MyPhotoWall *)photoWall cellAtIndex:(NSInteger)index
{
    MyPhotoWallCell *cell = [[MyPhotoWallCell alloc] init];

    if (index % 3) {
        cell.backgroundColor = [UIColor purpleColor];
    }
    else if (index % 2) {
        cell.backgroundColor = [UIColor magentaColor];
    }
    else {
        cell.backgroundColor = [UIColor cyanColor];
    }

//    if (index % 2) {
//        cell.backgroundColor = [UIColor purpleColor];
//    }
//    else {
//        cell.backgroundColor = [UIColor cyanColor];
//    }

    return cell;
}

- (void)photoWall:(MyPhotoWall *)photoWall didSelectedIndex:(NSInteger)index
{
    NSLog(@"Selected: %d", index);
}
@end