Свифт 3 :
Это решение может быть удобным, если вы уже установили изображение через конструктор интерфейса xCode. В основном у вас есть одно расширение для раскрашивания изображения:
extension UIImage {
public func image(withTintColor color: UIColor) -> UIImage{
UIGraphicsBeginImageContextWithOptions(self.size, false, self.scale)
let context: CGContext = UIGraphicsGetCurrentContext()!
context.translateBy(x: 0, y: self.size.height)
context.scaleBy(x: 1.0, y: -1.0)
context.setBlendMode(CGBlendMode.normal)
let rect: CGRect = CGRect(x: 0, y: 0, width: self.size.width, height: self.size.height)
context.clip(to: rect, mask: self.cgImage!)
color.setFill()
context.fill(rect)
let newImage: UIImage = UIGraphicsGetImageFromCurrentImageContext()!
UIGraphicsEndImageContext()
return newImage
}
}
Затем вы можете подготовить это расширение UIButton, чтобы раскрасить изображение для определенного состояния:
extension UIButton {
func imageWith(color:UIColor, for: UIControlState) {
if let imageForState = self.image(for: state) {
self.image(for: .normal)?.withRenderingMode(.alwaysTemplate)
let colorizedImage = imageForState.image(withTintColor: color)
self.setImage(colorizedImage, for: state)
}
}
}
Использование:
myButton.imageWith(.red, for: .normal)
PS (работает хорошо и в ячейках таблицы, вам не нужно вызывать setNeedDisplay()
метод, изменение цвета происходит немедленно из-за расширения UIImage ..