#include <CoreFoundation/CoreFoundation.h>
#include <CoreServices/CoreServices.h>
#include <QuickLook/QuickLook.h>
#import <Cocoa/Cocoa.h>
#include "ddjvuRef.h"
/* -----------------------------------------------------------------------------
Generate a preview for file
This function's job is to create preview for designated file
----------------------------------------------------------------------------- */
OSStatus
GeneratePreviewForURL(void *thisInterface,
QLPreviewRequestRef preview,
CFURLRef url,
CFStringRef contentTypeUTI,
CFDictionaryRef options)
{
CFBundleRef bundle = QLPreviewRequestGetGeneratorBundle(preview);
char *ddjvu = ddjvuPath(bundle);
if (ddjvu != NULL) {
NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init];
NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
NSDictionary *domain = [defaults persistentDomainForName:@"org.djvu.qlgenerator"];
CFUUIDRef uuid = NULL;
CFStringRef uuidString;
BOOL debug = FALSE;
NSString *tmpPath = nil;
if (domain && [domain objectForKey:@"previewpages"])
if ([[domain objectForKey:@"previewpages"] intValue] == 0)
goto poppool;
if (domain && [domain objectForKey:@"debug"]) {
debug = [[domain objectForKey:@"debug"] boolValue];
}
uuid = CFUUIDCreate(kCFAllocatorDefault);
if (uuid) {
uuidString = CFUUIDCreateString(kCFAllocatorDefault, uuid);
if (uuidString) {
tmpPath = [NSTemporaryDirectory() stringByAppendingPathComponent:
[NSString stringWithFormat:@"djvuql-%@", uuidString]];
CFRelease(uuidString);
}
CFRelease(uuid);
}
if (tmpPath) {
const char *cmd = NULL;
CFStringRef cmdRef;
int page = 0;
int pages = 1;
int maxpages = 5;
int width = 612; // default 8.5x11 in points
int height = 792;
CGSize size;
MDItemRef mditem = NULL;
NSString *source = (NSString *)CFURLCopyFileSystemPath(url, kCFURLPOSIXPathStyle);
NSString *dest = nil;
NSFileManager *fmgr = [NSFileManager defaultManager];
[fmgr createDirectoryAtPath:tmpPath attributes:nil];
mditem = MDItemCreate(kCFAllocatorDefault, (CFStringRef)source);
if (mditem) {
CFTypeRef ref = NULL;
ref = MDItemCopyAttribute(mditem, kMDItemNumberOfPages);
if (ref) {
CFNumberGetValue(ref, kCFNumberIntType, &pages);
CFRelease(ref);
}
ref = MDItemCopyAttribute(mditem, kMDItemPageWidth);
if (ref) {
CFNumberGetValue(ref, kCFNumberIntType, &width);
CFRelease(ref);
}
ref = MDItemCopyAttribute(mditem, kMDItemPageHeight);
if (ref) {
CFNumberGetValue(ref, kCFNumberIntType, &height);
CFRelease(ref);
}
CFRelease(mditem);
}
if (domain && [domain objectForKey:@"previewpages"])
maxpages = [[domain objectForKey:@"previewpages"] intValue];
if (debug) {
NSLog(@"metadata: pages=%d, width=%d, height=%d", pages, width, height);
NSLog(@"maxpages=%d", maxpages);
}
CGRect rect = CGRectMake(0, 0, width, height);
CGContextRef c;
c = QLPreviewRequestCreatePDFContext(preview, &rect, NULL, NULL);
page = 0;
do {
page++;
dest = [tmpPath stringByAppendingPathComponent:[[source lastPathComponent] stringByAppendingFormat:@"_t_p%04d.tiff", page]];
size = CGSizeMake(width, height);
cmdRef = CFStringCreateWithFormat(NULL, NULL, CFSTR("\"%s\" -format=tiff -page=%d -size=%dx%d \"%s\" \"%s\""), ddjvu, page, width, height, [source fileSystemRepresentation], [dest fileSystemRepresentation]);
cmd = CFStringGetCStringPtr(cmdRef, CFStringGetSystemEncoding());
if (cmd != NULL) {
if (debug)
NSLog(@"%s", cmd);
if ((system(cmd) == 0) && ([fmgr fileExistsAtPath:dest])) {
NSURL *durl = [NSURL fileURLWithPath:dest];
CGImageSourceRef sourceRef;
sourceRef = CGImageSourceCreateWithURL((CFURLRef)durl, NULL);
if(sourceRef) {
CGImageRef imageRef = NULL;
imageRef = CGImageSourceCreateImageAtIndex(sourceRef, 0, NULL);
if (imageRef) {
CGFloat offset = height * (page-1);
CGRectOffset(rect, 0.0, offset);
CGPDFContextBeginPage(c, NULL);
CGContextSaveGState(c);
CGContextDrawImage(c, rect, imageRef);
if (page == maxpages && maxpages < pages) {
CGAffineTransform m;
CFURLRef more = CFBundleCopyResourceURL(bundle, CFSTR("more_pages"), CFSTR("pdf"), NULL);
CGPDFDocumentRef doc = CGPDFDocumentCreateWithURL(more);
CGPDFPageRef pdf = CGPDFDocumentGetPage(doc, 1);
CGFloat offset = height * page;
CGRectOffset(rect, 0.0, offset);
CGContextSaveGState(c);
m = CGPDFPageGetDrawingTransform(pdf, kCGPDFMediaBox, rect, 0, true);
CGContextConcatCTM(c, m);
CGContextDrawPDFPage(c, pdf);
CGContextRestoreGState(c);
CFRelease(doc);
}
CGContextRestoreGState(c);
CGPDFContextEndPage(c);
CFRelease(imageRef);
}
CFRelease(sourceRef);
}
}
}
CFRelease(cmdRef);
} while (page < pages && page < maxpages);
CGPDFContextClose(c);
QLPreviewRequestFlushContext(preview, c);
CFRelease(c);
[fmgr removeFileAtPath:tmpPath handler:nil];
[source release];
}
poppool:
[pool release];
}
return noErr;
}
void CancelPreviewGeneration(void* thisInterface, QLPreviewRequestRef preview)
{
// implement only if supported
}