我无法快速从 CGContext 获取图像，有人可以帮助我吗？

Question

我可以在 UIView 中绘画，但无法从上下文中获取图像。 我想获取图像并将其设置到其他 ViewController

var image = UIImage()
        
override func draw(_ rect: CGRect) {
    super.draw(rect)
    UIGraphicsBeginImageContext(image.size)
    guard let context = UIGraphicsGetCurrentContext() else { return }
    context.setLineCap(.round)
    context.setLineJoin(.round)

    lines.forEach { line in
        for (i, p) in line.points.enumerated() {
            if i == 0 {
                context.move(to: p)
            } else {
                context.addLine(to: p)
            }
        }
        context.setStrokeColor(line.color)
        context.setLineWidth(line.width)
        context.strokePath()
    }
    guard let img = UIGraphicsGetImageFromCurrentImageContext() else { return }
    ViewModel.shared.image = img
    UIGraphicsEndImageContext()
    
}