from PIL import Image
import socket
import time
import math   

Particle_mini = False
localIP     = "192.168.3.2"
#localIP     = "192.168.1.138"
localPort   = 6454
bufferSize  = 530
CAP_MINI_01 =b'\x42\x4D\xB2\x4D\x00\x00\x00\x00\x00\x00\x36\x00\x00\x00\x28\x00\x00\x00\x39\x00\x00\x00\x74\x00\x00\x00\x01\x00\x18\x00\x00\x00\x00\x00\x7C\x4D\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
mode = 1
convert_jpeg = False

if Particle_mini==True:
    LEDS_TOTAL_1 = 57
    LEDS_TOTAL_2 = 43
    LEDS_TOTAL_3 = 29
    LEDS_TOTAL_4 = 15
else:
    LEDS_TOTAL_1 = 113
    LEDS_TOTAL_2 = 85
    LEDS_TOTAL_3 = 57
    LEDS_TOTAL_4 = 29

PARTS_1 = math.ceil(LEDS_TOTAL_1*2/4)*4
PARTS_2 = math.ceil(LEDS_TOTAL_2*2/4)*4
PARTS_3 = math.ceil(LEDS_TOTAL_3*2/4)*4
PARTS_4 = math.ceil(LEDS_TOTAL_4*2/4)*4
print("{f} {g} {h} {i}".format(f=PARTS_1, g=PARTS_2, h=PARTS_3, i=PARTS_4))
NUM_UNIVERSES_1 = math.ceil(LEDS_TOTAL_1*3*PARTS_1/510)
NUM_UNIVERSES_2 = math.ceil(LEDS_TOTAL_2*3*PARTS_2/510)
NUM_UNIVERSES_3 = math.ceil(LEDS_TOTAL_3*3*PARTS_3/510)
NUM_UNIVERSES_4 = math.ceil(LEDS_TOTAL_4*3*PARTS_4/510)
print("{f} {g} {h} {i}".format(f=NUM_UNIVERSES_1, g=NUM_UNIVERSES_2, h=NUM_UNIVERSES_3, i=NUM_UNIVERSES_4))
INI_UNIVERSE_1 = 0
INI_UNIVERSE_2 = NUM_UNIVERSES_1
INI_UNIVERSE_3 = NUM_UNIVERSES_1 + NUM_UNIVERSES_2
INI_UNIVERSE_4 = NUM_UNIVERSES_1 + NUM_UNIVERSES_2 + NUM_UNIVERSES_3
END_UNIVERSE = INI_UNIVERSE_4 + NUM_UNIVERSES_4
ART_DMX = 0X5000
ART_NET_ID = "Art-Net"
ART_DMX_START = 18
P01_numbers = [ [0 for i in range(530)] for j in range(NUM_UNIVERSES_1) ]
P02_numbers = [ [0 for i in range(530)] for j in range(NUM_UNIVERSES_2) ]
P03_numbers = [ [0 for i in range(530)] for j in range(NUM_UNIVERSES_3) ]
P04_numbers = [ [0 for i in range(530)] for j in range(NUM_UNIVERSES_4) ]

#P01_BIG_DMX = [[0] for j in range(18+NUM_UNIVERSES_1*510*3)]
P01_SIZE = 1400#18+NUM_UNIVERSES_1*510*3
P01_BIG_DMX = bytearray(P01_SIZE)
count_big_dmx_01 = 0
#print("{f}".format(f=P01_BIG_DMX))

mini01     =  ("192.168.3.10", 6454)
mini02     =  ("192.168.3.11", 6454)
mini03     =  ("192.168.3.12", 6454)
mini04     =  ("192.168.3.13", 6454)
##msgFromServer       = "Hello UDP Client"
##bytesToSend         = str.encode(msgFromServer)

# Create a datagram socket
UDPServerSocket = socket.socket(family=socket.AF_INET, type=socket.SOCK_DGRAM)

# Bind to address and ip
UDPServerSocket.bind((localIP, localPort))
print("UDP server up and listening")

