UITableViewCell에 Badge 달기

//

//  MMTableViewCell.h

//

//  Created by 장영준 on 10. 7. 16..

//  Copyright 2010 Wizsoft. All rights reserved.

//

#import <UIKit/UIKit.h>

@interface MMBadgeView : UIView

{

NSUInteger width;

NSUInteger badgeNumber;

UIFont *font;

UITableViewCell *parent;

UIColor *badgeColor;

UIColor *badgeColorHighlighted;

}

@property (nonatomic, readonly) NSUInteger width;

@property (nonatomic, assign) NSUInteger badgeNumber;

@property (nonatomic, assign) UITableViewCell *parent;

@property (nonatomic, retain) UIColor *badgeColor;

@property (nonatomic, retain) UIColor *badgeColorHighlighted;

@end

@interface MMTableViewCell : UITableViewCell {

NSInteger badgeNumber;

MMBadgeView *badge;

UIColor *badgeColor;

UIColor *badgeColorHighlighted;

}

@property NSInteger badgeNumber;

@property (readonly, retain) MMBadgeView *badge;

@property (nonatomic, retain) UIColor *badgeColor;

@property (nonatomic, retain) UIColor *badgeColorHighlighted;

@end

//

//  MMTableViewCell.m

//

//  Created by 장영준 on 10. 7. 16..

//  Copyright 2010 Wizsoft. All rights reserved.

//

#import "MMTableViewCell.h"

@interface MMBadgeView ()

@property (nonatomic, retain) UIFont *font;

@property (nonatomic, assign) NSUInteger width;

@end

@implementation MMBadgeView

@synthesize width, badgeNumber, parent, badgeColor, badgeColorHighlighted;

@synthesize font;

- (id) initWithFrame:(CGRect)frame

{

if (self = [super initWithFrame:frame])

{

font = [[UIFont boldSystemFontOfSize: 14] retain];

self.backgroundColor = [UIColor clearColor];

}

return self;

}

- (void) drawRect:(CGRect)rect

{

NSString *countString = [[NSString alloc] initWithFormat:@"%d", self.badgeNumber];

CGSize numberSize = [countString sizeWithFont: font];

self.width = numberSize.width + 16;

CGRect bounds = CGRectMake(0 , 0, numberSize.width + 16 , 18);

CGContextRef context = UIGraphicsGetCurrentContext();

float radius = bounds.size.height / 2.0;

CGContextSaveGState(context);

UIColor *col;

if (parent.highlighted || parent.selected) {

if (self.badgeColorHighlighted) {

col = self.badgeColorHighlighted;

} else {

col = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:1.000];

}

} else {

if (self.badgeColor) {

col = self.badgeColor;

} else {

col = [UIColor colorWithRed:0.530 green:0.600 blue:0.738 alpha:1.000];

}

}

CGContextSetFillColorWithColor(context, [col CGColor]);

CGContextBeginPath(context);

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

CGContextAddArc(context, bounds.size.width - radius, radius, radius, 3 * M_PI / 2, M_PI / 2, NO);

CGContextClosePath(context);

CGContextFillPath(context);

CGContextRestoreGState(context);

bounds.origin.x = (bounds.size.width - numberSize.width) / 2 +0.5;

CGContextSetBlendMode(context, kCGBlendModeClear);

[countString drawInRect:bounds withFont:self.font];

[countString release];

}

- (void) dealloc

{

parent = nil;

self.font = nil;

self.badgeColor = nil;

self.badgeColorHighlighted = nil;

[font release];

[badgeColor release];

[badgeColorHighlighted release];

[super dealloc];

}

@end

@implementation MMTableViewCell

@synthesize badgeNumber, badge, badgeColor, badgeColorHighlighted;

- (id)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reuseIdentifier {

    if (self = [super initWithStyle:style reuseIdentifier:reuseIdentifier]) {

        // Initialization code

badge = [[MMBadgeView alloc] initWithFrame:CGRectZero];

badge.parent = self;

//redraw cells in accordance to accessory

float version = [[[UIDevice currentDevice] systemVersion] floatValue];

if (version <= 3.0)

[self addSubview:self.badge];

else 

[self.contentView addSubview:self.badge];

[self.badge setNeedsDisplay];

    }

    return self;

}

