com.alee.extended.tree.FileTreeDataProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of weblaf-ui Show documentation
WebLaF UI components
There is a newer version: 1.2.14
/*
 * This file is part of WebLookAndFeel library.
 *
 * WebLookAndFeel library 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 3 of the License, or
 * (at your option) any later version.
 *
 * WebLookAndFeel library 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 WebLookAndFeel library.  If not, see .
 */

package com.alee.extended.tree;

import com.alee.utils.CollectionUtils;
import com.alee.utils.FileUtils;
import com.alee.utils.compare.Filter;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

/**
 * Asynchronous data provider for WebFileTree.
 *
 * @author Mikle Garin
 */

public class FileTreeDataProvider extends AbstractTreeDataProvider
{
    /**
     * Tree root files.
     */
    protected List rootFiles;

    /**
     * Constructs file tree data provider with the specified files as root.
     *
     * @param rootFiles tree root files
     */
    public FileTreeDataProvider ( final File... rootFiles )
    {
        super ();
        this.rootFiles = CollectionUtils.copy ( rootFiles );
        this.comparator = new FileTreeNodeComparator ();
        this.filter = WebFileTreeStyle.filter;
    }

    /**
     * Constructs file tree data provider with the specified files as root.
     *
     * @param rootFiles tree root files
     */
    public FileTreeDataProvider ( final List rootFiles )
    {
        super ();
        this.rootFiles = rootFiles;
        this.comparator = new FileTreeNodeComparator ();
        this.filter = WebFileTreeStyle.filter;
    }

    /**
     * Constructs file tree data provider with the specified files as root.
     *
     * @param filter    tree nodes filter
     * @param rootFiles tree root files
     */
    public FileTreeDataProvider ( final Filter filter, final File... rootFiles )
    {
        super ();
        this.rootFiles = CollectionUtils.copy ( rootFiles );
        this.comparator = new FileTreeNodeComparator ();
        this.filter = filter;
    }

    /**
     * Constructs file tree data provider with the specified files as root.
     *
     * @param filter    tree nodes filter
     * @param rootFiles tree root files
     */
    public FileTreeDataProvider ( final Filter filter, final List rootFiles )
    {
        super ();
        this.rootFiles = rootFiles;
        this.comparator = new FileTreeNodeComparator ();
        this.filter = filter;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public FileTreeNode getRoot ()
    {
        return rootFiles.size () == 1 ? new FileTreeNode ( rootFiles.get ( 0 ) ) : new FileTreeNode ( null );
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void loadChilds ( final FileTreeNode parent, final ChildsListener listener )
    {
        try
        {
            listener.childsLoadCompleted ( parent.getFile () == null ? getRootChilds () : getFileChilds ( parent ) );
        }
        catch ( final Throwable cause )
        {
            listener.childsLoadFailed ( cause );
        }
    }

    /**
     * Returns root child nodes.
     *
     * @return root child nodes
     */
    protected List getRootChilds ()
    {
        final List childs = new ArrayList ( rootFiles.size () );
        for ( final File rootFile : rootFiles )
        {
            childs.add ( new FileTreeNode ( rootFile ) );
        }
        return childs;
    }

    /**
     * Returns child nodes for specified node.
     *
     * @param node parent node
     * @return child nodes
     */
    public List getFileChilds ( final FileTreeNode node )
    {
        final File[] childsList = node.getFile ().listFiles ();
        if ( childsList == null || childsList.length == 0 )
        {
            return new ArrayList ( 0 );
        }
        else
        {
            final List childs = new ArrayList ( childsList.length );
            for ( final File f : childsList )
            {
                childs.add ( new FileTreeNode ( f ) );
            }
            return childs;
        }
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public Filter getChildsFilter ( final FileTreeNode node )
    {
        // We must not filter out given roots
        return node.getFile () == null ? null : super.getChildsFilter ( node );
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean isLeaf ( final FileTreeNode node )
    {
        return node.getFile () != null && !FileUtils.isDirectory ( node.getFile () );
    }
}