# Listen for incoming datagrams
while(True):
    bytesAddressPair = UDPServerSocket.recvfrom(bufferSize)
    message = bytesAddressPair[0]
    address = bytesAddressPair[1]
    clientMsg = "Message from Client:{}".format(message)
    msg = "{}".format(message)
    result = msg.find(ART_NET_ID)
    if (result>=0):
        opcode = message[8] | message[9] << 8
        if (opcode == ART_DMX):
            incomingUniverse = message[14] | message[15] << 8
            dmxDataLength = message[17] | message[16] << 8
            #print("{f} found with universe: {u}, DataLength: {n}.".format(f=ART_NET_ID, u= incomingUniverse, n=dmxDataLength))
            if mode==0: #echo
                for i in range(dmxDataLength):
                    if message[ART_DMX_START+i]!=P01_numbers[incomingUniverse][i]:
                        P01_numbers[incomingUniverse] =  message[ART_DMX_START:(dmxDataLength+ART_DMX_START)] ##Copia los 512 valores del ARRAY
                        UDPServerSocket.sendto(message, mini01)
                        #print("Sended")
                        break
            elif mode==1: #delay entre paquetes
                if incomingUniverse<INI_UNIVERSE_2:P01_numbers[incomingUniverse] =  message[:] ##Copia todos los valores del ARRAY
                elif incomingUniverse<(INI_UNIVERSE_3):P02_numbers[incomingUniverse-INI_UNIVERSE_2] =  message[:] ##Copia todos los valores del ARRAY
                elif incomingUniverse<(INI_UNIVERSE_4):P03_numbers[incomingUniverse-(INI_UNIVERSE_3)] =  message[:] ##Copia todos los valores del ARRAY
                elif incomingUniverse<(END_UNIVERSE):P04_numbers[incomingUniverse-(INI_UNIVERSE_4)] =  message[:] ##Copia todos los valores del ARRAY

                ##if incomingUniverse==(END_UNIVERSE-1):
                if incomingUniverse==(INI_UNIVERSE_2-1):
                    if convert_jpeg==True:
                        with open('mini01.bmp', 'wb') as fp:
                            in_x = 0
                            fp.write(CAP_MINI_01)
                            for h in range(NUM_UNIVERSES_1): 
                                #fp.write(P01_numbers[h][ART_DMX_START:528])
                                for j in range(ART_DMX_START,528): 
                                    in_x=in_x+1
                                    fp.write(P01_numbers[h][j:j+1])    
                                    if in_x==171: 
                                        fp.write(b'\x00')  
                                        in_x = 0     
                            for h in range(NUM_UNIVERSES_1): 
                                if h%3==0: fp.write(b'\x00')
                        image = Image.open('mini01.bmp')
                        # Specifying the RGB mode to the image
                        image.save("mini01.jpg")
                    for h in range(NUM_UNIVERSES_1):
                        UDPServerSocket.sendto(P01_numbers[h], mini01)
                    ##    if h<(NUM_UNIVERSES_2-1): 
                    ##            UDPServerSocket.sendto(P02_numbers[h], mini02)
                    ##    if h<(NUM_UNIVERSES_3-1): 
                    ##            UDPServerSocket.sendto(P03_numbers[h], mini03)
                    ##    if h<(NUM_UNIVERSES_4-1): 
                    ##            UDPServerSocket.sendto(P04_numbers[h], mini04)
                        if h%4==0:
                            time.sleep(0.0011) # Sleep for 3 seconds
            elif mode==2: #agrupacion
                if incomingUniverse<INI_UNIVERSE_2:
                    #print("{f}".format(f=incomingUniverse))
                    if incomingUniverse == INI_UNIVERSE_1:
                        #print("{f}".format(f=message))
                        for h in range(530):
                            P01_BIG_DMX[h] =  message[h] ##Copia todos los valores del ARRAY
                        P01_BIG_DMX[14] = 0x00FF & 0
                        P01_BIG_DMX[15] = 0x00FF&(0 >> 8)
                        P01_BIG_DMX[17] = 0x00FF & 18+P01_SIZE
                        P01_BIG_DMX[16] = 0x00FF&((18+P01_SIZE) >> 8)
                    #else:
                         #for h in range(510):
                            #P01_BIG_DMX[incomingUniverse*510 +h] =  message[18+h] ##Copia todos los valores del ARRAY
                if incomingUniverse==(END_UNIVERSE-1):
                    UDPServerSocket.sendto(P01_BIG_DMX, mini01)
                    incomingUniverse = P01_BIG_DMX[14] | P01_BIG_DMX[15] << 8
                    dmxDataLength = P01_BIG_DMX[17] | P01_BIG_DMX[16] << 8
                    ##print("{f} {g}".format(f=incomingUniverse, g=dmxDataLength))
                    ##print("{f}".format(f=P01_BIG_DMX))