/*
 * NSImage-Extensions.m
 * Sapphire
 *
 * Created by Warren Gavin on Oct. 7, 2007.
 * Copyright 2007 Sapphire Development Team and/or www.nanopi.net
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the GNU
 * General Public License as published by the Free Software Foundation; either version 3 of the License,
 * or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
 * Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along with this program; if not,
 * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#import "NSImage-Extensions.h"
#import <QTKit/QTKit.h>

static NSData * imageAtTime( QTMovie *movie, QTTime frameTime )
{
	NSImage *image = [movie frameImageAtTime:frameTime];

	if ( image == nil )
		return nil;

	NSApplicationLoad(); // TIFFRepresentation won't work without this
	NSBitmapImageRep * imageBitmap = [NSBitmapImageRep imageRepWithData:[image TIFFRepresentation]];

	return [imageBitmap representationUsingType:NSJPEGFileType properties:nil];
}

@implementation NSImage (QTImages)

+ (NSData *) imageFromMovie: (NSString *)path
{
	NSError * error     = nil;
	QTMovie * movie     = [QTMovie movieWithFile:path error:&error];	
	QTTime    imageTime = [movie duration];

	imageTime.timeValue /= 10;
	return imageAtTime( movie, imageTime );
}

+ (NSData *) imageFromMovie: (NSString *)path atTime: (unsigned int)instant
{
	NSError * error     = nil;
	QTMovie * movie     = [QTMovie movieWithFile:path error:&error];
	QTTime    imageTime = { instant, 1, 0 };

	return imageAtTime( movie, imageTime );
}

+ (NSArray *) imagesFromMovie: (NSString *)path forArraySize: (unsigned int) size
{
	NSLog( @"Getting array of size %d from %@", size, path );
	NSError * error    = nil;
	QTMovie * movie    = [QTMovie movieWithFile:path error:&error];
	NSLog(@"movie opened");
	QTTime    duration = [movie duration];

	unsigned int i;
	NSMutableArray * images = [NSMutableArray arrayWithCapacity:size];

	srand(time(NULL));

	// Split the duration into 'size' chunks and take a random
	// image from each chunk
	for ( i = 0; i < size; ++i )
	{
		QTTime imageTime = duration;
		imageTime.timeValue = (rand() % (duration.timeValue/size)) + (i*duration.timeValue/size);

		NSLog(@"get image");
		NSImage * image = [movie frameImageAtTime:imageTime];
		if ( image != nil )
			[images addObject:image];
	}

	NSLog( @"Returning array of size %d from %@", [images count], path );
	return images;
}

- (CGImageRef) asImageRef
{
	NSApplicationLoad(); // TIFFRepresentation won't work without this
	CGImageSourceRef sourceRef = CGImageSourceCreateWithData( (CFDataRef)[self TIFFRepresentation], NULL );
	
	return CGImageSourceCreateImageAtIndex( sourceRef, 0, NULL );
}

- (BOOL) writeToFile:(NSString *)path atomically:(BOOL)flag
{
	NSApplicationLoad(); // TIFFRepresentation won't work without this
	NSBitmapImageRep * imageBitmap = [NSBitmapImageRep imageRepWithData:[self TIFFRepresentation]];
	NSData           * imageData   = [imageBitmap representationUsingType:NSJPEGFileType properties:nil];

	return [imageData writeToFile:path atomically:flag];
}

@end