import serial
import time
from tkinter import *
import serial.tools.list_ports
from tkinter import filedialog as fd
import os
import threading

grbl = 0
i = 10
GCODE = 0
AXIS = 'X'
states = {'M3':'0', 'M8':'0', 'M6':'0', 'G10': '0'} #Spindle, Coolant, Toolchange

#GUI Main
buttonsize_x = 5
buttonsize_y = 3
increments = 0
BORDER = 2
freetosend = 1

#GUI Color Scheme
attention = 'red'
loaded = 'green'
cooling = 'blue'
toolchange = 'yellow'
standard = '#17223B'
feed = '#283B67'

def grblConnect2():
    global grbl

    #Serial Connection
    locations=['/dev/ttyACM0','/dev/ttyUSB0','/dev/ttyUSB1','/dev/ttyACM1','/dev/ttyACM2','/dev/ttyACM3',
        '/dev/ttyS0','/dev/ttyS1','/dev/ttyS2','/dev/ttyS3']
    
    for device in locations:
            try:
                #print([comport.device for comport in serial.tools.list_ports.comports()])
                print ("Trying...",device)
                grbl = serial.Serial(port= device, baudrate= 115200, timeout =3)
                #grbl.open()
                #print(grbl.readline())
                grbl.write(str.encode("\r\n\r\n"))
                time.sleep(2)   # Wait for grbl to initialize
                grbl.flushInput()  # Flush startup text in serial input
                connect_ser.config(bg= loaded)
                #print("connected")

                break
            except:
                #print ("Failed to connect on",device)
                grbl = 0

    # Stream g-code to grbl
    #Stream GCODE from -https://onehossshay.wordpress.com/2011/08/26/grbl-a-simple-python-interface/-    

   

def jogWrite(AXIS, CMD, scale):   
    DECIMAL = [0.1,1,10,100]
    scale = increments.get()
    MOVE = int(CMD) * DECIMAL[scale -1]     
    grbl_command = ('$J=G91' +' ' + AXIS + str(MOVE) + ' '+ 'F1000')    
    #print(grbl_command)
    if freetosend == 1:
            sendGRBL(grbl_command)

def directWrite(CMD):
    grbl_command = CMD
    sendGRBL(grbl_command)

def latchWrite(CMD):
    global states 
    if states[CMD] == '0':
        states[CMD] = '1'        
        if CMD == 'M3':
            spindle.config(bg= attention) #A31621
        if CMD == 'M6':
            tool.config(bg = toolchange)#E0CA3C
        if CMD == 'G10':
            zero_all.config(bg = loaded)        

    else:
        states[CMD] ='0'
        if CMD == 'M3':
            spindle.config(bg=loaded)#A2D729
        if CMD == 'M6':
            tool.config(bg='grey')
        #if CMD == 'G10':
        #    zero_all.config(bg= attention)

    
    if CMD == 'M3':
        if states['M3'] == '1':
            grbl_command = 'M3 S1000'
        else:
            grbl_command = 'M3 S0'
          
    elif CMD == 'M8':
        if states['M8'] == '1':
            grbl_command = (CMD)
            coolant.config(bg = cooling)#1F7A8C
        else:
            grbl_command = 'M9'
            coolant.config(bg ='grey')

    elif CMD == 'G10':
        grbl_command = 'G10 P0 L20 X0 Y0 Z0'

    else:
        grbl_command = (CMD)     
    
    #grbl_command = (CMD * int(states[CMD]) )   
    print(grbl_command)
    #print(states)
    sendGRBL(grbl_command)
    
def zeroWrite(CMD, AXIS):
    grbl_command = (CMD + ' ' + AXIS + '0')
    sendGRBL(grbl_command)

def terminalWrite():
    grbl_command = terminal.get()
    #print(grbl_command)
    sendGRBL(grbl_command)

def infoScreen(data):
    global i
    terminalFrame = Frame(terminal_recv, bg = 'white')
    terminal_recv.create_window(10,i, window = terminalFrame, anchor = 'nw')
    Label(terminalFrame, text = data, font = ('Calibri',10), bg ='white', fg ='black').pack()
    i += 22
    if i >=400:
        i=10
        terminal_recv.delete("all")

def openGCODE():
    global GCODE

    filetypes = (('GCODE', '*.nc'),('All files', '*.*'))
    GCODE = fd.askopenfile(title='Open a file', initialdir='/home/thomas/Nextcloud/CAM/', filetypes=filetypes)
    
    if GCODE != 0:
        fopen.config(bg= loaded)
    else:
        fopen.config(bg = 'grey')
      
    build_xy = findEnvelope()
    mill_table.create_rectangle(build_xy[0],build_xy[1], fill = 'blue', stipple = 'gray75')       
    
