# -*- coding: UTF-8 -*-
import random # Used for create_plots function (test purpose)

import serial

import matplotlib
matplotlib.use("TkAgg")
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2TkAgg
from matplotlib.figure import Figure
import matplotlib.animation as animation
from matplotlib import style

import Tkinter as tk
import ttk

import csv
import datetime
import os


T1_Arr=[]  # Create an empty array for saving temperatures values
I1_Arr=[]  # Create an empty array for saving current values
H1_Arr=[]  # Create an empty array for saving Hydrogen values
P1_Arr=[]  # Create an empty array for saving Hydrogen values

LARGE_FONT = ("Verdana",12)
NORM_FONT = ("Verdana",10)
SMALL_FONT = ("Verdana",8)

style.use('seaborn-darkgrid')

# --- default values ---
portCOM = 'com14'
baudRate = 115200
sampleSize = 50

# --- INFO : The menu item "Serial port/ Open serial COM..." is actually not working
# The command associated to this menu item is the openSerialPort(portCom, baudRate)
# defined few lines under. But as said... not working actually. So the serial communication
# is opened with default values with the line hereunder
arduinoData = serial.Serial()    #Creating our serial object named arduinoData
# ---

dataFileName = 'Data_logger.csv'
saveData = False


f = Figure()

##a = f.add_subplot(111)
##b = f.add_subplot(111)
##c = f.add_subplot(111)
##d = f.add_subplot(111)

a = f.add_subplot(221)
b = f.add_subplot(222)
c = f.add_subplot(223)
d = f.add_subplot(224)

# --------------------------
# function for test purposes

def qf(stringtoprint): # print a text in the python shell
    print(stringtoprint)
    
def create_plots(plotNumber, amplitude): # create random x, y values ; need random lib
    xs = []
    ys = []

    for i in range(plotNumber):
        x = i
        y= random.randrange(amplitude)

        xs.append(x)
        ys.append(y)
    return xs, ys

# --------------------------


def choosePort(): # Allow the user to choose the serial port for serial communication
    portComQ =tk.Tk()
    portComQ.wm_title('Serial port?')
    label = ttk.Label(portComQ, text = "Choose the serial port to listen.")
    label.pack(side="top", fill="x", pady=10)

    e = ttk.Entry(portComQ)
    e.insert(0,portCOM[3:]) # Getting the port number from the String
    e.pack()
    e.focus_set()

    def callback():
        global portCOM
        portCOM = 'com' + str(e.get())
        print('Port {} is now selected.'.format(portCOM))
        portComQ.destroy()
        
    b = ttk.Button(portComQ, text='Submit', width=10, command=callback)
    b.pack()
    tk.mainloop()
    

def chooseBaudRate(): # Allow the user to choose the baud rate of the serial communication
    baudRateQ =tk.Tk()
    baudRateQ.wm_title('Baud rate?')
    label = ttk.Label(baudRateQ, text = "Specify the baud rate value.")
    label.pack(side="top", fill="x", pady=10)

    e = ttk.Entry(baudRateQ)
    e.insert(0,baudRate)
    e.pack()
    e.focus_set()

    def callback():
        global baudRate
        baudRate = e.get()
        print('{} baudrate is now selected'.format(baudRate))
        baudRateQ.destroy()
        
    b = ttk.Button(baudRateQ, text='Submit', width=10, command=callback)
    b.pack()
    tk.mainloop()


def chooseSampleSize(): # Allow the user to choose the sample size of displayed values
    sampleSizeQ =tk.Tk()
    sampleSizeQ.wm_title('Sample size?')
    label = ttk.Label(sampleSizeQ, text = "Specify the sample size of displayed values.")
    label.pack(side="top", fill="x", pady=10)

    e = ttk.Entry(sampleSizeQ)
    e.insert(0,sampleSize)
    e.pack()
    e.focus_set()

    def callback():
        global sampleSize
        sampleSize = int(e.get())
        print('Sample size is now {}'.format(sampleSize))
        sampleSizeQ.destroy()
        
    b = ttk.Button(sampleSizeQ, text='Submit', width=10, command=callback)
    b.pack()
    tk.mainloop()
    

# --- Now working !!! ---
def openSerialPort(portCom, baudRate, openState): # Should open a serial communication with the portCom and baudRate defined
    global arduinoData
    arduinoData.port = portCOM
    arduinoData.baudrate = baudRate
    if openState == True:
        if arduinoData.isOpen() == False:
            try:
                arduinoData.open()
                print('Port {} is now opened'.format(arduinoData.port))
            except:
                msg = 'Unable to open serial port {}. Please check connection.'.format(arduinoData.port)
                print(msg)
                popupmsg(msg)
                
    elif openState == False:
        if arduinoData.isOpen() == True:
            try:
                arduinoData.close()
                print('Port {} is now closed'.format(arduinoData.port))
            except:
                msg = 'Unable to close serial port {}. Please check connection.'.format(arduinoData.port)
                print(msg)
                popupmsg(msg)
        else:
            msg = 'Serial port {} already closed.'.format(arduinoData.port)
            print(msg)
            popupmsg(msg)
            
