cb.util.PetalTree Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.crazybeans Show documentation
Java library to read, modify or create Rational Rose petal files
The newest version!
/**
 * Copyright (c) 2001 Markus Dahm
 * Copyright (C) 2015-2018 BITPlan GmbH http://www.bitplan.com
 *
 * This source is part of
 * https://github.com/BITPlan/CrazyBeans
 * and the license as outlined there applies
 */
package cb.util;

import java.util.Iterator;

import javax.swing.ImageIcon;
import javax.swing.JTree;
import javax.swing.ToolTipManager;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;

import cb.petal.PetalNodeList;
import cb.petal.PetalFile;
import cb.petal.PetalNode;
import cb.petal.PetalObject;
import cb.petal.StringLiteral;
import cb.petal.Tagged;
import cb.petal.Value;

/**
 * Display petal file in JTree. The tooltip text displays the property name, if
 * the currently selected node is a child of a PetalObject.
 *
 * @version $Id: PetalTree.java,v 1.1 2001/07/06 11:09:39 dahm Exp $
 * @author M. Dahm
 */
public class PetalTree extends JTree {
  /**
   * 
   */
  private static final long serialVersionUID = 3162615092990086860L;
  protected static final ImageIcon PLUS_ICON = getIcon("PLUS.GIF");
  protected static final ImageIcon MINUS_ICON = getIcon("MINUS.GIF");
  protected static final ImageIcon OPENED_ICON = getIcon("OPENED.GIF");
  protected static final ImageIcon CLOSED_ICON = getIcon("CLOSED.GIF");
  protected static final ImageIcon LEAF_ICON = getIcon("LEAF.GIF");

  /**
   * construct me from the given file
   * @param tree - the file to construct me from
   */
  public PetalTree(PetalFile tree) {
    super(tree);
    getSelectionModel().setSelectionMode(
        TreeSelectionModel.SINGLE_TREE_SELECTION);
    ToolTipManager.sharedInstance().registerComponent(this);
    putClientProperty("JTree.lineStyle", "Angled");
    setShowsRootHandles(false);
    setUI(new MyTreeUI());
    setCellRenderer(new MyTreeCellRenderer());
  }

  private static ImageIcon getIcon(String name) {
    try {
      return new ImageIcon(PetalTree.class.getResource("/images/" + name));
    } catch (Exception e) {
      System.err.println("Couldn't find image: " + name);
    }

    return new ImageIcon();
  }

  private static class MyTreeUI extends javax.swing.plaf.metal.MetalTreeUI {
    public javax.swing.Icon getExpandedIcon() {
      return MINUS_ICON;
    }

    public javax.swing.Icon getCollapsedIcon() {
      return PLUS_ICON;
    }
  }

  private static class MyTreeCellRenderer extends DefaultTreeCellRenderer {
    /**
     * 
     */
    private static final long serialVersionUID = 8446162207536434648L;

    public java.awt.Component getTreeCellRendererComponent(JTree tree,
        java.lang.Object value, boolean sel, boolean expanded, boolean leaf,
        int row, boolean hasFocus) {

      if (value instanceof PetalFile)
        value = "Root";
      else if (value instanceof PetalObject) {
        PetalObject obj = (PetalObject) value;
        StringBuffer buf = new StringBuffer(obj.getName());

        for (@SuppressWarnings("rawtypes")
        Iterator i = obj.getParameterList().iterator(); i.hasNext();)
          buf.append(" \"" + i.next() + "\"");

        if (obj instanceof Tagged)
          buf.append(" " + ((Tagged) obj).getTag());

        value = buf;
      } else if (value instanceof PetalNodeList) {
        value = "list " + ((PetalNodeList) value).getName();
      } else if ((value instanceof StringLiteral)) {
        value = '"' + (String) ((StringLiteral) value).getLiteralValue() + '"';
      } else if ((value instanceof Value)) {
        value = ((Value) value).getLiteralValue();
      }

      super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row,
          hasFocus);

      if (leaf) {
        setIcon(LEAF_ICON);
      } else {
        if (expanded)
          setIcon(OPENED_ICON);
        else
          setIcon(CLOSED_ICON);
      }

      return this;
    }
  }

  public String getToolTipText(java.awt.event.MouseEvent e) {
    if (e != null) {
      // int row = getRowForLocation(e.getX(), e.getY());
      TreePath path = getPathForLocation(e.getX(), e.getY());

      if ((path != null)) {
        PetalNode node = (PetalNode) path.getLastPathComponent();

        path = path.getParentPath();

        if (path != null) {
          PetalNode parent = (PetalNode) path.getLastPathComponent();

          if (parent instanceof PetalObject)
            return ((PetalObject) parent).getPropertyName(node);
        }
      }
    }

    return null;
  }
}