def findEnvelope(): #get the max used Buildspace and position of the job
    x_coords = []
    y_coords = []    
    coord_max = [0,1]
    coord_min = [0,1]   

    for line in GCODE:         
        l = line.strip(line) # Strip all EOL characters               
        l = line.replace('X', '')
        l2 = l.replace('Y', '')        
        l2 = l2.split()     
        
        if len(l2) == 3: 
            #print(l2)
            x = l2[0]
            if 'Z' not in x and 'X' in line and 'G0' not in x and 'G1' not in x:
                x_coords.append(float(x))
            y = l2[1]
            if 'Z' not in y and 'Y' in line and 'G0' not in x and 'G1' not in x:
                y_coords.append(float(y))        

    x_coords.sort()
    y_coords.sort()
    
    coord_max[0] = max(x_coords) +50
    coord_max[1] = 350 - max(y_coords) #invertierte Buildplattform mit 0 unten links statt oben links
    coord_min[0] = min(x_coords) +50
    coord_min[1] = 350 - min(y_coords)      

    return coord_min, coord_max

def grblWrite():   
    #print("write1")
    
    GCODE.seek(0)                
    for line in GCODE:        
        #print("write")
        l = line.strip() # Strip all EOL characters for streaming
        #l = line.split(";",1)  
        grbl_command = l
        print("GCODE",grbl_command)
        if freetosend == 1:
            sendGRBL(grbl_command)
           
    GCODE.close()
    fopen.config(bg = 'grey')

def grblClose():
    # Close file and serial port
    #f.close()
    try:
        grbl.close()
        print("closed")
        connect_ser.config(bg='grey')
    except:
        print("Connection still open")

def sendGRBL(grbl_command):
    global freetosend  
    freetosend = 0
    #print(grbl_command)  
    grbl.write(str.encode(grbl_command+ '\n')) # Send g-code block to grbl
    time.sleep(0.01) 
    grbl_out = grbl.readline() # Wait for grbl response with carriage return
           
    infoScreen(grbl_out)
    freetosend = 1
    return grbl_out
    
    #infoScreen("finished")  

def displayPosition():    
    if grbl != 0 and freetosend == 1:
        grbl_command = '?'          
        position = str(sendGRBL(grbl_command))
        #print(position)
        position = position.replace('Idle|', ',')
        position = position.replace('Run|', ',')
        position = position.replace('WPos:', '')
        position = position.replace('MPos:', '')
        position = position.replace('>', ',')
        position = position.replace('|', ',')  
        position.strip()
        coordinates_list = position.split(',')       
        #print(coordinates_list)
        try:
            show_ctrl_x.config(text = coordinates_list[1]) 
            show_ctrl_y.config(text = coordinates_list[2])
            show_ctrl_z.config(text = coordinates_list[3])

            #show_ctrl_x_w.config(text = coordinates_list[4]) 
            #show_ctrl_y_w.config(text = coordinates_list[5])
            #show_ctrl_z_w.config(text = coordinates_list[6])
        except:
            print("Listerror")
    else:
        print("Serial Busy")
    root.after(500,displayPosition) 

root = Tk()
root.title('touchCNC')
root.geometry('1024x600+0+0')
root.resizable(False,False)#17203b
root.attributes('-fullscreen', True)
root.tk_setPalette(background='#11192C', foreground='white',activeBackground='#283867', activeForeground='white' )

increments = IntVar()
movement = Frame(root, relief = 'ridge', bd = BORDER)
left = Button(root, text="-X",  width = buttonsize_x, height = buttonsize_y, command = lambda:jogWrite('X', '-1', increments),bd = BORDER, bg = standard)
right = Button(root, text="+X",width = buttonsize_x, height = buttonsize_y,command = lambda:jogWrite('X', '1', increments),bd = BORDER, bg = standard)
up = Button(root, text="+Y", width = buttonsize_x, height = buttonsize_y,command = lambda:jogWrite('Y', '1', increments),bd = BORDER, bg = standard)
down = Button(root, text="-Y",width = buttonsize_x, height = buttonsize_y,command = lambda:jogWrite('Y', '-1', increments),bd = BORDER, bg = standard)
z_up = Button(root, text="+Z",width = buttonsize_x, height = buttonsize_y,command = lambda:jogWrite('Z', '1', increments) ,bd = BORDER, bg = standard)
z_down = Button(root, text="-Z",width = buttonsize_x, height = buttonsize_y,command = lambda:jogWrite('Z', '-1', increments),bd = BORDER, bg = standard)

