filetree1.java

JTree

/**
 * <p>Title: </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2004</p>
 * <p>Company: </p>
 * @author not attributable
 * @version 1.0
 */


import java.awt.*;
import java.awt.event.*;
import java.io.*;
import java.util.*;

import javax.swing.*;
/**
 * <p>Title: </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2004</p>
 * <p>Company: </p>
 * @author not attributable
 * @version 1.0
 */


import java.awt.*;
import java.awt.event.*;
import java.io.*;
import java.util.*;

import javax.swing.*;
import javax.swing.tree.*;
import javax.swing.event.*;

public class FileTree1
 extends JFrame
{
  
 public static final ImageIcon ICON_COMPUTER =
  new ImageIcon("computer.gif");
 public static final ImageIcon ICON_DISK =
  new ImageIcon("disk.gif");
 public static final ImageIcon ICON_FOLDER =
  new ImageIcon("folder.gif");
 public static final ImageIcon ICON_EXPANDEDFOLDER =
  new ImageIcon("expandedfolder.gif");

 protected JTree  m_tree;
 protected DefaultTreeModel m_model;
 protected JTextField m_display;
 protected JToolBar toolbar;
 protected JButton downloadButton;
 protected JButton uploadButton;

 public FileTree1()
 {
  super("Directories Tree");
  init();
  setSize(400, 300);
  setVisible(true);
 }
	//初始化
	public void init(){
		DefaultMutableTreeNode top = new DefaultMutableTreeNode(
		   new IconData(ICON_COMPUTER, null, "我的硬盘"));//Tree的顶层目录
		  //DefaultMutableTreeNode netHardDisk=new DefaultMutableTreeNode(new IconData(ICON_COMPUTER,null,"网络硬盘"));//定义网络硬盘的目录
		  DefaultMutableTreeNode node;
		  File[] roots = File.listRoots();
		  for (int k=0; k<roots.length; k++)
		  {
		   node = new DefaultMutableTreeNode(new IconData(ICON_DISK,
			null, new FileNode(roots[k])));
		   top.add(node);
			node.add( new DefaultMutableTreeNode(new Boolean(true)));
			//node.add( new DefaultMutableTreeNode("OK"));
		  }
		  /*网络硬盘设置子文件夹
		  for (int k=0; k<roots.length; k++)
			{
			 node = new DefaultMutableTreeNode(new IconData(ICON_DISK,
			  null, new FileNode(roots[k])));
			 netHardDisk.add(node);
			  node.add( new DefaultMutableTreeNode(new Boolean(true)));
			  //node.add( new DefaultMutableTreeNode("OK"));
			}
			top.add(netHardDisk);
			*/
		  m_model = new DefaultTreeModel(top);
		  m_tree = new JTree(m_model);

		  m_tree.putClientProperty("JTree.lineStyle", "Angled");

			//设置Tree目录格式
		  TreeCellRenderer renderer = new IconCellRenderer();
		  m_tree.setCellRenderer(renderer);//

		  m_tree.addTreeExpansionListener(new
		   DirExpansionListener());

		  m_tree.addTreeSelectionListener(new
		   DirSelectionListener());

		  m_tree.getSelectionModel().setSelectionMode(
		   TreeSelectionModel.SINGLE_TREE_SELECTION);
		  m_tree.setShowsRootHandles(true);
		  m_tree.setEditable(false);
			
		  toolbar=new JToolBar();//
		//toolbar.setBorder(BorderFactory.createLineBorder(Color.black));
		  downloadButton=new JButton(new ImageIcon("download.gif"));
		  uploadButton=new JButton(new ImageIcon("upload.gif"));
		downloadButton.setBorder(BorderFactory.createRaisedBevelBorder());
		uploadButton.setBorder(BorderFactory.createRaisedBevelBorder());
		downloadButton.setToolTipText("下载");
		uploadButton.setToolTipText("上传");
		downloadButton.setActionCommand("downloadButton");
		  downloadButton.addActionListener(new ButtonAction());
		  //uploadButton.addActionListener();
		  toolbar.add(downloadButton);
		  toolbar.add(uploadButton);
		  getContentPane().add(toolbar,BorderLayout.NORTH);
		  
		  JScrollPane s = new JScrollPane();
		  s.getViewport().add(m_tree);
		  getContentPane().add(s, BorderLayout.CENTER);

		  m_display = new JTextField();
		  //m_display.setBackground(Color.BLUE);
		  m_display.setEditable(false);
		  getContentPane().add(m_display, BorderLayout.SOUTH);

		  WindowListener wndCloser = new WindowAdapter()
		  {
		   public void windowClosing(WindowEvent e)
		   {
			System.exit(0);
		   }
		  };
		  addWindowListener(wndCloser);
	}
	
 DefaultMutableTreeNode getTreeNode(TreePath path)
 {
  return (DefaultMutableTreeNode)(path.getLastPathComponent());
 }

 FileNode getFileNode(DefaultMutableTreeNode node)
 {
  if (node == null)
   return null;
  Object obj = node.getUserObject();
  if (obj instanceof IconData)
   obj = ((IconData)obj).getObject();
  if (obj instanceof FileNode)
   return (FileNode)obj;
  else
   return null;
 }

	// Make sure expansion is threaded and updating the tree model
	// only occurs within the event dispatching thread.
	class DirExpansionListener implements TreeExpansionListener
	{
		public void treeExpanded(TreeExpansionEvent event)
		{
			 final DefaultMutableTreeNode node = getTreeNode(
				event.getPath());
			 final FileNode fnode = getFileNode(node);

			Thread runner = new Thread()
			{
			  public void run()
			  {
				if (fnode != null && fnode.expand(node))
				{
				  Runnable runnable = new Runnable()
				  {
					public void run()
					{
					   m_model.reload(node);
					}
				  };
				  SwingUtilities.invokeLater(runnable);
				}
			  }
			};
			runner.start();
		}

		public void treeCollapsed(TreeExpansionEvent event) {}
	}


 class DirSelectionListener
  implements TreeSelectionListener
 {
  public void valueChanged(TreeSelectionEvent event)
  {
   DefaultMutableTreeNode node = getTreeNode(
	event.getPath());
   FileNode fnode = getFileNode(node);
   if (fnode != null)
	m_display.setText(fnode.getFile().
	 getAbsolutePath());
   else
	m_display.setText("");
  }
 }

 public static void main(String argv[])
 {
 	
   try {
	UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
	Font font=new Font("宋体",Font.PLAIN,12);
	String names[]={"Label","CheckBox","PopupMenu","TextPane",
				   "MenuItem","CheckBoxMenuItem","JRadioButtonMenuItem",
				   "ComboBox","Button","Tree","ScrollPane","TabbedPane",
				   "EditorPane","TitledBorder","Menu","TextArea","OptionPane",
				   "MenuBar","ToolBar","ToggleButton","ToolTip","ProgressBar",
				   "TableHeader","Panel","List","ColorChooser","PasswordField",
				   "TextField","Table","Label","Viewport","RadioButtonMenuItem",
				   "RadioButton"};
	for(int i=0;i<names.length;i++)UIManager.put(names[i]+".font",font);
	UIManager.put("Label.foreground",Color.black);
	UIManager.put("Border.foreground",Color.black);
	UIManager.put("TitledBorder.titleColor",Color.black);
	//UCoolManager ucm = new UCoolManager();
	
	new FileTree1();
	
 }
 catch(Exception e){
	e.printStackTrace();
 }
  
 }

}
class ButtonAction implements ActionListener{
 public void actionPerformed(ActionEvent e){
	if(e.getActionCommand()=="downloadButton"){
		String nameField="";
			File dir=new File("c:/");
			JFileChooser chooser=new JFileChooser(dir);
			chooser.setFileFilter(new AllFileFilter());
			int result = chooser.showSaveDialog(null);
				if (result == JFileChooser.APPROVE_OPTION) {
				  File file = chooser.getSelectedFile();
				  nameField=file.getAbsolutePath();

				} 
	}
	else if(e.getActionCommand()=="uploadButton"){
 		
	}
 }
}

