fixed audioinit

Makefile

+servo_control: servo_control.c
+	$(CC) -o $@ $^ -lwiringPi -lpthread
+	sudo chown root $@
+	sudo chmod u+x $@
+	sudo chown root $@

ksp_client.py

+import krpc
+import zmq
+import argparse
+
+
+class ModelRocket:
+    socket = None
+    realRocket = None 
+    topic = "events"
+    states = { -1:"abort", 0:"pre_launch",1:"liftoff", 2:"stage_1_separation", 3:"stage_2_separation"}
+    stage_map = {0:0,1:1,2:2,3:3}
+
+    def send(self,stage):
+        self.socket.send("%d %d" % (self.topic, self.stage_map[stage]))
+
+    def __init__(self,addr="localhost:5556"):
+        context = zmq.Context()
+        self.socket = context.socket(zmq.PUB)
+        self.socket.bind("tcp://%s" % addr)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='')
+    parser.add_argument('addr', type=str, default="localhost:5556", nargs="?")
+    args = parser.parse_args()
+
+    conn = krpc.connect()
+    vessel = conn.space_center.active_vessel
+    control = vessel.control
+    rocket = ModelRocket(args.addr)
+
+    stage = conn.add_stream(getattr, control, 'current_stage')
+    stage.add_callback(rocket.send)
+    stage.start()
+
+    while 42:
+        pass

lighting.py

 #!/usr/bin/env python3
 
+from time import sleep
+from random import randint
+import configparser
+
 import zmq
-import sys
+import spidev
+import ws2812
 
-import OPi.GPIO as GPIO
-from neopixel import *
-from random import randint
 
-GPIO.setmode(GPIO.BOARD)
-GPIO.setup(7, GPIO.OUT)
+def get_color():
+    rand = randint(0, 10)
+    if rand < 5:
+        return (248, 255, 255)
+    elif rand == 6:
+        return (111, 255, 0)
+    elif rand == 7:
+        return (234, 255, 0)
+    elif rand == 8:
+        return (87, 255, 34)
+    else:
+        return (0, 213, 0)
 
-# Create NeoPixel object with appropriate configuration.
-strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT)
-# Intialize the library (must be called once before other functions).
-strip.begin()
 
-# LED strip configuration:
-LED_COUNT   = 16      # Number of LED pixels.
-LED_PIN     = 18      # GPIO pin connected to the pixels (must support PWM!).
-LED_FREQ_HZ = 800000  # LED signal frequency in hertz (usually 800khz)
-LED_BRIGHTNESS = 255     # Set to 0 for darkest and 255 for brightest
-LED_DMA     = 5       # DMA channel to use for generating signal (try 5)
-LED_INVERT  = False   # True to invert the signal (when using NPN transistor level shift)
+def light_flame():
+    spi = spidev.SpiDev()
+    spi.open(1, 0)
 
-port = "5556"
-if len(sys.argv) > 1:
-    port = sys.argv[1]
-    int(port)
+    while True:
+        ws2812.write2812(spi, [get_color() * 4])
+        sleep(0.2)
 
-# Socket to talk to server
-context = zmq.Context()
-socket = context.socket(zmq.SUB)
 
-print("Connecting to coordinator...")
-socket.connect(f"tcp://localhost:{port}")
+def main():
+    config = configparser.ConfigParser({'host': '10.10.10.1', 'port': '5556'})
+    config.read('config.cfg')
 
-topicfilter = "events"
-socket.setsockopt(zmq.SUBSCRIBE, topicfilter)
+    port = config['port']
+    host = config['host']
 
+    # Socket to talk to server
+    context = zmq.Context()
+    socket = context.socket(zmq.SUB)
 
-while True:
-    string = socket.recv()
-    topic, messagedata = string.split()
-    print(topic, messagedata)
+    print("Connecting to coordinator...")
+    socket.connect(f"tcp://{host}:{port}")
 
-    if messagedata == "liftoff":
-        light_flame()
-    elif messagedata == "stage_1_separation":
-        light_flame()
-    else:
-        print("Unhandled message")
-
-def light_flame(): 
-    for i in range(0,strip.numPixels(),1):
-        rand = randint(0,10)
-        if rand<5:
-            color = (255, 248, 255)
-        else if rand==6:
-            color = (255, 111, 0)
-        else if rand==7:
-            color = (255, 234, 0)
-        else if rand==8:
-            color = (255, 87, 34)
-        else :
-            color = (213, 0, 0)
-        strip.setPixelColor(i, color)
-    strip.show()
+    topicfilter = "events"
+    socket.setsockopt(zmq.SUBSCRIBE, topicfilter)
+
+    while True:
+        string = socket.recv()
+        topic, messagedata = string.split()
+        print(topic, messagedata)
+
+        if messagedata == "liftoff":
+            light_flame()
+        elif messagedata == "stage_1_separation":
+            light_flame()
+        else:
+            print("Unhandled message")
+
+
+if __name__ == "__main__":
+    main()

