//

//  GrowlBubblesWindowView.m

//  Growl

//

//  Created by Nelson Elhage on Wed Jun 09 2004.

//  Name changed from KABubbleWindowView.m by Justin Burns on Fri Nov 05 2004.

//  Copyright (c) 2004-2006 The Growl Project. All rights reserved.

//

#import "GrowlBubblesWindowView.h"

#import "GrowlDefinesInternal.h"

#import "GrowlBubblesDefines.h"

#import "GrowlImageAdditions.h"

#import "GrowlBezierPathAdditions.h"

#import "NSMutableAttributedStringAdditions.h"

#import <WebKit/WebPreferences.h>

/* to get the limit pref */

#import "GrowlBubblesPrefsController.h"

/* Hardcoded geometry values */

#define PANEL_WIDTH_PX			270.0 /*!< Total width of the panel, including border */

#define BORDER_WIDTH_PX			  4.0

#define BORDER_RADIUS_PX		  9.0

#define PANEL_VSPACE_PX			 10.0 /*!< Vertical padding from bounds to content area */

#define PANEL_HSPACE_PX			 15.0 /*!< Horizontal padding from bounds to content area */

#define ICON_SIZE_PX			 32.0 /*!< The width and height of the (square) icon */

#define ICON_SIZE_LARGE_PX		 48.0 /*!< The width and height of the (square) icon */

#define ICON_HSPACE_PX			  8.0 /*!< Horizontal space between icon and title/description */

#define TITLE_VSPACE_PX			  5.0 /*!< Vertical space between title and description */

#define TITLE_FONT_SIZE_PTS		 13.0

#define DESCR_FONT_SIZE_PTS		 11.0

#define MAX_TEXT_ROWS				5  /*!< The maximum number of rows of text, used only if the limit preference is set. */

#define MIN_TEXT_HEIGHT			(PANEL_VSPACE_PX + PANEL_VSPACE_PX + iconSize)

#define TEXT_AREA_WIDTH			(PANEL_WIDTH_PX - PANEL_HSPACE_PX - PANEL_HSPACE_PX - iconSize - ICON_HSPACE_PX)

static void GrowlBubblesShadeInterpolate( void *info, const CGFloat *inData, CGFloat *outData ) {

	CGFloat *colors = (CGFloat *) info;

	register CGFloat a = inData[0];

	register CGFloat a_coeff = 1.0 - a;

	// SIMD could come in handy here

	// outData[0..3] = a_coeff * colors[4..7] + a * colors[0..3]

	outData[0] = a_coeff * colors[4] + a * colors[0];

	outData[1] = a_coeff * colors[5] + a * colors[1];

	outData[2] = a_coeff * colors[6] + a * colors[2];

	outData[3] = a_coeff * colors[7] + a * colors[3];

}

#pragma mark -

@implementation GrowlBubblesWindowView

- (id) initWithFrame:(NSRect) frame {

	if ((self = [super initWithFrame:frame])) {

		titleFont = [[NSFont boldSystemFontOfSize:TITLE_FONT_SIZE_PTS] retain];

		textFont = [[NSFont messageFontOfSize:DESCR_FONT_SIZE_PTS] retain];

		borderColor = [[NSColor colorWithCalibratedWhite:0.0 alpha:0.5] retain];

		highlightColor = [[NSColor colorWithCalibratedWhite:0.0 alpha:0.75] retain];

		textLayoutManager = [[NSLayoutManager alloc] init];

		titleLayoutManager = [[NSLayoutManager alloc] init];

		lineHeight = [textLayoutManager defaultLineHeightForFont:textFont];

		int size = GrowlBubblesSizePrefDefault;

		READ_GROWL_PREF_INT(GrowlBubblesSizePref, GrowlBubblesPrefDomain, &size);

		if (size == GrowlBubblesSizeLarge) {

			iconSize = ICON_SIZE_LARGE_PX;

		} else {

			iconSize = ICON_SIZE_PX;

		}

	}

	return self;

}

