plugins/org.eclipse.bpmn2.modeler.core/src/org/eclipse/bpmn2/modeler/core/features/ReconnectBaseElementFeature.java

/*******************************************************************************
 * Copyright (c) 2011, 2012 Red Hat, Inc.
 *  All rights reserved.
 * This program is 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:
 * Red Hat, Inc. - initial API and implementation
 *
 * @author Ivar Meikas
 ******************************************************************************/
package org.eclipse.bpmn2.modeler.core.features;

import org.eclipse.bpmn2.BaseElement;
import org.eclipse.bpmn2.DataAssociation;
import org.eclipse.bpmn2.di.BPMNEdge;
import org.eclipse.bpmn2.modeler.core.di.DIUtils;
import org.eclipse.bpmn2.modeler.core.utils.AnchorUtil;
import org.eclipse.bpmn2.modeler.core.utils.BusinessObjectUtil;
import org.eclipse.bpmn2.modeler.core.utils.FeatureSupport;
import org.eclipse.dd.di.DiagramElement;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.graphiti.datatypes.ILocation;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.IReconnectionContext;
import org.eclipse.graphiti.features.context.impl.ReconnectionContext;
import org.eclipse.graphiti.features.impl.DefaultReconnectionFeature;
import org.eclipse.graphiti.mm.pictograms.AnchorContainer;
import org.eclipse.graphiti.mm.pictograms.Connection;
import org.eclipse.graphiti.mm.pictograms.FixPointAnchor;

/**
 * The Class ReconnectBaseElementFeature.
 */
public class ReconnectBaseElementFeature extends DefaultReconnectionFeature {

	/** This flag will be set TRUE if changes were made and need to be committed. */
	protected boolean changesDone = false;
	
	/**
	 * Instantiates a new Reconnect Feature.
	 *
	 * @param fp the FeatureProvider
	 */
	public ReconnectBaseElementFeature(IFeatureProvider fp) {
		super(fp);
	}

	/* (non-Javadoc)
	 * @see org.eclipse.graphiti.features.impl.DefaultReconnectionFeature#canReconnect(org.eclipse.graphiti.features.context.IReconnectionContext)
	 */
	@Override
	public boolean canReconnect(IReconnectionContext context) {
		AnchorContainer sourceContainer = null;
		AnchorContainer targetContainer = null;
		Connection connection = context.getConnection();
		EObject businessObject = BusinessObjectUtil.getBusinessObjectForPictogramElement(context.getConnection());
		if (context.getReconnectType().equals(ReconnectionContext.RECONNECT_TARGET)) {
			sourceContainer = connection.getStart().getParent();
			if (context.getTargetPictogramElement() instanceof AnchorContainer)
				targetContainer = (AnchorContainer) context.getTargetPictogramElement();
		}
		else {
			targetContainer = connection.getEnd().getParent();
			if (context.getTargetPictogramElement() instanceof AnchorContainer)
				sourceContainer = (AnchorContainer) context.getTargetPictogramElement();
		}
		if (sourceContainer!=connection.getStart().getParent() || targetContainer!=connection.getEnd().getParent())
			if (!AbstractBpmn2CreateConnectionFeature.canCreateConnection(sourceContainer, targetContainer, businessObject.eClass(), context.getReconnectType()))
				return false;

		return super.canReconnect(context);
	}

	/* (non-Javadoc)
	 * @see org.eclipse.graphiti.features.impl.DefaultReconnectionFeature#preReconnect(org.eclipse.graphiti.features.context.IReconnectionContext)
	 */
	@Override
	public void preReconnect(IReconnectionContext context) {
		Connection connection = context.getConnection();
		FixPointAnchor newAnchor = null;

		if (!(context.getNewAnchor() instanceof FixPointAnchor)) {
			AnchorContainer source = connection.getStart().getParent();
			AnchorContainer target = connection.getEnd().getParent();
			if (context.getReconnectType().equals(ReconnectionContext.RECONNECT_TARGET)) {
				target = (AnchorContainer) context.getTargetPictogramElement();
				ILocation targetLoc = context.getTargetLocation();
				newAnchor = AnchorUtil.createAnchor(target, targetLoc.getX(), targetLoc.getY());
			}
			else {
				source = (AnchorContainer) context.getTargetPictogramElement();
				ILocation targetLoc = context.getTargetLocation();
				newAnchor = AnchorUtil.createAnchor(source, targetLoc.getX(), targetLoc.getY());
			}
	
			if (newAnchor!=null)
				((ReconnectionContext)context).setNewAnchor(newAnchor);
		}		
		super.preReconnect(context);
	}

	/* (non-Javadoc)
	 * @see org.eclipse.graphiti.features.impl.DefaultReconnectionFeature#postReconnect(org.eclipse.graphiti.features.context.IReconnectionContext)
	 */
	@Override
	public void postReconnect(IReconnectionContext context) {
		super.postReconnect(context);

		BPMNEdge edge = BusinessObjectUtil.getFirstElementOfType(context.getConnection(), BPMNEdge.class);
		DiagramElement de = BusinessObjectUtil.getFirstElementOfType(context.getTargetPictogramElement(), DiagramElement.class);
		if (context.getReconnectType().equals(ReconnectionContext.RECONNECT_TARGET)) {
			edge.setTargetElement(de);
		}
		else {
			edge.setSourceElement(de);
		}
		
		Connection connection = context.getConnection();
		BaseElement flow = BusinessObjectUtil.getFirstElementOfType(connection, BaseElement.class);
		if (!(flow instanceof DataAssociation)) {
			BaseElement be = BusinessObjectUtil.getFirstElementOfType(context.getTargetPictogramElement(), BaseElement.class);
			if (context.getReconnectType().equals(ReconnectionContext.RECONNECT_TARGET)) {
				EStructuralFeature feature = flow.eClass().getEStructuralFeature("targetRef"); //$NON-NLS-1$
				if (feature!=null)
					flow.eSet(feature, be);
			}
			else {
				EStructuralFeature feature = flow.eClass().getEStructuralFeature("sourceRef"); //$NON-NLS-1$
				if (feature!=null && !feature.isMany())
					flow.eSet(feature, be);
			}
		}
		
		FeatureSupport.updateConnection(getFeatureProvider(), connection, true);

		DIUtils.updateDIEdge(context.getConnection());
		
		changesDone = true;
	}

	/* (non-Javadoc)
	 * @see org.eclipse.graphiti.features.impl.AbstractFeature#hasDoneChanges()
	 */
	@Override
	public boolean hasDoneChanges() {
		return changesDone;
	}
}