bundles/common/org.eclipse.apogy.common.topology/src/org/eclipse/apogy/common/topology/util/TopologyUtils.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,
 *     Sebastien Gemme  - initial API and implementation
 * 
 * SPDX-License-Identifier: EPL-1.0
 *     
 *******************************************************************************/
package org.eclipse.apogy.common.topology.util;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.apogy.common.topology.GroupNode;
import org.eclipse.apogy.common.topology.Node;

public class TopologyUtils {

	private TopologyUtils() {
	}

	@SuppressWarnings("unchecked")
	public static <T> List<T> getChildren(Node node, Class<T> type) {
		ArrayList<T> lstNode = new ArrayList<T>();

		if (node instanceof GroupNode) {
			GroupNode grp = (GroupNode) node;
			Iterator<Node> iterator = grp.getChildren().iterator();
			while (iterator.hasNext()) {
				Node currentNode = iterator.next();
				if (type.isInstance(currentNode)) {
					lstNode.add((T) currentNode);
				}
				if (currentNode instanceof GroupNode)
					getChildren(currentNode, type, lstNode);
			}

		}

		return lstNode;
	}

	@SuppressWarnings("unchecked")
	public static <T> void getChildren(Node node, Class<T> type, ArrayList<T> lstNode) {
		if (node instanceof GroupNode) {
			GroupNode grp = (GroupNode) node;
			Iterator<Node> iterator = grp.getChildren().iterator();
			while (iterator.hasNext()) {
				Node currentNode = iterator.next();
				if (type.isInstance(currentNode)) {
					lstNode.add((T) currentNode);
				}
				if (currentNode instanceof GroupNode)
					getChildren(currentNode, type, lstNode);
			}

		}
	}
}