jdirchooser.java

java中的开源控件

package com.sunking.swing;

import java.io.*;

import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.event.*;

/**
 * <p>Title: OpenSwing</p>
 * <p>Description: JDirChooser 目录选择器</p>
 * <p>Copyright: Copyright (c) 2004</p>
 * <p>Company: </p>
 * @author <a href="mailto:[e-mail]">SunKing</a>
 *  && <a href="mailto:[e-mail]">zt9788</a>
 * @version 1.0
 */
public class JDirChooser extends JDialog implements TreeSelectionListener,
    ActionListener, Serializable{
    boolean hasCancel = true;

    JPanel pCenter = new JPanel(new BorderLayout());
    JFileTree fileTree = new JFileTree();
    JScrollPane spTree = new JScrollPane(fileTree);
    JPanel pSouth = new JPanel(new BorderLayout());
    JPanel pButtons = new JPanel(new FlowLayout(FlowLayout.RIGHT, 7, 0));
    JPanel pResult = new JPanel(new BorderLayout());
    JLabel lbFolder = new JLabel(OpenSwingUtil.getOpenResource("Folder"));
    JTextField txtResult = new JTextField();
    JButton bttCreateNew = new JButton(OpenSwingUtil.getOpenResource(
        "CreateNew"));
    JButton bttCancel = new JButton(OpenSwingUtil.getOpenResource("Cancel"));
    JButton bttOK = new JButton(OpenSwingUtil.getOpenResource("OK"));
    JLabel lbView = new JLabel();
    JPanel pAdjust = new JPanel(){
        public void paintChildren(Graphics g){
            super.paintChildren(g);
            int w = getWidth();
            int h = getHeight();
            Color oldColor = g.getColor();
            // draw ///
            g.setColor(Color.white);
            g.drawLine(w, h - 12, w - 12, h);
            g.drawLine(w, h - 8, w - 8, h);
            g.drawLine(w, h - 4, w - 4, h);
            g.setColor(new Color(128, 128, 128));
            g.drawLine(w, h - 11, w - 11, h);
            g.drawLine(w, h - 10, w - 10, h);
            g.drawLine(w, h - 7, w - 7, h);
            g.drawLine(w, h - 6, w - 6, h);
            g.drawLine(w, h - 3, w - 3, h);
            g.drawLine(w, h - 2, w - 2, h);
            g.setColor(oldColor);
        }
    };

    MouseInputAdapter adjustWindowListener = new MouseInputAdapter(){
        Point oldP = null;
        public void mouseDragged(MouseEvent e){
            if(oldP != null){
                Point newP = e.getPoint();
                JDirChooser c = JDirChooser.this;
                c.setBounds(c.getX(), c.getY(), c.getWidth() + (newP.x - oldP.x),
                            c.getHeight() + (newP.y - oldP.y));
                c.validate();
                oldP = newP;
            }
        }

        public void mouseMoved(MouseEvent e){
            Component c = e.getComponent();
            Rectangle r = new Rectangle(c.getWidth() - 12, 0, 12, c.getHeight());
            if(r.contains(e.getPoint())){
                JDirChooser.this.setCursor(Cursor.getPredefinedCursor(Cursor.
                    SE_RESIZE_CURSOR));
            } else{
                JDirChooser.this.setCursor(Cursor.getDefaultCursor());
            }
        }

        public void mousePressed(MouseEvent e){
            Component c = e.getComponent();
            Rectangle r = new Rectangle(c.getWidth() - 12, 0, 12, c.getHeight());
            if(r.contains(e.getPoint())){
                oldP = e.getPoint();
            } else{
                oldP = null;
            }
        }

        public void mouseExited(MouseEvent e){
            JDirChooser.this.setCursor(Cursor.getDefaultCursor());
        }

        public void mouseReleased(MouseEvent e){
            oldP = null;
        }
    };

    public static File showDialog(Component c, String title, boolean modal,
                                  File initDir, String msg){
        JDirChooser dialog;
        Window owner = getRootWindow(c);
        if(owner instanceof Dialog){
            dialog = new JDirChooser((Dialog)owner, title, modal);
        } else if(owner instanceof Frame){
            dialog = new JDirChooser((Frame)owner, title, modal);
        } else{
            dialog = new JDirChooser();
            dialog.setTitle(title);
        }
        if(initDir != null){
            try{
                dialog.setSelectFile(initDir);
            } catch(Exception ex){
            }
        }
        if(msg != null){
            dialog.setMsg(msg);
        }
        Dimension d = Toolkit.getDefaultToolkit().getScreenSize();
        dialog.setLocation((d.width - dialog.getSize().width) / 2,
                           (d.height - dialog.getSize().height) / 2);
        dialog.show();
        return dialog.getSelectFile();
    }

    /**
     * 取得根窗口
     * @param c Component
     * @return Window
     */
    static Window getRootWindow(Component c){
        if(c == null)return null;
        Container parent = c.getParent();
        if(c instanceof Window)
            return(Window)c;
        while(!(parent instanceof Window))
            parent = parent.getParent();
        return(Window)parent;
    }

    public JDirChooser(Frame frame, String title, boolean modal){
        super(frame, title, modal);
        try{
            jbInit();
        } catch(Exception ex){
            ex.printStackTrace();
        }
    }

    public JDirChooser(Dialog frame, String title, boolean modal){
        super(frame, title, modal);
        try{
            jbInit();
        } catch(Exception ex){
            ex.printStackTrace();
        }
    }

    public JDirChooser(){
        this((Frame)null, "", false);
        try{
            jbInit();
        } catch(Exception ex){
            ex.printStackTrace();
        }
    }

    private void jbInit() throws Exception{
        bttCreateNew.setPreferredSize(new Dimension(88, 21));
        bttCreateNew.setMargin(new Insets(0, 0, 0, 0));
        bttCancel.setPreferredSize(new Dimension(88, 21));
        bttCancel.setMargin(new Insets(0, 0, 0, 0));
        bttOK.setPreferredSize(new Dimension(88, 21));
        pCenter.setBorder(BorderFactory.createEmptyBorder(0, 24, 0, 24));
        lbView.setPreferredSize(new Dimension(190, 50));
        pButtons.setBorder(BorderFactory.createEmptyBorder(0, 0, 0, -7));
        pResult.setBorder(BorderFactory.createEmptyBorder(12, 0, 12, 0));
        pAdjust.setPreferredSize(new Dimension(10, 15));
        this.getContentPane().add(pCenter, BorderLayout.CENTER);
        pCenter.add(spTree, BorderLayout.CENTER);
        pCenter.add(pSouth, BorderLayout.SOUTH);
        pResult.add(txtResult, BorderLayout.CENTER);
        pResult.add(lbFolder, BorderLayout.WEST);
        pSouth.add(pButtons, BorderLayout.CENTER);
        pSouth.add(pResult, BorderLayout.NORTH);
        pButtons.add(bttOK, null);
        pButtons.add(bttCancel, null);
        pButtons.add(bttCreateNew, null);
        pCenter.add(lbView, BorderLayout.NORTH);
        this.getContentPane().add(pAdjust, BorderLayout.SOUTH);
        this.setSize(400, 305);
        this.setResizable(false);
        this.txtResult.setEditable(false);
        this.txtResult.setBackground(Color.white);

        fileTree.addTreeSelectionListener(this);
        getRootPane().registerKeyboardAction(this,
                                             KeyStroke.getKeyStroke(KeyEvent.
            VK_ESCAPE,
            0), JComponent.WHEN_IN_FOCUSED_WINDOW);
        bttOK.setEnabled(false);
        bttCreateNew.setEnabled(false);
        bttOK.addActionListener(this);
        bttCancel.addActionListener(this);
        bttCreateNew.addActionListener(this);

        pAdjust.setBorder(BorderFactory.createEmptyBorder(2, 2, 2, 2));
        pAdjust.addMouseListener(adjustWindowListener);
        pAdjust.addMouseMotionListener(adjustWindowListener);
    }

    public void setMsg(String msg){
        this.lbView.setText(msg);
    }

    public String getMsg(){
        return this.lbView.getText();
    }

    public void valueChanged(TreeSelectionEvent event){
        File f = fileTree.getSelectFile();
        boolean enabled = (f != null);
        bttOK.setEnabled(enabled);
        enabled = enabled && JFileTree.fileSystemView.isFileSystem(f);
        if(f != null && JFileTree.fileSystemView.isDrive(f)){
            enabled = enabled && f.canWrite();
        }
        bttCreateNew.setEnabled(enabled);
        if(f != null){
            txtResult.setText(JFileTree.fileSystemView.getSystemDisplayName(f));
            //fileTree.setEditable(f.renameTo(f));
        }
    }

    public JFileTree getFileTree(){
        return this.fileTree;
    }

    public void setFileTree(JFileTree tree){
        if(tree == null || tree == this.fileTree){
            return;
        }
        this.spTree.getViewport().setView(tree);
        this.spTree.doLayout();
    }

    public File getSelectFile(){
        if(hasCancel)
            return null;
        return fileTree.getSelectFile();
    }

    public void setSelectFile(File f) throws Exception{
        fileTree.setSelectFile(f);
    }

    public void actionPerformed(ActionEvent actionEvent){
        Object obj = actionEvent.getSource();
        if(obj == bttCreateNew){
            String dirName = JOptionPane.showInputDialog(this, "", "new");
            if(dirName == null || dirName.trim().length() == 0){
                return;
            }
            File f = fileTree.getSelectFile();
            f = new File(f.getAbsolutePath() + File.separator + dirName);
            if(f.mkdir()){
                fileTree.getSelectFileNode().removeAllChildren();
                fileTree.getSelectFileNode().setExplored(false);
                fileTree.getSelectFileNode().explore();
            } else{
                JOptionPane.showMessageDialog(this, "Error", "Error",
                                              JOptionPane.ERROR_MESSAGE);
            }
        } else{
            if(obj == bttOK){
                hasCancel = false;
            }
            this.setVisible(false);
        }
    }

    public static void main(String[] args){
        JFrame frame = OpenSwingUtil.createDemoFrame("JDirChooser Demo");
        JTextArea txt = new JTextArea();
        frame.getContentPane().add(new JScrollPane(txt), BorderLayout.CENTER);
        frame.setVisible(true);
        File f = JDirChooser.showDialog(frame, "Please Select directory", true, null,
                                        "Please Select");
        if(f != null){
            txt.setText(f.getAbsolutePath());
        }
    }
}