plugins/org.eclipse.epsilon.dt.epackageregistryexplorer/src/org/eclipse/epsilon/dt/epackageregistryexplorer/FeatureViewerContentProvider.java

/*******************************************************************************
 * Copyright (c) 2008 The University of York.
 * 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:
 *     Dimitrios Kolovos - initial API and implementation
******************************************************************************/

package org.eclipse.epsilon.dt.epackageregistryexplorer;

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

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EEnum;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;

public class FeatureViewerContentProvider implements ITreeContentProvider {
	
	protected PackageRegistryExplorerView view;
	
	public FeatureViewerContentProvider(PackageRegistryExplorerView view) {
		this.view = view;
	}
	
	protected Collection<EStructuralFeature> getFeatures(EClass eClass) {
		Collection<EStructuralFeature> features;
		
		if (view.isShowInheritedFeatures()) {
			features = eClass.getEAllStructuralFeatures();
		}
		else {
			features = eClass.getEStructuralFeatures();
		}
		
		if (view.isShowDerivedFeatures()) {
			return features;
		}
		else {
			ArrayList<EStructuralFeature> filtered = new ArrayList<EStructuralFeature>();
			for (EStructuralFeature sf : features) {
				if (!sf.isDerived()) {
					filtered.add(sf);
				}
			}
			return filtered;
		}
		
	}
	
	public Object[] getElements(Object inputElement) {
		if (inputElement instanceof EClass) {
			ArrayList<Object> eStructuralFeatures = new ArrayList<Object>();
			eStructuralFeatures.addAll(getFeatures((EClass) inputElement));
			eStructuralFeatures.addAll(DecoratorSupport.getHooks((EClass)inputElement, view.getEPackages(), view.isShowInheritedFeatures()));
			if (view.isShowOperations()) {
				eStructuralFeatures.addAll(((EClass)inputElement).getEAllOperations());
			}
			return eStructuralFeatures.toArray();
		}
		else if (inputElement instanceof EEnum) {
			return ((EEnum) inputElement).getELiterals().toArray();
		}
		return new Object[0];
	}

	public Object[] getChildren(Object parentElement) {
		// TODO Auto-generated method stub
		return null;
	}

	public Object getParent(Object element) {
		// TODO Auto-generated method stub
		return null;
	}

	public boolean hasChildren(Object element) {
		// TODO Auto-generated method stub
		return false;
	}

	public void dispose() {
		// TODO Auto-generated method stub
		
	}

	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
		// TODO Auto-generated method stub
		
	}
}