- (void) dealloc {

	[titleFont          release];

	[textFont           release];

	[icon               release];

	[textColor          release];

	[bgColor            release];

	[lightColor         release];

	[borderColor        release];

	[highlightColor     release];

	[textStorage        release];

	[titleStorage       release];

	[textLayoutManager  release];

	[titleLayoutManager release];

	[super dealloc];

}

- (CGFloat) titleHeight {

	return haveTitle ? titleHeight : 0.0;

}

- (void) drawRect:(NSRect) rect {

#pragma unused (rect)

	//Make sure that we don't draw in the main thread

	//if ([super dispatchDrawingToThread:rect]) {

		NSRect b = [self bounds];

		CGRect bounds = CGRectMake(b.origin.x, b.origin.y, b.size.width, b.size.height);

		CGRect shape = CGRectInset(bounds, BORDER_WIDTH_PX*0.5, BORDER_WIDTH_PX*0.5);

		CGContextRef context = (CGContextRef)[[NSGraphicsContext currentContext] graphicsPort];

		// Create a path with enough room to strike the border and remain inside our frame.

		// Since the path is in the middle of the line, this means we must inset it by half the border width.

		addRoundedRectToPath(context, shape, BORDER_RADIUS_PX);

		CGContextSetLineWidth(context, BORDER_WIDTH_PX);

		CGContextSaveGState(context);

		CGContextClip(context);

		// Create a callback function to generate the

		// fill clipped graphics context with our gradient

		struct CGFunctionCallbacks callbacks = { 0U, GrowlBubblesShadeInterpolate, NULL };

		CGFloat colors[8];

		[lightColor getRed:&colors[0]

					 green:&colors[1]

					  blue:&colors[2]

					 alpha:&colors[3]];

		[bgColor getRed:&colors[4]

				  green:&colors[5]

				   blue:&colors[6]

				  alpha:&colors[7]];

		CGFunctionRef function = CGFunctionCreate( (void *) colors,

												   1U,

												   /*domain*/ NULL,

												   4U,

												   /*range*/ NULL,

												   &callbacks );

		CGColorSpaceRef cspace = CGColorSpaceCreateDeviceRGB();

		CGPoint src, dst;

		src.x = CGRectGetMinX(bounds);

		src.y = CGRectGetMaxY(bounds);

		dst.x = src.x;

		dst.y = CGRectGetMinY(bounds);

		CGShadingRef shading = CGShadingCreateAxial(cspace, src, dst,

													function, false, false);

		CGContextDrawShading(context, shading);

		CGShadingRelease(shading);

		CGColorSpaceRelease(cspace);

		CGFunctionRelease(function);

		CGContextRestoreGState(context);

		addRoundedRectToPath(context, shape, BORDER_RADIUS_PX);

		CGContextSetLineWidth(context, BORDER_WIDTH_PX);

		if (mouseOver)

			[highlightColor set];

		else

			[borderColor set];

		CGContextStrokePath(context);

		NSRect drawRect;

		drawRect.origin.x = PANEL_HSPACE_PX;

		drawRect.origin.y = PANEL_VSPACE_PX;

		drawRect.size.width = iconSize;

		drawRect.size.height = iconSize;

		[icon setFlipped:YES];

		[icon drawScaledInRect:drawRect

					 operation:NSCompositeSourceOver

					  fraction:1.0];

		drawRect.origin.x += iconSize + ICON_HSPACE_PX;

		if (haveTitle) {

			[titleLayoutManager drawGlyphsForGlyphRange:titleRange atPoint:drawRect.origin];

			drawRect.origin.y += titleHeight + TITLE_VSPACE_PX;

		}

		if (haveText)

			[textLayoutManager drawGlyphsForGlyphRange:textRange atPoint:drawRect.origin];

		[[self window] invalidateShadow];

		[super drawRect:rect];

	//}

}

#pragma mark -

