domtreemodelfactory.java

来自「基于xml的元数据系统 解析xml文档」· Java 代码 · 共 49 行

package xml;

import javax.swing.tree.DefaultTreeModel;

import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

public class DomTreeModelFactory
{
	private Node root = null;
	private DomDefaultMutableTreeNode rootNode = null;
	private DefaultTreeModel treeModel = null;
	
	public DomTreeModelFactory(Document doc)
	{
		root = doc.getDocumentElement();
		rootNode = new DomDefaultMutableTreeNode(root);
		treeModel = new DefaultTreeModel(rootNode);
	}
	
	public DefaultTreeModel getDomTreeModel()
	{
		if(root.hasChildNodes())
		{
			NodeList nl = root.getChildNodes();
			for(int i=0; i<nl.getLength(); i++)
				if(!(nl.item(i).getNodeType()==3 && nl.item(i).getNodeValue().trim().equals("")))
					addToNode(nl.item(i), rootNode);
		}
		return treeModel;
	}
	
	public void addToNode(Node child, DomDefaultMutableTreeNode parentNode)
	{
		DomDefaultMutableTreeNode node = new DomDefaultMutableTreeNode(child);
		treeModel.insertNodeInto(node, parentNode, parentNode.getChildCount());
		
		if(child.hasChildNodes())
		{
			NodeList nl = child.getChildNodes();
			for(int i=0; i<nl.getLength(); i++)
				if(!(nl.item(i).getNodeType()==3 && nl.item(i).getNodeValue().trim().equals("")))
					addToNode(nl.item(i), node);
		}
	}

}