| /******************************************************************************* |
| * <copyright> |
| * |
| * Copyright (c) 2005, 2010 SAP AG. |
| * 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: |
| * SAP AG - initial API, implementation and documentation |
| * |
| * </copyright> |
| * |
| *******************************************************************************/ |
| package org.eclipse.graphiti.testtool.sketch.features; |
| |
| import org.eclipse.graphiti.features.IFeatureProvider; |
| import org.eclipse.graphiti.features.context.IAddContext; |
| import org.eclipse.graphiti.features.impl.AbstractAddShapeFeature; |
| import org.eclipse.graphiti.mm.pictograms.Diagram; |
| import org.eclipse.graphiti.mm.pictograms.PictogramElement; |
| import org.eclipse.graphiti.mm.pictograms.Shape; |
| |
| /** |
| * The Class AddDiagramFeature. |
| */ |
| public class AddDiagramFeature extends AbstractAddShapeFeature { |
| |
| /** |
| * Instantiates a new adds the diagram feature. |
| * |
| * @param fp |
| * the fp |
| */ |
| public AddDiagramFeature(IFeatureProvider fp) { |
| super(fp); |
| } |
| |
| public PictogramElement add(IAddContext context) { |
| if (context.getNewObject() instanceof Shape) { |
| Shape shape = (Shape) context.getNewObject(); |
| context.getTargetContainer().getChildren().add(shape); |
| } |
| return null; |
| } |
| |
| public boolean canAdd(IAddContext context) { |
| if (context.getTargetContainer() != null && context.getNewObject() instanceof Diagram) { |
| return true; |
| } |
| return false; |
| } |
| |
| } |