UIButton 更改样式和链式编程

作者: DDY | 来源:发表于2017-07-21 20:22 被阅读190次

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

UIButton 更改样式和链式编程

相关文章

网友评论

延伸阅读

深度阅读

栏目导航

热点阅读

iOS之功能细节

iOS实战