/*
* Copyright (c) 2016 Vivid Solutions.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Eclipse Distribution License v. 1.0 which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
*
* http://www.eclipse.org/org/documents/edl-v10.php.
*/
package org.locationtech.jtstest.testbuilder;
import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.event.*;
import java.util.*;
import javax.swing.*;
import javax.swing.event.*;
import javax.swing.border.*;
import javax.swing.tree.*;
import org.locationtech.jtstest.function.*;
import org.locationtech.jtstest.testbuilder.event.GeometryFunctionEvent;
import org.locationtech.jtstest.testbuilder.event.GeometryFunctionListener;
import org.locationtech.jtstest.util.*;
/**
* @version 1.7
*/
public class GeometryFunctionTreePanel extends JPanel {
JScrollPane jScrollPane = new JScrollPane();
JTree tree = new JTree();
BorderLayout borderLayout = new BorderLayout();
Border border1;
private static GeometryFunction getFunctionFromNode(Object value) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
if (node == null)
return null;
Object nodeValue = node.getUserObject();
if (nodeValue instanceof GeometryFunction)
return (GeometryFunction) nodeValue;
return null;
}
private class GeometryFunctionRenderer extends DefaultTreeCellRenderer {
private final ImageIcon binaryIcon = new ImageIcon(this.getClass()
.getResource("BinaryGeomFunction.png"));
private final ImageIcon unaryIcon = new ImageIcon(this.getClass()
.getResource("UnaryGeomFunction.png"));
public GeometryFunctionRenderer() {
}
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 (leaf) {
GeometryFunction func = getFunctionFromNode(value);
setIcon(func.isBinary() ? binaryIcon : unaryIcon);
// setToolTipText("This book is in the Tutorial series.");
String name = StringUtil.capitalize(func.getName());
setText(name);
setToolTipText(func.getSignature() + func.getDescription()); // no tool tip
} else {
setToolTipText(null); // no tool tip
}
return this;
}
}
public GeometryFunctionTreePanel() {
try {
jbInit();
} catch (Exception ex) {
ex.printStackTrace();
}
}
private void jbInit() throws Exception {
setSize(200, 250);
border1 = BorderFactory.createEmptyBorder(4, 4, 4, 4);
setLayout(borderLayout);
setBorder(border1);
add(jScrollPane, BorderLayout.CENTER);
jScrollPane.getViewport().add(tree, null);
tree.setRootVisible(false);
tree.setShowsRootHandles(true);
tree.setCellRenderer(new GeometryFunctionRenderer());
tree.getSelectionModel().setSelectionMode(
TreeSelectionModel.SINGLE_TREE_SELECTION);
tree.addMouseListener(new MouseAdapter() {
public void mouseClicked(MouseEvent e) {
if (e.getClickCount() == 2) {
GeometryFunction fun = getFunction();
if (fun != null)
fireFunctionInvoked(new GeometryFunctionEvent(fun));
}
}
});
tree.addTreeSelectionListener(new TreeSelectionListener() {
public void valueChanged(TreeSelectionEvent e) {
GeometryFunction fun = getFunction();
if (fun != null)
fireFunctionSelected(new GeometryFunctionEvent(fun));
}
});
}
public GeometryFunction getFunction() {
return getFunctionFromNode(tree.getLastSelectedPathComponent());
}
public void populate(DoubleKeyMap funcs) {
tree.setModel(createModel(funcs));
}
private TreeModel createModel(DoubleKeyMap funcMap) {
DefaultMutableTreeNode top = new DefaultMutableTreeNode();
Collection categories = funcMap.keySet();
for (Iterator i = categories.iterator(); i.hasNext();) {
String category = (String) i.next();
DefaultMutableTreeNode catNode = new DefaultMutableTreeNode(category);
top.add(catNode);
Collection funcs = funcMap.values(category);
for (Iterator j = funcs.iterator(); j.hasNext();) {
Object func = j.next();
catNode.add(new DefaultMutableTreeNode(func));
}
}
return new DefaultTreeModel(top);
}
private transient Vector eventListeners;
public synchronized void removeGeometryFunctionListener(
GeometryFunctionListener l) {
if (eventListeners != null && eventListeners.contains(l)) {
Vector v = (Vector) eventListeners.clone();
v.removeElement(l);
eventListeners = v;
}
}
public synchronized void addGeometryFunctionListener(
GeometryFunctionListener l) {
Vector v = eventListeners == null ? new Vector(2) : (Vector) eventListeners
.clone();
if (!v.contains(l)) {
v.addElement(l);
eventListeners = v;
}
}
protected void fireFunctionSelected(GeometryFunctionEvent e) {
if (eventListeners != null) {
Vector listeners = eventListeners;
int count = listeners.size();
for (int i = 0; i < count; i++) {
((GeometryFunctionListener) listeners.elementAt(i)).functionSelected(e);
}
}
}
protected void fireFunctionInvoked(GeometryFunctionEvent e) {
if (eventListeners != null) {
Vector listeners = eventListeners;
int count = listeners.size();
for (int i = 0; i < count; i++) {
((GeometryFunctionListener) listeners.elementAt(i)).functionInvoked(e);
}
}
}
}