iOS UISegmentControl实现自定义分栏效果

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

@interface BCLCommunitySegmentView : UIView

//定义一下block
@property (nonatomic, strong) PassValueBlock returnBlock;
@property (nonatomic, strong) UIImageView *selectImage;//这个就是短红线

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

//block传递值方法
- (void)setReturnBlock:(PassValueBlock)returnBlock;
@end

- (void)creatSegmentView {
? ? //设置按钮的宽度
? ? _itemWidth = self.frame.size.width / _itemCounts;
? ? //循环创建按钮
? ? for (int i = 0; i < _itemCounts; i++) {
? ? ? ? UIButton *button ?= [[UIButton alloc]initWithFrame:CGRectMake((i + 1) *_itemWidth/2, 0, _itemWidth/2, self.frame.size.height)];
? ? ? ? [self addSubview:button];
? ? ? ??
? ? ? ? //设置button的字
? ? ? ? [button setTitle:_titleArray[i] forState:UIControlStateNormal];
? ? ? ? //设置button的字颜色
? ? ? ??
? ? ? ? [button setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
? ? ? ? //设置字体大小
? ? ? ? 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;
? ? ? ? }
? ? }
? ??
? ??
? ? //添加一个select
? ? _selectImage = [[UIImageView alloc]initWithFrame:CGRectMake(_itemWidth / 2, self.frame.size.height - 2, _itemWidth / 2, 2)];
? ? _selectImage.image = [UIImage imageNamed:@"bcl_bg_community_segment_color_line"];
? ? [self addSubview:_selectImage];
}

-(void)buttonAction:(UIButton *)button{
? ??
? ? //当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;
? ??
? ? //通过当前的tag值设置select的位置
? ? NSInteger index = button.tag - 1000;
? ? [UIView animateWithDuration:0.3 animations:^{
? ? ? ? self->_selectImage.frame = CGRectMake((1 + index)*_itemWidth/2, _selectImage.frame.origin.y, self->_selectImage.frame.size.width, _selectImage.frame.size.height);
? ? }];
? ??
? ? _returnBlock(index);
}

@interface BCLCommunityView : UIView

@property (nonatomic, strong) UIScrollView *scrollerView;
@property(nonatomic ,strong) UITableView *circleTableView;
@property(nonatomic ,strong) UITableView *squreTableView;
@property (nonatomic, strong)BCLCommunitySegmentView *segmentView;

@end

- (instancetype) initWithFrame:(CGRect)frame {
? ? if(self = [super initWithFrame:frame]) {
? ? ? ? [self setSegmentView];
? ? ? ??
? ? ? ? _circleTableView = [self loadTableView];
? ? ? ? _squreTableView = [self loadTableView];
? ? ? ??
? ? ? ? _circleTableView.tag = 1;
? ? ? ? _squreTableView.tag = 2;
? ? ? ??
? ? ? ? _scrollerView = [[UIScrollView alloc] init];
? ? ? ? _scrollerView.frame = CGRectMake(0, 104, KScreenW, KScreenH);
? ? ? ? _scrollerView.pagingEnabled = YES;
? ? ? ? _scrollerView.scrollEnabled = YES;
? ? ? ? _scrollerView.contentSize = CGSizeMake(KScreenW * 2, KScreenH);
? ? ? ? _scrollerView.bounces = YES;
? ? ? ? _scrollerView.delegate = self;
? ? ? ??
? ? ? ? [_scrollerView addSubview:_circleTableView];
? ? ? ? [_scrollerView addSubview:_squreTableView];
? ? ? ??
? ? ? ? _circleTableView.frame = CGRectMake(0, 0, KScreenW, KScreenH);
? ? ? ? _squreTableView.frame = CGRectMake(KScreenW, 0, KScreenW, KScreenH);
? ? ? ? [self addSubview:_scrollerView];
? ? }
? ? return self;
}
- (UITableView *)loadTableView
{
? ? UITableView ?*tableView = [[UITableView alloc] initWithFrame:CGRectMake(0, 0, KScreenW, KScreenH) style:UITableViewStyleGrouped];
? ? tableView.showsVerticalScrollIndicator = NO;
? ? [tableView registerClass:[UITableViewCell class] forCellReuseIdentifier:@"cell"];
? ??
? ? tableView.dataSource = self;
? ??
? ? [self addSubview:tableView];
? ? return tableView;
}
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
? ? if(tableView.tag == 1) {
? ? ? ? return 3;
? ? } else {
? ? ? ? ?return 2;
? ? }
? ?
}
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
? ? return 1;
}
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
? ? if(tableView.tag == 1) {
? ? ? ? UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"cell" forIndexPath:indexPath];
? ? ? ? if(!cell) {
? ? ? ? ? ? cell = [[UITableViewCell alloc]initWithStyle:UITableViewCellStyleDefault reuseIdentifier:@"cell"];
? ? ? ? ? ??
? ? ? ? }
? ? ? ? cell.backgroundColor = [UIColor redColor];
? ? ? ??
? ? ? ? cell.textLabel.text = @"11111";
? ? ? ? return cell;
? ? } else {
? ? ? ? UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"cell" forIndexPath:indexPath];
? ? ? ? if(!cell) {
? ? ? ? ? ? cell = [[UITableViewCell alloc]initWithStyle:UITableViewCellStyleDefault reuseIdentifier:@"cell"];
? ? ? ? }
? ? ? ? return cell;
? ? }
? ??
}

- (void)scrollViewDidEndDecelerating:(UIScrollView *)scrollView {
? ? CGRect frame = _segmentView.selectImage.frame;
? ? if(scrollView.contentOffset.x / KScreenW == 0) {
? ? ? ? [UIView animateWithDuration:0.1 animations:^{
? ? ? ? _segmentView.selectImage.frame = CGRectMake(KScreenW / 4, frame.origin.y, frame.size.width, frame.size.height);
? ? ? ? }];
? ? } else if(scrollView.contentOffset.x / KScreenW == 1){
? ? ? ? [UIView animateWithDuration:0.1 animations:^{
? ? ? ? ? ? _segmentView.selectImage.frame = CGRectMake(KScreenW / 2, frame.origin.y, frame.size.width, frame.size.height);
? ? ? ? }];
? ? }
}