BC: Edit: Add plot display mode 'time' and 'date'.

mesh
Pierre-Antoine Rouby 2023-05-11 16:34:29 +02:00
parent 7a7e714ea4
commit 67af95cb40
2 changed files with 22 additions and 10 deletions

View File

@ -14,13 +14,16 @@ from View.BoundaryCondition.Edit.translate import *
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
class Plot(APlot): class Plot(APlot):
def __init__(self, canvas=None, data=None, toolbar=None): def __init__(self, canvas=None, data=None,
mode = "time", toolbar=None):
super(Plot, self).__init__( super(Plot, self).__init__(
canvas=canvas, canvas=canvas,
data=data, data=data,
toolbar=toolbar toolbar=toolbar
) )
self._mode = mode
def custom_ticks(self): def custom_ticks(self):
t0 = datetime.fromtimestamp(0) t0 = datetime.fromtimestamp(0)
nb = len(self.data.data) nb = len(self.data.data)
@ -36,6 +39,7 @@ class Plot(APlot):
) )
) )
xx = list(map(lambda v: v[0], fx)) xx = list(map(lambda v: v[0], fx))
if self._mode == "time":
xt = list( xt = list(
map( map(
lambda v: str( lambda v: str(
@ -46,6 +50,13 @@ class Plot(APlot):
fx fx
) )
) )
else:
xt = list(
map(
lambda v: str(datetime.fromtimestamp(v[0]).date()),
fx
)
)
self.canvas.axes.set_xticks(ticks=xx, labels=xt, rotation=45) self.canvas.axes.set_xticks(ticks=xx, labels=xt, rotation=45)

View File

@ -96,6 +96,7 @@ class EditBoundaryConditionWindow(ASubMainWindow, ListedSubWindow):
self.plot = Plot( self.plot = Plot(
canvas = self.canvas, canvas = self.canvas,
data = self._data, data = self._data,
mode = self._study.time_system,
) )
self.plot.draw() self.plot.draw()