servo.py

 #!/usr/bin/env python3
 
 import zmq
-import sys
+import subprocess
+import configparser
 
-import actions
 
-port = "5556"
-if len(sys.argv) > 1:
-    port = sys.argv[1]
-    int(port)
+def prime_stages():
+    subprocess.Popen('./servo_control', '1', '1')
+    subprocess.Popen('./servo_control', '7', '1')
 
-# Socket to talk to server
-context = zmq.Context()
-socket = context.socket(zmq.SUB)
 
-print("Connecting to coordinator...")
-socket.connect(f"tcp://localhost:{port}")
+def separate_stage_one():
+    subprocess.Popen('./servo_control', '1', '0')
 
-topicfilter = "events"
-socket.setsockopt(zmq.SUBSCRIBE, topicfilter)
 
-while True:
-    string = socket.recv()
-    topic, messagedata = string.split()
+def separate_stage_two():
+    subprocess.Popen('./servo_control', '7', '0')
 
-    if messagedata == "pre_launch":
-        actions.pre_launch()
-    elif messagedata == "liftoff":
-        actions.liftoff()
-    elif messagedata == "stage_1_separation":
-        actions.stage_1_separation()
-    elif messagedata == "stage_2_separation":
-        actions.stage_2_separation()
-    else:
-        print("Unhandled message")
 
-    print(topic, messagedata)
+def main():
+    config = configparser.ConfigParser({'host': '10.10.10.1', 'port': '5556'})
+    config.read('config.cfg')
+
+    port = config['port']
+    host = config['host']
+
+    # Socket to talk to server
+    context = zmq.Context()
+    socket = context.socket(zmq.SUB)
+
+    print("Connecting to coordinator...")
+    socket.connect(f"tcp://{host}:{port}")
+
+    topicfilter = "events"
+    socket.setsockopt(zmq.SUBSCRIBE, topicfilter)
+
+    prime_stages()
+
+    while True:
+        string = socket.recv()
+        topic, messagedata = string.split()
+
+        if messagedata == "stage_1_separation":
+            separate_stage_one()
+        elif messagedata == "stage_2_separation":
+            separate_stage_two()
+        else:
+            print("Unhandled message")
+
+        print(topic, messagedata)
+
+
+if __name__ == "__main__":
+    main()

sound.py

+#!/usr/bin/env python3
+
+import pygame as pg
+from threading import Timer
+import zmq
+import configparser
+
+
+# fade out
+def endMission():
+    pg.mixer.Channel(0).fadeout(5)
+    pass
+
+
+def main():
+    config = configparser.ConfigParser({'host': '10.10.10.1', 'port': '5556'})
+    config.read('config.cfg')
+
+    port = config['port']
+    host = config['host']
+
+    pg.mixer.init()
+    pg.init(44100, size=-16, channels=2, buffer=4096)
+    pg.mixer.set_num_channels(3)  # default is 8
+
+    # carico i file audio
+    lancio = pg.mixer.Sound('sounds/lancio.wav')
+    stageOne = pg.mixer.Sound('sounds/stageOne.wav')
+    stageTwo = pg.mixer.Sound('sounds/stageTwo.wav')
+
+    # tempo di fine suono dopo secondo modulo
+    t = Timer(30.0, endMission)
+
+    # Socket to talk to server
+    context = zmq.Context()
+    socket = context.socket(zmq.SUB)
+
+    print("Connecting to coordinator...")
+    socket.connect(f"tcp://{host}:{port}")
+
+    topicfilter = "events"
+    socket.setsockopt(zmq.SUBSCRIBE, topicfilter)
+
+    while True:
+        string = socket.recv()
+        topic, messagedata = string.split()
+
+        if messagedata == "pre_launch":
+            pg.mixer.Channel(0).play(lancio)
+        elif messagedata == "stage_1_separation":
+            pg.mixer.Channel(1).play(stageOne)
+        elif messagedata == "stage_2_separation":
+            pg.mixer.Channel(2).play(stageTwo)
+            t.start()
+        else:
+            print("Unhandled message")
+
+        print(topic, messagedata)
+
+
+if __name__ == "__main__":
+    main()