|
|
@@ -0,0 +1,289 @@
|
|
|
+# -*- 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, 600))
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+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.5
|
|
|
+ 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():
|
|
|
+ poll_flag = 1
|
|
|
+ xx = xsocket.recive()
|
|
|
+ k = xx["host"] +":"+ str(xx["head"][6])
|
|
|
+ sdata[k] = xx
|
|
|
+ print(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()
|
|
|
+_x=0
|
|
|
+while running:
|
|
|
+ x=int(_x)
|
|
|
+ clock.tick(310)
|
|
|
+ #screen.fill((0, 0, 0))
|
|
|
+
|
|
|
+ sdata = e.get()
|
|
|
+ #print(sdata)
|
|
|
+ sdata={}
|
|
|
+ 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
|
|
|
+
|
|
|
+ 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]
|
|
|
+ data.append(y)
|
|
|
+ y = xx["dmx"][3-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:
|
|
|
+ _x=0
|
|
|
+ x=0
|
|
|
+ rec = pygame.Rect(x,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-1,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)
|
|
|
+ c+=50
|
|
|
+ if c >255:
|
|
|
+ c=255
|
|
|
+ T.y-=265
|
|
|
+ pygame.display.flip()
|
|
|
+ _x+=0.5
|
|
|
+
|
|
|
+
|
