BlueBadge

#import <UIKit/UIKit.h>

@interface BlueBadge : UIView {

NSInteger count;

}

@property (nonatomic) NSInteger count;

- (void)drawWithCount:(NSInteger)i;

@end

#import "BlueBadge.h"

@implementation BlueBadge

@synthesize count;

- (id)initWithFrame:(CGRect)frame {

if (self = [super initWithFrame:frame]) {

// Initialization code

[self setBackgroundColor: [UIColor clearColor]];

[self setCount: 0];

}

return self;

}

- (void)drawRect:(CGRect)rect {

if (count == 0) return;

NSString *countString = [NSString stringWithFormat: @"%d", count];

UIFont *font = [UIFont boldSystemFontOfSize: 16];

CGSize numberSize = [countString sizeWithFont: font];

CGContextRef context = UIGraphicsGetCurrentContext();

float radius = numberSize.height / 2.0;

float startPoint = (rect.size.width - (numberSize.width + numberSize.height))/2.0;

CGContextSetRGBFillColor(context, 0.55, 0.6, 0.70, 1.0);

CGContextBeginPath(context);

CGContextAddArc(context, startPoint + radius, radius, radius, M_PI / 2 , 3 * M_PI / 2, NO);

CGContextAddArc(context, startPoint + radius + numberSize.width, radius, radius, 3 * M_PI / 2, M_PI / 2, NO);

CGContextClosePath(context);

CGContextFillPath(context);

[[UIColor whiteColor] set];

[countString drawInRect: CGRectMake(startPoint + radius, rect.origin.y, rect.size.width, rect.size.height) withFont: font];

}

- (void)drawWithCount:(NSInteger)i {

self.count = i;

[self setNeedsDisplay];

}

- (void)dealloc {

[super dealloc];

}

@end