NJDevice.m

   1 //
   2 //  NJDevice.m
   3 //  Enjoy
   4 //
   5 //  Created by Sam McCall on 4/05/09.
   6 //
   7
   8 #import "NJDevice.h"
   9
  10 #import "NJInput.h"
  11 #import "NJInputAnalog.h"
  12 #import "NJInputHat.h"
  13 #import "NJInputButton.h"
  14
  15 static NSArray *InputsForElement(IOHIDDeviceRef device, id base) {
  16     CFArrayRef elements = IOHIDDeviceCopyMatchingElements(device, NULL, kIOHIDOptionsTypeNone);
  17     NSMutableArray *children = [NSMutableArray arrayWithCapacity:CFArrayGetCount(elements)];
  18
  19     int buttons = 0;
  20     int axes = 0;
  21     int hats = 0;
  22
  23     for (int i = 0; i < CFArrayGetCount(elements); i++) {
  24         IOHIDElementRef element = (IOHIDElementRef)CFArrayGetValueAtIndex(elements, i);
  25         int type = IOHIDElementGetType(element);
  26         unsigned usage = IOHIDElementGetUsage(element);
  27         unsigned usagePage = IOHIDElementGetUsagePage(element);
  28         long max = IOHIDElementGetPhysicalMax(element);
  29         long min = IOHIDElementGetPhysicalMin(element);
  30         CFStringRef elName = IOHIDElementGetName(element);
  31
  32         NJInput *input = nil;
  33
  34         if (!(type == kIOHIDElementTypeInput_Misc
  35               || type == kIOHIDElementTypeInput_Axis
  36               || type == kIOHIDElementTypeInput_Button))
  37              continue;
  38
  39         if (max - min == 1 || usagePage == kHIDPage_Button || type == kIOHIDElementTypeInput_Button) {
  40             input = [[NJInputButton alloc] initWithName:(__bridge NSString *)elName
  41                                                     idx:++buttons
  42                                                     max:max];
  43         } else if (usage == kHIDUsage_GD_Hatswitch) {
  44             input = [[NJInputHat alloc] initWithIndex:++hats];
  45         } else if (usage >= kHIDUsage_GD_X && usage <= kHIDUsage_GD_Rz) {
  46             input = [[NJInputAnalog alloc] initWithIndex:++axes
  47                                                   rawMin:min
  48                                                   rawMax:max];
  49         } else {
  50             continue;
  51         }
  52
  53         // TODO(jfw): Should be moved into better constructors.
  54         input.base = base;
  55         input.cookie = IOHIDElementGetCookie(element);
  56         [children addObject:input];
  57     }
  58
  59     CFRelease(elements);
  60     return children;
  61 }
  62
  63 @implementation NJDevice {
  64     int vendorId;
  65     int productId;
  66 }
  67
  68 - (id)initWithDevice:(IOHIDDeviceRef)dev {
  69     if ((self = [super init])) {
  70         self.device = dev;
  71         self.productName = (__bridge NSString *)IOHIDDeviceGetProperty(dev, CFSTR(kIOHIDProductKey));
  72         vendorId = [(__bridge NSNumber *)IOHIDDeviceGetProperty(dev, CFSTR(kIOHIDVendorIDKey)) intValue];
  73         productId = [(__bridge NSNumber *)IOHIDDeviceGetProperty(dev, CFSTR(kIOHIDProductIDKey)) intValue];
  74         self.children = InputsForElement(dev, self);
  75     }
  76     return self;
  77 }
  78
  79 - (NSString *)name {
  80     return [NSString stringWithFormat:@"%@ #%d", _productName, _index];
  81 }
  82
  83 - (id)base {
  84     return nil;
  85 }
  86
  87 - (NSString *)uid {
  88     return [NSString stringWithFormat: @"%d:%d:%d", vendorId, productId, _index];
  89 }
  90
  91 - (NJInput *)findInputByCookie:(IOHIDElementCookie)cookie {
  92     for (NJInput *child in _children)
  93         if (child.cookie == cookie)
  94             return child;
  95     return nil;
  96 }
  97
  98 - (NJInput *)handlerForEvent:(IOHIDValueRef)value {
  99     NJInput *mainInput = [self inputForEvent:value];
 100     return [mainInput findSubInputForValue:value];
 101 }
 102
 103 - (NJInput *)inputForEvent:(IOHIDValueRef)value {
 104     IOHIDElementRef elt = IOHIDValueGetElement(value);
 105     IOHIDElementCookie cookie = IOHIDElementGetCookie(elt);
 106     return [self findInputByCookie:cookie];
 107 }
 108
 109 @end