UIImageExtensions.swift

import UIKit

extension UIImage {

    public convenience init(color: UIColor, size: CGSize) {
        UIGraphicsBeginImageContextWithOptions(size, true, 0.0)
        defer {
            UIGraphicsEndImageContext()
        }

        let context = UIGraphicsGetCurrentContext()
        assert(context != nil)

        context!.setFillColor(color.cgColor)
        context!.fill(CGRect(origin: .zero, size: size))

        let image = context!.makeImage()
        assert(image != nil)

        self.init(cgImage: image!, scale: UIScreen.main.scale, orientation: UIImageOrientation.up)
    }
}