macosx/HBLanguagesSelection.m

   1 /*  HBLanguagesSelection.m $
   2
   3  This file is part of the HandBrake source code.
   4  Homepage: <http://handbrake.fr/>.
   5  It may be used under the terms of the GNU General Public License. */
   6
   7 #import "HBLanguagesSelection.h"
   8 #include "lang.h"
   9
  10 @implementation HBLang
  11
  12 - (instancetype)init
  13 {
  14     @throw nil;
  15 }
  16
  17 - (instancetype)initWithLanguage:(NSString *)value iso639_2code:(NSString *)code
  18 {
  19     self = [super init];
  20     if (self)
  21     {
  22         _language = value;
  23         _iso639_2 = code;
  24     }
  25     return self;
  26 }
  27
  28 - (instancetype)copyWithZone:(NSZone *)zone
  29 {
  30     HBLang *lang = [[self class] allocWithZone:zone];
  31
  32     lang->_isSelected = self.isSelected;
  33     lang->_language = self.language;
  34     lang->_iso639_2 = self.iso639_2;
  35
  36     return lang;
  37 }
  38
  39 - (NSString *)description
  40 {
  41     return self.language;
  42 }
  43
  44 @end
  45
  46 @implementation HBLanguagesSelection
  47
  48 - (instancetype)init
  49 {
  50     self = [self initWithLanguages:nil];
  51     return self;
  52 }
  53
  54 - (instancetype)initWithLanguages:(NSArray *)languages
  55 {
  56     self = [super init];
  57     if (self)
  58     {
  59         NSMutableArray *internal = [[NSMutableArray alloc] init];
  60         NSMutableArray *selected = [[NSMutableArray alloc] init];
  61
  62         const iso639_lang_t *lang = lang_get_next(NULL);
  63         for (lang = lang_get_next(lang); lang != NULL; lang = lang_get_next(lang))
  64         {
  65             NSString *nativeLanguage = strlen(lang->native_name) ? @(lang->native_name) : @(lang->eng_name);
  66
  67             HBLang *item = [[HBLang alloc] initWithLanguage:nativeLanguage
  68                                                 iso639_2code:@(lang->iso639_2)];
  69             if ([languages containsObject:item.iso639_2])
  70             {
  71                 item.isSelected = YES;
  72                 [selected addObject:item];
  73             }
  74             else
  75             {
  76                 [internal addObject:item];
  77             }
  78
  79         }
  80
  81         // Add the (Any) item.
  82         HBLang *item = [[HBLang alloc] initWithLanguage:@"(Any)"
  83                                             iso639_2code:@"und"];
  84         if ([languages containsObject:item.iso639_2])
  85         {
  86             item.isSelected = YES;
  87         }
  88         [internal insertObject:item atIndex:0];
  89
  90         // Insert the selected items
  91         // in the original order.
  92         [selected sortUsingComparator:^NSComparisonResult(id obj1, id obj2) {
  93             return [languages indexOfObject:[obj1 iso639_2]] - [languages indexOfObject:[obj2 iso639_2]];
  94         }];
  95
  96         [internal insertObjects:selected
  97                       atIndexes:[NSIndexSet indexSetWithIndexesInRange:(NSMakeRange(1, selected.count))]];
  98
  99         _languagesArray = internal;
 100     }
 101
 102     return self;
 103 }
 104
 105 - (NSArray *)selectedLanguages
 106 {
 107     NSMutableArray *selected = [[NSMutableArray alloc] init];
 108     for (HBLang *lang in self.languagesArray)
 109     {
 110         if (lang.isSelected)
 111         {
 112             [selected addObject:lang.iso639_2];
 113         }
 114     }
 115
 116     return [selected copy];
 117 }
 118
 119
 120 @end
 121
 122 NSString *kHBLanguagesDragRowsType = @"kHBLanguagesDragRowsType";
 123
 124 @implementation HBLanguageArrayController
 125
 126 - (void)setShowSelectedOnly:(BOOL)showSelectedOnly
 127 {
 128     _showSelectedOnly = showSelectedOnly;
 129     [self rearrangeObjects];
 130 }
 131
 132 - (NSArray *)arrangeObjects:(NSArray *)objects
 133 {
 134     if (!self.showSelectedOnly) {
 135         return [super arrangeObjects:objects];
 136     }
 137
 138     // Returns a filtered array with only the selected items
 139     NSMutableArray *filteredObjects = [NSMutableArray arrayWithCapacity:[objects count]];
 140     for (id item in objects)
 141     {
 142         if ([[item valueForKeyPath:@"isSelected"] boolValue])
 143         {
 144             [filteredObjects addObject:item];
 145         }
 146     }
 147
 148     return [super arrangeObjects:filteredObjects];
 149 }
 150
 151 - (void) awakeFromNib
 152 {
 153         [self.tableView registerForDraggedTypes:@[kHBLanguagesDragRowsType]];
 154         self.isDragginEnabled = YES;
 155 }
 156
 157 #pragma mark - NSTableView Delegate
 158
 159 - (NSString *)tableView:(NSTableView *)tableView typeSelectStringForTableColumn:(NSTableColumn *)tableColumn
 160                     row:(NSInteger)row
 161 {
 162     if ([[tableColumn identifier] isEqualToString:@"name"])
 163     {
 164         NSUInteger tableColumnIndex = [[tableView tableColumns] indexOfObject:tableColumn];
 165         return [[tableView preparedCellAtColumn:tableColumnIndex
 166                                             row:row] stringValue];
 167     }
 168     return nil;
 169 }
 170
 171 - (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard
 172 {
 173     if (self.isDragginEnabled)
 174         {
 175         NSData *data = nil;
 176         if (self.showSelectedOnly)
 177         {
 178             // If the show selected only filter
 179             // is enabled, we need to modify the rowIndexes
 180             // to match the position of the elements in the unfiltered array
 181             NSArray *filteredArray = [[self arrangedObjects] copy];
 182             self.showSelectedOnly = NO;
 183             NSArray *unfilteredArray = [self arrangedObjects];
 184
 185             NSMutableIndexSet *unfilteredIndexes = [NSMutableIndexSet indexSet];
 186             NSUInteger currentIndex = [rowIndexes firstIndex];
 187             while (currentIndex != NSNotFound)
 188             {
 189                 NSUInteger newIndex = [unfilteredArray indexOfObject:filteredArray[currentIndex]];
 190                 [unfilteredIndexes addIndex:newIndex];
 191                 currentIndex = [rowIndexes indexGreaterThanIndex:currentIndex];
 192             }
 193
 194             data = [NSKeyedArchiver archivedDataWithRootObject:unfilteredIndexes];
 195         }
 196         else
 197         {
 198             data = [NSKeyedArchiver archivedDataWithRootObject:rowIndexes];
 199         }
 200
 201         [pboard declareTypes:@[kHBLanguagesDragRowsType] owner:self];
 202         [pboard setData:data forType:kHBLanguagesDragRowsType];
 203     }
 204
 205     return self.isDragginEnabled;
 206 }
 207
 208 - (NSDragOperation)tableView:(NSTableView *)view
 209                 validateDrop:(id <NSDraggingInfo>)info
 210                  proposedRow:(NSInteger)row
 211        proposedDropOperation:(NSTableViewDropOperation)operation
 212 {
 213     NSDragOperation dragOp = NSDragOperationNone;
 214
 215     // if drag source is our own table view, it's a move or a copy
 216     if ([info draggingSource] == view)
 217         {
 218                 // At a minimum, allow move
 219                 dragOp =  NSDragOperationMove;
 220     }
 221
 222     [view setDropRow:row dropOperation:NSTableViewDropAbove];
 223
 224     return dragOp;
 225 }
 226
 227 - (BOOL)tableView:(NSTableView *)view acceptDrop:(id <NSDraggingInfo>)info row:(NSInteger)row dropOperation:(NSTableViewDropOperation)operation
 228 {
 229     if (([info draggingSourceOperationMask] == NSDragOperationCopy))
 230     {
 231         return NO;
 232     }
 233
 234     if ([info draggingSource] == view)
 235     {
 236         // Get the index set from the pasteBoard.
 237         NSData *rowData = [[info draggingPasteboard] dataForType:kHBLanguagesDragRowsType];
 238         NSIndexSet *indexSet = [NSKeyedUnarchiver unarchiveObjectWithData:rowData];
 239
 240         NSUInteger i = [indexSet countOfIndexesInRange:NSMakeRange(0, row)];
 241
 242         // Rearrage the objects.
 243         [self moveObjectsInArrangedObjectsFromIndexes:indexSet toIndex:row];
 244
 245         // Update the selection.
 246         row -= i;
 247         NSIndexSet *selectionSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(row, [indexSet count])];
 248         [view selectRowIndexes:selectionSet byExtendingSelection:NO];
 249
 250                 return YES;
 251     }
 252
 253     return NO;
 254 }
 255
 256 - (void)moveObjectsInArrangedObjectsFromIndexes:(NSIndexSet *)indexSet toIndex:(NSUInteger)insertIndex
 257 {
 258     NSArray     *objects = [self arrangedObjects];
 259
 260     NSInteger aboveInsertIndexCount = 0;
 261     NSInteger removeIndex;
 262
 263     NSUInteger thisIndex = [indexSet lastIndex];
 264     while (thisIndex != NSNotFound)
 265         {
 266                 if (thisIndex >= insertIndex)
 267                 {
 268                         removeIndex = thisIndex + aboveInsertIndexCount;
 269                         aboveInsertIndexCount += 1;
 270                 }
 271                 else
 272                 {
 273                         removeIndex = thisIndex;
 274                         insertIndex -= 1;
 275                 }
 276
 277                 // Get the object we're moving
 278                 id object = objects[removeIndex];
 279
 280                 // Move the object
 281                 [self removeObjectAtArrangedObjectIndex:removeIndex];
 282                 [self insertObject:object atArrangedObjectIndex:insertIndex];
 283
 284                 thisIndex = [indexSet indexLessThanIndex:thisIndex];
 285     }
 286 }
 287
 288 @end