自定义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