| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* This program is free software; you can redistribute it and/or

* modify it under the terms of the GNU General Public License

* as published by the Free Software Foundation; either version 2

* of the License, or (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

* MA  02110-1301, USA.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2010 {Prodevelop}   {Task}

*/

package org.gvsig.installer.swing.impl.creation.model;

import java.io.File;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JTree;

import javax.swing.tree.TreeSelectionModel;

/**

 * @author <a href="mailto:jpiera@gvsig.org">Jorge Piera Llodr&aacute;</a>

 */

public class SelectFilesTree extends JTree{

        private File applicationDirectory = null;

        public SelectFilesTree() {

                super();

        }

        public void setApplicationDirectory(File applicationDirectory){

                this.applicationDirectory = applicationDirectory;

                setModel(new SelectFilesTreeModel(applicationDirectory));

                setCellRenderer(new SelectFilesTreeCellRenderer());

                getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);

                addMouseListener(new SelectFilesTreeNodeSelectionListener(this));

        }

        public List<File> getSelectedFiles(){

                List<File> selectedFiles = new ArrayList<File>();

                addSelectedFiles((SelectFilesTreeCheckNode)getModel().getRoot(), selectedFiles);

                return selectedFiles;

        }

        public void addSelectedFiles(SelectFilesTreeCheckNode node, List<File> selectedFiles){

                if (node.isSelected){

                        String absolutePath = node.getFile().getAbsolutePath();

                        String relativePath = absolutePath.substring(applicationDirectory.getAbsolutePath().length(), absolutePath.length());

                        selectedFiles.add(new File(relativePath));

                }else{

                        if (node.getFile().isDirectory()){

                                for (int i=0 ; i< node.getChildCount() ; i++){

                                        addSelectedFiles((SelectFilesTreeCheckNode)node.getChildAt(i), selectedFiles);

                                }

                        }

                }

        }

}