UIButton 更改样式和链式编程

Paste_Image.png

请关注，防止你用了，我改了，有问题连个商量的人都找不到...

继承方式

DDYButton.h

#import <UIKit/UIKit.h>

typedef NS_ENUM(NSInteger, DDYBtnStyle) {
    DDYBtnStyleDefault   = 0,     // 左图右文，整体居中，默认状态
    DDYBtnStyleImgLeft   = 1,     // 左图右文，整体居中，设置间隙
    DDYBtnStyleImgRight  = 2,     // 左文右图，整体居中，设置间隙
    DDYBtnStyleImgTop    = 3,     // 上图下文，整体居中，设置间隙
    DDYBtnStyleImgDown   = 4,     // 下图上文，整体居中，设置间隙
};

@interface DDYButton : UIButton

+ (instancetype)customDDYBtn;

/** 布局方式 */
@property (nonatomic, assign) DDYBtnStyle btnStyle;
/** 图文间距 默认5 */
@property (nonatomic, assign) CGFloat padding;
/** 文字字体 */
@property (nonatomic, strong) UIFont *textFont;

@end

DDYButton.m

#import "DDYButton.h"
#import <objc/runtime.h>

@implementation DDYButton

+ (instancetype)customDDYBtn
{
    return [[self class] buttonWithType:UIButtonTypeCustom];
}

- (instancetype)initWithFrame:(CGRect)frame
{
    if (self = [super initWithFrame:frame])
    {
        self.padding = 5;
        self.btnStyle = DDYStyleDefault;
    }
    return self;
}

- (void)layoutSubviews
{
    [super layoutSubviews];
    
    if (self.imageView.image && self.titleLabel.text)
    {
        [self setAttributes];
        switch (self.btnStyle)
        {
            case DDYBtnStyleImgRight:
                [self layoutHorizontalWithLeftView:self.titleLabel rightView:self.imageView];
                break;
            case DDYBtnStyleImgLeft:
                [self layoutHorizontalWithLeftView:self.imageView rightView:self.titleLabel];
                break;
            case DDYBtnStyleImgTop:
                [self layoutVerticalWithUpView:self.imageView downView:self.titleLabel];
                break;
            case DDYBtnStyleImgDown:
                [self layoutVerticalWithUpView:self.titleLabel downView:self.imageView];
                break;
            default:
                break;        
        }
    }
}

- (void)layoutHorizontalWithLeftView:(UIView *)leftView rightView:(UIView *)rightView
{
    CGFloat totalW = leftView.ddy_w + self.padding + rightView.ddy_w;
    
    leftView.ddy_x = (self.ddy_w - totalW)/2.0;
    leftView.ddy_y = (self.ddy_h - leftView.ddy_h)/2.0;
    
    rightView.ddy_x = leftView.ddy_right + self.padding;
    rightView.ddy_y = (self.ddy_h - rightView.ddy_h)/2.0;
}

- (void)layoutVerticalWithUpView:(UIView *)upView downView:(UIView *)downView
{
    CGFloat totalH = upView.ddy_h + self.padding + downView.ddy_h;
    
    upView.ddy_x = (self.ddy_w - upView.ddy_w)/2.0;
    upView.ddy_y = (self.ddy_h - totalH)/2.0;
    
    downView.ddy_x = (self.ddy_w - downView.ddy_w)/2.0;
    downView.ddy_y = upView.ddy_bottom + self.padding;

}

- (void)setImage:(UIImage *)image forState:(UIControlState)state
{
    [super setImage:image forState:state];
    [self.imageView sizeToFit];
}

- (void)setTitle:(NSString *)title forState:(UIControlState)state
{
    [super setTitle:title forState:state];
    [self.titleLabel sizeToFit];
}

- (void)setAttributes
{
    if (_textFont)
    {
        self.titleLabel.font = _textFont;
    }
    [self.titleLabel sizeToFit];
}

@end

分类方式

UIButton+DDYStyle.h

#import <UIKit/UIKit.h>

typedef NS_ENUM(NSInteger, DDYStyle) {
    DDYStyleDefault = 0,        // 左图右文，整体居中，默认状态
    DDYStyleImgLeft = 1,        // 左图右文，整体居中，设置间隙
    DDYStyleImgRight = 2,       // 左文右图，整体居中，设置间隙
    DDYStyleImgTop = 3,         // 上图下文，整体居中，设置间隙
    DDYStyleImgBottom = 4,      // 下图上文，整体居中，设置间隙
};

