transmission/macosx/TorrentCell.m

332 lines
10 KiB
Objective-C

/******************************************************************************
* $Id$
*
* Copyright (c) 2006 Transmission authors and contributors
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*****************************************************************************/
#import "TorrentCell.h"
#import "StringAdditions.h"
#define BAR_HEIGHT 12
@implementation TorrentCell
/***********************************************************************
* Static tables
***********************************************************************
* We use these tables to optimize the drawing. They contain packed
* RGBA pixels for every color we might need.
**********************************************************************/
static uint32_t kBorder[] =
{ 0x00000005, 0x00000010, 0x00000015, 0x00000015,
0x00000015, 0x00000015, 0x00000015, 0x00000015,
0x00000015, 0x00000015, 0x00000010, 0x00000005 };
static uint32_t kBack[] =
{ 0xB4B4B4FF, 0xE3E3E3FF, 0xE8E8E8FF, 0xDEDEDEFF,
0xDBDBDBFF, 0xE5E5E5FF, 0xE7E7E7FF, 0xF5F5F5FF,
0xFAFAFAFF, 0xDEDEDEFF, 0x0000003F, 0x00000015 };
/* Coefficients for the "3D effect":
0.59, 0.91, 0.97, 0.92, 0.79, 0.76, 0.85, 0.93, 1.00, 0.99 */
/* 255, 100, 80 */
static uint32_t kRed[] =
{ 0x963A2FFF, 0xE85B48FF, 0xF7614DFF, 0xEA5C49FF,
0xC94F3FFF, 0xC14C3CFF, 0xD85544FF, 0xED5D4AFF,
0xFF6450FF, 0xFC634FFF, 0x0000003F, 0x00000015 };
/* 160, 220, 255 */
static uint32_t kBlue1[] =
{ 0x5E8196FF, 0x91C8E8FF, 0x9BD5F7FF, 0x93CAEAFF,
0x7EADC9FF, 0x79A7C1FF, 0x88BBD8FF, 0x94CCEDFF,
0xA0DCFFFF, 0x9ED9FCFF, 0x0000003F, 0x00000015 };
/* 120, 190, 255 */
static uint32_t kBlue2[] =
{ 0x467096FF, 0x6DACE8FF, 0x74B8F7FF, 0x6EAEEAFF,
0x5E96C9FF, 0x5B90C1FF, 0x66A1D8FF, 0x6FB0EDFF,
0x78BEFFFF, 0x76BCFCFF, 0x0000003F, 0x00000015 };
/* 80, 160, 255 */
static uint32_t kBlue3[] =
{ 0x2F5E96FF, 0x4891E8FF, 0x4D9BF7FF, 0x4993EAFF,
0x3F7EC9FF, 0x3C79C1FF, 0x4488D8FF, 0x4A94EDFF,
0x50A0FFFF, 0x4F9EFCFF, 0x0000003F, 0x00000015 };
/* 30, 70, 180 */
static uint32_t kBlue4[] =
{ 0x11296AFF, 0x1B3FA3FF, 0x1D43AEFF, 0x1B40A5FF,
0x17378EFF, 0x163588FF, 0x193B99FF, 0x1B41A7FF,
0x1E46B4FF, 0x1D45B2FF, 0x0000003F, 0x00000015 };
/* 130, 130, 130 */
static uint32_t kGray[] =
{ 0x4C4C4CFF, 0x767676FF, 0x7E7E7EFF, 0x777777FF,
0x666666FF, 0x626262FF, 0x6E6E6EFF, 0x787878FF,
0x828282FF, 0x808080FF, 0x0000003F, 0x00000015 };
/* 0, 255, 0 */
static uint32_t kGreen[] =
{ 0x009600FF, 0x00E800FF, 0x00F700FF, 0x00EA00FF,
0x00C900FF, 0x00C100FF, 0x00D800FF, 0x00ED00FF,
0x00FF00FF, 0x00FC00FF, 0x0000003F, 0x00000015 };
- (void) setTorrent: (Torrent *) torrent
{
fTorrent = torrent;
}
- (void) setTextColor: (NSColor *) color
{
fTextColor = color;
}
/***********************************************************************
* init
***********************************************************************
* Prepares the NSBitmapImageReps we are going to need in order to
* draw.
**********************************************************************/
- (id) init
{
self = [super init];
return self;
}
/***********************************************************************
* buildSimpleBar
**********************************************************************/
- (void) buildSimpleBar
{
int h, w, end, pixelsPerRow;
uint32_t * p;
uint32_t * colors;
pixelsPerRow = [fBitmap bytesPerRow] / 4;
p = (uint32_t *) [fBitmap bitmapData] + 1;
end = lrintf( floor( [fTorrent progress] * ( fWidth - 2 ) ) );
if( [fTorrent isSeeding] )
colors = kGreen;
else if( [fTorrent isActive] )
colors = kBlue2;
else
colors = kGray;
for( h = 0; h < BAR_HEIGHT; h++ )
{
for( w = 0; w < end; w++ )
{
p[w] = htonl( colors[h] );
}
for( w = end; w < fWidth - 2; w++ )
{
p[w] = htonl( kBack[h] );
}
p += pixelsPerRow;
}
}
/***********************************************************************
* buildAdvancedBar
**********************************************************************/
- (void) buildAdvancedBar
{
int h, w, end, pixelsPerRow;
uint32_t * p, * colors;
uint8_t * bitmapData = [fBitmap bitmapData];
int bytesPerRow = [fBitmap bytesPerRow];
fPieces = malloc( fWidth );
[fTorrent getAvailability: fPieces size: fWidth];
if( [fTorrent isSeeding] )
{
/* All green, same as the simple bar */
[self buildSimpleBar];
return;
}
pixelsPerRow = [fBitmap bytesPerRow] / 4;
/* First two lines: dark blue to show progression */
end = lrintf( floor( [fTorrent progress] * ( fWidth - 2 ) ) );
for( h = 0; h < 2; h++ )
{
p = (uint32_t *) ( bitmapData + h * bytesPerRow ) + 1;
for( w = 0; w < end; w++ )
{
p[w] = htonl( kBlue4[h] );
}
for( w = end; w < fWidth - 2; w++ )
{
p[w] = htonl( kBack[h] );
}
}
/* Lines 2 to 14: blue or grey depending on whether
we have the piece or not */
for( w = 0; w < fWidth - 2; w++ )
{
/* Point to pixel ( 2 + w, 2 ). We will then draw
"vertically" */
p = (uint32_t *) ( bitmapData + 2 * bytesPerRow ) + 1 + w;
if( fPieces[w] < 0 )
{
colors = kGray;
}
else if( fPieces[w] < 1 )
{
colors = kRed;
}
else if( fPieces[w] < 2 )
{
colors = kBlue1;
}
else if( fPieces[w] < 3 )
{
colors = kBlue2;
}
else
{
colors = kBlue3;
}
for( h = 2; h < BAR_HEIGHT; h++ )
{
p[0] = htonl( colors[h] );
p = (uint32_t *) ( (uint8_t *) p + bytesPerRow );
}
}
free( fPieces );
}
- (void) buildBar
{
int h;
uint32_t * p;
/* Left and right borders */
p = (uint32_t *) [fBitmap bitmapData];
for( h = 0; h < BAR_HEIGHT; h++ )
{
p[0] = htonl( kBorder[h] );
p[fWidth - 1] = htonl( kBorder[h] );
p += [fBitmap bytesPerRow] / 4;
}
/* ...and redraw the progress bar on the top of it */
if( [[NSUserDefaults standardUserDefaults]
boolForKey:@"UseAdvancedBar"])
{
[self buildAdvancedBar];
}
else
{
[self buildSimpleBar];
}
}
/***********************************************************************
* drawWithFrame
***********************************************************************
* We have the strings, we have the bitmap. Let's just draw them where
* they belong.
**********************************************************************/
- (void) drawWithFrame: (NSRect) cellFrame inView: (NSView *) view
{
if( ![view lockFocusIfCanDraw] )
{
return;
}
NSMutableDictionary * attributes;
attributes = [NSMutableDictionary dictionaryWithCapacity: 2];
[attributes setObject: fTextColor
forKey: NSForegroundColorAttributeName];
NSPoint pen = cellFrame.origin;
/* Draw the icon */
pen.x += 5; pen.y += 10;
NSImage * icon = [fTorrent icon];
[icon drawAtPoint: pen fromRect:
NSMakeRect( 0, 0, [icon size].width, [icon size].height )
operation: NSCompositeSourceOver fraction: 1.0];
NSString * string;
fWidth = NSWidth( cellFrame ) - [icon size].width - 15;
/* Draw file or folder name */
pen.x += [icon size].width + 5; pen.y -= 7;
[attributes setObject: [NSFont messageFontOfSize: 11]
forKey: NSFontAttributeName];
NSString * sizeString = [NSString stringWithFormat: @" (%@)",
[NSString stringForFileSize: [fTorrent size]]];
string = [[[fTorrent name] stringFittingInWidth: fWidth - 50 -
[sizeString sizeWithAttributes: attributes].width
withAttributes: attributes] stringByAppendingString: sizeString];
[string drawAtPoint: pen withAttributes: attributes];
/* Draw the progress bar */
pen.y += 17;
fBitmap = [[NSBitmapImageRep alloc]
initWithBitmapDataPlanes: nil pixelsWide: fWidth
pixelsHigh: BAR_HEIGHT bitsPerSample: 8 samplesPerPixel: 4
hasAlpha: YES isPlanar: NO colorSpaceName:
NSCalibratedRGBColorSpace bytesPerRow: 0 bitsPerPixel: 0];
NSImage * img = [[NSImage alloc] initWithSize: [fBitmap size]];
[img addRepresentation: fBitmap];
[img setFlipped: YES];
[self buildBar];
[img drawAtPoint: pen fromRect: NSMakeRect( 0, 0,
[img size].width, [img size].height )
operation: NSCompositeSourceOver fraction: 1.0];
[img release];
[fBitmap release];
/* Status strings */
[attributes setObject: [NSFont messageFontOfSize: 9]
forKey: NSFontAttributeName];
pen.y += BAR_HEIGHT + 2;
[[fTorrent statusString] drawAtPoint: pen withAttributes: attributes];
pen.y += 13;
string = [[fTorrent infoString] stringFittingInWidth:
( fWidth - 80 ) withAttributes: attributes];
[string drawAtPoint: pen withAttributes: attributes];
/* Rates */
pen.x += fWidth - 70; pen.y -= 13;
[[fTorrent downloadString] drawAtPoint: pen withAttributes: attributes];
pen.y += 13;
[[fTorrent uploadString] drawAtPoint: pen withAttributes: attributes];
[view unlockFocus];
}
@end