JTree DefaultTreeCellRenderer color based on levels

Question

I am extending DefaultTreeCellRenderer in order to paint Node background based on level, however I am currently only able to get the current row and not the current level.

How can I get the current level inside getTreeCellRendererComponent(JTree aTree, Object aValue, boolean aSelected, boolean aExpanded, boolean aLeaf, int aRow, boolean aHasFocus) function?

In the instance below I am only able to alternate background of nodes based on row, how do I get level on the tree and if level == 1 set background to yellow??

import java.awt.*;
import javax.swing.*;
import javax.swing.tree.*;

public class Testing
{
    public void buildGUI()
    {
        final MyJTree tree = new MyJTree();
        JFrame f = new JFrame();
        f.getContentPane().add(new JScrollPane(tree), BorderLayout.CENTER);
        f.setSize(200, 200);
        f.setLocationRelativeTo(null);
        f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        f.setVisible(true);
    }

    public static void main(String[] args)
    {
        SwingUtilities.invokeLater(new Runnable()
        {
            @Override
            public void run()
            {
                new Testing().buildGUI();
            }
        });
    }
}

class MyJTree extends JTree
{
    public MyJTree()
    {
        super();
        setCellRenderer(new NiceTreeCellRenderer());
    }
}

class NiceTreeCellRenderer extends DefaultTreeCellRenderer
{

    DefaultTreeCellRenderer defaultRenderer = new DefaultTreeCellRenderer();

    @Override
    public Component getTreeCellRendererComponent(JTree aTree, Object aValue, boolean aSelected,
            boolean aExpanded, boolean aLeaf, int aRow, boolean aHasFocus)
    {
        JPanel panel = new JPanel(); // Create a new panel where we will show the data.
        String text = (String) ((DefaultMutableTreeNode) aValue).getUserObject();
        panel.add(new JLabel(text)); // Actually show the data.

        // If the value is not null and is a tree node and a leaf then paint it.
        if ((aValue != null) && (aValue instanceof DefaultMutableTreeNode) && aLeaf)
        {
                if (aRow % 2 == 0)
                {
                    panel.setBackground(Color.WHITE);
                }
                else
                {
                    panel.setBackground(new Color(230, 230, 230));
                }
            panel.setEnabled(aTree.isEnabled());
            return panel;
        }
        // For everything else use default renderer.
        return defaultRenderer.getTreeCellRendererComponent(aTree, aValue, aSelected, aExpanded, aLeaf,
                aRow, aHasFocus);
    }
}

Answer 1

use DefaultMutableTreeNode#getLevel()