ドラッグ＆ドロップ用のデータソースオブジェクトを作る【実践的Macintoshプログラミング解説】

//
//  DragDropDataSource.h
//  RepeatingMotifGenerator
//
//  Copyright NovemberKou 2008 . All rights reserved.
//

#import <Cocoa/Cocoa.h>


@interface DragDropDataSource : NSObject
{
    IBOutlet    NSArrayController   *arrayController;

    int     insertionRow;
}

@end

//
//  DragDropDataSource.m
//  RepeatingMotifGenerator
//
//  Copyright NovemberKou 2008 . All rights reserved.
//

#import "DragDropDataSource.h"

#import "Lissajous.h"


#undef      UNDO_MANAGER
#define     UNDO_MANAGER    [[arrayController managedObjectContext] undoManager]

// Pboard type
NSString    *RMGRowsType = @"RMGRowIndexesPboardType";

@implementation DragDropDataSource

//
//  DragDropDataSource.m
//

- (void)awakeFromNib
{
    NSNotificationCenter    *nc = [NSNotificationCenter defaultCenter];

    insertionRow = -1;
    [nc addObserver:self
           selector:@selector(managedObjectChanged:)
               name:NSManagedObjectContextObjectsDidChangeNotification
             object:[arrayController managedObjectContext]];
}

- (void)dealloc
{
    [[NSNotificationCenter defaultCenter] removeObserver:self];
    [super dealloc];
}

//
//  DragDropDataSource.m
//

#pragma mark -
#pragma mark Data Source method

//  ゼロを返すとテーブルビューは後戻りしてバインディングからデータを得る
- (int)numberOfRowsInTableView:(NSTableView *)tableView
{
    return 0;
}

//  nilを返すとテーブルビューは後戻りしてバインディングからデータを得る
- (id)tableView:(NSTableView *)tableView objectValueForTableColumn:(NSTableColumn *)tableColumn
            row:(int)rowIndex
{
    return nil;  
}

//
//  DragDropDataSource.m
//

- (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes
     toPasteboard:(NSPasteboard *)pboard
{
    [Lissajous copyObjects:[[arrayController arrangedObjects] objectsAtIndexes:rowIndexes]
              toPasteboard:pboard];
    [pboard addTypes:[NSArray arrayWithObject:RMGRowsType] owner:nil];
    [pboard setData:[NSArchiver archivedDataWithRootObject:rowIndexes]
            forType:RMGRowsType];

    return  YES;
}

//
//  DragDropDataSource.m
//

- (NSDragOperation)tableView:(NSTableView *)tableView
                validateDrop:(id <NSDraggingInfo>)info
{
    NSDragOperation dragOperation;

    dragOperation = ([info draggingSource] != tableView) ? NSDragOperationCopy : NSDragOperationMove;

    if([[NSApp currentEvent] modifierFlags] & NSAlternateKeyMask)
        dragOperation = NSDragOperationCopy;

    return  dragOperation & [info draggingSourceOperationMask];
}

- (NSDragOperation)tableView:(NSTableView *)tableView
                validateDrop:(id <NSDraggingInfo>)info
                 proposedRow:(int)row
       proposedDropOperation:(NSTableViewDropOperation)operation
{
    if(operation == NSTableViewDropOn)
        [tableView setDropRow:row dropOperation:NSTableViewDropAbove];

    return  [self tableView:tableView validateDrop:info];
}

//
//  DragDropDataSource.m
//

- (void)rearrangeObjects
{
    [[UNDO_MANAGER prepareWithInvocationTarget:self] rearrangeObjects];
    [arrayController rearrangeObjects];
}

- (BOOL)tableView:(NSTableView*)tableView 
       acceptDrop:(id <NSDraggingInfo>)info 
              row:(int)row 
    dropOperation:(NSTableViewDropOperation)operation
{
    NSPasteboard    *pboard = [info draggingPasteboard];
    NSDragOperation dragOperation = [self tableView:tableView validateDrop:info];

    if(dragOperation == NSDragOperationCopy)
    {
        insertionRow = row;
        [Lissajous pasteFromPasteboard:pboard controller:arrayController];
        [UNDO_MANAGER setActionName:NSLocalizedString(@"dragCopyLissajous",nil)];
        return  YES;
    }
    else if(dragOperation == NSDragOperationMove)
    {
        int             i,order,index,first,last;
        NSIndexSet      *rowIndexes;
        RMGRoot         *object;
        NSArray         *arrangedObjects = [arrayController arrangedObjects];
        NSEnumerator    *enumerator = [arrangedObjects objectEnumerator];
        NSMutableArray  *orders = [NSMutableArray arrayWithCapacity:[arrangedObjects count]];

        while(object = [enumerator nextObject])
            [orders addObject:[object order]];

        rowIndexes = [NSUnarchiver unarchiveObjectWithData:[pboard dataForType:RMGRowsType]];

        //  挿入された行までのドラッグされていないオブジェクトの番号を振り直す
        first = row < [rowIndexes firstIndex] ? row : [rowIndexes firstIndex];
        if(first == 0)
            order = 0;
        else
            order = [[[arrangedObjects objectAtIndex:first-1] order] intValue] + 1;
        for(i=first;i<row;i++)
        {
            if([rowIndexes containsIndex:i])    continue;
            object = [arrangedObjects objectAtIndex:i];
            [object setValue:[orders objectAtIndex:order++] forKey:@"order"];
        }
        //  ドラッグされているオブジェクトの番号を振り直す
        index = [rowIndexes firstIndex];
        while(index != NSNotFound)
        {
            object = [arrangedObjects objectAtIndex:index];
            [object setValue:[orders objectAtIndex:order++] forKey:@"order"];
            index = [rowIndexes indexGreaterThanIndex:index];
        }
        //  挿入された行以降のドラッグされていないオブジェクトの番号を振り直す
        last = row > [rowIndexes lastIndex] ? row : [rowIndexes lastIndex];
        for(i=row;i<last;i++)
        {
            if([rowIndexes containsIndex:i])    continue;
            object = [arrangedObjects objectAtIndex:i];
            [object setValue:[orders objectAtIndex:order++] forKey:@"order"];
        }

        [self rearrangeObjects];
        [UNDO_MANAGER setActionName:NSLocalizedString(@"dragOperation",nil)];
        return YES;
    }
    else
        return  NO;
}