如何在PDFKit的PDFDocument与Core Graphics的CGPDFContext间迁移数据？

嘿，我明白你现在卡在PDFDocument和CGPDFContext的迁移上了——确实这俩不是直接免费桥接的，尤其是在PyObjC环境下得绕个小弯儿。我给你捋清楚步骤，附上Python代码示例：

首先得明确：你之前用CGPDFContextCreate的思路错啦——这个API是用来创建新的PDF输出上下文（比如用来生成新PDF）的，不是用来从现有PDF数据加载文档的。要把PDFDocument的内容转到Core Graphics的生态里，得先把它转成CGPDFDocument对象，再结合CGPDFContext来处理。

下面是具体的实现步骤和代码：

1. 导入必要的框架 #

在PyObjC里，我们需要用到Quartz（包含Core Graphics的PDF相关API）和Cocoa的PDFDocument：

import Quartz
from Cocoa import PDFDocument, NSURL

2. 加载PDFDocument并转换为CGPDFDocument #

先加载你的PDF文件到PDFDocument，然后通过它的dataRepresentation()拿到NSData，再转成Core Graphics能识别的数据提供者，最终创建CGPDFDocument：

# 加载PDF到PDFDocument
pdf_url = NSURL.fileURLWithPath_("/path/to/your/pdf/file.pdf")
pdf_doc = PDFDocument.alloc().initWithURL_(pdf_url)

# 从PDFDocument获取NSData，转成CGDataProvider
pdf_data = pdf_doc.dataRepresentation()
data_provider = Quartz.CGDataProviderCreateWithCFData(pdf_data)

# 创建CGPDFDocument对象
cg_pdf_doc = Quartz.CGPDFDocumentCreateWithProvider(data_provider)

3. 使用CGPDFContext处理内容（比如生成新PDF） #

如果你的需求是基于现有PDF内容，用CGPDFContext生成新的PDF文件，那可以遍历CGPDFDocument的页面，绘制到新的CGPDFContext中：

# 定义输出PDF的路径
output_path = "/path/to/your/output.pdf"
output_url = NSURL.fileURLWithPath_(output_path)

# 创建CGPDFContext（输出上下文）
media_box = Quartz.CGRectMake(0, 0, 612, 792)  # 标准A4尺寸
context = Quartz.CGPDFContextCreateWithURL(output_url, media_box, None)

# 获取PDF总页数
page_count = Quartz.CGPDFDocumentGetNumberOfPages(cg_pdf_doc)

# 遍历每一页，绘制到新上下文
for page_num in range(1, page_count + 1):
    cg_page = Quartz.CGPDFDocumentGetPage(cg_pdf_doc, page_num)
    if cg_page is None:
        continue
    
    # 获取当前页面的媒体框
    page_box = Quartz.CGPDFPageGetBoxRect(cg_page, Quartz.kCGPDFMediaBox)
    
    # 开始绘制页面
    Quartz.CGPDFContextBeginPage(context, {"MediaBox": page_box})
    Quartz.CGContextDrawPDFPage(context, cg_page)
    Quartz.CGPDFContextEndPage(context)

# 结束上下文，写入文件
Quartz.CGPDFContextClose(context)

关键说明 #

• CGPDFDocumentCreateWithProvider才是从现有PDF数据创建CGPDFDocument的正确方式，而不是CGPDFContextCreate
• 如果你只是需要读取PDF内容（而不是生成新PDF），拿到cg_pdf_doc之后就可以直接用Core Graphics的API操作页面内容了
• 在PyObjC里通常不需要手动管理CG对象的内存，Python的垃圾回收会自动处理，但大量操作时可以留意内存占用情况

内容的提问来源于stack exchange，提问作者benwiggy