# -----------------------    

    

def saveDataAsCSV(state): # Create a new csv file for saving data.
    global dataFileName
    global saveData
    
    if state == True : # Create the file name.
        print("START recording data...")
        dataFileName = 'Data_logger_ {date:%Y-%m-%d_%H-%M-%S}.csv'.format( date=datetime.datetime.now() )
        saveData = True
        print(dataFileName)
        
    elif state == False : # Close the csv file and rename it with the actual time code format.
        saveData = False
        dataFileNameClosed = 'Data_logger_ {date:%Y-%m-%d_%H-%M-%S}.csv'.format( date=datetime.datetime.now() )
        os.rename(dataFileName, dataFileNameClosed)
        print("STOP recording data and saving record as csv file.")
        print(dataFileNameClosed)

        

def popupmsg(msg): # Popup the given message "msg"
    popup = tk.Tk()

    popup.wm_title("!")
    label = ttk.Label(popup, text=msg, font=NORM_FONT)
    label.pack(side="top", fill="x", pady=10)
    B1 = ttk.Button(popup, text="Okay", command = popup.destroy)
    B1.pack()
    popup.mainloop()

    
    
    
def animate(i): # This function create graph, read data from serail port and update the graph
    if arduinoData.isOpen() == True:
        while (arduinoData.inWaiting()==0):     # Wait here until there is data
            pass
            
        arduinoString = arduinoData.readline()  # read the line of text from the serial port
        dataArray = arduinoString.split(',')    # Split it into an array called dataArray
            
        T1 = float( dataArray[0])   # Convert 1st element (temperature) to floating number and put in T1
        I1 = float( dataArray[1])   # Convert 2nd element (Current) to floating number and put in I1
        H1 = float( dataArray[2])   # Convert 3rd element (hydrogen) to integer number and put in H1
        P1 = float( dataArray[3])   # Convert 4th element (Pressure) to floating number and put in P1
        
        T1_Arr.append(T1)   # Build our T1 array by appending T1 readings
        I1_Arr.append(I1)   # Building our current array by appending I1 readings
        H1_Arr.append(H1)   # Building our Hydrogen array by appending H1 readings
        P1_Arr.append(P1)   # Building our Hydrogen array by appending P1 readings
        
        a.clear() # clean the last graph in order to allow the new values to be drawn
        b.clear()
        c.clear()
        d.clear()
        
        a.plot(T1_Arr, 'ro-', label=u'Temp °C' , linewidth=0.5 , markersize=2)   # plot the temperature data
        a.set_title('Temperature')
        a.set_ylim(10,80)
        a.set_ylabel (u'Temperature (°C)')
        
        b.plot(I1_Arr, 'b^-', label='Current mA', linewidth=1 , markersize=2) # plot the current data
        b.set_title('Current')
        b.set_ylim(0,1000)
        b.set_ylabel ('current (mA)')
        
        c.plot(H1_Arr, 'g^-', label='Hydrogen', linewidth=1 , markersize=2) # plot hydrogen data
        c.set_title('Hydrogen')
        c.set_ylim(0,1000)
        c.set_ylabel ('Hydrogen (ppm)')
        
        d.plot(P1_Arr, 'm^-', label='Pressure', linewidth=1 , markersize=2) # plot pressure data
        d.set_title('Pressure')
        d.set_ylim(0,100)
        d.set_ylabel ('Pressure (KPa)')

        if(len(T1_Arr)>sampleSize):     # If you have "sampleSize" or more points, delete the first one from the array            
            T1_Arr.pop(0)       # This allows us to just see the last "sampleSize" data points
            I1_Arr.pop(0)       # Have to be improved later by selecting the number of points in a menu
            H1_Arr.pop(0)
            P1_Arr.pop(0)

        
        if saveData ==True: # --- Saving data as csv file --- (see saveDataAsCSV)
            myFile = open(dataFileName,'ab')
            try:
                theWriter = csv.writer(myFile, dialect='excel')  
                theWriter.writerow([T1,I1,H1,P1])
            finally:
                myFile.close()


