Swift中在透明画布上放置NSImage的实现方法问询

我来帮你完善这个NSImage扩展，实现将原图居中放置到指定尺寸的透明画布上的功能。你之前的思路是对的，但缺少了创建新图像并在其绘制上下文中执行绘制的关键步骤——macOS里操作NSImage需要先锁定它的绘制焦点，才能在上面绘制内容。

下面是完整的实现代码，附带详细注释：

extension NSImage {
    func squareIt(canvasSize: CGSize) -> NSImage? {
        // 先检查原图尺寸是否有效，避免无效输入
        guard size.width > 0, size.height > 0 else { return nil }
        
        // 创建指定尺寸的新NSImage，设置为支持alpha通道（透明背景）
        let newImage = NSImage(size: canvasSize, flipped: false) { rect in
            // 填充透明背景，确保画布初始是透明的
            NSColor.clear.setFill()
            rect.fill()
            
            // 计算原图居中的位置
            let originX = max(0, (canvasSize.width - self.size.width) / 2)
            let originY = max(0, (canvasSize.height - self.size.height) / 2)
            let centeredRect = CGRect(
                x: originX,
                y: originY,
                width: self.size.width,
                height: self.size.height
            )
            
            // 在新画布上绘制原图
            self.draw(in: centeredRect)
            return true
        }
        
        // 确保新图像的表示是正确的（可选，但能避免一些显示/保存问题）
        newImage.addRepresentation(NSBitmapImageRep(
            bitmapDataPlanes: nil,
            pixelsWide: Int(canvasSize.width),
            pixelsHigh: Int(canvasSize.height),
            bitsPerSample: 8,
            samplesPerPixel: 4,
            hasAlpha: true,
            isPlanar: false,
            colorSpaceName: .calibratedRGB,
            bytesPerRow: 0,
            bitsPerPixel: 0
        )!)
        
        return newImage
    }
}

关键步骤说明： #

• 创建带alpha的新图像：使用NSImage(size:flipped:handler:)初始化方法，这个闭包会自动处理绘制上下文的锁定/解锁，比手动调用lockFocus()更简洁安全。
• 填充透明背景：默认情况下新图像的背景可能是白色，所以先填充NSColor.clear来确保透明效果。
• 居中计算：用max(0, ...)处理画布尺寸小于原图的情况（此时原图会从左上角开始绘制，不会被裁剪），如果你的场景只需要画布比原图大，可以去掉max判断。
• 添加位图表示：这一步是为了确保新图像在后续保存或展示时，能正确保留透明通道，避免出现背景变黑的异常问题。

使用示例： #

假设你有一张200×300的NSImage，想要生成300×300的带透明边的图像：

if let originalImage = NSImage(named: "YourImage"),
   let squaredImage = originalImage.squareIt(canvasSize: CGSize(width: 300, height: 300)) {
    // 在这里使用生成的squaredImage
}

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