#import "DZNSegmentedControl.h"

@interface DZNSegmentedControl ()

@property (nonatomic) BOOL initializing;

@property (nonatomic, strong) UIView *selectionIndicator;

@property (nonatomic, strong) UIView *hairline;

@property (nonatomic, getter = isTransitioning) BOOL transitioning;

@implementation DZNSegmentedControl

@synthesize items = _items;

@synthesize selectedSegmentIndex = _selectedSegmentIndex;

@synthesize barPosition = _barPosition;

- (id)init

{

_initializing = YES;

if (self = [super init]) {

_selectedSegmentIndex = -1;

_font = [UIFont systemFontOfSize:1];

_height = 56.0;

_selectionIndicatorHeight = 2.0;

_animationDuration = 0.2;

_displayCount = YES;

_autoAdjustSelectionIndicatorWidth = YES;

_selectionIndicator = [UIView new];

[self addSubview:_selectionIndicator];

_hairline = [UIView new];

_hairline.backgroundColor = [UIColor lightGrayColor];

[self addSubview:_hairline];

self.backgroundColor = [UIColor whiteColor];

}

_initializing = NO;

return self;

}

- (id)initWithItems:(NSArray *)items

{

if (self = [self init]) {

self.items = items;

}

return self;

}

#pragma mark - UIView Methods

- (CGSize)sizeThatFits:(CGSize)size

{

return CGSizeMake(self.superview.frame.size.width, _height);

}

- (void)sizeToFit

{

CGRect rect = self.frame;

rect.size = [self sizeThatFits:rect.size];

self.frame = rect;

}

- (void)layoutSubviews

{

[super layoutSubviews];

[self sizeToFit];

if ([self buttons].count == 0) {

_selectedSegmentIndex = -1;

}

else if (_selectedSegmentIndex < 0) {

_selectedSegmentIndex = 0;

}

for (int i = 0; i < [self buttons].count; i++) {

UIButton *button = [[self buttons] objectAtIndex:i];

[button setFrame:CGRectMake(roundf(self.frame.size.width/self.numberOfSegments)*i, 0, roundf(self.frame.size.width/self.numberOfSegments), self.frame.size.height)];

CGFloat topInset = (_barPosition > UIBarPositionBottom) ? -4.0 : 4.0;

[button setTitleEdgeInsets:UIEdgeInsetsMake(0, 0, topInset, 0)];

if (i == _selectedSegmentIndex) {

button.selected = YES;

}

}

_selectionIndicator.frame = [self selectionIndicatorRect];

_hairline.frame = [self hairlineRect];

[self bringSubviewToFront:_selectionIndicator];

}

- (void)willMoveToSuperview:(UIView *)newSuperview

{

[super willMoveToSuperview:newSuperview];

[self layoutIfNeeded];

}

- (void)didMoveToWindow

{

if (!self.backgroundColor) {

self.backgroundColor = [UIColor whiteColor];

}

}

#pragma mark - Getter Methods

- (NSUInteger)numberOfSegments

{

return _items.count;

}

- (NSArray *)buttons

{

NSMutableArray *_buttons = [NSMutableArray new];

for (UIView *view in self.subviews) {

if ([view isKindOfClass:[UIButton class]]) {

[_buttons addObject:view];

}

}

return _buttons;

}

- (UIButton *)buttonAtIndex:(NSUInteger)segment

{

if (_items.count > 0 && segment < [self buttons].count) {

return (UIButton *)[[self buttons] objectAtIndex:segment];

}

return nil;

}

- (UIButton *)selectedButton

{

if (_selectedSegmentIndex >= 0) {

return [self buttonAtIndex:_selectedSegmentIndex];

}

return nil;

}

- (NSString *)stringForSegmentAtIndex:(NSUInteger)segment

{

UIButton *button = [self buttonAtIndex:segment];

return [[button attributedTitleForState:UIControlStateNormal] string];

}

- (NSString *)titleForSegmentAtIndex:(NSUInteger)segment

{

if (_displayCount) {

NSString *title = [self stringForSegmentAtIndex:segment];

NSArray *components = [title componentsSeparatedByString:@"\n"];

return [components objectAtIndex:1];

}

return [self stringForSegmentAtIndex:segment];

}

- (NSNumber *)countForSegmentAtIndex:(NSUInteger)segment

{

NSString *title = [self stringForSegmentAtIndex:segment];

NSArray *components = [title componentsSeparatedByString:@"\n"];

return @([[components firstObject] intValue]);

}

- (CGRect)selectionIndicatorRect

