封装自定义View,创建N个按钮,按钮根据title长度Size-fit显示

为了实现  创建N多个按钮, 并且让按钮根据所给的title长度自适应, 并且高度自适应,封装了这个类

效果如图:

ButtonView.h

//
//  ButtonView.h
//  AiComic
//
//  Created by Damon on 15/2/26.
//  Copyright (c) 2015年 Damon. All rights reserved.
//

#import "BaseView.h"

@protocol searchButtonViewDelegate <NSObject>
// 协议方法:
//点击每个按钮时的协议方法
- (void)searchButtonViewClicked:(NSString *)str;
// 传出此时布局后View的高度
- (void)searchButtonViewFrame:(CGFloat)y;

@end

@interface ButtonView : BaseView
// 代理
@property (nonatomic, retain) id<searchButtonViewDelegate>delegate;
// 数组, 存储title
@property (nonatomic, retain) NSMutableArray *arr;
// View的高度
@property (nonatomic, assign) CGFloat y;
// 重写初始化方法
- (id)initWithFrame:(CGRect)frame buttonNumber:(NSInteger)number;
// 给数组赋值
- (void)setMyArr:(NSMutableArray *)arr;

@end

ButtonView.m文件

//
//  ButtonView.m
//  AiComic
//
//  Created by Damon on 15/2/26.
//  Copyright (c) 2015年 Damon. All rights reserved.
//

#import "ButtonView.h"

@interface ButtonView ()
// 按钮的数量
@property (nonatomic, assign) NSInteger number;
// 字体大小
@property (nonatomic,retain)UIFont *font;
@end

@implementation ButtonView

// 初始化
- (id)initWithFrame:(CGRect)frame buttonNumber:(NSInteger)number
{
    
    self = [super initWithFrame:frame];
    
    if (self) {
        
        self.arr = [NSMutableArray array];
        self.number = number;
        for (int i = 1; i <= number; i++) {
            
            [self.arr addObject:[NSString stringWithFormat:@"%d",i]];
            UIButton *button = [UIButton buttonWithType:UIButtonTypeSystem];
            [button addTarget:self action:@selector(buttonClicked:) forControlEvents:UIControlEventTouchUpInside];
            button.tag = i;
            //  self.button0.backgroundColor = [UIColor redColor];
            [self addSubview:button];
            self.font = button.titleLabel.font;
            
        }
    }
    
    return self;
}

- (void)layoutSubviews
{
    [super layoutSubviews];
    CGFloat width = self.frame.size.width;
    CGFloat frontWidth = 10;
    CGFloat widthOfButton = 20;
    CGFloat buttonHeight = 20;
    self.y = 20;
    
    // 布局
    for (int i = 1; i <=self.arr.count; i++) {
        
        if (i == 1) {
            
            UIButton *button = (UIButton *)[self viewWithTag:i];
            button.frame = CGRectMake(frontWidth, self.y, [self getLabelSize:[self.arr objectAtIndex:i - 1]].width, buttonHeight);
            
        } else {
            
            UIButton *upButton = (UIButton *)[self viewWithTag:i - 1];
            
            UIButton *button = (UIButton *)[self viewWithTag:i];
            
            
            if ((upButton.frame.size.width + upButton.frame.origin.x + widthOfButton + frontWidth + [self getLabelSize:[self.arr objectAtIndex:i - 1]].width) > width) {
                self.y = self.y + 40;
                button.frame = CGRectMake(frontWidth, self.y, [self getLabelSize:[self.arr objectAtIndex:i - 1]].width, buttonHeight);
            } else {
                
                button.frame = CGRectMake(upButton.frame.origin.x + upButton.frame.size.width + widthOfButton, self.y, [self getLabelSize:[self.arr objectAtIndex:i - 1]].width, buttonHeight);
            }
            
        }
        
    }
    // 布局后  把当前高度传给代理
    [self.delegate searchButtonViewFrame:self.y];
    
}



// 获取字符串的宽度
- (CGSize)getLabelSize:(NSString *)str
{
    CGSize size = CGSizeMake(150, 20);
    NSDictionary *dic = [NSDictionary dictionaryWithObjectsAndKeys:self.font, NSFontAttributeName, nil];
    
    CGSize Labelsize = [str boundingRectWithSize:size options:NSStringDrawingUsesLineFragmentOrigin attributes:dic context:nil].size;
    
    return Labelsize;
}

// //点击每个按钮时的协议方法的实现
- (void)buttonClicked:(UIButton *)button
{
    // 让代理执行方法
    [self.delegate searchButtonViewClicked:button.titleLabel.text];
    
}
// 给标题数组赋值
- (void)setMyArr:(NSMutableArray *)arr
{
    //self.number = arr.count;
    [self.arr removeAllObjects];
    [self.arr addObjectsFromArray:arr];
    [self setTitleOfButton];
}

// 给每个button赋值
- (void)setTitleOfButton
{
    NSInteger i = 1;
    for (NSString *str in self.arr) {
        
        UIButton *button = (UIButton *)[self viewWithTag:i];
        
        [button setTitle:str forState:UIControlStateNormal];
        i++;
    }
    
    [self layoutSubviews];
}





/*
 // Only override drawRect: if you perform custom drawing.
 // An empty implementation adversely affects performance during animation.
 - (void)drawRect:(CGRect)rect {
 // Drawing code
 }
 */

@end<span style="color:#ff0000;">
</span>