/*******************************************************************************
* Copyright (c) 2009 Red Hat, Inc.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Red Hat - initial API and implementation
*******************************************************************************/
package org.eclipse.linuxtools.internal.callgraph.treeviewer;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.linuxtools.internal.callgraph.StapData;
import org.eclipse.linuxtools.internal.callgraph.StapGraph;
public class StapTreeContentProvider implements ITreeContentProvider{
private StapGraph graph;
@Override
public Object[] getChildren(Object parentElement) {
List<StapData> empty = new ArrayList<>();
if (parentElement instanceof StapData) {
StapData parent = ((StapData) parentElement);
List<Integer> childrenIDs = parent.collapsedChildren;
for (int val : childrenIDs) {
if (graph.getNodeData(val) != null) {
empty.add(graph.getNodeData(val));
}
}
}
return empty.toArray();
}
@Override
public Object getParent(Object element) {
if (element instanceof StapData) {
return graph.getNodeData(((StapData) element).collapsedParent);
}
return null;
}
@Override
public boolean hasChildren(Object element) {
if (element instanceof StapData) {
return ((StapData) element).children.size() > 0;
}
return false;
}
@Override
public Object[] getElements(Object inputElement) {
return getChildren(inputElement);
}
@Override
public void dispose() {
//Empty
}
@Override
public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
//Empty
}
public void setGraph(StapGraph graph) {
this.graph = graph;
}
}