blob: f101b3b9cb84bbb3a0269530334e85089cd8b8d6 [file] [log] [blame]
/*******************************************************************************
* Copyright (c) 2012, 2018 Willink Transformations and others.
* 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:
* E.D.Willink - initial API and implementation
*******************************************************************************/
package org.eclipse.qvtd.xtext.qvtbase.tests;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.eclipse.emf.codegen.ecore.genmodel.GenModel;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelPackage;
import org.eclipse.emf.codegen.ecore.genmodel.GenPackage;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EcorePackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.eclipse.ocl.examples.codegen.dynamic.OCL2JavaFileObject;
import org.eclipse.ocl.pivot.PivotTables;
import org.eclipse.ocl.pivot.internal.manager.MetamodelManagerInternal;
import org.eclipse.ocl.pivot.internal.resource.StandaloneProjectMap;
import org.eclipse.ocl.pivot.internal.utilities.PivotUtilInternal;
import org.eclipse.ocl.pivot.messages.StatusCodes;
import org.eclipse.ocl.pivot.resource.ASResource;
import org.eclipse.ocl.pivot.resource.CSResource;
import org.eclipse.ocl.pivot.resource.ProjectManager;
import org.eclipse.ocl.pivot.utilities.ClassUtil;
import org.eclipse.ocl.pivot.utilities.OCL;
import org.eclipse.ocl.pivot.utilities.PivotConstants;
import org.eclipse.ocl.pivot.utilities.PivotUtil;
import org.eclipse.qvtd.codegen.qvti.QVTiCodeGenOptions;
import org.eclipse.qvtd.codegen.qvti.java.QVTiCodeGenerator;
import org.eclipse.qvtd.compiler.AbstractCompilerChain;
import org.eclipse.qvtd.compiler.CompilerChain;
import org.eclipse.qvtd.compiler.CompilerOptions;
import org.eclipse.qvtd.compiler.DefaultCompilerOptions;
import org.eclipse.qvtd.compiler.internal.qvtm2qvts.ScheduleManager;
import org.eclipse.qvtd.compiler.internal.utilities.CompilerUtil;
import org.eclipse.qvtd.pivot.qvtbase.Transformation;
import org.eclipse.qvtd.pivot.qvtbase.utilities.QVTbase;
import org.eclipse.qvtd.pivot.qvtimperative.ImperativeTransformation;
import org.eclipse.qvtd.pivot.qvtimperative.evaluation.BasicQVTiExecutor;
import org.eclipse.qvtd.pivot.qvtimperative.evaluation.QVTiEnvironmentFactory;
import org.eclipse.qvtd.pivot.qvtimperative.evaluation.QVTiIncrementalExecutor;
import org.eclipse.qvtd.pivot.qvtimperative.evaluation.QVTiTransformationExecutor;
import org.eclipse.qvtd.pivot.qvtimperative.utilities.QVTimperative;
import org.eclipse.qvtd.pivot.qvtimperative.utilities.QVTimperativeUtil;
import org.eclipse.qvtd.pivot.qvtschedule.Region;
import org.eclipse.qvtd.pivot.qvtschedule.ScheduleModel;
import org.eclipse.qvtd.pivot.qvtschedule.ScheduledRegion;
import org.eclipse.qvtd.pivot.qvtschedule.utilities.QVTscheduleUtil;
import org.eclipse.qvtd.runtime.evaluation.Transformer;
import org.eclipse.qvtd.xtext.qvtimperativecs.QVTimperativeCSPackage;
import org.eclipse.xtext.resource.XtextResource;
import junit.framework.TestCase;
public abstract class AbstractTestQVT extends QVTimperative
{
// FIXME move following clones to a Util class
public static @NonNull XtextResource as2cs(@NonNull OCL ocl, @NonNull ResourceSet resourceSet, @NonNull ASResource asResource, @NonNull URI outputURI, /*@NonNull*/ String csContentType) throws IOException {
XtextResource xtextResource = ClassUtil.nonNullState((XtextResource) resourceSet.createResource(outputURI, csContentType));
ocl.as2cs(asResource, (CSResource) xtextResource);
LoadTestCase.assertNoResourceErrors("Conversion failed", xtextResource);
//
// CS save
//
URI savedURI = ClassUtil.nonNullState(asResource.getURI());
asResource.setURI(outputURI.trimFileExtension().trimFileExtension().appendFileExtension(PivotConstants.OCL_AS_FILE_EXTENSION));
asResource.save(DefaultCompilerOptions.defaultSavingOptions);
asResource.setURI(savedURI);
try {
LoadTestCase.assertNoDiagnosticErrors("Concrete Syntax validation failed", xtextResource);
xtextResource.save(DefaultCompilerOptions.defaultSavingOptions);
}
catch (Throwable e) {
e.printStackTrace();
URI xmiURI = outputURI.appendFileExtension(".xmi");
Resource xmiResource = resourceSet.createResource(xmiURI);
xmiResource.getContents().addAll(ClassUtil.nullFree(xtextResource.getContents()));
xmiResource.save(DefaultCompilerOptions.defaultSavingOptions);
LoadTestCase.fail(e.toString());
}
return xtextResource;
}
public static @NonNull ASResource loadQVTiAS(@NonNull OCL ocl, @NonNull URI inputURI) {
Resource asResource = ocl.getMetamodelManager().getASResourceSet().getResource(inputURI, true);
assert asResource != null;
// List<String> conversionErrors = new ArrayList<String>();
// RootPackageCS documentCS = Ecore2OCLinEcore.importFromEcore(resourceSet, null, ecoreResource);
// Resource eResource = documentCS.eResource();
LoadTestCase.assertNoResourceErrors("Load failed", asResource);
// Resource xtextResource = resourceSet.createResource(outputURI, OCLinEcoreCSTPackage.eCONTENT_TYPE);
// XtextResource xtextResource = (XtextResource) resourceSet.createResource(outputURI);
// xtextResource.getContents().add(documentCS);
return (ASResource) asResource;
}
/**
* The test bundle in which intermediate and result files are written.
*/
protected final @NonNull URI testBundleURI;
/**
* The source transformation.
*/
protected final @NonNull URI txURI;
/**
* A prefix for output filenames.
*/
protected final @NonNull URI prefixURI;
/**
* The file folder for generated Java sources.
*/
protected final @NonNull URI srcFileURI;
/**
* The file folder for compiled Java classes.
*/
protected final @NonNull URI binFileURI;
/**
* Project names needed on the classpath in addition to the defaults.
*/
private @Nullable List<@NonNull String> additionalProjectNames = null;
private final @NonNull Map<@NonNull Class<? extends Region>, @NonNull Integer> regionClass2count = new HashMap<>();
protected AbstractCompilerChain compilerChain = null;
private BasicQVTiExecutor interpretedExecutor = null;
private QVTiTransformationExecutor generatedExecutor = null;
private Set<@NonNull String> nsURIs = new HashSet<@NonNull String>();
private boolean suppressFailureDiagnosis = false; // FIXME BUG 511028
protected final @NonNull String testProjectName;
private Collection<@NonNull GenPackage> usedGenPackages = null;
private Collection<@NonNull EPackage> loadedEPackages = null;
private @Nullable String copyright = null;
public AbstractTestQVT(@NonNull ProjectManager projectManager, @NonNull String testProjectName, @NonNull URI testBundleURI, @NonNull URI txURI, @NonNull URI prefixURI, @NonNull URI srcFileURI, @NonNull URI binFileURI) {
super(new QVTiEnvironmentFactory(projectManager, null));
assert testBundleURI.isPlatform();
this.testProjectName = testProjectName;
this.testBundleURI = testBundleURI;
this.txURI = txURI;
this.prefixURI = prefixURI;
this.srcFileURI = srcFileURI;
this.binFileURI = binFileURI;
assert srcFileURI.isFile();
assert srcFileURI.hasAbsolutePath();
assert binFileURI.isFile();
assert binFileURI.hasAbsolutePath();
// installEPackages(eInstances);
//
// http://www.eclipse.org/emf/2002/Ecore is referenced by just about any model load
// Ecore.core is referenced from Ecore.genmodel that is used by the CG to coordinate Ecore objects with their Java classes
// therefore suppress diagnostics about confusing usage.
//
URI ecoreURI = URI.createURI(EcorePackage.eNS_URI);
getProjectManager().getPackageDescriptor(ecoreURI).configure(getResourceSet(), StandaloneProjectMap.LoadFirstStrategy.INSTANCE,
StandaloneProjectMap.MapToFirstConflictHandler.INSTANCE);
}
public void addClasspathProjectName(@NonNull String projectName) {
List<@NonNull String> additionalProjectNames2 = additionalProjectNames;
if (additionalProjectNames2 == null) {
additionalProjectNames = additionalProjectNames2 = new ArrayList<>();
}
additionalProjectNames2.add(projectName);
}
public void addRegisteredPackage(@NonNull String ePackageClassName) throws Exception {
Class<?> ePackageClass = Class.forName(ePackageClassName);
Field instanceField = ePackageClass.getField("eINSTANCE");
EPackage ePackage = (EPackage) instanceField.get(null);
EPackage.Registry.INSTANCE.put(ePackage.getNsURI(), ePackage);
}
public @NonNull GenPackage addUsedGenPackage(@NonNull String resourcePath, @Nullable String fragment) {
if (usedGenPackages == null) {
usedGenPackages = new ArrayList<>();
}
URI uri = URI.createPlatformResourceURI(resourcePath, false);
if (fragment != null) {
uri = uri.appendFragment(fragment);
}
GenPackage genPackage = ClassUtil.nonNullState((GenPackage)getResourceSet().getEObject(uri, true));
usedGenPackages.add(genPackage);
return genPackage;
}
public void assertRegionCount(@NonNull Class<? extends Region> regionClass, @NonNull Integer count) {
TestCase.assertEquals("Region " + regionClass.getSimpleName() + " count:", count != 0 ? count : null, regionClass2count.get(regionClass));
}
public @NonNull Class<? extends Transformer> buildTransformation(@NonNull String outputName,
boolean isIncremental, @NonNull String @NonNull... genModelFiles) throws Exception {
CompilerOptions options = createBuildCompilerChainOptions(isIncremental);
return doBuild(txURI, prefixURI, outputName, options, genModelFiles);
}
public @NonNull Class<? extends Transformer> buildTransformation_486938(@NonNull String outputName,
boolean isIncremental, @NonNull String @NonNull... genModelFiles) throws Exception {
CompilerOptions options = createBuildCompilerChainOptions(isIncremental);
options.setOption(CompilerChain.JAVA_STEP, CompilerChain.JAVA_EXTRA_PREFIX_KEY, "cg");
return doBuild(txURI, prefixURI, outputName, options, genModelFiles);
}
// protected void checkOutput(@NonNull Resource outputResource, @NonNull String expectedFilePath, @Nullable ModelNormalizer normalizer) throws IOException, InterruptedException {
// URI referenceModelURI = testSamplesUri.appendSegments(expectedFilePath.split("/"));
// checkOutput(outputResource, referenceModelURI, normalizer);
// }
protected void checkOutput(@NonNull Resource outputResource, @NonNull URI referenceModelURI, @Nullable ModelNormalizer normalizer) throws IOException, InterruptedException {
Resource referenceResource = outputResource.getResourceSet().getResource(referenceModelURI, true);
assert referenceResource != null;
if (normalizer != null) {
normalizer.normalize(referenceResource);
normalizer.normalize(outputResource);
}
LoadTestCase.assertSameModel(referenceResource, outputResource);
}
public @NonNull ImperativeTransformation compileTransformation(@NonNull String outputName) throws Exception {
return doCompile(txURI, prefixURI, outputName, createCompilerChainOptions());
}
protected @NonNull CompilerOptions createBuildCompilerChainOptions(boolean isIncremental) {
DefaultCompilerOptions compilerOptions = createCompilerChainOptions();
compilerOptions.setGenerateClassesOptions(srcFileURI, binFileURI, createClassProjectNames(), isIncremental);
if (generateGenModel()) {
URI genModelURI = prefixURI.appendFileExtension(AbstractCompilerChain.getDefaultExtension(CompilerChain.GENMODEL_STEP));
compilerOptions.setGenerateGenModelOptions(genModelURI, getBasePrefix(), copyright, usedGenPackages);
}
return compilerOptions;
}
/**
* Return a list of project names that need to be on the class path.
*/
protected @NonNull List<@NonNull String> createClassProjectNames() {
List<@NonNull String> classProjectNames = CompilerUtil.createClasspathProjectNameList(testProjectName);
if (additionalProjectNames != null) {
for (@NonNull String projectName : additionalProjectNames) {
classProjectNames.add(0, projectName);
}
}
return classProjectNames;
}
protected abstract @NonNull AbstractCompilerChain createCompilerChain(@NonNull URI txURI, @NonNull URI prefixURI,
@NonNull CompilerOptions options);
protected @NonNull DefaultCompilerOptions createCompilerChainOptions() {
DefaultCompilerOptions compilerOptions = createCompilerOptions();
// if (copyright != null) {
// compilerOptions.setCopyright(copyright);
// }
// compilerOptions.setBasePrefix(getBasePrefix());
// if (usedGenPackages != null) {
// compilerOptions.setUsedGenPackages(usedGenPackages);
// }
return compilerOptions;
}
protected @NonNull DefaultCompilerOptions createCompilerOptions() {
DefaultCompilerOptions compilerOptions = new DefaultCompilerOptions();
compilerOptions.setOption(CompilerChain.DEFAULT_STEP, CompilerChain.DEBUG_KEY, true);
return compilerOptions;
}
public @NonNull Class<? extends Transformer> createGeneratedClass(@NonNull Transformation asTransformation, @NonNull String @NonNull... genModelFiles) throws Exception {
ResourceSet resourceSet = getResourceSet();
resourceSet.getPackageRegistry().put(GenModelPackage.eNS_URI, GenModelPackage.eINSTANCE);
loadGenModels(genModelFiles);
QVTiCodeGenerator cg = new QVTiCodeGenerator(getEnvironmentFactory(), asTransformation);
QVTiCodeGenOptions options = cg.getOptions();
options.setUseNullAnnotations(true);
setPackagePrefixOption(options);
cg.generateClassFile();
cg.saveSourceFile(getJavaSourceURI().toString() + "/");
File explicitClassPath = new File(getJavaClassURI().toString() + "/");
String qualifiedClassName = cg.getQualifiedName();
String javaCodeSource = cg.generateClassFile();
OCL2JavaFileObject.saveClass(ClassUtil.nonNullState(explicitClassPath.toString()), qualifiedClassName, javaCodeSource);
@SuppressWarnings("unchecked")
Class<? extends Transformer> txClass = (Class<? extends Transformer>) OCL2JavaFileObject.loadExplicitClass(explicitClassPath, qualifiedClassName/*, null*/);
if (txClass == null) {
TestCase.fail("Failed to compile transformation");
throw new UnsupportedOperationException();
}
return txClass;
}
public void createGeneratedExecutor(@NonNull Transformation asTransformation, @NonNull String @NonNull... genModelFiles) throws Exception {
Class<? extends Transformer> txClass = createGeneratedClass(asTransformation, genModelFiles);
createGeneratedExecutor(txClass);
}
public QVTiTransformationExecutor createGeneratedExecutor(@NonNull Class<? extends Transformer> txClass) throws Exception {
return generatedExecutor = new QVTiTransformationExecutor(getEnvironmentFactory(), txClass);
}
public @NonNull BasicQVTiExecutor createInterpretedExecutor(@NonNull ImperativeTransformation asTransformation) throws Exception {
interpretedExecutor = new QVTiIncrementalExecutor(getEnvironmentFactory(), asTransformation, QVTiIncrementalExecutor.Mode.LAZY);
return interpretedExecutor;
}
public @Nullable Resource createModel(@NonNull String modelName, @NonNull URI modelURI) {
return interpretedExecutor.createModel(modelName, modelURI, null);
}
@Override
public synchronized void dispose() {
if (loadedEPackages != null) {
for (@NonNull EPackage ePackage : loadedEPackages) {
EPackage.Registry.INSTANCE.remove(ePackage.getNsURI());
}
}
super.dispose();
if (interpretedExecutor != null) {
interpretedExecutor.dispose();
}
if (compilerChain != null) {
compilerChain.dispose();
}
/**
* Remove the eInstances from the EPackage.Registry.INSTANCE so that global registrations from the calling test
* do not confuse subsequent tests that may want to use dynamic models.
*/
for (String nsURI : nsURIs) {
EPackage.Registry.INSTANCE.remove(nsURI);
}
}
protected @NonNull Class<? extends Transformer> doBuild(@NonNull URI txURI, @NonNull URI prefixURI, @NonNull String outputName,
@NonNull CompilerOptions options, @NonNull String @NonNull ... genModelFiles) throws IOException, Exception {
compilerChain = createCompilerChain(txURI, prefixURI, options);
ImperativeTransformation asTransformation = compilerChain.compile(outputName);
URI asURI = asTransformation.eResource().getURI();
if (asURI != null) {
URI serializedURI = asURI.trimFileExtension().appendFileExtension("serialized.qvti");
doSerialize(asURI, serializedURI);
}
return compilerChain.generate(asTransformation, genModelFiles);
}
protected @NonNull ImperativeTransformation doCompile(@NonNull URI txURI, @NonNull URI prefixURI, @NonNull String outputName,
@NonNull CompilerOptions options) throws Exception {
compilerChain = createCompilerChain(txURI, prefixURI, options);
ImperativeTransformation transformation = compilerChain.compile(outputName);
URI txASURI = transformation.eResource().getURI();
if (txASURI != null) {
URI inputURI = txASURI;
URI serializedURI = txASURI.trimFileExtension().appendFileExtension("serialized.qvti");
doSerialize(inputURI, serializedURI);
}
return transformation;
}
protected XtextResource doSerialize(@NonNull URI inputURI, @NonNull URI serializedURI) throws Exception {
ResourceSet resourceSet = new ResourceSetImpl();
//
// Load QVTiAS
//
OCL ocl = QVTbase.newInstance(getTestProjectManager());
ocl.getEnvironmentFactory().setSeverity(PivotTables.STR_Variable_c_c_CompatibleInitialiserType, StatusCodes.Severity.IGNORE);
try {
ASResource asResource = loadQVTiAS(ocl, inputURI);
LoadTestCase.assertNoResourceErrors("Serializing to " + serializedURI, asResource);
LoadTestCase.assertNoUnresolvedProxies("Serializing to " + serializedURI, asResource);
LoadTestCase.assertNoValidationErrors("Serializing to " + serializedURI, asResource);
//
// Pivot to CS
//
XtextResource xtextResource = as2cs(ocl, resourceSet, asResource, serializedURI, QVTimperativeCSPackage.eCONTENT_TYPE);
resourceSet.getResources().clear();
QVTimperative qvti = QVTimperative.newInstance(getTestProjectManager(), null);
try {
ImperativeTransformation asTransformation = QVTimperativeUtil.loadTransformation(qvti.getEnvironmentFactory(), serializedURI, false);
Resource asResource2 = asTransformation.eResource();
assert asResource2 != null;
LoadTestCase.assertNoResourceErrors("Loading " + serializedURI, asResource2);
LoadTestCase.assertNoUnresolvedProxies("Loading " + serializedURI, asResource2);
LoadTestCase.assertNoValidationErrors("Loading " + serializedURI, asResource2);
}
finally {
qvti.dispose();
qvti = null;
}
return xtextResource;
}
finally {
ocl.dispose();
ocl = null;
}
}
public Transformer executeTransformation() throws Exception {
if (interpretedExecutor != null) {
interpretedExecutor.execute();
interpretedExecutor.saveModels(DefaultCompilerOptions.defaultSavingOptions);
return null;
}
else {
Transformer transformer = generatedExecutor.getTransformer();
if (!transformer.run()) {
if (!suppressFailureDiagnosis) { // FIXME BUG 511028
StringBuilder s = new StringBuilder();
transformer.getInvocationManager().diagnoseWorkLists(s);
throw new Exception("Failed to execute" + s.toString());
}
}
return transformer;
}
}
/**
* Return true if this chanin generates the GenModel.
*/
protected boolean generateGenModel() {
return false;
}
protected abstract @NonNull String getBasePrefix();
@Override
public @NonNull QVTiEnvironmentFactory getEnvironmentFactory() {
return super.getEnvironmentFactory();
}
@Deprecated /** @deprecated use known writeable area in caller */
private @NonNull URI getJavaClassURI() {
return testBundleURI.appendSegment("bin");
}
@Deprecated /** @deprecated use known writeable area in caller */
private @NonNull URI getJavaSourceURI() {
return testBundleURI.appendSegment("test-gen");
}
public @NonNull Collection<@NonNull ? extends Object> getRootObjects(@NonNull String modelName) {
if (interpretedExecutor != null) {
return interpretedExecutor.getRootObjects(modelName);
}
else {
return generatedExecutor.getTransformer().getRootObjects(modelName);
}
}
public @NonNull Map<Object, Object> getSaveOptions() {
Map<Object, Object> saveOptions = new HashMap<Object, Object>(DefaultCompilerOptions.defaultSavingOptions);
saveOptions.put(ASResource.OPTION_NORMALIZE_CONTENTS, Boolean.TRUE);
return saveOptions;
}
protected @NonNull String getTestBundleName() {
return testBundleURI.segment(1);
}
protected @NonNull ProjectManager getTestProjectManager() throws Exception {
return ProjectManager.NO_PROJECTS;
}
public void installClassName(@NonNull String className) throws ClassNotFoundException, NoSuchFieldException, IllegalAccessException {
Class<?> middleClass = Class.forName(className);
Field middleField = middleClass.getDeclaredField("eINSTANCE");
EPackage middleEPackage = (EPackage) middleField.get(null);
getResourceSet().getPackageRegistry().put(middleEPackage.getNsURI(), middleEPackage);
}
protected void instrumentRegion(@NonNull ScheduleManager scheduleManager) {
ScheduleModel scheduleModel = scheduleManager.getScheduleModel();
for (@NonNull ScheduledRegion scheduledRegion : QVTscheduleUtil.getOwnedScheduledRegions(scheduleModel)) {
instrumentRegion(scheduledRegion);
}
}
protected void instrumentRegion(@NonNull Region parentRegion) {
Class<? extends @NonNull Region> regionClass = parentRegion.getClass();
Integer count = regionClass2count.get(regionClass);
regionClass2count.put(regionClass, count == null ? 1 : count+1);
for (@NonNull Region childRegion : parentRegion.getCallableChildren()) {
instrumentRegion(childRegion);
}
}
public void loadEPackage(@NonNull Class<?> txClass, @NonNull String qualifiedClassName) throws ClassNotFoundException, IllegalArgumentException, IllegalAccessException, NoSuchFieldException, SecurityException {
Class<?> ePackageClass = txClass.getClassLoader().loadClass(getBasePrefix() + "." + qualifiedClassName);
EPackage ePackage = (EPackage)ePackageClass.getField("eINSTANCE").get(null);
assert ePackage != null;
if (loadedEPackages == null) {
loadedEPackages = new ArrayList<>();
}
loadedEPackages.add(ePackage);
}
/**
* Explicitly install the eInstances that would normally make it into the ProjectMap from extension point registrations.
* Test models are not registered via extension point so we have to do this manually.
*
public void installEPackages(EPackage... eInstances) {
ResourceSetImpl resourceSet = (ResourceSetImpl) getResourceSet();
for (EPackage eInstance : eInstances) {
String nsURI = eInstance.getNsURI();
if (nsURI != null) {
nsURIs.add(nsURI);
}
resourceSet.getURIResourceMap().put(testPackageURI.appendSegment(eInstance.getName()+".ecore"), eInstance.eResource());
}
} */
public void loadEcoreFile(URI fileURI, EPackage ePackage) {
ResourceSet rSet = getResourceSet();
rSet.getPackageRegistry().put(fileURI.toString(), ePackage);
}
protected void loadGenModel(@NonNull URI genModelURI) {
ResourceSet resourceSet = getResourceSet();
MetamodelManagerInternal metamodelManager = getMetamodelManager();
Resource csGenResource = resourceSet.getResource(genModelURI, true);
for (EObject eObject : csGenResource.getContents()) {
if (eObject instanceof GenModel) {
GenModel genModel = (GenModel)eObject;
genModel.reconcile();
metamodelManager.addGenModel(genModel);
}
}
}
protected void loadGenModels(@NonNull String @NonNull... genModelFiles) {
URI primaryGenModelURI = compilerChain.getURI(CompilerChain.GENMODEL_STEP, CompilerChain.URI_KEY);
loadGenModel(primaryGenModelURI);
for (String genModelFile : genModelFiles) {
URI genModelURI = URI.createURI(genModelFile).resolve(testBundleURI);
loadGenModel(genModelURI);
}
}
public @Nullable Resource loadInput(@NonNull String modelName, @NonNull URI modelURI) {
if (interpretedExecutor != null) {
return interpretedExecutor.loadModel(modelName, modelURI);
}
else {
// Resource inputResource = getResourceSet().getResource(modelURI, true);
// ResourceSet resourceSet = environmentFactory.getMetamodelManager().getASResourceSet(); // FIXME get package registrations in exteranl RespurcSet
ResourceSet resourceSet = environmentFactory.getResourceSet(); // FIXME get package registrations in exteranl RespurcSet
PivotUtil.initializeLoadOptionsToSupportSelfReferences(resourceSet);
Resource inputResource = resourceSet.getResource(modelURI, true);
generatedExecutor.getTransformer().addRootObjects(modelName, ClassUtil.nonNullState(inputResource.getContents()));
return inputResource;
}
}
public void removeRegisteredPackage(@NonNull String ePackageClassName, boolean exceptionThrown) throws Exception {
if (exceptionThrown) { // Don't compound an earlier failure
try {
Class<?> ePackageClass = Class.forName(ePackageClassName);
Field eNsURIField = ePackageClass.getField("eNS_URI");
String nsURI = String.valueOf(eNsURIField.get(null));
EPackage.Registry.INSTANCE.remove(nsURI);
} catch (Exception e) {
e.printStackTrace();
}
}
else {
Class<?> ePackageClass = Class.forName(ePackageClassName);
Field eNsURIField = ePackageClass.getField("eNS_URI");
String nsURI = String.valueOf(eNsURIField.get(null));
EPackage.Registry.INSTANCE.remove(nsURI);
}
}
public void removeRegisteredPackage(@NonNull Class<?> txClass, @NonNull String ePackageClassName, boolean exceptionThrown) throws Exception {
if (exceptionThrown) { // Don't compound an earlier failure
try {
Class<?> ePackageClass = txClass.getClassLoader().loadClass(ePackageClassName);
Field eNsURIField = ePackageClass.getField("eNS_URI");
String nsURI = String.valueOf(eNsURIField.get(null));
EPackage.Registry.INSTANCE.remove(nsURI);
} catch (Exception e) {
e.printStackTrace();
}
}
else {
Class<?> ePackageClass = Class.forName(ePackageClassName);
Field eNsURIField = ePackageClass.getField("eNS_URI");
String nsURI = String.valueOf(eNsURIField.get(null));
EPackage.Registry.INSTANCE.remove(nsURI);
}
}
public @NonNull Resource saveOutput(@NonNull String modelName, @NonNull URI modelURI, @Nullable URI expectedURI, @Nullable ModelNormalizer normalizer) throws IOException, InterruptedException {
ResourceSet resourceSet;
if (PivotUtilInternal.isASURI(modelURI)) {
resourceSet = environmentFactory.getMetamodelManager().getASResourceSet(); // Need PivotSave to allocate xmi:ids
}
else {
resourceSet = getResourceSet();
}
Resource outputResource;
if (interpretedExecutor != null) {
outputResource = interpretedExecutor.saveModel(modelName, modelURI, null, getSaveOptions());
}
else {
outputResource = resourceSet.createResource(modelURI);
outputResource.getContents().addAll(generatedExecutor.getTransformer().getRootEObjects(modelName));
outputResource.save(getSaveOptions());
}
assert outputResource != null;
if (expectedURI != null) {
checkOutput(outputResource, expectedURI, normalizer);
}
return outputResource;
}
public void setCopyright(@Nullable String copyright) {
this.copyright = copyright;
}
protected void setPackagePrefixOption(@NonNull QVTiCodeGenOptions options) {}
public void setSuppressFailureDiagnosis(boolean suppressFailureDiagnosis) { // FIXME BUG 511028
this.suppressFailureDiagnosis = suppressFailureDiagnosis;
}
}