trunk/ImageSplitter/CRTableViewController.m

//
//  CRTableViewController.m
//  ImageSpliter
//
//  Created by hiroshi tomioka on 13/02/12.
//  Copyright (c) 2013å¹´ hiroshi tomioka. All rights reserved.
//

#import "CRTableViewController.h"
#import <Cocoa/Cocoa.h>
#import "CRDropView.h"
#import "NSImage+data.h"
#import "ZipArchive.h"

@implementation CRTableViewController {
        NSMutableArray *_dataArray;
        IBOutlet CRDropView *mainView;
        IBOutlet NSTableView *_tableView;
        IBOutlet NSProgressIndicator *_indicator;
}
@synthesize _dirPath;
//@synthesize mainView;

- (id)init
{
    self = [super init];
    if (self) {
        NSLog(@"%s", __func__);

                [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(respondFileDropedNotification:) name:@"files droped" object:nil];
                
        _dataArray = [[NSMutableArray alloc] init];
    }
    return self;
}

- (void)awakeFromNib {

}


#pragma mark - NSTableView data source

- (NSInteger)numberOfRowsInTableView:(NSTableView*)tableView
{
    NSLog(@"%s", __func__);
    
    return _dataArray.count;
}

- (id)tableView:(NSTableView *)tableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row
{
    NSDictionary *data = [_dataArray objectAtIndex:row];
    if ([[tableColumn identifier] isEqualToString:@"TITLE"]) {
        return [data objectForKey:@"title"];
    } else {
        return [data objectForKey:@"description"];
    }
}

//ãƒ‰ãƒãƒƒãƒ—ãƒ•ã‚¡ã‚¤ãƒ«ã�®é€šçŸ¥
- (void)respondFileDropedNotification:(NSNotification*)notification {
        NSDictionary* dic = notification.userInfo;
        NSArray *files = [dic objectForKey:@"files"];
        
        _dataArray = [[NSMutableArray alloc] init];
        for (int i = 0; i < [files count]; i++) {
                NSDictionary *data = @{@"title": [files[i] lastPathComponent],
                                                 @"description": files[i]};
                [_dataArray addObject:data];
        }
        [_tableView reloadData];
}

- (NSString*)imageTypeExp:(NSInteger)type {
        NSString *exp = @"";
        switch (type) {
                case IMAGE_TYPE_JPEG:
                        exp = @"jpg";
                        break;
                case IMAGE_TYPE_PNG:
                        exp = @"png";
                        break;
                case IMAGE_TYPE_TIFF:
                        exp = @"tif";
                        break;
                case IMAGE_TYPE_JPEG2000:
                        exp = @"jpg";
                        break;
                case IMAGE_TYPE_BMP:
                        exp = @"bmp";
                        break;
        }
        return exp;
}

