add: fast pygame GUI (oszi/grid) to visualize and debug dmx data

monitor/loop_oszi.sh

@@ -2,7 +2,7 @@
 
 while :
 do
-  python3 '/opt/LibreLight/ASP/monitor/oszi_sdl.py'
+  python3 '/opt/LibreLight/ASP/monitor/oszi_grid.py'
   PS1="X:"
   PS1="$PS1 $(ip netns identify $$)"
   export PS1

monitor/oszi_grid.py

@@ -0,0 +1,336 @@
+# -*- coding: UTF-8 -*-
+import os
+import fcntl
+import time
+import socket
+import struct
+import random
+
+from optparse import OptionParser
+parser = OptionParser()
+parser.add_option("-r", "--recive", dest="recive",
+                          help="set recive ip like --recive 10.")
+parser.add_option("-s", "--sendto", dest="sendto",
+                          help="set sender ip like --sendto 2.255.255.255")
+parser.add_option("-t", "--test", dest="testuniv",
+                          help="set test univers like --test [0-16]")
+parser.add_option("", "--inmap", dest="inmap",
+                          help="set test univers like --test [0-16]")
+#parser.add_option("-q", "--quiet",
+#                  action="store_false", dest="verbose", default=True,
+#                  help="don't print status messages to stdout")
+
+(options, args) = parser.parse_args()
+print("option",options)
+print(options.sendto)
+
+
+def unpack_art_dmx(data):
+    dmx = []
+    for i in range(len(data[18:]) ):
+        x=data[18+i]
+        #print("x",x)
+        #print( "data",b'!B', data[18+i])
+        #x=struct.unpack( b'!B',data[18+i])
+        #print( "data",b'!B', data[18+i],x)
+        #x=x[0]
+        dmx += [x]
+    return dmx
+
+        
+class Socket():
+    def __init__(self,bind='',port=6454):
+        self.__port =port
+        self.__bind =bind
+        self.__poll = 0
+        self.__data = []
+        self.__addr = "NONE"
+        #self.__hosts = {"host":{"9":[0]*512}}
+        self.__hosts = {}
+        self.hosts = self.__hosts
+        self.open()
+        self._poll_clean_time = time.time()
+        self._poll_clean_count = 0
+    def open(self):
+        try:
+            print("connecting to ArtNet bind:",self.__bind,"Port",self.__port)
+            self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+            
+            self.sock.bind((self.__bind, self.__port))
+            fcntl.fcntl(self.sock, fcntl.F_SETFL, os.O_NONBLOCK)
+            #self.sock.setblocking(0)
+            
+        except socket.error as e:
+            print("Socket ",self.__bind,self.__port, "ERR: {0} ".format(e.args))
+            #raw_input()
+            #sys.exit()
+    def poll_clean(self):
+        if self._poll_clean_time+(1/25.) <= time.time():
+            self._poll_clean_time = time.time()
+            self._poll_clean()
+            x = self._poll_clean_count 
+            self._poll_clean_count = 0
+            return x
+    def _poll_clean(self):
+        while 1:
+            try:
+                self.__data, self.__addr = self.sock.recvfrom(self.__port)
+                self._poll_clean_count += 1
+                #return 1
+            except socket.timeout as e:
+                err = e.args[0]
+                if err == 'timed out':
+                    time.sleep(1)
+                    print('recv timed out, retry later')
+                else:
+                    print(e)
+                break
+            except socket.error as e:
+                break
+    def poll(self):
+        if not self.__poll:
+            try:
+                self.__data, self.__addr = self.sock.recvfrom(self.__port)
+
+
+                data, addr = (self.__data,self.__addr)
+                self.host = addr[0]
+                head    = data[:18]
+                rawdmx  = data[18:]
+                #print([head],addr)
+                self.univ = -1
+                try:
+                    self.head = struct.unpack("!8sHBBBBHBB" , head )
+                except Exception as e:
+                    pass#print( "======E09823" , e)
+                univ = self.head[6]/255 # /512  # * 512
+                self.univ = int(univ)
+
+                if self.host.startswith("127."): #allways recive localhost on port 
+                    self.__poll = 1
+                    return 1
+                elif not options.recive:
+                    self.__poll = 1
+                    return 1
+                elif self.host.startswith(options.recive): 
+                    self.__poll = 1
+                    return 1
+                else:
+                    self.__poll = 0
+                
+                addr = str(addr)
+                univ = str(univ)
+                if self.__poll:
+                    if addr not in self.__hosts:
+                        self.__hosts[addr] = {}
+                    if univ not in self.__hosts[addr]:
+                        self.__hosts[addr][univ] = {}
+			
+                    self.__hosts[addr][univ] = {"head":head,"addr":addr,"univ":univ,"dmx":rawdmx}
+                    self.hosts = self.__hosts
+
+            except socket.timeout as e:
+                err = e.args[0]
+                if err == 'timed out':
+                    time.sleep(1)
+                    print('recv timed out, retry later')
+                else:
+                    print(e)
+            except socket.error as e:
+                pass
+    
+    def recive(self):
+        if self.__poll:
+            self.__poll = 0
+
+            data, addr = (self.__data,self.__addr)
+            #print( self.univ,self.head)
+
+            self.dmx  = unpack_art_dmx(data)
+
+            return { "host":self.host,"dmx":self.dmx,"univ":self.univ,"head":self.head,"data":data,"addr":addr}
+    
+
+import pygame
+import pygame.gfxdraw
+pygame.init()
+
+screen = pygame.display.set_mode((800, 800))
+
+
+
+
+pygame.display.set_caption("pygame: DMX OSZI")
+pygame.mouse.set_visible(1)
+pygame.key.set_repeat(1, 30)
+clock = pygame.time.Clock()
+#sf = pygame.Surface(300,300)
+pygame.init() 
+
+x=0
+y=0
+running = True
+xsocket = Socket()
+
+class Trans():
+    def __init__(self,y=150):
+        self.y = 150
+        self.s = 0.25
+    def get_y(self,y):
+        return int((y*self.s)+(self.y*self.s))*-1
+
+
+import copy
+import _thread as thread
+
+
+class E():
+    def __init__(self):
+        self.sdata = {}
+        self.lock = thread.allocate_lock()
+    def loop(self):
+        sdata = {}
+        print("loop")
+        while 1:
+            flag = 0
+            while xsocket.poll():
+                xx = xsocket.recive()
+                k = xx["host"] +":"+ str(xx["head"][6])
+                sdata[k] = xx
+                flag = 1
+
+            if flag:
+                try:
+                    self.lock.acquire()
+                    self.sdata = copy.deepcopy(sdata)
+                finally:
+                    self.lock.release()
+            time.sleep(0.001)
+    def get(self):
+        
+        try:
+            self.lock.acquire()
+            x = self.sdata #= copy.deepcopy(asdata)
+            self.sdata = {}
+            return x
+        finally:
+            self.lock.release()
+e = E()
+thread.start_new_thread(e.loop,())
+T = Trans()
+import sys
+font = pygame.font.SysFont("FreeSans", 20) #,color=(255,0,0))
+font = pygame.font.SysFont("monospace", 14,"bold") #,color=(255,0,0))
+
+_x=0
+sdata={}
+grid_timer = time.time()
+lz = time.time()
+while running:
+    x=int(_x)
+    clock.tick(30)
+    #screen.fill((0, 0, 0))
+
+    sdata = e.get()
+    #print(sdata)
+    if 0:
+        #while xsocket.poll():
+        #print(1)
+        poll_flag = 1
+        xx = xsocket.recive()
+        k = xx["host"] +":"+ str(xx["head"][6])
+        sdata[k] = xx
+        #print(xx)
+        flag = 1
+    xsdata = copy.deepcopy(sdata)
+    data = []
+    if int(time.time()*10) % 20 == 0:
+        for k in sdata:
+            print(k)
+    for k in sdata:
+        xx = sdata[k]
+        if xx["host"] == '2.0.0.88' and xx["head"][6]==0:
+            y = xx["dmx"][2-1]
+            y = xx["dmx"][21-1]
+            data.append(y)
+            y = xx["dmx"][3-1]
+            y = xx["dmx"][31-1]
+            data.append(y)
+            y = xx["dmx"][261-1]
+            data.append(y)
+            y = xx["dmx"][263-1]
+            data.append(y)
+    for event in pygame.event.get():
+        if event.type == pygame.QUIT:
+            running = False
+
+        if event.type == pygame.KEYDOWN:
+            print(event.type)
+            if event.key == pygame.K_ESCAPE:
+                pygame.event.post(pygame.event.Event(pygame.QUIT))
+
+
+    #if _x > 800:
+    if time.time() > lz:
+        lz = time.time()+2
+        _x=0
+        x=0
+        rec = pygame.Rect(x+1,T.get_y(10),30,245) # clear balken
+        pygame.draw.rect(screen,(10,10,0),rec)
+
+        #screen.fill((0, 0, 0))
+    T.y=-260
+    c=0
+    for d in data:
+        y=d
+        pygame.gfxdraw.pixel(screen,x,T.get_y(255),(255,0,0))
+        pygame.gfxdraw.pixel(screen,x,T.get_y(0),(0,0,255))
+        #rec = pygame.Rect(10+x,get_y(y),3,3) 
+        #pygame.draw.rect(screen,(255,255,0),rec)
+        rec = pygame.Rect(x+4,T.get_y(0),20,-80) # clear balken
+        pygame.draw.rect(screen,(c,210,110),rec)
+        rec = pygame.Rect(x+2,T.get_y(0),30,-80) # clear balken
+        pygame.draw.rect(screen,(c,10,110),rec)
+        text = font.render( str(y), True, (0,0,0))
+        #screen.blit(text, ( x+5, T.get_y(100) ) )
+        #print( dir(pygame.draw))
+        #sys.exit()
+
+        pygame.draw.circle(screen,(255,155,0),(x,T.get_y(y)),2)
+        c+=50
+        if c >255:
+            c=255
+        T.y-=265
+
+    if grid_timer < time.time():
+        #print("delta",round(grid_timer-time.time(),4))
+        grid_timer=time.time()+.015
+        for d in xsdata:
+            xx=sdata[d]
+            print(xx["head"],xx["host"])
+            if xx["host"] == '2.0.0.88' and xx["head"][6]==0:
+                rx=10
+                ry=T.get_y(200)
+                rec = pygame.Rect(rx,ry,800,600) # clear balken
+                pygame.draw.rect(screen,(0,10,210),rec)
+                line = []
+                for i,dmx in enumerate(xx["dmx"]):
+                    #dmx = xx["dmx"][1]
+                    text = font.render( str(dmx).rjust(3," "), True, (255,255,255))
+                    screen.blit(text, ( rx+10, ry+10 ) )
+                    #print(rx,ry,"dmx",dmx)
+                    #line.append(str(dmx).rjust(3," "))
+                    rx+=30
+                    #if rx > 600:
+
+                    if  (i+1) % 20 == 0:
+                        rx=10
+                        ry+=15
+                        
+                        #text = font.render( " ".join(line), True, (200,200,200))
+                        #line = []
+                        #screen.blit(text, ( rx+10, ry+10 ) )
+    pygame.display.flip()
+    _x+=3.5*2
+
+

