qmk_firmware/util/polling_rate.py

83 lines
2.8 KiB
Python

#!/usr/bin/env python3
import usb
USB_INTERFACE_CLASS_HID = 0x03
def usb_device_spec(spec):
major = spec >> 8
minor = (spec >> 4) & 0xF
return f"{major}.{minor}"
def usb_device_speed(speed):
if speed == 1:
return "Low-speed"
elif speed == 2:
return "Full-speed"
elif speed == 3:
return "High-speed"
elif speed == 4:
return "SuperSpeed"
elif speed == 5:
return "SuperSpeed+"
return "Speed unknown"
def usb_hid_interface_subclass(subclass):
if subclass == 0x00:
return "None"
elif subclass == 0x01:
return "Boot"
else:
return f"Unknown (0x{subclass:02X})"
def usb_hid_interface_protocol(subclass, protocol):
if subclass == 0x00 and protocol == 0x00:
return "None"
elif subclass == 0x01:
if protocol == 0x00:
return "None"
elif protocol == 0x01:
return "Keyboard"
elif protocol == 0x02:
return "Mouse"
return f"Unknown (0x{protocol:02X})"
def usb_interface_polling_rate(speed, interval):
if speed >= 3:
return f"{interval * 125} μs ({8000 // interval} Hz)"
else:
return f"{interval} ms ({1000 // interval} Hz)"
if __name__ == '__main__':
devices = usb.core.find(find_all=True)
for device in devices:
try:
configuration = device.get_active_configuration()
except NotImplementedError:
continue
hid_interfaces = []
for interface in configuration.interfaces():
if interface.bInterfaceClass == USB_INTERFACE_CLASS_HID:
hid_interfaces.append(interface)
if len(hid_interfaces) > 0:
print(f"{device.manufacturer} {device.product} ({device.idVendor:04X}:{device.idProduct:04X}:{device.bcdDevice:04X}), {usb_device_spec(device.bcdUSB)} {usb_device_speed(device.speed)}")
for interface in hid_interfaces:
print(f"└─ HID Interface {interface.bInterfaceNumber}")
subclass = interface.bInterfaceSubClass
protocol = interface.bInterfaceProtocol
print(f" ├─ Subclass: {usb_hid_interface_subclass(subclass)}")
print(f" ├─ Protocol: {usb_hid_interface_protocol(subclass, protocol)}")
for endpoint in interface.endpoints():
endpoint_address = endpoint.bEndpointAddress & 0xF
endpoint_direction = "IN" if endpoint.bEndpointAddress & 0x80 else "OUT"
print(f" └─ Endpoint {endpoint_address} {endpoint_direction}")
print(f" ├─ Endpoint Size: {endpoint.wMaxPacketSize} bytes")
print(f" └─ Polling Rate: {usb_interface_polling_rate(device.speed, endpoint.bInterval)}")