mirror of https://github.com/Trivernis/spydian.git
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.
101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
import os
|
|
import pygame
|
|
|
|
|
|
class Joystick:
|
|
def __init__(self, navigator, light, configuration, camera):
|
|
# joystick
|
|
pygame.joystick.init()
|
|
joystick = pygame.joystick.Joystick(0)
|
|
joystick.init()
|
|
# sound
|
|
pygame.mixer.init()
|
|
sound = pygame.mixer.Sound('./sounds/gasgasgas.wav')
|
|
sound.set_volume(0.3)
|
|
self.camera = camera
|
|
self.sound = sound
|
|
self.joystick = joystick
|
|
self.navigator = navigator
|
|
self.light = light
|
|
self.configuration = configuration
|
|
self.pressed = []
|
|
self.splaying = False
|
|
|
|
def _button_check(self, name, i, c, button):
|
|
""" Checks if a button was pressed and wasn't pressed before. """
|
|
if i == c[name]:
|
|
if button == 1 and i not in self.pressed:
|
|
self.pressed.append(i)
|
|
return 1
|
|
elif button == 0 and i in self.pressed:
|
|
self.pressed.remove(i)
|
|
return 2
|
|
return 0
|
|
|
|
def _handle_buttons(self, axis_dict, c):
|
|
for i in range(self.joystick.get_numbuttons()):
|
|
button = self.joystick.get_button(i)
|
|
axis_dict['Button {}'.format(i)] = button
|
|
|
|
if self._button_check('LIGHT', i, c, button) == 1:
|
|
self.light.switch()
|
|
|
|
elif self._button_check('MUSIC', i, c, button) == 1:
|
|
if self.splaying:
|
|
self.sound.stop()
|
|
self.splaying = False
|
|
else:
|
|
self.sound.play()
|
|
self.splaying = True
|
|
|
|
elif self._button_check('VOLIN', i, c, button) == 1:
|
|
self.sound.set_volume(self.sound.get_volume() + 0.1)
|
|
elif self._button_check('VOLIN', i, c, button) == 2:
|
|
self.sound.set_volume(self.sound.get_volume() - 0.1)
|
|
|
|
elif self._button_check('VOLDE', i, c, button):
|
|
pass
|
|
|
|
elif self._button_check('REC', i, c, button) == 1:
|
|
self._save_camimg()
|
|
return axis_dict
|
|
|
|
def handle(self):
|
|
c = self.configuration
|
|
nav = self.navigator
|
|
axis_dict = {}
|
|
# axes
|
|
for i in range(self.joystick.get_numaxes()):
|
|
axis = self.joystick.get_axis(i)
|
|
axis_dict['Axis {}'.format(i)] = axis
|
|
|
|
if i == c['GAS'] and axis > 0.1:
|
|
nav.forward()
|
|
elif i == c['GAS'] and axis < 0.1:
|
|
nav.stop()
|
|
axis_dict = self._handle_buttons(axis_dict, c)
|
|
# hats
|
|
for i in range(self.joystick.get_numhats()):
|
|
hat = self.joystick.get_hat(i)
|
|
axis_dict['Hat {}'.format(i)] = hat
|
|
|
|
if hat == (-1, 0):
|
|
nav.left()
|
|
elif hat == (1, 0):
|
|
nav.right()
|
|
else:
|
|
nav.straight()
|
|
|
|
axis_dict['Volume'] = self.sound.get_volume()
|
|
return axis_dict
|
|
|
|
def _save_camimg(self):
|
|
self.camera.new_frame()
|
|
img = self.camera.image
|
|
if os.path.isfile('image.jpg'):
|
|
count = 0
|
|
while os.path.isfile('./images/img{}.jpg'.format(count)):
|
|
count += 1
|
|
os.rename('image.jpg', 'images/img{}.jpg'.format(count))
|
|
pygame.image.save(img, 'image.jpg')
|