lookAndFeel/plugins/org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout/src/org/eclipse/papyrus/uml/diagram/clazz/lf/associationlabellayout/edit/policies/providers/NoMoveEditPolicyProvider.java

/*****************************************************************************
 * Copyright (c) 2019 CEA LIST.
 *
 *
 * 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:
 *  Xavier Le Pallec (for CEA LIST) xlepallec@lilo.org - Bug 558456
 *
 *****************************************************************************/

package org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.policies.providers;

import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gmf.runtime.common.core.service.AbstractProvider;
import org.eclipse.gmf.runtime.common.core.service.IOperation;
import org.eclipse.gmf.runtime.diagram.ui.services.editpolicy.CreateEditPoliciesOperation;
import org.eclipse.gmf.runtime.diagram.ui.services.editpolicy.IEditPolicyProvider;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.parts.SelfLaidOutAssociationEndSourceEditPart;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.parts.SelfLaidOutAssociationEndTargetEditPart;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.parts.SelfLaidOutAssociationMultiplicitySourceEditPart;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.parts.SelfLaidOutAssociationMultiplicityTargetEditPart;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.edit.policies.NoMoveEditPolicy;
import org.eclipse.papyrus.uml.diagram.clazz.lf.associationlabellayout.preferences.Settings;

public class NoMoveEditPolicyProvider extends AbstractProvider implements IEditPolicyProvider {

	@Override
	public boolean provides(IOperation operation) {
		if (Settings.getInstance().isActivated() && operation instanceof CreateEditPoliciesOperation) {
			CreateEditPoliciesOperation creationOperation = CreateEditPoliciesOperation.class.cast(operation);
			EditPart editPart = creationOperation.getEditPart();
			if (editPart instanceof SelfLaidOutAssociationEndTargetEditPart ||
					editPart instanceof SelfLaidOutAssociationEndSourceEditPart ||
					editPart instanceof SelfLaidOutAssociationMultiplicitySourceEditPart ||
					editPart instanceof SelfLaidOutAssociationMultiplicityTargetEditPart) {
				return true;
			}
		}
		return false;
	}

	@Override
	public void createEditPolicies(EditPart editPart) {
		NoMoveEditPolicy editPolicy = new NoMoveEditPolicy(editPart);
		editPart.installEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE, editPolicy);

	}

}