source: java/main/src/main/java/com/framsticks/gui/view/TreeCellRenderer.java @ 77

package com.framsticks.gui.view;

import com.framsticks.gui.ImageProvider;
import com.framsticks.gui.TreeNode;

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

/**
 * Renderer of Tree. Sets appropriate icons for nodes.
 */
@SuppressWarnings("serial")
public class TreeCellRenderer extends DefaultTreeCellRenderer {

    public TreeCellRenderer() {
        setOpenIcon(null);
        setClosedIcon(null);
        setLeafIcon(null);
    }

        @Override
        public Component getTreeCellRendererComponent(JTree tree, Object value,
                                                      boolean sel, boolean expanded, boolean leaf, int row,
                                                      boolean hasFocus) {

                //super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf,
                //              row, hasFocus);
                if (value == null) {
                        return this;
                }
                if (!(value instanceof TreeNode)) {
            setIcon(ImageProvider.loadImage(ImageProvider.SERVER));
            setText("framsticks");
                        return this;
                }
                TreeNode treeNode = (TreeNode)value;
        assert treeNode.getFrame().isActive();
        setToolTipText(treeNode.getTooltip());
        setIcon(ImageProvider.loadImage(treeNode.getIconName()));
        setText(treeNode.getName());
                return this;
        }

        public static String findIconName(String nodeName, String path) {
        if (path == null || nodeName == null) {
            return null;
        }
        if (path.equals("/")) {
                        return ImageProvider.SERVER;
                }
                if (path.endsWith("simulator")) {
                        return ImageProvider.SIMULATOR;
                }
                if (path.endsWith("cli")) {
                        return ImageProvider.CLI;
                }
                if (path.endsWith("world")) {
                        return ImageProvider.WORLD;
                }
                if (path.endsWith("genepools")) {
                        return ImageProvider.GENEPOOLS;
                }
                if (path.endsWith("populations")) {
                        return ImageProvider.POPULATIONS;
                }
                if (path.endsWith("experiment")) {
                        return ImageProvider.EXPERIMENT;
                }
                if (path.endsWith("stats")) {
                        return ImageProvider.STATISTIC;
                }
                if (nodeName.equals("creatures") || path.endsWith("creatures")) {
                        return ImageProvider.CREATURES_GROUP;
                }
                if (nodeName.equals("Favourite Fields")) {
                        return ImageProvider.FAVORITE_FIELDS;
                }
                if (nodeName.equals("genotypes") || path.endsWith("creatures")) {
                        return ImageProvider.GENOTYPES_GROUP;
                }
                if (nodeName.equals("parts") || path.endsWith("parts")) {
                        return ImageProvider.PART_GROUP;
                }
                if (nodeName.equals("joints") || path.endsWith("joints")) {
                        return ImageProvider.JOINT_GROUP;
                }
                if (nodeName.equals("neurodefs") || path.endsWith("neurodefs")) {
                        return ImageProvider.NEURON_DEF_GROUP;
                }
                if (nodeName.equals("mechparts") || path.endsWith("mechparts")) {
                        return ImageProvider.MECH_PART_GROUP;
                }
                if (nodeName.equals("mechjoints") || path.endsWith("mechjoints")) {
                        return ImageProvider.MECH_JOINT_GROUP;
                }
                if (nodeName.equals("neurons") || path.endsWith("neurons")) {
                        return ImageProvider.NEURON_GROUP;
                }
                if (path.matches("^.*parts/[0-9]+$")) {
                        return ImageProvider.PART;
                }
                if (path.matches("^.*joints/[0-9]+$")) {
                        return ImageProvider.JOINT;
                }
                if (path.matches("^.*neurodefs/[0-9]+$")) {
                        return ImageProvider.NEURON_DEF;
                }
                if (path.matches("^.*mechparts/[0-9]+$")) {
                        return ImageProvider.MECH_PART;
                }
                if (path.matches("^.*mechjoints/[0-9]+$")) {
                        return ImageProvider.MECH_JOINT;
                }
                if (path.matches("^.*neurons/[0-9]+$")) {
                        return ImageProvider.NEURON;
                }
                if (path.matches("^.*creatures/c[0-9]+$")) {
                        return ImageProvider.CREATURE;
                }
                if (path.matches("^.*genotypes/g[0-9]+$")) {
                        return ImageProvider.GENOTYPES;
                }
                if (path.matches("^.*populations/groups/[0-9]+$")) {
                        return ImageProvider.POPULATION_GROUP;
                }
                if (path.matches("^.*genepools/groups/[0-9]+$")) {
                        return ImageProvider.GENEPOOLS_GROUP;
                }
                if (path.matches("^.*events/e[0-9]+$")) {
                        return ImageProvider.EVENT;
                }
                return ImageProvider.SERVER;
        }
        
        
}