import csv
import matplotlib.pyplot as plt

# Fonction pour lire un fichier CSV et retourner les 1000 premières lignes
def lire_fichier_par_bloc(fichier, start_line, end_line):
    valeurs = []
    with open(fichier, 'r') as f:
        lecteur_csv = csv.reader(f,delimiter=';')
        # Lire les lignes du fichier et les ajouter à la liste
        for i, ligne in enumerate(lecteur_csv):
            if i >= start_line and i < end_line:
                valeurs.append(float(ligne[0]))  # On suppose qu'il y a une seule valeur par ligne
            if i >= end_line:
                break
    return valeurs

# Fonction pour tracer les graphiques
def tracer_graphe(data1, data2, start_line):
    plt.figure(figsize=(10, 6))
    plt.plot(data1, label='Fichier 1', color='blue')
    plt.plot(data2, label='Fichier 2', color='red')
    plt.title(f'Graphique des lignes {start_line} à {start_line + 1000}')
    plt.xlabel('Index')
    plt.ylabel('Valeur')
    plt.legend()
    plt.savefig("data.png")

# Fonction principale
def traiter_fichiers(fichier1, fichier2, taille_bloc=1000):
    # Initialiser les indices de départ pour chaque bloc
    start_line = 0

    while True:
        # Lire les 1000 lignes suivantes de chaque fichier
        lignes_fichier1 = lire_fichier_par_bloc(fichier1, start_line, start_line + taille_bloc)
        lignes_fichier2 = lire_fichier_par_bloc(fichier2, start_line, start_line + taille_bloc)
        
        # Si l'une des listes est vide, arrêter le traitement
        if not lignes_fichier1 or not lignes_fichier2:
            break
        
        # Tracer le graphique pour ce bloc de lignes
        tracer_graphe(lignes_fichier1, lignes_fichier2, start_line)
        
        # Passer au bloc suivant
        start_line += taille_bloc

# Exemple d'appel de la fonction
fichier1 = 'data_eval_protocol1.csv'  # Remplacer par le chemin du premier fichier
fichier2 = 'data_eval_protocol2.csv'  # Remplacer par le chemin du deuxième fichier

traiter_fichiers(fichier1, fichier2)