Signal-iOS/SignalMessaging/ViewControllers/Stickers/LinearHorizontalLayout.swift
2019-05-03 09:24:12 -04:00

101 lines
3.1 KiB
Swift

//
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
//
import UIKit
// A trivial layout that places each item in a horizontal line.
// Each item has uniform size.
class LinearHorizontalLayout: UICollectionViewLayout {
private let itemSize: CGSize
private let spacing: CGFloat
private var itemAttributesMap = [UICollectionViewLayoutAttributes]()
private var contentSize = CGSize.zero
// MARK: Initializers and Factory Methods
@available(*, unavailable, message:"use other constructor instead.")
required init?(coder aDecoder: NSCoder) {
notImplemented()
}
required init(itemSize: CGSize, spacing: CGFloat = 0) {
self.itemSize = itemSize
self.spacing = spacing
super.init()
}
// MARK: Methods
override func invalidateLayout() {
super.invalidateLayout()
itemAttributesMap.removeAll()
}
override func invalidateLayout(with context: UICollectionViewLayoutInvalidationContext) {
super.invalidateLayout(with: context)
itemAttributesMap.removeAll()
}
override func prepare() {
super.prepare()
guard let collectionView = collectionView else {
return
}
guard collectionView.numberOfSections == 1 else {
owsFailDebug("This layout only support a single section.")
return
}
let itemCount = collectionView.numberOfItems(inSection: 0)
let insets = collectionView.contentInset
guard itemCount > 0 else {
contentSize = .zero
return
}
for row in 0..<itemCount {
// TODO: We should ultimately make this layout RTL.
let itemX: CGFloat = insets.left + CGFloat(row) * (itemSize.width + spacing)
let itemY: CGFloat = insets.top
let itemFrame = CGRect(x: itemX, y: itemY, width: itemSize.width, height: itemSize.height)
let indexPath = NSIndexPath(row: row, section: 0)
let itemAttributes = UICollectionViewLayoutAttributes(forCellWith: indexPath as IndexPath)
itemAttributes.frame = itemFrame
itemAttributesMap.append(itemAttributes)
}
contentSize = CGSize(width: insets.left + insets.right + CGFloat(itemCount) * itemSize.width + CGFloat(itemCount - 1) * spacing,
height: insets.top + insets.bottom + itemSize.height)
}
override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
return itemAttributesMap.filter { itemAttributes in
return itemAttributes.frame.intersects(rect)
}
}
override func layoutAttributesForItem(at indexPath: IndexPath) -> UICollectionViewLayoutAttributes? {
return itemAttributesMap[safe: indexPath.row]
}
override var collectionViewContentSize: CGSize {
return contentSize
}
override func shouldInvalidateLayout(forBoundsChange newBounds: CGRect) -> Bool {
guard let collectionView = collectionView else {
return false
}
return collectionView.width() != newBounds.size.width
}
}