mirror of
https://forge.apps.education.fr/blender-edutech/jumeaux-numeriques.git
synced 2024-01-27 06:56:18 +01:00
236 lines
8.2 KiB
Python
236 lines
8.2 KiB
Python
import sys
|
|
import importlib
|
|
import csv
|
|
import xml.etree.ElementTree as ET # Creating/parsing XML file
|
|
|
|
import wx
|
|
import matplotlib
|
|
from matplotlib.figure import Figure
|
|
from matplotlib.backends.backend_wxagg import (FigureCanvasWxAgg as FigureCanvas, NavigationToolbar2WxAgg as NavigationToolbar)
|
|
|
|
###############################################################################
|
|
# twin_plot_wx.py
|
|
# @title: Visualisation des données (wxPython + Matplotlib)
|
|
# @project: Blender-EduTech
|
|
# @lang: fr
|
|
# @authors: Philippe Roy <philippe.roy@ac-grenoble.fr>
|
|
# @copyright: Copyright (C) 2023 Philippe Roy
|
|
# @license: GNU GPL
|
|
###############################################################################
|
|
|
|
# Lecture des configurations
|
|
twin_config = ET.parse('twin_config.xml').getroot()
|
|
plot_config_tree = ET.parse('plot_config.xml').getroot()
|
|
plot_config={}
|
|
|
|
###############################################################################
|
|
# Configuration des graphiques
|
|
###############################################################################
|
|
|
|
##
|
|
# Génère le dictionnaire plot_config à partir du fichier XML plot_config.xml
|
|
##
|
|
|
|
def plot_config_dict():
|
|
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.update({var.text: var_dict})
|
|
|
|
##
|
|
# Returne une valeur du dictionnaire plot_config
|
|
##
|
|
|
|
def plot_config_get(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_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
|
|
|
|
|
|
###############################################################################
|
|
# Création des graphiques
|
|
###############################################################################
|
|
|
|
def plot_draw(plt):
|
|
|
|
# Lecture fichier CSV
|
|
fields = []
|
|
rows = []
|
|
with open(sys.argv[1], 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
|
|
|
|
# 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() ==1:
|
|
if twin_config[1][0].text == "True": # Configuration des plots activée
|
|
plt.plot(xdata[i], ydata[i], label=var, color=plot_config_get(var, 'color'), linewidth=plot_config_get(var, 'linewidth'),
|
|
linestyle=plot_config_get(var, 'linestyle'), marker=plot_config_get(var, 'marker'))
|
|
else:
|
|
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:
|
|
plt.set_ylabel(var)
|
|
else:
|
|
plt.legend()
|
|
|
|
# Création des subplots
|
|
if plot_nb() >1:
|
|
if twin_config[1][0].text == "True": # Configuration des plots activée
|
|
plt[plt_current].plot(xdata[i], ydata[i], label=var, color=plot_config_get(var, 'color'), linewidth=plot_config_get(var, 'linewidth'),
|
|
linestyle=plot_config_get(var, 'linestyle'), marker=plot_config_get(var, 'marker'))
|
|
else:
|
|
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:
|
|
plt[plt_current].set_ylabel(var)
|
|
else:
|
|
plt[plt_current].legend()
|
|
|
|
# Décoration
|
|
if plot_nb() ==1: # 1 plot
|
|
plt.set_xlabel("Temps (s)")
|
|
# self.canvas.plt[0].set_ylabel("Valeurs")
|
|
plt.set_title(sys.argv[1])
|
|
plt.axhline(linewidth=1, color='k')
|
|
plt.grid(True, linestyle='--')
|
|
# self.canvas.plt.legend()
|
|
else: # Plusieurs plots
|
|
plt[plt_i-1].set_xlabel("Temps (s)")
|
|
# self.canvas.plt[0].set_ylabel("Valeurs")
|
|
plt[0].set_title(sys.argv[1])
|
|
for i in range (plt_i):
|
|
plt[i].axhline(linewidth=1, color='k')
|
|
plt[i].grid(True, linestyle='--')
|
|
# self.canvas.plt[i].legend()
|
|
|
|
|
|
###############################################################################
|
|
# Zone de dessin
|
|
###############################################################################
|
|
|
|
|
|
class CanvasFrame(wx.Frame):
|
|
|
|
def __init__(self):
|
|
super().__init__(None, -1, 'Visualisation des données')
|
|
|
|
# Configuration des graphiques
|
|
plot_config_dict()
|
|
|
|
# Création des zones graphique (Plots)
|
|
self.figure = Figure()
|
|
if plot_nb() ==1: # plot_nb() : nombre de graphiques
|
|
plt = self.figure.subplots() # plot_nb() : nombre de graphiques
|
|
else:
|
|
plt = self.figure.subplots(plot_nb(), 1, sharex=True) # plot_nb() : nombre de graphiques
|
|
|
|
# Dessin
|
|
plot_draw(plt)
|
|
|
|
# Implantation de la fenêtre (canvas)
|
|
self.canvas = FigureCanvas(self, -1, self.figure)
|
|
self.sizer = wx.BoxSizer(wx.VERTICAL)
|
|
self.sizer.Add(self.canvas, 1, wx.TOP | wx.LEFT | wx.EXPAND)
|
|
self.add_toolbar()
|
|
self.SetSizer(self.sizer)
|
|
self.Fit()
|
|
|
|
def add_toolbar(self):
|
|
self.toolbar = NavigationToolbar(self.canvas)
|
|
self.toolbar.Realize()
|
|
self.sizer.Add(self.toolbar, 0, wx.LEFT | wx.EXPAND)
|
|
self.toolbar.update()
|
|
|
|
|
|
###############################################################################
|
|
# Application
|
|
###############################################################################
|
|
|
|
class App(wx.App):
|
|
def OnInit(self):
|
|
frame = CanvasFrame()
|
|
frame.Show(True)
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
app = App()
|
|
app.MainLoop()
|