@interface UIButton (DDYStyle)

+ (instancetype)customBtn;
/** style:图文样式 padding:图文间隙 */
- (void)setDDYStyle:(DDYStyle)style padding:(CGFloat)padding;
- (void)DDYStyle:(DDYStyle)style padding:(CGFloat)padding;

@end

UIButton+DDYStyle.m

#import "UIButton+DDYStyle.h"

@implementation UIButton (DDYStyle)

+ (instancetype)customBtn
{
    return [[self class] buttonWithType:UIButtonTypeCustom];
}

- (void)setDDYStyle:(DDYStyle)style padding:(CGFloat)padding
{
    if (self.imageView.image != nil && self.titleLabel.text != nil)
    {
        // 先还原
        self.titleEdgeInsets = UIEdgeInsetsZero;
        self.imageEdgeInsets = UIEdgeInsetsZero;
        
        // 上图下文或者上文下图情况重新计算高
        CGFloat imageX = self.imageView.ddy_x;
        CGFloat titleX = self.titleLabel.ddy_x;
        CGFloat imageY = self.imageView.ddy_y;
        CGFloat titleY = self.titleLabel.ddy_y;
        CGFloat imageW = self.imageView.ddy_w;
        CGFloat titleW = self.titleLabel.ddy_w;
        CGFloat imageH = self.imageView.ddy_h;
        CGFloat titleH = self.titleLabel.ddy_h;
        CGFloat totalH = imageH + titleH + padding;
        CGFloat selfH = self.ddy_h;
        CGFloat selfW = self.ddy_w;
        
        switch (style)
        {
            case DDYStyleImgLeft:
                self.titleEdgeInsets = UIEdgeInsetsMake(0, padding/2.0, 0, -padding/2.0);
                self.imageEdgeInsets = UIEdgeInsetsMake(0, -padding/2.0, 0, padding/2.0);
                break;
                
            case DDYStyleImgRight:
                self.titleEdgeInsets = UIEdgeInsetsMake(0, -(imageW + padding/2.0), 0, (imageW + padding/2.0));
                self.imageEdgeInsets = UIEdgeInsetsMake(0, (titleW + padding/2.0), 0, -(titleW + padding/2.0));
                break;
                
            case DDYStyleImgTop:
                self.titleEdgeInsets = UIEdgeInsetsMake(((selfH - totalH)/2.0 + imageH + padding - titleY),
                                                        -(titleX + titleW/2.0 - selfW/2.0),
                                                        -((selfH - totalH)/2.0 + imageH + padding - titleY),
                                                        (titleX + titleW/2.0 - selfW/2.0));
                self.imageEdgeInsets = UIEdgeInsetsMake(((selfH - totalH)/2.0 - imageY),
                                                        (selfW/2.0 - imageX - imageW/2.0),
                                                        -((selfH - totalH)/2.0 - imageY),
                                                        -(selfW/2.0 - imageX - imageW/2.0));
                break;
                
            case DDYStyleImgBottom:
                NSLog(@"%f",titleX);
                self.titleEdgeInsets = UIEdgeInsetsMake(((selfH - totalH)/2.0 - titleY),
                                                        -(titleX + titleW/2.0 - selfW/2.0),
                                                        -((selfH - totalH)/2.0 - titleY),
                                                        (titleX + titleW/2.0 - selfW/2.0));
                self.imageEdgeInsets = UIEdgeInsetsMake(((selfH - totalH)/2.0 + titleH + padding - imageY),
                                                        (selfW/2.0 - imageX - imageW/2.0),
                                                        -((selfH - totalH)/2.0 + titleH + padding - imageY),
                                                        -(selfW/2.0 - imageX - imageW/2.0));
                break;
                
            default:
                break;
        }
    }
}
- (void)DDYStyle:(DDYStyle)style padding:(CGFloat)padding {
    [self setDDYStyle:style padding:padding];
}
@end

链式编程分类

DDYButton+DDYLinkBlock.h

#import "LinkBlockDefine.h"
#import "DDYButton.h"

@interface NSObject (DDYLinkBlock)

////////////////////////////// 链式编程 /////////////////////////////

