parent
e7a8fd6033
commit
d31e0a4d37
@ -0,0 +1,98 @@ |
||||
#!/usr/bin/env python |
||||
|
||||
""" |
||||
LUFA Library |
||||
Copyright (C) Dean Camera, 2013. |
||||
|
||||
dean [at] fourwalledcubicle [dot] com |
||||
www.lufa-lib.org |
||||
""" |
||||
|
||||
""" |
||||
LUFA Generic HID device demo host test script. This script will send a |
||||
continuous stream of generic reports to the device, to show a variable LED |
||||
pattern on the target board. Send and received report data is printed to |
||||
the terminal. |
||||
|
||||
Requires the PyUSB library (http://sourceforge.net/apps/trac/pyusb/). |
||||
""" |
||||
|
||||
import sys |
||||
from time import sleep |
||||
import usb.core |
||||
import usb.util |
||||
|
||||
# Generic HID device VID, PID and report payload length (length is increased |
||||
# by one to account for the Report ID byte that must be pre-pended) |
||||
device_vid = 0x03EB |
||||
device_pid = 0x204F |
||||
|
||||
def get_and_init_hid_device(): |
||||
device = usb.core.find(idVendor=device_vid, idProduct=device_pid) |
||||
|
||||
if device is None: |
||||
sys.exit("Could not find USB device.") |
||||
|
||||
if device.is_kernel_driver_active(0): |
||||
try: |
||||
device.detach_kernel_driver(0) |
||||
except usb.core.USBError as exception: |
||||
sys.exit("Could not detatch kernel driver: %s" % str(exception)) |
||||
|
||||
try: |
||||
device.set_configuration() |
||||
except usb.core.USBError as exception: |
||||
sys.exit("Could not set configuration: %s" % str(exception)) |
||||
|
||||
return device |
||||
|
||||
def send_led_pattern(device, led1, led2, led3, led4): |
||||
# Report data for the demo is LED on/off data |
||||
report_data = [led1, led2, led3, led4] |
||||
|
||||
# Send the generated report to the device |
||||
number_of_bytes_written = device.ctrl_transfer( # Set Report control request |
||||
0b00100001, # bmRequestType (constant for this control request) |
||||
0x09, # bmRequest (constant for this control request) |
||||
0, # wValue (MSB is report type, LSB is report number) |
||||
0, # wIndex (interface number) |
||||
report_data # report data to be sent |
||||
); |
||||
assert number_of_bytes_written == len(report_data) |
||||
|
||||
print("Sent LED Pattern: {0}".format(report_data)) |
||||
|
||||
def receive_led_pattern(hid_device): |
||||
endpoint = hid_device[0][(0,0)][0] |
||||
report_data = hid_device.read(endpoint.bEndpointAddress, endpoint.wMaxPacketSize) |
||||
return list(report_data) |
||||
|
||||
def main(): |
||||
hid_device = get_and_init_hid_device() |
||||
|
||||
print("Connected to device 0x%04X/0x%04X - %s [%s]" % |
||||
(hid_device.idVendor, hid_device.idProduct, |
||||
usb.util.get_string(hid_device, 256, hid_device.iProduct), |
||||
usb.util.get_string(hid_device, 256, hid_device.iManufacturer))) |
||||
|
||||
p = 0 |
||||
while (True): |
||||
# Convert the current pattern index to a bit-mask and send |
||||
send_led_pattern(hid_device, |
||||
(p >> 3) & 1, |
||||
(p >> 2) & 1, |
||||
(p >> 1) & 1, |
||||
(p >> 0) & 1) |
||||
|
||||
# Receive and print the current LED pattern |
||||
led_pattern = receive_led_pattern(hid_device)[0:4] |
||||
print("Received LED Pattern: {0}".format(led_pattern)) |
||||
|
||||
# Compute next LED pattern in sequence |
||||
p = (p + 1) % 16 |
||||
|
||||
# Delay a bit for visual effect |
||||
sleep(.2) |
||||
|
||||
if __name__ == '__main__': |
||||
main() |
Loading…
Reference in new issue