/*******************************************************************************
* Copyright (c) 2010-2015 Henshin developers. 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:
* TU Berlin, University of Luxembourg, SES S.A.
*******************************************************************************/
package de.tub.tfs.henshin.editor.editparts.rule.graphical;
import org.eclipse.emf.henshin.model.Graph;
import org.eclipse.emf.henshin.model.Mapping;
import org.eclipse.emf.henshin.model.Node;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.requests.CreateConnectionRequest;
import de.tub.tfs.henshin.editor.commands.rule.CreateNodeMappingCommand;
import de.tub.tfs.henshin.editor.editparts.graph.graphical.NodeEditPart;
import de.tub.tfs.henshin.editor.editparts.graph.graphical.NodeGraphicalEditPartPolicy;
import de.tub.tfs.henshin.editor.ui.rule.RuleView;
/**
* A derived {@link NodeGraphicalEditPartPolicy} for {@link NodeEditPart}s
* contained in {@link RuleView rule views} with handling of {@link Mapping node
* mappings}.
*
* @author Johann
*/
public class RuleNodeGraphicalEditPartPolicy extends
NodeGraphicalEditPartPolicy {
/*
* (non-Javadoc)
*
* @seeorg.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#
* getConnectionCompleteCommand
* (org.eclipse.gef.requests.CreateConnectionRequest)
*/
@Override
protected Command getConnectionCompleteCommand(
CreateConnectionRequest request) {
final Command command = request.getStartCommand();
if (command instanceof CreateNodeMappingCommand) {
final CreateNodeMappingCommand mapNodesCommand = (CreateNodeMappingCommand) command;
final Node image = (Node) getHost().getModel();
mapNodesCommand.setImage(image);
final Graph imageGraph = image.getGraph();
// set container
if (imageGraph.isNestedCondition()) {
mapNodesCommand.setContainer(imageGraph.eContainer());
} else if (imageGraph.isRhs()) {
mapNodesCommand.setContainer(imageGraph.getRule());
}
return mapNodesCommand;
}
return super.getConnectionCompleteCommand(request);
}
/*
* (non-Javadoc)
*
* @seeorg.eclipse.gef.editpolicies.GraphicalNodeEditPolicy#
* getConnectionCreateCommand
* (org.eclipse.gef.requests.CreateConnectionRequest)
*/
@Override
protected Command getConnectionCreateCommand(CreateConnectionRequest request) {
final Object newObject = request.getNewObject();
if (newObject instanceof Mapping) {
final Mapping newMapping = (Mapping) newObject;
final Node origin = (Node) getHost().getModel();
CreateNodeMappingCommand command = new CreateNodeMappingCommand(
newMapping, origin);
request.setStartCommand(command);
return command;
}
return super.getConnectionCreateCommand(request);
}
}