1 files changed, 90 insertions, 0 deletions
diff --git a/broadcast.py b/broadcast.py
new file mode 100644
index 0000000..0483a67
--- /dev/null
+++ b/broadcast.py
@@ -0,0 +1,90 @@
+import socket
+import sys
+import struct
+import time
+import xml.etree.ElementTree as ET
+import threading
+
+PORT = 13676
+if len(sys.argv) > 2:
+	factor = float(sys.argv[2])
+else:
+	factor = 1
+
+print 'Factor:', factor
+
+class Packet(object):
+	def __init__(self, cmd, *data):
+		self.cmd = cmd
+		self.data = data
+		if len(data) >= 8:
+			raise ValueError('Too many data')
+		self.data = list(self.data) + [0] * (8-len(self.data))
+	def __str__(self):
+		return struct.pack('>L'+('L'*len(self.data)), self.cmd, *self.data)
+
+class CMD:
+	KA = 0 # No important data
+	PING = 1 # Data are echoed exactly
+	QUIT = 2 # No important data
+	PLAY = 3 # seconds, microseconds, frequency (Hz), amplitude (0-255)
+
+s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+
+clients = []
+
+s.sendto(str(Packet(CMD.PING)), ('255.255.255.255', PORT))
+s.settimeout(0.5)
+
+try:
+	while True:
+		data, src = s.recvfrom(4096)
+		clients.append(src)
+except socket.timeout:
+	pass
+
+print 'Clients:'
+for cl in clients:
+	print cl
+	if sys.argv[1] == '-t':
+		s.sendto(str(Packet(CMD.PLAY, 0, 250000, 440, 255)), cl)
+		time.sleep(0.25)
+		s.sendto(str(Packet(CMD.PLAY, 0, 250000, 880, 255)), cl)
+	if sys.argv[1] == '-q':
+		s.sendto(str(Packet(CMD.QUIT)), cl)
+
+try:
+	iv = ET.parse(sys.argv[1]).getroot()
+except IOError:
+	print 'Bad file'
+	exit()
+
+notestreams = iv.findall("./streams/stream[@type='ns']")
+
+class NSThread(threading.Thread):
+	def run(self):
+		nsq, cl = self._Thread__args
+		for note in nsq:
+			ttime = float(note.get('time'))
+			pitch = int(note.get('pitch'))
+			vel = int(note.get('vel'))
+			dur = factor*float(note.get('dur'))
+			while time.time() - BASETIME < factor*ttime:
+				time.sleep(factor*ttime - (time.time() - BASETIME))
+			s.sendto(str(Packet(CMD.PLAY, int(dur), int((dur*1000000)%1000000), int(440.0 * 2**((pitch-69)/12.0)), vel*2)), cl)
+			time.sleep(dur)
+
+threads = []
+for ns in notestreams:
+	if not clients:
+		print 'WARNING: Out of clients!'
+		break
+	nsq = ns.findall('note')
+	threads.append(NSThread(args=(nsq, clients.pop(0))))
+
+BASETIME = time.time()
+for thr in threads:
+	thr.start()
+for thr in threads:
+	thr.join()