1 | //
|
---|
2 | // CTBadge.m
|
---|
3 | // CTWidgets
|
---|
4 | //
|
---|
5 | // Created by Chad Weider on 2/14/07.
|
---|
6 | // Copyright (c) 2007 Chad Weider.
|
---|
7 | // Some rights reserved: <http://creativecommons.org/licenses/by/2.5/>
|
---|
8 | //
|
---|
9 | // Version: 1.5
|
---|
10 |
|
---|
11 | #import "CTBadge.h"
|
---|
12 |
|
---|
13 | const float CTLargeBadgeSize = 46.;
|
---|
14 | const float CTSmallBadgeSize = 23.;
|
---|
15 | const float CTLargeLabelSize = 24.;
|
---|
16 | const float CTSmallLabelSize = 11.;
|
---|
17 |
|
---|
18 | @interface CTBadge (Private)
|
---|
19 | - (NSImage *)badgeMaskOfSize:(float)size length:(unsigned)length; //return a badge with height of <size> to fit <length> characters
|
---|
20 | - (NSAttributedString *)labelForString:(NSString *)string size:(unsigned)size; //returns appropriately attributed label string (not autoreleased)
|
---|
21 | - (CTGradient *)badgeGradient; //gradient used to fill badge mask
|
---|
22 | @end
|
---|
23 |
|
---|
24 | @implementation CTBadge
|
---|
25 |
|
---|
26 | - (id)init
|
---|
27 | {
|
---|
28 | self = [super init];
|
---|
29 |
|
---|
30 | if (self != nil)
|
---|
31 | {
|
---|
32 | badgeColor = nil;
|
---|
33 | labelColor = nil;
|
---|
34 |
|
---|
35 | [self setBadgeColor:[NSColor redColor]];
|
---|
36 | [self setLabelColor:[NSColor whiteColor]];
|
---|
37 | }
|
---|
38 | return self;
|
---|
39 | }
|
---|
40 |
|
---|
41 | - (void)dealloc
|
---|
42 | {
|
---|
43 | if(badgeColor != nil)
|
---|
44 | [badgeColor release];
|
---|
45 | if(labelColor != nil)
|
---|
46 | [labelColor release];
|
---|
47 |
|
---|
48 | [super dealloc];
|
---|
49 | }
|
---|
50 |
|
---|
51 | + (CTBadge *)systemBadge
|
---|
52 | {
|
---|
53 | id newInstance = [[[self class] alloc] init];
|
---|
54 |
|
---|
55 | return [newInstance autorelease];
|
---|
56 | }
|
---|
57 |
|
---|
58 | + (CTBadge *)badgeWithColor:(NSColor *)badgeColor labelColor:(NSColor *)labelColor;
|
---|
59 | {
|
---|
60 | id newInstance = [[[self class] alloc] init];
|
---|
61 |
|
---|
62 | [newInstance setBadgeColor:badgeColor];
|
---|
63 | [newInstance setLabelColor:labelColor];
|
---|
64 |
|
---|
65 | return [newInstance autorelease];
|
---|
66 | }
|
---|
67 | #pragma mark -
|
---|
68 |
|
---|
69 |
|
---|
70 | #pragma mark Appearance
|
---|
71 | - (void)setBadgeColor:(NSColor *)theColor;
|
---|
72 | {
|
---|
73 | if(badgeColor != nil)
|
---|
74 | [badgeColor release];
|
---|
75 |
|
---|
76 | badgeColor = theColor;
|
---|
77 | [badgeColor retain];
|
---|
78 | }
|
---|
79 | - (void)setLabelColor:(NSColor *)theColor;
|
---|
80 | {
|
---|
81 | if(labelColor != nil)
|
---|
82 | [labelColor release];
|
---|
83 |
|
---|
84 | labelColor = theColor;
|
---|
85 | [labelColor retain];
|
---|
86 | }
|
---|
87 |
|
---|
88 | - (NSColor *)badgeColor
|
---|
89 | {
|
---|
90 | return badgeColor;
|
---|
91 | }
|
---|
92 | - (NSColor *)labelColor
|
---|
93 | {
|
---|
94 | return labelColor;
|
---|
95 | }
|
---|
96 | #pragma mark -
|
---|
97 |
|
---|
98 |
|
---|
99 | #pragma mark Drawing
|
---|
100 | - (NSImage *)smallBadgeForValue:(unsigned)value //does drawing in it's own special way
|
---|
101 | {
|
---|
102 | return [self badgeOfSize:CTSmallBadgeSize forValue:value];
|
---|
103 | }
|
---|
104 |
|
---|
105 | - (NSImage *)smallBadgeForString:(NSString *)string
|
---|
106 | {
|
---|
107 | return [self badgeOfSize:CTSmallBadgeSize forString:string];
|
---|
108 | }
|
---|
109 |
|
---|
110 | - (NSImage *)largeBadgeForValue:(unsigned)value
|
---|
111 | {
|
---|
112 | return [self badgeOfSize:CTLargeBadgeSize forValue:value];
|
---|
113 | }
|
---|
114 |
|
---|
115 | - (NSImage *)largeBadgeForString:(NSString *)string
|
---|
116 | {
|
---|
117 | return [self badgeOfSize:CTLargeBadgeSize forString:string];
|
---|
118 | }
|
---|
119 |
|
---|
120 | - (NSImage *)badgeOfSize:(float)size forValue:(unsigned)value
|
---|
121 | {
|
---|
122 | return [self badgeOfSize:(float)size forString:[NSString stringWithFormat:@"%u", value]];
|
---|
123 | }
|
---|
124 |
|
---|
125 | - (NSImage *)badgeOfSize:(float)size forString:(NSString *)string
|
---|
126 | {
|
---|
127 | float scaleFactor = 1;
|
---|
128 |
|
---|
129 | if(size <= 0)
|
---|
130 | [NSException raise:NSInvalidArgumentException format:@"%@ %@: size (%f) must be positive", [self class], NSStringFromSelector(_cmd), size];
|
---|
131 | else if(size <= CTSmallBadgeSize)
|
---|
132 | scaleFactor = size/CTSmallBadgeSize;
|
---|
133 | else
|
---|
134 | scaleFactor = size/CTLargeBadgeSize;
|
---|
135 |
|
---|
136 | //Label stuff -----------------------------------------------
|
---|
137 | NSAttributedString *label;
|
---|
138 | NSSize labelSize;
|
---|
139 |
|
---|
140 | if(size <= CTSmallBadgeSize)
|
---|
141 | label = [self labelForString:string size:CTSmallLabelSize*scaleFactor];
|
---|
142 | else
|
---|
143 | label = [self labelForString:string size:CTLargeLabelSize*scaleFactor];
|
---|
144 |
|
---|
145 | labelSize = [label size];
|
---|
146 |
|
---|
147 | //Badge stuff -----------------------------------------------
|
---|
148 | NSImage *badgeImage; //this the image with the gradient fill
|
---|
149 | NSImage *badgeMask ; //we nock out this mask from the gradient
|
---|
150 |
|
---|
151 | CTGradient *badgeGradient = [self badgeGradient];
|
---|
152 |
|
---|
153 | float shadowOpacity,
|
---|
154 | shadowOffset,
|
---|
155 | shadowBlurRadius;
|
---|
156 |
|
---|
157 | int angle;
|
---|
158 |
|
---|
159 | if(size <= CTSmallBadgeSize)
|
---|
160 | {
|
---|
161 | shadowOpacity = .6;
|
---|
162 | shadowOffset = floorf(1*scaleFactor);
|
---|
163 | shadowBlurRadius = ceilf(1*scaleFactor);
|
---|
164 | }
|
---|
165 | else
|
---|
166 | {
|
---|
167 | shadowOpacity = .8;
|
---|
168 | shadowOffset = ceilf(1*scaleFactor);
|
---|
169 | shadowBlurRadius = ceilf(2*scaleFactor);
|
---|
170 | }
|
---|
171 |
|
---|
172 | if ([label length] <= 3) //Badges have different gradient angles
|
---|
173 | angle = -45;
|
---|
174 | else
|
---|
175 | angle = -30;
|
---|
176 |
|
---|
177 | badgeMask = [self badgeMaskOfSize:size length:[label length]];
|
---|
178 |
|
---|
179 | NSSize badgeSize = [badgeMask size];
|
---|
180 | NSPoint origin = NSMakePoint(shadowBlurRadius, shadowBlurRadius+shadowOffset);
|
---|
181 |
|
---|
182 | badgeImage = [[NSImage alloc] initWithSize:NSMakeSize(badgeSize.width + 2*shadowBlurRadius, //sometimes it needs more
|
---|
183 | badgeSize.height + 2*shadowBlurRadius - shadowOffset + (size <= CTSmallBadgeSize))]; //space when small
|
---|
184 |
|
---|
185 | [badgeImage lockFocus];
|
---|
186 | [badgeGradient fillRect:NSMakeRect(origin.x, origin.y, floorf(badgeSize.width), floorf(badgeSize.height)) angle:angle]; //apply the gradient
|
---|
187 | [badgeMask compositeToPoint:origin operation: NSCompositeDestinationAtop]; //knock out the badge area
|
---|
188 | [label drawInRect:NSMakeRect(origin.x+floorf((badgeSize.width-labelSize.width)/2), origin.y+floorf((badgeSize.height-labelSize.height)/2), badgeSize.width, labelSize.height)]; //draw label in center
|
---|
189 | [badgeImage unlockFocus];
|
---|
190 |
|
---|
191 |
|
---|
192 | //Final stuff -----------------------------------------------
|
---|
193 | NSImage *image = [[NSImage alloc] initWithSize:[badgeImage size]];
|
---|
194 |
|
---|
195 | [image lockFocus];
|
---|
196 | [NSGraphicsContext saveGraphicsState];
|
---|
197 | NSShadow *theShadow = [[NSShadow alloc] init];
|
---|
198 | [theShadow setShadowOffset: NSMakeSize(0,-shadowOffset)];
|
---|
199 | [theShadow setShadowBlurRadius:shadowBlurRadius];
|
---|
200 | [theShadow setShadowColor:[[NSColor blackColor] colorWithAlphaComponent:shadowOpacity]];
|
---|
201 | [theShadow set];
|
---|
202 | [theShadow release];
|
---|
203 | [badgeImage compositeToPoint:NSZeroPoint operation:NSCompositeSourceOver];
|
---|
204 | [NSGraphicsContext restoreGraphicsState];
|
---|
205 | [image unlockFocus];
|
---|
206 |
|
---|
207 | [label release];
|
---|
208 | [badgeImage release];
|
---|
209 |
|
---|
210 | return [image autorelease];
|
---|
211 | }
|
---|
212 |
|
---|
213 |
|
---|
214 | - (NSImage *)badgeOverlayImageForString:(NSString *)string insetX:(float)dx y:(float)dy;
|
---|
215 | {
|
---|
216 | NSImage *badgeImage = [self largeBadgeForString:string];
|
---|
217 | NSImage *overlayImage = [[NSImage alloc] initWithSize:NSMakeSize(128,128)];
|
---|
218 |
|
---|
219 | //draw large icon in the upper right corner of the overlay image
|
---|
220 | [overlayImage lockFocus];
|
---|
221 | NSSize badgeSize = [badgeImage size];
|
---|
222 | [badgeImage compositeToPoint:NSMakePoint(128-dx-badgeSize.width,128-dy-badgeSize.height) operation:NSCompositeSourceOver];
|
---|
223 | [overlayImage unlockFocus];
|
---|
224 |
|
---|
225 | return [overlayImage autorelease];
|
---|
226 | }
|
---|
227 |
|
---|
228 | - (void)badgeApplicationDockIconWithString:(NSString *)string insetX:(float)dx y:(float)dy;
|
---|
229 | {
|
---|
230 | NSImage *appIcon = [NSImage imageNamed:@"NSApplicationIcon"];
|
---|
231 | NSImage *badgeOverlay = [self badgeOverlayImageForString:string insetX:dx y:dy];
|
---|
232 |
|
---|
233 | //Put the appIcon underneath the badgeOverlay
|
---|
234 | [badgeOverlay lockFocus];
|
---|
235 | [appIcon compositeToPoint:NSZeroPoint operation:NSCompositeDestinationOver];
|
---|
236 | [badgeOverlay unlockFocus];
|
---|
237 |
|
---|
238 | [NSApp setApplicationIconImage:badgeOverlay];
|
---|
239 | }
|
---|
240 |
|
---|
241 | - (NSImage *)badgeOverlayImageForValue:(unsigned)value insetX:(float)dx y:(float)dy
|
---|
242 | {
|
---|
243 | return [self badgeOverlayImageForString:[NSString stringWithFormat:@"%u", value] insetX:dx y:dy];
|
---|
244 | }
|
---|
245 |
|
---|
246 | - (void)badgeApplicationDockIconWithValue:(unsigned)value insetX:(float)dx y:(float)dy
|
---|
247 | {
|
---|
248 | [self badgeApplicationDockIconWithString:[NSString stringWithFormat:@"%u", value] insetX:dx y:dy];
|
---|
249 | }
|
---|
250 | #pragma mark -
|
---|
251 |
|
---|
252 |
|
---|
253 | #pragma mark Misc.
|
---|
254 | - (CTGradient *)badgeGradient
|
---|
255 | {
|
---|
256 | CTGradient *aGradient = [CTGradient gradientWithBeginningColor:[self badgeColor] endingColor:[[self badgeColor] shadowWithLevel:(1./3.)]];
|
---|
257 |
|
---|
258 | aGradient = [aGradient addColorStop:[self badgeColor] atPosition:(1./3.)];
|
---|
259 |
|
---|
260 | return aGradient;
|
---|
261 | }
|
---|
262 |
|
---|
263 | - (NSAttributedString *)labelForString:(NSString *)label size:(unsigned)size
|
---|
264 | {
|
---|
265 | //set Attributes to use on String ---------------------------
|
---|
266 | NSFont *labelFont;
|
---|
267 |
|
---|
268 | if(size <= CTSmallLabelSize)
|
---|
269 | labelFont = [NSFont boldSystemFontOfSize:size];
|
---|
270 | else
|
---|
271 | labelFont = [NSFont fontWithName:@"Helvetica-Bold" size:size];
|
---|
272 |
|
---|
273 | NSMutableParagraphStyle *pStyle = [[NSMutableParagraphStyle alloc] init];[pStyle setAlignment:NSCenterTextAlignment];
|
---|
274 | NSDictionary *attributes = [[NSDictionary alloc] initWithObjectsAndKeys:[self labelColor], NSForegroundColorAttributeName,
|
---|
275 | labelFont , NSFontAttributeName , nil];
|
---|
276 | [pStyle release];
|
---|
277 |
|
---|
278 | //Label stuff
|
---|
279 | if([label length] >= 6)
|
---|
280 | label = [NSString stringWithUTF8String:"\xe2\x88\x9e"];
|
---|
281 |
|
---|
282 | NSAttributedString *attributedString = [[NSAttributedString alloc] initWithString:label attributes:attributes];
|
---|
283 | [attributes release];
|
---|
284 |
|
---|
285 | return attributedString;
|
---|
286 | }
|
---|
287 |
|
---|
288 | - (NSImage *)badgeMaskOfSize:(float)size length:(unsigned)length;
|
---|
289 | {
|
---|
290 | NSImage *badgeMask;
|
---|
291 |
|
---|
292 | if(length <=2)
|
---|
293 | badgeMask = [NSImage imageNamed:@"CTBadge_1.pdf"];
|
---|
294 | else if(length <=3)
|
---|
295 | badgeMask = [NSImage imageNamed:@"CTBadge_3.pdf"];
|
---|
296 | else if(length <=4)
|
---|
297 | badgeMask = [NSImage imageNamed:@"CTBadge_4.pdf"];
|
---|
298 | else
|
---|
299 | badgeMask = [NSImage imageNamed:@"CTBadge_5.pdf"];
|
---|
300 |
|
---|
301 | if(size > 0 && size != [badgeMask size].height)
|
---|
302 | {
|
---|
303 | [badgeMask setName:nil];
|
---|
304 | [badgeMask setScalesWhenResized:YES];
|
---|
305 | [badgeMask setSize:NSMakeSize([badgeMask size].width*(size/[badgeMask size].height), size)];
|
---|
306 | }
|
---|
307 |
|
---|
308 | return badgeMask;
|
---|
309 | }
|
---|
310 |
|
---|
311 | @end
|
---|