PGThumbnailBrowser.m

/* Copyright © 2007-2009, The Sequential Project
All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
    * Redistributions of source code must retain the above copyright
      notice, this list of conditions and the following disclaimer.
    * Redistributions in binary form must reproduce the above copyright
      notice, this list of conditions and the following disclaimer in the
      documentation and/or other materials provided with the distribution.
    * Neither the name of the the Sequential Project nor the
      names of its contributors may be used to endorse or promote products
      derived from this software without specific prior written permission.

THIS SOFTWARE IS PROVIDED BY THE SEQUENTIAL PROJECT ''AS IS'' AND ANY
EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE SEQUENTIAL PROJECT BE LIABLE FOR ANY
DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
#import "PGThumbnailBrowser.h"

// Views
#import "PGBezelPanel.h"

// Other Sources
#import "PGFoundationAdditions.h"

@interface PGThumbnailBrowser(Private)

- (void)_addColumnWithItem:(id)item;

@end

@implementation PGThumbnailBrowser

#pragma mark -PGThumbnailBrowser

@synthesize dataSource;
- (void)setDataSource:(NSObject<PGThumbnailBrowserDataSource, PGThumbnailViewDataSource> *)obj
{
	if(obj == dataSource) return;
	dataSource = obj;
	[[self views] makeObjectsPerformSelector:@selector(setDataSource:) withObject:obj];
}
@synthesize delegate;
@synthesize thumbnailOrientation = _thumbnailOrientation;
- (void)setThumbnailOrientation:(PGOrientation)orientation
{
	_thumbnailOrientation = orientation;
	NSUInteger i = [self numberOfColumns];
	while(i--) [[self viewAtIndex:i] setThumbnailOrientation:orientation];
}
- (NSSet *)selection
{
	PGThumbnailView *const lastView = [[self views] lastObject];
	NSSet *const selection = [lastView selection];
	if([selection count]) return selection;
	id const item = [lastView representedObject];
	return item ? [NSSet setWithObject:item] : nil;
}
- (void)setSelection:(NSSet *)aSet
{
	++_updateCount;
	NSUInteger const initialNumberOfColumns = [self numberOfColumns];
	if(!initialNumberOfColumns) [self _addColumnWithItem:nil];
	else [[self viewAtIndex:0] reloadData];

	NSMutableArray *const path = [NSMutableArray array];
	id obj = [aSet anyObject];
	while((obj = [[self dataSource] thumbnailBrowser:self parentOfItem:obj])) [path insertObject:obj atIndex:0];

	NSUInteger i = 0;
	for(; i < [path count]; i++) {
		PGThumbnailView *const view = [self viewAtIndex:i];
		id const item = [path objectAtIndex:i];
		NSParameterAssert([[self dataSource] thumbnailBrowser:self itemCanHaveChildren:item]);
		[view setSelection:[NSSet setWithObject:item]];
		if(i + 1 < [self numberOfColumns]) {
			PGThumbnailView *const nextView = [self viewAtIndex:i + 1];
			[nextView setRepresentedObject:item];
			[nextView reloadData];
		} else [self _addColumnWithItem:item];
	}

	PGThumbnailView *const lastView = [self viewAtIndex:i];
	[self removeColumnsAfterView:lastView];
	if([lastView representedObject] == [path lastObject]) [lastView setSelection:aSet];

	--_updateCount;
	if(!_updateCount) {
		[[self window] makeFirstResponder:[[self views] lastObject]];
		[[self delegate] thumbnailBrowser:self numberOfColumnsDidChangeFrom:initialNumberOfColumns];
	}
	if([self numberOfColumns] > initialNumberOfColumns) [self scrollToLastColumnAnimate:YES];
}

#pragma mark -

- (void)redisplayItem:(id)item recursively:(BOOL)flag
{
	if(flag) return [self setNeedsDisplay:YES];
	id const parent = [[self dataSource] thumbnailBrowser:self parentOfItem:item];
	for(PGThumbnailView *const view in [self views]) {
		id const rep = [view representedObject];
		if(rep == parent) {
			NSUInteger const i = [[view items] indexOfObjectIdenticalTo:item];
			if(NSNotFound != i) [view setNeedsDisplayInRect:[view frameOfItemAtIndex:i withMargin:YES]];
		}
	}
}

#pragma mark -PGThumbnailBrowser(Private)

- (void)_addColumnWithItem:(id)item
{
	if(item && dataSource && ![dataSource thumbnailBrowser:self itemCanHaveChildren:item]) return;
	PGThumbnailView *const thumbnailView = [[[PGThumbnailView alloc] init] autorelease];
	[thumbnailView setDataSource:[self dataSource]];
	[thumbnailView setDelegate:self];
	[thumbnailView setRepresentedObject:item];
	[thumbnailView setThumbnailOrientation:[self thumbnailOrientation]];
	[thumbnailView reloadData];
	if(![self numberOfColumns]) [self setColumnWidth:NSWidth([thumbnailView frame])];
	[self addColumnWithView:thumbnailView];
}

#pragma mark -PGColumnView

- (void)insertColumnWithView:(NSView *)aView atIndex:(NSUInteger)index
{
	NSUInteger const columns = [self numberOfColumns];
	[super insertColumnWithView:aView atIndex:index];
	if(!_updateCount) [[self delegate] thumbnailBrowser:self numberOfColumnsDidChangeFrom:columns];
}
- (void)removeColumnsAfterView:(NSView *)aView
{
	NSUInteger const columns = [self numberOfColumns];
	[super removeColumnsAfterView:aView];
	if(!_updateCount) [[self delegate] thumbnailBrowser:self numberOfColumnsDidChangeFrom:columns];
}

#pragma mark -NSResponder

- (IBAction)moveLeft:(id)sender
{
	NSUInteger const i = [[self views] indexOfObjectIdenticalTo:[[self window] firstResponder]];
	if(NSNotFound == i || !i) return;
	[[[self views] objectAtIndex:i] setSelection:[NSSet set]];
	[[self window] makeFirstResponder:[[self views] objectAtIndex:i - 1]];
}
- (IBAction)moveRight:(id)sender
{
	NSUInteger const i = [[self views] indexOfObjectIdenticalTo:[[self window] firstResponder]];
	if(NSNotFound == i || i + 1 >= [self numberOfColumns]) return;
	PGThumbnailView *const view = [[self views] objectAtIndex:i + 1];
	[[self window] makeFirstResponder:view];
	NSArray *const items = [view items];
	if([items count] && ![[view selection] count]) [view selectItem:[items objectAtIndex:0] byExtendingSelection:NO];
}

#pragma mark -<PGThumbnailViewDelegate>

- (void)thumbnailViewSelectionDidChange:(PGThumbnailView *)sender
{
	if(_updateCount) return;
	NSSet *const newSelection = [sender selection];
	id const selectedItem = [newSelection anyObject];
	if([newSelection count] != 1 || (dataSource && ![dataSource thumbnailBrowser:self itemCanHaveChildren:selectedItem])) {
		[self removeColumnsAfterView:sender];
		[[self delegate] thumbnailBrowserSelectionDidChange:self];
		return;
	}
	NSArray *const views = [self views];
	NSUInteger const col = [views indexOfObjectIdenticalTo:sender];
	NSParameterAssert(NSNotFound != col);
	if(col + 1 < [views count]) {
		PGThumbnailView *const nextView = [views objectAtIndex:col + 1];
		if([nextView representedObject] == selectedItem) return;
		[nextView setSelection:nil];
		[nextView setRepresentedObject:selectedItem];
		[nextView reloadData];
		[self scrollToTopOfColumnWithView:nextView];
	} else [self _addColumnWithItem:selectedItem];
	[self scrollToLastColumnAnimate:YES];
	[[self delegate] thumbnailBrowserSelectionDidChange:self];
}

@end

@implementation NSObject(PGThumbnailBrowserDataSource)

- (id)thumbnailBrowser:(PGThumbnailBrowser *)sender parentOfItem:(id)item
{
	return nil;
}
- (BOOL)thumbnailBrowser:(PGThumbnailBrowser *)sender itemCanHaveChildren:(id)item
{
	return YES;
}

@end

@implementation NSObject(PGThumbnailBrowserDelegate)

- (void)thumbnailBrowserSelectionDidChange:(PGThumbnailBrowser *)sender {}
- (void)thumbnailBrowser:(PGThumbnailBrowser *)sender numberOfColumnsDidChangeFrom:(NSUInteger)oldCount {}

@end