自定义segmentControl

#import <UIKit/UIKit.h>

@interface PDESegmentControl : UIView

{
    
    NSMutableArray *segmentButtons;
    
    NSMutableArray *buttonImgNames;
    
}

@property (readonly, nonatomic)  NSInteger selectedSegmentIndex;
-(id) initWithFrame:(CGRect)frame items:(NSArray*)itemArray;

-(void)setSegmentIndex:(NSInteger)index;

//segment control

//    NSArray * segmentItems = [NSArray arrayWithObjects: @"未审批", @"已审批", nil];

//    UISegmentedControl *segmentedControl = [[UISegmentedControl alloc] initWithItems: segmentItems];

//    segmentedControl.frame = CGRectMake(310, 295, 160, 36);

//    [segmentedControl addTarget:self action:@selector(segmentAction:) forControlEvents:UIControlEventValueChanged];

//    segmentedControl.segmentedControlStyle = UISegmentedControlStyleBordered;

//    segmentedControl.selectedSegmentIndex = 0;

//    myBorrowRequestType = segmentedControl.selectedSegmentIndex;

//    [self.view addSubview:segmentedControl];
@end

//
//  PDESegmentControl.m
//  xxx
//
//  Created by Tony on 15/4/2.
//  Copyright (c) 2015年 Tony. All rights reserved.
//

#import "PDESegmentControl.h"

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

#define SEGMENT_SELECTED 1


@implementation PDESegmentControl


- (id)initWithFrame:(CGRect)frame

{

self = [super initWithFrame:frame];

if (self) {

// Initialization code

}

return self;

}


- (id) initWithFrame:(CGRect)frame items:(NSArray*)itemArray

{

    self = [super initWithFrame:frame];

    if (self) {

    int segmentCount = [itemArray count];

    segmentButtons = [[NSMutableArray alloc] init];

    buttonImgNames = [[NSMutableArray alloc] init];

    float segmentWidth = frame.size.width/segmentCount;
    
    for (int i=0; i<segmentCount; i++) {
        
        UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
        
        button.frame = CGRectMake(segmentWidth*i, 0,
                                  
                                  segmentWidth, frame.size.height);
        
        if (i==0) { //left
            
            [buttonImgNames addObject:@"seg_btn_left_nor.png"];
            
            [buttonImgNames addObject:@"seg_btn_left_sel.png"];
            
        }else if(i==segmentCount-1){ //right
            
            [buttonImgNames addObject:@"seg_btn_right_nor.png"];
            
            [buttonImgNames addObject:@"seg_btn_right_sel.png"];
            
        }else{ //middle
            
            
            
        }
        
        button.tag = i;
        
        [button addTarget:self action:@selector(segmentAction:) forControlEvents:UIControlEventTouchUpInside];
        
        [button setTitle:[itemArray objectAtIndex:i] forState:UIControlStateNormal];
        
        [segmentButtons addObject:button];
        
        [self addSubview:button];
        
    }
    
    [self setSegmentIndex:0];
    
}
    
    return self;
    
}


-(void)setSegmentIndex:(NSInteger)index

{
    
    _selectedSegmentIndex = index;
    
    [self segmentAction:[segmentButtons objectAtIndex:index]];
    
}


-(void)segmentAction:(id)sender

{
    
    UIButton *button = (UIButton*)sender;
    
    int tag =  button.tag;
    
    for(int i=0; i<[segmentButtons count]; i++){
        
        int nameOffset = SEGMENT_UNSELECTED;
        
        if (tag == i) {
            
            nameOffset = SEGMENT_SELECTED;
            
        }
        
        UIButton *segButton = [segmentButtons objectAtIndex:i];
        
        [segButton setBackgroundImage:[UIImage imageNamed:[buttonImgNames objectAtIndex:i*2+nameOffset]]
         
                             forState:UIControlStateNormal];
        
    }
    
}

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



@end