propertytreeprovider.java

Python Development Environment (Python IDE plugin for Eclipse). Features editor, code completion, re

package org.python.pydev.refactoring.ui.model.generateproperties;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;
import org.python.pydev.refactoring.ast.adapters.IClassDefAdapter;
import org.python.pydev.refactoring.ui.model.tree.ITreeNode;

public class PropertyTreeProvider implements ITreeContentProvider {

	private List<IClassDefAdapter> adapters;

	public PropertyTreeProvider(List<IClassDefAdapter> adapters) {
		this.adapters = adapters;
	}

	public Object[] getChildren(Object parentElement) {
		if (parentElement instanceof ITreeNode) {
			return ((ITreeNode) parentElement).getChildren();
		}
		return null;
	}

	public Object getParent(Object element) {
		return null;
	}

	public boolean hasChildren(Object element) {
		if (element instanceof ITreeNode) {
			ITreeNode node = (ITreeNode) element;
			return node.hasChildren();
		}
		return false;
	}

	public Object[] getElements(Object inputElement) {
		Collection<TreeClassNode> elements = new ArrayList<TreeClassNode>();
		for (IClassDefAdapter elem : adapters) {
			if (elem.hasAttributes()) {
				elements.add(new TreeClassNode(elem));
			}
		}
		return elements.toArray();
	}

	public void dispose() {
	}

	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
	}

}