jumeaux-numeriques/twin_plot.py

217 lines
7.9 KiB
Python

import csv # Creating/parsing CSV file
import xml.etree.ElementTree as ET # Creating/parsing XML file
###############################################################################
# twin_plot.py
# @title: Visualisation des données (Matplotlib)
# @project: Blender-EduTech
# @lang: fr
# @authors: Philippe Roy <philippe.roy@ac-grenoble.fr>
# @copyright: Copyright (C) 2023 Philippe Roy
# @license: GNU GPL
###############################################################################
###############################################################################
# Configuration des plots
###############################################################################
def plot_config_get_enable():
twin_config = ET.parse('twin_config.xml').getroot()
if twin_config[1][0].text == "True":
return True
else:
return False
##
# Génère le dictionnaire plot_config à partir du fichier XML plot_config.xml
##
def plot_config_get_dict():
plot_config_tree = ET.parse('plot_config.xml').getroot()
plot_config_tmp={}
for var in plot_config_tree[0][0]:
var_dict={}
for i in range (len(var)):
if var[i] is not None:
var_dict.update({var[i].tag : var[i].text})
plot_config_tmp.update({var.text: var_dict})
return plot_config_tmp
##
# Returne une valeur du dictionnaire plot_config
##
def plot_config_get(plot_config, var, key):
if key in plot_config[var]:
return plot_config[var][key]
else:
return None
##
# Nombre de groupe de courbe(s)
##
def plot_nb(plot_config):
plot_config_list=list(plot_config)
nbgroup = 0
# Regroupement ('group' > 0)
for var in plot_config_list:
if ('group' in plot_config[var]):
if nbgroup < int(plot_config[var]['group']):
nbgroup = int(plot_config[var]['group'])
# Plot solitaire ('group' = 0)
# 'group' = -1 -> variable non affichée
for var in plot_config_list:
if ('group' in plot_config[var]):
if int(plot_config[var]['group']) == 0:
nbgroup +=1
return nbgroup
###############################################################################
# Fichier CSV
###############################################################################
##
# Fichier CSV -> Tableau X,Y
##
def csv_read(file):
# Lecture fichier CSV
fields = []
rows = []
with open(file, newline='') as csv_buff:
csv_reader = csv.reader(csv_buff, delimiter=';')
fields = next(csv_reader)
for row in csv_reader:
rows.append(row)
# Mise en tableau à deux colonnes (xdata,ydata)
xdata=[]
ydata=[]
i=0
for field in fields:
xdata_row=[]
ydata_row=[]
for row in rows:
xdata_row.append(float(row[0].replace(',', '.'))) # Revenir au format US des décimaux
ydata_row.append(float(row[i].replace(',', '.'))) # Revenir au format US des décimaux
xdata.append(xdata_row)
ydata.append(ydata_row)
i+=1
return fields, xdata, ydata
###############################################################################
# Création des graphiques
###############################################################################
# Création d'une courbe
def plot_draw(plt, var, xdata, ydata, plot_config):
# Courbe d'une variable analogique (continue)
if plot_config_get(plot_config, var, 'type') =='a':
plt.plot(xdata, ydata, label=var, color=plot_config_get(plot_config, var, 'color'), linewidth=plot_config_get(plot_config, var, 'linewidth'),
linestyle=plot_config_get(plot_config, var, 'linestyle'), marker=plot_config_get(plot_config, var, 'marker'))
# Courbe d'une variable binaire (discret)
if plot_config_get(plot_config, var, 'type') =='d':
plt.plot(xdata, ydata, label=var, drawstyle='steps-post', color=plot_config_get(plot_config, var, 'color'), linewidth=plot_config_get(plot_config, var, 'linewidth'),
linestyle=plot_config_get(plot_config, var, 'linestyle'), marker=plot_config_get(plot_config, var, 'marker'))
plt.set_ylim(0,1) # niveau 0 et niveau 1
# Courbe d'une variable numérique (discret)
if plot_config_get(plot_config, var, 'type') =='n':
plt.plot(xdata, ydata, label=var, drawstyle='steps-post', color=plot_config_get(plot_config, var, 'color'), linewidth=plot_config_get(plot_config, var, 'linewidth'),
linestyle=plot_config_get(plot_config, var, 'linestyle'), marker=plot_config_get(plot_config, var, 'marker'))
# Création des graphiques statiques
def plots_static(canvas, fields, xdata, ydata, plot_config, title):
# Configuration des plots activée
plot_config_enable=plot_config_get_enable()
# Plots
# Groupe de plots : si group = 0 -> nouveau groupe, plot solitaire
# si group = -1 -> pas de plot
# si group > 0 -> numéro du groupe
plt_i=0 # Compteur de plot
plt_grp=[] # Groupe de plot [[numéro du plot, groupe du fichier CSV]]
for i in range(len(fields)):
var = fields[i]
plt_current=-1 # Numéro du plot à créer
if ('group' in plot_config[var]): # Pas de Plot
if int(plot_config[var]['group']) !=-1: # Pas de Plot
# Plot solitaire
if int(plot_config[var]['group']) ==0:
plt_current = plt_i
plt_grp.append([plt_i, 0])
plt_i +=1
# Plot groupé
else:
plt_new = True # Flag d'un nouveau groupe
for j in range(len(plt_grp)):
if plt_grp[j][1] == int(plot_config[var]['group']): # Groupe déjà existant
plt_current = plt_grp[j][0]
plt_new = False
break
# Nouveau groupe
if plt_new:
plt_current = plt_i
plt_grp.append([plt_i, int(plot_config[var]['group'])])
plt_i +=1
# Création du plot unique
if plot_nb(plot_config) ==1:
if plot_config_enable: # Configuration des plots activée
plot_draw(canvas.plt, var, xdata[i], ydata[i], plot_config)
else:
canvas.plt.plot(xdata[i], ydata[i], '.-', label=var) # Configuration matplotlib par défaut
# Légende ou titre d'axe y
if plt_grp[plt_current][1]==0:
canvas.plt.set_ylabel(var)
else:
canvas.plt.legend()
# Création des subplots
if plot_nb(plot_config) >1:
if plot_config_enable: # Configuration des plots activée
plot_draw(canvas.plt[plt_current], var, xdata[i], ydata[i], plot_config)
else:
canvas.plt[plt_current].plot(xdata[i], ydata[i], '.-', label=var) # Configuration matplotlib par défaut
# Légende ou titre d'axe y
if plt_grp[plt_current][1]==0:
canvas.plt[plt_current].set_ylabel(var)
else:
canvas.plt[plt_current].legend()
# Décoration
if plot_nb(plot_config) ==1: # 1 plot
canvas.plt.set_xlabel("Temps (s)")
canvas.plt.set_title(title)
canvas.plt.axhline(linewidth=1, color='k')
canvas.plt.grid(True, linestyle='--')
# canvas.plt.legend()
else: # Plusieurs plots
canvas.plt[plt_i-1].set_xlabel("Temps (s)")
canvas.plt[0].set_title(title)
for i in range (plt_i):
canvas.plt[i].axhline(linewidth=1, color='k')
canvas.plt[i].grid(True, linestyle='--')
# canvas.plt[i].legend()
# Redraw
canvas.draw()