JoystickController.m

   1 //
   2 //  JoystickController.m
   3 //  Enjoy
   4 //
   5 //  Created by Sam McCall on 4/05/09.
   6 //
   7
   8 #import "JoystickController.h"
   9
  10 @implementation JoystickController {
  11     IOHIDManagerRef hidManager;
  12     BOOL programmaticallySelecting;
  13     NSTimer *continuousTimer;
  14 }
  15
  16 @synthesize joysticks;
  17 @synthesize runningTargets;
  18 @synthesize selectedAction;
  19 @synthesize frontWindowOnly;
  20 @synthesize mouseLoc;
  21
  22 - (id)init {
  23     if ((self = [super init])) {
  24         joysticks = [[NSMutableArray alloc] initWithCapacity:16];
  25         runningTargets = [[NSMutableArray alloc] initWithCapacity:32];
  26     }
  27     return self;
  28 }
  29
  30 - (void)dealloc {
  31     [continuousTimer invalidate];
  32     IOHIDManagerClose(hidManager, kIOHIDOptionsTypeNone);
  33     CFRelease(hidManager);
  34 }
  35
  36 - (void)expandRecursive:(id)handler {
  37     if ([handler base])
  38         [self expandRecursive:[handler base]];
  39     [outlineView expandItem:handler];
  40 }
  41
  42 - (void)addRunningTarget:(Target *)target {
  43     if (![runningTargets containsObject:target])
  44         [runningTargets addObject:target];
  45     if (!continuousTimer) {
  46         continuousTimer = [NSTimer scheduledTimerWithTimeInterval:1.f/60.f
  47                                                            target:self
  48                                                          selector:@selector(updateContinuousActions:)
  49                                                          userInfo:nil
  50                                                           repeats:YES];
  51         NSLog(@"Scheduled continuous target timer.");
  52     }
  53 }
  54
  55 static void input_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDValueRef value) {
  56     JoystickController *controller = (__bridge JoystickController *)ctx;
  57     IOHIDDeviceRef device = IOHIDQueueGetDevice(inSender);
  58
  59     Joystick *js = [controller findJoystickByRef:device];
  60     if (((ApplicationController *)[NSApplication sharedApplication].delegate).active) {
  61         JSAction *mainAction = [js actionForEvent:value];
  62         [mainAction notifyEvent:value];
  63         NSArray *children = mainAction.children ? mainAction.children : mainAction ? @[mainAction] : @[];
  64         for (JSAction *subaction in children) {
  65             Target *target = [controller.currentConfig getTargetForAction:subaction];
  66             target.magnitude = mainAction.magnitude;
  67             target.running = subaction.active;
  68             if (target.running && target.isContinuous)
  69                 [controller addRunningTarget:target];
  70         }
  71     } else if ([NSApplication sharedApplication].isActive
  72                && [NSApplication sharedApplication].mainWindow.isVisible) {
  73         // joysticks not active, use it to select stuff
  74         JSAction *handler = [js handlerForEvent:value];
  75         if (!handler)
  76             return;
  77
  78         [controller expandRecursive:handler];
  79         controller->programmaticallySelecting = YES;
  80         [controller->outlineView selectRowIndexes:[NSIndexSet indexSetWithIndex:[controller->outlineView rowForItem:handler]] byExtendingSelection: NO];
  81     }
  82 }
  83
  84 static int findAvailableIndex(NSArray *list, Joystick *js) {
  85     for (int index = 1; ; index++) {
  86         BOOL available = YES;
  87         for (Joystick *used in list) {
  88             if ([used.productName isEqualToString:js.productName] && used.index == index) {
  89                 available = NO;
  90                 break;
  91             }
  92         }
  93         if (available)
  94             return index;
  95     }
  96 }
  97
  98 static void add_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDDeviceRef device) {
  99     JoystickController *controller = (__bridge JoystickController *)ctx;
 100     IOHIDDeviceRegisterInputValueCallback(device, input_callback, (__bridge void*)controller);
 101     Joystick *js = [[Joystick alloc] initWithDevice:device];
 102     js.index = findAvailableIndex(controller.joysticks, js);
 103     [[controller joysticks] addObject:js];
 104     [controller->outlineView reloadData];
 105 }
 106
 107 - (Joystick *)findJoystickByRef:(IOHIDDeviceRef)device {
 108     for (Joystick *js in joysticks)
 109         if (js.device == device)
 110             return js;
 111     return nil;
 112 }
 113
 114 static void remove_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDDeviceRef device) {
 115     JoystickController *controller = (__bridge JoystickController *)ctx;
 116     Joystick *match = [controller findJoystickByRef:device];
 117     IOHIDDeviceRegisterInputValueCallback(device, NULL, NULL);
 118     if (match) {
 119         [controller.joysticks removeObject:match];
 120         [controller->outlineView reloadData];
 121     }
 122 }
 123
 124 - (void)updateContinuousActions:(NSTimer *)timer {
 125     self.mouseLoc = [NSEvent mouseLocation];
 126     for (Target *target in [self.runningTargets copy]) {
 127         if (![target update:self])
 128             [self.runningTargets removeObject:target];
 129     }
 130     if (!self.runningTargets.count) {
 131         [continuousTimer invalidate];
 132         continuousTimer = nil;
 133         NSLog(@"Unscheduled continuous target timer.");
 134     }
 135 }
 136
 137 #define NSSTR(e) ((NSString *)CFSTR(e))
 138
 139 - (void)setup {
 140     hidManager = IOHIDManagerCreate(kCFAllocatorDefault, kIOHIDOptionsTypeNone);
 141     NSArray *criteria = @[ @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
 142                               NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_Joystick) },
 143                            @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
 144                               NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_GamePad) },
 145                            @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
 146                               NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_MultiAxisController) }
 147                            ];
 148     IOHIDManagerSetDeviceMatchingMultiple(hidManager, (__bridge CFArrayRef)criteria);
 149
 150     IOHIDManagerScheduleWithRunLoop(hidManager, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode);
 151     IOHIDManagerOpen(hidManager, kIOHIDOptionsTypeNone); // FIXME: If an error happens, report it!
 152
 153     IOHIDManagerRegisterDeviceMatchingCallback(hidManager, add_callback, (__bridge void *)self);
 154     IOHIDManagerRegisterDeviceRemovalCallback(hidManager, remove_callback, (__bridge void *)self);
 155 }
 156
 157 - (Config *)currentConfig {
 158     return configsController.currentConfig;
 159 }
 160
 161 - (JSAction *)selectedAction {
 162     id item = [outlineView itemAtRow:outlineView.selectedRow];
 163     return [item children] ? nil : item;
 164 }
 165
 166 - (int)outlineView:(NSOutlineView *)outlineView numberOfChildrenOfItem:(id)item {
 167     return item ? [[item children] count] : [joysticks count];
 168 }
 169
 170 - (BOOL)outlineView:(NSOutlineView *)outlineView isItemExpandable:(id)item {
 171     return item ? [[item children] count] > 0: YES;
 172 }
 173
 174 - (id)outlineView:(NSOutlineView *)outlineView child:(int)index ofItem:(id)item {
 175     return item ? [item children][index] : joysticks[index];
 176 }
 177
 178 - (id)outlineView:(NSOutlineView *)outlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item  {
 179     if(item == nil)
 180         return @"root";
 181     return [item name];
 182 }
 183
 184 - (void)outlineViewSelectionDidChange: (NSNotification*) notification {
 185     [targetController reset];
 186     [targetController load];
 187     if (programmaticallySelecting)
 188         [targetController focusKey];
 189     programmaticallySelecting = NO;
 190 }
 191
 192 @end