1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
|
package swing.tree;
import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JProgressBar;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.SwingUtilities;
import javax.swing.SwingWorker;
import javax.swing.UIManager;
import javax.swing.UnsupportedLookAndFeelException;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import com.sun.java.swing.plaf.nimbus.NimbusLookAndFeel;
public class MyFileTree {
private JFrame frame;
private JButton b;
private JTree tree;
private JProgressBar progress;
public MyFileTree() {
frame = new JFrame();
progress = new JProgressBar();
progress.setIndeterminate(true);
frame.add(progress, BorderLayout.NORTH);
progress.setVisible(false);
tree = new JTree(new DefaultTreeModel(new DefaultMutableTreeNode("Vide")));
tree.setCellRenderer(new FileTreeCellRenderer());
b = new JButton("Changer Arbo");
b.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent arg0) {
JFileChooser chooser = new JFileChooser();
chooser.setMultiSelectionEnabled(false);
chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
int flag = chooser.showDialog(frame, "Choisir répertoire");
if(flag == JFileChooser.APPROVE_OPTION) {
DirectoryWorker worker = new DirectoryWorker(chooser.getSelectedFile());
progress.setVisible(true);
frame.validate();
worker.execute();
}
}
});
frame.setSize(640, 480);
frame.add(new JScrollPane(tree), BorderLayout.CENTER);
frame.add(b, BorderLayout.SOUTH);
frame.setLocationRelativeTo(null);
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
}
public JFrame getFrame() {
return frame;
}
private class FileTreeCellRenderer extends DefaultTreeCellRenderer {
/* (non-Javadoc)
* @see javax.swing.tree.DefaultTreeCellRenderer#getTreeCellRendererComponent(javax.swing.JTree, java.lang.Object, boolean, boolean, boolean, int, boolean)
*/
@Override
public Component getTreeCellRendererComponent(JTree arg0, Object arg1, boolean arg2, boolean arg3, boolean arg4, int arg5, boolean arg6) {
// TODO Module de remplacement de méthode auto-généré
super.getTreeCellRendererComponent(arg0, arg1, arg2, arg3, arg4, arg5, arg6);
System.out.println(arg1.getClass());
if(DefaultMutableTreeNode.class.equals(arg1.getClass())) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) arg1;
if(File.class.equals(node.getUserObject().getClass())) {
File f = (File) node.getUserObject();
setText(f.getName());
}
}
return this;
}
}
private class DirectoryWorker extends SwingWorker<DefaultMutableTreeNode, Void> {
private DefaultMutableTreeNode root=null;
private File f;
public DirectoryWorker(File f) {
super();
this.f = f;
}
@Override
protected DefaultMutableTreeNode doInBackground() throws Exception {
root = buildFileTree(f);
return null;
}
@Override
protected void done() {
tree.setModel(new DefaultTreeModel(root));
progress.setVisible(false);
frame.validate();
super.done();
}
private DefaultMutableTreeNode buildFileTree(File f) {
System.out.println(f.getAbsolutePath());
DefaultMutableTreeNode node = new DefaultMutableTreeNode(f);
if(f.isDirectory()) {
File[] children = f.listFiles();
for (File child : children) {
node.add(buildFileTree(child));
}
}
return node;
}
}
public static void main(String[] args) {
Runnable r = new Runnable() {
public void run() {
try {
UIManager.setLookAndFeel(new NimbusLookAndFeel());
new MyFileTree().getFrame().setVisible(true);
} catch (UnsupportedLookAndFeelException e) {
// TODO Bloc catch auto-généré
e.printStackTrace();
}
}
};
SwingUtilities.invokeLater(r);
}
} |
Partager