transmission/macosx/PiecesView.m

227 lines
7.1 KiB
Mathematica
Raw Normal View History

2007-09-16 01:02:06 +00:00
/******************************************************************************
* $Id$
*
2009-01-10 23:37:37 +00:00
* Copyright (c) 2006-2009 Transmission authors and contributors
2007-09-16 01:02:06 +00:00
*
* 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 "PiecesView.h"
2008-11-01 15:22:32 +00:00
#import "Torrent.h"
2007-09-16 01:02:06 +00:00
#import "InfoWindowController.h"
#import "utils.h"
2007-09-16 01:02:06 +00:00
#define MAX_ACROSS 18
#define BETWEEN 1.0
2007-09-16 01:02:06 +00:00
#define HIGH_PEERS 30
#define PIECE_NONE 0
#define PIECE_SOME 1
#define PIECE_HIGH_PEERS 2
#define PIECE_FINISHED 3
#define PIECE_FLASHING 4
2007-09-16 01:02:06 +00:00
@implementation PiecesView
- (void) awakeFromNib
{
2008-06-17 17:17:15 +00:00
//back image
fBack = [[NSImage alloc] initWithSize: [self bounds].size];
[fBack lockFocus];
2008-12-26 05:57:51 +00:00
NSGradient * gradient = [[NSGradient alloc] initWithStartingColor: [NSColor colorWithCalibratedWhite: 0.0f alpha: 0.4f]
2008-10-28 00:08:49 +00:00
endingColor: [NSColor colorWithCalibratedWhite: 0.2f alpha: 0.4f]];
2008-12-26 05:57:51 +00:00
[gradient drawInRect: [self bounds] angle: 90.0f];
[gradient release];
2008-06-17 17:17:15 +00:00
[fBack unlockFocus];
//store box colors
2008-10-28 00:08:49 +00:00
fGreenAvailabilityColor = [[NSColor colorWithCalibratedRed: 0.0f green: 1.0f blue: 0.4f alpha: 1.0f] retain];
fBluePieceColor = [[NSColor colorWithCalibratedRed: 0.0f green: 0.4f blue: 0.8f alpha: 1.0f] retain];
2008-06-17 17:17:15 +00:00
//actually draw the box
[self setTorrent: nil];
2007-09-16 01:02:06 +00:00
}
- (void) dealloc
{
2008-06-09 21:20:01 +00:00
tr_free(fPieces);
2007-09-16 01:02:06 +00:00
[fBack release];
[fGreenAvailabilityColor release];
[fBluePieceColor release];
2007-09-16 01:02:06 +00:00
[super dealloc];
}
- (void) setTorrent: (Torrent *) torrent
{
[self clearView];
2007-09-16 01:02:06 +00:00
2008-06-09 21:20:01 +00:00
fTorrent = torrent;
if (fTorrent)
2007-09-16 01:02:06 +00:00
{
//determine relevant values
2008-06-09 21:20:01 +00:00
fNumPieces = MIN([fTorrent pieceCount], MAX_ACROSS * MAX_ACROSS);
fAcross = ceil(sqrt(fNumPieces));
2007-09-16 01:02:06 +00:00
CGFloat width = [self bounds].size.width;
2007-09-16 01:02:06 +00:00
fWidth = (width - (fAcross + 1) * BETWEEN) / fAcross;
fExtraBorder = (width - ((fWidth + BETWEEN) * fAcross + BETWEEN)) / 2;
}
//reset the view to blank
NSImage * newBack = [fBack copy];
[self setImage: newBack];
[newBack release];
[self setNeedsDisplay];
2007-09-16 01:02:06 +00:00
}
- (void) clearView
2008-06-09 21:20:01 +00:00
{
tr_free(fPieces);
fPieces = NULL;
}
- (void) updateView
2007-09-16 01:02:06 +00:00
{
if (!fTorrent)
return;
2008-06-09 21:20:01 +00:00
//determine if first time
const BOOL first = fPieces == NULL;
if (first)
2008-06-09 21:20:01 +00:00
fPieces = (int8_t *)tr_malloc(fNumPieces * sizeof(int8_t));
2007-09-16 01:02:06 +00:00
2008-06-09 21:20:01 +00:00
int8_t * pieces = NULL;
float * piecesPercent = NULL;
2007-09-16 01:02:06 +00:00
const BOOL showAvailablity = [[NSUserDefaults standardUserDefaults] boolForKey: @"PiecesViewShowAvailability"];
2007-09-16 01:02:06 +00:00
if (showAvailablity)
{
2008-06-09 21:20:01 +00:00
pieces = (int8_t *)tr_malloc(fNumPieces * sizeof(int8_t));
2007-09-16 01:02:06 +00:00
[fTorrent getAvailability: pieces size: fNumPieces];
}
else
{
piecesPercent = (float *)tr_malloc(fNumPieces * sizeof(float));
2007-09-16 01:02:06 +00:00
[fTorrent getAmountFinished: piecesPercent size: fNumPieces];
}
NSImage * image = [self image];
NSRect fillRects[fNumPieces];
NSColor * fillColors[fNumPieces];
NSInteger index = -1, usedCount = 0;
for (NSInteger i = 0; i < fAcross; i++)
for (NSInteger j = 0; j < fAcross; j++)
2007-09-16 01:02:06 +00:00
{
index++;
if (index >= fNumPieces)
{
i = fAcross;
break;
}
NSColor * pieceColor = nil;
if (showAvailablity ? pieces[index] == -1 : piecesPercent[index] == 1.0)
{
if (first || fPieces[index] != PIECE_FINISHED)
2008-06-10 20:15:00 +00:00
{
if (!first && fPieces[index] != PIECE_FLASHING)
{
pieceColor = [NSColor orangeColor];
fPieces[index] = PIECE_FLASHING;
}
else
{
pieceColor = fBluePieceColor;
fPieces[index] = PIECE_FINISHED;
}
}
}
else if (showAvailablity ? pieces[index] == 0 : piecesPercent[index] == 0.0)
{
if (first || fPieces[index] != PIECE_NONE)
{
pieceColor = [NSColor whiteColor];
fPieces[index] = PIECE_NONE;
2008-06-10 20:15:00 +00:00
}
}
else if (showAvailablity && pieces[index] >= HIGH_PEERS)
{
if (first || fPieces[index] != PIECE_HIGH_PEERS)
2007-09-16 01:02:06 +00:00
{
pieceColor = fGreenAvailabilityColor;
fPieces[index] = PIECE_HIGH_PEERS;
2007-09-16 01:02:06 +00:00
}
}
else
{
//always redraw "mixed"
CGFloat percent = showAvailablity ? (CGFloat)pieces[index]/HIGH_PEERS : piecesPercent[index];
NSColor * fullColor = showAvailablity ? fGreenAvailabilityColor : fBluePieceColor;
pieceColor = [[NSColor whiteColor] blendedColorWithFraction: percent ofColor: fullColor];
fPieces[index] = PIECE_SOME;
2007-09-16 01:02:06 +00:00
}
if (pieceColor)
{
fillRects[usedCount] = NSMakeRect(j * (fWidth + BETWEEN) + BETWEEN + fExtraBorder,
[image size].width - (i + 1) * (fWidth + BETWEEN) - fExtraBorder,
fWidth, fWidth);
fillColors[usedCount] = pieceColor;
usedCount++;
}
2007-09-16 01:02:06 +00:00
}
if (usedCount > 0)
{
[image lockFocus];
NSRectFillListWithColors(fillRects, fillColors, usedCount);
[image unlockFocus];
[self setNeedsDisplay];
}
2007-09-16 01:02:06 +00:00
2008-06-09 21:20:01 +00:00
tr_free(pieces);
tr_free(piecesPercent);
2007-09-16 01:02:06 +00:00
}
- (BOOL) acceptsFirstMouse: (NSEvent *) event
{
return YES;
}
- (void) mouseDown: (NSEvent *) event
{
if (fTorrent)
[[[self window] windowController] setPiecesViewForAvailable:
![[NSUserDefaults standardUserDefaults] boolForKey: @"PiecesViewShowAvailability"]];
[super mouseDown: event];
}
@end