Fixed unregistering controllers and emulating extra buttons

Limelight/Input/Controller.h

@@ -12,6 +12,7 @@
 
 @property (nonatomic) int playerIndex;
 @property (nonatomic) int lastButtonFlags;
+@property (nonatomic) int emulatingButtonFlags;
 @property (nonatomic) char lastLeftTrigger;
 @property (nonatomic) char lastRightTrigger;
 @property (nonatomic) short lastLeftStickX;

Limelight/Input/Controller.m

@@ -10,6 +10,6 @@
 
 @implementation Controller
 @synthesize playerIndex;
-@synthesize lastButtonFlags, lastLeftTrigger, lastRightTrigger;
+@synthesize lastButtonFlags, emulatingButtonFlags, lastLeftTrigger, lastRightTrigger;
 @synthesize lastLeftStickX, lastLeftStickY, lastRightStickX, lastRightStickY;
 @end

Limelight/Input/ControllerSupport.m

@@ -23,10 +23,9 @@
     
 #define EMULATING_SELECT     0x1
 #define EMULATING_SPECIAL    0x2
-    int _emulatingButtonFlags;
 }
 
-// UPDATE_BUTTON_FLAG(flag, pressed)
+// UPDATE_BUTTON_FLAG(controller, flag, pressed)
 #define UPDATE_BUTTON_FLAG(controller, x, y) \
 ((y) ? [self setButtonFlag:controller flags:x] : [self clearButtonFlag:controller flags:x])
 
@@ -70,16 +69,16 @@
 
 -(void) handleSpecialCombosReleased:(Controller*)controller releasedButtons:(int)releasedButtons
 {
-    if ((_emulatingButtonFlags & EMULATING_SELECT) &&
+    if ((controller.emulatingButtonFlags & EMULATING_SELECT) &&
         ((releasedButtons & LB_FLAG) || (releasedButtons & PLAY_FLAG))) {
         controller.lastButtonFlags &= ~BACK_FLAG;
-        _emulatingButtonFlags &= ~EMULATING_SELECT;
+        controller.emulatingButtonFlags &= ~EMULATING_SELECT;
     }
-    if ((_emulatingButtonFlags & EMULATING_SPECIAL) &&
+    if ((controller.emulatingButtonFlags & EMULATING_SPECIAL) &&
         ((releasedButtons & RB_FLAG) || (releasedButtons & PLAY_FLAG) ||
          (releasedButtons & BACK_FLAG))) {
             controller.lastButtonFlags &= ~SPECIAL_FLAG;
-            _emulatingButtonFlags &= ~EMULATING_SPECIAL;
+            controller.emulatingButtonFlags &= ~EMULATING_SPECIAL;
         }
 }
 
@@ -91,13 +90,13 @@
         if (controller.lastButtonFlags & LB_FLAG) {
             controller.lastButtonFlags |= BACK_FLAG;
             controller.lastButtonFlags &= ~(pressedButtons & (PLAY_FLAG | LB_FLAG));
-            _emulatingButtonFlags |= EMULATING_SELECT;
+            controller.emulatingButtonFlags |= EMULATING_SELECT;
         }
         // If (RB or select) and start are down, trigger special
         else if ((controller.lastButtonFlags & RB_FLAG) || (controller.lastButtonFlags & BACK_FLAG)) {
             controller.lastButtonFlags |= SPECIAL_FLAG;
             controller.lastButtonFlags &= ~(pressedButtons & (PLAY_FLAG | RB_FLAG | BACK_FLAG));
-            _emulatingButtonFlags |= EMULATING_SPECIAL;
+            controller.emulatingButtonFlags |= EMULATING_SPECIAL;
         }
     }
     
@@ -145,11 +144,8 @@
     [_controllerStreamLock unlock];
 }
 
--(void) unregisterControllerCallbacks
+-(void) unregisterControllerCallbacks:(GCController*) controller
 {
-    for (int i = 0; i < [[GCController controllers] count]; i++) {
-        GCController *controller = [GCController controllers][i];
-        
     if (controller != NULL) {
         controller.controllerPausedHandler = NULL;
         
@@ -161,14 +157,9 @@
         }
     }
 }
-    [_controllers removeAllObjects];
-}
 
--(void) registerControllerCallbacks
+-(void) registerControllerCallbacks:(GCController*) controller
 {
-    for (int i = 0; i < [[GCController controllers] count]; i++) {
-        GCController *controller = [GCController controllers][i];
-        
     if (controller != NULL) {
         controller.controllerPausedHandler = ^(GCController *controller) {
             Controller* limeController = [_controllers objectForKey:[NSNumber numberWithInteger:controller.playerIndex]];
@@ -236,11 +227,11 @@
                 [self updateFinished:limeController];
             };
         }
+    } else {
+        NSLog(@"ERROR: Tried to register controller callbacks on NULL controller");
     }
 }
 
-}
-
 -(void) updateAutoOnScreenControlMode
 {
     // Auto on-screen control support may not be enabled
@@ -300,19 +291,21 @@
     _controllers = [[NSMutableDictionary alloc] init];
     _controllerNumbers = 0;
     
-    NSLog(@"Number of controllers connected: %ld", [[GCController controllers] count]);
+    NSLog(@"Number of controllers connected: %ld", (long)[[GCController controllers] count]);
     for (GCController* controller in [GCController controllers]) {
         [self assignController:controller];
+        [self registerControllerCallbacks:controller];
+        [self updateAutoOnScreenControlMode];
     }
     
     self.connectObserver = [[NSNotificationCenter defaultCenter] addObserverForName:GCControllerDidConnectNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification *note) {
         NSLog(@"Controller connected!");
         
-        GCController* controller = (GCController*) note.object;
+        GCController* controller = note.object;
         [self assignController:controller];
         
         // Register callbacks on the new controller
-        [self registerControllerCallbacks];
+        [self registerControllerCallbacks:controller];
         
         // Re-evaluate the on-screen control mode
         [self updateAutoOnScreenControlMode];
@@ -321,6 +314,7 @@
         NSLog(@"Controller disconnected!");
         
         GCController* controller = note.object;
+        [self unregisterControllerCallbacks:controller];
         [_controllers removeObjectForKey:[NSNumber numberWithInteger:controller.playerIndex]];
         _controllerNumbers &= ~(1 << controller.playerIndex);
         NSLog(@"Unassigning controller index: %ld", (long)controller.playerIndex);
@@ -329,9 +323,6 @@
         [self updateAutoOnScreenControlMode];
     }];
     
-    // Register for controller callbacks on any existing controllers
-    [self registerControllerCallbacks];
-    
     return self;
 }
 
@@ -339,8 +330,11 @@
 {
     [[NSNotificationCenter defaultCenter] removeObserver:self.connectObserver];
     [[NSNotificationCenter defaultCenter] removeObserver:self.disconnectObserver];
-    
+    [_controllers removeAllObjects];
-    [self unregisterControllerCallbacks];
+    _controllerNumbers = 0;
+    for (GCController* controller in [GCController controllers]) {
+        [self unregisterControllerCallbacks:controller];
+    }
 }
 
 @end