zero_x = Button(root, text="zero X",width = buttonsize_x, height = 1, command = lambda:directWrite('G10 P0 L20 X0'),bd = BORDER)
zero_y = Button(root, text="zero Y",width = buttonsize_x, height = 1, command = lambda:directWrite('G10 P0 L20 Y0' ),bd = BORDER)
zero_z = Button(root, text="zero Z",width = buttonsize_x, height = 1, command = lambda:directWrite('G10 P0 L20 Z0' ),bd = BORDER)
zero_all=Button(root, text="zeroAll",width = buttonsize_x, height = 3, command = lambda:latchWrite('G10'),bd = BORDER, bg= 'magenta')

setzero =Button(root, text="SetPOS",width = buttonsize_x, height = buttonsize_y, command = lambda:directWrite('G28.1'),bd = BORDER)
gozero =Button(root, text="GoPOS",width = buttonsize_x, height = buttonsize_y, command = lambda:directWrite('G28'),bd = BORDER) 

connect_ser = Button(root, text="Cnnct",width = buttonsize_x, height = buttonsize_y, command = grblConnect2, bg = 'grey',bd = BORDER)
discon_ser = Button(root, text="Dsconct",width = buttonsize_x, height = buttonsize_y, command = lambda:grblClose(),bd = BORDER)
unlock = Button(root, text="Unlock",width = buttonsize_x, height = buttonsize_y, command = lambda:directWrite('$X'),bd = BORDER)
start = Button(root, text="START",width = buttonsize_x, height = buttonsize_y, bg = attention, command = lambda: threading.Thread(target = grblWrite).start(),bd = BORDER)
stop = Button(root, text="STOP",width = buttonsize_x, height = buttonsize_y,bd = BORDER, command = lambda: directWrite('') )
pause = Button(root, text="PAUSE",width = buttonsize_x, height = buttonsize_y, bg = cooling,bd = BORDER,command = lambda: directWrite('!')  )
resume = Button(root, text="RESUME",width = buttonsize_x, height = buttonsize_y,bd = BORDER,command = lambda: directWrite('~'))

fopen = Button(root, text="GCODE",width = buttonsize_x , height = buttonsize_y, bg = 'grey',fg = 'black', command = openGCODE,bd = BORDER)

spindle = Button(root, text="Spindle",width = buttonsize_x, height = buttonsize_y,command = lambda:latchWrite('M3'))
coolant = Button(root, text="Coolant",width = buttonsize_x, height = buttonsize_y,command = lambda:latchWrite('M8') )
tool = Button(root, text="Tool",width = buttonsize_x, height = buttonsize_y,command = lambda:latchWrite('M6') )
macro = Button(root, text="Macro1",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite(' G90 G0 X10 Y10 Z50 F1000') )

inc1 = Button(root, text="Inc 1%",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite('‘'),bg= feed)
inc10 = Button(root,text="Inc 10%",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite('“'),bg= feed )
dec1 = Button(root, text="Dec 1%",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite('”'),bg= feed )
dec10 = Button(root,text="Dec 10%",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite('’'),bg= feed )
reset = Button(root,text="<RESET",width = buttonsize_x, height = buttonsize_y,command = lambda:directWrite(''),bg= 'grey' )

reboot= Button(root,text="REBOOT",width = buttonsize_x, height = buttonsize_y,command = lambda: os.system('reboot'),bg= 'grey' )

step_incr1 = Radiobutton(root, text= '0,1', value = 1 , variable = increments,width = buttonsize_x, height = buttonsize_y, indicatoron = 0 )
step_incr2 = Radiobutton(root, text= '1', value = 2 , variable = increments,width = buttonsize_x, height = buttonsize_y, indicatoron = 0 )
step_incr3 = Radiobutton(root, text= '10', value = 3 , variable = increments,width = buttonsize_x, height = buttonsize_y, indicatoron = 0 )
step_incr4 = Radiobutton(root, text= '100', value = 4 , variable = increments,width = buttonsize_x, height = buttonsize_y, indicatoron = 0 )
step_incr2.select()

terminal = Entry(root, width =8, text="GCODE")
terminal_send = Button(root, text="SEND",width = buttonsize_x, height = buttonsize_y, bd= 3, command = lambda: terminalWrite())
terminal_recv = Canvas(root, width = 200, height =400, bg = 'white')

