UICollectionViewLayout：仅将DecorationView添加到特定单元格

Question

我开发了一个自定义的CollectionViewLayout，它使用DecorationView来显示单元格后面的阴影。

但是，我想将此装饰仅添加到某些单元格中。 UICollectionView为vertical，但可能在单元格中包含嵌入的horizontal UICollectionView。嵌入UICollectionView的单元格不应进行修饰，如图所示：

以下是我用来添加阴影的代码。 UICollectionViewLayout没有提供如何检索单元格类的方法，因此可以决定是否添加阴影：

  override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
    let parent = super.layoutAttributesForElements(in: rect)
    guard let attributes = parent, !attributes.isEmpty else {
      return parent
    }

    let sections = attributes.map{$0.indexPath.section}
    let unique = Array(Set(sections))


    // Need to detect, which sections contain an embedded UICollectionView and exclude them from the UNIQUE set


    let backgroundShadowAttributes: [UICollectionViewLayoutAttributes] = unique.compactMap{ section in
      let indexPath = IndexPath(item: 0, section: section)
      return self.layoutAttributesForDecorationView(ofKind: backgroundViewClass.reuseIdentifier(),
                                                    at: indexPath)
    }

    return attributes + backgroundShadowAttributes + separators
  }

有没有办法有条件地指定哪些视图应该装饰？

Answer 1

完成此代码： 一种直接询问DataSource的协议，是否为特定部分显示阴影：

protocol SectionBackgroundFlowLayoutDataSource {
  func shouldDisplayBackgroundFor(section: Int) -> Bool
}

利用func layoutAttributesForElements(in rect: CGRect)方法中的协议：

  override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
    let parent = super.layoutAttributesForElements(in: rect)
    guard let attributes = parent, !attributes.isEmpty else {
      return parent
    }

    attributes.forEach(configureRoundCornersAttributes)

    // Display shadows for every section by default
    var sectionsWithShadow = Set(attributes.map{$0.indexPath.section})
    if let dataSource = collectionView?.dataSource as? SectionBackgroundFlowLayoutDataSource {
    // Ask DataSource for sections with shadows, if it supports the protocol
      sectionsWithShadow = sectionsWithShadow.filter{dataSource.shouldDisplayBackgroundFor(section: $0)}
    }

    let backgroundShadowAttributes: [UICollectionViewLayoutAttributes] = sectionsWithShadow.compactMap{ section in
      let indexPath = IndexPath(item: 0, section: section)
      return self.layoutAttributesForDecorationView(ofKind: backgroundViewClass.reuseIdentifier(),
                                                    at: indexPath)
    }

    return attributes + backgroundShadowAttributes + separators
  }

func shouldDisplayBackgroundFor(section: Int) -> Bool可能会比cellForItemAtIndexPath更快地返回，因为它不需要完整的单元格配置。