iOS UICollectionView 瀑布流、标签(UICollectionViewFlowLayout)

瀑布流

import UIKit

public protocol SPWaterflowLayoutDelegate: NSObjectProtocol {
    func waterflowLayout(view: SPWaterflowLayout, heightForItemAtIndex: IndexPath, itemWidth: CGFloat) -> CGFloat
}

public class SPWaterflowLayout: UICollectionViewFlowLayout {
    
    weak open var delegate: SPWaterflowLayoutDelegate?
    
    public var columnCount: NSInteger = 2
    public var columnMargin: CGFloat = 10
    public var rowMargin: CGFloat = 10
    public var edgeInsets: UIEdgeInsets = UIEdgeInsets(top: 10, left: 10, bottom: 10, right: 10)
    
    private var attrsArray = [UICollectionViewLayoutAttributes]()
    private var columnHeights = [CGFloat]()
    private var contentHeight: CGFloat = 0
}

extension SPWaterflowLayout {
    
    public override func prepare() {
        super.prepare()
        
        contentHeight = 0
        columnHeights.removeAll()
        
        for _ in 0..<columnCount {
            columnHeights.append(edgeInsets.top)
        }
        
        attrsArray.removeAll()
        guard let sections = collectionView?.numberOfSections else { return }
        for section in 0..<sections {
            guard let count = collectionView?.numberOfItems(inSection: section) else { return }
            for i in 0..<count {
                if let attrs = layoutAttributesForItem(at: IndexPath(item: i, section: section)) {
                    attrsArray.append(attrs)
                }
            }
        }
    }
    
    public override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        return attrsArray
    }
    
    public override func layoutAttributesForItem(at indexPath: IndexPath) -> UICollectionViewLayoutAttributes? {
        
        let attrs = UICollectionViewLayoutAttributes(forCellWith: indexPath)
        guard let collectionViewWidth = collectionView?.frame.width else { return attrs }
        
        let width = (collectionViewWidth - edgeInsets.left - edgeInsets.right - (CGFloat(columnCount) - 1) * columnMargin) / CGFloat(columnCount)
        guard let height = delegate?.waterflowLayout(view: self, heightForItemAtIndex: indexPath, itemWidth: width) else { return attrs }
        
        var destColumn: NSInteger = 0
        var minColumnHeight: CGFloat = columnHeights[0]
        
        for i in 0..<columnCount {
            let columnHeight = columnHeights[i]
            if minColumnHeight > columnHeight {
                minColumnHeight = columnHeight
                destColumn = i
            }
        }
        
        let x = edgeInsets.left + CGFloat(destColumn) * (width + columnMargin)
        var y = minColumnHeight
        
        if y != edgeInsets.top {
            y += rowMargin
        }
        
        attrs.frame = CGRect(x: x, y: y, width: width, height: height)
        
        columnHeights[destColumn] = attrs.frame.maxY
        
        let columnHeight = columnHeights[destColumn]
        if contentHeight < columnHeight {
            contentHeight = columnHeight
        }
        
        return attrs
    }
    
    public override var collectionViewContentSize: CGSize {
        return CGSize(width: 0, height: contentHeight + edgeInsets.bottom)
    }
}
image.png

标签

import UIKit

public protocol SPTagsListLayoutDelegate: NSObjectProtocol {
    func tagsListLayout(view: SPTagsListLayout, widthForItemAtIndex: IndexPath, itemHeight: CGFloat) -> CGFloat
    func tagsListLayout(view: SPTagsListLayout, maxHeight: CGFloat)
}

public class SPTagsListLayout: UICollectionViewFlowLayout {
    
    weak open var delegate: SPTagsListLayoutDelegate?
    
    public var itemHeight: CGFloat = 25
    public var columnMargin: CGFloat = 10
    public var rowMargin: CGFloat = 10
    public var edgeInsets: UIEdgeInsets = UIEdgeInsets(top: 10, left: 10, bottom: 10, right: 10)
    
    private var attrsArray = [UICollectionViewLayoutAttributes]() 
    private var preAttrs = [String : UICollectionViewLayoutAttributes]()
    private var maxY: CGFloat = 0
}

