File folder Tree with icons : Tree « Swing JFC « Java






File folder Tree with icons

File folder Tree with icons
    
/*
Swing, Second Edition
by Matthew Robinson, Pavel Vorobiev

*/
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;

  public FileTree1()
  {
    super("Directories Tree");
    setSize(400, 300);

    DefaultMutableTreeNode top = new DefaultMutableTreeNode(
      new IconData(ICON_COMPUTER, null, "Computer"));

    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)));
    }

    m_model = new DefaultTreeModel(top);
    m_tree = new JTree(m_model);

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

    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);

    JScrollPane s = new JScrollPane();
    s.getViewport().add(m_tree);
    getContentPane().add(s, BorderLayout.CENTER);

    m_display = new JTextField();
    m_display.setEditable(false);
    getContentPane().add(m_display, BorderLayout.NORTH);

    WindowListener wndCloser = new WindowAdapter()
    {
      public void windowClosing(WindowEvent e) 
      {
        System.exit(0);
      }
    };
    addWindowListener(wndCloser);
    
    setVisible(true);
  }

  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[]) 
  {
    new FileTree1();
  }
}

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();
  }
}

class FileNode
{
  protected File m_file;

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

  public File getFile() 
  { 
    return m_file;
  }

  public String 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;
    }
  }
}

           
         
    
    
    
  








Related examples in the same category

1.Build a tree based on DefaultMutableTreeNodeBuild a tree based on DefaultMutableTreeNode
2.Add Tree to JScrollPaneAdd Tree to JScrollPane
3.Genealogy TreeGenealogy Tree
4.Tree LinesTree Lines
5.DefaultMutableTreeNode Node Tree SampleDefaultMutableTreeNode Node Tree Sample
6.Display a file system in a JTree viewDisplay a file system in a JTree view
7.implements TreeSelectionListener to create your own listenerimplements TreeSelectionListener  to create your own listener
8.File Tree with Popup MenuFile Tree with Popup Menu
9.File Tree with TooltipsFile Tree with Tooltips
10.Ancestor Tree with IconsAncestor Tree with Icons
11.Tree Icon DemoTree Icon Demo
12.DefaultMutableTreeNode and user objectDefaultMutableTreeNode and user object
13.Display user object in a treeDisplay user object in a tree
14.Tree Expand Event DemoTree Expand Event Demo
15.Tree: Drag and DropTree: Drag and Drop
16.Tree open IconTree open Icon
17.Traverse TreeTraverse Tree
18.Tree based on Array structureTree based on Array structure
19.Tree will Expand event and listenerTree will Expand event and listener
20.Set the Tree LineSet the Tree Line
21.Tree Selection RowTree Selection Row
22.JTree.DynamicUtilTreeNode.createChildrenJTree.DynamicUtilTreeNode.createChildren
23.Install ToolTips for Tree (JTree)Install ToolTips for Tree (JTree)
24.Tree Expand Event Demo 2Tree Expand Event Demo 2
25.A tree with componentA tree with component
26.A sample component for dragging and dropping a collection of files into a tree.A sample component for dragging and dropping a collection of files into a tree.
27.DnD (drag and drop)JTree code DnD (drag and drop)JTree code
28.Build a tree and populate it from hashtablesBuild a tree and populate it from hashtables
29.A simple test to see how we can build a tree and populate itA simple test to see how we can build a tree and populate it
30.Installs custom iconsInstalls custom icons
31.Build a tree and customize its iconsBuild a tree and customize its icons
32.Displaying Hierarchical Data within a JTreeDisplaying Hierarchical Data within a JTree
33.Add and remove tree Node and expand the tree node
34.File System Tree
35.TreeExpansionListener and TreeExpansionEventTreeExpansionListener and TreeExpansionEvent
36.Enabling and Disabling Multiple Selections in a JTree Component
37.Allow only a single node to be selected (default)
38.Allow selection to span one vertical contiguous set of visible nodes
39.Allow multiple selections of visible nodes
40.Setting the Row Height of a JTree
41.All rows will be given 15 pixels of height
42.Have the row height for each row computed individually
43.Flush the internal cache of Row height
44.Preventing Expansion or Collapse of a Node in a JTree: override JTree.setExpandedState()
45.Removing a Node to a JTree Component
46.JTree root cannot be removed with removeNodeFromParent(), use DefaultTreeModel.setRoot() to remove the root
47.Listening for Expansion and Collapse Events in a JTree Component
48.Expansion and Collapse Events in a JTree are fired before a node is expanded or collapsed can be vetoed, thereby preventing the operation.
49.Creating a JTree Component
50.Changing and Removing the Default Icons in a JTree Component
51.Use UIManager to change the default icon for JTree
52.Getting the Selected Nodes in a JTree Component
53.Visiting All the Nodes in a JTree Component
54.Traverse all expanded nodes in tree
55.Finding a Node in a JTree Component
56.Search backward from last visible row looking for any visible node whose name starts with prefix.
57.Find the path regardless of visibility that matches the specified sequence of names
58.Adding a Node to a JTree Component
59.Returns a TreePath containing the specified node.
60.Converting All Nodes in a JTree Component to a TreePath Array
61.Get path for all expanded or not expanded tree pathes
62.Expanding or Collapsing All Nodes in a JTree Component
63.Preventing the Expansion or Collapse of a Node in a JTree Component
64.Listening for Selection Events in a JTree Component
65.Have a popup attached to a JTree
66.Deleting nodes from JTree
67.Adding editable nodes to JTree
68.Searching node in a JTree
69.Drag and drop of a group of files into a tree
70.JTree drag and drop utilites
71.Expand All for a tree path
72.Get tree path from TreeNode
73.JTree Utilities
74.Expand JTree