From be67f49d43a56158256a1da48cd249cd950c1b2c Mon Sep 17 00:00:00 2001 From: Simon Iteanu Date: Tue, 23 Jun 2020 14:56:06 +0200 Subject: [PATCH] Adding some descriptions to functions --- plotter.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/plotter.py b/plotter.py index 2211e1c..87b144e 100644 --- a/plotter.py +++ b/plotter.py @@ -517,13 +517,18 @@ class Plotter(Aggregator, BaseProcessor): dmap_vh = dmap_vh_node.read() dmap_vv = self.save.get_node("/maps/speed_v_{}".format(ax_los)).read() + label, unit_old, unit = self._ax_label_unit(dmap_vh_node, "", unit, unit_coeff) + vel_red = self.pp_params.plot.vel_red radius = self.save.root.maps._v_attrs.radius center = self.save.root.maps._v_attrs.center lbox = self.save.root._v_attrs.lbox - map_vh_red = dmap_vh[::vel_red, ::vel_red] # take only a subset of velocities - map_vv_red = dmap_vv[::vel_red, ::vel_red] + map_vh_red = dmap_vh[::vel_red, ::vel_red] * unit_old.express( + unit + ) # take only a subset of velocities + map_vv_red = dmap_vv[::vel_red, ::vel_red] * unit_old.express(unit) + nh = map_vh_red.shape[0] nv = map_vv_red.shape[1] vec_h = ( @@ -538,7 +543,6 @@ class Plotter(Aggregator, BaseProcessor): min_v = np.min(norm_v) Q = P.quiver(hh, vv, map_vh_red, map_vv_red, units="width", **kwargs) - label, unit_old, unit = self._ax_label_unit(dmap_vh_node, "", unit, unit_coeff) if key_v is None: key_v = (max_v + min_v) / 2.0