plugins/sysml11/requirements/org.eclipse.papyrus.requirements.sysml.testers/src/org/eclipse/papyrus/requirements/sysml/testers/SysMLRequirementsProfileTester.java

/*****************************************************************************
 * Copyright (c) 2016 CEA LIST.
 *
 *
 * 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:
 * 
 * 		Mauricio Alferez (mauricio.alferez@cea.fr) CEA LIST - Initial API and implementation
 *
 *****************************************************************************/
package org.eclipse.papyrus.requirements.sysml.testers;

import org.eclipse.core.expressions.PropertyTester;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.papyrus.sysml.requirements.RequirementsPackage;
import org.eclipse.uml2.uml.Package;
import org.eclipse.uml2.uml.util.UMLUtil;

/**
 * This class helps to test if the selected element applies the SysMLRequirements Profile or not
 */
public class SysMLRequirementsProfileTester extends PropertyTester {

	public static final String propertyToTest = "appliesSysMLRequirementsProfile";//NLS
	
	@Override
	public boolean test(Object receiver, String property, Object[] args, Object expectedValue) {
		if (propertyToTest.equals(property)) {
			boolean testResult = isExpected((IStructuredSelection) receiver);
			
			return expectedValue.equals(testResult);
		}
		return false;
	}
	private boolean isExpected(IStructuredSelection receiver) {
		for (Object current : receiver.toArray()) {
			// Adapt current selection to EObject
			if (current instanceof IAdaptable) {
				EObject eObj = (EObject) ((IAdaptable) current)
						.getAdapter(EObject.class);
				if (eObj instanceof Package){
					final String packageQN = UMLUtil.getProfile(RequirementsPackage.eINSTANCE, eObj).getQualifiedName();
					return (((Package)eObj).getAppliedProfile(packageQN, true) != null);
				}
			}
			return false;
		}
		return false;
	}

}