X-Git-Url: https://git.yukkurigames.com/?p=enjoyable.git;a=blobdiff_plain;f=Classes%2FNJDeviceViewController.m;h=5d0bf6abe72593530525f63d3de7abb73a0d0ade;hp=f9ff141d1e53a0c01e91d8478f397469e74623af;hb=e7c0b2d96e9e3209b5ba80cc1fdc8e7213cb5302;hpb=5bf06d6aca4bf51b0a347021d69c90dd198c7cf8

diff --git a/Classes/NJDeviceViewController.m b/Classes/NJDeviceViewController.m
index f9ff141..5d0bf6a 100644
--- a/Classes/NJDeviceViewController.m
+++ b/Classes/NJDeviceViewController.m
@@ -36,6 +36,14 @@
     [self expandRecursive:[self.delegate deviceViewController:self elementForUID:uid]];
 }
 
+- (void)beginUpdates {
+    [self.inputsTree beginUpdates];
+}
+
+- (void)endUpdates {
+    [self.inputsTree endUpdates];
+}
+
 - (void)reexpandAll {
     for (NSString *uid in [_expanded copy])
         [self expandRecursiveByUID:uid];
@@ -47,22 +55,18 @@
 }
 
 - (void)addedDevice:(NJDevice *)device atIndex:(NSUInteger)idx {
-    [self.inputsTree beginUpdates];
     [self.inputsTree insertItemsAtIndexes:[[NSIndexSet alloc] initWithIndex:idx]
                                   inParent:nil
                              withAnimation:NSTableViewAnimationEffectFade];
     [self reexpandAll];
-    [self.inputsTree endUpdates];
     self.noDevicesNotice.hidden = YES;
 }
 
 - (void)removedDevice:(NJDevice *)device atIndex:(NSUInteger)idx {
     BOOL anyDevices = !![self.delegate numberOfDevicesInDeviceList:self];
-    [self.inputsTree beginUpdates];
     [self.inputsTree removeItemsAtIndexes:[[NSIndexSet alloc] initWithIndex:idx]
                                   inParent:nil
                              withAnimation:NSTableViewAnimationEffectFade];
-    [self.inputsTree endUpdates];
     self.noDevicesNotice.hidden = anyDevices || !self.hidStoppedNotice.isHidden;
 }