Swift 3 :
Essa solução pode ser confortável se você já configurou sua imagem através do construtor de interface xCode. Basicamente, você tem uma extensão para colorir uma imagem:
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
}
}
Em seguida, você pode preparar esta extensão UIButton para colorir a imagem para um estado específico:
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)
}
}
}
Uso:
myButton.imageWith(.red, for: .normal)
PS (funcionando bem também nas células da tabela, você não precisa chamar o setNeedDisplay()
método, a alteração da cor é imediata devido à extensão UIImage .