UIImage+Color.swift

import UIKit

extension UIImage {
  class func colorAsImage(from color:UIColor, for size:CGSize) -> UIImage? {
    var img:UIImage?
    let rect = CGRect(x:0.0, y:0.0, width: size.width, height: size.height)
    UIGraphicsBeginImageContextWithOptions(rect.size, false, 0)
    guard let context = UIGraphicsGetCurrentContext() else { return nil }
    context.setFillColor(color.cgColor)
    context.fill(rect)
    img = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    guard let colorImage = img else { return nil }
    return colorImage
  }
}