public static void main(String[] args) {
MutableTreeNode node = new DefaultMutableTreeNode(1);
node.insert(new DefaultMutableTreeNode(2), 0);
node.insert(new DefaultMutableTreeNode(5), 1);
MutableTreeNode node1 = (MutableTreeNode) node.getChildAt(0);
node1.insert(new DefaultMutableTreeNode(3), 0);
node1.insert(new DefaultMutableTreeNode(4), 1);
MutableTreeNode node2 = (MutableTreeNode) node1.getChildAt(0);
node2.insert(new DefaultMutableTreeNode(6), 0);
System.out.println(getHeight(node));
}
/** Inserts the given child node into the given parent at the given index. */
public void insertNodeInto(MutableTreeNode child, MutableTreeNode parent, int idx) {
parent.insert(child, idx);
fireTreeNodesInserted(this, getPathToRoot(parent), new int[] {idx}, new Object[] {child});
}
public void drop(final DropTargetDropEvent e) {
_timerHover.stop(); // Prevent hover timer from doing an unwanted expandPath or collapsePath
if (!isDropAcceptable(e)) {
e.rejectDrop();
return;
}
e.acceptDrop(e.getDropAction());
final Transferable transferable = e.getTransferable();
final DataFlavor[] flavors = transferable.getTransferDataFlavors();
for (int i = 0; i < flavors.length; i++) {
final DataFlavor flavor = flavors[i];
if (flavor.isMimeTypeEqual(DataFlavor.javaJVMLocalObjectMimeType)) {
try {
final Point pt = e.getLocation();
final TreePath pathTarget = getClosestPathForLocation(pt.x, pt.y);
final TreePath pathSource = (TreePath) transferable.getTransferData(flavor);
if ((pathTarget == null) || (pathSource == null)) {
e.dropComplete(false);
return;
}
final MutableTreeNode sourceNode = (MutableTreeNode) pathSource.getLastPathComponent();
final MutableTreeNode oldParent = (MutableTreeNode) sourceNode.getParent();
final MutableTreeNode targetNode = (MutableTreeNode) pathTarget.getLastPathComponent();
final MutableTreeNode newParent = (MutableTreeNode) targetNode.getParent();
if (!sourceNode.isLeaf() && (targetNode.getParent() == sourceNode)) {
// trying to drag a folder into its own childs
e.dropComplete(false);
return;
}
final DefaultTreeModel model = (DefaultTreeModel) getModel();
final TreePath pathNewChild = null;
if (targetNode.isLeaf() || JDragTree.this.isCollapsed(pathTarget)) {
// collapsed tree node or leaf
// dropped on a leaf, insert into leaf's parent AFTER leaf
int idx = newParent.getIndex(targetNode);
if (idx < 0) {
JDragTree.logger.warning("child not found in parent!!!");
e.dropComplete(false);
return;
} else {
idx++; // insert AFTER targetNode
// remove node from oldParent ...
final Object[] removedChilds = {sourceNode};
final int[] childIndices = {oldParent.getIndex(sourceNode)};
sourceNode.removeFromParent();
model.nodesWereRemoved(oldParent, childIndices, removedChilds);
// ... and insert into newParent
if (idx >= newParent.getChildCount()) {
// newParent.add( sourceNode );
newParent.insert(sourceNode, newParent.getChildCount());
final int insertedIndex[] = {newParent.getChildCount() - 1};
model.nodesWereInserted(newParent, insertedIndex);
} else {
newParent.insert(sourceNode, idx);
final int insertedIndex[] = {idx};
model.nodesWereInserted(newParent, insertedIndex);
}
}
} else {
// expanded node, insert UNDER the node (before first child)
// remove node from oldParent ...
final Object[] removedChilds = {sourceNode};
final int[] childIndices = {oldParent.getIndex(sourceNode)};
sourceNode.removeFromParent();
model.nodesWereRemoved(oldParent, childIndices, removedChilds);
// ... and add to newParent
targetNode.insert(sourceNode, 0);
final int insertedIndex[] = {0};
model.nodesWereInserted(targetNode, insertedIndex);
}
if (pathNewChild != null)
setSelectionPath(pathNewChild); // Mark this as the selected path in the tree
break; // No need to check remaining flavors
} catch (final UnsupportedFlavorException ufe) {
JDragTree.logger.log(
Level.SEVERE, "Exception thrown in drop(DropTargetDropEvent e)", ufe);
e.dropComplete(false);
return;
} catch (final IOException ioe) {
JDragTree.logger.log(
Level.SEVERE, "Exception thrown in drop(DropTargetDropEvent e)", ioe);
e.dropComplete(false);
return;
}
}
}
e.dropComplete(true);
}
private static void addChildrenTo(final MutableTreeNode node, final List<TreeNode> children) {
for (final Object aChildren : children) {
final MutableTreeNode child = (MutableTreeNode) aChildren;
node.insert(child, node.getChildCount());
}
}