class DataLogApp(tk.Tk): # Design of the main application and menubar

    def __init__(self, *arg, **kwargs):

        tk.Tk.__init__(self, *arg, **kwargs)

        tk.Tk.wm_title(self, "Data logger of the poor")

        container = tk.Frame(self)
        container.pack(side='top', fill="both", expand = True)
        
        container.grid_rowconfigure(0, weight=1)
        container.grid_columnconfigure(0,weight=1)

        menubar = tk.Menu(container)                        # create a container named "menubar" that will contain all menu items
        
        # --- File menu ---
        filemenu = tk.Menu(menubar, tearoff=0)              # create a Menu element named "filemenu"
        filemenu.add_command(label="Save settings", command = lambda: popupmsg("Not supported just yet!")) # add a command button/label labelled "Save setting"
        filemenu.add_separator()                            # add a separator horizontal line in the menu "filemenu"
        filemenu.add_command(label="Exit", command = quit)  # add a command button/label labelled "Exit" and assign the quit command to it
        menubar.add_cascade(label="File", menu=filemenu)    # add the filemenu element to the menubar element

        # --- Record data menu ---
        data_menu = tk.Menu(menubar, tearoff=1)       # create a Menu element named "data_menu"
        data_menu.add_command(label="Start record",   # add a command button/label labelled
                                command = lambda: saveDataAsCSV(True)) 
        data_menu.add_command(label="Stop record",    # add a command button/label labelled
                                command = lambda: saveDataAsCSV(False)) 
        data_menu.add_command(label="Sample size",  # add a command button/label labelled
                                command = chooseSampleSize) 
        data_menu.add_command(label="Clear data",       # add a command button/label labelled 
                                command = lambda: popupmsg("Not supported just yet!")) 
        menubar.add_cascade(label="Data",               # add the data_menu element to the menubar element
                                 menu=data_menu)

        
        # --- Serial port communication menu ---
        serialPort_menu = tk.Menu(menubar, tearoff=1)               # create a Menu element named "serialPort_menu"
        serialPort_menu.add_command(label='Choose port ...',    # add a command button/label labelled "Choose serial port"
                                command = choosePort)
        serialPort_menu.add_command(label= 'Choose baud rate... ', # add a command button/label labelled "Choose baud rate"
                                command = chooseBaudRate)
        serialPort_menu.add_command(label='Open Serial COM...',     # add a command button/label labelled "Start listening"
                                command = lambda: openSerialPort(portCOM,baudRate,True))
        serialPort_menu.add_command(label='Close Serial COM...',     # add a command button/label labelled "Start listening"
                                command = lambda: openSerialPort(portCOM,baudRate,False))
 
        menubar.add_cascade(label="Serial port",                    # add the serialPort_menu element to the menubar element
                                 menu=serialPort_menu) 

        

        tk.Tk.config(self, menu=menubar)

        self.frames = {}

        for F in (StartPage, graph_Page):

            frame = F(container, self)

            self.frames[F] = frame

            frame.grid(row=0, column=0, sticky="nsew")

        self.show_frame(StartPage)

    def show_frame(self, cont):

        frame = self.frames[cont]
        frame.tkraise()

        



class StartPage(tk.Frame): # Start page ...

    def __init__(self, parent, controller):
        tk.Frame.__init__(self,parent)
        label = tk.Label(self, text="""Data logger of the poor application
        use at your own risk.
        There is no promise of waranty""", font= LARGE_FONT)
        label.pack(pady=10,padx=10)

        button1 = ttk.Button(self,
                            text="Agree",
                            command=lambda: controller.show_frame(graph_Page))
        button1.pack()
        button2 = ttk.Button(self,
                            text="Disagree",
                            command=quit)
        button2.pack()
        


##class PageOne(tk.Frame):
##
##    def __init__(self, parent, controller):
##        tk.Frame.__init__(self,parent)
##        label = tk.Label(self, text="Page one", font= LARGE_FONT)
##        label.pack(pady=10,padx=10)
##
##        button1 = ttk.Button(self,
##                            text="Back to home",
##                            command=lambda: controller.show_frame(StartPage))
##        button1.pack()




class graph_Page(tk.Frame): # Page of the monitoring graph

    def __init__(self, parent, controller):

        tk.Frame.__init__(self,parent)
        
        tk.Label(self, text="Monitoring Page", font= LARGE_FONT).pack(pady=10,padx=10)



        canvas = FigureCanvasTkAgg(f, self)
        canvas.show

        canvas.get_tk_widget().pack(side=tk.TOP, fill=tk.BOTH, expand = True)

        toolbar = NavigationToolbar2TkAgg(canvas, self)
        toolbar.update()
        canvas._tkcanvas.pack(side=tk.TOP, fill=tk.BOTH, expand=True)
        
        
        

app = DataLogApp()
app.geometry("1280x720")
ani = animation.FuncAnimation(f, animate, interval=500)
app.mainloop()