[plotter] [snap] improve vector overlays

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
-56
View File
@@ -910,34 +910,6 @@ class SnapshotProcessor(HDF5Container):
datamap = self._rt.process(self._cam[ax_los], surf_qty=True)
return datamap.map.T
def _vector_h(self, name, unit, ax_los, z=0.0):
h_op = ScalarOperator(
lambda dset: dset[name][:, self._ax_nb[self._axes_h[ax_los]]],
self._ro.info[unit],
)
dmap_h = slicing.SliceMap(self._amr, self._cam[ax_los], h_op, z=z).map.T
return dmap_h
def _vector_v(self, name, unit, ax_los, z=0.0):
v_op = ScalarOperator(
lambda dset: dset[name][:, self._ax_nb[self._axes_v[ax_los]]],
self._ro.info[unit],
)
dmap_v = slicing.SliceMap(self._amr, self._cam[ax_los], v_op, z=z).map.T
return dmap_v
def _speed_h(self, ax_los, z=0.0):
return self._vector_h("vel", "unit_velocity", ax_los, z)
def _speed_v(self, ax_los, z=0.0):
return self._vector_v("vel", "unit_velocity", ax_los, z)
def _B_h(self, ax_los, z=0.0):
return self._vector_h("Br", "unit_mag", ax_los, z)
def _B_v(self, ax_los, z=0.0):
return self._vector_v("Br", "unit_mag", ax_los, z)
def _B_int(self, ax_los, z=0.0):
"""
Slice ont the intensity of the magnetic field
@@ -1595,34 +1567,6 @@ class SnapshotProcessor(HDF5Container):
unit=U.none,
dependencies=["avg_map_coldens", "avg_map_T_mwavg"],
),
"speed_h": Rule(
self,
self._speed_h,
"Horizontal speed slice wrt the line of sight",
"/maps",
unit=self.info["unit_velocity"],
),
"speed_v": Rule(
self,
self._speed_v,
"Vertical speed slice wrt the line of sight",
"/maps",
unit=self.info["unit_velocity"],
),
"B_h": Rule(
self,
self._B_h,
"Horizontal slice of the magnetic field wrt the line of sight",
"/maps",
unit=self.info["unit_mag"],
),
"B_v": Rule(
self,
self._B_v,
"Vertical slice of the magnetic field wrt the line of sight",
"/maps",
unit=self.info["unit_mag"],
),
"T": Rule(
self,
self._temperature,