contenttreemodel.java

来自「用Swing实现的CHM制作工具」· Java 代码 · 共 45 行

package g2w.app.gchm.gui;

import javax.swing.tree.DefaultTreeModel;

/**
 * The customized tree model.
 * 
 * @author GreatGhoul
 * @version 010 2009-3-19 11:09:35
 */
public class ContentTreeModel extends DefaultTreeModel{

	public ContentTreeModel(ContentTreeNode root) {
		super(root);
	}
	
	public void removeNodeFromParent(ContentTreeNode node) {
		node.getUserObject().removeFromParent();
		super.removeNodeFromParent(node);
	}
	
	public void insertNodeInto(ContentTreeNode newChild, 
			ContentTreeNode parent, 
			int index) {
		parent.getUserObject().addChild(newChild.getUserObject(), index);
		super.insertNodeInto(newChild, parent, index);
	}

	/**
	 * Get the neighbor of the given node.
	 * 
	 * @param node The given node.
	 * @return The neighbor.
	 */
	public ContentTreeNode getNeighbor(ContentTreeNode node) {
		ContentTreeNode neighbor = (ContentTreeNode) node.getNextSibling();
		if (neighbor == null)
			neighbor = (ContentTreeNode) node.getPreviousSibling();
		if (neighbor == null)
			neighbor = (ContentTreeNode) node.getParent();
		
		return neighbor;
	}
}