1 import  java.awt. * ;
  2 import  java.awt.event. * ;
  3 import  java.io. * ;
  4 import  java.util. * ;
  5
  6 import  javax.swing. * ;
  7 import  javax.swing.tree. * ;
  8 import  javax.swing.event. * ;
  9
 10 public   class  FileTree1 
 11   extends  JFrame 
 12 {
 13   public   static   final  ImageIcon ICON_COMPUTER  =  
 14    new  ImageIcon( " computer.gif " );
 15   public   static   final  ImageIcon ICON_DISK  =  
 16    new  ImageIcon( " disk.gif " );
 17   public   static   final  ImageIcon ICON_FOLDER  =  
 18    new  ImageIcon( " folder.gif " );
 19   public   static   final  ImageIcon ICON_EXPANDEDFOLDER  =  
 20    new  ImageIcon( " expandedfolder.gif " );
 21
 22   protected  JTree  m_tree;
 23   protected  DefaultTreeModel m_model;
 24   protected  JTextField m_display;
 25
 26   public  FileTree1()
 27   {
 28    super ( " Directories Tree " );
 29   setSize( 400 ,  300 );
 30
 31   DefaultMutableTreeNode top  =   new  DefaultMutableTreeNode(
 32     new  IconData(ICON_COMPUTER,  null ,  " Computer " ));
 33
 34   DefaultMutableTreeNode node;
 35   File[] roots  =  File.listRoots();
 36    for  ( int  k = 0 ; k < roots.length; k ++ )
 37    {
 38    node  =   new  DefaultMutableTreeNode( new  IconData(ICON_DISK, 
 39      null ,  new  FileNode(roots[k])));
 40    top.add(node);
 41                         node.add(  new  DefaultMutableTreeNode( new  Boolean( true )));
 42   }
 43
 44   m_model  =   new  DefaultTreeModel(top);
 45   m_tree  =   new  JTree(m_model);
 46
 47                 m_tree.putClientProperty( " JTree.lineStyle " ,  " Angled " );
 48
 49   TreeCellRenderer renderer  =   new  
 50    IconCellRenderer();
 51   m_tree.setCellRenderer(renderer);
 52
 53   m_tree.addTreeExpansionListener( new  
 54    DirExpansionListener());
 55
 56   m_tree.addTreeSelectionListener( new  
 57    DirSelectionListener());
 58
 59   m_tree.getSelectionModel().setSelectionMode(
 60    TreeSelectionModel.SINGLE_TREE_SELECTION); 
 61   m_tree.setShowsRootHandles( true ); 
 62   m_tree.setEditable( false );
 63
 64   JScrollPane s  =   new  JScrollPane();
 65   s.getViewport().add(m_tree);
 66   getContentPane().add(s, BorderLayout.CENTER);
 67
 68   m_display  =   new  JTextField();
 69   m_display.setEditable( false );
 70   getContentPane().add(m_display, BorderLayout.NORTH);
 71
 72   WindowListener wndCloser  =   new  WindowAdapter()
 73    {
 74     public   void  windowClosing(WindowEvent e) 
 75     {
 76     System.exit( 0 );
 77    }
 78   } ;
 79   addWindowListener(wndCloser);
 80   
 81   setVisible( true );
 82  }
 83
 84  DefaultMutableTreeNode getTreeNode(TreePath path)
 85   {
 86    return  (DefaultMutableTreeNode)(path.getLastPathComponent());
 87  }
 88
 89  FileNode getFileNode(DefaultMutableTreeNode node)
 90   {
 91    if  (node  ==   null )
 92     return   null ;
 93   Object obj  =  node.getUserObject();
 94    if  (obj  instanceof  IconData)
 95    obj  =  ((IconData)obj).getObject();
 96    if  (obj  instanceof  FileNode)
 97     return  (FileNode)obj;
 98    else
 99     return   null ;
100  }
101
102      //  Make sure expansion is threaded and updating the tree model
103      //  only occurs within the event dispatching thread.
104      class  DirExpansionListener  implements  TreeExpansionListener
105      {
106          public   void  treeExpanded(TreeExpansionEvent event)
107          {
108               final  DefaultMutableTreeNode node  =  getTreeNode(
109                 event.getPath());
110               final  FileNode fnode  =  getFileNode(node);
111
112             Thread runner  =   new  Thread() 
113              {
114                public   void  run() 
115                {
116                  if  (fnode  !=   null   &&  fnode.expand(node)) 
117                  {
118                   Runnable runnable  =   new  Runnable() 
119                    {
120                      public   void  run() 
121                      {
122                        m_model.reload(node);
123                     }
124                   } ;
125                   SwingUtilities.invokeLater(runnable);
126                 }
127               }
128             } ;
129             runner.start();
130         }
131
132          public   void  treeCollapsed(TreeExpansionEvent event)  {}
133     }
134
135
136   class  DirSelectionListener 
137    implements  TreeSelectionListener 
138   {
139    public   void  valueChanged(TreeSelectionEvent event)
140    {
141    DefaultMutableTreeNode node  =  getTreeNode(
142     event.getPath());
143    FileNode fnode  =  getFileNode(node);
144     if  (fnode  !=   null )
145     m_display.setText(fnode.getFile().
146      getAbsolutePath());
147     else
148     m_display.setText( "" );
149   }
150  }
151
152   public   static   void  main(String argv[]) 
153   {
154    new  FileTree1();
155  }
156 }
157
158 class  IconCellRenderer 
159   extends     JLabel 
160   implements  TreeCellRenderer
161 {
162   protected  Color m_textSelectionColor;
163   protected  Color m_textNonSelectionColor;
164   protected  Color m_bkSelectionColor;
165   protected  Color m_bkNonSelectionColor;
166   protected  Color m_borderSelectionColor;
167
168   protected   boolean  m_selected;
169
170   public  IconCellRenderer()
171   {
172    super ();
173   m_textSelectionColor  =  UIManager.getColor(
174     " Tree.selectionForeground " );
175   m_textNonSelectionColor  =  UIManager.getColor(
176     " Tree.textForeground " );
177   m_bkSelectionColor  =  UIManager.getColor(
178     " Tree.selectionBackground " );
179   m_bkNonSelectionColor  =  UIManager.getColor(
180     " Tree.textBackground " );
181   m_borderSelectionColor  =  UIManager.getColor(
182     " Tree.selectionBorderColor " );
183   setOpaque( false );
184  }
185
186   public  Component getTreeCellRendererComponent(JTree tree, 
187   Object value,  boolean  sel,  boolean  expanded,  boolean  leaf, 
188    int  row,  boolean  hasFocus) 
189   
190   {
191   DefaultMutableTreeNode node  =  
192    (DefaultMutableTreeNode)value;
193   Object obj  =  node.getUserObject();
194   setText(obj.toString());
195
196                  if  (obj  instanceof  Boolean)
197                   setText( " Retrieving data " );
198
199    if  (obj  instanceof  IconData)
200    {
201    IconData idata  =  (IconData)obj;
202     if  (expanded)
203     setIcon(idata.getExpandedIcon());
204     else
205     setIcon(idata.getIcon());
206   }
207    else
208    setIcon( null );
209
210   setFont(tree.getFont());
211   setForeground(sel  ?  m_textSelectionColor : 
212    m_textNonSelectionColor);
213   setBackground(sel  ?  m_bkSelectionColor : 
214    m_bkNonSelectionColor);
215   m_selected  =  sel;
216    return   this ;
217  }
218     
219  
220   public   void  paintComponent(Graphics g) 
221   {
222   Color bColor  =  getBackground();
223   Icon icon  =  getIcon();
224
225   g.setColor(bColor);
226    int  offset  =   0 ;
227    if (icon  !=   null   &&  getText()  !=   null ) 
228    offset  =  (icon.getIconWidth()  +  getIconTextGap());
229   g.fillRect(offset,  0 , getWidth()  -   1   -  offset,
230    getHeight()  -   1 );
231   
232    if  (m_selected) 
233    {
234    g.setColor(m_borderSelectionColor);
235    g.drawRect(offset,  0 , getWidth() - 1 - offset, getHeight() - 1 );
236   }
237    super .paintComponent(g);
238     }
239 }
240
241 class  IconData
242 {
243   protected  Icon   m_icon;
244   protected  Icon   m_expandedIcon;
245   protected  Object m_data;
246
247   public  IconData(Icon icon, Object data)
248   {
249   m_icon  =  icon;
250   m_expandedIcon  =   null ;
251   m_data  =  data;
252  }
253
254   public  IconData(Icon icon, Icon expandedIcon, Object data)
255   {
256   m_icon  =  icon;
257   m_expandedIcon  =  expandedIcon;
258   m_data  =  data;
259  }
260
261   public  Icon getIcon() 
262   { 
263    return  m_icon;
264  }
265
266   public  Icon getExpandedIcon() 
267   { 
268    return  m_expandedIcon != null   ?  m_expandedIcon : m_icon;
269  }
270
271   public  Object getObject() 
272   { 
273    return  m_data;
274  }
275
276   public  String toString() 
277   { 
278    return  m_data.toString();
279  }
280 }
281
282 class  FileNode
283 {
284   protected  File m_file;
285
286   public  FileNode(File file)
287   {
288   m_file  =  file;
289  }
290
291   public  File getFile() 
292   { 
293    return  m_file;
294  }
295
296   public  String toString() 
297   { 
298    return  m_file.getName().length()  >   0   ?  m_file.getName() : 
299    m_file.getPath();
300  }
301
302   public   boolean  expand(DefaultMutableTreeNode parent)
303   {
304   DefaultMutableTreeNode flag  =  
305    (DefaultMutableTreeNode)parent.getFirstChild();
306    if  (flag == null )    //  No flag
307     return   false ;
308   Object obj  =  flag.getUserObject();
309    if  ( ! (obj  instanceof  Boolean))
310     return   false ;       //  Already expanded
311
312   parent.removeAllChildren();   //  Remove Flag
313
314   File[] files  =  listFiles();
315    if  (files  ==   null )
316     return   true ;
317
318   Vector v  =   new  Vector();
319
320    for  ( int  k = 0 ; k < files.length; k ++ )
321    {
322    File f  =  files[k];
323     if  ( ! (f.isDirectory()))
324      continue ;
325
326    FileNode newNode  =   new  FileNode(f);
327    
328     boolean  isAdded  =   false ;
329     for  ( int  i = 0 ; i < v.size(); i ++ )
330     {
331     FileNode nd  =  (FileNode)v.elementAt(i);
332      if  (newNode.compareTo(nd)  <   0 )
333      {
334      v.insertElementAt(newNode, i);
335      isAdded  =   true ;
336       break ;
337     }
338    }
339     if  ( ! isAdded)
340     v.addElement(newNode);
341   }
342
343    for  ( int  i = 0 ; i < v.size(); i ++ )
344    {
345    FileNode nd  =  (FileNode)v.elementAt(i);
346    IconData idata  =   new  IconData(FileTree1.ICON_FOLDER, 
347     FileTree1.ICON_EXPANDEDFOLDER, nd);
348    DefaultMutableTreeNode node  =   new  
349     DefaultMutableTreeNode(idata);
350    parent.add(node);
351     
352     if  (nd.hasSubDirs())
353     node.add( new  DefaultMutableTreeNode( 
354       new  Boolean( true ) ));
355   }
356
357    return   true ;
358  }
359
360   public   boolean  hasSubDirs()
361   {
362   File[] files  =  listFiles();
363    if  (files  ==   null )
364     return   false ;
365    for  ( int  k = 0 ; k < files.length; k ++ )
366    {
367     if  (files[k].isDirectory())
368      return   true ;
369   }
370    return   false ;
371  }
372  
373   public   int  compareTo(FileNode toCompare)
374   { 
375    return   m_file.getName().compareToIgnoreCase(
376    toCompare.m_file.getName() ); 
377  }
378
379   protected  File[] listFiles()
380   {
381    if  ( ! m_file.isDirectory())
382     return   null ;
383    try
384    {
385     return  m_file.listFiles();
386   }
387    catch  (Exception ex)
388    {
389    JOptionPane.showMessageDialog( null , 
390      " Error reading directory  " + m_file.getAbsolutePath(),
391      " Warning " , JOptionPane.WARNING_MESSAGE);
392     return   null ;
393   }
394  }
395 }
396
397