diff --git a/plotter.py b/plotter.py index 8758edb..09dc37d 100644 --- a/plotter.py +++ b/plotter.py @@ -487,7 +487,7 @@ class Plotter(Aggregator, BaseProcessor): """ if label is None: try: - label = self.current_processor.get_attribute("node_name", "label") + label = self.current_processor.get_attribute(node_name, "label") except KeyError: if os.path.basename(node_name) in self.label_convert: label = self.label_convert[os.path.basename(node_name)] @@ -495,7 +495,7 @@ class Plotter(Aggregator, BaseProcessor): label = os.path.basename(node_name) try: - unit_old = self.current_processor.get_attribute("node_name", "unit") + unit_old = self.current_processor.get_attribute(node_name, "unit") except KeyError: unit_old = U.none @@ -564,7 +564,7 @@ class Plotter(Aggregator, BaseProcessor): ax_h = self._axes_h[ax_los] ax_v = self._axes_v[ax_los] - im_extent = np.array(self.current_processor.im_extent) + im_extent = np.array(self.current_processor.get_attribute("/maps", "im_extent")) unit_length = self.current_processor.info["unit_length"] if embeded: @@ -784,8 +784,8 @@ class Plotter(Aggregator, BaseProcessor): sinks = pd.DataFrame(self.current_processor.get_value("/datasets/sinks")) part_pos = sinks[["x", "y", "z"]].values mass = sinks.M - - if parts: + unit_length /= self.current_processor.lbox + elif parts: # Open particle HDF5 filetype_from_ext self.current_processor.load_parts(keys=["pos", "mass"]) part_pos = self.current_processor.parts.pos