# -*- coding: utf-8 -*-
import os
#import easygui
from tkinter import filedialog
import numpy as np
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
import DTAmeasurement

def process_data(data_in):
    t = []
    T1 = []
    T2 = []
    dT = []
    for i in range(len(data_in)):
        t.append(data_in[i][0])
        dT.append(data_in[i][1])
        T1.append(data_in[i][2])
        T2.append(data_in[i][2] - data_in[i][1])
    return t,T1,T2,dT


def new(*args):
    if(len(args)==0):
        #path = easygui.fileopenbox(filetypes=[["*","All files"],["*.*","All files"]]) 
        path = filedialog.askopenfilename(initialdir = ".",
                                          title = "Select a File",
                                          filetypes = (("all files",
                                                        "*"),
                                                        ("dat files",
                                                        "*.dat")))
    else:
        path = args[0]
    if(path):
        return DTAmeasurement.Measurement(path)
    else:
        print("Error <DTAutil.new>: no file found in path")
    return False



def flin(x, a, b):
    return a*x+b

def fexp(x,A,b,c):
    return A*np.exp(-b*x)+c

def trapcalc(x,y):
    sum = 0
    for i in range(len(x)-1):
        if(y[i] < y[i+1]):
            sum += y[i] * (x[i+1]-x[i])  +  (y[i+1]-y[i]) * (x[i+1]-x[i])/2
        else:
            sum += y[i+1] * (x[i+1]-x[i])  +  (y[i]-y[i+1]) * (x[i+1]-x[i])/2
    return sum

def trapadd(sum, x0, y0, x1, y1):
    if(y0 < y1):
        return sum + y0 * (x1-x0)  +  (y1-y0) * (x1-x0)/2
    return sum + y1 * (x1-x0)  +  (y0-y1) * (x1-x0)/2



def autozoom(plt, meas, region):
    xlim = [0,0]
    ylim = [9999,0]
    if(region == "base"):
        xlim[0] = 0
        for i in range(1,len(meas.y),1):
            if(meas.y[i] < ylim[0]):
                ylim[0] = meas.y[i]
            if(meas.y[i] > ylim[1]):
                ylim[1] = meas.y[i]
            if(abs(np.average(meas.y[0:i:1]) - meas.y[i]) > 0.1):
                xlim[1] = meas.x[i]
                break

    if(region == "exp"):
        imax = meas.y.index(max(meas.y))
        xlim = [meas.x[imax], meas.x[-1]]
        ylim = [meas.y[-1], meas.y[imax]]


    if(region == "main"):
        for i in range(1,len(meas.y),1):
            if(abs(np.average(meas.y[0:i:1]) - meas.y[i]) > 0.1):
                xlim[0] = meas.x[i-10]
                ylim[0] = meas.y[i-10]
                break
        imax = meas.y.index(max(meas.y))
        xlim[1] = meas.x[imax]
        ylim[1] = meas.Ts[imax]



    if(region == "int"):
        for i in range(1,len(meas.y),1):
            if(abs(np.average(meas.y[0:i:1]) - meas.y[i]) > 0.1):
                xlim[0] = meas.x[i-20]
                ylim[0] = meas.y[i-20]
                break
        xlim[1] = meas.x[-1]
        if(meas.bTs):
            ylim[1] = max(meas.Ts)
        else:
            ylim[1] = max(meas.y)


    xlim[0] -= (xlim[1] - xlim[0])/18
    xlim[1] += (xlim[1] - xlim[0])/18
    ylim[0] -= (ylim[1] - ylim[0])/18
    ylim[1] += (ylim[1] - ylim[0])/18

    plt.gca().set_xlim(tuple(xlim))
    plt.gca().set_ylim(tuple(ylim))