class IconCellRenderer
 extends    JLabel
 implements TreeCellRenderer
{
 protected Color m_textSelectionColor;
 protected Color m_textNonSelectionColor;
 protected Color m_bkSelectionColor;
 protected Color m_bkNonSelectionColor;
 protected Color m_borderSelectionColor;

 protected boolean m_selected;

 public IconCellRenderer()
 {
  super();
  m_textSelectionColor = UIManager.getColor(
   "Tree.selectionForeground");
  m_textNonSelectionColor = UIManager.getColor(
   "Tree.textForeground");
  m_bkSelectionColor = UIManager.getColor(
   "Tree.selectionBackground");
  m_bkNonSelectionColor = UIManager.getColor(
   "Tree.textBackground");
  m_borderSelectionColor = UIManager.getColor(
   "Tree.selectionBorderColor");
  setOpaque(false);
 }

 public Component getTreeCellRendererComponent(JTree tree,
  Object value, boolean sel, boolean expanded, boolean leaf,
  int row, boolean hasFocus)

 {
  DefaultMutableTreeNode node =
   (DefaultMutableTreeNode)value;
  Object obj = node.getUserObject();
  setText(obj.toString());

				if (obj instanceof Boolean)
				  setText("Retrieving data...");

  if (obj instanceof IconData)
  {
   IconData idata = (IconData)obj;

   if (expanded){
	setIcon(idata.getExpandedIcon());
   }
   else
	setIcon(idata.getIcon());
  }
  else
   setIcon(null);

  setFont(tree.getFont());
  setForeground(sel ? m_textSelectionColor :
   m_textNonSelectionColor);
  setBackground(sel ? m_bkSelectionColor :
   m_bkNonSelectionColor);
  m_selected = sel;
  return this;
 }


 public void paintComponent(Graphics g)
 {
  Color bColor = getBackground();
  Icon icon = getIcon();

  g.setColor(bColor);
  int offset = 0;
  if(icon != null && getText() != null)
   offset = (icon.getIconWidth() + getIconTextGap());
  g.fillRect(offset, 0, getWidth() - 1 - offset,
   getHeight() - 1);

  if (m_selected)
  {
   g.setColor(m_borderSelectionColor);
   g.drawRect(offset, 0, getWidth()-1-offset, getHeight()-1);
  }
  super.paintComponent(g);
	}
}

