/*
Copyright 2008-2010 Gephi
Authors : Mathieu Bastian <mathieu.bastian@gephi.org>
Website : http://www.gephi.org
This file is part of Gephi.
Gephi is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
Gephi 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 Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with Gephi. If not, see <http://www.gnu.org/licenses/>.
*/
package org.gephi.desktop.filters.query;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import javax.swing.Action;
import org.gephi.desktop.filters.library.FilterBuilderNode;
import org.gephi.filters.api.FilterController;
import org.gephi.filters.api.Query;
import org.gephi.filters.spi.FilterBuilder;
import org.openide.nodes.AbstractNode;
import org.openide.nodes.Children;
import org.openide.nodes.Node;
import org.openide.util.Lookup;
import org.openide.util.NbBundle;
import org.openide.util.datatransfer.PasteType;
/**
*
* @author Mathieu Bastian
*/
public class QueryChildren extends Children.Array {
private Query query;
private Query[] topQuery;
public QueryChildren(Query query) {
this.query = query;
}
public QueryChildren(Query[] topQuery) { //Only for root node
if (topQuery.length > 0) {
this.topQuery = topQuery;
}
}
@Override
protected Collection<Node> initCollection() {
Collection<Node> nodesChildren = new ArrayList<Node>();
if (query == null && topQuery == null) {
nodesChildren.add(new HelpNode());
} else {
Query[] children = topQuery != null ? topQuery : query.getChildren();
boolean hasParameters = query == null ? false : query.getPropertiesCount() > 0;
int slots = topQuery != null ? topQuery.length : query.getChildrenSlotsCount();
if (slots == Integer.MAX_VALUE) {
slots = children != null ? children.length + 1 : 1;
}
if (hasParameters) {
nodesChildren.add(new ParameterNode(query));
}
for (int i = 0; i < slots; i++) {
if (children != null && i < children.length) {
nodesChildren.add(new QueryNode(children[i]));
} else {
nodesChildren.add(new SlotNode(query));
}
}
}
return nodesChildren;
}
private static class HelpNode extends AbstractNode {
public HelpNode() {
super(Children.LEAF);
setIconBaseWithExtension("org/gephi/desktop/filters/query/resources/drop.png");
}
@Override
public String getHtmlDisplayName() {
return NbBundle.getMessage(QueryChildren.class, "HelpNode.name");
}
@Override
public PasteType getDropType(Transferable t, int action, int index) {
if (t.isDataFlavorSupported(FilterBuilderNode.DATA_FLAVOR)) {
try {
final FilterBuilder fb = (FilterBuilder) t.getTransferData(FilterBuilderNode.DATA_FLAVOR);
return new PasteType() {
@Override
public Transferable paste() throws IOException {
FilterController filterController = Lookup.getDefault().lookup(FilterController.class);
Query f = filterController.createQuery(fb.getFilter());
filterController.add(f);
return null;
}
};
} catch (UnsupportedFlavorException ex) {
ex.printStackTrace();
} catch (IOException ex) {
ex.printStackTrace();
}
}
return null;
}
@Override
public Action[] getActions(boolean context) {
return new Action[0];
}
}
}