sejf/raspi/pySejf.py

617 lines
22 KiB
Python

#!/usr/bin/python
import os
os.chdir(os.path.dirname(os.path.realpath(__file__)))
os.environ['SDL_VIDEO_CENTERED'] = '1'
import serial
import webapp
for c in ['/dev/ttyUSB0', '/dev/ttyACM0']:
try:
ser = serial.Serial(c, 115200)
except:
continue
print('Using ', c)
webapp.init(ser)
import random, math, pygame
from pygame.locals import *
from virtualKeyboard import VirtualKeyboard
import threading
from math import cos
import time
import RPi.GPIO as GPIO
pygame.init()
#gameplay
#wynik w snake, ktory wygrywa
SNAKE_SCORE = 2
#predkosc gry w snake - w zakresie od 1 do 10
TICK_SNAKE = 10
#ilosc segmentow dodanych do snake po zlapaniu jablka
SNAKE_ADDSIZE = 10
#hasla klawiatury ekranowej dla obu zagadek
password1 = 'one'
password2 = 'two'
#ilosc powtorzen mp3
MP3_LOOPS = 4
#layout
#font i komunikat do snake
FONT_SNAKE = pygame.font.SysFont("Courier", 28)
MSG_SNAKE = 'Koniec gry! Twoj wynik: '
#font klawiatury
FONT_KEYBOARD = pygame.font.SysFont("Purisa", 28)
#kolor klawiatury w RGB
COLOR_KEYBOARD = [0, 0, 127]
#kolor tekstu na klawiaturze (na klawiszach, guzikach i wpisywanego hasla) w RGB
COLOR_KEYBOARD_TEXT = (0, 255, 0)
#tekst na przyciskach klawiatury
MSG_BTN1 = "Wpisz ja"
MSG_BTN2 = "Kasuj"
#grafika - jablko do snake, segmenty snake
IMG_APPLE = pygame.image.load("media/apple.gif")
IMG_SNAKEBODY = pygame.image.load("media/snake.gif")
#obrazki poszczegolnych slajdow
IMG_HELLO = pygame.image.load('media/hello_background.png')
IMG_KEY = pygame.image.load('media/key_background.gif')
IMG_WIN = pygame.image.load('media/congratulation_background.gif')
IMG_POT = pygame.image.load('media/pot_background.png')
IMG_FOAMBTN = pygame.image.load('media/foambtn_background.png')
IMG_KEYPAD = pygame.image.load('media/keypad_background.png')
IMG_SNAKE = pygame.image.load('media/snake_background.png')
IMG_MAZE = pygame.image.load('media/maze_background.png')
IMG_MP3 = pygame.image.load('media/sound_background.jpg')
IMG_KEYBOARD = pygame.image.load('media/keyboard_background.png')
IMG_DYNAMO = pygame.image.load('media/dynamo_background.jpg')
wall_a = 0 #czy to RPi jest na scianie A?
#pinologia
#przypisania pinow - fizyczne przyciski
PIN_BTNY = 17
PIN_BTNG = 27
#przypisania pinow - piny z RPi do Ardu zapalajace diode
PIN_SUC4 = 2
PIN_SUC9 = 3
PIN_SUC10 = 4
#przypisania pinow - piny z Ardu do Rpi wywolujace slajd
PIN_SET1 = 25
PIN_SET2 = 8
PIN_SET3 = 7
PIN_SET5 = 12
PIN_SET6 = 16
PIN_SET_WIN = 20
#przypisania pinow - zworka do GND na tym pinie ustawia, ze ta Pi jest na scianie A (uruchamia dzwiek i obrotowa lampe)
PIN_WALL_A = 26
#przypisania pinow - na tym pinie jest przekaznik wlaczajacy lampe obrotowa i otwierajacy szuflade #5
PIN_RELAY = 21
#ustawianie stanow i trybow pracy pinow
GPIO.setmode(GPIO.BCM) # set up BCM GPIO numbering
GPIO.setup(PIN_BTNY, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_BTNG, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_SET1, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_SET2, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_SET3, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_SET5, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_SET6, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
GPIO.setup(PIN_WALL_A, GPIO.IN, pull_up_down=GPIO.PUD_UP) # set GPIO2 as input (button)
if GPIO.input(PIN_WALL_A) == 0:
print "this RPi is on wall A."
wall_a = 1
if(wall_a):
GPIO.setup(PIN_SET_WIN, GPIO.IN, pull_up_down=GPIO.PUD_UP) # ten pin jest dzielony miedzy wszystkie rpi i arduina - pullup robi tylko RPi A
else:
GPIO.setup(PIN_SET_WIN, GPIO.IN)
print "this RPi is."
GPIO.setup(PIN_SUC4, GPIO.OUT)
GPIO.output(PIN_SUC4, 1)
GPIO.setup(PIN_SUC9, GPIO.OUT)
GPIO.setup(PIN_SUC10, GPIO.OUT)
GPIO.setup(PIN_RELAY, GPIO.OUT)
GPIO.output(PIN_SUC9, 1)
GPIO.output(PIN_SUC10, 1)
GPIO.output(PIN_RELAY, 1)
WINSIZEX, WINSIZEY = 900, 680
counter = 0
timeSec = 0
UP = 1
DOWN = 3
RIGHT = 2
LEFT = 4
lastJoyState = RIGHT
score = 0
dynamoValSPI = [0,0,0]
#numery slajdow
SLIDE_START = 0
SLIDE_KEY = 1
SLIDE_POT = 2
SLIDE_FOAMBTN = 3
SLIDE_SNAKE = 4
SLIDE_DYNAMO = 5
SLIDE_KEYPAD = 6
SLIDE_MAZE = 7
SLIDE_MP3 = 8
SLIDE_KEYBOARD = 9
SLIDE_CONGRATS = 11
SLIDE_SNAKE_DEAD = 10 #to nie slajd, tylko wartosc potrzebna do resetowania snake po przegranej
global slide
#slajd poczatkowy
slide = SLIDE_START
slides = [SLIDE_START,SLIDE_KEY,SLIDE_POT,SLIDE_FOAMBTN,SLIDE_SNAKE,SLIDE_KEYPAD,SLIDE_MAZE,SLIDE_MP3,SLIDE_KEYBOARD,SLIDE_CONGRATS]
def win():
print "GAME OVER - YOU WIN"
GPIO.output(PIN_RELAY, 0)
def signalWin(success):
pin = 0
if success == 4:
pin = PIN_SUC4
elif success == 9:
pin = PIN_SUC9
elif success == 10:
pin = PIN_SUC10
print("signalling win for task: ")
print(success)
print("using GPIO pin: ")
print(pin)
GPIO.output(pin, 0)
pygame.time.wait(200)
GPIO.output(pin, 1)
print("done signalling.")
def waitForEvents():
global slide
print("waiting for events")
while 1:
time.sleep(0.01)
if GPIO.input(PIN_SET1) == 0 and slide != SLIDE_KEY:
print "Ardu said go to key slide"
slide = SLIDE_KEY
return
if GPIO.input(PIN_SET2) == 0 and slide != SLIDE_POT:
print "Ardu said go to pot slide"
slide = SLIDE_POT
return
if GPIO.input(PIN_SET3) == 0 and slide != SLIDE_FOAMBTN:
print "Ardu said go to foam button slide"
slide = SLIDE_FOAMBTN
return
if GPIO.input(PIN_SET5) == 0 and slide != SLIDE_DYNAMO:
print "Ardu said go to dynamo slide"
slide = SLIDE_DYNAMO
return
if GPIO.input(PIN_SET6) == 0 and slide != SLIDE_KEYPAD:
print "Ardu said go to keypad slide"
slide = SLIDE_KEYPAD
return
if GPIO.input(PIN_SET_WIN) == 0 and slide != SLIDE_CONGRATS:
print "Ardu said win"
slide = SLIDE_CONGRATS
return
if GPIO.input(PIN_BTNY) == 0 and slide != SLIDE_MAZE:
print "yellow button pressed - go to maze slide"
slide = SLIDE_MAZE
return
if GPIO.input(PIN_BTNG) == 0 and slide != SLIDE_MP3:
print "green button pressed - go to mp3 slide"
slide = SLIDE_MP3
return
for event in pygame.event.get():
if event.type == pygame.MOUSEBUTTONDOWN:
slide = SLIDE_KEYBOARD
print("mouse event")
return
elif event.type == pygame.KEYDOWN:
if slide != SLIDE_SNAKE:
slide = SLIDE_SNAKE
else:
slide = SLIDE_SNAKE_DEAD
print("keydown event")
return
def checkEvents():
global slide
if GPIO.input(PIN_SET1) == 0 and slide != SLIDE_KEY:
print "Ardu said go to key slide"
slide = SLIDE_KEY
return
if GPIO.input(PIN_SET2) == 0 and slide != SLIDE_POT:
print "Ardu said go to pot slide"
slide = SLIDE_POT
return
if GPIO.input(PIN_SET3) == 0 and slide != SLIDE_FOAMBTN:
print "Ardu said go to foam button slide"
slide = SLIDE_FOAMBTN
return
if GPIO.input(PIN_SET5) == 0 and slide != SLIDE_DYNAMO:
print "Ardu said go to dynamo slide"
slide = SLIDE_DYNAMO
return
if GPIO.input(PIN_SET6) == 0 and slide != SLIDE_KEYPAD:
print "Ardu said go to keypad slide"
slide = SLIDE_KEYPAD
return
if GPIO.input(PIN_SET_WIN) == 0 and slide != SLIDE_CONGRATS:
print "Ardu said win"
slide = SLIDE_KEYPAD
return
if GPIO.input(PIN_BTNY) == 0 and slide != SLIDE_MAZE:
print "yellow button pressed - go to maze slide"
slide = SLIDE_MAZE
return
if GPIO.input(PIN_BTNG) == 0 and slide != SLIDE_MP3:
print "green button pressed - go to mp3 slide"
slide = SLIDE_MP3
return
for event in pygame.event.get():
if event.type == pygame.MOUSEBUTTONDOWN and slide != SLIDE_KEYBOARD:
slide = SLIDE_KEYBOARD
print("mouse event")
return 1
elif event.type == pygame.KEYDOWN:
slide = SLIDE_SNAKE
print("keydown event")
return
else: return 0
def drawSlide(n):
print("drawing slide:")
print(n)
screen.fill(WHITE)
screen.blit(slides[n], (0, 0))
pygame.display.flip()
waitForEvents()
def collide(x1, x2, y1, y2, w1, w2, h1, h2):
if x1+w1>x2 and x1<x2+w2 and y1+h1>y2 and y1<y2+h2:return True
else:return False
def die(screen, score):
f=FONT_SNAKE
t=f.render(MSG_SNAKE+str(score), True, (0, 0, 0))
if(score >= SNAKE_SCORE):
signalWin(4)
screen.blit(t, (10, 270))
pygame.display.update()
pygame.time.wait(2000)
waitForEvents()
def main():
global slide
startonce = 0
showstartscreen = 0
while startonce == 0:
startonce = 0
######## CONSTANTS
WINSIZE = [WINSIZEX,WINSIZEY]
BLOCKSIZE = [18,18]
UP = 1
DOWN = 3
RIGHT = 2
LEFT = 4
WHITE = [255, 255, 255]
MINX = 100
MAXX = WINSIZEX - MINX
MINY = 100
MAXY = WINSIZEY - MINY
SNAKESTEP = 20
TRUE = 1
FALSE = 0
######## VARIABLES
direction = RIGHT # 1=up,2=right,3=down50,4=left
snakexy = [300,400]
snakelist = [[300,400],[280,400],[260,400]]
counter = 0
score = 0
appleonscreen = 0
applexy = [0,0]
newdirection = RIGHT
snakedead = FALSE
gameregulator = 6
gamepaused = 0
growsnake = 0 # added to grow tail by two each time
snakegrowunit = 2 # added to grow tail by two each time
clock = pygame.time.Clock()
screen = pygame.display.set_mode(WINSIZE, pygame.NOFRAME)#FULLSCREEN)
#screen = pygame.display.set_mode((0, 0), pygame.FULLSCREEN)
pygame.display.set_caption('PozySejf')
pygame.FULLSCREEN
#screen = display.set_mode([1024,768], FULLSCREEN)
#screen.fill(BLACK)
dynamoValSPI = [0,0,0]
dynamoRectY = [50, 50, 50]
dynamoRectX = 50
dynamoXpos = [-100, 0, 100]
distanceDynamoRect = 300
textInputDotNumber = 0
pushButtonNumber = 100
valueSound = 100
passwordLetter = ''
#### show initial start screen
### every while loop for each slide
while slide == SLIDE_START:
print("showing slide 0")
screen.fill(WHITE)
screen.blit(IMG_HELLO, (0, 0))
pygame.display.flip()
pygame.time.wait(5000)
waitForEvents()
break
while slide == SLIDE_KEY:
print("showing slide 1 - key")
screen.fill(WHITE)
screen.blit(IMG_KEY, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_POT:
print("showing slide 2 - pot")
screen.fill(WHITE)
screen.blit(IMG_POT, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_FOAMBTN:
print("showing slide 3 - foam button")
screen.fill(WHITE)
screen.blit(IMG_FOAMBTN, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_SNAKE: ## snake
print("snake is starting...")
xs = [290, 290, 290, 290, 290];
ys = [290, 270, 250, 230, 210];
dirs = 0
score = 0
applepos = (random.randint(0, WINSIZEX), random.randint(0, WINSIZEY))
pygame.init()
#s=pygame.display.set_mode((600, 600))
appleimage = IMG_APPLE
img = IMG_SNAKEBODY#pygame.Surface((20, 20))
#img.fill((255, 0, 0))
f = FONT_SNAKE
clock = pygame.time.Clock()
while slide == SLIDE_SNAKE:
clock.tick(TICK_SNAKE)
if GPIO.input(PIN_SET1) == 0 and slide != SLIDE_KEY:
print "Ardu said go to key slide"
slide = SLIDE_KEY
break
if GPIO.input(PIN_SET2) == 0 and slide != SLIDE_POT:
print "Ardu said go to pot slide"
slide = SLIDE_POT
break
if GPIO.input(PIN_SET3) == 0 and slide != SLIDE_FOAMBTN:
print "Ardu said go to foam button slide"
slide = SLIDE_FOAMBTN
break
if GPIO.input(PIN_SET5) == 0 and slide != SLIDE_DYNAMO:
print "Ardu said go to dynamo slide"
slide = SLIDE_DYNAMO
break
if GPIO.input(PIN_SET6) == 0 and slide != SLIDE_KEYPAD:
print "Ardu said go to keypad slide"
slide = SLIDE_KEYPAD
break
if GPIO.input(PIN_SET_WIN) == 0 and slide != SLIDE_CONGRATS:
print "Ardu said win"
slide = SLIDE_KEYPAD
break
if GPIO.input(PIN_BTNY) == 0:
print "yellow button pressed - go to maze slide"
slide = SLIDE_MAZE
break
if GPIO.input(PIN_BTNG) == 0:
print "green button pressed - go to mp3 slide"
slide = SLIDE_MP3
break
for e in pygame.event.get():
#if e.type == QUIT:
# sys.exit(0) #quit - nie chcemy miec mozliwosci wyjscia z aplikacji
if e.type == KEYDOWN:
if e.key == K_UP and dirs != 0:dirs = 2
elif e.key == K_DOWN and dirs != 2:dirs = 0
elif e.key == K_LEFT and dirs != 1:dirs = 3
elif e.key == K_RIGHT and dirs != 3:dirs = 1
elif e.type == pygame.MOUSEBUTTONDOWN:
slide = SLIDE_KEYBOARD
print("mouse event")
break
i = len(xs)-1
while i >= 2:
if collide(xs[0], xs[i], ys[0], ys[i], 20, 20, 20, 20):
die(screen, score)
score = 0
break
i-= 1
if collide(xs[0], applepos[0], ys[0], applepos[1], 20, 10, 20, 10):
score+=1
for i in range(0, SNAKE_ADDSIZE):
xs.append(700)
ys.append(700)
applepos=(random.randint(0,WINSIZEX),random.randint(0,WINSIZEY))
if xs[0] < 0 or xs[0] > WINSIZEX or ys[0] < 0 or ys[0] > WINSIZEY:
die(screen, score)
score = 0
break
i = len(xs)-1
while i >= 1:
xs[i] = xs[i-1];ys[i] = ys[i-1];i -= 1
if dirs==0:ys[0] += 20
elif dirs==1:xs[0] += 20
elif dirs==2:ys[0] -= 20
elif dirs==3:xs[0] -= 20
#rysuj tlo
screen.fill(WHITE)
screen.blit(IMG_SNAKE, (0, 0))
print("drawing snake...")
if(score > SNAKE_SCORE):
die(screen, score)
break
for i in range(0, len(xs)):
screen.blit(img, (xs[i], ys[i]))
screen.blit(appleimage, applepos);t=f.render(str(score), True, (0, 0, 0));screen.blit(t, (10, 10));pygame.display.update()
while slide == SLIDE_SNAKE_DEAD:
print("resetting snake.")
slide = SLIDE_SNAKE
while slide == SLIDE_DYNAMO:
print("showing slide 5 - dynamo")
screen.fill(WHITE)
screen.blit(IMG_DYNAMO, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_KEYPAD:
print("showing slide 6 - drawer keypad")
screen.fill(WHITE)
screen.blit(IMG_KEYPAD, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_MAZE:
print("showing slide 7 - maze")
screen.fill(WHITE)
screen.blit(IMG_MAZE, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_MP3:
print("showing slide 8 - mp3 player")
screen.fill(WHITE)
screen.blit(IMG_MP3, (0, 0))
pygame.display.flip()
waitForEvents()
break
while slide == SLIDE_KEYBOARD: ## keyboard
#print("checking events")
checkEvents()
#render backdrop
screen.fill(WHITE)
screen.blit(IMG_KEYBOARD, (0, 0))
font = FONT_KEYBOARD
#render keys
height1 = 480-40
height2 = 560-40
height3 = 640-40
letterMeaning = ['q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p', 'a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l', 'z', 'x', 'c', 'v', 'b', 'n', 'm', ' ']
letterQM = []
for i in range(10):
letterQM.append(pygame.draw.rect(screen,COLOR_KEYBOARD,Rect([i*80 +120,height1],[70 , 70])))
tt = font.render(letterMeaning[i], True, COLOR_KEYBOARD_TEXT)
#tt = font.render(str(i), True, (0, 0, 0))
screen.blit(tt, (i*80 +120+20,height1+15))
for j in range(9):
letterQM.append(pygame.draw.rect(screen,COLOR_KEYBOARD,Rect([j*80 +160,height2],[70 , 70])))
tt = font.render(letterMeaning[j+10], True, COLOR_KEYBOARD_TEXT)
#tt = font.render(str(j+10), True, (0, 0, 0))
screen.blit(tt, (j*80 +160+20,height2+15))
for k in range(8):
letterQM.append(pygame.draw.rect(screen,COLOR_KEYBOARD,Rect([k*80 +200,height3],[70 , 70])))
tt = font.render(letterMeaning[k+19], True, COLOR_KEYBOARD_TEXT)
#tt = font.render(str(k+19), True, (0, 0, 0))
screen.blit(tt, (k*80 +200+20,height3+15))
#render buttons
enterButton = pygame.draw.rect(screen,COLOR_KEYBOARD,Rect([WINSIZEX/2 - 100 - 70,300],[140 , 70]))
tt = font.render(MSG_BTN1, True, COLOR_KEYBOARD_TEXT)
screen.blit(tt, [WINSIZEX/2 - 100 - 70,300])
clearButton = pygame.draw.rect(screen,COLOR_KEYBOARD,Rect([WINSIZEX/2 + 100 - 70,300],[140 , 70]))
tt = font.render(MSG_BTN2, True, COLOR_KEYBOARD_TEXT)
screen.blit(tt, [WINSIZEX/2 + 100 - 70,300])
#render password as it is typed
tt = font.render(passwordLetter, True, COLOR_KEYBOARD_TEXT)
screen.blit(tt, [WINSIZEX/2 - 70,380])
#interactive features
for event in pygame.event.get():
#if event.type == pygame.QUIT:
# exit()
if event.type == pygame.MOUSEBUTTONDOWN:
# Set the x, y postions of the mouse click
x, y = event.pos
if enterButton.collidepoint(x, y):
if passwordLetter == password1:
print 'password correct for task 9.'
signalWin(9)
if passwordLetter == password2:
print 'password correct for task 10.'
signalWin(10)
passwordLetter = ''
if clearButton.collidepoint(x, y):
passwordLetter = ''
for i in range(10+9+8):
if letterQM[i].collidepoint(x, y):
passwordLetter = passwordLetter + letterMeaning[i]
print passwordLetter
pygame.display.flip()
while slide == SLIDE_CONGRATS: ## congratulation
print("showing slide 10 - win")
screen.fill(WHITE)
screen.blit(IMG_WIN, (0, 0))
pygame.display.flip()
if(wall_a):
win()
pygame.mixer.init()
pygame.mixer.music.load("media/win.mp3")
pygame.mixer.music.play(MP3_LOOPS)
pygame.mixer.music.set_volume(1.0)
while pygame.mixer.music.get_busy() == True:
continue
while 1:
print "you win."
waitForEvents()
if __name__ == '__main__':
main()