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

Adding some descriptions to functions

Browse Source
This commit is contained in:
Simon Iteanu
2020-06-23 14:56:06 +02:00
committed by Noe Brucy
parent 850cc4991e
commit be67f49d43
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plotter.py
+7 -3
View File
@@ -517,13 +517,18 @@ class Plotter(Aggregator, BaseProcessor):
dmap_vh = dmap_vh_node.read() dmap_vh = dmap_vh_node.read()
dmap_vv = self.save.get_node("/maps/speed_v_{}".format(ax_los)).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 vel_red = self.pp_params.plot.vel_red
radius = self.save.root.maps._v_attrs.radius radius = self.save.root.maps._v_attrs.radius
center = self.save.root.maps._v_attrs.center center = self.save.root.maps._v_attrs.center
lbox = self.save.root._v_attrs.lbox lbox = self.save.root._v_attrs.lbox
map_vh_red = dmap_vh[::vel_red, ::vel_red] # take only a subset of velocities map_vh_red = dmap_vh[::vel_red, ::vel_red] * unit_old.express(
map_vv_red = dmap_vv[::vel_red, ::vel_red] 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] nh = map_vh_red.shape[0]
nv = map_vv_red.shape[1] nv = map_vv_red.shape[1]
vec_h = ( vec_h = (
@@ -538,7 +543,6 @@ class Plotter(Aggregator, BaseProcessor):
min_v = np.min(norm_v) min_v = np.min(norm_v)
Q = P.quiver(hh, vv, map_vh_red, map_vv_red, units="width", **kwargs) 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: if key_v is None:
key_v = (max_v + min_v) / 2.0 key_v = (max_v + min_v) / 2.0