Блог

Я создал пользовательский макет представления коллекции, используя UICollectionViewLayout. Макет и разделы отображаются правильно, но я пытаюсь добавить заголовок раздела к каждому из 4 разделов с фоном и меткой, используя viewForSupplementaryElementOfKind. Добавление функции в класс не приводит к отображению заголовка.

 import UIKit

protocol CustomCollectionViewDelegate: class {
     func theNumberOfItemsInCollectionView() -> Int
}

extension CustomCollectionViewDelegate {
    func heightForContentInItem(inCollectionView collectionView: UICollectionView, at indexPath: IndexPath) -> CGFloat {
        return 0
    }
}

class CustomCollectionViewLayout: UICollectionViewLayout {
    fileprivate var numberOfColumns = 3
    fileprivate var cellPadding: CGFloat = 0
    fileprivate let cellHeight: CGFloat = 110
    
    
    weak var delegate: CustomCollectionViewDelegate?
    
    
    //An array to cache the calculated attributes
    fileprivate var cache = [UICollectionViewLayoutAttributes]()
    
    //For content size
    fileprivate var contentHeight: CGFloat = 0
    fileprivate var contentWidth: CGFloat {
        guard let collectionView = collectionView else {return 0}
        let insets = collectionView.contentInset
        return collectionView.bounds.width - (insets.left   insets.right)
    }
    
    //Setting the content size
    override var collectionViewContentSize: CGSize {
        return CGSize(width: contentWidth, height: contentHeight)
    }
    
    override func prepare() {
        //We begin measuring the location of items only if the cache is empty
        guard cache.isEmpty == true, let collectionView = collectionView else {return}
        
        let columnWidth = contentWidth / CGFloat(numberOfColumns)
        var xOffset = [CGFloat]()
        
        //Getting the xOffset based on the column and column width
        for column in 0..<numberOfColumns {
            
            if column == 0 {
                 xOffset.append(0)
            }
            if column == 1 {
                 xOffset.append(2 * columnWidth)
            }
            if column == 2
            {
                 xOffset.append( columnWidth)
            }
            
           // xOffset.append(CGFloat(column) * columnWidth)
        }
        
        var column = 0
        var yOffset = [CGFloat]()
        
        
        //For each section in the collection view
        
        for the section in 0...3 {
        //Header code goes here
            


            
        for item in 0..<collectionView.numberOfItems(inSection: section) {
            let indexPath = IndexPath(item: item, section: section)
            
            //Different offset based on what column the item is
            for column in 0..<numberOfColumns {
                switch column {
                case 0:
                    yOffset.append(2 * cellPadding)
                case 1:
                    yOffset.append(2 * cellPadding)
                case 2:
                    
                    yOffset.append(cellPadding   cellHeight)
                default:
                    break
                }
            }
            
            //Measuring the frame
            let height = cellPadding * 2   cellHeight
            let frame = CGRect(x: xOffset[column], y: yOffset[column], width: columnWidth, height: columnWidth)
            let insetFrame = frame.insetBy(dx: cellPadding, dy: cellPadding)
            
            //Creating attributres for the layout and caching them
            let attributes = UICollectionViewLayoutAttributes(forCellWith: indexPath)
            attributes.frame = insetFrame
            cache.append(attributes)
            print ("frame.maxY", frame.maxY)
            
            //We increase the max height of the content as we get more items
            contentHeight = max(collectionView.frame.height   10, frame.maxY)
            
            
            //We increase the yOffset, too
            yOffset[column] = yOffset[column]   2 * (height - cellPadding)
            
            // print ("column: (column), yOffset: (yOffset[column])")
            
            let numberOfItems = delegate?.theNumberOfItemsInCollectionView()
            
            //Changing column so the next item will be added to a different column
            
            if let numberOfItems = numberOfItems, indexPath.item == numberOfItems - 1
            {
                //In case we get to the last cell, we check the column of the cell before
                //The last one, and based on that, we change the column
                print ("indexPath.item: (indexPath.item), numberOfItems: (numberOfItems)")
                print ("A")
                switch column {
                case 0:
                    column = 2
                case 2:
                    column = 0
                case 1:
                    column = 2
                default:
                    return
                }
            } else  {
                print ("B")
                column = column < (numberOfColumns - 1) ? (column   1) : 0
            }
        }
    }
        
    }
    
    
    //Is called  to determine which items are visible in the given rect
    override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        var visibleLayoutAttributes = [UICollectionViewLayoutAttributes]()
        
        //Loop through the cache and look for items in the rect
        for attribute in cache {
            if attribute.frame.intersects(rect) {
                visibleLayoutAttributes.append(attribute)
            }
        }
        
        return visibleLayoutAttributes
    }
    
    //The attributes for the item at the indexPath
    override func layoutAttributesForItem(at indexPath: IndexPath) -> UICollectionViewLayoutAttributes? {
        return cache[indexPath.item]
    }
    
    fileprivate let collectionViewHeaderFooterReuseIdentifier = "MyHeaderFooterClass"
    
    func collectionView(collectionView: UICollectionView, viewForSupplementaryElementOfKind kind: String, atIndexPath indexPath: NSIndexPath) -> UICollectionReusableView {
            print("ip = (indexPath.item)")
            var supplementaryView = collectionView.dequeueReusableSupplementaryView(ofKind: kind, withReuseIdentifier: collectionViewHeaderFooterReuseIdentifier, for: indexPath as IndexPath)
            supplementaryView.backgroundColor = UIColor.blue
            return supplementaryView
        }

    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, referenceSizeForHeaderInSection section: Int) -> CGSize {
            return CGSize(width: collectionView.frame.width, height: 180.0)
    }
    }