core/plugins/org.eclipse.dltk.ui/src/org/eclipse/dltk/internal/ui/typehierarchy/TraditionalHierarchyViewer.java

/*******************************************************************************
 * Copyright (c) 2000, 2018 IBM Corporation 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.dltk.internal.ui.typehierarchy;

import java.util.Arrays;
import java.util.List;

import org.eclipse.dltk.core.IType;
import org.eclipse.dltk.core.ITypeHierarchy;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.ui.IWorkbenchPart;

/**
 * A TypeHierarchyViewer that looks like the type hierarchy view of VA/Java:
 * Starting form Object down to the element in focus, then all subclasses from
 * this element. Used by the TypeHierarchyViewPart which has to provide a
 * TypeHierarchyLifeCycle on construction (shared type hierarchy)
 */
public class TraditionalHierarchyViewer extends TypeHierarchyViewer {

	public TraditionalHierarchyViewer(Composite parent,
			TypeHierarchyLifeCycle lifeCycle, IWorkbenchPart part,
			IPreferenceStore store) {
		super(parent, new TraditionalHierarchyContentProvider(lifeCycle),
				lifeCycle, part, store);
	}

	@Override
	public String getTitle() {
		if (isMethodFiltering()) {
			return TypeHierarchyMessages.TraditionalHierarchyViewer_filtered_title;
		}
		return TypeHierarchyMessages.TraditionalHierarchyViewer_title;
	}

	@Override
	public void updateContent(boolean expand) {
		getTree().setRedraw(false);
		refresh();

		if (expand) {
			TraditionalHierarchyContentProvider contentProvider = (TraditionalHierarchyContentProvider) getContentProvider();
			int expandLevel = contentProvider.getExpandLevel();
			if (isMethodFiltering()) {
				expandLevel++;
			}
			expandToLevel(expandLevel);
		}
		getTree().setRedraw(true);
	}

	/**
	 * Content provider for the 'traditional' type hierarchy.
	 */
	public static class TraditionalHierarchyContentProvider
			extends TypeHierarchyContentProvider {

		public TraditionalHierarchyContentProvider(
				TypeHierarchyLifeCycle provider) {
			super(provider);
		}

		public int getExpandLevel() {
			ITypeHierarchy hierarchy = getHierarchy();
			if (hierarchy != null) {
				IType input = hierarchy.getType();
				if (input != null) {
					return getDepth(hierarchy, input) + 2;
				}
				return 5;
			}
			return 2;
		}

		private int getDepth(ITypeHierarchy hierarchy, IType input) {
			int max = 0;
			IType[] superType = hierarchy.getSuperclass(input);
			if (superType == null) {
				return 0;
			}
			for (int q = 0; q < superType.length; ++q) {
				int c = getDepth(hierarchy, superType[q]);
				if (c > max)
					max = c;
			}

			return max;
		}

		@Override
		protected final void getRootTypes(List res) {
			ITypeHierarchy hierarchy = getHierarchy();
			if (hierarchy != null) {
				IType input = hierarchy.getType();
				if (input == null) {
					IType[] classes = hierarchy.getRootClasses();
					for (int i = 0; i < classes.length; i++) {
						res.add(classes[i]);
					}
				} else {
					// if (Flags.isInterface(hierarchy.getCachedFlags(input))) {
					// res.add(input);
					// } else if (isAnonymousFromInterface(input)) {
					// res.add(hierarchy.getSuperInterfaces(input)[0]);
					// } else {
					IType[] roots = hierarchy.getRootClasses();
					res.addAll(Arrays.asList(roots)); // something wrong with
														// the hierarchy
					// }
				}
			}
		}

		@Override
		protected final void getTypesInHierarchy(IType type, List res) {
			ITypeHierarchy hierarchy = getHierarchy();
			if (hierarchy != null) {
				IType[] types = hierarchy.getSubtypes(type);
				/*
				 * if (isObject(type)) { for (int i= 0; i < types.length; i++) {
				 * IType curr= types[i]; if (!isAnonymousFromInterface(curr)) {
				 * // no anonymous classes on 'Object' -> will be children of
				 * interface res.add(curr); } } } else
				 */ {
					boolean isHierarchyOnType = (hierarchy.getType() != null);
					boolean isClass = true;// !Flags.isInterface(hierarchy.getCachedFlags(type));
					if (isClass || isHierarchyOnType) {
						for (int i = 0; i < types.length; i++) {
							res.add(types[i]);
						}
					} else {
						// for (int i = 0; i < types.length; i++) {
						// IType curr = types[i];
						// no classes implementing interfaces, only if
						// anonymous
						// if
						// (Flags.isInterface(hierarchy.getCachedFlags(curr))
						// || isAnonymous(curr)) {
						// res.add(curr);
						// }
						// }
					}
				}
			}
		}

		@Override
		protected IType[] getParentType(IType type) {
			ITypeHierarchy hierarchy = getHierarchy();
			if (hierarchy != null) {
				return hierarchy.getSuperclass(type);
				// don't handle interfaces
			}
			return null;
		}

	}
}