UIImage+Color.swift

extension UIImage {

     class func imageFrom(color: UIColor, size: CGSize = CGSize(width: 1, height: 1) ) -> UIImage {
        let rect = CGRect(origin: CGPoint(x: 0, y:0), size: size)

        UIGraphicsBeginImageContextWithOptions(rect.size, true, 0)
        defer {
            UIGraphicsEndImageContext()
        }
          
        if let context = UIGraphicsGetCurrentContext() {
            context.setFillColor(color.cgColor)
            context.fill(rect)
        }
          
        return UIGraphicsGetImageFromCurrentImageContext()!
    }

    class func imageFrom(layer: CALayer) -> UIImage {
        UIGraphicsBeginImageContextWithOptions(layer.frame.size, layer.isOpaque, 0.0)
        defer {
            UIGraphicsEndImageContext()
        }
         
        if let context = UIGraphicsGetCurrentContext() {
            layer.render(in: context)
        }
         
        return UIGraphicsGetImageFromCurrentImageContext()!

    }
}