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)来判断是否在点击范围内。

推荐阅读更多精彩内容