plugins/extension/elementgroup/org.eclipse.papyrus.sysml16.elementgroup.ui/src/org/eclipse/papyrus/sysml16/elementgroup/ui/queries/element/DisplayElementGroupNameQuery.java

/*****************************************************************************
 *  Copyright (c) 2021 CEA LIST, 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.sysml16.elementgroup.ui.queries.element;

import org.eclipse.emf.edit.provider.IItemLabelProvider;
import org.eclipse.papyrus.emf.facet.efacet.core.IFacetManager;
import org.eclipse.papyrus.emf.facet.efacet.core.exception.DerivedTypedElementException;
import org.eclipse.papyrus.emf.facet.query.java.core.IJavaQuery2;
import org.eclipse.papyrus.emf.facet.query.java.core.IParameterValueList2;
import org.eclipse.papyrus.sysml16.modelelements.ElementGroup;
import org.eclipse.papyrus.uml.tools.providers.DelegatingItemLabelProvider;
import org.eclipse.uml2.uml.Element;
import org.eclipse.uml2.uml.util.UMLUtil;

public class DisplayElementGroupNameQuery implements IJavaQuery2<Element, String> {

	private static final IItemLabelProvider labelProvider = new DelegatingItemLabelProvider();

	public String evaluate(final Element context,
			final IParameterValueList2 parameterValues,
			final IFacetManager facetManager)
			throws DerivedTypedElementException {

		if (parameterValues.getParameterValueByName("eReference") != null) { //$NON-NLS-1$
			if (parameterValues.getParameterValueByName("eReference").getValue() != null) { //$NON-NLS-1$
				return labelProvider.getText(parameterValues.getParameterValueByName("eReference").getValue()); //$NON-NLS-1$
			}
		}
		ElementGroup elementGroup = UMLUtil.getStereotypeApplication(context, ElementGroup.class);
		if (elementGroup != null) {
			if (elementGroup.getName() != null) {
				return elementGroup.getName();
			}
			return ""; //$NON-NLS-1$
		} else {
			return labelProvider.getText(context);
		}
	}
}