|
9 | 9 | import Foundation |
10 | 10 | import UIKit |
11 | 11 |
|
| 12 | +public extension UIImage { |
| 13 | + func resizedImage(size: CGSize) -> UIImage { |
| 14 | + if self.size.width < size.width { |
| 15 | + return self |
| 16 | + } |
| 17 | + |
| 18 | + let renderer = UIGraphicsImageRenderer(size: size) |
| 19 | + return renderer.image { (context) in |
| 20 | + self.draw(in: CGRect(origin: .zero, size: size)) |
| 21 | + } |
| 22 | + } |
| 23 | + |
| 24 | + func resizedImage(width: CGFloat) -> UIImage { |
| 25 | + let h = (self.size.height / self.size.width) * width |
| 26 | + return self.resizedImage(size: CGSize(width: width, height: h)) |
| 27 | + } |
| 28 | + |
| 29 | + func fixedOrientation() -> UIImage? { |
| 30 | + guard imageOrientation != UIImage.Orientation.up else { |
| 31 | + return self.copy() as? UIImage |
| 32 | + } |
| 33 | + |
| 34 | + guard let cgImage = self.cgImage else { |
| 35 | + return nil |
| 36 | + } |
| 37 | + |
| 38 | + guard let colorSpace = cgImage.colorSpace, let ctx = CGContext(data: nil, width: Int(size.width), height: Int(size.height), bitsPerComponent: cgImage.bitsPerComponent, bytesPerRow: 0, space: colorSpace, bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue) else { |
| 39 | + return nil //Not able to create CGContext |
| 40 | + } |
| 41 | + |
| 42 | + var transform: CGAffineTransform = CGAffineTransform.identity |
| 43 | + |
| 44 | + switch imageOrientation { |
| 45 | + case .down, .downMirrored: |
| 46 | + transform = transform.translatedBy(x: size.width, y: size.height) |
| 47 | + transform = transform.rotated(by: CGFloat.pi) |
| 48 | + break |
| 49 | + case .left, .leftMirrored: |
| 50 | + transform = transform.translatedBy(x: size.width, y: 0) |
| 51 | + transform = transform.rotated(by: CGFloat.pi / 2.0) |
| 52 | + break |
| 53 | + case .right, .rightMirrored: |
| 54 | + transform = transform.translatedBy(x: 0, y: size.height) |
| 55 | + transform = transform.rotated(by: CGFloat.pi / -2.0) |
| 56 | + break |
| 57 | + case .up, .upMirrored: |
| 58 | + break |
| 59 | + @unknown default: |
| 60 | + print("") |
| 61 | + } |
| 62 | + //Flip image one more time if needed to, this is to prevent flipped image |
| 63 | + switch imageOrientation { |
| 64 | + case .upMirrored, .downMirrored: |
| 65 | + transform = transform.translatedBy(x: size.width, y: 0) |
| 66 | + transform = transform.scaledBy(x: -1, y: 1) |
| 67 | + break |
| 68 | + case .leftMirrored, .rightMirrored: |
| 69 | + transform = transform.translatedBy(x: size.height, y: 0) |
| 70 | + transform = transform.scaledBy(x: -1, y: 1) |
| 71 | + case .up, .down, .left, .right: |
| 72 | + break |
| 73 | + @unknown default: |
| 74 | + print("") |
| 75 | + } |
| 76 | + ctx.concatenate(transform) |
| 77 | + switch imageOrientation { |
| 78 | + case .left, .leftMirrored, .right, .rightMirrored: |
| 79 | + ctx.draw(self.cgImage!, in: CGRect(x: 0, y: 0, width: size.height, height: size.width)) |
| 80 | + default: |
| 81 | + ctx.draw(self.cgImage!, in: CGRect(x: 0, y: 0, width: size.width, height: size.height)) |
| 82 | + break |
| 83 | + } |
| 84 | + guard let newCGImage = ctx.makeImage() else { return nil } |
| 85 | + return UIImage.init(cgImage: newCGImage, scale: 1, orientation: .up) |
| 86 | + } |
| 87 | +} |
| 88 | + |
| 89 | +public extension UIImage { |
| 90 | + static func withColor(_ color: UIColor, size: CGSize = CGSize(width: 1, height: 1)) -> UIImage { |
| 91 | + let format = UIGraphicsImageRendererFormat() |
| 92 | + format.scale = 1 |
| 93 | + let image = UIGraphicsImageRenderer(size: size, format: format).image { rendererContext in |
| 94 | + color.setFill() |
| 95 | + rendererContext.fill(CGRect(origin: .zero, size: size)) |
| 96 | + } |
| 97 | + return image |
| 98 | + } |
| 99 | +} |
| 100 | + |
| 101 | + |
12 | 102 | public extension UIImage { |
13 | 103 | /// 根据文字生成包含第一个文字的图片 |
14 | 104 | /// - Parameters: |
|
0 commit comments