UIImage with color in Swift

UIImage.swift

//
//  UIImage.swift
//  ProjectIceCream
//
//  Created by Luca D'Incà on 24/11/14.
//  Copyright (c) 2014 B3LAB. All rights reserved.
//

import Foundation
import Swift
import UIKit

internal extension UIImage {
    
    /**
    Creates and returns a UIImage with a specific color
    
    :param: color, rect: image size
    :returns: image with a specific color and size
    */
    
    func imageWithColor(color: UIColor, andRect rect:CGRect) -> UIImage {
        UIGraphicsBeginImageContextWithOptions(rect.size, false, 0)
        
        color.setFill()
        UIRectFill(rect)
        
        var image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        return image
    }
}