show_ctrl_x_label = Label(root,text = "X")
show_ctrl_y_label = Label(root,text = "Y")
show_ctrl_z_label = Label(root,text = "Z")
show_ctrl_x =Label(root, text = "X_POS", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')
show_ctrl_y =Label(root, text = "Y_POS", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')
show_ctrl_z =Label(root, text = "Z_POS", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')

show_ctrl_x_w =Label(root, text = "X_POS_W", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')
show_ctrl_y_w =Label(root, text = "Y_POS_W", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')
show_ctrl_z_w =Label(root, text = "Z_POS_W", width = 8, height = 2, bg ='white', relief = SUNKEN, fg= 'black')

threading.Thread(target= displayPosition()).start() 

#feed_control = Scale(root, orient = HORIZONTAL, length = 400, label = "Feedrate",tickinterval = 20)

#Milling Area and Gcode preview with grid generation

mill_table= Canvas(root, width= 400, height = 400, bg = 'grey')

mill_table.create_rectangle(50,50,350,350, fill ='white')
mill_table.create_text(200,25,text = 'Fräsbereich 300mm x 300mm')

for x in range(50,350,50):
    mill_table.create_text(x,400- x, text = x-50)

for x in range(0,400,50):
    for y in range(0,400,50):
        gitter_x = mill_table.create_line(x,0,x,400)
        gitter_y = mill_table.create_line(0,y,400,y)

movement.grid(row = 0, column = 0, columnspan = 3, rowspan = 1)
left.grid(row=1, column=0, padx=3, pady=2)
right.grid(row=1, column=2,padx=3, pady=2)
up.grid(row=0, column=1, padx=3, pady=10)
down.grid(row=1, column=1,padx=3, pady=2)
z_up.grid(row=0, column=3,padx=10, pady=10)
z_down.grid(row=1, column=3,padx=10, pady=2)

step_incr2.select()

step_incr1.grid(row=2, column=0,padx=3, pady=10)
step_incr2.grid(row=2, column=1,padx=3, pady=10)
step_incr3.grid(row=2, column=2,padx=3, pady=10)
step_incr4.grid(row=2, column=3,padx=3, pady=10)

show_ctrl_x_label.grid(row=3, column=0,padx=3, pady=10)
show_ctrl_y_label.grid(row=4, column=0,padx=3, pady=10)
show_ctrl_z_label.grid(row=5, column=0,padx=3, pady=10)

show_ctrl_x.grid(row=3, column=1,padx=0, pady=0, columnspan =1)
show_ctrl_y.grid(row=4, column=1,padx=0, pady=0, columnspan =1)
show_ctrl_z.grid(row=5, column=1,padx=0, pady=0, columnspan =1)

show_ctrl_x_w.grid(row=3, column=2,padx=0, pady=0, columnspan =1)
show_ctrl_y_w.grid(row=4, column=2,padx=0, pady=0, columnspan =1)
show_ctrl_z_w.grid(row=5, column=2,padx=0, pady=0, columnspan =1)

zero_x.grid(row=3, column=3)
zero_y.grid(row=4, column=3)
zero_z.grid(row=5, column=3)
zero_all.grid(row=6, column=3,padx=10, pady=10)

setzero.grid(row=6, column=0,padx=10, pady=10)
gozero.grid(row=6, column=1,padx=10, pady=10)

connect_ser.grid(row=7, column=0,padx=10, pady=10)
discon_ser.grid(row=7, column=1,padx=10, pady=10)
unlock.grid(row=8, column=1,padx=10, pady=10)
start.grid(row=7, column=2,padx=10, pady=10)
stop.grid(row=7, column=3,padx=10, pady=10)
pause.grid(row=8, column=2,padx=10, pady=10)
resume.grid(row=8, column=3,padx=10, pady=10)
fopen.grid(row=8, column=0,padx=10, pady=10)

spindle.grid(row=7, column=4,padx=1, pady=10)
coolant.grid(row=7, column=5,padx=1, pady=10)
tool.grid(row=7, column=6,padx=1, pady=10)
macro.grid(row=7, column=7,padx=1, pady=10)

dec10.grid(row=8, column=4,padx=1, pady=10)
dec1.grid(row=8, column=5,padx=1, pady=10)
inc1.grid(row=8, column=6,padx=1, pady=10)
inc10.grid(row=8, column=7,padx=1, pady=10)

reset.grid(row=8, column=8,padx=1, pady=10)
reboot.grid(row=8, column=9,padx=1, pady=10)


terminal.grid(row = 7, column = 8, padx =2, pady =10)
terminal_send.grid(row = 7, column = 9, padx =2, pady =10)
terminal_recv.grid(row = 0, column = 8, padx =10, pady =10,rowspan = 7, columnspan =2)

#feed_control.grid(row = 8, column = 4, columnspan =4)

mill_table.grid(row=0, column=4,padx=10, pady=10,columnspan = 4, rowspan = 7)
  
root.mainloop()