Tweak sub/superscript spacing implementation.

anntzer · anntzer · commit feecd7c2a21e · 2025-11-01T23:24:12.000+01:00
Rename _in_subscript_or_superscript to the more descriptive
_needs_space_after_subsuper; simplify its setting in operatorname();
avoid the need to introduce an extra explicitly-typed spaced_nucleus
variable.
diff --git a/lib/matplotlib/_mathtext.py b/lib/matplotlib/_mathtext.py
@@ -2147,7 +2147,7 @@ def csnames(group: str, names: Iterable[str]) -> Regex:
         self._math_expression = p.math
 
         # To add space to nucleus operators after sub/superscripts
-        self._in_subscript_or_superscript = False
+        self._needs_space_after_subsuper = False
 
     def parse(self, s: str, fonts_object: Fonts, fontsize: float, dpi: float) -> Hlist:
         """
@@ -2165,7 +2165,7 @@ def parse(self, s: str, fonts_object: Fonts, fontsize: float, dpi: float) -> Hli
             # explain becomes a plain method on pyparsing 3 (err.explain(0)).
             raise ValueError("\n" + ParseException.explain(err, 0)) from None
         self._state_stack = []
-        self._in_subscript_or_superscript = False
+        self._needs_space_after_subsuper = False
         # prevent operator spacing from leaking into a new expression
         self._em_width_cache = {}
         ParserElement.reset_cache()
@@ -2275,7 +2275,7 @@ def symbol(self, s: str, loc: int,
             prev_char = next((c for c in s[:loc][::-1] if c != ' '), '')
             # Binary operators at start of string should not be spaced
             # Also, operators in sub- or superscripts should not be spaced
-            if (self._in_subscript_or_superscript or (
+            if (self._needs_space_after_subsuper or (
                     c in self._binary_operators and (
                     len(s[:loc].split()) == 0 or prev_char in {
                         '{', *self._left_delims, *self._relation_symbols}))):
@@ -2381,13 +2381,9 @@ def operatorname(self, s: str, loc: int, toks: ParseResults) -> T.Any:
             # Add thin space except when followed by parenthesis, bracket, etc.
             hlist_list += [self._make_space(self._space_widths[r'\,'])]
         self.pop_state()
-        # if followed by a super/subscript, set flag to true
-        # This flag tells subsuper to add space after this operator
-        if next_char in {'^', '_'}:
-            self._in_subscript_or_superscript = True
-        else:
-            self._in_subscript_or_superscript = False
-
+        # If followed by a sub/superscript, set flag to true to tell subsuper
+        # to add space after this operator.
+        self._needs_space_after_subsuper = next_char in {'^', '_'}
         return Hlist(hlist_list)
 
     def start_group(self, toks: ParseResults) -> T.Any:
@@ -2500,9 +2496,9 @@ def subsuper(self, s: str, loc: int, toks: ParseResults) -> T.Any:
             result = Hlist([
                 vlt,
                 *([self._make_space(self._space_widths[r'\,'])]
-                  if self._in_subscript_or_superscript else []),
+                  if self._needs_space_after_subsuper else []),
             ])
-            self._in_subscript_or_superscript = False
+            self._needs_space_after_subsuper = False
             return [result]
 
         # We remove kerning on the last character for consistency (otherwise
@@ -2594,12 +2590,12 @@ def subsuper(self, s: str, loc: int, toks: ParseResults) -> T.Any:
 
         # Do we need to add a space after the nucleus?
         # To find out, check the flag set by operatorname
-        spaced_nucleus: list[Node] = [nucleus, x]
-        if self._in_subscript_or_superscript:
-            spaced_nucleus += [self._make_space(self._space_widths[r'\,'])]
-            self._in_subscript_or_superscript = False
-
-        result = Hlist(spaced_nucleus)
+        result = Hlist([
+            nucleus, x,
+            *([self._make_space(self._space_widths[r'\,'])]
+              if self._needs_space_after_subsuper else []),
+        ])
+        self._needs_space_after_subsuper = False
         return [result]
 
     def _genfrac(self, ldelim: str, rdelim: str, rule: float | None, style: _MathStyle,
