core/plugins/org.eclipse.dltk.ui/src corext/org/eclipse/dltk/internal/corext/callhierarchy/CallSearchResultCollector.java

/*******************************************************************************
 * Copyright (c) 2000, 2018 IBM Corporation and others.
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 *******************************************************************************/
package org.eclipse.dltk.internal.corext.callhierarchy;

import java.util.HashMap;
import java.util.Map;

import org.eclipse.dltk.core.IMember;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.IType;

public class CallSearchResultCollector {
	private Map fCalledMembers;

	public CallSearchResultCollector() {
		this.fCalledMembers = createCalledMethodsData();
	}

	public Map getCallers() {
		return fCalledMembers;
	}

	protected void addMember(IModelElement member, IModelElement calledMember,
			int start, int end) {
		addMember(member, calledMember, start, end,
				CallLocation.UNKNOWN_LINE_NUMBER);
	}

	protected void addMember(IModelElement member, IModelElement calledMember,
			int start, int end, int lineNumber) {
		if ((member != null) && (calledMember != null)) {
			if (!isIgnored(calledMember)) {
				MethodCall methodCall = (MethodCall) fCalledMembers
						.get(calledMember.getHandleIdentifier());

				if (methodCall == null) {
					methodCall = new MethodCall(calledMember);
					fCalledMembers.put(calledMember.getHandleIdentifier(),
							methodCall);
				}

				methodCall.addCallLocation(new CallLocation(member,
						calledMember, start, end, lineNumber));
			}
		}
	}

	protected Map createCalledMethodsData() {
		return new HashMap();
	}

	/**
	 * Method isIgnored.
	 *
	 * @param enclosingElement
	 * @return boolean
	 */
	private boolean isIgnored(IModelElement enclosingElement) {
		IType type = getTypeOfElement(enclosingElement);
		if (type != null) {
			String fullyQualifiedName = type.getFullyQualifiedName();

			return CallHierarchy.getDefault().isIgnored(fullyQualifiedName);
		}
		return false;
	}

	private IType getTypeOfElement(IModelElement element) {
		if (element.getElementType() == IModelElement.TYPE) {
			return (IType) element;
		}
		if (element instanceof IMember) {
			return ((IMember) element).getDeclaringType();
		}
		return null;
	}
}