33 lines
859 B
Swift
33 lines
859 B
Swift
|
//Copyright © 2019 Vincode, Inc. All rights reserved.
|
||
|
|
|
||
|
|
import UIKit
|
||
|
|
|
||
|
|
extension UIImage {
|
||
|
|
|
||
|
|
func maskWithColor(color: UIColor) -> UIImage? {
|
||
|
|
|
||
|
|
let maskImage = cgImage!
|
||
|
|
|
||
|
|
let width = size.width
|
||
|
|
let height = size.height
|
||
|
|
let bounds = CGRect(x: 0, y: 0, width: width, height: height)
|
||
|
|
|
||
|
|
let colorSpace = CGColorSpaceCreateDeviceRGB()
|
||
|
|
let bitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedLast.rawValue)
|
||
|
|
let context = CGContext(data: nil, width: Int(width), height: Int(height), bitsPerComponent: 8, bytesPerRow: 0, space: colorSpace, bitmapInfo: bitmapInfo.rawValue)!
|
||
|
|
|
||
|
|
context.clip(to: bounds, mask: maskImage)
|
||
|
|
context.setFillColor(color.cgColor)
|
||
|
|
context.fill(bounds)
|
||
|
|
|
||
|
|
if let cgImage = context.makeImage() {
|
||
|
|
let coloredImage = UIImage(cgImage: cgImage)
|
||
|
|
return coloredImage
|
||
|
|
} else {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|