IJInventoryWindowController.m 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391
  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 "IJItemMatrix.h"
  12. #import "IJItemPickerWindowController.h"
  13. @implementation IJInventoryWindowController
  14. @synthesize outlineView;
  15. @synthesize worldSelectionControl;
  16. @synthesize statusTextField;
  17. @synthesize inventoryMatrixContainer, quickMatrixContainer, armorMatrixContainer;
  18. - (void)awakeFromNib
  19. {
  20. armorItem = [NSMutableArray array];
  21. quickItem = [NSMutableArray array];
  22. inventoryItem = [NSMutableArray array];
  23. rootItems = [[NSArray alloc] initWithObjects:armorItem, quickItem, inventoryItem, nil];
  24. statusTextField.stringValue = @"";
  25. inventoryMatrix = [IJItemMatrix itemMatrixWithFrame:inventoryMatrixContainer.bounds rows:3 columns:9];
  26. [inventoryMatrixContainer addSubview:inventoryMatrix];
  27. quickMatrix = [IJItemMatrix itemMatrixWithFrame:quickMatrixContainer.bounds rows:1 columns:9];
  28. [quickMatrixContainer addSubview:quickMatrix];
  29. armorMatrix = [IJItemMatrix itemMatrixWithFrame:armorMatrixContainer.bounds rows:4 columns:1];
  30. [armorMatrixContainer addSubview:armorMatrix];
  31. }
  32. - (void)dealloc
  33. {
  34. [inventory release];
  35. [rootItems release];
  36. [level release];
  37. [super dealloc];
  38. }
  39. #pragma mark -
  40. #pragma mark World Selection
  41. - (void)loadWorldAtIndex:(int)worldIndex
  42. {
  43. [armorItem removeAllObjects];
  44. [quickItem removeAllObjects];
  45. [inventoryItem removeAllObjects];
  46. // Reload data here because we have just invalidated all of the items used in the outline view.
  47. [outlineView reloadData];
  48. [self willChangeValueForKey:@"worldTime"];
  49. [level release];
  50. level = nil;
  51. [inventory release];
  52. inventory = nil;
  53. [self didChangeValueForKey:@"worldTime"];
  54. if (![IJMinecraftLevel worldExistsAtIndex:worldIndex])
  55. {
  56. 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.");
  57. return;
  58. }
  59. sessionLockValue = [IJMinecraftLevel writeToSessionLockAtIndex:worldIndex];
  60. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  61. {
  62. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable obtain the session lock.");
  63. return;
  64. }
  65. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  66. NSData *fileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:levelPath]];
  67. if (!fileData)
  68. {
  69. // Error loading
  70. [outlineView reloadData];
  71. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", levelPath);
  72. return;
  73. }
  74. [self willChangeValueForKey:@"worldTime"];
  75. level = [[IJMinecraftLevel nbtContainerWithData:fileData] retain];
  76. inventory = [[level inventory] retain];
  77. [self didChangeValueForKey:@"worldTime"];
  78. // Add placeholder inventory items:
  79. for (int i = 0; i < IJInventorySlotQuickLast + 1 - IJInventorySlotQuickFirst; i++)
  80. [quickItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotQuickFirst + i]];
  81. for (int i = 0; i < IJInventorySlotNormalLast + 1 - IJInventorySlotNormalFirst; i++)
  82. [inventoryItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotNormalFirst + i]];
  83. for (int i = 0; i < IJInventorySlotArmorLast + 1 - IJInventorySlotArmorFirst; i++)
  84. [armorItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotArmorFirst + i]];
  85. // Overwrite the placeholders with actual inventory:
  86. for (IJInventoryItem *item in inventory)
  87. {
  88. if (IJInventorySlotQuickFirst <= item.slot && item.slot <= IJInventorySlotQuickLast)
  89. {
  90. [quickItem replaceObjectAtIndex:item.slot - IJInventorySlotQuickFirst withObject:item];
  91. }
  92. else if (IJInventorySlotNormalFirst <= item.slot && item.slot <= IJInventorySlotNormalLast)
  93. {
  94. [inventoryItem replaceObjectAtIndex:item.slot - IJInventorySlotNormalFirst withObject:item];
  95. }
  96. else if (IJInventorySlotArmorFirst <= item.slot && item.slot <= IJInventorySlotArmorLast)
  97. {
  98. [armorItem replaceObjectAtIndex:item.slot - IJInventorySlotArmorFirst withObject:item];
  99. }
  100. }
  101. [outlineView reloadData];
  102. [outlineView expandItem:nil expandChildren:YES];
  103. for (IJInventoryItem *item in inventoryItem)
  104. {
  105. int slot = item.slot - IJInventorySlotNormalFirst;
  106. NSImageCell *cell = [inventoryMatrix cellAtRow:slot/9 column:slot%9];
  107. cell.image = item.image;
  108. }
  109. for (IJInventoryItem *item in quickItem)
  110. {
  111. int slot = item.slot - IJInventorySlotQuickFirst;
  112. NSImageCell *cell = [quickMatrix cellAtRow:0 column:slot];
  113. cell.image = item.image;
  114. }
  115. for (IJInventoryItem *item in armorItem)
  116. {
  117. int slot = item.slot - IJInventorySlotArmorFirst;
  118. NSImageCell *cell = [armorMatrix cellAtRow:3-slot column:0];
  119. cell.image = item.image;
  120. }
  121. dirty = NO;
  122. statusTextField.stringValue = @"";
  123. }
  124. - (void)saveToWorldAtIndex:(int)worldIndex
  125. {
  126. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  127. {
  128. NSBeginCriticalAlertSheet(@"Another application has modified this world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"The session lock was changed by another application.");
  129. return;
  130. }
  131. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  132. NSMutableArray *newInventory = [NSMutableArray array];
  133. for (NSArray *items in rootItems)
  134. {
  135. for (IJInventoryItem *item in items)
  136. {
  137. if (item.count > 0 && item.itemId > 0)
  138. [newInventory addObject:item];
  139. }
  140. }
  141. [level setInventory:newInventory];
  142. NSString *backupPath = [levelPath stringByAppendingPathExtension:@"insidejobbackup"];
  143. BOOL success;
  144. NSError *error = nil;
  145. success = [[NSFileManager defaultManager] removeItemAtPath:backupPath error:&error];
  146. success = [[NSFileManager defaultManager] copyItemAtPath:levelPath
  147. toPath:backupPath
  148. error:&error];
  149. if (!success)
  150. {
  151. 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.");
  152. return;
  153. }
  154. [[level writeData] writeToURL:[NSURL fileURLWithPath:levelPath] atomically:NO];
  155. dirty = NO;
  156. statusTextField.stringValue = @"Saved.";
  157. }
  158. - (void)markDirty
  159. {
  160. dirty = YES;
  161. statusTextField.stringValue = @"World has unsaved changes.";
  162. }
  163. #pragma mark -
  164. #pragma mark Actions
  165. - (IBAction)worldSelectionChanged:(id)sender
  166. {
  167. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  168. [self loadWorldAtIndex:worldIndex];
  169. }
  170. - (void)saveDocument:(id)sender
  171. {
  172. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  173. [self saveToWorldAtIndex:worldIndex];
  174. }
  175. - (void)delete:(id)sender
  176. {
  177. IJInventoryItem *item = [outlineView itemAtRow:[outlineView selectedRow]];
  178. item.count = 0;
  179. item.itemId = 0;
  180. item.damage = 0;
  181. [self markDirty];
  182. [outlineView reloadItem:item];
  183. }
  184. - (BOOL)validateUserInterfaceItem:(id <NSValidatedUserInterfaceItem>)anItem
  185. {
  186. if (anItem.action == @selector(delete:))
  187. {
  188. return [outlineView selectedRow] != -1 && ![rootItems containsObject:[outlineView itemAtRow:[outlineView selectedRow]]];
  189. }
  190. return YES;
  191. }
  192. - (NSNumber *)worldTime
  193. {
  194. return [level worldTimeContainer].numberValue;
  195. }
  196. - (void)setWorldTime:(NSNumber *)number
  197. {
  198. [self willChangeValueForKey:@"worldTime"];
  199. [level worldTimeContainer].numberValue = number;
  200. [self didChangeValueForKey:@"worldTime"];
  201. [self markDirty];
  202. }
  203. #pragma mark -
  204. #pragma mark Inventory Outline View
  205. - (id)outlineView:(NSOutlineView *)theOutlineView child:(NSInteger)index ofItem:(id)item
  206. {
  207. if (item == nil)
  208. {
  209. return [rootItems objectAtIndex:index];
  210. }
  211. else
  212. {
  213. return [item objectAtIndex:index];
  214. }
  215. }
  216. - (BOOL)outlineView:(NSOutlineView *)theOutlineView isItemExpandable:(id)item
  217. {
  218. return item == nil || [rootItems containsObject:item];
  219. }
  220. - (NSInteger)outlineView:(NSOutlineView *)theOutlineView numberOfChildrenOfItem:(id)item
  221. {
  222. if (item == nil)
  223. {
  224. return 3;
  225. }
  226. else if ([rootItems containsObject:item])
  227. {
  228. return [(NSArray *)item count];
  229. }
  230. else
  231. {
  232. return 0;
  233. }
  234. }
  235. - (id)outlineView:(NSOutlineView *)theOutlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
  236. {
  237. if ([rootItems containsObject:item])
  238. {
  239. if ([tableColumn.identifier isEqual:@"slot"])
  240. {
  241. if (item == armorItem)
  242. return @"Armor";
  243. else if (item == quickItem)
  244. return @"Quick Inventory";
  245. else if (item == inventoryItem)
  246. return @"Inventory";
  247. }
  248. else
  249. {
  250. return nil;
  251. }
  252. }
  253. IJInventoryItem *invItem = item;
  254. if ([tableColumn.identifier isEqual:@"slot"])
  255. {
  256. return [NSString stringWithFormat:@"%d", invItem.slot];
  257. }
  258. else if ([tableColumn.identifier isEqual:@"id"])
  259. {
  260. if (invItem.itemId)
  261. return [NSNumber numberWithShort:invItem.itemId];
  262. else
  263. return nil;
  264. }
  265. else if ([tableColumn.identifier isEqual:@"item"])
  266. {
  267. if (invItem.itemId)
  268. return invItem.itemName;
  269. else
  270. return @"";
  271. }
  272. else if ([tableColumn.identifier isEqual:@"count"])
  273. {
  274. if (invItem.count)
  275. return [NSNumber numberWithUnsignedChar:invItem.count];
  276. else
  277. return nil;
  278. }
  279. else if ([tableColumn.identifier isEqual:@"damage"])
  280. {
  281. if (invItem.damage)
  282. return [NSNumber numberWithShort:invItem.damage];
  283. else
  284. return nil;
  285. }
  286. return nil;
  287. }
  288. - (void)outlineView:(NSOutlineView *)theOutlineView setObjectValue:(id)object forTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
  289. {
  290. IJInventoryItem *invItem = item;
  291. if ([tableColumn.identifier isEqual:@"id"])
  292. {
  293. invItem.itemId = [object shortValue];
  294. [self markDirty];
  295. }
  296. else if ([tableColumn.identifier isEqual:@"count"])
  297. {
  298. invItem.count = [object unsignedCharValue];
  299. if (invItem.count > 64)
  300. invItem.count = 64;
  301. [self markDirty];
  302. }
  303. else if ([tableColumn.identifier isEqual:@"damage"])
  304. {
  305. invItem.damage = [object shortValue];
  306. [self markDirty];
  307. }
  308. }
  309. - (BOOL)outlineView:(NSOutlineView *)theOutlineView shouldEditTableColumn:(NSTableColumn *)tableColumn item:(id)item
  310. {
  311. if ([rootItems containsObject:item])
  312. return NO;
  313. else if ([tableColumn.identifier isEqual:@"item"])
  314. {
  315. IJInventoryItem *invItem = item;
  316. [[IJItemPickerWindowController sharedController] showPickerWithInitialItemId:invItem.itemId completionBlock:^(uint16_t itemId) {
  317. invItem.itemId = itemId;
  318. [outlineView reloadItem:item];
  319. [self markDirty];
  320. }];
  321. return NO;
  322. }
  323. else
  324. {
  325. return [tableColumn.identifier isEqual:@"slot"] == NO;
  326. }
  327. }
  328. @end