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

Go to download
/*
 * 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 AbstractAsyncTreeDataProvider
{
    /**
     * 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.asList ( rootFiles );
        this.comparator = new FileTreeNodeComparator ();
        this.filter = new FileTreeNodeFilter ();
    }

    /**
     * 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 = new FileTreeNodeFilter ();
    }

    /**
     * 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.asList ( 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;
    }

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

    @Override
    public void loadChildren ( final FileTreeNode parent, final NodesLoadCallback listener )
    {
        try
        {
            listener.completed ( parent.getFile () == null ? getRootChildren () : getFileChildren ( parent ) );
        }
        catch ( final Exception cause )
        {
            listener.failed ( cause );
        }
    }

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

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

    @Override
    public Filter getChildrenFilter ( final FileTreeNode parent, final List children )
    {
        // We must not filter out given roots
        return parent.getFile () != null ? super.getChildrenFilter ( parent, children ) : null;
    }

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