使 table-layout 支持区头/区尾吸附效果 ```ts enum _yx_table_layout_supplementary_kinds { HEADER = 'header', FOOTER = 'footer', } export class YXTableLayout extends YXLayout { /** * 行高 */ rowHeight: number | ((indexPath: YXIndexPath) => number) = 100 /** * 内容上边距 */ top: number = 0 /** * 内容下边距 */ bottom: number = 0 /** * 节点之间间距 */ spacing: number = 0 /** * 区头高度 */ sectionHeaderHeight: number | ((section: number) => number) = null /** * 区尾高度 */ sectionFooterHeight: number | ((section: number) => number) = null /** * 钉住 header 的位置 ( header 吸附在列表可见范围内 ) */ sectionHeadersPinToVisibleBounds: boolean = false /** * 钉住 footer 的位置 ( footer 吸附在列表可见范围内 ) */ sectionFootersPinToVisibleBounds: boolean = false /** * 区头/区尾标识 */ static SupplementaryKinds = _yx_table_layout_supplementary_kinds protected originalHeaderRect: Map = new Map() // 保存所有 header 的原始位置 protected originalFooterRect: Map = new Map() // 保存所有 footer 的原始位置 prepare(collectionView: YXCollectionView): void { // 设置列表的滚动方向(这套布局固定为垂直方向滚动) collectionView.scrollView.horizontal = false collectionView.scrollView.vertical = true if (collectionView.scrollDirection === YXCollectionView.ScrollDirection.HORIZONTAL) { // 由于这套布局规则只支持垂直方向布局,当外部配置了水平方向滚动时这里可以给个警告 warn(`YXTableLayout 仅支持垂直方向排列`) } // 清空一下布局属性数组 this.attributes = [] this.originalHeaderRect.clear() this.originalFooterRect.clear() // 获取列表宽度 const contentWidth = collectionView.node.getComponent(UITransform).width // 声明一个临时变量,用来记录当前所有内容的总高度 let contentHeight = 0 // 获取列表一共分多少个区 let numberOfSections = collectionView.getNumberOfSections() // 为每条数据对应的生成一个布局属性 for (let section = 0; section < numberOfSections; section++) { // 创建一个区索引 let sectionIndexPath = new YXIndexPath(section, 0) // 通过区索引创建一个区头节点布局属性 let sectionHeaderHeight = 0 if (this.sectionHeaderHeight) { sectionHeaderHeight = this.sectionHeaderHeight instanceof Function ? this.sectionHeaderHeight(section) : this.sectionHeaderHeight } if (sectionHeaderHeight > 0) { let headerAttr = YXLayoutAttributes.layoutAttributesForSupplementary(sectionIndexPath, YXTableLayout.SupplementaryKinds.HEADER) // 确定这个节点的位置 headerAttr.frame.x = 0 headerAttr.frame.width = contentWidth headerAttr.frame.height = sectionHeaderHeight headerAttr.frame.y = contentHeight // 调整层级 headerAttr.zIndex = 1 // 重要: 保存布局属性 this.attributes.push(headerAttr) this.originalHeaderRect.set(section, headerAttr.frame.clone()) // 更新整体内容高度 contentHeight = headerAttr.frame.yMax } // 将 top 配置应用到每个区 contentHeight = contentHeight + this.top // 获取这个区内的内容数量,注意这里传入的是 section let numberOfItems = collectionView.getNumberOfItems(section) for (let item = 0; item < numberOfItems; item++) { // 创建索引,注意这里的 section 已经改为正确的 section 了 let indexPath = new YXIndexPath(section, item) // 通过索引创建一个 cell 节点的布局属性 let attr = YXLayoutAttributes.layoutAttributesForCell(indexPath) // 通过索引获取这个节点的高度 let rowHeight = this.rowHeight instanceof Function ? this.rowHeight(indexPath) : this.rowHeight // 确定这个节点的位置 attr.frame.x = 0 attr.frame.width = contentWidth attr.frame.height = rowHeight attr.frame.y = contentHeight + (item > 0 ? this.spacing : 0) // 重要: 保存布局属性 this.attributes.push(attr) // 更新当前内容高度 contentHeight = attr.frame.yMax } // 高度补一个底部间距,跟 top 一样,也是应用到每个区 contentHeight = contentHeight + this.bottom // 通过区索引创建一个区尾节点布局属性 let sectionFooterHeight = 0 if (this.sectionFooterHeight) { sectionFooterHeight = this.sectionFooterHeight instanceof Function ? this.sectionFooterHeight(section) : this.sectionFooterHeight } if (sectionFooterHeight > 0) { let footerAttr = YXLayoutAttributes.layoutAttributesForSupplementary(sectionIndexPath, YXTableLayout.SupplementaryKinds.FOOTER) // 确定这个节点的位置 footerAttr.frame.x = 0 footerAttr.frame.width = contentWidth footerAttr.frame.height = sectionFooterHeight footerAttr.frame.y = contentHeight // 调整层级 footerAttr.zIndex = 1 // 重要: 保存布局属性 this.attributes.push(footerAttr) this.originalFooterRect.set(section, footerAttr.frame.clone()) // 更新整体内容高度 contentHeight = footerAttr.frame.yMax } } // 重要: 设置内容区域总大小,只有确定了滚动区域的大小列表才能滚动 this.contentSize = new math.Size(contentWidth, contentHeight) } initOffset(collectionView: YXCollectionView): void { // 列表首次刷新时,调整一下列表的偏移位置 collectionView.scrollView.scrollToTop() } layoutAttributesForElementsInRect(rect: math.Rect, collectionView: YXCollectionView): YXLayoutAttributes[] { let numberOfSections = collectionView.getNumberOfSections() let scrollOffset = collectionView.scrollView.getScrollOffset() for (let index = 0; index < this.attributes.length; index++) { const element = this.attributes[index]; if (element.elementCategory === 'Supplementary') { if (this.sectionHeadersPinToVisibleBounds && element.supplementaryKinds === YXTableLayout.SupplementaryKinds.HEADER) { const originalFrame = this.originalHeaderRect.get(element.indexPath.section) element.frame.y = originalFrame.y if (scrollOffset.y > originalFrame.y) { element.frame.y = scrollOffset.y } const nextOriginalFrame = this.getNextOriginalFrame(element.indexPath.section, YXTableLayout.SupplementaryKinds.FOOTER, numberOfSections) if (nextOriginalFrame) { if (element.frame.yMax > nextOriginalFrame.y) { element.frame.y = nextOriginalFrame.y - element.frame.height } } } if (this.sectionFootersPinToVisibleBounds && element.supplementaryKinds === YXTableLayout.SupplementaryKinds.FOOTER) { let bottom = scrollOffset.y + collectionView.scrollView.view.height const originalFrame = this.originalFooterRect.get(element.indexPath.section) const previousOriginalFrame = this.getPreviousOriginalFrame(element.indexPath.section, YXTableLayout.SupplementaryKinds.HEADER) element.frame.y = originalFrame.y if (bottom < originalFrame.yMax) { element.frame.y = bottom - element.frame.height if (previousOriginalFrame) { if (element.frame.y < previousOriginalFrame.yMax) { element.frame.y = previousOriginalFrame.yMax } } } } } } return this.attributes } shouldUpdateAttributesZIndex(): boolean { return this.sectionHeadersPinToVisibleBounds || this.sectionFootersPinToVisibleBounds } shouldUpdateAttributesForBoundsChange(): boolean { return this.sectionHeadersPinToVisibleBounds || this.sectionFootersPinToVisibleBounds } /** * 获取 `section` 下一个 header 或者 footer 的位置 */ protected getNextOriginalFrame(section: number, kinds: _yx_table_layout_supplementary_kinds, total: number) { if (section >= total) { return null } if (kinds === YXTableLayout.SupplementaryKinds.HEADER) { let result = this.originalHeaderRect.get(section) if (result) { return result } return this.getNextOriginalFrame(section, YXTableLayout.SupplementaryKinds.FOOTER, total) } if (kinds === YXTableLayout.SupplementaryKinds.FOOTER) { let result = this.originalFooterRect.get(section) if (result) { return result } return this.getNextOriginalFrame(section + 1, YXTableLayout.SupplementaryKinds.HEADER, total) } return null } /** * 获取 `section` 前一个 header 或者 footer 的位置 */ protected getPreviousOriginalFrame(section: number, kinds: _yx_table_layout_supplementary_kinds) { if (section < 0) { return null } if (kinds === YXTableLayout.SupplementaryKinds.HEADER) { let result = this.originalHeaderRect.get(section) if (result) { return result } return this.getPreviousOriginalFrame(section - 1, YXTableLayout.SupplementaryKinds.FOOTER) } if (kinds === YXTableLayout.SupplementaryKinds.FOOTER) { let result = this.originalFooterRect.get(section) if (result) { return result } return this.getPreviousOriginalFrame(section, YXTableLayout.SupplementaryKinds.HEADER) } return null } } ```