from secrets import choice
from tracemalloc import stop
import numpy as np
import matplotlib.pyplot as plt
import csv

#"C:\Users\quent\OneDrive\Bureau\ENSC\TransD\Framboisier\02400001.TXT"
#! /usr/bin/env python3
# -*- coding: UTF-8 -*-

def function():
    tab = np.zeros([1,9])

    #def tab_choix():
    while True:
        choice = input("Choisis un fichier csv de 1 à 1862: \n")
        print("\n")
        
        if float(choice) in range(1, 1863):
            rawdata = '../RawDataFiles/RawData' + choice + '.csv'
            with open(rawdata, newline='') as csvfile:
                reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC)
                for row in reader:
                    t = np.zeros([1,9])
                    t[0] = row
                    tab = np.append(tab ,t , axis = 0)
            break
        
        elif choice > '1862' or choice < '1':
            print("{} n'est pas un nombre valide.\n".format(choice))
            exit()
    
    x = tab[:,0]
    y = tab[:,1]
    print(tab)
    return x , y

#Fonction d'affichage du graphique
def printFig(axe_x , axe_y , title = "new plot" , xlabel = "Temps" , ylabel = "Tension"):
    plt.plot(axe_x , axe_y)
    plt.xlabel(xlabel)
    plt.ylabel(ylabel)
    plt.title(title)
    plt.show()
    return 0

#print(x)
#print("y = " + str(y))
def final():
    x, y = function()
    printFig(x,y)

final()