org.eclipse.xtext.xbase.util.XbaseUsageCrossReferencer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.eclipse.xtend.standalone Show documentation

There is a newer version: 2.4.3

Show newest version

/*******************************************************************************
 * Copyright (c) 2011 itemis AG (http://www.itemis.eu) and others.
 * 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
 *******************************************************************************/
package org.eclipse.xtext.xbase.util;

import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;

import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.emf.common.util.TreeIterator;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.util.EcoreUtil.ContentTreeIterator;
import org.eclipse.emf.ecore.util.EcoreUtil.UsageCrossReferencer;
import org.eclipse.xtext.xbase.XAbstractFeatureCall;
import org.eclipse.xtext.xbase.XExpression;

import com.google.common.collect.Iterables;
/**
 * @author Holger Schill - Initial contribution and API
 */
public class XbaseUsageCrossReferencer extends UsageCrossReferencer {

	protected XbaseUsageCrossReferencer(EObject eObject) {
		super(eObject);
	}

	public static Collection find(EObject eObjectOfInterest, EObject eObject) {
		return new XbaseUsageCrossReferencer(eObject).findUsage(eObjectOfInterest);
	}

	private static final long serialVersionUID = 1L;

	@Override
	protected TreeIterator newContentsIterator() {
		return new ContentTreeIterator(emfObjects) {
			private static final long serialVersionUID = 1L;

			@Override
			protected Iterator getEObjectChildren(EObject eObject) {
				if(eObject instanceof XAbstractFeatureCall){
					Iterable result = eObject.eContents();
					XAbstractFeatureCall featureCall = (XAbstractFeatureCall) eObject;
					XExpression implicitReceiver = featureCall.getImplicitReceiver();
					if(implicitReceiver != null) 
						result = Iterables.concat(result, Collections.singleton(implicitReceiver));
					XExpression implicitFirstArgument = featureCall.getImplicitFirstArgument();
					if(implicitFirstArgument != null)
						result = Iterables.concat(result, Collections.singleton(implicitFirstArgument));
					return result.iterator();
				} else 
					return super.getEObjectChildren(eObject);
			}
		};
	}

}