Adicione cores e formas de nós à legenda em tikz-picture usando Forest Tree e pgfplots

Question

Quero adicionar os elementos da minha árvore à legenda (veja a legenda para descrição). eu encontreiesseeesseresposta, mas não explicam como adicionar outras formas à legenda. Não encontrei nada útil nos documentos pgfplots. Não estou restrito a pgfplots para a legenda, se você tiver outra solução terei prazer em usá-la.

\documentclass{standalone} 
\usepackage[dvipsnames,fixpdftex]{xcolor}
\usepackage{forest}
\usetikzlibrary{calc,shadings}

\usepackage{pgfplots}
\pgfplotsset{compat=1.15}

% Code from Christian Feuersänger
% https://tex.stackexchange.com/questions/54794/using-a-pgfplots-style-legend-in-a-plain-old-tikzpicture#54834

% argument #1: any options
\newenvironment{customlegend}[1][]{%
    \begingroup
    % inits/clears the lists (which might be populated from previous
    % axes):
    \csname pgfplots@init@cleared@structures\endcsname
    \pgfplotsset{#1}%
}{%
    % draws the legend:
    \csname pgfplots@createlegend\endcsname
    \endgroup
}%

% makes \addlegendimage available (typically only available within an
% axis environment):
\def\addlegendimage{\csname pgfplots@addlegendimage\endcsname}

%%--------------------------------

% definition to insert numbers
\pgfkeys{/pgfplots/number in legend/.style={%
        /pgfplots/legend image code/.code={%
            \node at (0.295,-0.0225){#1};
        },%
    },
}
\begin{document}

\begin{forest}
for tree={circle,draw,minimum size = 2.25em, l sep=15pt, s sep=3.0em}
[0, fill=Apricot  
    [1, fill=SpringGreen, shape=diamond, for tree={s sep = 3.0em}  
      [$\dots$, draw=none]
    ]
    [2, fill=SpringGreen, shape=star, for tree={s sep = 3.0em}
      [$\dots$, draw=none]
      [8]
    ]
    [3, fill=SpringGreen, shape=diamond      
      [$\dots$, draw=none]
    ]
    [4, fill=Apricot
      [$\dots$, draw=none]      
    ]
    [5, fill=SpringGreen, shape=regular polygon
      [$\dots$, draw=none]      
    ]
]
\begin{customlegend}[
  legend entries={ % <= in the following there are the entries
  Green circle/node here,
  Red circle/node here,
  White Star here, 
  White Diamond here,
  White Polygon here
  },
  legend style={at={(4.5,3.5)},font=\footnotesize}] % <= to define position and font legend
  % the following are the "images" and numbers in the legend
      \addlegendimage{mark=ball,ball color=SpringGreen, draw=white}
      \addlegendimage{mark=ball,ball color=Apricot, draw=white}
      \addlegendimage{stealth-stealth,red,opacity=0.4}
      \addlegendimage{number in legend=1,blue}
      \addlegendimage{number in legend=0,red}
  \end{customlegend}
\end{forest}

\end{document}

Answer 1

Você pode adicionar um estilo para formas na legenda semelhante ao seu number in legendestilo atual, por exemplo chamado shape in legend. O estilo das formas pode ser definido para aceitar dois argumentos, a forma e a cor, usando .style 2 args. Isso é chamado com shape in legend={arg1}{arg2}(vejaComo plotar dados de vários filtros usando pgfplots). O próprio estilo desenha um nó normal que usa a forma e a cor de preenchimento como argumentos.

MWE:

\documentclass{standalone} 
\usepackage[dvipsnames]{xcolor}
\usepackage{forest}
\usetikzlibrary{calc,shadings}

\usepackage{pgfplots}
\pgfplotsset{compat=1.15}

% Code from Christian Feuersänger
% https://tex.stackexchange.com/questions/54794/using-a-pgfplots-style-legend-in-a-plain-old-tikzpicture#54834

% argument #1: any options
\newenvironment{customlegend}[1][]{%
    \begingroup
    % inits/clears the lists (which might be populated from previous
    % axes):
    \csname pgfplots@init@cleared@structures\endcsname
    \pgfplotsset{#1}%
}{%
    % draws the legend:
    \csname pgfplots@createlegend\endcsname
    \endgroup
}%

% makes \addlegendimage available (typically only available within an
% axis environment):
\def\addlegendimage{\csname pgfplots@addlegendimage\endcsname}

%%--------------------------------

% definition to insert numbers and shapes
\pgfkeys{/pgfplots/number in legend/.style={%
        /pgfplots/legend image code/.code={%
            \node at (0.295,-0.0225){#1};
        },%
        },
        /pgfplots/shape in legend/.style 2 args={%
        /pgfplots/legend image code/.code={%
            \node[draw,#1,fill=#2,minimum width=3mm] at (0.295,-0.0225){};
        },%
        },
}
\begin{document}

\begin{forest}
for tree={circle,draw,minimum size = 2.25em, l sep=15pt, s sep=3.0em}
[0, fill=Apricot  
    [1, fill=SpringGreen, shape=diamond, for tree={s sep = 3.0em}  
      [$\dots$, draw=none]
    ]
    [2, fill=SpringGreen, shape=star, for tree={s sep = 3.0em}
      [$\dots$, draw=none]
      [8]
    ]
    [3, fill=SpringGreen, shape=diamond      
      [$\dots$, draw=none]
    ]
    [4, fill=Apricot
      [$\dots$, draw=none]      
    ]
    [5, fill=SpringGreen, shape=regular polygon
      [$\dots$, draw=none]      
    ]
]
\begin{customlegend}[
  legend entries={ % <= in the following there are the entries
  Green circle/node here,
  Red circle/node here,
  White Star here, 
  White Diamond here,
  White Polygon here
  },
  legend style={at={(4.5,3.5)},font=\footnotesize}] % <= to define position and font legend
  % the following are the "images" and numbers in the legend
      \addlegendimage{shape in legend={circle}{SpringGreen}}
      \addlegendimage{shape in legend={circle}{red}}
      \addlegendimage{shape in legend={star}{white}}
      \addlegendimage{shape in legend={diamond}{white}}
      \addlegendimage{shape in legend={regular polygon}{white}}
  \end{customlegend}
\end{forest}

\end{document}

Resultado: