swift瀑布流自定义布局实现

demo github下载地址
https://github.com/liwei5bao/SwiftWaterfallProject
//  ViewController.swift

//  SwiftWaterfall

import UIKit

import UIKit

class ViewController:UIViewController{

    
   var collectionView:UICollectionView?

    ///重用item的标识
   let itwmID =
"shopID"

    ///商品数组
   lazy
var shops:NSMutableArray? = {

        

        returnNSMutableArray()

        
    }()

    
   override
func viewDidLoad() {
       super.viewDidLoad()

        
       //初始化布局
       self.setupLayout()

        //初始化上拉加载更多,下拉加载

        self.setupRefresh()

        self.loadNewShops()
    }

    

    //MARK:初始化数据
   private
func setupList(){

        
       for i
in 0...10{

            
           var model:NSDictionary?

            
           if
Int(i % 2) ==
0{

                

                model = ["w":NSNumber(int:200),"h":NSNumber(int:300),"img":"http://s6.mogujie.cn/b7/bao/131008/q2o17_kqyvcz3ckfbewv3wgfjeg5sckzsew_330x445.jpg_200x999.jpg","price":""]

                
            }else{

                

                model = ["w":NSNumber(int:200),"h":NSNumber(int:400),"img":"http://s6.mogujie.cn/b7/bao/131008/q2o17_kqyvcz3ckfbewv3wgfjeg5sckzsew_330x445.jpg_200x999.jpg","price":""]
            }

            
           self.shops?.addObject(model!)
        }

        self.collectionView?.header.endRefreshing()

        self.collectionView?.footer.endRefreshing()
    }

    

    //MARK:初始化的内容
   private
func setupRefresh(){

        

        self.collectionView?.addLegendHeaderWithRefreshingTarget(self,
refreshingAction: "loadNewShops")//header = MJRefreshNormalHeader(refreshingTarget: self, refreshingAction: "loadNewShops")

        self.collectionView?.addLegendFooterWithRefreshingTarget(self,
refreshingAction: "loadMoreShops")//footer = MJRefreshAutoFooter(refreshingTarget: self, refreshingAction: "loadMoreShops")

        self.collectionView?.footer.hidden =true
    }

    
   private
func setupLayout(){

        
       let layout =
OOWaterflowLayout()
        layout.delegate =self

        //创建collectionView
       let collectionView =
UICollectionView(frame:
self.view.bounds, collectionViewLayout: layout)
        collectionView.backgroundColor =UIColor.whiteColor()
        collectionView.dataSource =self
       self.view.addSubview(collectionView)
        collectionView.registerNib(UINib(nibName:"OOShopCell", bundle:
nil), forCellWithReuseIdentifier:
itwmID)
       self.collectionView = collectionView

        
    }

    

    

    ///加载新数据
   func loadNewShops(){

        

        self.shops?.removeAllObjects()
       self.setupList()
       //刷新数据

        self.collectionView?.reloadData()

        self.collectionView?.header.endRefreshing()
    }

    

    ///加载更多的数据
   func loadMoreShops(){

        
       self.setupList()
       //刷新数据

        self.collectionView?.reloadData()

        self.collectionView?.footer.endRefreshing()

        
    }

    
   override
func didReceiveMemoryWarning() {

        super.didReceiveMemoryWarning()

        
    }

    
}

extension ViewController:UICollectionViewDataSource{

    

    //MARK:UICollectionViewDataSource的数据源
   func collectionView(collectionView:
UICollectionView, numberOfItemsInSection section:
Int) -> Int {

        

        self.collectionView?.footer.hidden = (self.shops?.count
== 0)
       return (self.shops?.count)!
    }

    
   func collectionView(collectionView:
UICollectionView, cellForItemAtIndexPath indexPath:
NSIndexPath) -> UICollectionViewCell {

        
       let cell = collectionView.dequeueReusableCellWithReuseIdentifier(itwmID,
forIndexPath: indexPath)as?
OOShopCell
        cell?.shops =self.shops![indexPath.item]as?
NSDictionary
       return cell!
    }

    
}

extension ViewController:OOWaterflowLayoutDeledate{

    

    ///必须实现的方法
   func waterflowLayout(waterflowLayout waterflowLayout:OOWaterflowLayout, heightForItemAtIndex index:NSInteger,
itemWidth width: CGFloat) ->
CGFloat {

        
       let shop =
self.shops![index]as!
NSDictionary
       return width *
CGFloat(shop["h"]as!
NSNumber) /CGFloat(shop["w"] as!
NSNumber)
    }

    

    //MARK:下边的代理可以不实现有默认值

    ///返回列数
   func columnCountInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->
NSInteger {

        
       return
2
    }

    

    ///返回列间距
   func columnMarginInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->
CGFloat {

        
       return
10
    }

    

    ///返回行间距
   func rowMarginInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->
CGFloat {

        
       return
10
    }

    

    ///返回内边距
   func edgeInsetsInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->
UIEdgeInsets {

        
       return
UIEdgeInsetsMake(20,10,
10, 10)
    }
}

//  OOWaterflowLayout.swift

//  swift瀑布流

import UIKit

class OOWaterflowLayout:UICollectionViewLayout {

    ///代理

    weakvar delegate:OOWaterflowLayoutDeledate?

    ///默认的列数
   let OODefaultColumnCount:NSInteger =2

    /**
每一列之间的间距 */
   let OODefaultColumnMargin:CGFloat =10

    /**
每一行之间的间距 */
   let OODefaultRowMargin:CGFloat =10

    /**
边缘间距 */
   let OODefaultEdgeInsets:UIEdgeInsets =UIEdgeInsetsMake(10,10,
10, 10)

    

    /**
存放所有cell的布局属性 */