- (void) layoutSubviews

{

[super layoutSubviews];

if(self.badgeNumber > 0)

{

//force badges to hide on edit.

if(self.editing)

[self.badge setHidden:YES];

else

[self.badge setHidden:NO];

CGSize badgeSize = [[NSString stringWithFormat: @"%d", self.badgeNumber] sizeWithFont:[UIFont boldSystemFontOfSize: 14]];

//CGSize badgeSize = CGSizeMake(16, 16);

float version = [[[UIDevice currentDevice] systemVersion] floatValue];

CGRect badgeframe;

if (version <= 3.0)

{

badgeframe = CGRectMake(self.contentView.frame.size.width - (badgeSize.width+16), round((self.contentView.frame.size.height - 18) / 2), badgeSize.width+16, 18);

}

else

{

badgeframe = CGRectMake(self.contentView.frame.size.width - (badgeSize.width+16) - 10, round((self.contentView.frame.size.height - 22) / 2), badgeSize.width+16, 18);

}

[self.badge setFrame:badgeframe];

[badge setBadgeNumber:self.badgeNumber];

[badge setParent:self];

if ((self.textLabel.frame.origin.x + self.textLabel.frame.size.width) >= badgeframe.origin.x)

{

CGFloat badgeWidth = self.textLabel.frame.size.width - badgeframe.size.width - 10.0;

self.textLabel.frame = CGRectMake(self.textLabel.frame.origin.x, self.textLabel.frame.origin.y, badgeWidth, self.textLabel.frame.size.height);

}

if ((self.detailTextLabel.frame.origin.x + self.detailTextLabel.frame.size.width) >= badgeframe.origin.x)

{

CGFloat badgeWidth = self.detailTextLabel.frame.size.width - badgeframe.size.width - 10.0;

self.detailTextLabel.frame = CGRectMake(self.detailTextLabel.frame.origin.x, self.detailTextLabel.frame.origin.y, badgeWidth, self.detailTextLabel.frame.size.height);

}

//set badge highlighted colours or use defaults

if(self.badgeColorHighlighted)

badge.badgeColorHighlighted = self.badgeColorHighlighted;

else 

badge.badgeColorHighlighted = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:1.000];

//set badge colours or impose defaults

if(self.badgeColor)

badge.badgeColor = self.badgeColor;

else

badge.badgeColor = [UIColor colorWithRed:0.530 green:0.600 blue:0.738 alpha:1.000];

}

else

{

[self.badge setHidden:YES];

}

}

- (void)setHighlighted:(BOOL)highlighted animated:(BOOL)animated

{

[super setHighlighted:highlighted animated:animated];

[badge setNeedsDisplay];

}

- (void)setSelected:(BOOL)selected animated:(BOOL)animated

{

[super setSelected:selected animated:animated];

[badge setNeedsDisplay];

}

- (void)setEditing:(BOOL)editing animated:(BOOL)animated

{

[super setEditing:editing animated:animated];

if (editing) {

badge.hidden = YES;

[badge setNeedsDisplay];

[self setNeedsDisplay];

}

else 

{

badge.hidden = NO;

[badge setNeedsDisplay];

[self setNeedsDisplay];

}

}

- (void)dealloc {

self.badgeColor = nil;

self.badgeColorHighlighted = nil;

[badge release];

[badgeColor release];

[badgeColorHighlighted release];

    [super dealloc];

}

@end

// cell에 뱃지 삽입 샘플

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath 

{

int row = [indexPath row];

        static NSString *CellIdentifier = @"MMCategoryCell";

    

    MMTableViewCell *cell = (MMTableViewCell *)[tableView dequeueReusableCellWithIdentifier:CellIdentifier];

    if (cell == nil) {

        cell = [[[MMTableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:CellIdentifier] autorelease];

}

cell.textLabel.text = @"..............";

cell.badgeNumber = 10;

cell.accessoryType = UITableViewCellAccessoryDisclosureIndicator;

    return cell;

}