- (void) setPriority:(int)priority {

	NSString *key;

	NSString *textKey;

	NSString *topKey;

	switch (priority) {

		case -2:

			key = GrowlBubblesVeryLowColor;

			textKey = GrowlBubblesVeryLowTextColor;

			topKey = GrowlBubblesVeryLowTopColor;

			break;

		case -1:

			key = GrowlBubblesModerateColor;

			textKey = GrowlBubblesModerateTextColor;

			topKey = GrowlBubblesModerateTopColor;

			break;

		case 1:

			key = GrowlBubblesHighColor;

			textKey = GrowlBubblesHighTextColor;

			topKey = GrowlBubblesHighTopColor;

			break;

		case 2:

			key = GrowlBubblesEmergencyColor;

			textKey = GrowlBubblesEmergencyTextColor;

			topKey = GrowlBubblesEmergencyTopColor;

			break;

		case 0:

		default:

			key = GrowlBubblesNormalColor;

			textKey = GrowlBubblesNormalTextColor;

			topKey = GrowlBubblesNormalTopColor;

			break;

	}

	NSData *data = nil;

	CGFloat backgroundAlpha = 95.0;

	READ_GROWL_PREF_FLOAT(GrowlBubblesOpacity, GrowlBubblesPrefDomain, &backgroundAlpha);

	backgroundAlpha *= 0.01;

	Class NSDataClass = [NSData class];

	READ_GROWL_PREF_VALUE(key, GrowlBubblesPrefDomain, NSData *, &data);

	if(data)

		CFMakeCollectable(data);		

	if (data && [data isKindOfClass:NSDataClass]) {

			bgColor = [NSUnarchiver unarchiveObjectWithData:data];

			bgColor = [bgColor colorWithAlphaComponent:backgroundAlpha];

	} else {

		bgColor = [NSColor colorWithCalibratedRed:0.69412

											green:0.83147

											 blue:0.96078

											alpha:backgroundAlpha];

	}

	[bgColor retain];

	[data release];

	data = nil;

	READ_GROWL_PREF_VALUE(textKey, GrowlBubblesPrefDomain, NSData *, &data);

	if(data)

		CFMakeCollectable(data);		

	if (data && [data isKindOfClass:NSDataClass]) {

		textColor = [NSUnarchiver unarchiveObjectWithData:data];

	} else {

		textColor = [NSColor controlTextColor];

	}

	[textColor retain];

	[data release];

	data = nil;

	READ_GROWL_PREF_VALUE(topKey, GrowlBubblesPrefDomain, NSData *, &data);

	if(data)

		CFMakeCollectable(data);		

	if (data && [data isKindOfClass:NSDataClass]) {

		lightColor = [NSUnarchiver unarchiveObjectWithData:data];

		lightColor = [lightColor colorWithAlphaComponent:backgroundAlpha];

	} else {

		lightColor = [NSColor colorWithCalibratedRed:0.93725

											   green:0.96863

												blue:0.99216

											   alpha:backgroundAlpha];

	}

	[lightColor retain];

	[data release];

}

- (void) setIcon:(NSImage *) anIcon {

	[icon release];

	icon = [anIcon retain];

	[self setNeedsDisplay:YES];

}

