forked from nmilford/stormLauncher
-
Notifications
You must be signed in to change notification settings - Fork 0
/
stormLauncher.py
126 lines (96 loc) · 4.48 KB
/
stormLauncher.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/python
# Copyright 2012, Nathan Milford
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# The following script will control the Dream Cheeky Storm & Thunder USB
# Missile Launchers. There are a few projects for using older launchers
# in Linux, but I couldn't find any for this launcher, so... enjoy.
# Thunder: http://www.dreamcheeky.com/thunder-missile-launcher
# O.I.C Storm: http://www.dreamcheeky.com/storm-oic-missile-launcher
# This script requires:
# * PyUSB 1.0+, apt in Debian/Ubuntu installs 0.4.
# * The ImageTk library. On Debian/Ubuntu 'sudo apt-get install python-imaging-tk'
# Also, unless you want to toggle with udev rules, it needs to be run as root
# Use arrows to aim. Sse the left enter to fire.
# BTW, Leeroy Jenkins Mode .wav is from: http://www.leeroyjenkins.net/soundbites/warcry.wav
import os
import sys
import time
import pygame
import usb.core
from Tkinter import *
from PIL import Image, ImageTk
wavFile = "warcry.wav"
logoFile = "stormLauncher.png"
class launchControl(Frame):
def __init__(self):
self.dev = usb.core.find(idVendor=0x2123, idProduct=0x1010)
if self.dev is None:
raise ValueError('Launcher not found.')
if self.dev.is_kernel_driver_active(0) is True:
self.dev.detach_kernel_driver(0)
self.dev.set_configuration()
Frame.__init__(self)
self.pack()
self.master.title("Launch Control")
self.master.geometry("400x90")
self.logo = ImageTk.PhotoImage(Image.open(logoFile))
self.panel1 = Label(self, image=self.logo)
self.panel1.pack(side='top', fill='both', expand='yes')
self.panel1.image = self.logo
self.message1 = StringVar()
self.line1 = Label(self, textvariable = self.message1)
self.message1.set("Aim (arrow keys) & Fire (return)!")
self.line1.pack()
if os.path.isfile(wavFile):
self.hasSound = IntVar()
self.check1 = Checkbutton(self, text = "Leeroy Jenkins Mode?", variable = self.hasSound, onvalue = 1, offvalue = 0)
self.check1.pack()
self.master.bind("<KeyPress-Up>", self.turretUp)
self.master.bind("<KeyRelease-Up>", self.turretStop)
self.master.bind("<KeyPress-Down>", self.turretDown)
self.master.bind("<KeyRelease-Down>", self.turretStop)
self.master.bind("<KeyPress-Left>", self.turretLeft)
self.master.bind("<KeyRelease-Left>", self.turretStop)
self.master.bind("<KeyPress-Right>", self.turretRight)
self.master.bind("<KeyRelease-Right>", self.turretStop)
self.master.bind("<KeyPress-Return>", self.turretFire)
def turretUp(self, event):
self.message1.set("Turret Up.")
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x02,0x00,0x00,0x00,0x00,0x00,0x00])
def turretDown(self, event):
self.message1.set("Turret Down.")
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x01,0x00,0x00,0x00,0x00,0x00,0x00])
def turretLeft(self, event):
self.message1.set("Turret Left.")
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x04,0x00,0x00,0x00,0x00,0x00,0x00])
def turretRight(self, event):
self.message1.set("Turret Right.")
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x08,0x00,0x00,0x00,0x00,0x00,0x00])
def turretStop(self, event):
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x20,0x00,0x00,0x00,0x00,0x00,0x00])
def turretFire(self, event):
self.message1.set("FIRE!")
if os.path.isfile(wavFile):
if self.hasSound.get() == 1:
pygame.init()
sound = pygame.mixer.Sound("warcry.wav")
sound.play()
time.sleep(3)
self.dev.ctrl_transfer(0x21,0x09,0,0,[0x02,0x10,0x00,0x00,0x00,0x00,0x00,0x00])
if __name__ == '__main__':
if not os.geteuid() == 0:
sys.exit("Script must be run as root.")
launchControl().mainloop()