sitter3 for python3

ethertools/sitter3.py

@@ -0,0 +1,422 @@
+#!/usr/bin/python
+# j4cDAC "sitter"
+#
+# Copyright 2012 Jacob Potter
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# python 3.11 macos : brew install tcl-tk
+
+import socket
+import time
+import struct
+
+def pack_point(x, y, r, g, b, i = -1, u1 = 0, u2 = 0, flags = 0):
+	"""Pack some color values into a struct dac_point.
+
+	Values must be specified for x, y, r, g, and b. If a value is not
+	passed in for the other fields, i will default to max(r, g, b); the 
+	rest default to zero.
+	"""
+	
+	if i < 0:
+		i = max(r, g, b)
+
+	return struct.pack("<HhhHHHHHH", flags, x, y, i, r, g, b, u1, u2)
+
+
+class ProtocolError(Exception):
+	"""Exception used when a protocol error is detected."""
+	pass
+
+
+class Status(object):
+	"""Represents a status response from the DAC."""
+
+	def __init__(self, data):
+		"""Initialize from a chunk of data."""
+		if isinstance(data, str):
+			data = data.encode('utf-8')
+		self.protocol_version, self.le_state, self.playback_state, \
+		  self.source, self.le_flags, self.playback_flags, \
+		  self.source_flags, self.fullness, self.point_rate, \
+		  self.point_count = \
+			struct.unpack("<BBBBHHHHII", data)
+
+	def dump(self, prefix = " - "):
+		"""Dump to a string."""
+
+		lines = [
+			"Light engine: state %d, flags 0x%x" %
+				(self.le_state, self.le_flags),
+			"Playback: state %d, flags 0x%x" %
+				(self.playback_state, self.playback_flags),
+			"Buffer: %d points" %
+				(self.fullness, ),
+			"Playback: %d kpps, %d points played" %
+				(self.point_rate, self.point_count),
+			"Source: %d, flags 0x%x" %
+				(self.source, self.source_flags)
+		]
+		for l in lines:
+			print(prefix + l)
+
+
+class BroadcastPacket(object):
+	"""Represents a broadcast packet from the DAC."""
+
+	def __init__(self, st, ip=None):
+		"""Initialize from a chunk of data."""
+		self.mac = st[:6]
+		self.hw_rev, self.sw_rev, self.buffer_capacity, \
+		self.max_point_rate = struct.unpack("<HHHI", st[6:16])
+		self.status = Status(st[16:36])
+		self.ip = ip
+
+	def dump(self, prefix = " - "):
+		"""Dump to a string."""
+
+		lines = [
+			"MAC: " + ":".join(
+				"%02x" % (ord(o), ) for o in self.mac),
+			"HW %d, SW %d" %
+				(self.hw_rev, self.sw_rev),
+			"Capabilities: max %d points, %d kpps" %
+				(self.buffer_capacity, self.max_point_rate)
+		]
+		for l in lines:
+			print(prefix + l)
+		self.status.dump(prefix)
+
+	def macstr(self):
+		#print("macstr mac : ",self.mac, "ip ", self.ip)
+		return "".join("%02x" % (c, ) for c in self.mac)
+
+
+class DAC(object):
+	"""A connection to a DAC."""
+
+	def got_broadcast(self, bp):
+		self.last_broadcast = bp
+		self.last_broadcast_time = time.time()
+
+	def read(self, l):
+		"""Read exactly length bytes from the connection."""
+		while l > len(self.buf):
+			self.buf += self.conn.recv(4096).decode('utf-8')
+
+		obuf = self.buf
+		self.buf = obuf[l:]
+		return obuf[:l]
+
+	def readresp(self, cmd):
+		"""Read a response from the DAC."""
+
+		data = self.read(22)
+		response = data[0]
+		cmdR = data[1]
+		status = Status(data[2:])
+
+#       status.dump()
+
+		if cmdR != cmd:
+			raise ProtocolError("expected resp for %r, got %r"
+				% (cmd, cmdR))
+
+		if response != "a":
+			raise ProtocolError("expected ACK, got %r"
+				% (response, ))
+
+		self.last_status = status
+		return status
+
+	def __init__(self, macstr, bp):
+		self.macstr = macstr
+		self.firmware_string = "-"
+		self.got_broadcast(bp)
+
+		try:
+			t1 = time.time()
+			self.connect(self.last_broadcast.ip[0])
+			t = time.time() - t1
+			self.conn_status = "ok (%d ms)" % (t * 500)
+
+			if self.last_broadcast.sw_rev < 2:
+				self.firmware_string = "(old)"
+			else:
+				self.conn.sendall('v')
+				self.firmware_string = self.read(32).replace("\x00", " ").strip()
+		except (Exception, e):
+			print(e)
+			self.conn_status = str(e)
+	def connect(self, host, port = 7765):
+		"""Connect to the DAC over TCP."""
+
+		conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+		conn.settimeout(0.2)
+		conn.connect((host, port))
+		self.conn = conn
+		self.buf = ""
+
+		# Read the "hello" message
+		first_status = self.readresp("?")
+		first_status.dump()
+
+
+
+	def begin(self, lwm, rate):
+		cmd = struct.pack("<cHI", "b", lwm, rate)
+		self.conn.sendall(cmd)
+		return self.readresp("b")
+
+	def update(self, lwm, rate):
+		cmd = struct.pack("<cHI", "u", lwm, rate)
+		self.conn.sendall(cmd)
+		return self.readresp("u")
+
+	def encode_point(self, point):
+		return pack_point(*point)
+
+	def write(self, points):
+		epoints = list(map(self.encode_point, points))
+		cmd = struct.pack("<cH", "d", len(epoints))
+		self.conn.sendall(cmd + "".join(epoints))
+		return self.readresp("d")
+
+	def prepare(self):
+		self.conn.sendall("p")
+		return self.readresp("p")
+
+	def stop(self):
+		self.conn.sendall("s")
+		return self.readresp("s")
+
+	def estop(self):
+		self.conn.sendall("\xFF")
+		return self.readresp("\xFF")
+
+	def clear_estop(self):
+		self.conn.sendall("c")
+		return self.readresp("c")
+
+	def ping(self):
+		self.conn.sendall("?")
+		return self.readresp("?")
+
+	def play_stream(self, stream):
+		# First, prepare the stream
+		if self.last_status.playback_state == 2:
+			raise Exception("already playing?!")
+		elif self.last_status.playback_state == 0:
+			self.prepare()
+
+		started = 0
+
+		while True:
+			# How much room?
+			cap = 1799 - self.last_status.fullness
+			points = stream.read(cap)
+
+			if cap < 100:
+				time.sleep(0.005)
+				cap += 150
+
+#           print "Writing %d points" % (cap, )
+			t0 = time.time()
+			self.write(points)
+			t1 = time.time()
+#           print "Took %f" % (t1 - t0, )
+
+			if not started:
+				self.begin(0, 30000)
+				started = 1
+
+
+def find_dac():
+	"""Listen for broadcast packets."""
+
+	s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+	s.bind(("0.0.0.0", 7654))
+
+	while True:
+		data, addr = s.recvfrom(1024)
+		bp = BroadcastPacket(data)
+		
+		print("Packet from %s: " % (addr, ))
+		bp.dump()
+
+
+from tkinter import *
+import socket
+import queue
+import _thread
+import time
+
+class DacDisplay(LabelFrame):
+	def __init__(self, master):
+		LabelFrame.__init__(self, master, width=600, height=400)
+		self.grid_propagate(0)
+
+		Label(self, text="IP Address:").grid(row=0, column=0, sticky=N+E)
+		Label(self, text="Version:").grid(row=1, column=0, sticky=N+E)
+		Label(self, text="Status:").grid(row=2, column=0, sticky=N+E)
+		Label(self, text="Source:").grid(row=3, column=0, sticky=N+E)
+		Label(self, text="Network:").grid(row=4, column=0, sticky=N+E)
+		Label(self, text="Firmware:").grid(row=5, column=0, sticky=N+E)
+
+		self.iplabel = Label(self, text = "")
+		self.iplabel.grid(row=0, column=1, sticky=N+W)
+		self.verslabel = Label(self, text = "")
+		self.verslabel.grid(row=1, column=1, sticky=N+W)
+		self.stlabel = Label(self, text = "")
+		self.stlabel.grid(row=2, column=1, sticky=N+W) 
+		self.srclabel = Label(self, text = "")
+		self.srclabel.grid(row=3, column=1, sticky=N+W) 
+		self.netlabel = Label(self, text = "")
+		self.netlabel.grid(row=4, column=1, sticky=N+W) 
+		self.fwlabel = Label(self, text = "")
+		self.fwlabel.grid(row=5, column=1, sticky=N+W) 
+
+		self.display_none()
+
+	def display_none(self):
+		self['text'] = "Ether Dream"
+
+		for l in self.iplabel, self.verslabel, self.stlabel, self.srclabel, self.netlabel, self.fwlabel:
+			l['text'] = ""
+
+	def display_dac(self, dac):
+		b = dac.last_broadcast
+		self['text'] = "Ether Dream " + b.macstr()[6:]
+		self.iplabel['text'] = str(b.ip[0])
+		self.verslabel['text'] = "hardware %d, software %d" % (b.hw_rev, b.sw_rev)
+
+		st_str = ""
+		if b.status.le_state == 0:
+			st_str = "online, "
+		else:
+			st_str = "ESTOP ACTIVE (%d), " % (b.status.le_flags, )
+
+		if b.status.playback_state == 0:
+			st_str += "idle"
+		elif b.status.playback_state == 1:
+			st_str += "prepared"
+		elif b.status.playback_state == 2:
+			st_str += "playing (%d buffered, %d played)" % (b.status.fullness, b.status.point_count)
+		else:
+			st_str += "DAC state %d" % (b.status.playback_state, )
+
+		if b.status.point_rate:
+			st_str += ", %d pps" % (b.status.point_rate)
+
+		self.stlabel['text'] = st_str
+
+		if b.status.source == 0:
+			src_str = "network"
+		elif b.status.source == 1:
+			src_str = "file playback: %s, repeat %s" % (
+				b.status.source_flags & 1 and "playing" or "not playing",
+				b.status.source_flags & 2 and "on" or "off"
+			)
+		elif b.status.source == 2:
+			src_str = "abstract generator: %s" % (
+				b.status.source_flags & 1 and "playing" or "not playing",
+			)
+		else:
+			src_str = "unknown %d" % (b.status.source)
+		
+		self.srclabel['text'] = src_str
+		self.netlabel['text'] = dac.conn_status
+		self.fwlabel['text'] = dac.firmware_string
+
+class DacTracker(Listbox):
+	def __init__(self, master, *args, **kwargs):
+		Listbox.__init__(self, master, *args, **kwargs)
+
+		self.dac_list = []
+		self.dac_macstr_map = {}
+
+		self.bind("<<ListboxSelect>>", self.update_selection)
+
+	def update_selection(self, lb=None):
+		if self.dac_display:
+			try:
+				dac_obj = self.dac_list[self.index(ACTIVE)]
+			except:
+				return
+			self.dac_display.display_dac(dac_obj)
+
+	def got_packet(self, bp):
+		macstr = bp.macstr()
+		if macstr not in self.dac_macstr_map:
+			new_dac = DAC(macstr, bp)
+			self.insert(END, macstr[6:])
+			self.dac_list.append(new_dac)
+			self.dac_macstr_map[macstr] = new_dac
+			dac_obj = new_dac
+		else:
+			dac_obj = self.dac_macstr_map[macstr]
+			dac_obj.got_broadcast(bp)
+
+		if len(self.dac_list) == 1:
+			self.selection_set(0)
+			self.update_selection()
+
+		def check_on_dac():
+			if time.time() - dac_obj.last_broadcast_time < 2:
+				return
+
+			idx = self.dac_list.index(dac_obj)
+			self.dac_list.remove(dac_obj)
+			del self.dac_macstr_map[macstr]
+			self.delete(idx)
+			self.dac_display.display_none()
+
+
+		self.after(2000, check_on_dac)
+
+
+
+# Set up the basic window
+root = Tk()
+root.title("Ether Dream")
+root.resizable(FALSE, FALSE)
+frame = Frame(root)
+frame.grid()
+
+disp = DacDisplay(root)
+disp.grid(row=0, column=1, padx=5, pady=5)
+tracker = DacTracker(root, height=22)
+tracker.grid(row=0, column=0, padx=5, pady=5)
+tracker.dac_display = disp
+
+# Set up queue checker
+packet_queue = queue.Queue()
+def queue_check():
+	try:
+		while True:
+			data, addr = packet_queue.get_nowait()
+			tracker.got_packet(BroadcastPacket(data, addr))
+	except queue.Empty:
+		root.after(100, queue_check)
+
+root.after(100, queue_check)
+
+# Set up listening socket and thread
+s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+s.bind(("0.0.0.0", 7654))
+def socket_thread():
+	while True:
+		packet_queue.put(s.recvfrom(1024))
+_thread.start_new(socket_thread, ())
+
+root.mainloop()