pgfplots com mapas de cores em escala não linear

Question

Esta é uma pergunta que apareceu na lista de discussão do pgfplots; Respondo aqui pois permite uma resposta de maior qualidade.

---

Eu tenho uma imagem que usa um mapa de cores divergentes.

Neste caso, o mínimo e o máximo não têm o mesmo valor absoluto (mas sim -0,2 e +0,5)

Quero poder fazer um "mapa de cores centralizado", onde “0” é a cor do meio, todos os pontos >0 usam a metade superior do mapa e todos <0 a metade inferior.

A barra de cores deve ser distorcida de acordo com os valores reais (ou seja, a metade inferior do mapa deve ocupar 2/7 da barra, e a metade superior os 5/7 restantes)

\documentclass{standalone}
\usepackage{pgfplots}
\pgfplotsset{compat=1.9} 


\begin{document}
\begin{tikzpicture}

\begin{axis}[
    enlargelimits=false,
    % I want the color to be distributed in a nonlinear way, not like this
    % I want the tick labels to reflect the centered colorbar
    colorbar,
]
    \addplot[line width=3pt,mesh,domain=-0.2:0.5] {x};
\end{axis}
\end{tikzpicture}

\end{document}

talvez uma chave de ponto meta center = [auto,], com auto significando um valor calculado de (ponto meta max + ponto meta min) ÷ 2

Answer 1

É possível escalar o point meta. Naturalmente, isso também dimensionará as colorbardescrições de seus eixos. Mas como a colorbarna verdade nada mais é do que normal axis, podemos definir transformações personalizadas para "desfazer" o efeito.

O código a seguir define um novo estilo nonlinear colormap around 0={<min>}{<max>}que redimensiona o meta do ponto (assumindo que seria a ycoordenada por padrão). Ele também redimensiona a barra de cores de maneira não linear para restaurar as descrições corretas:

\documentclass{standalone}
\usepackage{pgfplots}
\pgfplotsset{compat=1.9} 

\pgfplotsset{
    % this transformation ensures that every input argument is
    % transformed from -0.2 : 0.5 -> -0.5,0.5 
    % and every tick label is transformed back:
    nonlinear colormap trafo/.code 2 args={
        \def\nonlinearscalefactor{((#2)/(#1))}%
        \pgfkeysalso{%
            y coord trafo/.code={%
                \pgfmathparse{##1 < 0 ? -1*##1*\nonlinearscalefactor : ##1}%
            },
            y coord inv trafo/.code={%
                \pgfmathparse{##1 < 0 ? -1*##1/\nonlinearscalefactor : ##1}%
            },
        }%
    },
    nonlinear colormap around 0/.code 2 args={
        \def\nonlinearscalefactor{((#2)/(#1))}%
        \pgfkeysalso{
            colorbar style={
                nonlinear colormap trafo={#1}{#2},
                %
                % OVERRIDE this here. The value is *only* used to
                % generate a nice axis, it does not affect the data.
                % Note that these values will be mapped through the
                % colormap trafo as defined above.
                point meta min={#1},
                point meta max={#2},
            },
            %
            % this here is how point meta is computed for the plot.
            % It means that a point meta of -0.2 will actually become -0.5
            % Thus, the *real* point meta min is -0.5... but we
            % override it above.
            point meta={y < 0 ? -y*\nonlinearscalefactor : y},
        }%
    },
}

\begin{document}
\begin{tikzpicture}

\begin{axis}[
    enlargelimits=false,
    colorbar,
    %
    % activate the nonlinear colormap:
    nonlinear colormap around 0={-0.2}{0.5},
    %
    % reconfigure it - the default yticks are typically unsuitable
    % (because they are chosen in a linear way)
    colorbar style={
        ytick={-0.2,-0.1,0,0.25,0.5},
    },
]
    \addplot[line width=3pt,mesh,domain=-0.2:0.5] {x};
\end{axis}
\end{tikzpicture}

\end{document}