{

UIButton *button = [self selectedButton];

NSString *title = [self titleForSegmentAtIndex:button.tag];

CGRect frame = _selectionIndicator.frame;

frame.origin.y = (_barPosition > UIBarPositionBottom) ? 0.0 : (button.frame.size.height-frame.size.height);

if (_autoAdjustSelectionIndicatorWidth) {

frame.size = CGSizeMake([title sizeWithAttributes:nil].width, _selectionIndicatorHeight);

frame.origin.x = (button.frame.size.width*(_selectedSegmentIndex))+(button.frame.size.width-frame.size.width)/2;

} else {

frame.size = CGSizeMake(button.frame.size.width, _selectionIndicatorHeight);

frame.origin.x = (button.frame.size.width*(_selectedSegmentIndex));

}

return frame;

}

- (UIColor *)hairlineColor

{

return _hairline.backgroundColor;

}

- (CGRect)hairlineRect

{

CGRect frame = CGRectMake(0, 0, self.frame.size.width, 0.5);

frame.origin.y = (_barPosition > UIBarPositionBottom) ? 0 : self.frame.size.height;

return frame;

}

#pragma mark - Setter Methods

- (void)setTintColor:(UIColor *)color

{

if (!color || !_items || _initializing) {

return;

}

[super setTintColor:color];

[self setTitleColor:color forState:UIControlStateHighlighted];

[self setTitleColor:color forState:UIControlStateSelected];

}

- (void)setItems:(NSArray *)items

{

if (_items) {

[self removeAllSegments];

}

if (items) {

_items = [NSArray arrayWithArray:items];

[self configure];

}

}

- (void)setDelegate:(id<DZNSegmentedControlDelegate>)delegate

{

_delegate = delegate;

_barPosition = [delegate positionForBar:self];

}

- (void)setSelectedSegmentIndex:(NSInteger)segment

{

if (segment > self.numberOfSegments-1) {

segment = 0;

}

[self setSelected:YES forSegmentAtIndex:segment];

}

- (void)setTitle:(NSString *)title forSegmentAtIndex:(NSUInteger)segment

{

if (!title) {

return;

}

NSAssert(segment >= 0, @"Cannot assign a title to a negative segment.");

NSMutableArray *items = [NSMutableArray arrayWithArray:self.items];

if (segment >= self.numberOfSegments) {

[items insertObject:title atIndex:self.numberOfSegments];

_items = items;

[self addButtonForSegment:segment];

}

else {

[items replaceObjectAtIndex:segment withObject:title];

_items = items;

[self setCount:[self countForSegmentAtIndex:segment] forSegmentAtIndex:segment];

}

}

- (void)setCount:(NSNumber *)count forSegmentAtIndex:(NSUInteger)segment

{

if (!count || !_items) {

return;

}

NSAssert(segment < self.numberOfSegments, @"Cannot assign a count to non-existing segment.");

NSAssert(segment >= 0, @"Cannot assign a title to a negative segment.");

NSString *text = [NSString stringWithFormat:@"%@\n%@", count ,[_items objectAtIndex:segment]];

NSMutableAttributedString *attributedString = [[NSMutableAttributedString alloc] initWithString:text];

NSString* title;

if (_displayCount) {

title = [NSString stringWithFormat:@"%@\n%@", count ,[_items objectAtIndex:segment]];

}

else {

title = [NSString stringWithFormat:@"%@",[_items objectAtIndex:segment]];

}

NSMutableAttributedString *string = [[NSMutableAttributedString alloc] initWithString:title];

[string addAttribute:NSFontAttributeName value:[UIFont fontWithName:_font.fontName size:19.0] range:[title rangeOfString:[count stringValue]]];

UIButton *button = [self buttonAtIndex:segment];

[button setAttributedTitle:attributedString forState:UIControlStateNormal];

[button setAttributedTitle:attributedString forState:UIControlStateHighlighted];

[button setAttributedTitle:attributedString forState:UIControlStateSelected];

[button setAttributedTitle:attributedString forState:UIControlStateDisabled];

[self setTitleColor:[UIColor darkGrayColor] forState:UIControlStateNormal];

[self setTitleColor:self.tintColor forState:UIControlStateHighlighted];

[self setTitleColor:self.tintColor forState:UIControlStateSelected];

[self setTitleColor:[UIColor lightGrayColor] forState:UIControlStateDisabled];

}

- (void)setTitleColor:(UIColor *)color forState:(UIControlState)state