class IconData
{
 protected Icon   m_icon;
 protected Icon   m_expandedIcon;
 protected Object m_data;

 public IconData(Icon icon, Object data)
 {
  m_icon = icon;
  m_expandedIcon = null;
  m_data = data;
 }

 public IconData(Icon icon, Icon expandedIcon, Object data)
 {

  m_icon = icon;
  m_expandedIcon = expandedIcon;
  m_data = data;
 }

 public Icon getIcon()
 {
  return m_icon;
 }

 public Icon getExpandedIcon()
 {
  return m_expandedIcon!=null ? m_expandedIcon : m_icon;
 }

 public Object getObject()
 {
  return m_data;
 }

 public String toString()
 {
  return m_data.toString();
 }
 
 //增加
 public boolean isDirectory(){
	if(m_data instanceof File){
		return ((File)m_data).isDirectory();
	}
	else
	return false;
 }
}

class FileNode
{
 protected File m_file;

 public FileNode(File file)
 {
  m_file = file;
 }

 public File getFile()
 {
  return m_file;
 }

 public String toString()
 {
	//System.out.println(m_file.getPath().toString());
  return m_file.getName().length() > 0 ? m_file.getName() :
   m_file.getPath();
   
 }
//展开一个父节点
 public boolean expand(DefaultMutableTreeNode parent)
 {
  DefaultMutableTreeNode flag =
   (DefaultMutableTreeNode)parent.getFirstChild();
  if (flag==null)   // No flag
   return false;
  Object obj = flag.getUserObject();
  if (!(obj instanceof Boolean))
   return false;      // Already expanded

  parent.removeAllChildren();  // Remove Flag

  File[] files = listFiles();
  if (files == null)
   return true;

  Vector v = new Vector();

  for (int k=0; k<files.length; k++)
  {
   File f = files[k];
   
   //是否显示文件
   if (!(f.isDirectory()))
	continue;//
   FileNode newNode = new FileNode(f);

   boolean isAdded = false;
   for (int i=0; i<v.size(); i++)
   {
	FileNode nd = (FileNode)v.elementAt(i);
	if (newNode.compareTo(nd) < 0)
	{
	 v.insertElementAt(newNode, i);
	 isAdded = true;
	 break;
	}
   }
   if (!isAdded)
	v.addElement(newNode);
  }

  for (int i=0; i<v.size(); i++)
  {
   FileNode nd = (FileNode)v.elementAt(i);
   IconData idata = new IconData(FileTree1.ICON_FOLDER,
   FileTree1.ICON_EXPANDEDFOLDER, nd);
   DefaultMutableTreeNode node = new
	DefaultMutableTreeNode(idata);
   parent.add(node);

   if (nd.hasSubDirs())
	node.add(new DefaultMutableTreeNode(
	 new Boolean(true) ));
  }

  return true;
 }

 public boolean hasSubDirs()
 {
  File[] files = listFiles();
  if (files == null)
   return false;
  for (int k=0; k<files.length; k++)
  {
   if (files[k].isDirectory())
	return true;
  }
  return false;
 }

 public int compareTo(FileNode toCompare)
 {
  return  m_file.getName().compareToIgnoreCase(
   toCompare.m_file.getName() );
 }

 protected File[] listFiles()
 {
  if (!m_file.isDirectory())
   return null;
  try
  {
   return m_file.listFiles();
  }
  catch (Exception ex)
  {
   JOptionPane.showMessageDialog(null,
	"Error reading directory "+m_file.getAbsolutePath(),
	"Warning", JOptionPane.WARNING_MESSAGE);
   return null;
  }
 }
 
}