soundboard-host/host.py

#!/usr/bin/env python3

import sys
import os
import time
import random
import socket
import pygame

from pygame.locals import *

os.environ["SDL_VIDEODRIVER"] = "dummy"

# "we'll fix it in post!"
MAPPING = [
    8,
    7,
    6,
    5,
    1,
    2,
    3,
    4,
    9,
    10,
    11,
    0
]

SPATH = "sounds/"

def logmesohard(name):
    s = socket.socket()
    s.connect(('10.8.1.1', 2003))
    s.sendall('hackerspace.soundboard.{} 1'.format(name.replace('.', '_'_)))
    s.close()

class SimpleSound:
    def __init__(self, path):
        self.path = path
    def create(self):
        print("Loading {}".format(self.path))
        self.s = pygame.mixer.Sound(SPATH + self.path)
    def play(self):
        print("Playing {}".format(self.path))
        logmesohard(self.path)
        self.s.play()
        return self.s.get_length() * 0.5

class RandomSound:
    def __init__(self, paths):
        self.paths = paths
    def create(self):
        print("Loading {}".format(", ".join(self.paths)))
        self.sounds = [(pygame.mixer.Sound(SPATH + p), p) for p in self.paths]
    def play(self):
        s, path = random.choice(self.sounds)
        logmesohard(self.paths[0])
        print("Playing {}".format(path))
        print(s)
        s.play()
        return s.get_length() * 0.5

SOUNDS = [
    SimpleSound("nope.wav"),
    SimpleSound("yesyes.wav"),
    SimpleSound("ba-dum-tish.wav"),
    SimpleSound("crickets.wav"),
    SimpleSound("wind.wav"),
    SimpleSound("trombone.wav"),
    RandomSound(["whip.wav", "whip2.wav", "whip4.wav"]),
    SimpleSound("finishim.wav"),
    SimpleSound("wtfboom.wav"),
    SimpleSound("noooooooo.wav"),
    SimpleSound("fanfare.wav"),
    SimpleSound("dundundun.wav"),
]

play_again_at = time.time()


if __name__ == "__main__":
    # are we a fucking rpi?
    f = open("/proc/cpuinfo", "r")
    d = f.read()
    f.close()
    if "BCM2708" in d:
        print("We are on an rpi. Doing ALSA hack.")
        os.system("amixer cset numid=3 1")

    pygame.mixer.init()
    pygame.display.init()
    pygame.display.set_mode((1,1))
    for s in SOUNDS:
        s.create()
    pygame.joystick.init()
    j = pygame.joystick.Joystick(0)
    j.init()
    print("Ready!")

    while True:
        pygame.event.pump()
        for event in pygame.event.get():
            if event.type == JOYBUTTONDOWN:
                n = MAPPING.index(event.button)
                if n < len(SOUNDS) and time.time() >= play_again_at:
                   play_again_at = time.time() + SOUNDS[n].play()

        time.sleep(0.1)