UIButton 更改样式和链式编程
2017-07-21 本文已影响190人
DDY
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];
}
