利用分类（category）扩充UIImage，实现按钮点击改变背景颜色

原文地址：iOS button 点击状态下改变背景颜色（by韩万杰）

1.为UIImage添加分类

//添加image分类  .m文件代码
#import "UIImage+Extension.h"
 
@implementation UIImage (Extension)
 
+ (UIImage *)imageWithColor:(UIColor *)color
{
    CGFloat imageW = 3;
    CGFloat imageH = 3;
    // 1.开启基于位图的图形上下文
    UIGraphicsBeginImageContextWithOptions(CGSizeMake(imageW, imageH), NO, 0.0);
    // 2.画一个color颜色的矩形框
    [color set];
    UIRectFill(CGRectMake(0, 0, imageW, imageH));
     
    // 3.拿到图片
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
     
    // 4.关闭上下文
    UIGraphicsEndImageContext();
     
    return image;
}
 
@end
 
 
//image分类  .h文件代码
 
#import <UIKit/UIKit.h>
 
@interface UIImage (Extension)
 
/**
 *  生成的图片的rect默认为100,100
 */
+ (UIImage *)imageWithColor:(UIColor *)color;
@end
 
//导入头文件   调用  分类方法   可改变image背景颜色
    
 UIButton *loginButton = [[UIButton alloc]initWithFrame:self.view.bounds];
    [loginButton setTitle:@"登 录" forState:UIControlStateNormal];
    loginButton.backgroundColor = [UIColor whiteColor];
    [loginButton setBackgroundImage:[UIImage imageWithColor:[UIColor lightGrayColor]] forState:UIControlStateHighlighted];
    [self.view addSubview:loginButton];

扩展阅读：

1.category和extension的区别，查了一下，网上同胞写的很多都不准确，如果你同我一样是新手建议谨慎，个人觉得《Effective Objective-C 2.0》里面说的比较清晰准确。

2.iOS开发系列——打造自己的“美图秀秀”  Quartz 2D绘制2D图形和Core Image相关的知识