main_pyplot_script.py

# Python 3.10
# UTF-8
# Alexandre Doneux

from matplotlib import pyplot
from pack_points.point import Point
from pack_points.figure import Figure
from random import randint
import argparse


my_parser = argparse.ArgumentParser(description="Commande permettant de lire dans un fichier txt d'un certain format. "
                                                "On affiche avec matplotlib les points et les figurees qu'ils créent "
                                                "quand ils sont regroupés")
my_parser.add_argument('-file', type=str, help="Fichier dans lequel on va aller chercher les groupes de points pour "
                                               "créer les figures.")
my_parser.add_argument('-savefile', type=str, help="Fichier dans lequel on va sauver le png du résultat.")
args = my_parser.parse_args()

file_name = args.file.strip()
try:
    save_file = args.savefile.strip()
except:
    # si rien n'est mis en paramètre pour le fichier où sauver le résultat: on mets au même endroit que le
    # fichier de base
    save_file = file_name[:-3] + "png"

# Lignes sous la forme: "nomPoint x y"


class LineSyntaxError(Exception):
    """
    Erreur pour indiquer un problème de format sur une certaine ligne.
    """
    def __int__(self, line="None"):
        self.line = line


def clean_lines(all_lines):
    """
    Fonction supprimant les valeurs doublons dans une liste de lignes d'un fichier.
    :param - all_lines: Liste contenant les lignes du fichier texte analysé.
    :return - clean_lines: Liste des lignes sans doublons
    """
    new_lines = []  # liste des lignes sans les doublons
    first_line = True
    for line in all_lines:
        line = line.strip("\n")
        line = line.strip()  # .strip() -> on enlève les espaces en trop au début et à la fine

        if first_line:
            previous_line = line
            first_line = False
            new_lines.append(line)
            continue  # on ne peut pas verifier un doublon à la première ligne

        # On saute l'écriture dans la nouvelle liste les lignes en doublon l'une à la suite des autres.
        # Notamment les doubles lignes vides
        if line == previous_line:
            continue

        new_lines.append(line)
        previous_line = line

    return new_lines

def draw(main_list):
    """

    """
    color_list = ["goldenrod", "blueviolet", "crimson", "navy", "seagreen", "steelblue", "maroon", "coral",
                  "olivedrab", "lime", "darkorange", "aqua", "dimgray", "cornflowerblue"]

    for group in main_list:
        draw_color = color_list[randint(0, len(color_list) - 1)]
        if len(group) == 1:
            group[0].draw_point(markerfacecolor=draw_color)  # Quand un seul point on le dessine avec Point.draw_point()
            continue
        my_figure = Figure(group)
        my_figure.draw_figure(color=draw_color)

def group_figures(all_lines):
    """
    Fonction qui prends en paramètres une liste des lignes du fichier à analyser. Renvoie une liste de listes. La
    fonction supprime les lignes doublons les unes à la suite des autres. Les différentes listes représentent les
    groupements de figures. Les groupements sont déterminés par les lignes vides dans le fichier texte analysé.
    :param - all_lines: Liste des lignes du fichier.
    :return - grouped_points: Liste contenant une liste pour chaque groupement de points.
    """
    main_list = []  # liste qui contiendra les groupes
    i = 0
    # indice de main_list. On aura main_list = [[point1, point2],[point3],[........],[.........]]
    # l'indice augmentera de 1 lorsqu'on tombe sur une liste vide -> début d'une nouvelle figure

    maxx = 0
    minx = 0
    maxy = 0
    miny = 0

    new_lines = clean_lines(all_lines)

    # on retire la première ligne si c'est une ligne vide
    if new_lines[0] == "":
        new_lines = new_lines[1:]

    # on pré-construit l'array final avec les groupements de points
    number_of_groups = new_lines.count("")
    for j in range(number_of_groups):
        main_list.append([])

    for line in new_lines:
        if line == "":
            i += 1  # on commence une nouvelle sous-liste
            continue

        line = line.split()  # division en une liste

        # Vérification de si la liste a la bonne taille.
        if len(line) != 3:
            print("Erreur taille")
            raise LineSyntaxError(line)


        # marche pas pour l'instant
        # Vérification de si les valeurs dans une ligne sont justes
        try:
            float(line[1])
            float(line[2])
        except ValueError:
            print("Erreur type")
            raise LineSyntaxError(line)


        line[1] = float(line[1])
        line[2] = float(line[2])

        if line[1] > maxx:
            maxx = line[1]
        if line[2] > maxy:
            maxy = line[2]
        if line[1] < minx:
            minx = line[1]
        if line[2] < miny:
            miny = line[2]

        # Création du point dans la sous-liste
        main_list[i].append(Point(line[0], line[1], line[2]))

    extremes = [minx, maxx, miny, maxy]

    return main_list, extremes


if __name__ == "__main__":

    try:
        with open(file_name) as file:
            lines = file.readlines()
            lines.append("\n")  # pour être sur que e fichier fini par une ligne au vide. Au pire elle sera supprimée


            main_list = group_figures(lines)[0]
            extremes = group_figures(lines)[1]


    except FileNotFoundError:
        print("Erreur. Le fichier n'existe pas.")
    except IOError:
        print("Erreur IO.")
    except LineSyntaxError as e:
        #print("Erreur de syntaxe à la ligne {}".format(" ".join(e.line)))  # On indique à quelle ligne il y a une erreur
        # Pas réussi à affihcer la ligne ->  ... object has no attribute 'line'
        exit()

    # desiin des points et figures
    draw(main_list)

    pyplot.xlim(extremes[0] - 2, extremes[1] + 2)
    pyplot.ylim(extremes[2] - 2, extremes[3] + 2)
    pyplot.savefig(save_file)
    pyplot.show()