send.py 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. import time
  2. import socket
  3. import struct
  4. import random
  5. class ArtNetNode():
  6. """simple Object to generate ArtNet Network packages
  7. works in Python2 and Python3 2021-12-05
  8. (only basic implementation)
  9. "Art-Net™ Designed by and Copyright Artistic Licence Holdings Ltd"
  10. https://art-net.org.uk/
  11. """
  12. def __init__(self, to="10.10.10.255",univ=7,port=6454):
  13. try:
  14. univ = int(univ)
  15. except:
  16. print("errror univ",univ ,"is not int ... set to 7")
  17. univ = 7
  18. self.univ=univ
  19. self.sendto = to
  20. self.portto = port
  21. print(__name__,"bind",to,port,univ)
  22. self.s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
  23. self.s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
  24. self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  25. self.stamp = time.time()
  26. self.test_stamp = time.time()
  27. self.dmx=[33]*512
  28. self.v=0
  29. self.d=1
  30. def head(self):
  31. self._header = []
  32. self._header.append(b"Art-Net\x00") # Name, 7byte + 0x00
  33. self._header.append(struct.pack('<H', 0x5000)) # OpCode ArtDMX -> 0x5000, Low Byte first
  34. self._header.append(struct.pack('>H', 14)) # Protocol Version 14, High Byte first
  35. self._header.append(b"\x00") # Order -> nope -> 0x00
  36. self._header.append(struct.pack('B',1)) # Eternity Port
  37. # Address
  38. #if 0 <= universe <= 15 and 0 <= net <= 127 and 0 <= subnet <= 15
  39. net, subnet, universe = (0,0,self.univ) #address
  40. self._header.append(struct.pack('<H', net << 8 | subnet << 4 | universe))
  41. self._header = b"".join(self._header)
  42. def send(self,dmx=None,port=''):
  43. if dmx is None:
  44. dmx = self.dmx
  45. else:
  46. self.dmx = dmx
  47. self.head()
  48. c=[self._header]
  49. c.append( struct.pack('>H', len(dmx) ) )
  50. #print([c])
  51. dmx_count = 0
  52. for v in dmx:
  53. if type(v) is not int:
  54. v=0
  55. elif v > 255: # max dmx value 255
  56. v = 255
  57. elif v < 0: # min dmx value 0
  58. v = 0
  59. dmx_count += 1
  60. c.append(struct.pack("B",v))
  61. c = b"".join(c)
  62. if port:
  63. self.s.sendto(c, (self.sendto, port)) # default 6454
  64. else:
  65. self.s.sendto(c, (self.sendto, self.portto)) # default 6454
  66. print(self.v)
  67. time.sleep(0.0001)
  68. return c
  69. def _test_frame(self):
  70. if self.test_stamp+(.01) < time.time():
  71. self.test_stamp = time.time()
  72. dmx = [0]*512
  73. dmx[201-1] = self.v
  74. self.dmx = dmx
  75. if self.v >= 255:
  76. self.d=0
  77. elif self.v <=0:
  78. self.d=1
  79. if self.d:
  80. self.v+=1
  81. else:
  82. self.v-=1
  83. #print( self.v)
  84. #time.sleep(1/30.)
  85. def next(self):
  86. if self.stamp + (1/30.) < time.time():
  87. self.send()
  88. def artnet_test():
  89. #artnet = ArtNetNode(to="127.0.0.1",port=6555,univ=12)
  90. #artnet = ArtNetNode(to="127.0.0.1",port=6555,univ=0)
  91. artnet = ArtNetNode(to="10.10.10.255",univ=0)
  92. #artnet = ArtNetNode(to="10.10.10.255",univ=1)
  93. while 1:
  94. artnet._test_frame()
  95. artnet.next()
  96. time.sleep(0.01)
  97. artnet_test()