UIImage+Extension

UIImage+Extension.swift

extension UIImage {
    
    public func cropped(to rect: CGRect) -> UIImage? {
        UIGraphicsBeginImageContextWithOptions(rect.size, true, scale)
        draw(at: rect.origin.applying(CGAffineTransform(scaleX: -1, y: -1)))
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image
    }
    
    public func squareCropped() -> UIImage? {
        let rect = size.width > size.height
            ? CGRect(x: (size.width - size.height) * 0.5, y: 0, width: size.height, height: size.height)
            : CGRect(x: 0, y: (size.height - size.width) * 0.5, width: size.width, height: size.width)
        return cropped(to: rect)
    }
    
    public func resized(to size: CGSize) -> UIImage? {
        UIGraphicsBeginImageContextWithOptions(size, true, scale)
        draw(in: CGRect(origin: .zero, size: size))
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image
    }
    
}