iOS-分类的实现

一、原理

分类是运行时决议，在编译的过程中只是转化为可执行文件，并没有为类生成新方法。

分类结构

由图可以看出我们可以为分类添加实例方法、类方法、协议、属性。但是要注意添加的属性只是实现了get、set方法，没有实例变量。

大致流程：

1. 在运行APP的时候，加载完动态链接库，会加载可执行文件，通过runtime生成类、成员变量、方法列表，在宿主类的方法列表生成完之后，会开始加载分类的方法列表。
2. 取到所有分类的列表数组（按编译时的顺序排序），然后按倒序从分类列表里取出每个分类的方法列表，生成一个二维数组。
3. 把二维数组中的方法，按正序即从0索引开始，插入到宿主类的方法列表中（一维的数组），在原有类的方法之前。
4. 所以分类才拥有了“覆盖”原有类的方法功能，其实是原方法是存在的。

二、应用

我们在为分类添加属性的时候，实际没有生成实例变量，那么如果我们想要这么做怎么实现呢？答案是关联对象。
比如我们想通过分类给UIButton添加一个属性，来设置button的点击范围(比如点击范围上下左右都扩大20坐标)。

#import <UIKit/UIKit.h>

@interface UIButton (HitRect)
/**
 自定义响应边界 UIEdgeInsetsMake(-3, -4, -5, -6). 表示扩大
 例如： self.btn.hitEdgeInsets = UIEdgeInsetsMake(-3, -4, -5, -6);
 */
@property(nonatomic, assign) UIEdgeInsets hitEdgeInsets;
/**
 自定义响应边界 自定义的边界的范围 范围扩大3.0
 例如：self.btn.hitScale = 3.0;
 */
@property(nonatomic, assign) CGFloat hitScale;
@end

#import "UIButton+HitRect.h"
#import <objc/runtime.h>


static const char * kHitEdgeInsets = "hitEdgeInsets";
static const char * kHitScale      = "hitScale";

@implementation UIButton (HitRect)

#pragma mark - set Method
-(void)setHitEdgeInsets:(UIEdgeInsets)hitEdgeInsets{
    NSValue *value = [NSValue value:&hitEdgeInsets withObjCType:@encode(UIEdgeInsets)];
    objc_setAssociatedObject(self,kHitEdgeInsets, value, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

-(void)setHitScale:(CGFloat)hitScale{
    CGFloat width = self.bounds.size.width * hitScale;
    CGFloat height = self.bounds.size.height * hitScale;
    self.hitEdgeInsets = UIEdgeInsetsMake(-height, -width,-height, -width);
    objc_setAssociatedObject(self, kHitScale, @(hitScale), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
#pragma mark - get Method
-(UIEdgeInsets)hitEdgeInsets{
    NSValue *value = objc_getAssociatedObject(self, kHitEdgeInsets);
    UIEdgeInsets edgeInsets;
    [value getValue:&edgeInsets];
    return value ? edgeInsets:UIEdgeInsetsZero;
}
-(CGFloat)hitScale{
    return [objc_getAssociatedObject(self, kHitScale) floatValue];
}
#pragma mark - override super method
-(BOOL)pointInside:(CGPoint)point withEvent:(UIEvent *)event{
    //如果 button 边界值无变化  失效 隐藏 或者透明 直接返回
    if(UIEdgeInsetsEqualToEdgeInsets(self.hitEdgeInsets, UIEdgeInsetsZero) || !self.enabled || self.hidden || self.alpha == 0 ) {
        return [super pointInside:point withEvent:event];
    }else{
        CGRect relativeFrame = self.bounds;
        CGRect hitFrame = UIEdgeInsetsInsetRect(relativeFrame, self.hitEdgeInsets);
        return CGRectContainsPoint(hitFrame, point);
    }
}

这个例子中我们添加了两个属性，我们拿hitEdgeInsets来讲解，有两个功能点：

• 通过关联对象给分类添加了属性hitEdgeInsets，设置、获取点击范围。
• 重写pointInside:withEvent方法，在这个方法里获取根据自身bounds和已设置的hitEdgeInsets，来获取最新的响应范围，然后通过CGRectContainsPoint(hitFrame,point)来判断是否在点击范围内。