自定义UISegmentControl控件

平时我们在项目开发中，对于一般的需求使用系统默认的按钮组控件即可完成。但是有的时候，由于用户的需求，默认控件既不美观也不是很实用，所以这里就自己封装一个稍微美观的控件。

SegmentView.h

//定义block，用来传递点击的第几个按钮
typedef void (^PassValueBlock)(NSInteger index);

在.h文件中需要声明一个block用来传递点击按钮组的标记，通知使用该类点击的是第几个按钮。

SegmentView.m

    NSInteger _itemCounts;//控件的数目
    NSArray *_titleArray;//存放title
    float _itemWidth;//单个按钮的宽度
    UIImageView *_selectImage;  //选中背景图

在这里需要设置几个属性，需要几个按钮组，每个按钮组的名称，单个按钮的宽度，选中按钮的背景（可以不设置）。
在这里还需要声明一个方法，供其它类来调用，传入一个title数组即可。我这里用的是复写init方法。

//初始化按钮组，传入frame和名称
-(id)initWithFrame:(CGRect)frame withTitleArray:(NSArray *)array;

这样拿到title数组，就可以设置控件的数目，名称和每个按钮的宽度。

利用for循环创建按钮组，根据每个按钮的个数和宽度来进行设置。

 //循环创建按钮
    for (int i = 0; i < _itemCounts; i++) {
        UIButton *button  = [[UIButton alloc]initWithFrame:CGRectMake(i *_itemWidth, 0, _itemWidth, self.frame.size.height)];
        [self addSubview:button];

        //设置button的字
        [button setTitle:_titleArray[i] forState:UIControlStateNormal];
        //设置button的字颜色

        [button setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
        [button setTitleColor:[UIColor orangeColor] forState:UIControlStateSelected];

        //设置字体大小
        button.titleLabel.font = [UIFont systemFontOfSize:20];
        //设置居中显示
        button.titleLabel.textAlignment = NSTextAlignmentCenter;
        //设置tag值
        button.tag = 1000 + i;
        //添加点击事件
        [button addTarget:self action:@selector(buttonAction:) forControlEvents:UIControlEventTouchUpInside];
        //如果是第一个，默认被选中
        if (i == 0) {
            button.selected = YES;
            button.titleLabel.font = [UIFont systemFontOfSize:30];
        }
    }

同时设置默认选中第一个按钮，之间对按钮组的样式进行了设计，可以根据自己的需求灵活修改。

在按钮点击事件触发后，先把所有的按钮设置为未点击状态，然后再将点击的按钮设置为选中状态。并且通过tag值来设置背景图片的位置。

    //当button被点击，所有的button都设为未选中状态
    for (UIView *view in self.subviews) {
        if ([view isKindOfClass:[UIButton class]]) {
            UIButton *subButton = (UIButton*)view;
            subButton.selected = NO;
            subButton.titleLabel.font = [UIFont systemFontOfSize:20];
        }
    }  

    //然后将选中的这个button变为选中状态
    button.selected = YES;
    button.titleLabel.font = [UIFont systemFontOfSize:30];

    //通过当前的tag值设置select的位置
    NSInteger index = button.tag - 1000;
    [UIView animateWithDuration:.3 animations:^{
    _selectImage.frame = CGRectMake(index*_itemWidth, _selectImage.frame.origin.y, _selectImage.frame.size.width, _selectImage.frame.size.height);
    }];

最后一步，在通过声明的block将点击的按钮是第几个传递出来即可。

_returnBlock(index);

ViewController.m

直接在特定的类中使用即可，这里需要注意的是，一定不要忘了获取block传递的值，不获取会导致程序崩溃。

    SegmentView *segmentV = [[SegmentView alloc]initWithFrame:frame     withTitleArray:titleArr];
    [self.view addSubview:segmentV];

    //获取返回值
    [segmentV setReturnBlock:^(NSInteger index) {
        NSLog(@"%ld",(long)index);
    }];

附上Demo：
https://github.com/mayuwen/MYWSegmentView