2 // NJDeviceController.m
5 // Created by Sam McCall on 4/05/09.
8 #import "NJDeviceController.h"
11 #import "NJMappingsController.h"
15 #import "NJOutputController.h"
18 @implementation NJDeviceController {
19 IOHIDManagerRef hidManager;
20 NSTimer *continuousTimer;
21 NSMutableArray *runningOutputs;
22 NSMutableArray *_devices;
26 if ((self = [super init])) {
27 _devices = [[NSMutableArray alloc] initWithCapacity:16];
28 runningOutputs = [[NSMutableArray alloc] initWithCapacity:32];
34 [continuousTimer invalidate];
35 IOHIDManagerClose(hidManager, kIOHIDOptionsTypeNone);
36 CFRelease(hidManager);
39 - (void)expandRecursive:(id <NJInputPathElement>)pathElement {
41 [self expandRecursive:pathElement.base];
42 [outlineView expandItem:pathElement];
46 - (void)addRunningOutput:(NJOutput *)output {
47 if (![runningOutputs containsObject:output]) {
48 [runningOutputs addObject:output];
50 if (!continuousTimer) {
51 continuousTimer = [NSTimer scheduledTimerWithTimeInterval:1.f/60.f
53 selector:@selector(updateContinuousInputs:)
56 NSLog(@"Scheduled continuous output timer.");
60 - (void)runOutputForDevice:(IOHIDDeviceRef)device value:(IOHIDValueRef)value {
61 NJDevice *dev = [self findDeviceByRef:device];
62 NJInput *mainInput = [dev inputForEvent:value];
63 [mainInput notifyEvent:value];
64 NSArray *children = mainInput.children ? mainInput.children : mainInput ? @[mainInput] : @[];
65 for (NJInput *subInput in children) {
66 NJOutput *output = mappingsController.currentMapping[subInput];
67 output.magnitude = mainInput.magnitude;
68 output.running = subInput.active;
69 if (output.running && output.isContinuous)
70 [self addRunningOutput:output];
74 - (void)showOutputForDevice:(IOHIDDeviceRef)device value:(IOHIDValueRef)value {
75 NJDevice *dev = [self findDeviceByRef:device];
76 NJInput *handler = [dev handlerForEvent:value];
80 [self expandRecursive:handler];
81 [outlineView selectRowIndexes:[NSIndexSet indexSetWithIndex:[outlineView rowForItem:handler]] byExtendingSelection: NO];
82 [outputController focusKey];
85 static void input_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDValueRef value) {
86 NJDeviceController *controller = (__bridge NJDeviceController *)ctx;
87 IOHIDDeviceRef device = IOHIDQueueGetDevice(inSender);
89 if (controller.translatingEvents) {
90 [controller runOutputForDevice:device value:value];
91 } else if ([NSApplication sharedApplication].mainWindow.isVisible) {
92 [controller showOutputForDevice:device value:value];
96 static int findAvailableIndex(NSArray *list, NJDevice *dev) {
97 for (int index = 1; ; index++) {
99 for (NJDevice *used in list) {
100 if ([used.productName isEqualToString:dev.productName] && used.index == index) {
110 - (void)addDeviceForDevice:(IOHIDDeviceRef)device {
111 IOHIDDeviceRegisterInputValueCallback(device, input_callback, (__bridge void*)self);
112 NJDevice *dev = [[NJDevice alloc] initWithDevice:device];
113 dev.index = findAvailableIndex(_devices, dev);
114 [_devices addObject:dev];
115 [outlineView reloadData];
118 static void add_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDDeviceRef device) {
119 NJDeviceController *controller = (__bridge NJDeviceController *)ctx;
120 [controller addDeviceForDevice:device];
123 - (NJDevice *)findDeviceByRef:(IOHIDDeviceRef)device {
124 for (NJDevice *dev in _devices)
125 if (dev.device == device)
130 static void remove_callback(void *ctx, IOReturn inResult, void *inSender, IOHIDDeviceRef device) {
131 NJDeviceController *controller = (__bridge NJDeviceController *)ctx;
132 [controller removeDeviceForDevice:device];
135 - (void)removeDeviceForDevice:(IOHIDDeviceRef)device {
136 NJDevice *match = [self findDeviceByRef:device];
137 IOHIDDeviceRegisterInputValueCallback(device, NULL, NULL);
139 [_devices removeObject:match];
140 [outlineView reloadData];
145 - (void)updateContinuousInputs:(NSTimer *)timer {
146 self.mouseLoc = [NSEvent mouseLocation];
147 for (NJOutput *output in [runningOutputs copy]) {
148 if (![output update:self]) {
149 [runningOutputs removeObject:output];
152 if (!runningOutputs.count) {
153 [continuousTimer invalidate];
154 continuousTimer = nil;
155 NSLog(@"Unscheduled continuous output timer.");
159 #define NSSTR(e) ((NSString *)CFSTR(e))
162 hidManager = IOHIDManagerCreate(kCFAllocatorDefault, kIOHIDOptionsTypeNone);
163 NSArray *criteria = @[ @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
164 NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_Joystick) },
165 @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
166 NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_GamePad) },
167 @{ NSSTR(kIOHIDDeviceUsagePageKey) : @(kHIDPage_GenericDesktop),
168 NSSTR(kIOHIDDeviceUsageKey) : @(kHIDUsage_GD_MultiAxisController) }
170 IOHIDManagerSetDeviceMatchingMultiple(hidManager, (__bridge CFArrayRef)criteria);
172 IOHIDManagerScheduleWithRunLoop(hidManager, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode);
173 IOReturn ret = IOHIDManagerOpen(hidManager, kIOHIDOptionsTypeNone);
174 if (ret != kIOReturnSuccess) {
175 [[NSAlert alertWithMessageText:@"Input devices are unavailable"
179 informativeTextWithFormat:@"Error 0x%08x occured trying to access your devices. "
180 @"Input may not be correctly detected or mapped.",
182 beginSheetModalForWindow:outlineView.window
188 IOHIDManagerRegisterDeviceMatchingCallback(hidManager, add_callback, (__bridge void *)self);
189 IOHIDManagerRegisterDeviceRemovalCallback(hidManager, remove_callback, (__bridge void *)self);
192 - (NJInput *)selectedInput {
193 id <NJInputPathElement> item = [outlineView itemAtRow:outlineView.selectedRow];
194 return (!item.children && item.base) ? item : nil;
197 - (NSInteger)outlineView:(NSOutlineView *)outlineView
198 numberOfChildrenOfItem:(id <NJInputPathElement>)item {
199 return item ? item.children.count : _devices.count;
202 - (BOOL)outlineView:(NSOutlineView *)outlineView
203 isItemExpandable:(id <NJInputPathElement>)item {
204 return item ? [[item children] count] > 0: YES;
207 - (id)outlineView:(NSOutlineView *)outlineView
208 child:(NSInteger)index
209 ofItem:(id <NJInputPathElement>)item {
210 return item ? item.children[index] : _devices[index];
213 - (id)outlineView:(NSOutlineView *)outlineView
214 objectValueForTableColumn:(NSTableColumn *)tableColumn
215 byItem:(id <NJInputPathElement>)item {
216 return item ? item.name : @"root";
219 - (void)outlineViewSelectionDidChange:(NSNotification *)notification {
221 [outputController loadCurrent];
224 - (void)setTranslatingEvents:(BOOL)translatingEvents {
225 if (translatingEvents != _translatingEvents) {
226 _translatingEvents = translatingEvents;
227 translatingEventsSetting.selectedSegment = !!translatingEvents;
228 NSString *name = translatingEvents
229 ? NJEventTranslationActivated
230 : NJEventTranslationDeactivated;
231 [NSNotificationCenter.defaultCenter postNotificationName:name
236 - (IBAction)translatingEventsChanged:(id)sender {
237 self.translatingEvents = translatingEventsSetting.selectedSegment;