bundles/common/org.eclipse.apogy.common.topology.ui/src/org/eclipse/apogy/common/topology/ui/NodePresentationAdapterFactory.java

/*******************************************************************************
 * Copyright (c) 2018 Agence spatiale canadienne / Canadian Space Agency
 * 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:
 *     Pierre Allard, 
 *     Regent L'Archeveque,
<<<<<<< HEAD
 *     Sebastien Gemme
 * 
=======
 *     Sebastien Gemme  - initial API and implementation
 *     
>>>>>>> refs/heads/eclipse_pa
 * SPDX-License-Identifier: EPL-1.0
 *     
 *******************************************************************************/
package org.eclipse.apogy.common.topology.ui;

import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.SortedSet;
import java.util.TreeSet;

import org.eclipse.apogy.common.topology.ContentNode;

public class NodePresentationAdapterFactory {

	/**
	 * This is a sorted map, the keys are sorted from the most to the less specific.
	 */
	private SortedSet<Class<?>> classes;
	private Map<Class<?>, NodePresentationAdapter> classesToAdapterMap;
	private final List<NodePresentationAdapter> adapters;

	private SortedSet<Class<?>> getClasses() {
		this.classes = new TreeSet<Class<?>>(new ClassComparator());

		// We get the registered adapters.
		Iterator<NodePresentationAdapter> iterator = this.adapters.iterator();

		while (iterator.hasNext()) {
			NodePresentationAdapter adapter = iterator.next();

			getClassesToAdapterMap().put(adapter.getAdaptedClass(), adapter);

			this.classes.add(adapter.getAdaptedClass());

		}

		return this.classes;
	}

	private Map<Class<?>, NodePresentationAdapter> getClassesToAdapterMap() {

		if (this.classesToAdapterMap == null) {

			this.classesToAdapterMap = new HashMap<Class<?>, NodePresentationAdapter>();

		}

		return this.classesToAdapterMap;
	}

	/**
	 * 
	 * @param adapters
	 */
	public NodePresentationAdapterFactory(List<NodePresentationAdapter> adapters) {
		this.adapters = adapters;

	}

	/**
	 * Retrieves the appropriate adapter for node 'node'.
	 * 
	 * @param obj the node we wish to get the adapter.
	 * @return the appropriate adapter for node 'node' or <code>null</code> if none
	 *         is found.
	 */
	public NodePresentationAdapter getAdapterFor(Object obj) {

		NodePresentationAdapter adapter = null;

		// We go through all the classes.
		boolean found = false;

		Iterator<Class<?>> iterator = getClasses().iterator();

		Object nodeContent = null;

		if (obj instanceof ContentNode<?>) {
			ContentNode<?> cNode = (ContentNode<?>) obj;

			nodeContent = cNode.getContent();
		} else {
			nodeContent = obj;
		}

		while (iterator.hasNext() && !found) {
			Class<?> adaptedClass = iterator.next();

			if (adaptedClass.isAssignableFrom(nodeContent.getClass())) {
				found = true;
				adapter = getClassesToAdapterMap().get(adaptedClass);
			}
		}

		return adapter;
	}

}