IGKMatteSegmentedControl.m

//
//  IGKMatteSegmentedControl.m
//  Ingredients
//
//  Created by Alex Gordon on 27/02/2010.
//  Written in 2010 by Fileability.
//

#import "IGKMatteSegmentedControl.h"
#import "NSShadow+MCAdditions.h"
#import "NSBezierPath+MCAdditions.h"

@interface IGKMatteSegmentedControl ()

- (float)igk_drawSegment:(NSUInteger)segment runningY:(float)runningY selected:(BOOL)isSelected;

@end


@implementation IGKMatteSegmentedControl

- (id)initWithFrame:(NSRect)frame {
    self = [super initWithFrame:frame];
    if (self) {
        // Initialization code here.
    }
    return self;
}

- (void)drawRect:(NSRect)rect
{
    rect = [self bounds];
	
	//Setting isFlipped makes the menu appear in the wrong place. So we need to fake it
	//NSAffineTransform doesn't play well with inner shadows, so we're using an image instead
	
	NSImage *image = [[NSImage alloc] initWithSize:rect.size];
	[image lockFocus];
	
	NSUInteger selseg = [self selectedSegment];
	
	int i;
	float runningY = 0.0;
	float sely = 0.0;
	for (i = 0; i < [self segmentCount]; i++)
	{
		float width = [self widthForSegment:i];
		
		//if (i != selseg)
		[self igk_drawSegment:i runningY:runningY selected:NO];
		//else
		if (i == selseg)
			sely = runningY;
		
		runningY += width - 1.0;
	}
	
	if (selseg != -1)
		[self igk_drawSegment:selseg runningY:sely selected:YES];
	
	[image unlockFocus];
	
	[image drawInRect:NSMakeRect(0, 0, rect.size.width, rect.size.height)
			  fromRect:NSZeroRect
			 operation:NSCompositeSourceOver
			  fraction:[self isActive] ? 1.0 : 0.7
	   respectFlipped:YES
				hints:nil];
}
- (float)igk_drawSegment:(NSUInteger)segment runningY:(float)runningY selected:(BOOL)isSelected
{
	BOOL isLeft = (segment == 0);
	BOOL isRight = (segment + 1 == [self segmentCount]);
	
	float width = [self widthForSegment:segment];
	
	if (isLeft)
	{
		width += 1.0;
	}
	
	width += 1.0;
	
	//Create a bezier path
	const double radius = 4.0;
	
	NSRect strokeRect = NSMakeRect(runningY, 1, width, [self bounds].size.height - 1.0);
	
	if (!isSelected)
	{
		strokeRect.size.height -= 1.0;
		NSBezierPath *bottomHighlightPath = [[self class] roundedBezierInRect:strokeRect radius:radius hasLeft:isLeft hasRight:isRight];
		[[NSColor colorWithCalibratedWhite:1.0 alpha:0.25] set];
		[bottomHighlightPath fill];

		strokeRect.origin.y += 1.0;
		NSBezierPath *strokePath = [[self class] roundedBezierInRect:strokeRect radius:radius hasLeft:isLeft hasRight:isRight];
		
		NSRect fillRect = NSInsetRect(strokeRect, 1.0, 1.0);
		NSBezierPath *fillHighlightPath = [[self class] roundedBezierInRect:fillRect radius:radius - 1 hasLeft:isLeft hasRight:isRight];
		fillRect.size.height -= 1.0;
		NSBezierPath *fillPath = [[self class] roundedBezierInRect:fillRect radius:radius - 1 hasLeft:isLeft hasRight:isRight];
		
		NSGradient *strokeGradient = [[NSGradient alloc] initWithStartingColor:[NSColor colorWithCalibratedRed:0.484 green:0.484 blue:0.484 alpha:1.000] endingColor:[NSColor colorWithCalibratedRed:0.484 green:0.484 blue:0.484 alpha:1.000]];	
		[strokeGradient drawInBezierPath:strokePath angle:90];
		
		[[NSColor colorWithCalibratedRed:0.907 green:0.907 blue:0.907 alpha:1.000] set];
		[fillHighlightPath fill];
		
		NSGradient *fillGradient = [[NSGradient alloc] initWithStartingColor:[NSColor colorWithCalibratedRed:0.728 green:0.728 blue:0.728 alpha:1.000] endingColor:[NSColor colorWithCalibratedRed:0.878 green:0.878 blue:0.878 alpha:1.000]];	
		[fillGradient drawInBezierPath:fillPath angle:90];
	}
	else
	{
		NSRect pressedRect = strokeRect;
		pressedRect.size.width -= isLeft ? 2.0 : isRight ? 0.0 : 1.0;// segment - 2.0; 
		pressedRect.size.height -= 1.0;
		pressedRect.origin.y += 1.0;
		
		NSBezierPath *pressedPath = [[self class] roundedBezierInRect:pressedRect radius:radius hasLeft:isLeft hasRight:isRight];
		
		NSShadow *insideShadow = [[NSShadow alloc] initWithColor:[NSColor colorWithCalibratedWhite:0.0 alpha:0.6] offset:NSMakeSize(0.0, -1.0) blurRadius:4.0];//;
		
		[[NSColor colorWithCalibratedRed:0.728 green:0.728 blue:0.728 alpha:1.000] set];
		
		[pressedPath fill];
				
		[pressedPath fillWithInnerShadow:insideShadow];
		
		NSGradient *pressedGradient = [[NSGradient alloc] initWithStartingColor:[NSColor colorWithCalibratedWhite:0.0 alpha:0.34]
																	endingColor:[NSColor colorWithCalibratedWhite:0.0 alpha:0.06 + 0.34]];
		
		[pressedGradient drawInBezierPath:pressedPath angle:90];
	}
	
	if (isSelected || [self selectedSegment] != segment)
	{
		NSImage *image = [self imageForSegment:segment];
		
		if (isSelected)
		{
			//*** MASSIVE HACK ***
			image = [NSImage imageNamed:[[image name] stringByAppendingString:@"_S"]];
		}
		
		if (image)
		{
			[image drawAtPoint:NSMakePoint(floor(runningY + width / 2.0 - [image size].width / 2.0 - 1.0 + segment), ceil([self bounds].size.height / 2.0 - [image size].height / 2.0 + 1.0))
					  fromRect:NSZeroRect
					 operation:NSCompositeSourceOver
					  fraction:([self isEnabledForSegment:segment] ? 1.0 : 0.5)];	
		}
	}
	
	//Calculate the Y coord of the next segment	
	return runningY + width - 1.0;
}
+ (NSBezierPath *)roundedBezierInRect:(NSRect)rect radius:(float)radius hasLeft:(BOOL)hasLeft hasRight:(BOOL)hasRight
{
	NSBezierPath *b = [NSBezierPath bezierPath];
	
	NSPoint bottomLeft = rect.origin;
	NSPoint topLeft = NSMakePoint(NSMinX(rect), NSMaxY(rect));
	NSPoint topRight = NSMakePoint(NSMaxX(rect), NSMaxY(rect));
	NSPoint bottomRight = NSMakePoint(NSMaxX(rect), NSMinY(rect));
	
	if (hasLeft)
		[b moveToPoint:NSMakePoint(NSMidX(rect), NSMaxY(rect))];
	else
		[b moveToPoint:topLeft];
	
	if (hasLeft)
		[b appendBezierPathWithArcFromPoint:topLeft toPoint:bottomLeft radius:radius];
	else
		[b lineToPoint:bottomLeft];
	
	if (hasLeft)
		[b appendBezierPathWithArcFromPoint:bottomLeft toPoint:bottomRight radius:radius];
	else
		[b appendBezierPathWithArcFromPoint:bottomLeft toPoint:bottomRight radius:radius];
	
	if (hasRight)
		[b appendBezierPathWithArcFromPoint:bottomRight toPoint:topRight radius:radius];
	else
		[b lineToPoint:bottomRight];
	
	if (hasRight)
		[b appendBezierPathWithArcFromPoint:topRight toPoint:topLeft radius:radius];
	else
		[b lineToPoint:topRight];
	
	[b closePath];
	
	return b;
}