- (IBAction)buildButton:(id)sender {
        NSLog(@"pushed");
        BOOL isDirectory;
        NSInteger imageType = [[NSUserDefaults standardUserDefaults] integerForKey:@"image type"];
        NSInteger zipFlag = [[NSUserDefaults standardUserDefaults] integerForKey:@"zip flag"];
        NSInteger sizeFlag = [[NSUserDefaults standardUserDefaults] integerForKey:@"fit size"];
        int fitWidth = (int)[[NSUserDefaults standardUserDefaults] integerForKey:@"fit width"];
        int fitHeight = (int)[[NSUserDefaults standardUserDefaults] integerForKey:@"fit height"];
        double qualityJpeg2000 = [[NSUserDefaults standardUserDefaults] doubleForKey:@"jpeg2000 quality"];
        double qualityJpeg = [[NSUserDefaults standardUserDefaults] doubleForKey:@"jpeg quality"];
        
        NSString *dir = NSTemporaryDirectory();
        NSLog(@"dir %@", dir);
        
        if (0 == zipFlag) {
                NSInteger result;
                NSOpenPanel* openPanel;
                
                openPanel = [NSOpenPanel openPanel];
                [openPanel setCanChooseDirectories:YES];
                [openPanel setCanCreateDirectories:YES];
                result = [openPanel runModal];
                if(result == NSOKButton) {
                        NSString*       dirPath = [[openPanel URL] path];
                        BOOL isDirectory;
                        if ([[NSFileManager defaultManager] fileExistsAtPath:dirPath isDirectory:&isDirectory]) {
                                if (isDirectory) {
                                        self._dirPath = dirPath;
                                        dir = dirPath;
                                } else {
                                        NSRunAlertPanel(@"info", @"set a folder",
                                                                        nil, nil, nil);
                                }
                        }
                }
        } else {
                
                NSInteger result;
                NSSavePanel* savePanel;
                
                savePanel = [NSSavePanel savePanel];
                NSArray *allowedFileTypes = [NSArray arrayWithObjects:@"zip",@"'ZIP'",nil];
                [savePanel setAllowedFileTypes:allowedFileTypes];
        
                [savePanel setCanCreateDirectories:YES];
                result = [savePanel runModal];
                if(result == NSOKButton) {
                        NSString*       dirPath = [[savePanel URL] path];
                        self._dirPath = dirPath;
                }
        }


        [_indicator setDoubleValue:0];
        if ([[NSFileManager defaultManager] fileExistsAtPath:dir isDirectory:&isDirectory]) {
                if (isDirectory) {
                        _indicator.minValue = 0;
                        _indicator.maxValue = [mainView._files count] * 2;
                        
                        int count = 0;
                        int progress = 0;
                        for (int i = 0; i < [mainView._files count]; i++) {
                                
                                NSImage *image;
                                NSData *data;
                                NSBitmapImageRep *bitmapImageRep;
                                
                                image = [[NSImage alloc] initWithContentsOfFile:mainView._files[i]];
                                data = [image TIFFRepresentation];
                                bitmapImageRep = [NSBitmapImageRep imageRepWithData:data];
                                NSSize pointSize = NSMakeSize(bitmapImageRep.pixelsWide, bitmapImageRep.pixelsHigh);
                                
                                if (pointSize.width < pointSize.height) {
                                        NSString *filename;
                                        filename = [dir stringByAppendingFormat:@"/%06d.%@", count++, [self imageTypeExp:imageType]];
                                        
                                        NSImage *newImage = image;
                                        if (0 == sizeFlag) {
                                                newImage = [image fitImageInSize:NSMakeSize(fitWidth, fitHeight)];
                                        }
                                        switch (imageType) {
                                                case IMAGE_TYPE_JPEG:
                                                        [[newImage JPEGDataWithQuality:qualityJpeg] writeToFile:filename atomically:YES];
                                                        break;
                                                case IMAGE_TYPE_PNG:
                                                        [[newImage PNGDataWithInterlaced:YES] writeToFile:filename atomically:YES];
                                                        break;
                                                case IMAGE_TYPE_TIFF:
                                                        [[newImage TIFFRepresentation] writeToFile:filename atomically:YES];
                                                        break;
                                                case IMAGE_TYPE_JPEG2000:
                                                        [[newImage JPEG2000DataWithQuality:qualityJpeg2000] writeToFile:filename atomically:YES];
                                                        break;
                                                case IMAGE_TYPE_BMP:
                                                        [[newImage BMPData] writeToFile:filename atomically:YES];
                                                        break;
                                        }
                                        
                                        progress += 2;
                                        [_indicator setDoubleValue:progress];
                                } else {
                                        for (int j = 0; j < 2; j++) {
                                                NSString *filename;
                                                filename = [dir stringByAppendingFormat:@"/%06d.%@", count++, [self imageTypeExp:imageType]];
                                                NSRect rc = NSMakeRect((pointSize.width / 2.0 * (1 - j)), 0, pointSize.width / 2.0, pointSize.height);
                
                                                NSImage *newImage = [image trimImageByRect:rc];
                                                if (0 == sizeFlag) {
                                                        newImage = [newImage fitImageInSize:NSMakeSize(fitWidth, fitHeight)];
                                                }
                                                switch (imageType) {
                                                        case IMAGE_TYPE_JPEG:
                                                                [[newImage JPEGDataWithQuality:qualityJpeg] writeToFile:filename atomically:YES];
                                                                break;
                                                        case IMAGE_TYPE_PNG:
                                                                [[newImage PNGDataWithInterlaced:YES] writeToFile:filename atomically:YES];
                                                                break;
                                                        case IMAGE_TYPE_TIFF:
                                                                [[newImage TIFFRepresentation] writeToFile:filename atomically:YES];
                                                                break;
                                                        case IMAGE_TYPE_JPEG2000:
                                                                [[newImage JPEG2000DataWithQuality:qualityJpeg2000] writeToFile:filename atomically:YES];
                                                                break;
                                                        case IMAGE_TYPE_BMP:
                                                                [[newImage BMPData] writeToFile:filename atomically:YES];
                                                                break;
                                                }
                                                
                                                progress++;
                                                [_indicator setDoubleValue:progress];
                                        }
                                }
                                [image release];
                                
                        }
                        
//                      NSError *error;
//                      for (int i = 0; i < count; i++) {
//                              NSString* atPath = [dir stringByAppendingFormat:@"/%06d.jpg", i];
//                              NSString* toPath = [self._dirPath stringByAppendingFormat:@"/a%06d.jpg", i];
//                              if (![[NSFileManager defaultManager] moveItemAtPath:atPath toPath:toPath error:&error]) {
//                                      
//                                      NSRunAlertPanel(@"info", error.debugDescription,
//                                                                      nil, nil, nil);
//                                      break;
//                              }
//                      }
                        
                        if (0 != zipFlag) {
                                NSString* zipFile = self._dirPath;
                                ZipArchive* archiver = [[ZipArchive alloc] init];
                                [archiver CreateZipFile2:zipFile];
                                
                                for (int i = 0; i < count; i++) {
                                        NSString* atPath = [dir stringByAppendingFormat:@"/%06d.%@", i, [self imageTypeExp:imageType]];
                                        NSString* toPath = [NSString stringWithFormat:@"%06d.%@", i, [self imageTypeExp:imageType]];
                                        [archiver addFileToZip:atPath
                                                                   newname:toPath];
                                }
                                [archiver CloseZipFile2];
                                [archiver release];
                                
                                for (int i = 0; i < count; i++) {
                                        NSError *error;
                                        NSString* atPath = [dir stringByAppendingFormat:@"/%06d.%@", i, [self imageTypeExp:imageType]];
                                        
                                        if (![[NSFileManager defaultManager] removeItemAtPath:atPath error:&error]) {
                                                NSLog(@"error %@", error.debugDescription);
                                        }
                                }
                        }
                        
                        NSRunAlertPanel(@"info", @"finish",
                                                        nil, nil, nil);
                        
                        
                }
        } else {
                NSRunAlertPanel(@"info", @"set a folder",
                                                nil, nil, nil);
        }
}

