noe/pipeline
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[log extractor] Add parser for fine step and plot

Browse Source
This commit is contained in:
Noe Brucy
2021-05-19 17:22:50 +02:00
parent fe62be0550
commit d5c8241e1d
2 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plotter.py
+13 -4
View File
@@ -1208,13 +1208,17 @@ class Plotter(Aggregator, BaseProcessor):
plt.xlim(tmin, tmax)
plt.ylim(ylim)
def _gen_from_log(self, logrule, name, description="Generated"):
self.rules[name] = PlotRule(
def _gen_from_log(self, logrule, name_y, name_x="time", description="Generated"):
if name_x == "time":
name_rule = name_y
else:
name_rule = name_y + "_" + name_x
self.rules[name_rule] = PlotRule(
self,
partial(
self._plot,
"/series/" + logrule + "/time",
"/series/" + logrule + "/" + name,
"/series/" + logrule + "/" + name_x,
"/series/" + logrule + "/" + name_y,
xunit=U.Myr,
),
description=description,
@@ -1677,6 +1681,11 @@ class Plotter(Aggregator, BaseProcessor):
for name in ["step", "mcons", "econs", "epot", "ekin", "eint", "emag"]:
self._gen_from_log("cons_from_log", name)
for name in ["fine_step", "dt", "a", "mempc1", "mempc2"]:
self._gen_from_log("fine_step_from_log", name)
for name in ["time", "dt", "a", "mempc1", "mempc2"]:
self._gen_from_log("fine_step_from_log", name_y=name, name_x="fine_step")
# Generic rules directly from Ramses fields
for field in self.pp_params.pymses.variables: