astropi/menu_nojoystick.py at master · TechnologyClubOfThrace/astropi · GitHub

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
#!/usr/bin/python
from sense_hat import SenseHat
import os
import time
import RPi.GPIO as GPIO
import subprocess

homedir = "/home/pi/astropi/"

r = (255, 0, 0)
g = (0, 255, 0)
b = (0, 255, 255)
y = (255, 255, 0)

menu = [
  ("Clock",       b, "/usr/bin/python", "clock/clock.py"),
  ("Colour",      b, "/usr/bin/python", "sensehat/colour_cycle.py"),

  ("Maze",        y, "/usr/bin/python", "marble-maze/marble-maze.py"),
  ("Dice",        g, "/usr/bin/python", "dice/dice.py"),
  ("Ball",        y, "/usr/bin/python", "gravity/gravity_ball.py"),

  ("Rainbow",     b, "/usr/bin/python", "sensehat/rainbow.py"),

  ("Conway",      g, "/usr/bin/python", "sensehat2/conway.py"),
  ("Sparkles",    b, "/usr/bin/python", "sensehat2/random_sparkles.py"),
  ("Temp",        b, "/usr/bin/python", "sensehat2/temperature.py"),
  ("Humidity",    b, "/usr/bin/python", "sensehat2/humidity.py"),

  ("Text",        b, "/usr/bin/python", "sensehat/text_scroll.py"),

#  ("Egg",         g, "/usr/bin/python", "egg/eggdrop.py"),
#  ("Compass",     b, "/usr/bin/python", "examples/compass.py"),

  ("IPAddr",      r, "/usr/bin/python", "ipaddr/displayip.py"),
  ("Poweroff",    r, "poweroff", "")
]

current_proc =  None

GPIO.setmode(GPIO.BCM)
GPIO.setup(26, GPIO.IN, pull_up_down=GPIO.PUD_UP) #UP
GPIO.setup(13, GPIO.IN, pull_up_down=GPIO.PUD_UP) #DOWN
GPIO.setup(20, GPIO.IN, pull_up_down=GPIO.PUD_UP) #LEFT
GPIO.setup(19, GPIO.IN, pull_up_down=GPIO.PUD_UP) #RIGHT
GPIO.setup(16, GPIO.IN, pull_up_down=GPIO.PUD_UP) #2LEFT
GPIO.setup(21, GPIO.IN, pull_up_down=GPIO.PUD_UP) #2RIGHT

time.sleep(1)

sense = SenseHat()
sense.clear()  # Blank the LED matrix
sense.set_rotation(270)  # Flight orientation

def print_menu(choice):
    sense.show_message(menu[choice][0], scroll_speed=0.05, text_colour=menu[choice][1])


def run_process(choice):
    global current_proc
    if (current_proc != None):
       current_proc.kill()

    cmdline = menu[choice][3]
    if cmdline != "":
        cmdline = homedir + cmdline

    print "executing " + menu[choice][2] + " " + cmdline

    if (choice == len(menu) -1):
        current_proc = subprocess.Popen( "poweroff" )
    else:
        current_proc = subprocess.Popen( [menu[choice][2], cmdline])

    print current_proc

def stop_process():
    global current_proc

    if (current_proc != None):
       print "killing process"
       print current_proc
       current_proc.kill()
    current_proc = None

    print_menu(current)

running = True

current = 0
print_menu(current)


while running:
    if GPIO.input(26) == 0:
        print ("button 26 - UP")
        current = current-1
        if current<0:
            current = len(menu)-1
        print_menu(current)

    if GPIO.input(13) == 0:
        print ("button 13 - DOWN")
        current=current+1
        if current>len(menu)-1:
            current = 0
        print_menu(current)

    if GPIO.input(19) == 0:
        print ("button 19")

    if GPIO.input(20) == 0:
        print ("button 20")

    if GPIO.input(16) == 0:
        print ("button 16 - right click")
        stop_process()

    if GPIO.input(21) == 0:
        print ("button 21 - left click")
        run_process(current)

    time.sleep(0.2)