extension SPTagsListLayout {
    
    public override func prepare() {
        super.prepare()
        
        attrsArray.removeAll()
        guard let sections = collectionView?.numberOfSections else { return }
        for section in 0..<sections {
            guard let count = collectionView?.numberOfItems(inSection: section) else { return }
            for i in 0..<count {
                if let attrs = layoutAttributesForItem(at: IndexPath(item: i, section: section)) {
                    attrsArray.append(attrs)
                }
            }
        }
    }
    
    public override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        return attrsArray
    }
    
    public override func layoutAttributesForItem(at indexPath: IndexPath) -> UICollectionViewLayoutAttributes? {
        
        let attrs = UICollectionViewLayoutAttributes(forCellWith: indexPath)
        guard let collectionViewWidth = collectionView?.frame.width,
              var currentWidth = delegate?.tagsListLayout(view: self, widthForItemAtIndex: indexPath, itemHeight: itemHeight) else { return attrs }
        currentWidth = min(currentWidth, collectionViewWidth - edgeInsets.left - edgeInsets.right)
        
        let key = "preAttrs_key_\(indexPath.section)_\(indexPath.row)"
        var x: CGFloat = edgeInsets.left
        var y: CGFloat = edgeInsets.top
        
        if indexPath.row - 1 >= 0 {
            let preKey = "preAttrs_key_\(indexPath.section)_\(indexPath.row - 1)"
            if let preAttr = preAttrs[preKey] {
                y = preAttr.frame.origin.y
                x = (preAttr.frame.maxX + columnMargin)
            }
        }
        
        if x + currentWidth > collectionViewWidth - edgeInsets.right {
            x = edgeInsets.left
            y += (rowMargin + itemHeight)
        }
        
        let attr = UICollectionViewLayoutAttributes(forCellWith: indexPath)
        attr.frame = CGRect(x: x, y: y, width: currentWidth, height: itemHeight)
        
        preAttrs[key] = attr
        maxY = attr.frame.maxY
        
        return attr
    }
    
    public override var collectionViewContentSize: CGSize {
        delegate?.tagsListLayout(view: self, maxHeight: maxY + edgeInsets.bottom)
        return CGSize(width: 0, height: maxY + edgeInsets.bottom)
    }
}
标签使用 :

代理 func tagsListLayout(view: SPTagsListLayout, maxHeight: CGFloat) 返回的 maxHeight 不要超过 UICollectionView 的高度

import UIKit

class ViewController: UIViewController, SPTagsListLayoutDelegate, UICollectionViewDelegate, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout {
    
    private lazy var contentView: UIView = {
        let view = UIView(frame: CGRect(x: 20, y: 100, width: UIScreen.main.bounds.width - 40, height: 0))
        view.clipsToBounds = true
        return view
    }()
    
    private lazy var collectionView: UICollectionView = {
        
        let layout = SPTagsListLayout()
        layout.delegate = self
        
        let view = UICollectionView(frame: CGRect(x: 0, y: 0, width: UIScreen.main.bounds.width - 40, height: UIScreen.main.bounds.height), collectionViewLayout: layout)
        view.register(UICollectionViewCell.self, forCellWithReuseIdentifier: "UICollectionViewCellID")
        view.delegate = self
        view.dataSource = self
        view.backgroundColor = .white
        
        return view
    }()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        view.backgroundColor = UIColor.lightGray
        
        view.addSubview(contentView)
        contentView.addSubview(collectionView)
    }
    
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return 50
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "UICollectionViewCellID", for: indexPath)
        cell.backgroundColor = UIColor(red: CGFloat(arc4random() % 256) / 255.0, green: CGFloat(arc4random() % 256) / 255.0, blue: CGFloat(arc4random() % 256) / 255.0, alpha: 1.0)
        return cell
    }
    
    func tagsListLayout(view: SPTagsListLayout, widthForItemAtIndex: IndexPath, itemHeight: CGFloat) -> CGFloat {
        return CGFloat(arc4random_uniform(100) + 10)
    }
    
    func tagsListLayout(view: SPTagsListLayout, maxHeight: CGFloat) {
        contentView.frame.size.height = maxHeight
    }
}
image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349