mirror of
https://github.com/transmission/transmission
synced 2024-12-25 09:13:06 +00:00
189 lines
5.2 KiB
Objective-C
189 lines
5.2 KiB
Objective-C
//
|
|
// Badger.m
|
|
// Transmission
|
|
//
|
|
// Created by Mitchell Livingston on 12/21/05.
|
|
//
|
|
|
|
#import "Badger.h"
|
|
|
|
@interface Badger (Private)
|
|
|
|
- (void) badgeString: (NSString *) string forRect: (NSRect) rect;
|
|
|
|
@end
|
|
|
|
@implementation Badger
|
|
|
|
- (id) init
|
|
{
|
|
if ((self = [super init]))
|
|
{
|
|
fBadge = [NSImage imageNamed: @"Badge"];
|
|
fDockIcon = [[NSApp applicationIconImage] copy];
|
|
fBadgedDockIcon = [fDockIcon copy];
|
|
fUploadBadge = [NSImage imageNamed: @"UploadBadge"];
|
|
fDownloadBadge = [NSImage imageNamed: @"DownloadBadge"];
|
|
|
|
NSShadow * stringShadow = [[NSShadow alloc] init];
|
|
[stringShadow setShadowOffset: NSMakeSize(2, -2)];
|
|
[stringShadow setShadowBlurRadius: 4];
|
|
|
|
fAttributes = [[NSDictionary dictionaryWithObjectsAndKeys:
|
|
[NSColor whiteColor], NSForegroundColorAttributeName,
|
|
[NSFont fontWithName: @"Helvetica-Bold" size: 28], NSFontAttributeName,
|
|
stringShadow, NSShadowAttributeName,
|
|
nil] retain];
|
|
[stringShadow release];
|
|
|
|
fCompleted = 0;
|
|
fSpeedShown = NO;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void) dealloc
|
|
{
|
|
[fDockIcon release];
|
|
[fBadgedDockIcon release];
|
|
|
|
[fAttributes release];
|
|
|
|
[super dealloc];
|
|
}
|
|
|
|
- (void) updateBadgeWithCompleted: (int) completed
|
|
uploadRate: (NSString *) uploadRate
|
|
downloadRate: (NSString *) downloadRate
|
|
{
|
|
NSImage * dockIcon = nil;
|
|
NSSize iconSize = [fDockIcon size];
|
|
|
|
//set seeding and downloading badges if there was a change
|
|
if (completed != fCompleted)
|
|
{
|
|
fCompleted = completed;
|
|
|
|
dockIcon = [fDockIcon copy];
|
|
|
|
//set completed badge to top right
|
|
if (completed > 0)
|
|
{
|
|
NSRect badgeRect;
|
|
badgeRect.size = [fBadge size];
|
|
badgeRect.origin.x = iconSize.width - badgeRect.size.width;
|
|
badgeRect.origin.y = iconSize.height - badgeRect.size.height;
|
|
|
|
[dockIcon lockFocus];
|
|
|
|
//place badge
|
|
[fBadge compositeToPoint: badgeRect.origin
|
|
operation: NSCompositeSourceOver];
|
|
|
|
//ignore shadow of badge when placing string
|
|
float badgeBottomExtra = 5.0;
|
|
badgeRect.size.height -= badgeBottomExtra;
|
|
badgeRect.origin.y += badgeBottomExtra;
|
|
|
|
//place badge text
|
|
[self badgeString: [NSString stringWithFormat: @"%d", completed]
|
|
forRect: badgeRect];
|
|
|
|
[dockIcon unlockFocus];
|
|
}
|
|
|
|
[fBadgedDockIcon release];
|
|
fBadgedDockIcon = [dockIcon copy];
|
|
}
|
|
|
|
//display upload and download rates
|
|
BOOL speedShown = NO;
|
|
if (uploadRate || downloadRate)
|
|
{
|
|
speedShown = YES;
|
|
|
|
NSRect badgeRect, stringRect;
|
|
badgeRect.size = [fUploadBadge size];
|
|
badgeRect.origin = NSZeroPoint;
|
|
|
|
//ignore shadow of badge when placing string
|
|
float badgeBottomExtra = 2.0;
|
|
stringRect = badgeRect;
|
|
stringRect.size.height -= badgeBottomExtra;
|
|
stringRect.origin.y += badgeBottomExtra;
|
|
|
|
if (!dockIcon)
|
|
dockIcon = [fBadgedDockIcon copy];
|
|
|
|
[dockIcon lockFocus];
|
|
|
|
if (uploadRate)
|
|
{
|
|
//place badge
|
|
[fUploadBadge compositeToPoint: badgeRect.origin
|
|
operation: NSCompositeSourceOver];
|
|
|
|
//place badge text
|
|
[self badgeString: uploadRate forRect: stringRect];
|
|
}
|
|
|
|
if (downloadRate)
|
|
{
|
|
//download rate above upload rate
|
|
if (uploadRate)
|
|
{
|
|
float spaceBetween = badgeRect.size.height + 2.0;
|
|
badgeRect.origin.y += spaceBetween;
|
|
stringRect.origin.y += spaceBetween;
|
|
}
|
|
|
|
//place badge
|
|
[fDownloadBadge compositeToPoint: badgeRect.origin
|
|
operation: NSCompositeSourceOver];
|
|
|
|
//place badge text
|
|
[self badgeString: downloadRate forRect: stringRect];
|
|
}
|
|
|
|
[dockIcon unlockFocus];
|
|
}
|
|
|
|
if (dockIcon || fSpeedShown)
|
|
{
|
|
if (!dockIcon)
|
|
dockIcon = [fBadgedDockIcon copy];
|
|
|
|
[NSApp setApplicationIconImage: dockIcon];
|
|
|
|
[dockIcon release];
|
|
}
|
|
|
|
fSpeedShown = speedShown;
|
|
}
|
|
|
|
- (void) clearBadge
|
|
{
|
|
[fBadgedDockIcon release];
|
|
fBadgedDockIcon = [fDockIcon copy];
|
|
|
|
[NSApp setApplicationIconImage: fDockIcon];
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation Badger (Private)
|
|
|
|
//dock icon must have locked focus
|
|
- (void) badgeString: (NSString *) string forRect: (NSRect) rect
|
|
{
|
|
NSSize stringSize = [string sizeWithAttributes: fAttributes];
|
|
|
|
//string is in center of image
|
|
rect.origin.x += (rect.size.width - stringSize.width) * 0.5;
|
|
rect.origin.y += (rect.size.height - stringSize.height) * 0.5;
|
|
|
|
[string drawAtPoint: rect.origin withAttributes: fAttributes];
|
|
}
|
|
|
|
@end
|