IJInventoryWindowController.m 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442
  1. //
  2. // IJInventoryWindowController.m
  3. // InsideJob
  4. //
  5. // Created by Adam Preble on 10/7/10.
  6. // Copyright 2010 Adam Preble. All rights reserved.
  7. //
  8. #import "IJInventoryWindowController.h"
  9. #import "IJMinecraftLevel.h"
  10. #import "IJInventoryItem.h"
  11. #import "IJInventoryView.h"
  12. #import "IJItemPropertiesViewController.h"
  13. #import "MAAttachedWindow.h"
  14. @interface IJInventoryWindowController ()
  15. - (void)removePropertiesWindow;
  16. @end
  17. @implementation IJInventoryWindowController
  18. @synthesize worldSelectionControl;
  19. @synthesize statusTextField;
  20. @synthesize inventoryView, armorView, quickView;
  21. @synthesize itemSearchField, itemTableView;
  22. - (void)awakeFromNib
  23. {
  24. armorInventory = [[NSMutableArray alloc] init];
  25. quickInventory = [[NSMutableArray alloc] init];
  26. normalInventory = [[NSMutableArray alloc] init];
  27. statusTextField.stringValue = @"";
  28. [inventoryView setRows:3 columns:9];
  29. [quickView setRows:1 columns:9];
  30. [armorView setRows:4 columns:1];
  31. inventoryView.delegate = self;
  32. quickView.delegate = self;
  33. armorView.delegate = self;
  34. // Item Table View setup
  35. NSArray *keys = [[IJInventoryItem itemIdLookup] allKeys];
  36. keys = [keys sortedArrayUsingSelector:@selector(compare:)];
  37. allItemIds = [[NSArray alloc] initWithArray:keys];
  38. filteredItemIds = [allItemIds retain];
  39. }
  40. - (void)dealloc
  41. {
  42. [propertiesViewController release];
  43. [armorInventory release];
  44. [quickInventory release];
  45. [normalInventory release];
  46. [inventory release];
  47. [level release];
  48. [super dealloc];
  49. }
  50. #pragma mark -
  51. #pragma mark World Selection
  52. - (void)loadWorldAtIndex:(int)worldIndex
  53. {
  54. [armorInventory removeAllObjects];
  55. [quickInventory removeAllObjects];
  56. [normalInventory removeAllObjects];
  57. [self willChangeValueForKey:@"worldTime"];
  58. [level release];
  59. level = nil;
  60. [inventory release];
  61. inventory = nil;
  62. [self didChangeValueForKey:@"worldTime"];
  63. if (![IJMinecraftLevel worldExistsAtIndex:worldIndex])
  64. {
  65. NSBeginCriticalAlertSheet(@"No world exists in that slot.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Please create a new single player world in this slot using Minecraft and try again.");
  66. return;
  67. }
  68. sessionLockValue = [IJMinecraftLevel writeToSessionLockAtIndex:worldIndex];
  69. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  70. {
  71. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable obtain the session lock.");
  72. return;
  73. }
  74. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  75. NSData *fileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:levelPath]];
  76. if (!fileData)
  77. {
  78. // Error loading
  79. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", levelPath);
  80. return;
  81. }
  82. [self willChangeValueForKey:@"worldTime"];
  83. level = [[IJMinecraftLevel nbtContainerWithData:fileData] retain];
  84. inventory = [[level inventory] retain];
  85. [self didChangeValueForKey:@"worldTime"];
  86. // Add placeholder inventory items:
  87. for (int i = 0; i < IJInventorySlotQuickLast + 1 - IJInventorySlotQuickFirst; i++)
  88. [quickInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotQuickFirst + i]];
  89. for (int i = 0; i < IJInventorySlotNormalLast + 1 - IJInventorySlotNormalFirst; i++)
  90. [normalInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotNormalFirst + i]];
  91. for (int i = 0; i < IJInventorySlotArmorLast + 1 - IJInventorySlotArmorFirst; i++)
  92. [armorInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotArmorFirst + i]];
  93. // Overwrite the placeholders with actual inventory:
  94. for (IJInventoryItem *item in inventory)
  95. {
  96. if (IJInventorySlotQuickFirst <= item.slot && item.slot <= IJInventorySlotQuickLast)
  97. {
  98. [quickInventory replaceObjectAtIndex:item.slot - IJInventorySlotQuickFirst withObject:item];
  99. }
  100. else if (IJInventorySlotNormalFirst <= item.slot && item.slot <= IJInventorySlotNormalLast)
  101. {
  102. [normalInventory replaceObjectAtIndex:item.slot - IJInventorySlotNormalFirst withObject:item];
  103. }
  104. else if (IJInventorySlotArmorFirst <= item.slot && item.slot <= IJInventorySlotArmorLast)
  105. {
  106. [armorInventory replaceObjectAtIndex:item.slot - IJInventorySlotArmorFirst withObject:item];
  107. }
  108. }
  109. NSLog(@"normal: %@", normalInventory);
  110. NSLog(@"quick: %@", quickInventory);
  111. [inventoryView setItems:normalInventory];
  112. [quickView setItems:quickInventory];
  113. [armorView setItems:armorInventory];
  114. dirty = NO;
  115. statusTextField.stringValue = @"";
  116. }
  117. - (void)saveToWorldAtIndex:(int)worldIndex
  118. {
  119. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  120. {
  121. NSBeginCriticalAlertSheet(@"Another application has modified this world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"The session lock was changed by another application.");
  122. return;
  123. }
  124. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  125. NSMutableArray *newInventory = [NSMutableArray array];
  126. for (NSArray *items in [NSArray arrayWithObjects:armorInventory, quickInventory, normalInventory, nil])
  127. {
  128. for (IJInventoryItem *item in items)
  129. {
  130. if (item.count > 0 && item.itemId > 0)
  131. [newInventory addObject:item];
  132. }
  133. }
  134. [level setInventory:newInventory];
  135. NSString *backupPath = [levelPath stringByAppendingPathExtension:@"insidejobbackup"];
  136. BOOL success;
  137. NSError *error = nil;
  138. success = [[NSFileManager defaultManager] removeItemAtPath:backupPath error:&error];
  139. success = [[NSFileManager defaultManager] copyItemAtPath:levelPath
  140. toPath:backupPath
  141. error:&error];
  142. if (!success)
  143. {
  144. NSBeginCriticalAlertSheet(@"An error occurred while saving.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable to create a backup of the existing level file.");
  145. return;
  146. }
  147. [[level writeData] writeToURL:[NSURL fileURLWithPath:levelPath] atomically:NO];
  148. dirty = NO;
  149. statusTextField.stringValue = @"Saved.";
  150. }
  151. - (void)markDirty
  152. {
  153. dirty = YES;
  154. statusTextField.stringValue = @"World has unsaved changes.";
  155. }
  156. #pragma mark -
  157. #pragma mark Actions
  158. - (IBAction)worldSelectionChanged:(id)sender
  159. {
  160. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  161. [self loadWorldAtIndex:worldIndex];
  162. }
  163. - (void)saveDocument:(id)sender
  164. {
  165. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  166. [self saveToWorldAtIndex:worldIndex];
  167. }
  168. - (void)delete:(id)sender
  169. {
  170. // IJInventoryItem *item = [outlineView itemAtRow:[outlineView selectedRow]];
  171. // item.count = 0;
  172. // item.itemId = 0;
  173. // item.damage = 0;
  174. // [self markDirty];
  175. // [outlineView reloadItem:item];
  176. }
  177. - (BOOL)validateUserInterfaceItem:(id <NSValidatedUserInterfaceItem>)anItem
  178. {
  179. return YES;
  180. }
  181. - (NSNumber *)worldTime
  182. {
  183. return [level worldTimeContainer].numberValue;
  184. }
  185. - (void)setWorldTime:(NSNumber *)number
  186. {
  187. [self willChangeValueForKey:@"worldTime"];
  188. [level worldTimeContainer].numberValue = number;
  189. [self didChangeValueForKey:@"worldTime"];
  190. [self markDirty];
  191. }
  192. #pragma mark -
  193. #pragma mark IJInventoryViewDelegate
  194. - (NSMutableArray *)itemArrayForInventoryView:(IJInventoryView *)theInventoryView slotOffset:(int*)slotOffset
  195. {
  196. if (theInventoryView == inventoryView)
  197. {
  198. if (slotOffset) *slotOffset = IJInventorySlotNormalFirst;
  199. return normalInventory;
  200. }
  201. else if (theInventoryView == quickView)
  202. {
  203. if (slotOffset) *slotOffset = IJInventorySlotQuickFirst;
  204. return quickInventory;
  205. }
  206. else if (theInventoryView == armorView)
  207. {
  208. if (slotOffset) *slotOffset = IJInventorySlotArmorFirst;
  209. return armorInventory;
  210. }
  211. return nil;
  212. }
  213. - (void)inventoryView:(IJInventoryView *)theInventoryView removeItemAtIndex:(int)itemIndex
  214. {
  215. int slotOffset = 0;
  216. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  217. if (itemArray)
  218. {
  219. IJInventoryItem *item = [IJInventoryItem emptyItemWithSlot:slotOffset + itemIndex];
  220. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  221. [theInventoryView setItems:itemArray];
  222. }
  223. [self markDirty];
  224. }
  225. - (void)inventoryView:(IJInventoryView *)theInventoryView setItem:(IJInventoryItem *)item atIndex:(int)itemIndex
  226. {
  227. int slotOffset = 0;
  228. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  229. if (itemArray)
  230. {
  231. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  232. item.slot = slotOffset + itemIndex;
  233. [theInventoryView setItems:itemArray];
  234. }
  235. [self markDirty];
  236. }
  237. - (void)inventoryView:(IJInventoryView *)theInventoryView selectedItemAtIndex:(int)itemIndex
  238. {
  239. // Show the properties window for this item.
  240. IJInventoryItem *lastItem = propertiesViewController.item;
  241. //[self removePropertiesWindow];
  242. NSPoint itemLocationInView = [theInventoryView pointForItemAtIndex:itemIndex];
  243. NSPoint point = [theInventoryView convertPoint:itemLocationInView toView:nil];
  244. point.x += 16 + 8;
  245. point.y -= 16;
  246. NSArray *items = [self itemArrayForInventoryView:theInventoryView slotOffset:nil];
  247. IJInventoryItem *item = [items objectAtIndex:itemIndex];
  248. if (item.itemId == 0 || lastItem == item)
  249. return; // can't show info on nothing
  250. if (!propertiesViewController)
  251. {
  252. propertiesViewController = [[IJItemPropertiesViewController alloc] initWithNibName:@"ItemPropertiesView" bundle:nil];
  253. }
  254. propertiesViewController.item = item;
  255. propertiesWindow = [[MAAttachedWindow alloc] initWithView:propertiesViewController.view
  256. attachedToPoint:point
  257. inWindow:self.window
  258. onSide:MAPositionRight
  259. atDistance:0];
  260. [propertiesWindow setDelegate:propertiesViewController]; // Be the delegate so it can be part of the responder chain (to get cancelOperation:).
  261. [propertiesWindow setBackgroundColor:[NSColor controlBackgroundColor]];
  262. [propertiesWindow setViewMargin:10.0];
  263. [propertiesWindow setAlphaValue:1.0];
  264. [[self window] addChildWindow:propertiesWindow ordered:NSWindowAbove];
  265. [propertiesWindow makeKeyAndOrderFront:nil];
  266. observerObject = [[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidResignKeyNotification
  267. object:propertiesWindow
  268. queue:[NSOperationQueue mainQueue]
  269. usingBlock:^(NSNotification *notification) {
  270. [self removePropertiesWindow];
  271. }];
  272. }
  273. #pragma mark -
  274. #pragma mark Item Picker
  275. - (IBAction)updateItemSearchFilter:(id)sender
  276. {
  277. NSString *filterString = [sender stringValue];
  278. if (filterString.length == 0)
  279. {
  280. [filteredItemIds autorelease];
  281. filteredItemIds = [allItemIds retain];
  282. [itemTableView reloadData];
  283. return;
  284. }
  285. NSMutableArray *results = [NSMutableArray array];
  286. for (NSNumber *itemId in allItemIds)
  287. {
  288. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  289. NSRange range = [name rangeOfString:filterString options:NSCaseInsensitiveSearch];
  290. if (range.location != NSNotFound)
  291. [results addObject:itemId];
  292. }
  293. [filteredItemIds autorelease];
  294. filteredItemIds = [results retain];
  295. [itemTableView reloadData];
  296. }
  297. - (NSInteger)numberOfRowsInTableView:(NSTableView *)theTableView
  298. {
  299. return filteredItemIds.count;
  300. }
  301. - (id)tableView:(NSTableView *)theTableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row
  302. {
  303. // TODO: Change this, because the row will not correspond once we support sorting.
  304. NSNumber *itemId = [filteredItemIds objectAtIndex:row];
  305. if ([tableColumn.identifier isEqual:@"itemId"])
  306. {
  307. return itemId;
  308. }
  309. else if ([tableColumn.identifier isEqual:@"image"])
  310. {
  311. return [IJInventoryItem imageForItemId:[itemId shortValue]];
  312. }
  313. else
  314. {
  315. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  316. return name;
  317. }
  318. }
  319. - (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard
  320. {
  321. [pboard declareTypes:[NSArray arrayWithObjects:IJPasteboardTypeInventoryItem, nil] owner:nil];
  322. NSNumber *itemId = [filteredItemIds objectAtIndex:[rowIndexes firstIndex]];
  323. IJInventoryItem *item = [[IJInventoryItem alloc] init];
  324. item.itemId = [itemId shortValue];
  325. item.count = 1;
  326. item.damage = 0;
  327. item.slot = 0;
  328. [pboard setData:[NSKeyedArchiver archivedDataWithRootObject:item]
  329. forType:IJPasteboardTypeInventoryItem];
  330. [item release];
  331. return YES;
  332. }
  333. #pragma mark -
  334. #pragma mark
  335. - (void)reloadInventoryViewForItem:(IJInventoryItem *)item
  336. {
  337. if ([normalInventory containsObject:item])
  338. [inventoryView reloadItemAtIndex:[normalInventory indexOfObject:item]];
  339. else if ([quickInventory containsObject:item])
  340. [quickView reloadItemAtIndex:[quickInventory indexOfObject:item]];
  341. else if ([armorInventory containsObject:item])
  342. [armorView reloadItemAtIndex:[armorInventory indexOfObject:item]];
  343. }
  344. - (void)removePropertiesWindow
  345. {
  346. if (observerObject)
  347. {
  348. [propertiesViewController commitEditing];
  349. [[NSNotificationCenter defaultCenter] removeObserver:observerObject];
  350. observerObject = nil;
  351. [self reloadInventoryViewForItem:propertiesViewController.item];
  352. [self.window removeChildWindow:propertiesWindow];
  353. [propertiesWindow orderOut:nil];
  354. //[propertiesWindow release];
  355. propertiesWindow = nil;
  356. propertiesViewController.item = nil;
  357. }
  358. }
  359. @end