coloração no modo matemático

Question

Quero colorir alguns dos símbolos no modo matemático, mas não seus subscritos. Porém, quando faço isso, os subscritos são quebrados e escapam do texto.

Preparei um exemplo mínimo para demonstrar meu problema:

\documentclass{article}
\usepackage{xcolor}
\begin{document}

$T_i$

${\color{red}T}_i$

$\textcolor[rgb]{1,0,0}{T}_{i}$

\end{document}

O resultado é:

Observe que T e i são separados quando T é colorido. Existe uma maneira de colorir corretamente sem adicionar espaço extra no subscrito?

Answer 1

Em vez de alterar as cores ou fechar um grupo imediatamente antes de um subscrito, altere a cor após inserir o subscrito.

\documentclass{article}
\usepackage{xcolor}
\begin{document}
$T_i$

${\color{red}T_{\color{black}i}}$
\end{document}

Answer 2

Você pode lembrar a cor atual e usá-la xparsepara obter os possíveis subscritos e sobrescritos, que são compostos na cor atual.

\documentclass{article}
\usepackage{xcolor}
\usepackage{xparse}

\NewDocumentCommand{\colornucleus}{omme{_^}}{%
  \begingroup\colorlet{currcolor}{.}%
  \IfValueTF{#1}
   {\textcolor[#1]{#2}}
   {\textcolor{#2}}
    {%
     #3% the nucleus
     \IfValueT{#4}{_{\textcolor{currcolor}{#4}}}% subscript
     \IfValueT{#5}{^{\textcolor{currcolor}{#5}}}% superscript
    }%
  \endgroup
}
\begin{document}

$T_i$ $T^j$ $T_i^j$

$\colornucleus{red}{T}_i$
$\colornucleus{red}{T}^j$
$\colornucleus{red}{T}_i^j$

$\colornucleus[rgb]{1,0,0}{T}_{i}$
$\colornucleus[rgb]{0,1,0}{T}^j$
$\colornucleus[rgb]{0,0,1}{T}_{i}^j$

\textcolor{blue!30!red}{Text $\colornucleus{black}{T}_i$ text}

\end{document}