Ticket #45: NSImage-Extensions.2.m

/*
 * 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>
#import <QTKit/QTMovie.h>

static const NSData * imageAsBitmapData( const NSImage * const image )
{
        const NSBitmapImageRep * imageBitmap = nil;
        const NSEnumerator     * const en = [[image representations] objectEnumerator];
        const NSImageRep       * rep = nil;
        
        while( nil != (rep = [en nextObject]) )
        {
                if ( [rep isKindOfClass:[NSBitmapImageRep class]] )
                {
                        imageBitmap = (NSBitmapImageRep *) rep;
                        break;
                }
        }
        
        return [imageBitmap representationUsingType:NSJPEGFileType properties:nil];
}

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

        if ( image == nil )
                return nil;

        return imageAsBitmapData( image );
}

@implementation NSImage (QTImages)

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

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

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

        return imageAtTime( movie, imageTime );
}

+ (const NSArray *) imagesFromMovie: (NSString *)path forArraySize: (const unsigned int) size
{
        const NSError * const error    = nil;
        const QTMovie * const movie    = [QTMovie movieWithFile:path error:&error];
        const QTTime          duration = [movie duration];

        unsigned int i;
        const NSMutableArray * const 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);

                const NSImage * const image = [movie frameImageAtTime:imageTime];
                if ( image != nil )
                        [images addObject:image];
        }

        return images;
}

- (CGImageRef) asImageRef
{
        CGImageSourceRef sourceRef = CGImageSourceCreateWithData( (CFDataRef)imageAsBitmapData( self ), NULL ); 
        return CGImageSourceCreateImageAtIndex( sourceRef, 0, NULL );
}

- (BOOL) writeToFile:(NSString *)path atomically:(BOOL)flag
{
        const NSData * imageData = imageAsBitmapData( self );
        return [imageData writeToFile:path atomically:flag];
}

@end