org.eclipse.gmf.examples.runtime.diagram.logic/src/org/eclipse/gmf/examples/runtime/diagram/logic/internal/editpolicies/TerminalCanonicalEditPolicy.java

/******************************************************************************
 * Copyright (c) 2005, 2006 IBM Corporation and others.
 * 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:
 *    IBM Corporation - initial API and implementation 
 ****************************************************************************/

package org.eclipse.gmf.examples.runtime.diagram.logic.internal.editpolicies;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.gmf.examples.runtime.diagram.logic.internal.editparts.ITerminalOwnerEditPart;
import org.eclipse.gmf.examples.runtime.diagram.logic.internal.l10n.ExampleDiagramLogicMessages;
import org.eclipse.gmf.examples.runtime.diagram.logic.internal.providers.LogicConstants;
import org.eclipse.gmf.examples.runtime.diagram.logic.semantic.Circuit;
import org.eclipse.gmf.examples.runtime.diagram.logic.semantic.Element;
import org.eclipse.gmf.examples.runtime.diagram.logic.semantic.Terminal;
import org.eclipse.gmf.runtime.common.core.command.CompositeCommand;
import org.eclipse.gmf.runtime.common.core.command.ICommand;
import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil;
import org.eclipse.gmf.runtime.diagram.ui.commands.SetBoundsCommand;
import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalEditPolicy;
import org.eclipse.gmf.runtime.diagram.ui.l10n.DiagramUIMessages;
import org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewRequest.ViewDescriptor;
import org.eclipse.gmf.runtime.notation.View;

/**
 * @author qili
 *
 * Sync notation views with semantic children
 */
public class TerminalCanonicalEditPolicy extends CanonicalEditPolicy {
	
	/* (non-Javadoc)
	 * @see org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalEditPolicy#getSemanticChildrenList()
	 */
	protected List getSemanticChildrenList() {
		Element logicElement = getLogicElement();
		
//		making sure that semantic element is not null before proceeding.
		if (logicElement == null)
			return Collections.EMPTY_LIST;
		
		List theElements = new ArrayList();
		
		if (logicElement instanceof Circuit) {
			theElements = logicElement.getInputTerminals();
		} else {
			List theInput = logicElement.getInputTerminals();
			List theOutput = logicElement.getOutputTerminals();
		
			theElements.addAll(theInput);
			theElements.addAll(theOutput);
		}
		return theElements;
	}
	
	/** 
	 * Returns <tt>false</tt> not to delete LogicShapeCompartmentview.
	 * 
	 * @see org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalEditPolicy#shouldDeleteView(org.eclipse.gmf.runtime.notation.View)
	 */
	protected boolean shouldDeleteView(View view) {
		return view.getType().equals(LogicConstants.LOGIC_SHAPE_COMPARTMENT) == false; 
	}
	
	
	/**
	 * @return semantic element for the logic shape
	 */
	private final Element getLogicElement() {
		return (Element)ViewUtil.resolveSemanticElement((View)(this.host().getModel()));
	}
	
		/* (non-Javadoc)
	 * @see org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalEditPolicy#getCreateViewCommand(org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewRequest.ViewDescriptor)
	 */
	protected ICommand getCreateViewCommand(ViewDescriptor descriptor) {
		CompositeCommand cc = new CompositeCommand(DiagramUIMessages.AddCommand_Label);
		
		ICommand viewCommand = super.getCreateViewCommand(descriptor);
		
		IAdaptable adapter = descriptor.getElementAdapter();
		if( adapter == null ) {
			return null;
		}
		Terminal element = (Terminal)adapter.getAdapter(Terminal.class);
	
        TransactionalEditingDomain editingDomain = ((IGraphicalEditPart) getHost()).getEditingDomain();
        
		ICommand boundsCommand = 
			new SetBoundsCommand(editingDomain, 
				ExampleDiagramLogicMessages.SetLocationCommand_Label_Resize,
				descriptor,
				(Point)((ITerminalOwnerEditPart)getHost()).createBoundsMap().get(element.getId()));
		
		cc.compose(viewCommand);
		cc.compose(boundsCommand);
		
		return cc;
	}
}