#pragma mark Redrawing when the window becomes Active/Inactive

- (void)viewWillMoveToWindow:(NSWindow *)window
{
	//NSLog(@"viewWillMoveToWindow: %d", [self isActive]);
	
	if (window)
	{
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(windowDidBecomeMain:) name:NSWindowDidBecomeKeyNotification object:window];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(windowDidResignMain:) name:NSWindowDidResignKeyNotification object:window];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(windowDidBecomeMain:) name:NSWindowDidBecomeMainNotification object:window];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(windowDidResignMain:) name:NSWindowDidResignMainNotification object:window];
	}
	else
	{
		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSWindowDidBecomeKeyNotification object:[self window]];
		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSWindowDidResignKeyNotification object:[self window]];
		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSWindowDidBecomeMainNotification object:[self window]];
		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSWindowDidResignMainNotification object:[self window]];
	}
	
	[self setNeedsDisplay:YES];
}
- (void)viewDidMoveToWindow
{
	[self setNeedsDisplay:YES];
}
- (void)windowDidBecomeMain:(NSNotification *)notif
{	
	[self setNeedsDisplay:YES];
}
- (void)windowDidResignMain:(NSNotification *)notif
{	
	[self setNeedsDisplay:YES];
}
- (BOOL)isActive
{	
	return [[self window] isMainWindow] || ([NSStringFromClass([[self window] class]) isEqual:@"_NSFullScreenWindow"] && [[self window] isKeyWindow]);
}

@end