FileListSelection.java

/*
 * Copyright (C) 2014 by Array Systems Computing Inc. http://www.array.ca
 *
 * 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 3 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, see http://www.gnu.org/licenses/
 */
package org.esa.snap.graphbuilder.rcp.utils;

import com.bc.ceres.swing.selection.AbstractSelection;

import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * File list selection
 */
public class FileListSelection extends AbstractSelection implements Transferable {

    private static final DataFlavor[] flavors = {
            DataFlavor.stringFlavor,
            DataFlavor.javaFileListFlavor
    };

    private final List<File> fileList = new ArrayList<>();

    public FileListSelection(File[] fileList) {
        this.fileList.addAll(Arrays.asList(fileList));
    }

    @Override
    public File getSelectedValue() {
        return fileList.get(0);
    }

    @Override
    public File[] getSelectedValues() {
        return fileList.toArray(new File[fileList.size()]);
    }

    /**
     * Returns an array of flavors in which this <code>Transferable</code>
     * can provide the data. <code>DataFlavor.stringFlavor</code>
     *
     * @return an array of flavors
     */
    public DataFlavor[] getTransferDataFlavors() {
        return flavors;
    }

    public boolean isDataFlavorSupported(final DataFlavor flavor) {
        for (DataFlavor f : flavors) {
            if (flavor.equals(f)) {
                return true;
            }
        }
        return false;
    }

    /**
     * Returns the <code>Transferable</code>'s data in the requested <code>DataFlavor</code> if possible.
     *
     * @param flavor the requested flavor for the data
     * @return the data in the requested flavor, as outlined above
     * @throws java.awt.datatransfer.UnsupportedFlavorException if the requested data flavor not supported
     */
    public Object getTransferData(final DataFlavor flavor)
            throws UnsupportedFlavorException {
        if (flavor.equals(DataFlavor.javaFileListFlavor)) {
            return fileList;
        } else if (flavor.equals(DataFlavor.stringFlavor)) {
            return null;
        } else {
            throw new UnsupportedFlavorException(flavor);
        }
    }
}