GUI improvements

2020-05-13 18:23:54 +02:00
parent 0818372e0c
commit bb6f1091eb
2 changed files with 80 additions and 22 deletions
@@ -118,7 +118,15 @@ class Plotter(Aggregator, BaseProcessor):
        )

    def _process_rule(
-        self, name, rule, arg, overwrite=False, ax=None, movie=False, **kwargs
+        self,
+        name,
+        rule,
+        arg,
+        overwrite=False,
+        ax=None,
+        movie=False,
+        from_cells=False,
+        **kwargs
    ):
        if not arg is None:
            name_full = name + "_" + str(arg)
@@ -143,7 +151,12 @@ class Plotter(Aggregator, BaseProcessor):
                    for num in nums:
                        plot_filename = self._find_filename(name_full, run, num)

-                        if rule.kind == "classic":
+                        if from_cells or rule.kind == "cells":
+                            if not os.exists(self.pp[run][num].cells_filename):
+                                self.pp[run][num].load_cells()
+                                self.pp[run][num].unload_cells()
+                            save = tables.open_file(self.pp[run][num].cells_filename)
+                        elif rule.kind == "classic":
                            save = tables.open_file(self.pp[run][num].filename)
                        else:
                            save = tables.open_file(self.comp.filename, "r")