JezK
Edit File: libwacom-show-stylus
#!/usr/bin/env python3 # # Permission to use, copy, modify, distribute, and sell this software # and its documentation for any purpose is hereby granted without # fee, provided that the above copyright notice appear in all copies # and that both that copyright notice and this permission notice # appear in supporting documentation, and that the name of Red Hat # not be used in advertising or publicity pertaining to distribution # of the software without specific, written prior permission. Red # Hat makes no representations about the suitability of this software # for any purpose. It is provided "as is" without express or implied # warranty. # # THE AUTHORS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, # INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN # NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY SPECIAL, INDIRECT OR # CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS # OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, # NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN # CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. # import argparse import configparser import sys from pathlib import Path try: import libevdev import pyudev except ModuleNotFoundError as e: print("Error: {}".format(str(e)), file=sys.stderr) print( "One or more python modules are missing. Please install those " "modules and re-run this tool." ) sys.exit(1) class Ansi: clearline = "\x1B[K" @classmethod def up(cls, count): return f"\x1B[{count}A" @classmethod def down(cls, count): return f"\x1B[{count}B" @classmethod def right(cls, count): return f"\x1B[{count}C" @classmethod def left(cls, count): return f"\x1B[{count}D" def die(msg): print(msg, file=sys.stderr) sys.exit(1) def select_device(): context = pyudev.Context() for device in context.list_devices(subsystem="input"): if device.get("ID_INPUT_TABLET", 0) and (device.device_node or "").startswith( "/dev/input/event" ): name = device.get("NAME", None) if not name: name = next( (p.get("NAME") for p in device.ancestors if p.get("NAME")), "unknown", ) print("Using {}: {}".format(name or "unknown", device.device_node)) return device.device_node die("Unable to find a tablet device.") def record_events(ns): with open(ns.device_path, "rb") as fd: d = libevdev.Device(fd) if not d.absinfo[libevdev.EV_ABS.ABS_MISC]: die("Device only supports generic styli") tool_bits = set( c for c in libevdev.EV_KEY.codes if c.name.startswith("BTN_TOOL_") ) styli = {} # dict of (type, serial) = proximity_state current_type, current_serial = 0, 0 in_prox = False dirty = False print("Please put tool in proximity") try: while True: for event in d.events(): if event.matches(libevdev.EV_ABS.ABS_MISC): if event.value != 0: current_type = event.value dirty = True elif event.matches(libevdev.EV_MSC.MSC_SERIAL): if event.value != 0: current_serial = event.value & 0xFFFFFFFF dirty = True elif event.code in tool_bits: # print(f'Current prox: {event.value}') in_prox = event.value != 0 dirty = True elif event.matches(libevdev.EV_SYN.SYN_REPORT) and dirty: dirty = False print( f"{Ansi.up(len(styli))}{Ansi.left(10000)}{Ansi.clearline}", end="", ) styli[(current_type, current_serial)] = in_prox for s, prox in styli.items(): tid, serial = s print( f"Tool id {tid:#x} serial {serial:#x} in-proximity: {prox} " ) except KeyboardInterrupt: print("Terminating") return [s[0] for s in styli.keys()] def load_data_files(): lookup_paths = ( ("./data/",), ("/usr/share/libwacom", "/etc/libwacom"), ("/usr/share/libwacom/", "/etc/libwacom/"), ) stylusfiles = [] for paths in lookup_paths: stylusfiles = [] for p in paths: files = list(Path(p).glob("*.stylus")) if files: stylusfiles += files if any(stylusfiles): break else: die("Unable to find a libwacom.stylus data file") print(f'Using stylus file(s): {", ".join([str(s) for s in stylusfiles])}') styli = {} for path in stylusfiles: config = configparser.ConfigParser() config.read(path) for stylus_id in config.sections(): sid = int(stylus_id, 16) styli[sid] = config[stylus_id].get("Group", sid) return styli def main(): parser = argparse.ArgumentParser(description="Tool to show tablet stylus ids") parser.add_argument( "device_path", nargs="?", default=None, help="Path to the /dev/input/event node" ) ns = parser.parse_args() if not ns.device_path: ns.device_path = select_device() all_styli = load_data_files() styli = record_events(ns) groups = [] for sid in styli: if sid in all_styli: groups.append(all_styli[sid]) else: print(f"Unknown stylus id {sid:#x}. New entry needed") print("Suggested line for .tablet file:") print(f"Styli={';'.join(set(groups))}") if __name__ == "__main__": try: main() except PermissionError: die("Insufficient permissions, please run me as root")