Компонент JGraphTree

Компонент JGraphTree позволяет отрисовывать деревья, представленные стандартным интерфейсом TreeModel, в виде графа. Компонент писался под собственные нужды и на идеальное исполнение не претендует. Архив с исходным кодом, документацией и библиотекой с компонентом можно взять тут. Под катом пример использования компонента и небольшие пояснения к его устройству.

Об алгоритме расчета координат узлов дерева читайте в предыдущей статье Как построить дерево в виде графа.

Компонент расчитан на построение дерева, описанного объектом, реализующим интерфейс TreeModel. Это интерфейс из пакета javax.swing.tree использующийся в стандартном компоненте JTree. События TreeModelEvent предлагаемым компонентом не обрабатываются, что, безусловно, не хорошо и возможно будет исправлено. Перерасчет разметки дерева происходит при вызове метода setModel у JGraphTree.

JGraphTree генерирует события взаимодействия мыши с узлами дерева: MouseEnteredEvent, MouseExitedEvent, MousePressedEvent, MouseReleasedEvent и MouseClickedEvent. Каждое событие сопровождается объектом MouseClickedEvent, возвращающим ссылку на узел, с которым связано событие и ссылку на стандартный объект MouseEvent.

Отрисовка узлов дерева вынесена в отдельный компонент, реализующий интерфейс NodeRender:

public interface NodeRender {

    /**
     * Возвращает компонент, рисующий узел дерева.
     * 
     * @param node
     *            отображаемый узел дерева.
     * @param pressed
     *            true, если узел находится в нажатом состоянии.
     * @param highlighted
     *            true, если узел подсвечен.
     * @return компонент, рисующий узел дерева.
     */
    Component getNodeRendererComponent(Object node, boolean pressed, boolean highlighted);
}

Метод getNodeRendererComponent должен возвращать компонент, в виде которого рисуется узел node. При этом ключи pressed и highlighted указывают на то, был ли узел нажат и подсвечен соответственно.

Пример реализации NodeRender с помощью объекта JButton

public class ButtonRender implements NodeRender {

    /**
     * Возвращает кнопку, по шаблону которой будут отображаться узлы дерева.
     */
    public JButton getButton() {
        return button;
    }

    @Override
    public Component getNodeRendererComponent(Object node, boolean selected,
            boolean highlighted) {
        button.setText(node.toString());
        button.getModel().setRollover(highlighted);
        button.getModel().setPressed(selected);
        button.getModel().setArmed(selected);
        return button;
    }

    /**
     * Создает компонент для отрисовки узла дерева в виде кнопки.
     */
    public ButtonRender() {
        button = new JButton();
    }

    private JButton button;
}

Пример реализации NodeRender с помощью собственного алгоритма отрисовки узла

public class CircleRender extends JComponent implements NodeRender {

    public Component getNodeRendererComponent(Object node, boolean pressed,
            boolean highlighted) {
        backgroundColor = (pressed) ? backgroun.darker() : backgroun;
        borderColor = (highlighted) ? border.brighter() : border;
        return this;
    }

    @Override
    public void paint(Graphics g) {
        Rectangle boundary = this.getBounds();
        g.setColor(borderColor);
        g.fillOval(0, 0, boundary.width, boundary.height);
        g.setColor(backgroundColor);
        g.fillOval(borderWidth, borderWidth, boundary.width - borderWidth * 2,
                boundary.height - borderWidth * 2);
    }

    private Color backgroundColor;

    private Color borderColor;

    private int borderWidth = 5;

    private Color backgroun = new Color(150, 200, 255);

    private Color border = new Color(70, 150, 255);

    private static final long serialVersionUID = -8249718835725058224L;
}

Пример использования

Код для генерирования модели дерева можно взять с сайта Oracle: How to Use Trees.

private TreeModel createTreeModel() {
  DefaultMutableTreeNode top = new DefaultMutableTreeNode("The Java Series");
  createNodes(top);
  return new DefaultTreeModel(top);
}

/**
 * @see http://docs.oracle.com/javase/tutorial/uiswing/components/tree.html
 */
private void createNodes(DefaultMutableTreeNode top) {
  DefaultMutableTreeNode category = null;
  DefaultMutableTreeNode book = null;

  category = new DefaultMutableTreeNode("Books for Java Programmers");
  top.add(category);

  // original Tutorial
  book = new DefaultMutableTreeNode(new BookInfo(
          "The Java Tutorial: A Short Course on the Basics", "tutorial.html"));
  category.add(book);

  // Tutorial Continued
  book = new DefaultMutableTreeNode(new BookInfo(
          "The Java Tutorial Continued: The Rest of the JDK", "tutorialcont.html"));
  category.add(book);

  // Swing Tutorial
  book = new DefaultMutableTreeNode(new BookInfo(
          "The Swing Tutorial: A Guide to Constructing GUIs", "swingtutorial.html"));
  category.add(book);

  // ...add more books for programmers...

  category = new DefaultMutableTreeNode("Books for Java Implementers");
  top.add(category);

  // VM
  book = new DefaultMutableTreeNode(new BookInfo(
          "The Java Virtual Machine Specification", "vm.html"));
  category.add(book);

  // Language Spec
  book = new DefaultMutableTreeNode(new BookInfo("The Java Language Specification",
          "jls.html"));
  category.add(book);
}

class BookInfo {
  public String bookName;
  public String bookURL;

  public BookInfo(String book, String url) {
      bookName = book;
      bookURL = url;
  }

  public String toString() {
      return bookName;
  }
}

DemoWindow - окно на котором расположен компонент. labelComment - компонент для вывода текстовой информации. Тоже располагается на DemoWindow. Идея примера в том, чтобы при нажатии на узел дерева выводить информацию о содержимом этого узла в labelComment.

public class App implements Runnable {
    /**
     * Launch the application.
     */
    public static void main(String[] args) {
        try {
            UIManager.setLookAndFeel(getLookAndFeel());
        } catch (Exception e) {
            e.printStackTrace();
        }
        EventQueue.invokeLater(new App());
    }

    private static String getLookAndFeel() {
        for (LookAndFeelInfo info : UIManager.getInstalledLookAndFeels()) {
            if ("Nimbus".equals(info.getName())) {
                return info.getClassName();
            }
        }
        return UIManager.getSystemLookAndFeelClassName();
    }

    public void run() {
        final DemoWindow window = new DemoWindow();        
        TreeModel model = createTreeModel();        
        window.graphTree.setModel(model);
        window.graphTree.setNodeRender(new ButtonRender());
        window.graphTree.addMouseNodeEventListener(new MouseNodeAdapter() {
            @Override
            public void mouseClicked(MouseNodeEvent event) {
                DefaultMutableTreeNode node = (DefaultMutableTreeNode) event.getNode();
                String text = null;
                if (node.getUserObject() instanceof BookInfo) {
                    BookInfo bi = (BookInfo) node.getUserObject();
                    text = bi.bookName + ": " + bi.bookURL;
                } else {
                    text = node.getUserObject().toString();
                }
                window.labelComment.setText(text);
            }
        });
        window.setVisible(true);
    }
}

Напоминаю, что архив с библиотекой, исходным кодом и документацие вы можете скачать здесь.

Для запуска демонстрационного примера, распакуйте архив и запустите myswing-1.0.jar:

java -jar myswing-1.0.jar

Репозиторий www.bitbucket.org/dok/myswing.