/*
Copyright 2008-2011 Gephi
Authors : Mathieu Bastian <mathieu.bastian@gephi.org>, Sébastien Heymann <sebastien.heymann@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.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import javax.swing.Action;
import javax.swing.SwingUtilities;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import javax.swing.tree.TreeSelectionModel;
import org.gephi.desktop.filters.FilterUIModel;
import org.gephi.filters.api.FilterController;
import org.gephi.filters.api.FilterModel;
import org.gephi.filters.api.Query;
import org.openide.explorer.ExplorerManager;
import org.openide.explorer.view.BeanTreeView;
import org.openide.nodes.AbstractNode;
import org.openide.nodes.Children;
import org.openide.nodes.Node;
import org.openide.util.Lookup;
/**
*
* @author Mathieu Bastian
*/
public class QueryExplorer extends BeanTreeView implements PropertyChangeListener, ChangeListener {
private ExplorerManager manager;
private FilterModel model;
private FilterUIModel uiModel;
private FilterController filterController;
//state
private boolean listenSelectedNodes = false;
public QueryExplorer() {
setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
}
public void unsetup() {
if (model != null) {
model.removeChangeListener(this);
model = null;
}
}
public void setup(final ExplorerManager manager, final FilterModel model, FilterUIModel uiModel) {
this.manager = manager;
this.model = model;
this.uiModel = uiModel;
this.filterController = Lookup.getDefault().lookup(FilterController.class);
if (model != null) {
model.addChangeListener(this);
SwingUtilities.invokeLater(new Runnable() {
public void run() {
manager.setRootContext(new RootNode(new QueryChildren(model.getQueries())));
}
});
} else {
SwingUtilities.invokeLater(new Runnable() {
public void run() {
manager.setRootContext(new AbstractNode(Children.LEAF) {
@Override
public Action[] getActions(boolean context) {
return new Action[0];
}
});
}
});
}
updateEnabled(model != null);
if (!listenSelectedNodes) {
manager.addPropertyChangeListener(this);
listenSelectedNodes = true;
}
}
public void propertyChange(PropertyChangeEvent evt) {
if (evt.getPropertyName().equals(ExplorerManager.PROP_SELECTED_NODES)) {
if (uiModel == null) {
return;
}
Node[] nodeArray = (Node[]) evt.getNewValue();
if (nodeArray.length > 0) {
Node node = ((Node[]) evt.getNewValue())[0];
if (node instanceof RootNode) {
uiModel.setSelectedQuery(null);
filterController.setCurrentQuery(null);
return;
}
while (!(node instanceof QueryNode)) {
node = node.getParentNode();
if (node.getParentNode() == null) {
uiModel.setSelectedQuery(null);
filterController.setCurrentQuery(null);
return;
}
}
QueryNode queryNode = (QueryNode) node;
final Query query = queryNode.getQuery();
new Thread(new Runnable() {
public void run() {
uiModel.setSelectedQuery(query);
model.removeChangeListener(QueryExplorer.this);
filterController.setCurrentQuery(uiModel.getSelectedRoot());
model.addChangeListener(QueryExplorer.this);
}
}).start();
}
}
}
public void stateChanged(ChangeEvent e) {
//System.out.println("model updated");
SwingUtilities.invokeLater(new Runnable() {
public void run() {
//uiModel.setSelectedQuery(model.getCurrentQuery());
saveExpandStatus(QueryExplorer.this.manager.getRootContext());
QueryExplorer.this.manager.setRootContext(new RootNode(new QueryChildren(QueryExplorer.this.model.getQueries())));
loadExpandStatus(QueryExplorer.this.manager.getRootContext());
}
});
}
private void updateEnabled(final boolean enabled) {
SwingUtilities.invokeLater(new Runnable() {
public void run() {
setRootVisible(enabled);
setEnabled(enabled);
}
});
}
private void loadExpandStatus(Node node) {
if (node instanceof RootNode) {
RootNode rootNode = (RootNode) node;
for (Node n : rootNode.getChildren().getNodes()) {
loadExpandStatus(n);
}
} else if (node instanceof QueryNode) {
QueryNode queryNode = (QueryNode) node;
if (uiModel.isExpanded(queryNode.getQuery())) {
expandNode(queryNode);
}
Node firstChild = queryNode.getChildren().getNodeAt(0);
if (firstChild != null && firstChild instanceof ParameterNode) {
if (uiModel.isParametersExpanded(queryNode.getQuery())) {
expandNode(firstChild);
}
}
for (Node n : queryNode.getChildren().getNodes()) {
loadExpandStatus(n);
}
}
}
private void saveExpandStatus(Node node) {
if (node instanceof RootNode) {
RootNode rootNode = (RootNode) node;
for (Node n : rootNode.getChildren().getNodes()) {
saveExpandStatus(n);
}
} else if (node instanceof QueryNode) {
QueryNode queryNode = (QueryNode) node;
Node firstChild = queryNode.getChildren().getNodeAt(0);
boolean parameterExpanded = false;
if (firstChild != null && firstChild instanceof ParameterNode) {
parameterExpanded = isExpanded(firstChild);
}
uiModel.setExpand(queryNode.getQuery(), isExpanded(queryNode), parameterExpanded);
for (Node n : queryNode.getChildren().getNodes()) {
saveExpandStatus(n);
}
}
}
}