iOS根据颜色生成图片

有时候，想要给button一个背景颜色，对于UIControlStateNormal和UIControlStateSelected这两种状态，设置不同的颜色，奈何，只有设置图片的方法setBackgroundImage，总不至于跟UI要切图吧，而且对我而言，能自己做出来的，就不用图片，所以就想到能不能把颜色转化为相应的图片。
代码如下：

- (UIImage*) GetImageWithColor:(UIColor*)color andHeight:(CGFloat)height
{
    CGRect r= CGRectMake(0.0f, 0.0f, 1.0f, height);
    UIGraphicsBeginImageContext(r.size);
    CGContextRef context = UIGraphicsGetCurrentContext();
    
    CGContextSetFillColorWithColor(context, [color CGColor]);
    CGContextFillRect(context, r);
    
    UIImage *img = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return img;
}