ReadViewSDK/Sources/RDReaderView/ReaderView/RDReaderView+CollectionView.swift

import UIKit

extension RDReaderView: UICollectionViewDataSource, RDReaderFlowLayoutDelegate, RDReaderFlowLayoutDataSoure {

    public func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {

        if let identifer = pageReuseIdentifier(for: indexPath.row) {
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: identifer, for: IndexPath(item: indexPath.row, section: 0)) as! RDReaderContentCell
            let preloadedView = preloadController.takePreloadedView(for: indexPath.row)
            let reusableView = cell.containerView ?? preloadedView
            cell.containerView = contentViewForPage(indexPath.row, reusableView: reusableView)

            if pageDirection == .rightToLeft && currentDisplayType != .verticalScroll {
                cell.contentView.transform = CGAffineTransform(scaleX: -1, y: 1)
            } else {
                cell.contentView.transform = .identity
            }
            return cell
        }

        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: NSStringFromClass(UICollectionViewCell.self), for: indexPath)

        return cell
    }

    public func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return numberOfPages()
    }

    public func pageNum(flowLayout: RDReaderFlowLayout, pageIndex: Int) {
        if currentPage >= 0, currentPage != pageIndex {
            predictedPageDirection = pageIndex >= currentPage
        }
        currentPage = pageIndex
        primePageCache(around: pageIndex, preferredForward: predictedPageDirection)
    }

    public func heigtOfVerticalScrollPage(flowLayout: RDReaderFlowLayout, pageIndex: Int) -> CGFloat? {
        return nil
    }
}