| /******************************************************************************* |
| * Copyright (c) 2011, 2012 Montages A.G. |
| * 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: |
| * Guillaume Hillairet (Montages A.G.) : initial implementation |
| *******************************************************************************/ |
| package org.eclipse.gmf.internal.bridge.genmodel; |
| |
| import java.net.MalformedURLException; |
| import java.net.URL; |
| |
| import org.eclipse.core.runtime.NullProgressMonitor; |
| import org.eclipse.emf.codegen.ecore.genmodel.GenModel; |
| import org.eclipse.emf.common.util.BasicEList; |
| import org.eclipse.emf.common.util.Diagnostic; |
| import org.eclipse.emf.common.util.EList; |
| import org.eclipse.emf.common.util.URI; |
| import org.eclipse.emf.ecore.EObject; |
| import org.eclipse.emf.ecore.EPackage; |
| import org.eclipse.emf.ecore.EPackage.Registry; |
| import org.eclipse.emf.ecore.resource.Resource; |
| import org.eclipse.emf.ecore.resource.ResourceSet; |
| import org.eclipse.gmf.codegen.gmfgen.GenEditorGenerator; |
| import org.eclipse.gmf.internal.bridge.VisualIdentifierDispenser; |
| import org.eclipse.gmf.mappings.Mapping; |
| import org.eclipse.m2m.internal.qvt.oml.InternalTransformationExecutor; |
| import org.eclipse.m2m.internal.qvt.oml.trace.Trace; |
| import org.eclipse.m2m.qvt.oml.BasicModelExtent; |
| import org.eclipse.m2m.qvt.oml.ExecutionContext; |
| import org.eclipse.m2m.qvt.oml.ExecutionDiagnostic; |
| import org.eclipse.m2m.qvt.oml.ModelExtent; |
| |
| /** |
| * @since 2.0 |
| */ |
| @SuppressWarnings("restriction") |
| public class QVTDiagramGenModelTransformer { |
| |
| private final static String myTransfPath = "platform:/plugin/org.eclipse.gmf.bridge/transforms/Map2Gen.qvto"; |
| private URI myTransfURI; |
| private Trace myTrace; |
| private final ResourceSet myResourceSet; |
| private Registry myRegistry; |
| |
| public QVTDiagramGenModelTransformer(ResourceSet resourceSet, VisualIdentifierDispenser idDespenser) { |
| myResourceSet = resourceSet; |
| VisualIdentifierDispenserFacade.Provider.setDispenser(idDespenser); |
| } |
| |
| public static URL getDefaultTransformation() { |
| try { |
| return new URL(myTransfPath); |
| } catch (MalformedURLException e) { |
| e.printStackTrace(); |
| } |
| return null; |
| } |
| |
| public ExecutionDiagnostic transform(final Mapping m, final GenModel genModel, final ModelExtent output, final ExecutionContext context) { |
| final Resource trace = m.eResource() == null ? myResourceSet.createResource(URI.createURI("trace.qvtotrace")) : myResourceSet.createResource( |
| m.eResource().getURI().trimFileExtension().appendFileExtension("qvtotrace")); |
| |
| final InternalTransformationExecutor executor = new InternalTransformationExecutor(getTransformation()) { |
| @Override |
| protected void handleExecutionTraces(Trace traces) { |
| super.handleExecutionTraces(traces); |
| trace.getContents().add(traces); |
| } |
| }; |
| RuntimeGenModelAccess runtimeAccess = new RuntimeGenModelAccess(); |
| runtimeAccess.ensure(); |
| |
| final ExecutionDiagnostic result = executor.execute(context, |
| getModelExtent(m), |
| getModelExtent(genModel), |
| getModelExtent(runtimeAccess.genPackage().getGenModel()), |
| output); |
| |
| this.myTrace = trace.getContents().isEmpty() ? null : (Trace) trace.getContents().get(0); |
| |
| return result; |
| } |
| |
| public void setRegistry(EPackage.Registry registry) { |
| this.myRegistry = registry; |
| } |
| |
| public ExecutionDiagnostic transform(final Mapping m, final GenModel genModel, ModelExtent output, final ExecutionContext context, final URI... extensions) { |
| ExecutionDiagnostic result = transform(m, genModel, output, context); |
| if (Diagnostic.OK != result.getSeverity()) { |
| return result; |
| } |
| |
| GenEditorGenerator outputGenModel = null; |
| if (output.getContents().size() == 1 && output.getContents().get(0) instanceof GenEditorGenerator) { |
| outputGenModel = (GenEditorGenerator) output.getContents().get(0); |
| } |
| if (outputGenModel == null) { |
| return result; |
| } |
| |
| output = getModelExtent(outputGenModel); |
| for (URI extension: extensions) { |
| final InternalTransformationExecutor exec = myRegistry == null ? |
| new InternalTransformationExecutor(extension) : new InternalTransformationExecutor(extension, myRegistry); |
| exec.loadTransformation(new NullProgressMonitor()); |
| |
| if (1 == exec.getTransformation().getModelParameter().size()) { |
| result = exec.execute(context, output); |
| } else { |
| result = exec.execute(context, getModelExtent(m), getModelExtent(myTrace), output); |
| } |
| } |
| |
| return result; |
| } |
| |
| public Trace getTrace() { |
| return myTrace; |
| } |
| |
| private BasicModelExtent getModelExtent(EObject rootObject) { |
| final EList<EObject> mapObjects = new BasicEList<EObject>(); |
| mapObjects.add(rootObject); |
| return new BasicModelExtent(mapObjects); |
| } |
| |
| public void setTransformationL(URL mainTransformation) { |
| myTransfURI = URI.createURI(mainTransformation.toString()); |
| } |
| |
| public URI getTransformation() { |
| if (myTransfURI == null) { |
| myTransfURI = URI.createURI(myTransfPath); |
| } |
| return myTransfURI; |
| } |
| } |