Merge pull request #3282 from chummels/phase_ion

BUG: Set the display name for new fields

yt/visualization/profile_plotter.py

@@ -730,15 +730,16 @@ def _get_field_label(self, field, field_info, field_unit, fractional=False):
  if isinstance(field, tuple):
  field = field[1]
  if field_name is None:
- field_name = r"$\rm{" + field + r"}$"
- field_name = r"$\rm{" + field.replace("_", r"\ ").title() + r"}$"
+ field_name = field_info.get_latex_display_name()
  elif field_name.find("$") == -1:
  field_name = field_name.replace(" ", r"\ ")
  field_name = r"$\rm{" + field_name + r"}$"
  if fractional:
  label = field_name + r"$\rm{\ Probability\ Density}$"
  elif field_unit is None or field_unit == "":
  label = field_name
+ elif r"\frac" in field_unit:
+ label = field_name + r"$\ \ \left(" + field_unit + r"\right)$"
  else:
  label = field_name + r"$\ \ (" + field_unit + r")$"
  return label
@@ -1018,15 +1019,16 @@ def _get_field_label(self, field, field_info, field_unit, fractional=False):
  if isinstance(field, tuple):
  field = field[1]
  if field_name is None:
- field_name = r"$\rm{" + field + r"}$"
- field_name = r"$\rm{" + field.replace("_", r"\ ").title() + r"}$"
+ field_name = field_info.get_latex_display_name()
  elif field_name.find("$") == -1:
  field_name = field_name.replace(" ", r"\ ")
  field_name = r"$\rm{" + field_name + r"}$"
  if fractional:
  label = field_name + r"$\rm{\ Probability\ Density}$"
  elif field_unit is None or field_unit == "":
  label = field_name
+ elif r"\frac" in field_unit:
+ label = field_name + r"$\ \ \left(" + field_unit + r"\right)$"
  else:
  label = field_name + r"$\ \ (" + field_unit + r")$"
  return label
@@ -1074,6 +1076,7 @@ def _setup_plots(self):
  zlim = (None, None)
  xlim = self._xlim
  ylim = self._ylim
+
  if f in self.plots:
  draw_colorbar = self.plots[f]._draw_colorbar
  draw_axes = self.plots[f]._draw_axes