长图PDF转Image性能优化探索之路

2023-07-25 17:24:44

长图 PDF 转 Image 性能优化

前言

在 iOS 开发中，经常需要将 PDF 文件转换为 Image 对象以用于应用程序的显示和处理。但对于长图 PDF（即高度远大于宽度），直接转换整个 PDF 可能导致严重的性能问题，包括内存消耗高和转换速度慢。

问题分析

当处理长图 PDF 时，主要消耗内存的步骤是在将 PDF 转换为 CGContextRef 对象后，再使用 CGContextDrawPDFPage 函数将 PDF 页面绘制到该对象中。由于长图 PDF 的高度远大于宽度，绘制过程需要更大的 context 大小，从而导致内存消耗激增。

优化方案

为了解决此问题，我们提出了以下优化方案：

调整 Context 的填充大小

我们可以调整 context 的填充大小来仅绘制 PDF 页面的一部分内容，从而降低内存消耗。根据屏幕大小确定 context 填充大小：小屏幕使用较小填充大小，大屏幕使用较大填充大小。

根据屏幕大小生成不同数量的 Image 对象

对于长图 PDF，我们可以根据屏幕大小生成不同数量的 Image 对象。小屏幕生成较少对象，每个对象包含部分 PDF 页面内容。大屏幕生成较多对象，每个对象包含更多内容。

优化效果

经过优化，长图 PDF 转换的内存消耗从数百兆字节降低到几十兆字节，转换速度也显著提升。测试表明，对于高度超过 10,000 像素的长图 PDF，优化后内存消耗下降了约 90%。

代码示例

以下代码示例演示了如何在调整 context 填充大小和生成不同数量的 Image 对象后进行优化：

// 根据屏幕大小调整 Context 填充大小
CGSize contextSize;
if (UIScreen.main.bounds.height < 1000) {
    contextSize = CGSize(width: UIScreen.main.bounds.width, height: 500)
} else {
    contextSize = CGSize(width: UIScreen.main.bounds.width, height: 1000)
}

// 根据屏幕大小生成不同数量的 Image 对象
var images: [UIImage] = []
let pageCount = document.pageCount
for page in 0..<pageCount {
    if pageCount < 10 {
        images.append(generateImage(forPage: page, contextSize: contextSize))
    } else if pageCount < 50 {
        images.append(generateImage(forPage: page, contextSize: contextSize))
        images.append(generateImage(forPage: page + 1, contextSize: contextSize))
    } else {
        images.append(generateImage(forPage: page, contextSize: contextSize))
        images.append(generateImage(forPage: page + 1, contextSize: contextSize))
        images.append(generateImage(forPage: page + 2, contextSize: contextSize))
    }
}