1 /* 2 * Parallel port to Walkera WK-0701 TX joystick 3 * 4 * Copyright (c) 2008 Peter Popovec 5 * 6 * More about driver: <file:Documentation/input/walkera0701.txt> 7 */ 8 9 /* 10 * This program is free software; you can redistribute it and/or modify it 11 * under the terms of the GNU General Public License version 2 as published by 12 * the Free Software Foundation. 13 */ 14 15 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 16 17 #define RESERVE 20000 18 #define SYNC_PULSE 1306000 19 #define BIN0_PULSE 288000 20 #define BIN1_PULSE 438000 21 22 #define ANALOG_MIN_PULSE 318000 23 #define ANALOG_MAX_PULSE 878000 24 #define ANALOG_DELTA 80000 25 26 #define BIN_SAMPLE ((BIN0_PULSE + BIN1_PULSE) / 2) 27 28 #define NO_SYNC 25 29 30 #include <linux/kernel.h> 31 #include <linux/module.h> 32 #include <linux/parport.h> 33 #include <linux/input.h> 34 #include <linux/hrtimer.h> 35 36 MODULE_AUTHOR("Peter Popovec <popovec@fei.tuke.sk>"); 37 MODULE_DESCRIPTION("Walkera WK-0701 TX as joystick"); 38 MODULE_LICENSE("GPL"); 39 40 static unsigned int walkera0701_pp_no; 41 module_param_named(port, walkera0701_pp_no, int, 0); 42 MODULE_PARM_DESC(port, 43 "Parallel port adapter for Walkera WK-0701 TX (default is 0)"); 44 45 /* 46 * For now, only one device is supported, if somebody need more devices, code 47 * can be expanded, one struct walkera_dev per device must be allocated and 48 * set up by walkera0701_connect (release of device by walkera0701_disconnect) 49 */ 50 51 struct walkera_dev { 52 unsigned char buf[25]; 53 u64 irq_time, irq_lasttime; 54 int counter; 55 int ack; 56 57 struct input_dev *input_dev; 58 struct hrtimer timer; 59 60 struct parport *parport; 61 struct pardevice *pardevice; 62 }; 63 64 static struct walkera_dev w_dev; 65 66 static inline void walkera0701_parse_frame(struct walkera_dev *w) 67 { 68 int i; 69 int val1, val2, val3, val4, val5, val6, val7, val8; 70 int magic, magic_bit; 71 int crc1, crc2; 72 73 for (crc1 = crc2 = i = 0; i < 10; i++) { 74 crc1 += w->buf[i] & 7; 75 crc2 += (w->buf[i] & 8) >> 3; 76 } 77 if ((w->buf[10] & 7) != (crc1 & 7)) 78 return; 79 if (((w->buf[10] & 8) >> 3) != (((crc1 >> 3) + crc2) & 1)) 80 return; 81 for (crc1 = crc2 = 0, i = 11; i < 23; i++) { 82 crc1 += w->buf[i] & 7; 83 crc2 += (w->buf[i] & 8) >> 3; 84 } 85 if ((w->buf[23] & 7) != (crc1 & 7)) 86 return; 87 if (((w->buf[23] & 8) >> 3) != (((crc1 >> 3) + crc2) & 1)) 88 return; 89 val1 = ((w->buf[0] & 7) * 256 + w->buf[1] * 16 + w->buf[2]) >> 2; 90 val1 *= ((w->buf[0] >> 2) & 2) - 1; /* sign */ 91 val2 = (w->buf[2] & 1) << 8 | (w->buf[3] << 4) | w->buf[4]; 92 val2 *= (w->buf[2] & 2) - 1; /* sign */ 93 val3 = ((w->buf[5] & 7) * 256 + w->buf[6] * 16 + w->buf[7]) >> 2; 94 val3 *= ((w->buf[5] >> 2) & 2) - 1; /* sign */ 95 val4 = (w->buf[7] & 1) << 8 | (w->buf[8] << 4) | w->buf[9]; 96 val4 *= (w->buf[7] & 2) - 1; /* sign */ 97 val5 = ((w->buf[11] & 7) * 256 + w->buf[12] * 16 + w->buf[13]) >> 2; 98 val5 *= ((w->buf[11] >> 2) & 2) - 1; /* sign */ 99 val6 = (w->buf[13] & 1) << 8 | (w->buf[14] << 4) | w->buf[15]; 100 val6 *= (w->buf[13] & 2) - 1; /* sign */ 101 val7 = ((w->buf[16] & 7) * 256 + w->buf[17] * 16 + w->buf[18]) >> 2; 102 val7 *= ((w->buf[16] >> 2) & 2) - 1; /*sign */ 103 val8 = (w->buf[18] & 1) << 8 | (w->buf[19] << 4) | w->buf[20]; 104 val8 *= (w->buf[18] & 2) - 1; /*sign */ 105 106 magic = (w->buf[21] << 4) | w->buf[22]; 107 magic_bit = (w->buf[24] & 8) >> 3; 108 pr_debug("%4d %4d %4d %4d %4d %4d %4d %4d (magic %2x %d)\n", 109 val1, val2, val3, val4, val5, val6, val7, val8, 110 magic, magic_bit); 111 112 input_report_abs(w->input_dev, ABS_X, val2); 113 input_report_abs(w->input_dev, ABS_Y, val1); 114 input_report_abs(w->input_dev, ABS_Z, val6); 115 input_report_abs(w->input_dev, ABS_THROTTLE, val3); 116 input_report_abs(w->input_dev, ABS_RUDDER, val4); 117 input_report_abs(w->input_dev, ABS_MISC, val7); 118 input_report_key(w->input_dev, BTN_GEAR_DOWN, val5 > 0); 119 } 120 121 static inline int read_ack(struct pardevice *p) 122 { 123 return parport_read_status(p->port) & 0x40; 124 } 125 126 /* falling edge, prepare to BIN value calculation */ 127 static void walkera0701_irq_handler(void *handler_data) 128 { 129 u64 pulse_time; 130 struct walkera_dev *w = handler_data; 131 132 w->irq_time = ktime_to_ns(ktime_get()); 133 pulse_time = w->irq_time - w->irq_lasttime; 134 w->irq_lasttime = w->irq_time; 135 136 /* cancel timer, if in handler or active do resync */ 137 if (unlikely(0 != hrtimer_try_to_cancel(&w->timer))) { 138 w->counter = NO_SYNC; 139 return; 140 } 141 142 if (w->counter < NO_SYNC) { 143 if (w->ack) { 144 pulse_time -= BIN1_PULSE; 145 w->buf[w->counter] = 8; 146 } else { 147 pulse_time -= BIN0_PULSE; 148 w->buf[w->counter] = 0; 149 } 150 if (w->counter == 24) { /* full frame */ 151 walkera0701_parse_frame(w); 152 w->counter = NO_SYNC; 153 if (abs64(pulse_time - SYNC_PULSE) < RESERVE) /* new frame sync */ 154 w->counter = 0; 155 } else { 156 if ((pulse_time > (ANALOG_MIN_PULSE - RESERVE) 157 && (pulse_time < (ANALOG_MAX_PULSE + RESERVE)))) { 158 pulse_time -= (ANALOG_MIN_PULSE - RESERVE); 159 pulse_time = (u32) pulse_time / ANALOG_DELTA; /* overtiping is safe, pulsetime < s32.. */ 160 w->buf[w->counter++] |= (pulse_time & 7); 161 } else 162 w->counter = NO_SYNC; 163 } 164 } else if (abs64(pulse_time - SYNC_PULSE - BIN0_PULSE) < 165 RESERVE + BIN1_PULSE - BIN0_PULSE) /* frame sync .. */ 166 w->counter = 0; 167 168 hrtimer_start(&w->timer, ktime_set(0, BIN_SAMPLE), HRTIMER_MODE_REL); 169 } 170 171 static enum hrtimer_restart timer_handler(struct hrtimer 172 *handle) 173 { 174 struct walkera_dev *w; 175 176 w = container_of(handle, struct walkera_dev, timer); 177 w->ack = read_ack(w->pardevice); 178 179 return HRTIMER_NORESTART; 180 } 181 182 static int walkera0701_open(struct input_dev *dev) 183 { 184 struct walkera_dev *w = input_get_drvdata(dev); 185 186 if (parport_claim(w->pardevice)) 187 return -EBUSY; 188 189 parport_enable_irq(w->parport); 190 return 0; 191 } 192 193 static void walkera0701_close(struct input_dev *dev) 194 { 195 struct walkera_dev *w = input_get_drvdata(dev); 196 197 parport_disable_irq(w->parport); 198 hrtimer_cancel(&w->timer); 199 200 parport_release(w->pardevice); 201 } 202 203 static int walkera0701_connect(struct walkera_dev *w, int parport) 204 { 205 int error; 206 207 w->parport = parport_find_number(parport); 208 if (!w->parport) { 209 pr_err("parport %d does not exist\n", parport); 210 return -ENODEV; 211 } 212 213 if (w->parport->irq == -1) { 214 pr_err("parport %d does not have interrupt assigned\n", 215 parport); 216 error = -EINVAL; 217 goto err_put_parport; 218 } 219 220 w->pardevice = parport_register_device(w->parport, "walkera0701", 221 NULL, NULL, walkera0701_irq_handler, 222 PARPORT_DEV_EXCL, w); 223 if (!w->pardevice) { 224 pr_err("failed to register parport device\n"); 225 error = -EIO; 226 goto err_put_parport; 227 } 228 229 if (parport_negotiate(w->pardevice->port, IEEE1284_MODE_COMPAT)) { 230 pr_err("failed to negotiate parport mode\n"); 231 error = -EIO; 232 goto err_unregister_device; 233 } 234 235 hrtimer_init(&w->timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL); 236 w->timer.function = timer_handler; 237 238 w->input_dev = input_allocate_device(); 239 if (!w->input_dev) { 240 pr_err("failed to allocate input device\n"); 241 error = -ENOMEM; 242 goto err_unregister_device; 243 } 244 245 input_set_drvdata(w->input_dev, w); 246 w->input_dev->name = "Walkera WK-0701 TX"; 247 w->input_dev->phys = w->parport->name; 248 w->input_dev->id.bustype = BUS_PARPORT; 249 250 /* TODO what id vendor/product/version ? */ 251 w->input_dev->id.vendor = 0x0001; 252 w->input_dev->id.product = 0x0001; 253 w->input_dev->id.version = 0x0100; 254 w->input_dev->dev.parent = w->parport->dev; 255 w->input_dev->open = walkera0701_open; 256 w->input_dev->close = walkera0701_close; 257 258 w->input_dev->evbit[0] = BIT(EV_ABS) | BIT_MASK(EV_KEY); 259 w->input_dev->keybit[BIT_WORD(BTN_GEAR_DOWN)] = BIT_MASK(BTN_GEAR_DOWN); 260 261 input_set_abs_params(w->input_dev, ABS_X, -512, 512, 0, 0); 262 input_set_abs_params(w->input_dev, ABS_Y, -512, 512, 0, 0); 263 input_set_abs_params(w->input_dev, ABS_Z, -512, 512, 0, 0); 264 input_set_abs_params(w->input_dev, ABS_THROTTLE, -512, 512, 0, 0); 265 input_set_abs_params(w->input_dev, ABS_RUDDER, -512, 512, 0, 0); 266 input_set_abs_params(w->input_dev, ABS_MISC, -512, 512, 0, 0); 267 268 error = input_register_device(w->input_dev); 269 if (error) { 270 pr_err("failed to register input device\n"); 271 goto err_free_input_dev; 272 } 273 274 return 0; 275 276 err_free_input_dev: 277 input_free_device(w->input_dev); 278 err_unregister_device: 279 parport_unregister_device(w->pardevice); 280 err_put_parport: 281 parport_put_port(w->parport); 282 return error; 283 } 284 285 static void walkera0701_disconnect(struct walkera_dev *w) 286 { 287 input_unregister_device(w->input_dev); 288 parport_unregister_device(w->pardevice); 289 parport_put_port(w->parport); 290 } 291 292 static int __init walkera0701_init(void) 293 { 294 return walkera0701_connect(&w_dev, walkera0701_pp_no); 295 } 296 297 static void __exit walkera0701_exit(void) 298 { 299 walkera0701_disconnect(&w_dev); 300 } 301 302 module_init(walkera0701_init); 303 module_exit(walkera0701_exit); 304