UIImage Extensions

ImageExtensions.swift

extension UIImage {

  func resizeImage(_ image: UIImage, newWidth: CGFloat) -> UIImage? {
    let scale = newWidth / image.size.width
    let newHeight = image.size.height * scale
    UIGraphicsBeginImageContext(CGSize(width: newWidth, height: newHeight))
    image.draw(in: CGRect(x: 0, y: 0, width: newWidth, height: newHeight))
    let newImage = UIGraphicsGetImageFromCurrentImageContext()
    guard let image = newImage else { return nil }
    UIGraphicsEndImageContext()
    return image
  }

  func drawFromView(view: UIView) -> UIImage? {
    UIGraphicsBeginImageContext(view.frame.size)
    view.layer.render(in: UIGraphicsGetCurrentContext()!)
    let newImage = UIGraphicsGetImageFromCurrentImageContext()
    guard let image = newImage else { return nil }
    UIGraphicsEndImageContext()
    return image
  }
}