/******************************************************************************
 * Copyright (c) 2005-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