plugins/req/org.eclipse.papyrus.pdp4eng.req.ui/src/org/eclipse/papyrus/pdp4eng/req/ui/view/ReviewsTreeContentProvider.java

/**
 *  Copyright (c)2020 CEA LIST, Committer Name, and others.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License 2.0
 *  which accompanies this distribution, and is available at
 *  https://www.eclipse.org/legal/epl-2.0/
 *
 *  SPDX-License-Identifier: EPL-2.0
 *
 *  Contributors:
 *  Patrick Tessier (CEA LIST) Patrick.tessier@cea.fr   - Initial API and implementation
 *
 */
package org.eclipse.papyrus.pdp4eng.req.ui.view;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.papyrus.infra.emf.utils.EMFHelper;
import org.eclipse.papyrus.pdp4eng.req.profile.constraints.TraceabilityIndexer;
import org.eclipse.uml2.uml.DirectedRelationship;
import org.eclipse.uml2.uml.Element;
import org.eclipse.uml2.uml.util.UMLUtil;

/**
 * This is the content provider to display reviews in the review editor
 *
 */
public class ReviewsTreeContentProvider implements ITreeContentProvider {

	private static Object[] EMPTY = {};
	private ArrayList<Class<?>> stereotypeFilter;

	public ReviewsTreeContentProvider(ArrayList<Class<?>> stereotypeFilter) {
		this.stereotypeFilter = stereotypeFilter;
	}

	@Override
	public void dispose() {

	}

	@Override
	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {

	}

	@Override
	public Object[] getElements(Object inputElement) {

		EObject eobject = EMFHelper.getEObject(inputElement);

		if (eobject instanceof Element) {
			TraceabilityIndexer.getInstance().loadTraceability((Element) eobject);
			ArrayList<Element> result = new ArrayList<>();
			if (TraceabilityIndexer.getInstance().getUpwardTraceabiltiy((Element) eobject) != null) {

				ArrayList<DirectedRelationship> links = TraceabilityIndexer.getInstance().getUpwardTraceabiltiy((Element) eobject);
				for (Iterator<DirectedRelationship> iterator = links.iterator(); iterator.hasNext();) {
					DirectedRelationship directedRelationship = iterator.next();
					result.addAll(directedRelationship.getTargets());
				}
			}
			if (TraceabilityIndexer.getInstance().getParentRequirement((Element) eobject) != null) {
				result.add(TraceabilityIndexer.getInstance().getParentRequirement((Element) eobject));
			}
			// return result.toArray();
			return filter(result).toArray();

		}

		return EMPTY;
	}

	@Override
	public Object[] getChildren(Object parentElement) {
		return computeChildreen(parentElement);
	}

	private Object[] computeChildreen(Object parentElement) {
		ArrayList<Element> result = new ArrayList<>();
		if (parentElement instanceof Element) {
			if (TraceabilityIndexer.getInstance().getUpwardTraceabiltiy((Element) parentElement) != null) {
				ArrayList<DirectedRelationship> links = TraceabilityIndexer.getInstance().getUpwardTraceabiltiy((Element) parentElement);
				for (Iterator<DirectedRelationship> iterator = links.iterator(); iterator.hasNext();) {
					DirectedRelationship directedRelationship = iterator.next();
					result.addAll(directedRelationship.getTargets());

				}
			}
			if (TraceabilityIndexer.getInstance().getParentRequirement((Element) parentElement) != null) {
				result.add(TraceabilityIndexer.getInstance().getParentRequirement((Element) parentElement));
			}
		}
		// return result.toArray();
		return filter(result).toArray();
	}

	@Override
	public Object getParent(Object element) {
		return null;
	}

	@Override
	public boolean hasChildren(Object element) {

		return computeChildreen(element).length > 0;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	protected HashSet<Element> filter(Collection<Element> collection) {
		HashSet<Element> result = new HashSet<>();
		for (Iterator<Element> iterElement = collection.iterator(); iterElement.hasNext();) {
			Element currentElement = iterElement.next();
			for (Iterator<Class<?>> iterator = stereotypeFilter.iterator(); iterator.hasNext();) {
				Class<?> stereoClass = iterator.next();
				if (UMLUtil.getStereotypeApplication(currentElement, (Class) stereoClass) != null) {
					result.add(currentElement);
				}

			}

		}
		return result;
	}

}