monitor/oszi_sdl.py

@@ -155,7 +155,7 @@ import pygame
 import pygame.gfxdraw
 pygame.init()
 
-screen = pygame.display.set_mode((800, 600))
+screen = pygame.display.set_mode((800, 300))
 
 
 
@@ -175,7 +175,7 @@ xsocket = Socket()
 class Trans():
     def __init__(self,y=150):
         self.y = 150
-        self.s = 0.5
+        self.s = 0.25
     def get_y(self,y):
         return int((y*self.s)+(self.y*self.s))*-1
 
@@ -220,10 +220,10 @@ T = Trans()
 _x=0
 while running:
     x=int(_x)
-    clock.tick(310)
+    clock.tick(30)
     #screen.fill((0, 0, 0))
 
-    sdata = e.get()
+    #sdata = e.get()
     #print(sdata)
     sdata={}
     while xsocket.poll():
@@ -263,7 +263,7 @@ while running:
     if _x > 800:
         _x=0
         x=0
-        rec = pygame.Rect(x,T.get_y(10),30,245) # clear balken
+        rec = pygame.Rect(x+1,T.get_y(10),30,245) # clear balken
         pygame.draw.rect(screen,(10,10,0),rec)
 
         #screen.fill((0, 0, 0))
@@ -275,7 +275,9 @@ while running:
         pygame.gfxdraw.pixel(screen,x,T.get_y(0),(0,0,255))
         #rec = pygame.Rect(10+x,get_y(y),3,3) 
         #pygame.draw.rect(screen,(255,255,0),rec)
-        rec = pygame.Rect(x-1,T.get_y(0),30,-127) # clear balken
+        rec = pygame.Rect(x+4,T.get_y(0),20,-127) # clear balken
+        pygame.draw.rect(screen,(c,210,110),rec)
+        rec = pygame.Rect(x+2,T.get_y(0),30,-127) # clear balken
         pygame.draw.rect(screen,(c,10,110),rec)
 
         pygame.draw.circle(screen,(255,155,0),(x,T.get_y(y)),2)
@@ -284,6 +286,6 @@ while running:
             c=255
         T.y-=265
     pygame.display.flip()
-    _x+=0.5
+    _x+=3.5
 
 

monitor/setup_oszi.sh

@@ -1,4 +1,5 @@
 sleep 10
 sudo sh /opt/LibreLight/ASP/netns.sh 23
+#sudo ip netns exec blue223 su - user -c 'sh /opt/LibreLight/ASP/monitor/loop_oszi.sh'
 sudo ip netns exec blue223 su user -c 'screen -d -m -S oszi sh /opt/LibreLight/ASP/monitor/loop_oszi.sh'