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

[plotter] [snap] improve vector overlays

Browse Source
This commit is contained in:
Noe Brucy
2021-11-24 19:05:23 +01:00
parent b7512bd5ed
commit 74c2f40bd2
2 changed files with 14 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plotter.py
+14 -41
View File
@@ -946,18 +946,21 @@ class Plotter(Aggregator, BaseProcessor):
self, name, ax_los, extent, unit=U.km_s, unit_coeff=1.0, reduce_res=1, kind="quiver", **kwargs
):
"""
Add an overlay : velocity vector field
Add an overlay : vector field
"""
self.current_processor.process(f"{name}_h", ax_los)
self.current_processor.process(f"{name}_v", ax_los)
ax_h = self._axes_h[ax_los]
ax_v = self._axes_v[ax_los]
map_h = self.current_processor.get_value("/maps/speed_h_{}".format(ax_los))
map_v = self.current_processor.get_value("/maps/speed_v_{}".format(ax_los))
label, unit_old, unit = self._ax_label_unit(f"/maps/speed_h_{ax_los}", "", unit, unit_coeff)
self.current_processor.process(f"slice_{name}{ax_h}", ax_los)
self.current_processor.process(f"slice_{name}{ax_v}", ax_los)
map_h = self.current_processor.get_value(f"/maps/slice_{name}{ax_h}_{ax_los}")
map_v = self.current_processor.get_value(f"/maps/slice_{name}{ax_v}_{ax_los}")
label, unit_old, unit = self._ax_label_unit(f"/maps/slice_{name}{ax_h}_{ax_los}", "", unit, unit_coeff)
# take only a subset of velocities
# take only a subset
map_h = map_h[::reduce_res, ::reduce_res] * unit_old.express(unit)
map_v = map_v[::reduce_res, ::reduce_res] * unit_old.express(unit)
@@ -969,7 +972,7 @@ class Plotter(Aggregator, BaseProcessor):
line_integral_convolution(plt.gca(), map_h, map_v, extent=extent, **kwargs)
def _overlay_speed(self, ax_los, extent, **kwargs):
self._overlay_vector("speed", ax_los, extent, **kwargs)
self._overlay_vector("vel", ax_los, extent, **kwargs)
def _overlay_B(self, ax_los, extent, **kwargs):
self._overlay_vector("B", ax_los, extent, **kwargs)
@@ -1453,31 +1456,7 @@ class Plotter(Aggregator, BaseProcessor):
overlays=[self._overlay_speed],
),
"Density slice with speed overlay",
dependencies=["slice_rho", "speed_h", "speed_v"],
),
"slice_rho_B": PlotRule(
self,
partial(
self._plot_map,
"slice_rho",
label=r"$\rho$",
unit=U.Msun_pc3,
overlays=[self._overlay_B],
),
"Density slice with magnetic field overlay",
dependencies=["slice_rho", "B_h", "B_v"],
),
"slice_rho_B_vel": PlotRule(
self,
partial(
self._plot_map,
"slice_rho",
label=r"$\rho$",
unit=U.Msun_pc3,
overlays=[self._overlay_B, self._overlay_speed],
),
"Density slice with magnetic field and velocity overlay",
dependencies=["slice_rho", "B_h", "B_v", "speed_h", "speed_v"],
dependencies=["slice_rho"],
),
"jeans_ratio": PlotRule(
self,
@@ -1548,14 +1527,6 @@ class Plotter(Aggregator, BaseProcessor):
"P-PDF on a 2D slice ",
dependencies=["P_pdf"],
),
"B_int": PlotRule(
self,
partial(
self._plot_map, "B_int", label=r"$\mid \mathrm{B} \mid$", unit=U.T
),
"Magnetic intensity map",
dependencies=["B_int"],
),
"Brho": PlotRule(
self,
partial(
@@ -1902,7 +1873,9 @@ class Plotter(Aggregator, BaseProcessor):
# Dict of overlays
self.overlays = {
"g": partial(self._overlay_vector, "g"),
"B": self._overlay_B,
"vel": self._overlay_speed,
"speed": self._overlay_speed,
"levels": self._overlay_levels,
"contour": self._overlay_contour,