- (IBAction)onClearButton:(id)sender
{
        [mainView._files removeAllObjects];
        _dataArray = [[NSMutableArray alloc] init];
        [_tableView reloadData];
}

- (IBAction)onOpenButton:(id)sender
{
        NSInteger result;
        NSOpenPanel* openPanel;
        
        openPanel = [NSOpenPanel openPanel];
        [openPanel setCanChooseDirectories:YES];
        [openPanel setCanCreateDirectories:YES];
        [openPanel setAllowsMultipleSelection:YES];
        
        result = [openPanel runModal];
        if(result == NSOKButton) {
                
                
                for (NSURL *url in [openPanel URLs]) {
                        NSString *path = [url path];
                        
                        BOOL isDirectory;
                        if ([[NSFileManager defaultManager] fileExistsAtPath:path isDirectory:&isDirectory]) {
                                if (isDirectory) {
                                        NSArray* array = [[NSFileManager defaultManager] subpathsAtPath:path];
                                        array = [array sortedArrayUsingSelector:@selector(caseInsensitiveCompare:)];
                                        
                                        for (NSString *filename in array) {
                                                filename = [path stringByAppendingPathComponent:filename];
                                                if ([NSImage isImageFile:filename]) {
                                                        NSLog(@"%@", filename);
                                                        [mainView._files addObject:filename];
                                                }
                                        }
                                } else {
                                        if ([NSImage isImageFile:path]) {
                                                [mainView._files addObject:path];
                                        }
                                }
                                
                        }
                }
        }
        
        NSArray *files = mainView._files;
        
        _dataArray = [[NSMutableArray alloc] init];
        for (int i = 0; i < [files count]; i++) {
                NSDictionary *data = @{@"title": [files[i] lastPathComponent],
                                                 @"description": files[i]};
                [_dataArray addObject:data];
        }
        [_tableView reloadData];
}

- (IBAction)onInfo:(id)sender {
        if (0 < [mainView._files count] && 0 <= [_tableView selectedRow]) {
                NSString *selectedStringValue = [mainView._files objectAtIndex:[_tableView selectedRow]];
                if (selectedStringValue) {
                        if ([[NSWorkspace sharedWorkspace] openFile:selectedStringValue]) { //withApplication
                                return;
                        }
                }
        }
        NSRunAlertPanel(@"info", @"can't open",
                                        nil, nil, nil);
}

//- (IBAction)savePDF:(id)sender
//{
//      // (1) setup filename
//      NSArray *paths = NSSearchPathForDirectoriesInDomains(
//                                                                                                               NSDesktopDirectory, NSUserDomainMask, YES);
//      NSString* path = [NSString stringWithFormat:@"%@/test.pdf",
//                                        [paths objectAtIndex:0], nil];
//      
//      // (2) setup WebView
//      NSView* view = [[[_web_view mainFrame] frameView] documentView];
//      
//      // (3) setup NSDictionary for NSPrintInfo
//      NSMutableDictionary* p_dict = [NSMutableDictionary
//                                                                 dictionaryWithDictionary:[[NSPrintInfo sharedPrintInfo] dictionary]];
//      [p_dict setObject:NSPrintSaveJob forKey:NSPrintJobDisposition];
//      [p_dict setObject:path forKey:NSPrintSavePath];
//      
//      // (4) setup NSPrintInfo
//      NSPrintInfo* p_info = [[NSPrintInfo alloc] initWithDictionary:p_dict];
//      [p_info setHorizontalPagination:NSAutoPagination];
//      [p_info setVerticalPagination:NSAutoPagination];
//      [p_info setVerticallyCentered:NO];
//      
//      // (5) setup NSPrintOperation
//      NSPrintOperation* p_ope =
//      [NSPrintOperation printOperationWithView:view
//                                                                 printInfo:p_info];
//      [p_ope setShowsPrintPanel:NO];
//      
//      // (6) print it
//      [p_ope runOperation];
//}

@end