Pref360Control/DeviceItem.m - master - dexMargera/360Controller

Pref360Control/DeviceItem.m
147 строк5 KB
Drew Mills
Collection of fixes for long standing bugs (#688)
01 июн 2018, 07:22
01 июн 2018, 07:22848187f
/*
    MICE Xbox 360 Controller driver for Mac OS X
    Copyright (C) 2006-2013 Colin Munro

    DeviceItem.m - implementation of device wrapper class

    This file is part of Xbox360Controller.

    Xbox360Controller is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    Xbox360Controller is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Foobar; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/
#import "DeviceItem.h"

static NSString* GetDeviceName(io_service_t device)
{
    CFMutableDictionaryRef serviceProperties;
    NSDictionary *properties;
    NSString *deviceName = nil;

    if (IORegistryEntryCreateCFProperties(device, &serviceProperties, kCFAllocatorDefault, kNilOptions) != KERN_SUCCESS)
        return nil;
    properties = CFBridgingRelease(serviceProperties);
    deviceName = properties[@kIOHIDProductKey];
    if (deviceName == nil)
        deviceName = properties[@"USB Product Name"];
    return deviceName;
}

static ControllerType GetControllerType(io_service_t device)
{
    io_service_t parent;
    CFMutableDictionaryRef serviceProperties;

    // Check for the DeviceData dictionary in device
    if (IORegistryEntryCreateCFProperties(device, &serviceProperties, kCFAllocatorDefault, kNilOptions) == KERN_SUCCESS)
    {
        NSDictionary *properties = CFBridgingRelease(serviceProperties);
        NSDictionary *deviceData = properties[@"DeviceData"];

        if (deviceData != nil)
        {
            NSNumber *controllerType = deviceData[@"ControllerType"];
            if (controllerType != nil)
                return (ControllerType)[controllerType intValue];
        }
    }

    // Check for the DeviceData dictionary in the device's parent
    if (IORegistryEntryGetParentEntry(device, kIOServicePlane, &parent) == KERN_SUCCESS)
    {
        if(IORegistryEntryCreateCFProperties(parent, &serviceProperties, kCFAllocatorDefault, kNilOptions) == KERN_SUCCESS)
        {
            NSDictionary *properties = CFBridgingRelease(serviceProperties);
            NSDictionary *deviceData = properties[@"DeviceData"];

            if (deviceData != nil)
            {
                NSNumber *controllerType = deviceData[@"ControllerType"];
                if (controllerType != nil)
                    return (ControllerType)[controllerType intValue];
            }
        }
    }

    NSLog(@"Error: couldn't find ControllerType");
    return Xbox360Controller;
}

@interface DeviceItem ()
@property (strong, readwrite) NSString *deviceName;
@property (readwrite) ControllerType controllerType;
@property (readwrite) io_service_t rawDevice;
@property (readwrite) FFDeviceObjectReference ffDevice;
@property (readwrite) IOHIDDeviceInterface122 **hidDevice;
@end

@implementation DeviceItem
@synthesize controllerType = controllerType;
@synthesize rawDevice = deviceHandle;
@synthesize ffDevice = forceFeedback;
@synthesize hidDevice = interface;

- (instancetype)initWithItemForDevice:(io_service_t)device
{
    if (self = [super init]) {
        IOReturn ret;
        IOCFPlugInInterface **plugInInterface;
        SInt32 score=0;

        ret = IOCreatePlugInInterfaceForService(device, kIOHIDDeviceUserClientTypeID, kIOCFPlugInInterfaceID, &plugInInterface, &score);
        if (ret != kIOReturnSuccess) {
            return nil;
        }
        ret = (*plugInInterface)->QueryInterface(plugInInterface, CFUUIDGetUUIDBytes(kIOHIDDeviceInterfaceID122), (LPVOID)&interface);
        (*plugInInterface)->Release(plugInInterface);
        if (ret != kIOReturnSuccess) {
            return nil;
        }
        forceFeedback = 0;
        FFCreateDevice(device, &forceFeedback);
        self.rawDevice = device;
        self.deviceName = GetDeviceName(device);
        self.controllerType = GetControllerType(device);
    }
    return self;
}

+ (instancetype)allocateDeviceItemForDevice:(io_service_t)device
{
    DeviceItem *item = [[[self class] alloc] initWithItemForDevice:device];

    if (item)
        return item;

    IOObjectRelease(device);
    return nil;
}

- (void)dealloc
{
    if (deviceHandle)
        IOObjectRelease(deviceHandle);
    if (interface)
        (*interface)->Release(interface);
    if (forceFeedback)
        FFReleaseDevice(forceFeedback);
}

- (NSString *)displayName {
    if (self.deviceName == nil)
        return @"Generic Controller";
    else
        return self.deviceName;
}

@end