You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
import usb.core
|
|
import usb.util
|
|
from array import array
|
|
import pdb
|
|
|
|
class GK1100():
|
|
COMMANDS = ["", "SET_ANIMATION"]
|
|
|
|
def __init__(self):
|
|
self.ANIMATIONS = ["OFF", "STATIC", "CIRCLE", "CUSTOM", "COLOR_CYCLE", "AFTER_GLOW", "STARS", "ROW_ROCKET"]
|
|
self.COLORS = ["RED", "WHITE", "GREEN", "BLUE", "PINK", "YELLOW", "LIGHT_BLUE", "VAR"]
|
|
dev = usb.core.find(idVendor=0x0b05, idProduct=0x1835)
|
|
if dev is None:
|
|
raise ValueError('GK1100 keyboard not found!')
|
|
|
|
cfg = dev.get_active_configuration()
|
|
led_interface = cfg[(1,0)]
|
|
if dev.is_kernel_driver_active(1):
|
|
dev.detach_kernel_driver(1)
|
|
usb.util.claim_interface(dev, 1)
|
|
|
|
self.led_out = led_interface.endpoints()[1]
|
|
self.led_in = led_interface.endpoints()[0]
|
|
|
|
def set_animation(self, animation, color, speed, brightness):
|
|
data = array("B", [0x06, 0x0b, 0x01, self.ANIMATIONS.index(animation), brightness, speed, self.COLORS.index(color)])
|
|
self.led_out.write(data)
|
|
self.pprint_reply(self.fetch_reply())
|
|
|
|
def pprint_reply(self, data):
|
|
# TODO: is byte 0 alway 0x06 (magic header?)?
|
|
# TODO: what is byte 1?
|
|
# TODO: what is byte 2?
|
|
animation = data[3]
|
|
brightness = data[4]
|
|
speed = data[5]
|
|
animation_color = data[6]
|
|
success = data[7]
|
|
# TODO: byte 8 could be some kind of checksum
|
|
print("Animation:", animation, "->", self.ANIMATIONS[animation])
|
|
if animation != 2 and animation != 4 and animation != 7 and animation != 0:
|
|
print("Color:", animation_color, "->", self.COLORS[animation_color])
|
|
print("Speed:", speed)
|
|
print("Brightness:", brightness)
|
|
print("Command was successful executed:", True if success == 1 else False)
|
|
print("Left-over data:", data[8:])
|
|
print()
|
|
|
|
|
|
def fetch_reply(self):
|
|
try:
|
|
data = self.led_in.read(128, timeout=None)
|
|
return data
|
|
except Exception:
|
|
return None
|
|
|
|
#class HID_message():
|
|
# def __init__(self, command, payload):
|
|
# self.data = array("B", [0]*64)
|
|
# self.set_header()
|
|
# if command in GK1100.COMMANDS:
|
|
# self.data[2] = GK1100.COMMAND.index(command)
|
|
# else:
|
|
# raise ValueError("Command \"{}\" not found".format(command))
|
|
#
|
|
# def set_header(self):
|
|
# self.data[0] = 6
|
|
# self.data[1] = 11
|
|
|
|
|
|
keyboard = GK1100()
|
|
keyboard.set_animation("STATIC", "PINK", 0, 3)
|
|
keyboard.set_animation("STATIC", "RED", 0, 3)
|
|
keyboard.set_animation("STATIC", "RED", 0, 4)
|
|
keyboard.set_animation("STATIC", "RED", 1, 3)
|
|
pdb.set_trace()
|