org.eclipse.emf.refactor.comrel.diagram/src/comrel/diagram/edit/helpers/ComrelBaseEditHelper.java

/*
 * 
 */
package comrel.diagram.edit.helpers;

import org.eclipse.gmf.runtime.common.core.command.CompositeCommand;
import org.eclipse.gmf.runtime.common.core.command.ICommand;
import org.eclipse.gmf.runtime.emf.type.core.ElementTypeRegistry;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.gmf.runtime.emf.type.core.edithelper.AbstractEditHelper;
import org.eclipse.gmf.runtime.emf.type.core.edithelper.IEditHelperAdvice;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateElementRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateRelationshipRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.DestroyElementRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.DestroyReferenceRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.IEditCommandRequest;

import comrel.CompositeRefactoring;
import comrel.Port;
import comrel.util.ComrelUtil;

/**
 * @generated
 */
public class ComrelBaseEditHelper extends AbstractEditHelper {

	/**
	 * @generated
	 */
	public static final String EDIT_POLICY_COMMAND = "edit policy command"; //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final String CONTEXT_ELEMENT_TYPE = "context element type"; //$NON-NLS-1$

	/**
	 * @generated
	 */
	protected IEditHelperAdvice[] getEditHelperAdvice(IEditCommandRequest req) {
		if (req.getParameter(CONTEXT_ELEMENT_TYPE) instanceof IElementType) {
			return ElementTypeRegistry.getInstance().getEditHelperAdvice(
					(IElementType) req.getParameter(CONTEXT_ELEMENT_TYPE));
		}
		return super.getEditHelperAdvice(req);
	}

	/**
	 * @generated
	 */
	protected ICommand getInsteadCommand(IEditCommandRequest req) {
		ICommand epCommand = (ICommand) req.getParameter(EDIT_POLICY_COMMAND);
		req.setParameter(EDIT_POLICY_COMMAND, null);
		ICommand ehCommand = super.getInsteadCommand(req);
		if (epCommand == null) {
			return ehCommand;
		}
		if (ehCommand == null) {
			return epCommand;
		}
		CompositeCommand command = new CompositeCommand(null);
		command.add(epCommand);
		command.add(ehCommand);
		return command;
	}

	/**
	 * @generated NOT
	 */
	protected ICommand getCreateCommand(CreateElementRequest req) {
		CompositeRefactoring cr = ComrelUtil
				.getContainedCompositeRefactoring(req.getContainer());
		if (cr != null) {
			cr.updateVisualization();
		}
		return null;
	}

	/**
	 * @generated NOT
	 */
	protected ICommand getCreateRelationshipCommand(
			CreateRelationshipRequest req) {
		CompositeRefactoring cr = ComrelUtil
				.getContainedCompositeRefactoring(req.getContainer());
		if (cr != null) {
			cr.updateVisualization();
		}
		return null;
	}

	/**
	 * @generated NOT
	 */
	protected ICommand getDestroyElementCommand(DestroyElementRequest req) {
		if (req.getElementToDestroy() instanceof Port) {
			return EditHelperUtil.getPortDestroy(req);
		}
		CompositeRefactoring cr = ComrelUtil
				.getContainedCompositeRefactoring(req.getContainer());
		if (cr != null) {
			cr.updateVisualization();
		}
		return null;
	}

	/**
	 * @generated NOT
	 */
	protected ICommand getDestroyReferenceCommand(DestroyReferenceRequest req) {
		CompositeRefactoring cr = ComrelUtil
				.getContainedCompositeRefactoring(req.getContainer());
		if (cr != null) {
			cr.updateVisualization();
		}
		return null;
	}

	@Override
	public void initializeDefaultFeature(CreateElementRequest req) {
		CompositeRefactoring cr = ComrelUtil
				.getContainedCompositeRefactoring(req.getContainer());
		if (cr != null) {
			cr.updateVisualization();
		}
		super.initializeDefaultFeature(req);
	}
}