/***************************************************************************** * Copyright (c) 2009 CEA * * * 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: * Atos Origin - Initial API and implementation * *****************************************************************************/ package org.eclipse.papyrus.uml.diagram.sequence.command; import org.eclipse.core.commands.ExecutionException; import org.eclipse.core.runtime.IAdaptable; import org.eclipse.core.runtime.IProgressMonitor; import org.eclipse.emf.ecore.EClass; import org.eclipse.emf.ecore.EObject; import org.eclipse.gmf.runtime.common.core.command.CommandResult; import org.eclipse.gmf.runtime.common.core.command.ICommand; import org.eclipse.gmf.runtime.emf.type.core.IElementType; import org.eclipse.gmf.runtime.emf.type.core.commands.EditElementCommand; import org.eclipse.gmf.runtime.emf.type.core.requests.ConfigureRequest; import org.eclipse.gmf.runtime.emf.type.core.requests.CreateElementRequest; import org.eclipse.gmf.runtime.notation.View; import org.eclipse.papyrus.uml.diagram.sequence.providers.ElementInitializers; import org.eclipse.uml2.uml.Package; import org.eclipse.uml2.uml.TimeObservation; import org.eclipse.uml2.uml.UMLFactory; /** * Command for creating time observation element * */ public class CustomTimeObservationCreateCommand extends EditElementCommand { private EClass eClass = null; private EObject eObject = null; public CustomTimeObservationCreateCommand(CreateElementRequest req, EObject eObject) { super(req.getLabel(), null, req); this.eObject = eObject; this.eClass = eObject != null ? eObject.eClass() : null; } public static CustomTimeObservationCreateCommand create(CreateElementRequest req, EObject eObject) { return new CustomTimeObservationCreateCommand(req, eObject); } public CustomTimeObservationCreateCommand(CreateElementRequest req) { super(req.getLabel(), null, req); } protected EObject getElementToEdit() { EObject container = ((CreateElementRequest)getRequest()).getContainer(); if(container instanceof View) { container = ((View)container).getElement(); } if(container != null) { return container; } return eObject; } public boolean canExecute() { return true; } protected CommandResult doExecuteWithResult(IProgressMonitor monitor, IAdaptable info) throws ExecutionException { TimeObservation newElement = UMLFactory.eINSTANCE.createTimeObservation(); // get the parent package as owner EObject container = getElementToEdit(); while(container != null && !(container instanceof Package)) { container = container.eContainer(); } if(container == null) { return CommandResult.newCancelledCommandResult(); } Package owner = (Package)container; owner.getPackagedElements().add(newElement); ElementInitializers.getInstance().init_TimeObservation_3020(newElement); doConfigure(newElement, monitor, info); ((CreateElementRequest)getRequest()).setNewElement(newElement); return CommandResult.newOKCommandResult(newElement); } protected void doConfigure(TimeObservation newElement, IProgressMonitor monitor, IAdaptable info) throws ExecutionException { IElementType elementType = ((CreateElementRequest)getRequest()).getElementType(); ConfigureRequest configureRequest = new ConfigureRequest(getEditingDomain(), newElement, elementType); configureRequest.setClientContext(((CreateElementRequest)getRequest()).getClientContext()); configureRequest.addParameters(getRequest().getParameters()); ICommand configureCommand = elementType.getEditCommand(configureRequest); if(configureCommand != null && configureCommand.canExecute()) { configureCommand.execute(monitor, info); } } }