LBDeclare DDYButton *(^btnFrame)(CGFloat x, CGFloat y, CGFloat w,CGFloat h);
LBDeclare DDYButton *(^btnX)(CGFloat x);
LBDeclare DDYButton *(^btnY)(CGFloat y);
LBDeclare DDYButton *(^btnW)(CGFloat w);
LBDeclare DDYButton *(^btnH)(CGFloat h);
LBDeclare DDYButton *(^btnTitleN)(NSString *title);
LBDeclare DDYButton *(^btnTitleS)(NSString *title);
LBDeclare DDYButton *(^btnTitleH)(NSString *title);
LBDeclare DDYButton *(^btnTitleColorN)(UIColor *color);
LBDeclare DDYButton *(^btnTitleColorS)(UIColor *color);
LBDeclare DDYButton *(^btnTitleColorH)(UIColor *color);
LBDeclare DDYButton *(^btnImageN)(UIImage *image);
LBDeclare DDYButton *(^btnImageS)(UIImage *image);
LBDeclare DDYButton *(^btnImageH)(UIImage *image);
LBDeclare DDYButton *(^btnImgNameN)(NSString *imgName);
LBDeclare DDYButton *(^btnImgNameS)(NSString *imgName);
LBDeclare DDYButton *(^btnImgNameH)(NSString *imgName);
LBDeclare DDYButton *(^btnLayoutStyle)(DDYBtnStyle style);
LBDeclare DDYButton *(^btnPadding)(CGFloat padding);
LBDeclare DDYButton *(^btnSuperView)(UIView *superView);
LBDeclare DDYButton *(^btnFont)(UIFont *font);

//- (DDYButton *(^)(UIFont *font))btnFont;

@end

DDYButton+DDYLinkBlock.m

#import "DDYButton+DDYLinkBlock.h"

@implementation DDYButton (DDYLinkBlock)

- (DDYButton *(^)(CGFloat, CGFloat, CGFloat, CGFloat))btnFrame
{
    return ^id(CGFloat x,CGFloat y, CGFloat w, CGFloat h) {
        LinkHandle_REF(DDYButton)   // 防止nil.method([nil method]是允许的)
        LinkGroupHandle_REF(btnFrame, x, y, w, h)
        _self.frame = CGRectMake(x, y, w, h);
        return _self;
    };
}

- (DDYButton *(^)(CGFloat))btnX
{
    return ^id(CGFloat x) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnX, x)
        _self.ddy_x = x;
        return _self;
    };
}

- (DDYButton *(^)(CGFloat))btnY
{
    return ^id(CGFloat y) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnY, y)
        _self.ddy_y = y;
        return _self;
    };
}

- (DDYButton *(^)(CGFloat))btnW
{
    return ^id(CGFloat w) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnW, w)
        _self.ddy_w = w;
        return _self;
    };
}

- (DDYButton *(^)(CGFloat))btnH
{
    return ^id(CGFloat h) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnH, h)
        _self.ddy_h = h;
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnTitleN
{
    return ^id(NSString *title) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleN, title)
        [_self setTitle:title forState:UIControlStateNormal];
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnTitleS
{
    return ^id(NSString *title) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleS, title)
        [_self setTitle:title forState:UIControlStateSelected];
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnTitleH
{
    return ^id(NSString *title) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleH, title)
        [_self setTitle:title forState:UIControlStateHighlighted];
        return _self;
    };
}

- (DDYButton *(^)(UIColor *))btnTitleColorN
{
    return ^id(UIColor *color) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleColorN, color)
        [_self setTitleColor:color forState:UIControlStateNormal];
        return _self;
    };
}

- (DDYButton *(^)(UIColor *))btnTitleColorS
{
    return ^id(UIColor *color) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleColorS, color)
        [_self setTitleColor:color forState:UIControlStateSelected];
        return _self;
    };
}

- (DDYButton *(^)(UIColor *))btnTitleColorH
{
    return ^id(UIColor *color) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnTitleColorH, color)
        [_self setTitleColor:color forState:UIControlStateHighlighted];
        return _self;
    };
}

- (DDYButton *(^)(UIImage *))btnImageN
{
    return ^id(UIImage *image) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImageN, image)
        [_self setImage:image forState:UIControlStateNormal];
        return _self;
    };
}

- (DDYButton *(^)(UIImage *))btnImageS
{
    return ^id(UIImage *image) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImageS, image)
        [_self setImage:image forState:UIControlStateSelected];
        return _self;
    };
}