    lazyvar attrsArray:[UICollectionViewLayoutAttributes]
= {return [UICollectionViewLayoutAttributes]()}()

    /**
存放所有列的当前高度 */
   lazy
var columnHeights:NSMutableArray = {returnNSMutableArray()}()
   var contentHeight:CGFloat?

    

    ///行间距
   var rowMargin:CGFloat{

        
       get{

            

            if ((self.delegate?.respondsToSelector("rowMarginInWaterflowLayout:"))
== true){

            
               return (self.delegate?.rowMarginInWaterflowLayout!(self))!
            }else{

            

                returnOODefaultRowMargin
            }

            
        }

        
    }

    

    ///列间距
   var columnMargin:CGFloat{

        
       get{

            

            if ((self.delegate?.respondsToSelector("columnMarginInWaterflowLayout:"))
== true){

                
               return (self.delegate?.columnMarginInWaterflowLayout!(self))!
            }else{

                

                returnOODefaultColumnMargin
            }

            
        }
    }

    

    ///列数
   var columnCount:NSInteger{

        
       get{

            

            if ((self.delegate?.respondsToSelector("columnMarginInWaterflowLayout:"))
== true){

                
               return (self.delegate?.columnCountInWaterflowLayout!(self))!
            }else{

                

                returnOODefaultColumnCount
            }

            
        }
    }

    

    ///边距
   var edgeInsets:UIEdgeInsets{

        
       get{

            

            if ((self.delegate?.respondsToSelector("edgeInsetsInWaterflowLayout:"))
== true){

                
               return (self.delegate?.edgeInsetsInWaterflowLayout!(self))!
            }else{

                

                returnOODefaultEdgeInsets
            }

            
        }
    }

}

//MARK:初始化和计算

extension OOWaterflowLayout{

    ///1.布局的初始化操作
   override
func prepareLayout() {

        super.prepareLayout()

        

        self.contentHeight =0

        //清除之前计算的高度

        self.columnHeights.removeAllObjects()
       for
_ in
0...self.columnCount -1{

        

            self.columnHeights.addObject(self.edgeInsets.top)
        }

        

        //清除之前所有的布局属性

        self.attrsArray.removeAll()

        

        //创建每一个cell的布局属性

        let count =self.collectionView?.numberOfItemsInSection(0)
       for
var i =0;i < count;i++ {

        
           let indexpath =
NSIndexPath(forItem: i, inSection:0)
           //获取布局属性

            let attrs =self.layoutAttributesForItemAtIndexPath(indexpath)
           self.attrsArray.append(attrs!)
        }

        
    }

    ///2.决定cell的排布
   override
func layoutAttributesForElementsInRect(rect:CGRect) -> [UICollectionViewLayoutAttributes]? {

        
       return
self.attrsArray
    }

    

    ///3.返回indexPath位置cell对应的布局属性
   override
func layoutAttributesForItemAtIndexPath(indexPath:NSIndexPath) ->
UICollectionViewLayoutAttributes? {

        

        //创建布局属性
       let attrs =
UICollectionViewLayoutAttributes(forCellWithIndexPath:indexPath)

        //创建collectionView的宽度
       let collectionViewW =
self.collectionView?.frame.size.width
       let coCount:CGFloat =CGFloat(NSNumber(integer:self.columnCount))

        //设置布局的frame
       let w:CGFloat = (collectionViewW! -self.edgeInsets.left
-self.edgeInsets.right - (coCount -1) *
self.columnMargin) / coCount
       let h:CGFloat = (self.delegate?.waterflowLayout(waterflowLayout:self,
heightForItemAtIndex: indexPath.item, itemWidth: w))!

        

        //找出高度最短的那一列
       var destColumn:CGFloat =0
       var minColumnHeight:CGFloat = (self.columnHeights[0]as?
CGFloat)!
       for
var i =1;i <
self.columnCount;i++ {

            
           let columnHeight = (self.columnHeights[i]as?
CGFloat)!
           if minColumnHeight > columnHeight{

            
                minColumnHeight = columnHeight
                destColumn =CGFloat(i)
            }
        }

        
       let x:CGFloat =self.edgeInsets.left
+ destColumn * (w +self.columnMargin)
       var y:CGFloat = minColumnHeight
       if y !=
self.edgeInsets.top{

        
            y +=self.rowMargin
        }

        
        attrs.frame =CGRectMake(x, y, w, h)

        

        //更新最短的那一列的高度
       self.columnHeights[NSInteger(destColumn)] =CGRectGetMaxY(attrs.frame)

        

        //记录内容的高度
       let columnHeight:CGFloat =self.columnHeights[NSInteger(destColumn)]as!
CGFloat

        
       if
self.contentHeight < columnHeight{

        
           self.contentHeight = columnHeight
        }

        
       return attrs
    }

    

    
   override
func collectionViewContentSize() ->CGSize {

        returnCGSizeMake(0,self.contentHeight!
+self.edgeInsets.bottom)

    }

    
}

//MARK:代理
@objcprotocol OOWaterflowLayoutDeledate:NSObjectProtocol{

    ///返回每个item的高度必须实现
   func waterflowLayout(waterflowLayout waterflowLayout:OOWaterflowLayout,heightForItemAtIndex index:NSInteger,itemWidth
width:CGFloat)->CGFloat

    ///返回的列数不实现默认为3列
   optional
func columnCountInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->NSInteger

    ///返回列间距,默认为10
   optional
func columnMarginInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->CGFloat

    ///返回行间距,默认为10
   optional
func rowMarginInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->CGFloat

    ///返回内边距,默认为(10,10,10,10)
   optional
func edgeInsetsInWaterflowLayout(waterflowLayout:OOWaterflowLayout) ->UIEdgeInsets

    
}