- (void) setTitle:(NSString *) aTitle {

	haveTitle = [aTitle length] != 0;

	if (!haveTitle) {

		[self setNeedsDisplay:YES];

		return;

	}

	if (!titleStorage) {

		NSSize containerSize;

		containerSize.width = TEXT_AREA_WIDTH;

		containerSize.height = FLT_MAX;

		titleStorage = [[NSTextStorage alloc] init];

		titleContainer = [[NSTextContainer alloc] initWithContainerSize:containerSize];

		[titleLayoutManager addTextContainer:titleContainer];	// retains textContainer

		[titleContainer release];

		[titleStorage addLayoutManager:titleLayoutManager];	// retains layoutManager

		[titleContainer setLineFragmentPadding:0.0];

	}

	NSMutableParagraphStyle *paragraphStyle = [[NSParagraphStyle defaultParagraphStyle] mutableCopy];

	[paragraphStyle setLineBreakMode:NSLineBreakByTruncatingTail];

	NSDictionary *defaultAttributes = [[NSDictionary alloc] initWithObjectsAndKeys:

		titleFont,      NSFontAttributeName,

		textColor,      NSForegroundColorAttributeName,

		paragraphStyle, NSParagraphStyleAttributeName,

		nil];

	[paragraphStyle release];

	[[titleStorage mutableString] setString:aTitle];

	[titleStorage setAttributes:defaultAttributes range:NSMakeRange(0, [titleStorage length])];

	[defaultAttributes release];

	titleRange = [titleLayoutManager glyphRangeForTextContainer:titleContainer];	// force layout

	titleHeight = [titleLayoutManager usedRectForTextContainer:titleContainer].size.height;

	[self setNeedsDisplay:YES];

}

- (void) setText:(NSString *) aText {

	haveText = [aText length] != 0;

	if (!haveText) {

		[self setNeedsDisplay:YES];

		return;

	}

	if (!textStorage) {

		NSSize containerSize;

		BOOL limitPref = YES;

		READ_GROWL_PREF_BOOL(GrowlBubblesLimitPref, GrowlBubblesPrefDomain, &limitPref);

		containerSize.width = TEXT_AREA_WIDTH;

		if (limitPref)

			containerSize.height = lineHeight * MAX_TEXT_ROWS;

		else

			containerSize.height = FLT_MAX;

		textStorage = [[NSTextStorage alloc] init];

  		textContainer = [[NSTextContainer alloc] initWithContainerSize:containerSize];

		[textLayoutManager addTextContainer:textContainer];	// retains textContainer

		[textContainer release];

		[textStorage addLayoutManager:textLayoutManager];	// retains layoutManager

		[textContainer setLineFragmentPadding:0.0];

	}

	NSDictionary *defaultAttributes = [[NSDictionary alloc] initWithObjectsAndKeys:

		textFont,  NSFontAttributeName,

		textColor, NSForegroundColorAttributeName,

		nil];

	[[textStorage mutableString] setString:aText];

	[textStorage setAttributes:defaultAttributes range:NSMakeRange(0, [textStorage length])];

	[defaultAttributes release];

	textRange = [textLayoutManager glyphRangeForTextContainer:textContainer];	// force layout

	textHeight = [textLayoutManager usedRectForTextContainer:textContainer].size.height;

	[self setNeedsDisplay:YES];

}

- (void) sizeToFit {

	CGFloat height = PANEL_VSPACE_PX + PANEL_VSPACE_PX + [self titleHeight] + [self descriptionHeight];

	if (haveTitle && haveText)

		height += TITLE_VSPACE_PX;

	if (height < MIN_TEXT_HEIGHT)

		height = MIN_TEXT_HEIGHT;

	// resize the window so that it contains the tracking rect

	NSWindow *window = [self window];

	NSRect windowRect = [window frame];

	windowRect.origin.y -= height - windowRect.size.height;

	windowRect.size.height = height;

	[window setFrame:windowRect display:YES animate:YES];

	if (trackingRectTag)

		[self removeTrackingRect:trackingRectTag];

	trackingRectTag = [self addTrackingRect:[self frame] owner:self userData:NULL assumeInside:NO];

}

- (BOOL) isFlipped {

	// Coordinates are based on top left corner

    return YES;

}

- (CGFloat) descriptionHeight {

	return haveText ? textHeight : 0.0;

}

- (NSInteger) descriptionRowCount {

	NSInteger rowCount = textHeight / lineHeight;

	BOOL limitPref = YES;

	READ_GROWL_PREF_BOOL(GrowlBubblesLimitPref, GrowlBubblesPrefDomain, &limitPref);

	if (limitPref)

		return MIN(rowCount, MAX_TEXT_ROWS);

	else

		return rowCount;

}

@end