如何在Swift中从自定义文本生成UIImage

import UIKit

func generateImage(from text: String, with font: UIFont, color: UIColor, size: CGSize) -> UIImage? {
    // 创建画布
    UIGraphicsBeginImageContextWithOptions(size, false, 0.0)
    
    // 获取图形上下文
    guard let context = UIGraphicsGetCurrentContext() else { return nil }
    
    // 设置文本属性
    let attributes: [NSAttributedString.Key: Any] = [
        .font: font,
        .foregroundColor: color,
        .paragraphStyle: NSMutableParagraphStyle().apply { $0.alignment = .center }
    ]
    
    // 绘制文本
    let textRect = CGRect(x: 0, y: 0, width: size.width, height: size.height)
    text.draw(in: textRect, withAttributes: attributes)
    
    // 生成UIImage
    let image = UIGraphicsGetImageFromCurrentImageContext()
    
    // 结束图形上下文
    UIGraphicsEndImageContext()
    
    return image
}

// 使用示例
let text = "Hello, World!"
let font = UIFont.systemFont(ofSize: 30)
let color = UIColor.red
let size = CGSize(width: 200, height: 100)

if let image = generateImage(from: text, with: font, color: color, size: size) {
    // 使用生成的UIImage
}