001    /*
002     * Copyright (c) 2009 The openGion Project.
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     *     http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
013     * either express or implied. See the License for the specific language
014     * governing permissions and limitations under the License.
015     */
016    package org.opengion.hayabusa.io;
017    
018    import javax.swing.tree.TreeModel;
019    import javax.swing.tree.DefaultTreeModel;
020    import javax.swing.tree.TreeNode;
021    import javax.swing.tree.DefaultMutableTreeNode;
022    import java.io.File;
023    
024    /**
025     * 蜀?Κ縺ｫ TreeModel 繧呈戟縺｣縺溘ヵ繧｡繧､繝ｫ髫主ｱ､陦ｨ迴ｾ繧定｡ｨ縺吶け繝ｩ繧ｹ縺ｧ縺吶?
026     * 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ繧呈欠螳壹＠縺ｦ縲√◎繧御ｻ･荳九?繝?ぅ繝ｬ繧ｯ繝医Μ髫主ｱ､繧呈ｧ狗ｯ峨＠縺ｾ縺吶?
027     * 縺薙?繧ｯ繝ｩ繧ｹ縺ｯ?後☆縺ｹ縺ｦ縺ｮ繝??繧ｿ繧定ｵｰ譟ｻ縺励※縺九ｉ縲ゝreeModel 繧呈ｧ狗ｯ峨＠縺ｾ縺吶?縺ｧ縲?
028     * 繝代ヵ繧ｩ繝ｼ繝槭Φ繧ｹ逧?↓縺ｯ蜉｣繧翫∪縺吶?
029     *
030     * @og.group 縺昴?莉門?蜉?
031     *
032     * @version  4.0
033     * @author   Kazuhiko Hasegawa
034     * @since    JDK5.0,
035     */
036    public class FileTreeModel {
037            TreeModel model = null;
038            String    dir   = null;
039    
040            /**
041             * 繝?ヵ繧ｩ繝ｫ繝医さ繝ｳ繧ｹ繝医Λ繧ｯ繧ｿ繝ｼ
042             *
043             */
044            public FileTreeModel() {
045                    initialise( null );
046            }
047    
048            /**
049             * 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ繧呈欠螳壹＠縺ｦ縲ゝreeModel 繧呈ｧ狗ｯ峨☆繧九さ繝ｳ繧ｹ繝医Λ繧ｯ繧ｿ繝ｼ
050             *
051             * @param   dir 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ譁?ｭ怜?
052             */
053            public FileTreeModel( final String dir ) {
054                    initialise( dir );
055            }
056    
057            /**
058             * 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ繧呈欠螳壹＠縺ｦ縲ゝreeModel 繧呈ｧ狗ｯ峨＠縺ｾ縺吶?
059             *
060             * @param   dir 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ譁?ｭ怜?
061             */
062            public void setDirectory( final String dir ) {
063                    initialise( dir );
064            }
065    
066            /**
067             * 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ繧呈欠螳壹＠縺ｦ縲ゝreeModel 繧呈ｧ狗ｯ峨＠縺ｾ縺吶?
068             *
069             * @param   dir 繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ譁?ｭ怜?
070             */
071            private void initialise( final String dir ) {
072                    this.dir = dir;
073                    if( this.dir == null ) { this.dir = "."; }
074                    TreeNode root = makeTree( new File(this.dir) );
075                    model = new DefaultTreeModel( root );
076            }
077    
078            /**
079             * TreeModel 繧貞叙蠕励＠縺ｾ縺吶?
080             * 繧ｳ繝ｳ繧ｹ繝医Λ繧ｯ繧ｿ繝ｼ 縺ｾ縺溘?縲《etDirectory()繝｡繧ｽ繝?ラ縺ｫ繧医▲縺ｦ讒狗ｯ峨＆繧後◆
081             * 繝?ぅ繝ｬ繧ｯ繝医Μ髫主ｱ､繧?TreeModel 縺ｫ繝槭ャ繝斐Φ繧ｰ縺励※霑斐＠縺ｾ縺吶?
082             *
083             * @return      繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ譁?ｭ怜?
084             */
085            public TreeModel getTreeModel() {
086                    return model;
087            }
088    
089            /**
090             * 蜀?Κ逧?↓ 繝?ぅ繝ｬ繧ｯ繝医Μ髫主ｱ､繧定｡ｨ迴ｾ縺励◆ TreeNode 繧定ｿ斐＠縺ｾ縺吶?
091             *
092             * @param       file    繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ縺ｮ繝輔ぃ繧､繝ｫ繧ｪ繝悶ず繧ｧ繧ｯ繝?
093             *
094             * @return      繝?ぅ繝ｬ繧ｯ繝医Μ髫主ｱ､繧定｡ｨ迴ｾ縺励◆TreeNode
095             */
096            private DefaultMutableTreeNode makeTree( final File file ) {
097                    DefaultMutableTreeNode node = new DefaultMutableTreeNode( file.getName() );
098                    if(file.isDirectory()) {
099                            String[] list = file.list();
100                            for( int i=0; i<list.length; i++ ) {
101                                    node.add( makeTree( new File(file, list[i]) ) );
102                            }
103                    }
104                    return node;
105            }
106    
107            /**
108             * Tree 縺ｮ陦ｨ遉ｺ逕ｨ繝｡繧ｽ繝?ラ
109             *
110             * 縺薙ｌ縺ｯ縲√ユ繧ｹ繝育畑縺ｫ菴ｿ逕ｨ縺吶ｋ縺溘ａ縺ｮ Tree 繧呈ｨ呎ｺ門?蜉帙↓ 蜃ｺ蜉帙☆繧九Γ繧ｽ繝?ラ縺ｧ縺吶?
111             *
112             * @param    root       繝医ャ繝励Ξ繝吶Ν縺ｮTreeNode繧ｪ繝悶ず繧ｧ繧ｯ繝?髫主ｱ､逧?↓蜊ｰ蟄励＠縺ｾ縺吶?)
113             * @param    model      TreeNode繧貞性繧?TreeModel繧ｪ繝悶ず繧ｧ繧ｯ繝?
114             * @param    level      髫主ｱ､繝ｬ繝吶Ν縲ゆｸ?分繝医ャ繝励ｒ 0 繝ｬ繝吶Ν縺ｨ縺吶ｋ縲?
115             */
116            public void printTree( final TreeNode root,final TreeModel model,final int level ) {
117                    int num = model.getChildCount( root );
118                    TreeNode[] nodes = new TreeNode[num];
119                    for( int i=0; i<num; i++ ) {
120                            nodes[i] = (TreeNode)model.getChild( root,i );
121                            if( nodes[i].isLeaf() ) {
122                                    System.out.println( level + ":" + nodes[i].toString() );
123                            }
124                            else {
125                                    System.out.println( level + ":" + nodes[i].toString() );
126                                    printTree( nodes[i],model,level+1 );
127                            }
128                    }
129            }
130    
131            /**
132             * main 繝｡繧ｽ繝?ラ
133             *
134             * 縺薙ｌ縺ｯ縲√ユ繧ｹ繝育畑縺ｫ菴ｿ逕ｨ縺吶ｋ縺溘ａ縺ｮ main 繝｡繧ｽ繝?ラ縺ｧ縺吶?
135             *
136             * @param  args  襍ｷ蜍墓凾縺ｮ蠑墓焚 args[0] 縺ｫ繝ｫ繝ｼ繝医ョ繧｣繝ｬ繧ｯ繝医Μ蜷?
137             */
138            public static void main( final String[] args ) {
139                    FileTreeModel fmodel = new FileTreeModel( args[0] );
140                    TreeModel model = fmodel.getTreeModel();
141                    TreeNode root = (TreeNode)model.getRoot() ;
142                    fmodel.printTree( root,model,0 );
143            }
144    }