2023-02-11 20:49:42 +00:00
|
|
|
// This file Copyright © 2007-2023 Transmission authors and contributors.
|
2022-01-20 18:27:56 +00:00
|
|
|
// It may be used under the MIT (SPDX: MIT) license.
|
|
|
|
// License text can be found in the licenses/ folder.
|
2007-11-01 03:20:29 +00:00
|
|
|
|
|
|
|
#import "BadgeView.h"
|
|
|
|
#import "NSStringAdditions.h"
|
|
|
|
|
2022-10-19 19:28:21 +00:00
|
|
|
static CGFloat const kBetweenPadding = 2.0;
|
2007-12-26 16:28:13 +00:00
|
|
|
|
2022-01-24 01:32:45 +00:00
|
|
|
@interface BadgeView ()
|
2007-11-01 03:20:29 +00:00
|
|
|
|
2022-02-22 16:04:20 +00:00
|
|
|
@property(nonatomic) NSMutableDictionary* fAttributes;
|
|
|
|
|
|
|
|
@property(nonatomic) CGFloat fDownloadRate;
|
|
|
|
@property(nonatomic) CGFloat fUploadRate;
|
|
|
|
|
2007-11-01 03:20:29 +00:00
|
|
|
@end
|
|
|
|
|
|
|
|
@implementation BadgeView
|
|
|
|
|
2022-12-21 20:21:16 +00:00
|
|
|
- (instancetype)init
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2011-03-03 03:53:21 +00:00
|
|
|
if ((self = [super init]))
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2022-02-22 16:04:20 +00:00
|
|
|
_fDownloadRate = 0.0;
|
|
|
|
_fUploadRate = 0.0;
|
2007-11-01 03:20:29 +00:00
|
|
|
}
|
|
|
|
return self;
|
|
|
|
}
|
|
|
|
|
2021-08-15 09:41:48 +00:00
|
|
|
- (BOOL)setRatesWithDownload:(CGFloat)downloadRate upload:(CGFloat)uploadRate
|
2008-04-24 21:24:42 +00:00
|
|
|
{
|
|
|
|
//only needs update if the badges were displayed or are displayed now
|
2022-02-22 16:04:20 +00:00
|
|
|
if (self.fDownloadRate == downloadRate && self.fUploadRate == uploadRate)
|
2021-08-15 09:41:48 +00:00
|
|
|
{
|
2009-01-03 06:06:21 +00:00
|
|
|
return NO;
|
2021-08-15 09:41:48 +00:00
|
|
|
}
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2022-02-22 16:04:20 +00:00
|
|
|
self.fDownloadRate = downloadRate;
|
|
|
|
self.fUploadRate = uploadRate;
|
2009-01-03 06:06:21 +00:00
|
|
|
return YES;
|
2008-04-24 21:24:42 +00:00
|
|
|
}
|
|
|
|
|
2021-08-15 09:41:48 +00:00
|
|
|
- (void)drawRect:(NSRect)rect
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2021-10-31 15:18:27 +00:00
|
|
|
[NSApp.applicationIconImage drawInRect:rect fromRect:NSZeroRect operation:NSCompositingOperationSourceOver fraction:1.0];
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2022-02-22 16:04:20 +00:00
|
|
|
BOOL const upload = self.fUploadRate >= 0.1;
|
|
|
|
BOOL const download = self.fDownloadRate >= 0.1;
|
2009-12-12 18:29:14 +00:00
|
|
|
CGFloat bottom = 0.0;
|
2022-11-07 19:21:49 +00:00
|
|
|
if (download)
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2022-11-07 19:21:49 +00:00
|
|
|
NSImage* downloadBadge = [NSImage imageNamed:@"DownloadBadge"];
|
|
|
|
[self badge:downloadBadge string:[NSString stringForSpeedAbbrev:self.fDownloadRate] atHeight:bottom];
|
2022-02-22 16:04:20 +00:00
|
|
|
|
2022-11-07 19:21:49 +00:00
|
|
|
if (upload)
|
2021-08-15 09:41:48 +00:00
|
|
|
{
|
2022-11-07 19:21:49 +00:00
|
|
|
bottom += downloadBadge.size.height + kBetweenPadding; //upload rate above download rate
|
2021-08-15 09:41:48 +00:00
|
|
|
}
|
2007-11-01 03:20:29 +00:00
|
|
|
}
|
2022-11-07 19:21:49 +00:00
|
|
|
if (upload)
|
2021-08-15 09:41:48 +00:00
|
|
|
{
|
2022-11-07 19:21:49 +00:00
|
|
|
[self badge:[NSImage imageNamed:@"UploadBadge"] string:[NSString stringForSpeedAbbrev:self.fUploadRate] atHeight:bottom];
|
2021-08-15 09:41:48 +00:00
|
|
|
}
|
2007-11-01 03:20:29 +00:00
|
|
|
}
|
|
|
|
|
2022-01-25 17:45:06 +00:00
|
|
|
- (void)badge:(NSImage*)badge string:(NSString*)string atHeight:(CGFloat)height
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2022-02-22 16:04:20 +00:00
|
|
|
if (!self.fAttributes)
|
2007-11-01 03:20:29 +00:00
|
|
|
{
|
2021-08-15 09:41:48 +00:00
|
|
|
NSShadow* stringShadow = [[NSShadow alloc] init];
|
2021-08-07 07:27:56 +00:00
|
|
|
stringShadow.shadowOffset = NSMakeSize(2.0, -2.0);
|
|
|
|
stringShadow.shadowBlurRadius = 4.0;
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2022-02-22 16:04:20 +00:00
|
|
|
self.fAttributes = [[NSMutableDictionary alloc] initWithCapacity:3];
|
|
|
|
self.fAttributes[NSForegroundColorAttributeName] = NSColor.whiteColor;
|
|
|
|
self.fAttributes[NSShadowAttributeName] = stringShadow;
|
2007-11-01 03:20:29 +00:00
|
|
|
}
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2009-10-03 14:10:52 +00:00
|
|
|
NSRect badgeRect;
|
2021-08-07 07:27:56 +00:00
|
|
|
badgeRect.size = badge.size;
|
2009-12-12 18:52:03 +00:00
|
|
|
badgeRect.origin.x = 0.0;
|
2007-12-05 18:23:51 +00:00
|
|
|
badgeRect.origin.y = height;
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2021-10-31 15:18:27 +00:00
|
|
|
[badge drawInRect:badgeRect fromRect:NSZeroRect operation:NSCompositingOperationSourceOver fraction:1.0];
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2009-12-12 18:29:14 +00:00
|
|
|
//make sure text fits on the badge
|
|
|
|
CGFloat fontSize = 26.0;
|
|
|
|
NSSize stringSize;
|
|
|
|
do
|
|
|
|
{
|
2022-02-22 16:04:20 +00:00
|
|
|
self.fAttributes[NSFontAttributeName] = [NSFont boldSystemFontOfSize:fontSize];
|
|
|
|
stringSize = [string sizeWithAttributes:self.fAttributes];
|
2009-12-12 18:29:14 +00:00
|
|
|
fontSize -= 1.0;
|
|
|
|
} while (NSWidth(badgeRect) < stringSize.width);
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2009-12-12 18:29:14 +00:00
|
|
|
//string is in center of image
|
2009-12-12 18:52:03 +00:00
|
|
|
NSRect stringRect;
|
2009-12-26 00:02:20 +00:00
|
|
|
stringRect.origin.x = NSMidX(badgeRect) - stringSize.width * 0.5;
|
2022-01-25 17:45:06 +00:00
|
|
|
stringRect.origin.y = NSMidY(badgeRect) - stringSize.height * 0.5 + 1.0; //adjust for shadow
|
2007-12-05 18:23:51 +00:00
|
|
|
stringRect.size = stringSize;
|
2017-01-24 17:53:16 +00:00
|
|
|
|
2022-02-22 16:04:20 +00:00
|
|
|
[string drawInRect:stringRect withAttributes:self.fAttributes];
|
2007-11-01 03:20:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@end
|