Swift缩放并填充图片功能的实现

摘要

直接操作图片来实现它的缩放或者填充多余空间，首选 UIGraphicsBeginImageContext 函数来实现，它就相当于一个画布，你甚至可以用它来涂鸦。

最近有一个需求，就是将图片先等比例缩放到指定大小，然后将空余出来空间填充为黑色，返回指定大小的图片。

这种直接操作图片的需求，就要考虑使用 UIGraphicsBeginImageContext 函数实现。它可以理解为一个画布，我们只需要把图片放在画布的对应位置，把画布的多余地方全部涂成黑色就完成。

实现

先看代码，然后再分析：

func rescaleAndPading(_ image: UIImage, targetSize: CGSize) -> UIImage? {
      let max = max(image.width, image.height)
      let ratio = Float(targetSize.width) / Float(max)
      
      let (newWidth, newHeight) = (

          Int(Float(image.width) * ratio),
          Int(Float(image.height) * ratio)
      )
      
      let (tarWidth, tarHeight) = (
          
          Int(targetSize.width),
          Int(targetSize.height)
      )
      
      let deltaW = tarWidth - newWidth
      let deltaH = tarHeight - newHeight
      
      let (y, x) = (
          
          deltaH / 2,
          deltaW / 2
      )
      
      // 创建绘图上下文环境
      UIGraphicsBeginImageContext(targetSize)
      let context = UIGraphicsGetCurrentContext()
      // 黄色背景
      context?.setFillColor(UIColor.yellow.cgColor)
      context?.fill(CGRect(x: 0, y: 0, width: tarWidth, height: tarHeight))
      image.draw(in: CGRect(x: x, y: y, width: newWidth, height: newHeight))
      // 获取上下文里的内容，将视图写入到新的图像对象
      let newImage = UIGraphicsGetImageFromCurrentImageContext()
      UIGraphicsEndImageContext()
      return newImage
  }

看代码，总结出逻辑很简单，就是首先根据目标的 size 来计算出需要缩放的比例（按照最大边来处理），计算出图片在画布中的对应位置和缩放后的宽高。

最后就是重头戏，调用 UIGraphicsBeginImageContext 来绘画。这里要留意几个参数的设置：