| /** |
| * 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; |
| } |
| |
| } |