blob: 3fd06c65e4e3d3128794150d4b899fb08be0a508 [file] [log] [blame]
#!/usr/bin/env python
# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
# Description:
#
# Class for handling linux 'evdev' input devices.
#
# Provides evtest-like functionality if run from the command line:
# $ input_device.py -d /dev/input/event6
""" Read properties and events of a linux input device. """
import array
import copy
import fcntl
import os.path
import re
import select
import struct
import time
from collections import OrderedDict
from linux_input import *
# The regular expression of possible keyboard types.
KEYBOARD_TYPES = '(keyboard|chromeos-ec-i2c|cros-ec-spi|cros-ec-i2c|cros_ec)'
_DEVICE_INFO_FILE = '/proc/bus/input/devices'
class Valuator:
""" A Valuator just stores a value """
def __init__(self):
self.value = 0
class SwValuator(Valuator):
""" A Valuator used for EV_SW (switch) events """
def __init__(self, value):
self.value = value
class AbsValuator(Valuator):
"""
An AbsValuator, used for EV_ABS events stores a value as well as other
properties of the corresponding absolute axis.
"""
def __init__(self, value, min_value, max_value, fuzz, flat, resolution):
self.value = value
self.min = min_value
self.max = max_value
self.fuzz = fuzz
self.flat = flat
self.resolution = resolution
class InputEvent:
"""
Linux evdev input event
An input event has the following fields which can be accessed as public
properties of this class:
tv_sec
tv_usec
type
code
value
"""
def __init__(self, tv_sec=0, tv_usec=0, type=0, code=0, value=0):
self.format = input_event_t
self.format_size = struct.calcsize(self.format)
(self.tv_sec, self.tv_usec, self.type, self.code,
self.value) = (tv_sec, tv_usec, type, code, value)
def read(self, stream):
""" Read an input event from the provided stream and unpack it. """
packed = stream.read(self.format_size)
(self.tv_sec, self.tv_usec, self.type, self.code,
self.value) = struct.unpack(self.format, packed)
def write(self, stream):
""" Pack an input event and write it to the provided stream. """
packed = struct.pack(self.format, self.tv_sec, self.tv_usec, self.type,
self.code, self.value)
stream.write(packed)
stream.flush()
def __str__(self):
t = EV_TYPES.get(self.type, self.type)
if self.type in EV_STRINGS:
c = EV_STRINGS[self.type].get(self.code, self.code)
else:
c = self.code
return ('%d.%06d: %s[%s] = %d' %
(self.tv_sec, self.tv_usec, t, c, self.value))
class InputDevice:
"""
Linux evdev input device
A linux kernel "evdev" device sends a stream of "input events".
These events are grouped together into "input reports", which is a set of
input events ending in a single EV_SYN/SYN_REPORT event.
Each input event is tagged with a type and a code.
A given input device supports a subset of the possible types, and for
each type it supports a subset of the possible codes for that type.
The device maintains a "valuator" for each supported type/code pairs.
There are two types of "valuators":
Normal valuators represent just a value.
Absolute valuators are only for EV_ABS events. They have more fields:
value, minimum, maximum, resolution, fuzz, flatness
Note: Relative and Absolute "Valuators" are also often called relative
and absolute axis, respectively.
The evdev protocol is stateful. Input events are only sent when the values
of a valuator actually changes. This dramatically reduces the stream of
events emenating from the kernel.
Multitouch devices are a special case. There are two types of multitouch
devices defined in the kernel:
Multitouch type "A" (MT-A) devices:
In each input report, the device sends an unordered list of all
active contacts. The data for each active contact is separated
in the input report by an EV_SYN/SYN_MT_REPORT event.
Thus, the MT-A contact event stream is not stateful.
Note: MT-A is not currently supported by this class.
Multitouch type "B" (MT-B) devices:
The device maintains a list of slots, where each slot contains a
single contact. In each input report, the device only sends
information about the slots that have changed.
Thus, the MT-B contact event stream is stateful.
When reporting multiple slots, the EV_ABS/MT_SLOT valuator is used
to indicate the 'current' slot for which subsequent EV_ABS/ABS_MT_*
valuator events apply.
An inactive slot has EV_ABS/ABS_MT_TRACKING_ID == -1
Active slots have EV_ABS/ABS_MT_TRACKING_ID >= 0
Besides maintaining the set of supported ABS_MT valuators in the supported
valuator list, a array of slots is also maintained. Each slot has its own
unique copy of just the supported ABS_MT valuators. This represents the
current state of that slot.
"""
def __init__(self, path, ev_syn_cb=None):
"""
Constructor opens the device file and probes its properties.
Note: The device file is left open when the constructor exits.
"""
self.path = path
self.ev_syn_cb = ev_syn_cb
self.events = {} # dict { ev_type : dict { ev_code : Valuator } }
self.mt_slots = [] # [ dict { mt_code : AbsValuator } ] * |MT-B slots|
# Open the device node, and use ioctls to probe its properties
self.f = None
self.f = open(path, 'rb+', buffering=0)
self._ioctl_version()
self._ioctl_id()
self._ioctl_name()
for t in self._ioctl_types():
self._ioctl_codes(t)
self._setup_mt_slots()
def __del__(self):
"""
Deconstructor closes the device file, if it is open.
"""
if self.f and not self.f.closed:
self.f.close()
def process_event(self, ev):
"""
Processes an incoming input event.
Returns True for EV_SYN/SYN_REPORT events to indicate that a complete
input report has been received.
Returns False for other events.
Events not supported by this device are silently ignored.
For MT events, updates the slot valuator value for the current slot.
If current slot is the 'primary' slot, also updates the events entry.
For all other events, updates the corresponding valuator value.
"""
if ev.type == EV_SYN and ev.code == SYN_REPORT:
return True
elif ev.type not in self.events or ev.code not in self.events[ev.type]:
return False
elif self.is_mt_b() and ev.type == EV_ABS and ev.code in ABS_MT_RANGE:
# TODO: Handle MT-A
slot = self._get_current_slot()
slot[ev.code].value = ev.value
# if the current slot is the "primary" slot,
# update the events[] entry, too.
if slot == self._get_mt_primary_slot():
self.events[ev.type][ev.code].value = ev.value
else:
self.events[ev.type][ev.code].value = ev.value
return False
def _ioctl_version(self):
""" Queries device file for version information. """
# Version is a 32-bit integer, which encodes 8-bit major version,
# 8-bit minor version and 16-bit revision.
version = array.array('I', [0])
fcntl.ioctl(self.f, EVIOCGVERSION, version, 1)
self.version = (version[0] >> 16, (version[0] >> 8) & 0xff,
version[0] & 0xff)
def _ioctl_id(self):
""" Queries device file for input device identification. """
# struct input_id is 4 __u16
gid = array.array('H', [0] * 4)
fcntl.ioctl(self.f, EVIOCGID, gid, 1)
self.id_bus = gid[ID_BUS]
self.id_vendor = gid[ID_VENDOR]
self.id_product = gid[ID_PRODUCT]
self.id_version = gid[ID_VERSION]
def _ioctl_name(self):
""" Queries device file for the device name. """
# Device name is a C-string up to 255 bytes in length.
name_len = 255
name = array.array('B', [0] * name_len)
name_len = fcntl.ioctl(self.f, EVIOCGNAME(name_len), name, 1)
self.name = name[0:name_len-1].tostring()
def _ioctl_get_switch(self, sw):
"""
Queries device file for current value of all switches and returns
a boolean indicating whether the switch sw is set.
"""
size = SW_CNT / 8 # Buffer size of one __u16
buf = array.array('H', [0])
fcntl.ioctl(self.f, EVIOCGSW(size), buf)
return SwValuator(((buf[0] >> sw) & 0x01) == 1)
def _ioctl_absinfo(self, axis):
"""
Queries device file for absinfo structure for given absolute axis.
"""
# struct input_absinfo is 6 __s32
a = array.array('i', [0] * 6)
fcntl.ioctl(self.f, EVIOCGABS(axis), a, 1)
return AbsValuator(a[0], a[1], a[2], a[3], a[4], a[5])
def _ioctl_codes(self, ev_type):
"""
Queries device file for supported event codes for given event type.
"""
self.events[ev_type] = {}
if ev_type not in EV_SIZES:
return
size = EV_SIZES[ev_type] / 8 # Convert bits to bytes
ev_code = array.array('B', [0] * size)
try:
count = fcntl.ioctl(self.f, EVIOCGBIT(ev_type, size), ev_code, 1)
for c in range(count * 8):
if test_bit(c, ev_code):
if ev_type == EV_ABS:
self.events[ev_type][c] = self._ioctl_absinfo(c)
elif ev_type == EV_SW:
self.events[ev_type][c] = self._ioctl_get_switch(c)
else:
self.events[ev_type][c] = Valuator()
except IOError as (errno, strerror):
# Errno 22 signifies that this event type has no event codes.
if errno != 22:
raise
def _ioctl_types(self):
""" Queries device file for supported event types. """
ev_types = array.array('B', [0] * (EV_CNT / 8))
fcntl.ioctl(self.f, EVIOCGBIT(EV_SYN, EV_CNT / 8), ev_types, 1)
types = []
for t in range(EV_CNT):
if test_bit(t, ev_types):
types.append(t)
return types
def _convert_slot_index_to_slot_id(self, index):
""" Convert a slot index in self.mt_slots to its slot id. """
return self.abs_mt_slot.min + index
def _ioctl_mt_slots(self):
"""Query mt slots values using ioctl.
The ioctl buffer argument should be binary equivalent to
struct input_mt_request_layout {
__u32 code;
__s32 values[num_slots];
Note that the slots information returned by EVIOCGMTSLOTS
corresponds to the slot ids ranging from abs_mt_slot.min to
abs_mt_slot.max which is not necessarily the same as the
slot indexes ranging from 0 to num_slots - 1 in self.mt_slots.
We need to map between the slot index and the slot id correctly.
};
"""
# Iterate through the absolute mt events that are supported.
for c in range(ABS_MT_FIRST, ABS_MT_LAST):
if c not in self.events[EV_ABS]:
continue
# Sync with evdev kernel driver for the specified code c.
mt_slot_info = array.array('i', [c] + [0] * self.num_slots)
mt_slot_info_len = (self.num_slots + 1) * mt_slot_info.itemsize
fcntl.ioctl(self.f, EVIOCGMTSLOTS(mt_slot_info_len), mt_slot_info)
values = mt_slot_info[1:]
for slot_index in range(self.num_slots):
slot_id = self._convert_slot_index_to_slot_id(slot_index)
self.mt_slots[slot_index][c].value = values[slot_id]
def _setup_mt_slots(self):
"""
Sets up the device's mt_slots array.
Each element of the mt_slots array is initialized as a deepcopy of a
dict containing all of the MT valuators from the events dict.
"""
# TODO(djkurtz): MT-A
if not self.is_mt_b():
return
ev_abs = self.events[EV_ABS]
# Create dict containing just the MT valuators
mt_abs_info = dict((axis, ev_abs[axis])
for axis in ev_abs
if axis in ABS_MT_RANGE)
# Initialize TRACKING_ID to -1
mt_abs_info[ABS_MT_TRACKING_ID].value = -1
# Make a copy of mt_abs_info for each MT slot
self.abs_mt_slot = ev_abs[ABS_MT_SLOT]
self.num_slots = self.abs_mt_slot.max - self.abs_mt_slot.min + 1
for s in range(self.num_slots):
self.mt_slots.append(copy.deepcopy(mt_abs_info))
self._ioctl_mt_slots()
def get_current_slot_id(self):
"""
Return the current slot id.
"""
if not self.is_mt_b():
return None
return self.events[EV_ABS][ABS_MT_SLOT].value
def _get_current_slot(self):
"""
Returns the current slot, as indicated by the last ABS_MT_SLOT event.
"""
current_slot_id = self.get_current_slot_id()
if current_slot_id is None:
return None
return self.mt_slots[current_slot_id]
def _get_tid(self, slot):
""" Returns the tracking_id for the given MT slot. """
return slot[ABS_MT_TRACKING_ID].value
def _get_mt_valid_slots(self):
"""
Returns a list of valid slots.
A valid slot is a slot whose tracking_id != -1.
"""
return [s for s in self.mt_slots if self._get_tid(s) != -1]
def _get_mt_primary_slot(self):
"""
Returns the "primary" MT-B slot.
The "primary" MT-B slot is arbitrarily chosen as the slot with lowest
tracking_id (> -1). It is used to make an MT-B device look like
single-touch (ST) device.
"""
slot = None
for s in self.mt_slots:
tid = self._get_tid(s)
if tid < 0:
continue
if not slot or tid < self._get_tid(slot):
slot = s
return slot
def _code_if_mt(self, type, code):
"""
Returns MT-equivalent event code for certain specific event codes
"""
if type != EV_ABS:
return code
elif code == ABS_X:
return ABS_MT_POSITION_X
elif code == ABS_Y:
return ABS_MT_POSITION_Y
elif code == ABS_PRESSURE:
return ABS_MT_PRESSURE
elif code == ABS_TOOL_WIDTH:
return ABS_TOUCH_MAJOR
else:
return code
def _get_valuator(self, type, code):
""" Returns Valuator for given event type and code """
if (not type in self.events) or (not code in self.events[type]):
return None
if type == EV_ABS:
code = self._code_if_mt(type, code)
return self.events[type][code]
def _get_value(self, type, code):
"""
Returns the value of the valuator with the give event (type, code).
"""
axis = self._get_valuator(type, code)
if not axis:
return None
return axis.value
def _get_min(self, type, code):
"""
Returns the min value of the valuator with the give event (type, code).
Note: Only AbsValuators (EV_ABS) have max values.
"""
axis = self._get_valuator(type, code)
if not axis:
return None
return axis.min
def _get_max(self, type, code):
"""
Returns the min value of the valuator with the give event (type, code).
Note: Only AbsValuators (EV_ABS) have max values.
"""
axis = self._get_valuator(type, code)
if not axis:
return None
return axis.max
""" Public accessors """
def get_num_fingers(self):
if self.is_mt_b():
return len(self._get_mt_valid_slots())
elif self.is_mt_a():
return 0 # TODO(djkurtz): MT-A
else: # Single-Touch case
if not self._get_value(EV_KEY, BTN_TOUCH) == 1:
return 0
elif self._get_value(EV_KEY, BTN_TOOL_TRIPLETAP) == 1:
return 3
elif self._get_value(EV_KEY, BTN_TOOL_DOUBLETAP) == 1:
return 2
elif self._get_value(EV_KEY, BTN_TOOL_FINGER) == 1:
return 1
else:
return 0
def get_x(self):
return self._get_value(EV_ABS, ABS_X)
def get_x_min(self):
return self._get_min(EV_ABS, ABS_X)
def get_x_max(self):
return self._get_max(EV_ABS, ABS_X)
def get_y(self):
return self._get_value(EV_ABS, ABS_Y)
def get_y_min(self):
return self._get_min(EV_ABS, ABS_Y)
def get_y_max(self):
return self._get_max(EV_ABS, ABS_Y)
def get_pressure(self):
return self._get_value(EV_ABS, ABS_PRESSURE)
def get_pressure_min(self):
return self._get_min(EV_ABS, ABS_PRESSURE)
def get_pressure_max(self):
return self._get_max(EV_ABS, ABS_PRESSURE)
def get_left(self):
return int(self._get_value(EV_KEY, BTN_LEFT) == 1)
def get_right(self):
return int(self._get_value(EV_KEY, BTN_RIGHT) == 1)
def get_middle(self):
return int(self._get_value(EV_KEY, BTN_MIDDLE) == 1)
def get_microphone_insert(self):
return self._get_value(EV_SW, SW_MICROPHONE_INSERT)
def get_headphone_insert(self):
return self._get_value(EV_SW, SW_HEADPHONE_INSERT)
def get_lineout_insert(self):
return self._get_value(EV_SW, SW_LINEOUT_INSERT)
def is_touchpad(self):
return ((EV_KEY in self.events) and
(BTN_TOOL_FINGER in self.events[EV_KEY]) and
(EV_ABS in self.events))
def is_keyboard(self):
return ((EV_KEY in self.events) and
(KEY_F2 in self.events[EV_KEY]))
def is_touchscreen(self):
return ((EV_KEY in self.events) and
(BTN_TOUCH in self.events[EV_KEY]) and
(not BTN_TOOL_FINGER in self.events[EV_KEY]) and
(EV_ABS in self.events))
def is_lid(self):
return ((EV_SW in self.events) and
(SW_LID in self.events[EV_SW]))
def is_mt_b(self):
return self.is_mt() and ABS_MT_SLOT in self.events[EV_ABS]
def is_mt_a(self):
return self.is_mt() and ABS_MT_SLOT not in self.events[EV_ABS]
def is_mt(self):
return (EV_ABS in self.events and
(set(self.events[EV_ABS]) & set(ABS_MT_RANGE)))
def is_hp_jack(self):
return (EV_SW in self.events and
SW_HEADPHONE_INSERT in self.events[EV_SW])
def is_mic_jack(self):
return (EV_SW in self.events and
SW_MICROPHONE_INSERT in self.events[EV_SW])
def is_audio_jack(self):
return (EV_SW in self.events and
((SW_HEADPHONE_INSERT in self.events[EV_SW]) or
(SW_MICROPHONE_INSERT in self.events[EV_SW] or
(SW_LINEOUT_INSERT in self.events[EV_SW]))))
""" Debug helper print functions """
def print_abs_info(self, axis):
if EV_ABS in self.events and axis in self.events[EV_ABS]:
a = self.events[EV_ABS][axis]
print ' Value %6d' % a.value
print ' Min %6d' % a.min
print ' Max %6d' % a.max
if a.fuzz != 0:
print ' Fuzz %6d' % a.fuzz
if a.flat != 0:
print ' Flat %6d' % a.flat
if a.resolution != 0:
print ' Resolution %6d' % a.resolution
def print_props(self):
print ('Input driver Version: %d.%d.%d' %
(self.version[0], self.version[1], self.version[2]))
print ('Input device ID: bus %x vendor %x product %x version %x' %
(self.id_bus, self.id_vendor, self.id_product, self.id_version))
print 'Input device name: "%s"' % (self.name)
for t in self.events:
print ' Event type %d (%s)' % (t, EV_TYPES.get(t, '?'))
for c in self.events[t]:
if (t in EV_STRINGS):
code = EV_STRINGS[t].get(c, '?')
print ' Event code %s (%d)' % (code, c)
else:
print ' Event code (%d)' % (c)
self.print_abs_info(c)
def get_slots(self):
""" Get those slots with positive tracking IDs. """
slot_dict = OrderedDict()
for slot_index in range(self.num_slots):
slot = self.mt_slots[slot_index]
if self._get_tid(slot) == -1:
continue
slot_id = self._convert_slot_index_to_slot_id(slot_index)
slot_dict[slot_id] = slot
return slot_dict
def print_slots(self):
slot_dict = self.get_slots()
for slot_id, slot in slot_dict.items():
print 'slot #%d' % slot_id
for a in slot:
abs = EV_STRINGS[EV_ABS].get(a, '?')
print ' %s = %6d' % (abs, slot[a].value)
def print_report(device):
print '----- EV_SYN -----'
if device.is_touchpad():
f = device.get_num_fingers()
if f == 0:
return
x = device.get_x()
y = device.get_y()
z = device.get_pressure()
l = device.get_left()
print 'Left=%d Fingers=%d X=%d Y=%d Pressure=%d' % (l, f, x, y, z)
if device.is_mt():
device.print_slots()
def get_device_node(device_type):
"""Get the keyboard device node through device info file.
Example of the keyboard device information looks like
I: Bus=0011 Vendor=0001 Product=0001 Version=ab41
N: Name="AT Translated Set 2 keyboard"
P: Phys=isa0060/serio0/input0
S: Sysfs=/devices/platform/i8042/serio0/input/input5
U: Uniq=
H: Handlers=sysrq kbd event5
"""
device_node = None
device_found = None
device_pattern = re.compile('N: Name=.*%s' % device_type, re.I)
event_number_pattern = re.compile(r'H: Handlers=.*event(\d?)', re.I)
with open(_DEVICE_INFO_FILE) as info:
for line in info:
if device_found:
result = event_number_pattern.search(line)
if result:
event_number = int(result.group(1))
device_node = '/dev/input/event%d' % event_number
break
else:
device_found = device_pattern.search(line)
return device_node
if __name__ == "__main__":
from optparse import OptionParser
import glob
parser = OptionParser()
parser.add_option("-a", "--audio_jack", action="store_true",
dest="audio_jack", default=False,
help="Find and use all audio jacks")
parser.add_option("-d", "--devpath", dest="devpath",
default="/dev/input/event0",
help="device path (/dev/input/event0)")
parser.add_option("-q", "--quiet", action="store_false", dest="verbose",
default=True, help="print less messages to stdout")
parser.add_option("-t", "--touchpad", action="store_true", dest="touchpad",
default=False, help="Find and use first touchpad device")
(options, args) = parser.parse_args()
# TODO: Use gudev to detect touchpad
devices = []
if options.touchpad:
for path in glob.glob('/dev/input/event*'):
device = InputDevice(path)
if device.is_touchpad():
print 'Using touchpad %s.' % path
options.devpath = path
devices.append(device)
break
else:
print 'No touchpad found!'
exit()
elif options.audio_jack:
for path in glob.glob('/dev/input/event*'):
device = InputDevice(path)
if device.is_audio_jack():
devices.append(device)
device = None
elif os.path.exists(options.devpath):
print 'Using %s.' % options.devpath
devices.append(InputDevice(options.devpath))
else:
print '%s does not exist.' % options.devpath
exit()
for device in devices:
device.print_props()
if device.is_touchpad():
print ('x: (%d,%d), y: (%d,%d), z: (%d, %d)' %
(device.get_x_min(), device.get_x_max(),
device.get_y_min(), device.get_y_max(),
device.get_pressure_min(), device.get_pressure_max()))
device.print_slots()
print 'Number of fingers: %d' % device.get_num_fingers()
print 'Current slot id: %d' % device.get_current_slot_id()
print '------------------'
print
ev = InputEvent()
while True:
_rl, _, _ = select.select([d.f for d in devices], [], [])
for fd in _rl:
# Lookup for the device which owns fd.
device = [d for d in devices if d.f == fd][0]
try:
ev.read(fd)
except KeyboardInterrupt:
exit()
is_syn = device.process_event(ev)
print ev
if is_syn:
print_report(device)