package com.mwq.frame.common;
// Download by http://www.codefans.net
import java.awt.BorderLayout;
import java.util.Iterator;
import java.util.Set;

import javax.swing.ImageIcon;
import javax.swing.JDialog;
import javax.swing.JOptionPane;
import javax.swing.JScrollPane;
import javax.swing.JTextField;
import javax.swing.JTree;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;

import com.mwq.hibernate.Dao;
import com.mwq.hibernate.mapping.TbDept;

public class DeptTreeDialog extends JDialog {

	private JTree tree;

	private Dao dao;

	public DeptTreeDialog(final JTextField textField) {
		super();
		dao = Dao.getInstance();
		setModal(true);// 設(shè)置對(duì)話框阻止當(dāng)前線程
		setUndecorated(true);// 設(shè)置對(duì)話框不提供標(biāo)題欄
		setBounds(100, 100, 125, 175);

		final JScrollPane scrollPane = new JScrollPane();// 創(chuàng)建滾動(dòng)面板
		getContentPane().add(scrollPane, BorderLayout.CENTER);

		TbDept company = (TbDept) dao.queryDeptById(1);
		DefaultMutableTreeNode root = new DefaultMutableTreeNode(company
				.getName());// 創(chuàng)建部門樹的根節(jié)點(diǎn)

		Set depts = company.getTbDepts();
		for (Iterator deptIt = depts.iterator(); deptIt.hasNext();) {
			TbDept dept = (TbDept) deptIt.next();
			DefaultMutableTreeNode deptNode = new DefaultMutableTreeNode(dept
					.getName());// 創(chuàng)建部門樹的二級(jí)子節(jié)點(diǎn)
			root.add(deptNode);
			Set sonDepts = dept.getTbDepts();
			for (Iterator sonDeptIt = sonDepts.iterator(); sonDeptIt.hasNext();) {
				TbDept sonDept = (TbDept) sonDeptIt.next();
				deptNode.add(new DefaultMutableTreeNode(sonDept.getName()));// 創(chuàng)建部門樹的葉子節(jié)點(diǎn)
			}
		}

		DefaultTreeModel treeModel = new DefaultTreeModel(root); // 利用根節(jié)點(diǎn)對(duì)象創(chuàng)建樹模型對(duì)象

		tree = new JTree(treeModel); // 利用樹模型對(duì)象創(chuàng)建樹對(duì)象
		DefaultTreeCellRenderer render = new DefaultTreeCellRenderer();
		render.setLeafIcon(new ImageIcon());// 設(shè)置葉子結(jié)點(diǎn)的圖標(biāo)
		render.setClosedIcon(new ImageIcon());// 設(shè)置結(jié)點(diǎn)折疊時(shí)的圖標(biāo)
		render.setOpenIcon(new ImageIcon());// 設(shè)置結(jié)點(diǎn)展開時(shí)的圖標(biāo)
		tree.setCellRenderer(render);
		tree.addTreeSelectionListener(new TreeSelectionListener() {// 捕獲樹節(jié)點(diǎn)被選中的事件
					public void valueChanged(TreeSelectionEvent e) {
						TreePath treePath = e.getPath();// 獲得被選中樹節(jié)點(diǎn)的路徑
						DefaultMutableTreeNode node = (DefaultMutableTreeNode) treePath
								.getLastPathComponent();// 獲得被選中樹節(jié)點(diǎn)的對(duì)象
						if (node.getChildCount() == 0) {// 被選中的節(jié)點(diǎn)為葉子節(jié)點(diǎn)
							textField.setText(node.toString());// 將選中節(jié)點(diǎn)的名稱顯示到傳入的文本框中
						} else {// 被選中的節(jié)點(diǎn)不是葉子節(jié)點(diǎn)
							JOptionPane.showMessageDialog(null, "請(qǐng)選擇所在的具體部門",
									"錯(cuò)誤提示", JOptionPane.ERROR_MESSAGE);
							return;
						}
						dispose();// 銷毀部門樹對(duì)話框
					}
				});
		scrollPane.setViewportView(tree); // 將部門樹放到滾動(dòng)面板中
		//
	}

}