Skip to content

Instantly share code, notes, and snippets.

@JZHeadley

JZHeadley/gist:904992fad766ee6245e62e81bef4a011

Created March 24, 2018 17:20
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save JZHeadley/904992fad766ee6245e62e81bef4a011 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save JZHeadley/904992fad766ee6245e62e81bef4a011 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. JZHeadley created this gist Mar 24, 2018.
    55 changes: 55 additions & 0 deletions gistfile1.txt
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,55 @@
    import platform
    import time
    import usb.core
    import usb.util


    class Armageddon(object):
    """
    Based on https://github.com/codedance/Retaliation
    """
    DOWN = 0x01
    UP = 0x02
    LEFT = 0x04
    RIGHT = 0x08
    FIRE = 0x10
    STOP = 0x20

    DEVICE_ORIGINAL = 'Original'
    DEVICE_THUNDER = 'Thunder'

    def __init__(self):
    self._get_device()
    self._detach_hid()
    self.DEVICE.set_configuration()

    def _get_device(self):
    self.DEVICE = usb.core.find(idVendor=0x2123, idProduct=0x1010)
    if self.DEVICE is None:
    self.DEVICE = usb.core.find(idVendor=0x0a81, idProduct=0x0701)
    if self.DEVICE is None:
    raise ValueError('Missile device not found')
    else:
    self.DEVICE_TYPE = self.DEVICE_ORIGINAL
    else:
    self.DEVICE_TYPE = self.DEVICE_THUNDER

    def _detach_hid(self):
    if "Linux" == platform.system():
    try:
    self.DEVICE.detach_kernel_driver(0)
    except Exception, e:
    pass

    def send_cmd(self, cmd):
    if self.DEVICE_THUNDER == self.DEVICE_TYPE:
    self.DEVICE.ctrl_transfer(0x21, 0x09, 0, 0,
    [0x02, cmd, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00])
    elif self.DEVICE_ORIGINAL == self.DEVICE_TYPE:
    self.DEVICE.ctrl_transfer(0x21, 0x09, 0x0200, 0,
    [cmd])

    def send_move(self, cmd, duration_ms):
    self.send_cmd(cmd)
    time.sleep(duration_ms / 1000.0)
    self.send_cmd(self.STOP)