sitter3 for python3
This commit is contained in:
Sam
parent
6b1b75858f
commit
e21fd57743
422
ethertools/sitter3.py
Executable file
422
ethertools/sitter3.py
Executable file
@ -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()
|
||||
Loading…
Reference in New Issue
Block a user