ModelTreeCellRenderer.java

package jetbrains.mps.ide.platform.modeltree;

/*Generated by MPS */

import com.intellij.ui.ColoredTreeCellRenderer;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.JTree;
import javax.swing.tree.TreeNode;

public class ModelTreeCellRenderer extends ColoredTreeCellRenderer implements TreeCellRenderer {
  public ModelTreeCellRenderer() {
  }
  @Override
  public void customizeCellRenderer(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {
    if (value instanceof TreeNode) {
      ModelTreeNode treeNode = (ModelTreeNode) value;
      append(treeNode.getText());
      setIcon(treeNode.getIcon(expanded));
    }
  }
}