- (DDYButton *(^)(UIImage *))btnImageH
{
    return ^id(UIImage *image) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImageH, image)
        [_self setImage:image forState:UIControlStateHighlighted];
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnImgNameN
{
    return ^id(NSString *imgName) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImgNameN, imgName)
        [_self setImage:[UIImage imageNamed:imgName] forState:UIControlStateNormal];
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnImgNameS
{
    return ^id(NSString *imgName) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImgNameS, imgName)
        [_self setImage:[UIImage imageNamed:imgName] forState:UIControlStateSelected];
        return _self;
    };
}

- (DDYButton *(^)(NSString *))btnImgNameh
{
    return ^id(NSString *imgName) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnImgNameH, imgName)
        [_self setImage:[UIImage imageNamed:imgName] forState:UIControlStateHighlighted];
        return _self;
    };
}

- (DDYButton *(^)(DDYBtnStyle))btnLayoutStyle
{
    return ^id(DDYBtnStyle style) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnLayoutStyle, style)
        _self.btnStyle = style;
        return _self;
    };
}

- (DDYButton *(^)(CGFloat))btnPadding
{
    return ^id(CGFloat padding) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnPadding, padding)
        _self.padding = padding;
        return _self;
    };
}

- (DDYButton *(^)(UIView *))btnSuperView
{
    return ^id(UIView *superView) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnSuperView, superView)
        [superView addSubview:_self];
        return _self;
    };
}

- (DDYButton *(^)(UIFont *))btnFont
{
    return ^id(UIFont *font) {
        LinkHandle_REF(DDYButton)
        LinkGroupHandle_REF(btnFont, font)
        _self.titleLabel.font = font;
        return _self;
    };
}

@end

生成按钮

- (void)setupButton
{
    // 链式编程 继承方式
    [DDYButton customDDYBtn].btnFrame(10, 10, 60, 80)
                            .btnFont(DDYFont(12))
                            .btnTitleN(@"上图下文")
                            .btnTitleColorN([UIColor blueColor])
                            .btnImageN([UIImage imageWithColor:[UIColor redColor] size:CGSizeMake(20, 20)])
                            .btnLayoutStyle(DDYBtnStyleImgTop)
                            .btnPadding(5)
                            .btnSuperView(self.view);
    
    // 原生编程 继承方式
    DDYButton *button = [DDYButton customDDYBtn];
    button.frame = CGRectMake(80, 10, 60, 80);
    button.titleLabel.font = DDYFont(12);
    [button setTitle:@"下图上文" forState:UIControlStateNormal];
    [button setTitleColor:[UIColor blueColor] forState:UIControlStateNormal];
    [button setImage:[UIImage imageWithColor:[UIColor redColor] size:CGSizeMake(20, 20)] forState:UIControlStateNormal];
    button.btnStyle = DDYBtnStyleImgDown;
    button.padding = 5;
    [self.view addSubview:button];
    
    // 原生编程 分类方式
    UIButton *btn = [UIButton buttonWithType:UIButtonTypeCustom];
    btn.frame = CGRectMake(150, 10, 85, 80);
    btn.titleLabel.font = DDYFont(12);
    [btn setTitle:@"左文右图" forState:UIControlStateNormal];
    [btn setTitleColor:[UIColor blueColor] forState:UIControlStateNormal];
    [btn setImage:[UIImage imageWithColor:[UIColor redColor] size:CGSizeMake(20, 20)] forState:UIControlStateNormal];
    [btn DDYStyle:DDYStyleImgRight padding:5];
    [self.view addSubview:btn];
    
}

码农不易点星星 scan code

最后编辑于：2017.11.17 14:38:27

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 213,558评论 6赞 492
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 91,002评论 3赞 387
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 159,036评论 0赞 349
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 57,024评论 1赞 285
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 66,144评论 6赞 385
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 50,255评论 1赞 292
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 39,295评论 3赞 412
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 38,068评论 0赞 268
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 44,478评论 1赞 305
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 36,789评论 2赞 327
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 38,965评论 1赞 341
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 34,649评论 4赞 336
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 40,267评论 3赞 318
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 30,982评论 0赞 21
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 32,223评论 1赞 267
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 46,800评论 2赞 365
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 43,847评论 2赞 351

UIButton 更改样式和链式编程

推荐阅读更多精彩内容