{

for (UIButton *button in [self buttons]) {

NSMutableAttributedString *attributedString = [[NSMutableAttributedString alloc] initWithAttributedString:[button attributedTitleForState:state]];

NSString *string = attributedString.string;

NSMutableParagraphStyle *style = [[NSMutableParagraphStyle alloc] init];

style.alignment = NSTextAlignmentCenter;

style.lineBreakMode = NSLineBreakByWordWrapping;

style.minimumLineHeight = 16.0;

NSArray *components = [attributedString.string componentsSeparatedByString:@"\n"];

NSString *count = [components objectAtIndex:0];

NSString *title = [components objectAtIndex:1];

[attributedString addAttribute:NSParagraphStyleAttributeName value:style range:NSMakeRange(0, string.length)];

[attributedString addAttribute:NSFontAttributeName value:[UIFont fontWithName:_font.fontName size:19.0] range:[string rangeOfString:count]];

[attributedString addAttribute:NSFontAttributeName value:[UIFont fontWithName:_font.fontName size:12.0] range:[string rangeOfString:title]];

if (_displayCount) {

if (state == UIControlStateNormal) {

[attributedString addAttribute:NSForegroundColorAttributeName value:color range:NSMakeRange(0, count.length)];

[attributedString addAttribute:NSForegroundColorAttributeName value:[color colorWithAlphaComponent:0.5] range:NSMakeRange(count.length, title.length+1)];

}

else {

[attributedString addAttribute:NSForegroundColorAttributeName value:color range:NSMakeRange(0, string.length)];

if (state == UIControlStateSelected) {

_selectionIndicator.backgroundColor = color;

}

}

} else {

[attributedString addAttribute:NSForegroundColorAttributeName value:color range:NSMakeRange(0, attributedString.string.length)];

}

[button setAttributedTitle:attributedString forState:state];

}

}

- (void)setSelected:(BOOL)selected forSegmentAtIndex:(NSUInteger)segment

{

if (_selectedSegmentIndex == segment || self.isTransitioning) {

return;

}

for (UIButton *_button in [self buttons]) {

_button.highlighted = NO;

_button.selected = NO;

_button.userInteractionEnabled = YES;

}

CGFloat duration = (_selectedSegmentIndex < 0) ? 0.0 : _animationDuration;

_selectedSegmentIndex = segment;

_transitioning = YES;

UIButton *button = [self buttonAtIndex:segment];

[UIView animateWithDuration:duration

delay:0

options:UIViewAnimationOptionBeginFromCurrentState|UIViewAnimationOptionCurveEaseInOut

animations:^{

_selectionIndicator.frame = [self selectionIndicatorRect];

}

completion:^(BOOL finished) {

button.userInteractionEnabled = NO;

_transitioning = NO;

}];

[self sendActionsForControlEvents:UIControlEventValueChanged];

}

- (void)setEnabled:(BOOL)enabled forSegmentAtIndex:(NSUInteger)segment

{

UIButton *button = [self buttonAtIndex:segment];

button.enabled = enabled;

}

- (void)setHairlineColor:(UIColor *)color

{

if (_initializing) {

return;

}

_hairline.backgroundColor = color;

}

#pragma mark - DZNSegmentedControl Methods

- (void)configure

{

for (int i = 0; i < self.numberOfSegments; i++) {

[self addButtonForSegment:i];

}

}

- (void)addButtonForSegment:(NSUInteger)segment

{

UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];

[button addTarget:self action:@selector(willSelectedButton:) forControlEvents:UIControlEventTouchDown];

[button addTarget:self action:@selector(didSelectedButton:) forControlEvents:UIControlEventTouchDragOutside|UIControlEventTouchDragInside|UIControlEventTouchDragEnter|UIControlEventTouchDragExit|UIControlEventTouchCancel|UIControlEventTouchUpInside|UIControlEventTouchUpOutside];

button.backgroundColor = nil;

button.opaque = YES;

button.clipsToBounds = YES;

button.titleLabel.numberOfLines = 2;

button.adjustsImageWhenHighlighted = NO;

button.adjustsImageWhenDisabled = NO;

button.exclusiveTouch = YES;

button.tag = segment;

[self addSubview:button];

[self setCount:@(0) forSegmentAtIndex:segment];

[self layoutIfNeeded];

}

- (void)willSelectedButton:(id)sender

{

UIButton *button = (UIButton *)sender;

if (!self.isTransitioning) {

self.selectedSegmentIndex = button.tag;

}

}

- (void)didSelectedButton:(id)sender

{

UIButton *button = (UIButton *)sender;

button.highlighted = NO;

button.selected = YES;

}

- (void)removeAllSegments

{

if (self.isTransitioning) {

return;

}

for (UIButton *_button in [self buttons]) {

[_button removeFromSuperview];

}

_items = nil;

}