DataHandler.m

   1 // MacTF Copyright 2004 Nathan Oates
   2
   3 /*
   4 *
   5  * This source code is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Public License as published
   7  * by the Free Software Foundation; either version 2 of the License,
   8  * or (at your option) any later version.
   9  *
  10  * This source code is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * Please refer to the GNU Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Public License along with
  16  * this source code; if not, write to:
  17  * Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19
  20 /* This file was modified by Kalle Olavi Niemitalo on 2007-10-18.  */
  21
  22
  23 #import "DataHandler.h"
  24
  25 @implementation DataHandler
  26
  27 #pragma mark -
  28 #pragma mark Startup and Shutdown
  29
  30 - (id) init
  31 {
  32     if (self = [super init])
  33     {
  34         fileList = [[NSMutableArray alloc] init];
  35                 searchList = [[NSMutableArray alloc] init];
  36     }
  37     return self;
  38 }
  39
  40 - (void) dealloc
  41 {
  42     [fileList release];
  43     [searchList release];
  44     [super dealloc];
  45 }
  46
  47
  48
  49 #pragma mark -
  50 #pragma mark Simple Accessors
  51
  52 - (NSMutableArray *) fileList
  53 {
  54     return fileList;
  55 }
  56
  57 - (NSMutableArray *) displayedList
  58 {
  59     if (searchIsActive) return searchList;
  60         else return fileList;
  61 }
  62
  63 - (void) setFileList: (NSArray *)newFileList
  64 {
  65     if (fileList != newFileList)
  66     {
  67         [fileList autorelease];
  68         fileList = [[NSMutableArray alloc] initWithArray: newFileList];
  69     }
  70 }
  71
  72 - (void) setSearchList: (NSArray *)newSearchList
  73 {
  74     if (searchList != newSearchList)
  75     {
  76         [searchList autorelease];
  77         searchList = [[NSMutableArray alloc] initWithArray: newSearchList];
  78     }
  79 }
  80
  81 -(void)awakeFromNib
  82 {
  83         searchIsActive = NO;
  84 }
  85
  86
  87
  88 -(id)tableView:(NSTableView *)aTableView
  89 objectValueForTableColumn:(NSTableColumn *)aTableColumn
  90            row:(int)rowIndex
  91 {
  92     id theRecord, theValue;
  93         if (searchIsActive)     theRecord = [searchList objectAtIndex:rowIndex];
  94     else theRecord = [fileList objectAtIndex:rowIndex];
  95     theValue = [theRecord objectForKey:[aTableColumn identifier]];
  96     return theValue;
  97 }
  98
  99 // just returns the number of items we have.
 100 - (int)numberOfRowsInTableView:(NSTableView *)aTableView
 101 {
 102         if (searchIsActive) return [searchList count];
 103     else return [fileList count];
 104 }
 105
 106 - (void) reloadTable {
 107     [tableView reloadData];
 108 }
 109
 110 - (IBAction) search:(id)sender {
 111         NSString *searchString = [sender stringValue];
 112         if ([searchString isEqualToString:@""]) {
 113                 searchIsActive = NO;
 114                 [self reloadTable];
 115         } else {
 116                 NSArray *searchTerms = [searchString componentsSeparatedByString:@" "];
 117                 NSEnumerator *enumerator = [searchTerms objectEnumerator];
 118                 id anObject;
 119                 BOOL firstToken = YES;
 120                 NSArray *results = [[NSArray alloc] init];
 121                 while (anObject = [enumerator nextObject]) {
 122                         if (![anObject isEqualTo:@""]) {
 123                                 NSPredicate *pred = [NSPredicate predicateWithFormat:@"(name contains[cd] %@)", anObject];
 124                                 if (firstToken)
 125                                         results = [fileList filteredArrayUsingPredicate:pred];
 126                                 else
 127                                         results = [results filteredArrayUsingPredicate:pred];
 128                                 firstToken = NO;
 129                         }
 130                 }
 131                 [self setSearchList:results];
 132                 searchIsActive = YES;
 133                 [self reloadTable];
 134         }
 135 }
 136
 137
 138 // Drag n Drop methods
 139 /*- (BOOL)tableView:(NSTableView *)aTableView writeRows:(NSArray*)rows toPasteboard:(NSPasteboard*)pboard {
 140      id list;
 141         if (searchIsActive)
 142                 list = searchList;
 143         else
 144                 list = seqs;
 145         NSMutableArray *tempArray = [NSMutableArray array];
 146     movedRows = [[NSMutableArray alloc] init];
 147     NSNumber *index;
 148     id tempObject;
 149     NSEnumerator *enumerator = [rows objectEnumerator];
 150     while ( (index = [enumerator nextObject]) ) {
 151         tempObject = [list objectAtIndex:[index intValue]];
 152         [tempArray addObject:tempObject];
 153     }
 154     movedRows = tempArray;
 155     NSData *data = [NSArchiver archivedDataWithRootObject: tempArray];
 156     [pboard declareTypes: [NSArray arrayWithObject: @"NSGeneralPboardType"] owner: nil];
 157     [pboard setData: data forType: @"NSGeneralPboardType"];
 158     return YES;
 159 }
 160 */
 161 - (BOOL)tableView:(NSTableView*)aTableView acceptDrop:(id <NSDraggingInfo>)info row:(int)row dropOperation:(NSTableViewDropOperation)operation {
 162         NSPasteboard *pboard = [info draggingPasteboard];
 163     NSArray *types = [NSArray arrayWithObjects:@"NSFilenamesPboardType", nil];
 164         NSString *desiredType = [pboard availableTypeFromArray:types];
 165         if (nil == [info draggingSource]) // From other application
 166     {
 167                 NSData *carriedData = [pboard dataForType:desiredType];
 168                 if (nil == carriedData)
 169                 {
 170                         //the operation failed for some reason
 171                         NSRunAlertPanel(@"Paste Error", @"Sorry, but the paste operation failed", nil, nil, nil);
 172                         return NO;
 173                 } else {
 174                         if ([desiredType isEqualToString:NSFilenamesPboardType]) {
 175                                 //we have a list of file names in an NSData object
 176                                 NSArray *fileArray = [pboard propertyListForType:@"NSFilenamesPboardType"]; //be caseful since this method returns id. We just happen to know that it will be an array.
 177                                 NSEnumerator* selected = [fileArray objectEnumerator];
 178                                 id currentSelected;
 179                                 while (currentSelected = [selected nextObject]) {
 180                                         NSLog(@"Upload:%@ to %@", currentSelected, [NSString stringWithString:[UIEl currentPath]]);
 181                                         [UIEl uploadPath:currentSelected toPath:[NSString stringWithString:[UIEl currentPath]]];
 182                                 }
 183                         } else {
 184                                 //this can't happen
 185                                 NSAssert(NO, @"This can't happen");
 186                                 return NO;
 187                         }
 188                         return YES;
 189                 }
 190     } else if (aTableView == [info draggingSource]) { // From self
 191         return YES;
 192     }
 193     // From other documents
 194         return YES;
 195 }
 196
 197     - (NSDragOperation) tableView: (NSTableView *) view
 198 validateDrop: (id <NSDraggingInfo>) info
 199 proposedRow: (int) row
 200 proposedDropOperation: (NSTableViewDropOperation) operation
 201     {
 202
 203         if (nil == [info draggingSource]) // From other application
 204     {
 205                 [view setDropRow: row dropOperation: NSTableViewDropAbove];
 206         return NSDragOperationCopy;
 207     }
 208     else if (tableView == [info draggingSource]) // From self
 209     {
 210          return NSDragOperationNone;
 211     }
 212         return NSDragOperationNone;
 213 }
 214
 215 // Delegate methods for tableview
 216
 217 - (NSMutableDictionary*) getTFFileFromSwappedHexData:(NSData*) input {
 218
 219         NSData* mjdHex = [input subdataWithRange:(NSRange) {0,2}];
 220         NSData* hrHex = [input subdataWithRange:(NSRange) {2,1}];
 221         NSData* minHex = [input subdataWithRange:(NSRange) {3,1}];
 222         NSData* secHex = [input subdataWithRange:(NSRange) {4,1}];
 223         NSData* typeHex = [input subdataWithRange:(NSRange) {5,1}];
 224         NSData* sizeMSBHex = [input subdataWithRange:(NSRange) {6,4}];
 225         NSData* sizeLSBHex = [input subdataWithRange:(NSRange) {10,4}];
 226         NSData* nameHex = [input subdataWithRange:(NSRange) {14,95}];
 227         // windows attributes here?
 228
 229         unsigned char c;
 230         UInt16 u16_bigendian;
 231         SInt32 s32_bigendian;
 232         NSMutableDictionary* tfFile = [[NSMutableDictionary alloc] init];
 233         [mjdHex getBytes:&u16_bigendian];
 234         [tfFile setObject:[NSNumber numberWithInt:EndianU16_BtoN(u16_bigendian)] forKey:@"rawMJD"];
 235         [hrHex getBytes:&c];
 236         [tfFile setObject:[NSNumber numberWithChar:c] forKey:@"hour"];
 237         [minHex getBytes:&c];
 238         [tfFile setObject:[NSNumber numberWithChar:c] forKey:@"min"];
 239         [secHex getBytes:&c];
 240         [tfFile setObject:[NSNumber numberWithChar:c] forKey:@"sec"];
 241         [typeHex getBytes:&c];
 242         [tfFile setObject:[NSNumber numberWithChar:c] forKey:@"type"];
 243         [sizeMSBHex getBytes:&s32_bigendian];
 244         [tfFile setObject:[NSNumber numberWithLong:EndianS32_BtoN(s32_bigendian)] forKey:@"sizeMSB"];
 245         [sizeLSBHex getBytes:&s32_bigendian];
 246         [tfFile setObject:[NSNumber numberWithLong:EndianS32_BtoN(s32_bigendian)] forKey:@"sizeLSB"];
 247         NSString* name = [[NSString alloc] initWithData:nameHex encoding:NSISOLatin1StringEncoding];
 248         name = [name substringToIndex:[name rangeOfCharacterFromSet:[NSCharacterSet characterSetWithRange:(NSRange){0,1}]].location]; //trim anything past the 0x00
 249         name = [name stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithRange:(NSRange){5,1}]]; //remove extraneous 0x05 on some peoples names.
 250         [tfFile setObject:name forKey:@"name"];
 251 //      NSLog([tfFile description]);
 252         return tfFile;
 253 }
 254
 255 - (NSDictionary*) extractAttributes:(NSDictionary*)fattrs {
 256         NSMutableDictionary* tfFile = [[NSMutableDictionary alloc] init];
 257         //dates
 258         NSCalendarDate* date = [[NSCalendarDate alloc] initWithTimeInterval:0 sinceDate:[fattrs fileModificationDate]];
 259         int min = [date minuteOfHour];
 260         int sec = [date secondOfMinute];
 261         int hour = [date hourOfDay];
 262         int MJDdate;
 263         NSCalendarDate *mjdref = [[[NSCalendarDate alloc] initWithYear:1858 month:11 day:17 hour:0 minute:0 second:0 timeZone:[NSTimeZone timeZoneWithAbbreviation:@"GMT"]] autorelease];
 264         [date years:NULL months:NULL days:&MJDdate hours:NULL minutes:NULL seconds:NULL sinceDate:mjdref];
 265         [tfFile setObject:[NSNumber numberWithInt:MJDdate] forKey:@"rawMJD"];
 266         [tfFile setObject:[NSNumber numberWithInt:hour] forKey:@"hour"];
 267         [tfFile setObject:[NSNumber numberWithInt:min] forKey:@"min"];
 268         [tfFile setObject:[NSNumber numberWithInt:sec] forKey:@"sec"];
 269
 270         //types
 271         NSString* type = [fattrs fileType];
 272         if ([type isEqualToString:@"NSFileTypeDirectory"])
 273                 [tfFile setObject:[NSNumber numberWithInt:1] forKey:@"type"];
 274         else
 275                 [tfFile setObject:[NSNumber numberWithInt:2] forKey:@"type"];
 276
 277         //sizes
 278         long long size = [fattrs fileSize];
 279         [tfFile setObject:[NSNumber numberWithLong:size/0xFFFFFFFF] forKey:@"sizeMSB"];
 280         [tfFile setObject:[NSNumber numberWithLong:size%0xFFFFFFFF] forKey:@"sizeLSB"];
 281         return tfFile;
 282 }
 283
 284 -(NSData*) getDataFromTFFile:(NSDictionary*) infile withName:(NSString*) inputName {
 285         UInt16 u16_bigendian = EndianU16_NtoB([[infile objectForKey:@"rawMJD"] shortValue]);
 286         NSMutableData* build = [NSMutableData dataWithBytes:&u16_bigendian length:2];
 287         char c = [[infile objectForKey:@"hour"]charValue];
 288         [build appendData:[NSData dataWithBytes:&c length:1]];
 289         c = [[infile objectForKey:@"min"] charValue];
 290         [build appendData:[NSData dataWithBytes:&c length:1]];
 291         c = [[infile objectForKey:@"sec"] charValue];
 292         [build appendData:[NSData dataWithBytes:&c length:1]];
 293         c = [[infile objectForKey:@"type"] charValue];
 294         [build appendData:[NSData dataWithBytes:&c length:1]];
 295         SInt32 s32_bigendian = EndianS32_NtoB([[infile objectForKey:@"sizeMSB"] longValue]);
 296         [build appendData:[NSData dataWithBytes:&s32_bigendian length:4]];
 297         s32_bigendian = EndianS32_NtoB([[infile objectForKey:@"sizeLSB"] longValue]);
 298         [build appendData:[NSData dataWithBytes:&s32_bigendian length:4]];
 299         NSData * name = [inputName dataUsingEncoding:NSISOLatin1StringEncoding];
 300         [build appendData:name];
 301         if ([name length] < 95)
 302                 [build increaseLengthBy:95-[name length]];
 303         // other file stuff (5 bytes) - maybe add this later?
 304         [build increaseLengthBy:5];
 305         return build;
 306 }
 307
 308 - (void) convertRawDataToUseful:(NSMutableDictionary*) input {
 309
 310         //turn MJD and time into a useful object
 311         if ([[input objectForKey:@"type"] intValue] == 1) { //folder
 312                 [input setObject:@"" forKey:@"date"];
 313                 [input setObject:[NSDate distantPast] forKey:@"sortdate"];
 314                 [input setObject:[NSImage imageNamed:@"folder.tif"] forKey:@"icon"];
 315                 if ([[input objectForKey:@"name"] isEqualToString:@".."])
 316                         [input setObject:@".. (Parent folder)" forKey:@"name"];
 317                 [input setObject:@"0000" forKey:@"suffix"];
 318         } else {
 319                 if ([[input objectForKey:@"name"] hasSuffix:@".tgd"]) [input setObject:[NSImage imageNamed:@"tgd.tif"] forKey:@"icon"];
 320                 else if ([[input objectForKey:@"name"] hasSuffix:@".tap"]) [input setObject:[NSImage imageNamed:@"package.tif"] forKey:@"icon"];
 321                 else if ([[input objectForKey:@"name"] hasSuffix:@".mp3"]) [input setObject:[NSImage imageNamed:@"mp3.tif"] forKey:@"icon"];
 322                 else if ([[input objectForKey:@"name"] hasSuffix:@".txt"] || [[input objectForKey:@"name"] hasSuffix:@".ini"]) [input setObject:[NSImage imageNamed:@"txt.tif"] forKey:@"icon"];
 323                 else [input setObject:[NSImage imageNamed:@"file.tif"] forKey:@"icon"];
 324                 [input setObject:[[input objectForKey:@"name"] pathExtension] forKey:@"suffix"];
 325
 326                 int daysSinceRef = [[input objectForKey:@"rawMJD"] intValue] - 51910; // ref date is 1/1/2001 51911??
 327                 float secsSinceRef = (((daysSinceRef * 24 + [[input objectForKey:@"hour"] intValue])* 60 + [[input objectForKey:@"min"] intValue]) * 60);
 328                 NSCalendarDate* tempDate = [NSCalendarDate dateWithTimeIntervalSinceReferenceDate:secsSinceRef];
 329                 //now adjust for timezone
 330                 int offset = [[NSTimeZone defaultTimeZone] secondsFromGMTForDate:tempDate];
 331                 //NSLog(@"TZ offset:%i td:%@", offset, [tempDate description]);
 332                 NSCalendarDate* finalDate = [tempDate dateByAddingYears:0 months:0 days:0 hours:0 minutes:0 seconds:-1*offset];
 333                 //NSLog(@"TZ td:%@", [finalDate description]);
 334                 [input setObject:finalDate forKey:@"date"];
 335                 [input setObject:finalDate forKey:@"sortdate"];
 336         }
 337
 338         //turn size fields into a proper number
 339         double size;
 340         double big =  pow( 256., 4. );
 341         double lsb = [[input objectForKey:@"sizeLSB"] doubleValue];
 342         double msb = [[input objectForKey:@"sizeMSB"] doubleValue];
 343         if (lsb < 0)
 344                 lsb = lsb + pow( 256., 4. ); //rollover
 345         if (msb < 0)
 346                 msb = msb + pow( 256., 4. ); //rollover
 347         size = msb * big + lsb;
 348         [input setObject:[NSNumber numberWithDouble:size] forKey:@"fileSize"];
 349
 350         NSString* ret = nil;
 351         if( size == 0. ) ret = NSLocalizedString( @" - ", "no file size" );
 352         else if( size > 0. && size < 1024. ) ret = [NSString stringWithFormat:NSLocalizedString( @"%.0f B", "file size measured in bytes" ), size];
 353         else if( size >= 1024. && size < pow( 1024., 2. ) ) ret = [NSString stringWithFormat:NSLocalizedString( @"%.1f KB", "file size measured in kilobytes" ), ( size / 1024. )];
 354         else if( size >= pow( 1024., 2. ) && size < pow( 1024., 3. ) ) ret = [NSString stringWithFormat:NSLocalizedString( @"%.2f MB", "file size measured in megabytes" ), ( size / pow( 1024., 2. ) )];
 355         else if( size >= pow( 1024., 3. ) && size < pow( 1024., 4. ) ) ret = [NSString stringWithFormat:NSLocalizedString( @"%.2f GB", "file size measured in gigabytes" ), ( size / pow( 1024., 3. ) )];
 356         else if( size >= pow( 1024., 4. ) ) ret = [NSString stringWithFormat:NSLocalizedString( @"%.4f TB", "file size measured in terabytes" ), ( size / pow( 1024., 4. ) )];
 357         else ret = NSLocalizedString( @" ? ", "unknown file size" );
 358         [input setObject:ret forKey:@"size"];
 359 }
 360
 361 - (NSDictionary*) extractDataFromRecHeader:(NSString*)file forModel:(NSString*)model {
 362         NSMutableDictionary* extracts = [NSMutableDictionary dictionaryWithCapacity:8];
 363         NSData* fileData = [NSData dataWithContentsOfFile:file];
 364
 365         int modelOffset = 0;
 366         if ([model isEqualToString:@"TF5800"]) modelOffset = 4;
 367         unsigned char c; //1byte
 368         UInt16 u16_bigendian;
 369         [[fileData subdataWithRange:(NSRange) {8,2}] getBytes:&u16_bigendian];
 370         [extracts setObject:[NSNumber numberWithShort:EndianU16_BtoN(u16_bigendian)] forKey:@"duration"];
 371         NSString* channel = [[NSString alloc] initWithData:[fileData subdataWithRange:(NSRange) {28,24}] encoding:NSISOLatin1StringEncoding];
 372         NSRange r = [channel rangeOfCharacterFromSet:[NSCharacterSet characterSetWithRange:(NSRange){0,1}]]; //trim anything past the 0x00
 373         if (r.location != NSNotFound)
 374                 channel = [channel substringToIndex:r.location];
 375         channel = [channel stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithRange:(NSRange){5,1}]]; //remove extraneous 0x05 on some peoples names.
 376         [extracts setObject:channel forKey:@"channel"];
 377         [[fileData subdataWithRange:(NSRange) {76+modelOffset,2}] getBytes:&u16_bigendian];
 378         [extracts setObject:[NSNumber numberWithInt:EndianU16_BtoN(u16_bigendian)] forKey:@"rawMJD"];
 379         int daysSinceRef = [[extracts objectForKey:@"rawMJD"] intValue] - 51910; // ref date is 1/1/2001 51911??
 380         [[fileData subdataWithRange:(NSRange) {78+modelOffset,1}] getBytes:&c];
 381         NSNumber* hour = [NSNumber numberWithChar:c];
 382         [[fileData subdataWithRange:(NSRange) {79+modelOffset,1}] getBytes:&c];
 383         NSNumber* min = [NSNumber numberWithChar:c];
 384         float secsSinceRef = (((daysSinceRef * 24 + [hour intValue])* 60 + [min intValue]) * 60);
 385         NSCalendarDate* finalDate = [NSCalendarDate dateWithTimeIntervalSinceReferenceDate:secsSinceRef];
 386         [extracts setObject:finalDate forKey:@"startTime"];
 387         [[fileData subdataWithRange:(NSRange) {85+modelOffset,1}] getBytes:&c];
 388         NSString* name = [[NSString alloc] initWithData:[fileData subdataWithRange:(NSRange) {87+modelOffset,c}] encoding:NSISOLatin1StringEncoding];
 389         [extracts setObject:name forKey:@"name"];
 390         NSString* desc = [[NSString alloc] initWithData:[fileData subdataWithRange:(NSRange) {87+c+modelOffset,255}] encoding:NSISOLatin1StringEncoding];
 391         r = [desc rangeOfCharacterFromSet:[NSCharacterSet characterSetWithRange:(NSRange){0,1}]]; //trim anything past the 0x00
 392         if (r.location != NSNotFound)
 393                 desc = [desc substringToIndex:r.location];
 394         desc = [desc stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithRange:(NSRange){5,1}]]; //remove extraneous 0x05 on some peoples names.
 395         [extracts setObject:desc forKey:@"description"];
 396         [[fileData subdataWithRange:(NSRange) {362+modelOffset,2}] getBytes:&u16_bigendian];
 397         NSString* extInfo = [[NSString alloc] initWithData:[fileData subdataWithRange:(NSRange) {369+modelOffset,EndianU16_BtoN(u16_bigendian)}] encoding:NSISOLatin1StringEncoding];
 398         [extracts setObject:extInfo forKey:@"extInfo"];
 399         //NSData* movieData = [fileData subdataWithRange:(NSRange) {1656,([fileData length]-1656)}];
 400         //[movieData writeToFile:@"/Volumes/Tyr/cazlar/testMovie.ts" atomically:YES];
 401         //NSLog([extracts description]);
 402         return extracts;
 403 }
 404
 405
 406
 407 @end