addons/AppleExtras/source/unimotion.c

   1  /*
   2  *  UniMotion - Unified Motion detection for Apple portables.
   3  *
   4  *  Copyright (c) 2006 Lincoln Ramsay. All rights reserved.
   5  *
   6  *  This library is free software; you can redistribute it and/or
   7  *  modify it under the terms of the GNU Lesser General Public
   8  *  License version 2.1 as published by the Free Software Foundation.
   9  *
  10  *  This library is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  *  Lesser General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU Lesser General Public
  16  *  License along with this library; if not, write to the Free Software
  17  *  Foundation Inc. 59 Temple Place, Suite 330, Boston MA 02111-1307 USA
  18  */
  19
  20 /*
  21  * HISTORY of Motion
  22  * Written by Christian Klein
  23  * Modified for iBook compatibility by Pall Thayer
  24  * Modified for Hi Res Powerbook compatibility by Pall Thayer
  25  * Modified for MacBook Pro compatibility by Randy Green
  26  * Disparate forks unified into UniMotion by Lincoln Ramsay
  27  */
  28
  29 // This license applies to the portions created by Cristian Klein.
  30 /* motion.c
  31  *
  32  * a little program to display the coords returned by
  33  * the powerbook motion sensor
  34  *
  35  * A fine piece of c0de, brought to you by
  36  *
  37  *               ---===---
  38  * *** teenage mutant ninja hero coders ***
  39  *               ---===---
  40  *
  41  * All of the software included is copyrighted by Christian Klein <chris@5711.org>.
  42  *
  43  * Copyright 2005 Christian Klein. All rights reserved.
  44  *
  45  * Redistribution and use in source and binary forms, with or without
  46  * modification, are permitted provided that the following conditions
  47  * are met:
  48  * 1. Redistributions of source code must retain the above copyright
  49  *    notice, this list of conditions and the following disclaimer.
  50  * 2. Redistributions in binary form must reproduce the above copyright
  51  *    notice, this list of conditions and the following disclaimer in the
  52  *    documentation and/or other materials provided with the distribution.
  53  * 3. The name of the author must not be used to endorse or promote
  54  *    products derived from this software
  55  *    without specific prior written permission.
  56  *
  57  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  58  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  59  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  60  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  61  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  62  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  63  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  64  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  65  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  66  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  67  * SUCH DAMAGE.
  68  *
  69  */
  70
  71 #include "unimotion.h"
  72 #include <IOKit/IOKitLib.h>
  73 #include <CoreFoundation/CoreFoundation.h>
  74 #include <string.h>
  75 #include <stdint.h>
  76
  77 enum data_type {
  78     PB_IB,
  79     MBP
  80 };
  81
  82 struct pb_ib_data {
  83     int8_t x;
  84     int8_t y;
  85     int8_t z;
  86     int8_t pad[57];
  87 };
  88
  89 struct mbp_data {
  90     int16_t x;
  91     int16_t y;
  92     int16_t z;
  93     int8_t pad[34];
  94 };
  95
  96 union motion_data {
  97     struct pb_ib_data pb_ib;
  98     struct mbp_data mbp;
  99 };
 100
 101
 102 static int set_values(int type, int *kernFunc, char **servMatch, int *dataType)
 103 {
 104     switch ( type ) {
 105         case powerbook:
 106             *kernFunc = 21;
 107             *servMatch = "IOI2CMotionSensor";
 108             *dataType = PB_IB;
 109             break;
 110         case ibook:
 111             *kernFunc = 21;
 112             *servMatch = "IOI2CMotionSensor";
 113             *dataType = PB_IB;
 114             break;
 115         case highrespb:
 116             *kernFunc = 21;
 117             *servMatch = "PMUMotionSensor";
 118             *dataType = PB_IB;
 119             break;
 120         case macbookpro:
 121             *kernFunc = 5;
 122             *servMatch = "SMCMotionSensor";
 123             *dataType = MBP;
 124             break;
 125         default:
 126             return 0;
 127     }
 128
 129     return 1;
 130 }
 131
 132 static int probe_sms(int kernFunc, char *servMatch, int dataType, void *data)
 133 {
 134     kern_return_t result;
 135     mach_port_t masterPort;
 136     io_iterator_t iterator;
 137     io_object_t aDevice;
 138     io_connect_t  dataPort;
 139
 140     IOItemCount structureInputSize;
 141     IOByteCount structureOutputSize;
 142
 143     union motion_data inputStructure;
 144     union motion_data *outputStructure;
 145
 146     outputStructure = (union motion_data *)data;
 147
 148     result = IOMasterPort(MACH_PORT_NULL, &masterPort);
 149
 150     CFMutableDictionaryRef matchingDictionary = IOServiceMatching(servMatch);
 151
 152     result = IOServiceGetMatchingServices(masterPort, matchingDictionary, &iterator);
 153
 154     if (result != KERN_SUCCESS) {
 155         //fputs("IOServiceGetMatchingServices returned error.\n", stderr);
 156         return 0;
 157     }
 158
 159     aDevice = IOIteratorNext(iterator);
 160     IOObjectRelease(iterator);
 161
 162     if (aDevice == 0) {
 163         //fputs("No motion sensor available\n", stderr);
 164         return 0;
 165     }
 166
 167     result = IOServiceOpen(aDevice, mach_task_self(), 0, &dataPort);
 168     IOObjectRelease(aDevice);
 169
 170     if (result != KERN_SUCCESS) {
 171         //fputs("Could not open motion sensor device\n", stderr);
 172         return 0;
 173     }
 174
 175     switch ( dataType ) {
 176         case PB_IB:
 177             structureInputSize = sizeof(struct pb_ib_data);
 178             structureOutputSize = sizeof(struct pb_ib_data);
 179             break;
 180         case MBP:
 181             structureInputSize = sizeof(struct mbp_data);
 182             structureOutputSize = sizeof(struct mbp_data);
 183             break;
 184         default:
 185             return 0;
 186     }
 187
 188     memset(&inputStructure, 0, sizeof(union motion_data));
 189     memset(outputStructure, 0, sizeof(union motion_data));
 190
 191     result = IOConnectMethodStructureIStructureO(dataPort, kernFunc, structureInputSize,
 192                 &structureOutputSize, &inputStructure, outputStructure);
 193
 194     IOServiceClose(dataPort);
 195
 196     if (result != KERN_SUCCESS) {
 197         //puts("no coords");
 198         return 0;
 199     }
 200     return 1;
 201 }
 202
 203 int detect_sms()
 204 {
 205     int kernFunc;
 206     char *servMatch;
 207     int dataType;
 208     union motion_data data;
 209     int i;
 210
 211     for ( i = 1; ; i++ ) {
 212         if ( !set_values(i, &kernFunc, &servMatch, &dataType) )
 213             break;
 214         if ( probe_sms(kernFunc, servMatch, dataType, &data) )
 215             return i;
 216     }
 217
 218     return unknown;
 219 }
 220
 221 int read_sms_raw(int type, int *x, int *y, int *z)
 222 {
 223     int kernFunc;
 224     char *servMatch;
 225     int dataType;
 226     union motion_data data;
 227
 228     if ( !set_values(type, &kernFunc, &servMatch, &dataType) )
 229         return 0;
 230     if ( probe_sms(kernFunc, servMatch, dataType, &data) ) {
 231         switch ( dataType ) {
 232             case PB_IB:
 233                 if ( x ) *x = data.pb_ib.x;
 234                 if ( y ) *y = data.pb_ib.y;
 235                 if ( z ) *z = data.pb_ib.z;
 236                 break;
 237             case MBP:
 238                 if ( x ) *x = data.mbp.x;
 239                 if ( y ) *y = data.mbp.y;
 240                 if ( z ) *z = data.mbp.z;
 241                 break;
 242             default:
 243                 return 0;
 244         }
 245         return 1;
 246     }
 247     return 0;
 248 }
 249
 250 int read_sms(int type, int *x, int *y, int *z)
 251 {
 252     int _x, _y, _z;
 253         int xoff, yoff, zoff;
 254         Boolean ok;
 255     int ret;
 256
 257     ret = read_sms_raw(type, &_x, &_y, &_z);
 258     if ( !ret )
 259         return 0;
 260
 261         CFStringRef app = CFSTR("com.ramsayl.UniMotion");
 262         CFStringRef xoffstr = CFSTR("x_offset");
 263         CFStringRef yoffstr = CFSTR("y_offset");
 264         CFStringRef zoffstr = CFSTR("z_offset");
 265
 266         xoff = CFPreferencesGetAppIntegerValue(xoffstr, app, &ok);
 267         if ( ok ) _x += xoff;
 268         yoff = CFPreferencesGetAppIntegerValue(yoffstr, app, &ok);
 269         if ( ok ) _y += yoff;
 270         zoff = CFPreferencesGetAppIntegerValue(zoffstr, app, &ok);
 271         if ( ok ) _z += zoff;
 272
 273         *x = _x;
 274         *y = _y;
 275         *z = _z;
 276
 277     return ret;
 278 }
 279
 280 int read_sms_real(int type, double *x, double *y, double *z)
 281 {
 282     int _x, _y, _z;
 283         int xscale, yscale, zscale;
 284     int ret;
 285         Boolean ok;
 286
 287     ret = read_sms_raw(type, &_x, &_y, &_z);
 288     if ( !ret )
 289         return 0;
 290
 291         CFStringRef app = CFSTR("com.ramsayl.UniMotion");
 292         CFStringRef xscalestr = CFSTR("x_scale");
 293         CFStringRef yscalestr = CFSTR("y_scale");
 294         CFStringRef zscalestr = CFSTR("z_scale");
 295         xscale = CFPreferencesGetAppIntegerValue(xscalestr, app, &ok);
 296         if ( !ok ) return 0;
 297         yscale = CFPreferencesGetAppIntegerValue(yscalestr, app, &ok);
 298         if ( !ok ) return 0;
 299         zscale = CFPreferencesGetAppIntegerValue(zscalestr, app, &ok);
 300         if ( !ok ) return 0;
 301
 302     *x = _x / (double)xscale;
 303     *y = _y / (double)yscale;
 304     *z = _z / (double)zscale;
 305
 306     return 1;
 307 }
 308