blob: 21e52792217fdfa8e971f485d016eb840faef9f7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
/*****************************************************************************
* Copyright (c) 2015 CEA LIST.
*
* 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:
* Quentin Le Menez quentin.lemenez@cea.fr - Initial API and implementation
*****************************************************************************/
package org.eclipse.papyrus.uml.templaterepository;
import java.io.IOException;
import java.util.Collections;
import org.eclipse.emf.common.command.CommandStack;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.transaction.RecordingCommand;
import org.eclipse.papyrus.infra.core.resource.ModelSet;
import org.eclipse.papyrus.uml.diagram.wizards.transformation.IGenerator;
//import org.eclipse.papyrus.uml.tools.model.UmlModel;
import org.eclipse.papyrus.uml.tools.model.UmlUtils;
import org.eclipse.uml2.uml.PackageImport;
//import org.eclipse.uml2.uml.UMLFactory;
public class ImportUMLPrimitiveTypesInModel implements IGenerator {
// This class is used when the element is checked inside SelectModelTemplateComposite's templateTableViewer
private ModelSet modelSet;
private org.eclipse.uml2.uml.Package rootElement;
private EObject primitiveTypes;
public void execute() {
// Get the umlModel from the modelSet
// UmlModel umlModel = (UmlModel) modelSet.getModel(UmlModel.MODEL_ID);
// Get its resource
// Resource umlResource = umlModel.getResource();
Resource umlResource = UmlUtils.getUmlModel(modelSet).getResource();
// Select the root element to add the wanted imports
rootElement = (org.eclipse.uml2.uml.Model) umlResource.getContents().get(0);
// Load the needed libraries. Calling them directly with the path renders the resource folder obsolete ?
URI primitiveTypesURI = URI.createURI("pathmap://UML_LIBRARIES/UMLPrimitiveTypes.library.uml").appendFragment("_0");
primitiveTypes = modelSet.getEObject(primitiveTypesURI, true);
// Creates the import packages at the root of the model (elements of type packageImport)
this.getCommandStack(modelSet).execute(new RecordingCommand(modelSet.getTransactionalEditingDomain()) {
@Override
protected void doExecute() {
// PackageImport packageImport = UMLFactory.eINSTANCE.createPackageImport();
// packageImport.setImportedPackage((org.eclipse.uml2.uml.Package) ImportUMLPrimitiveTypesInModel.this.primitiveTypes);
// rootElement.getPackageImports().add(packageImport);
PackageImport importPrimitivePack = rootElement.createPackageImport((org.eclipse.uml2.uml.Package) primitiveTypes);
if (!rootElement.getPackageImports().contains(importPrimitivePack)) {
rootElement.getPackageImports().add(importPrimitivePack);
}
try {
rootElement.eResource().save(Collections.EMPTY_MAP);
} catch (IOException e) {
Activator.log.error(e);
}
}
});
}
/**
* @see org.eclipse.papyrus.uml.diagram.wizards.transformation.IGenerator#setModelSet(org.eclipse.papyrus.uml.diagram.wizards.transformation.ModelSet)
*
* @param modelSet
*/
public void setModelSet(ModelSet modelSet) {
this.modelSet = modelSet;
}
protected final CommandStack getCommandStack(ModelSet modelSet) {
return modelSet.getTransactionalEditingDomain().getCommandStack();
}
public ModelSet getModelSet() {
return this.modelSet;
}
}
|
